// import {helloHelper} from 'helper.js'
// helloHelper2 = require('./helper.js')
// for organizational proposes

const helper = require('./helper')
const sdapi = require('./sdapi')
const exportHelper = require('./export_png')
const outpaint = require('./outpaint')
const psapi = require('./psapi')
const app = window.require('photoshop').app

const { batchPlay } = require('photoshop').action
const { executeAsModal } = require('photoshop').core
const dialog_box = require('./dialog_box')
const {entrypoints} = require('uxp')



async function getUniqueDocumentId () {
  try {
    uniqueDocumentId = await psapi.readUniqueDocumentIdExe()

    console.log(
      'btnLinkCurrentDocument.click():  uniqueDocumentId: ',
      uniqueDocumentId
    )

    // Regular expression to check if string is a valid UUID
    const regexExp =
      /^[0-9a-fA-F]{8}\b-[0-9a-fA-F]{4}\b-[0-9a-fA-F]{4}\b-[0-9a-fA-F]{4}\b-[0-9a-fA-F]{12}$/gi

    // String with valid UUID separated by dash
    // const str = 'a24a6ea4-ce75-4665-a070-57453082c256'

    const isValidId = regexExp.test(uniqueDocumentId) // true
    console.log('isValidId: ', isValidId)
    if (isValidId == false) {
      let uuid = self.crypto.randomUUID()
      console.log(uuid) // for example "36b8f84d-df4e-4d49-b662-bcde71a8764f"
      await psapi.saveUniqueDocumentIdExe(uuid)
      uniqueDocumentId = uuid
    }
  } catch (e) {
    console.warn('warning Document Id may not be valid', e)
  }
  return uniqueDocumentId
}

document
  .getElementById('btnLinkCurrentDocument')
  .addEventListener('click', async () => {
    await getUniqueDocumentId()
  })

  
// attach event listeners for tabs
Array.from(document.querySelectorAll(".sp-tab")).forEach(theTab => {
  theTab.onclick = () => {
    // localStorage.setItem("currentTab", theTab.getAttribute("id"));
    Array.from(document.querySelectorAll(".sp-tab")).forEach(aTab => {
      if (aTab.getAttribute("id") === theTab.getAttribute("id")) {
        aTab.classList.add("selected");
      } else {
        aTab.classList.remove("selected");
      }
    });
    Array.from(document.querySelectorAll(".sp-tab-page")).forEach(tabPage => {
      if (tabPage.getAttribute("id").startsWith(theTab.getAttribute("id"))) {
        tabPage.classList.add("visible");
      } else {
        tabPage.classList.remove("visible");
      }
    });
  }
});


// entrypoints.setup({

//   panels:{
//     vanilla: ()=>{
//       console.log("you are in the vanilla panel")
//     },
//     experimental_1: ()=>{
//       console.log("you are in the experimental_1 panel")
      
//     } 
//   }
// }
//   )
// just a number that shouldn't unique enough that we will use when save files.
// each session will get a number from 1 to 1000000
const random_session_id = Math.floor((Math.random() * 1000000) + 1);

function getSelectedText() // JavaScript
{
//     // Obtain the object reference for the <textarea>
    // const txtarea = document.getElementById("taPrompt");
    const promptTextarea = document.querySelector('#taPrompt')
    console.log("promptTextarea: ", promptTextarea.value)
//     // Obtain the index of the first selected character
    var start = promptTextarea.selectionStart;
    console.log("start: ",start)
//     // Obtain the index of the last selected character
//     var finish = txtarea.selectionEnd;
//     console.log("finish: ",finish)

//     // Obtain the selected text
//     var sel = txtarea.value.substring(start, finish);
//     console.log("selected textarea: ", sel)
    

    // Do something with the selected content
}
// setInterval(getSelectedText,2000)
function getCommentedString(){
  // const text = document.getElementById("taPrompt").value
  // let text = `Visit /*W3Schools 
  // cute, girl, painterly   
  // *\\ any text
  // and prompt`;
  // let text = `cute cat /*by greg 

  // and artgerm 
  
  //  */ and famous artist`
  let text = `Visit /*W3Schools 
  cute, girl, painterly   
  */ any text
  and prompt




 cute cat  /*by greg 

  and artgerm 
  
   */ and famous artist`
  console.log("getCommentedString: text: ",text)
 
  // let pattern = /(\/)(\*)(\s|\S)*\*\\/g;
  let pattern = /(\/)(\*)(\s|\S)*?(\*\/)/g;

  let result = text.match(pattern);
  console.log("getCommentedString: ",result)
}

//duplicate the active layer
async function duplication () {
  try {
    console.log('active layer id: ', app.activeDocument.activeLayers[0].id)
    await executeAsModal(async () => {
      let inner_new_layer = await app.activeDocument.activeLayers[0].duplicate()
      console.log('inner_new_layer id: ', inner_new_layer.id)
    })

    console.log('new active layer id: ', app.activeDocument.activeLayers[0].id)
  } catch (e) {
    console.log('duplication error:', e)
  }
}

function tempDisableElement(element,time){

  element.disabled = true;
  element.style.opacity = "0.65"; 
  element.style.cursor =  "not-allowed";
  setTimeout(function() {
    element.disabled = false;
    element.style.opacity = "1.0"; 
  element.style.cursor =  "default";
  }, time);
}


async function refreshUI(){
  
  await initSamplers()
  await refreshModels()
  await updateVersionUI()
}
async function refreshModels () {
  try{

    g_models = await sdapi.requestGetModels()
    // const models_menu_element = document.getElementById('mModelsMenu')
  // models_menu_element.value = ""
  document.getElementById('mModelsMenu').innerHTML = ''
  
  for (let model of g_models) {
    console.log(model.title)
    const menu_item_element = document.createElement('sp-menu-item')
    menu_item_element.innerHTML = model.title
    document.getElementById('mModelsMenu').appendChild(menu_item_element)
  }
}catch(e){
  console.warn(e)
}
}


async function updateVersionUI(){
  
    try{
      version = await sdapi.getVersionRequest()
    document.getElementById('lVersionNumber').textContent = version
    }
    catch (e){
      console.warn(e)
      document.getElementById('lVersionNumber').textContent = "v0.0.0"
    }
  
}

async function initSamplers () {
 try{

   let sampler_group = document.getElementById('sampler_group')
   sampler_group.innerHTML = ''

  samplers = await sdapi.requestGetSamplers()
  for (sampler of samplers) {
    console.log(sampler)
    // sampler.name
    // <sp-radio class="rbSampler" value="Euler">Euler</sp-radio>
    const rbSampler = document.createElement('sp-radio')
    
    rbSampler.innerHTML = sampler.name
    rbSampler.setAttribute('class', 'rbSampler')
    rbSampler.setAttribute('value', sampler.name)

    sampler_group.appendChild(rbSampler)
    //add click event on radio button for Sampler radio button, so that when a button is clicked it change g_sd_sampler globally
    
    rbSampler.addEventListener('click', evt => {
      g_sd_sampler = evt.target.value
      console.log(`You clicked: ${g_sd_sampler}`)
    })
  }
  document.getElementsByClassName('rbSampler')[0].setAttribute('checked', '')
}catch(e){
  console.warn(e)
}
}


function promptShortcutExample(){

  let prompt_shortcut_example = {
    "game_like": "Unreal Engine, Octane Render, arcane card game ui, hearthstone art style, epic fantasy style art",
    "large_building_1": "castle, huge building, large building",
    "painterly_style_1": "A full portrait of a beautiful post apocalyptic offworld arctic explorer, intricate, elegant, highly detailed, digital painting, artstation, concept art, smooth, sharp focus, illustration",
    "ugly": " ((((ugly)))), (((duplicate))), ((morbid)), ((mutilated)), out of frame, extra fingers, mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), ((ugly)), blurry, ((bad anatomy)), (((bad proportions))), ((extra limbs)), cloned face, (((disfigured))), out of frame, ugly, extra limbs, (bad anatomy), gross proportions, (malformed limbs), ((missing arms)), ((missing legs)), (((extra arms))), (((extra legs))), mutated hands, (fused fingers), (too many fingers), (((long neck)))"
  }
  var JSONInPrettyFormat = JSON.stringify(prompt_shortcut_example, undefined, 7);
  document.getElementById('taPromptShortcut').value = JSONInPrettyFormat
}
//steps to load init_image:
//duplicate the active layer
// duplication()
// create a mask from marquee selection

// export the layer as png
//load the image from disk to panel as <img /> tag
//store the relative path of the image into init_img_path to be load from the python server (serverMain.py)
//

//**********Start: global variables
let prompt_dir_name = ''
let gImage_paths = []
gCurrentImagePath = ''
let g_init_image_name = ''
let numberOfImages = document.querySelector('#tiNumberOfImages').value
let g_sd_mode = 'txt2img'
let g_sd_sampler = 'Euler a'
let g_denoising_strength = 0.7
let g_use_mask_image = false
let g_models = []
let g_model_title = ''
let gWidth = 512
let gHeight = 512
let g_inpainting_fill = 0
let g_last_outpaint_layers = []
let g_last_inpaint_layers = []
let g_last_snap_and_fill_layers = []

let g_metadatas = []
let g_can_request_progress = true
let g_saved_active_layers = []
let g_is_active_layers_stored = false
//********** End: global variables */

//***********Start: init function calls */

refreshModels() // get the models when the plugin loads
displayUpdate()
initSamplers()
updateVersionUI()
promptShortcutExample()
//***********End: init function calls */

//add click event on radio button mode, so that when a button is clicked it change g_sd_mode globally
rbModeElements = document.getElementsByClassName('rbMode')
for (let rbModeElement of rbModeElements) {
  rbModeElement.addEventListener('click', evt => {
    g_sd_mode = evt.target.value
    console.log(`You clicked: ${g_sd_mode}`)
    displayUpdate()
  })
}

rbMaskContentElements = document.getElementsByClassName('rbMaskContent')

for (let rbMaskContentElement of rbMaskContentElements) {
  rbMaskContentElement.addEventListener('click', evt => {
    g_inpainting_fill = evt.target.value

    console.log(`You clicked: ${g_inpainting_fill}`)
    displayUpdate()
  })
}

// //add click event on radio button for Sampler radio button, so that when a button is clicked it change g_sd_sampler globally
// rbSamplerElements = document.getElementsByClassName('rbSampler')
// for (let rbSamplerElement of rbSamplerElements) {
//   rbSamplerElement.addEventListener('click', evt => {
//     g_sd_sampler = evt.target.value
//     console.log(`You clicked: ${g_sd_sampler}`)
//   })
// }
// document.getElementById('chUseMaskImage').addEventListener("click",evt=>{
//   g_use_mask_image = evt.target.checked
//   console.log(`g_use_mask_image:? ${g_use_mask_image}`);

// })

// show the interface that need to be shown and hide the interface that need to be hidden
function displayUpdate () {
  if (g_sd_mode == 'txt2img') {
    document.getElementById('slDenoisingStrength').style.display = 'none' // hide denoising strength slider
    // document.getElementById("image_viewer").style.display = 'none' // hide images
    document.getElementById('init_image_container').style.display = 'none' // hide init image
    document.getElementById('init_image_mask_container').style.display = 'none' // hide init mask
    document.getElementById('slInpainting_fill').style.display = 'none' // hide inpainting fill mode
    
    document.getElementById('btnSnapAndFill').style.display = 'none'//"none" will  misaligned the table // hide snap and fill button
  }

  if (g_sd_mode == 'img2img') {
    document.getElementById('slDenoisingStrength').style.display = 'block' // show denoising strength
    // document.getElementById("image_viewer").style.display = 'block'
    document.getElementById('init_image_container').style.display = 'block' // hide init image

    document.getElementById('init_image_mask_container').style.display = 'none' // hide mask
    document.getElementById('slInpainting_fill').style.display = 'none' // hide inpainting fill mode
    document.getElementById('btnSnapAndFill').style.display = 'inline-flex' // hide snap and fill button mode
  }
  if (g_sd_mode == 'inpaint') {
    ///fix the misalignment problem in the ui (init image is not aligned with init mask when switching from img2img to inpaint ). note: code needs refactoring   
    document.getElementById('btnSnapAndFill').style.display = 'none'//"none" will  misaligned the table // hide snap and fill button
    document.getElementById('tableInitImageContainer').style.display = 'none' // hide the table 
    setTimeout(() => {
      document.getElementById('tableInitImageContainer').style.display = 'table' // show the table after some time so it gets rendered. 
    }, 100);
    

    document.getElementById('slDenoisingStrength').style.display = 'block'
    document.getElementById('init_image_mask_container').style.display = 'block'
    document.getElementById('slInpainting_fill').style.display = 'block'
    document.getElementById('init_image_container').style.display = 'block' // hide init image

    document.getElementById('btnInitOutpaint').style.display = 'inline-flex'
    document.getElementById('btnInitInpaint').style.display = 'inline-flex'
    // document.getElementById('btnInitOutpaint').style.display = 'none'
    // document.getElementById('btnInitInpaint').style.display = 'none'
  } else {//txt2img or img2img
    document.getElementById('btnInitOutpaint').style.display = 'none'
    document.getElementById('btnInitInpaint').style.display = 'none'
  }
}
// function showLayerNames () {
//   const app = window.require('photoshop').app
//   const allLayers = app.activeDocument.layers
//   const allLayerNames = allLayers.map(
//     layer => `${layer.name} (${layer.opacity} %)`
//   )

//   const sortedNames = allLayerNames.sort((a, b) => (a < b ? -1 : a > b ? 1 : 0))
//   document.getElementById('layers').innerHTML = `
//         <ul>${sortedNames.map(name => `<li>${name}</li>`).join('')}</ul>`
// }

async function addNewLayer () {
  document.getElementById('layers').innerHTML = `<span>
    addNewLayer was called
    </span>`
  await require('photoshop').core.executeAsModal(newNormalLayer)
  // app.activeDocument.createLayer({ name: "myLayer", opacity: 80, mode: "colorDodge" });
}

async function newColorDodgeLayer (executionContext) {
  document.getElementById('layers').innerHTML = `<span>
        newColorDogeLayer was called
        </span>`
  await app.activeDocument.createLayer({
    name: 'myLayer',
    opacity: 80,
    mode: 'colorDodge'
  })
}

async function newNormalLayer (executionContext) {
  document.getElementById('layers').innerHTML = `<span>
    Normal was called
    </span>`
  await app.activeDocument.createLayer({
    name: 'myLayer',
    opacity: 100,
    mode: 'normal'
  })
}

function selectTool () {
  var doc = app.activeDocument
  var activeTool = app.currentTool

  // if (activeTool !== toolName) {
  //   toolName = activeTool;
  //   doc.activeTool = toolName;
  // }
  // const util = require('util')

  // console.log(util.inspect(myObject, {showHidden: false, depth: null, colors: true}))
  console.dir(app, { depth: null })
  console.log('hello this is Abdullah')
  document.getElementById('layers').innerHTML = `<span>
    selectTool was called, ${activeTool}
    </span>`

  //rectanglemarquee
  // await require('photoshop').core.executeAsModal(newNormalLayer);
}

async function testServerPath () {
  // const serverPath = "https://api.github.com/users/abdullah"

  try {
    // const serverPath = 'https://api.weather.gov/points/123.4,342.5'
    // const serverPath = 'https://api.github.com/users/abdullah'
    // const serverPath = "https://api.coindesk.com/v1/bpi/currentprice.json"
    // const serverPath = "http://127.0.0.1:3000"
    // const serverPath = "http://localhost:3000"
    const serverPath = 'http://127.0.0.1:8000/txt2img/random%20prompt'
    // const serverPath = "https://3330-37-106-100-102.eu.ngrok.io"
    console.log('testServerPath function was called')

    let response = await fetch(serverPath)
    console.log('testServerPath finished fetch')

    if (!response.ok) {
      throw new Error(
        `HTTP error fetching weather station; status: ${response.status}`
      )
    }
    let stationJson = await response.json()
    console.dir(stationJson)
  } catch (err) {
    console.error('testServerPath error: ' + err.message)
  }
}

// User picks an image file
// open a explorer for user to select a image file
async function fillImage () {
  const storage = require('uxp').storage
  const fs = storage.localFileSystem
  let imageFile = await fs.getFileForOpening({
    types: storage.fileTypes.images
  })

  // Create ImageFill for this image
  const ImageFill = require('scenegraph').ImageFill
  let fill = new ImageFill(imageFile)

  // Set fill of first selected item
  selection.items[0].fill = fill
}
// fillImage()

function pastImage2Layer () {
  const { batchPlay } = require('photoshop').action
  const { executeAsModal } = require('photoshop').core

  executeAsModal(
    () => {
      // batchPlay([command], {})
      const result = batchPlay(
        [
          {
            _obj: 'paste',
            antiAlias: {
              _enum: 'antiAliasType',
              _value: 'antiAliasNone'
            },
            as: {
              _class: 'pixel'
            },
            _options: {
              dialogOptions: 'dontDisplay'
            }
          }
        ],
        {
          synchronousExecution: false,
          modalBehavior: 'fail'
        }
      )
    },
    {
      commandName: 'Create Label'
    }
  )
}
function sliderToResolution (sliderValue) {
  return sliderValue * 64
}

document.querySelector('#slHeight').addEventListener('input', evt => {
  gHeight = sliderToResolution(evt.target.value)
  document.querySelector('#lHeight').textContent = gHeight
})
document.querySelector('#slWidth').addEventListener('input', evt => {
  gWidth = sliderToResolution(evt.target.value)
  document.querySelector('#lWidth').textContent = gWidth
})
document.querySelector('#slInpaintPadding').addEventListener('input', evt => {
  padding = evt.target.value * 4
  document.querySelector('#lInpaintPadding').textContent = padding
})

document
  .querySelector('#slDenoisingStrength')
  .addEventListener('input', evt => {
    const denoising_string_value = evt.target.value / 100.0
    g_denoising_strength = denoising_string_value
    // document.querySelector('#lDenoisingStrength').value= Number(denoising_string_value)
    document.querySelector('#lDenoisingStrength').textContent =
      denoising_string_value
    document.getElementById(
      'lDenoisingStrength'
    ).innerHTML = `${denoising_string_value}`
    // console.log(`New denoising_string_value: ${document.querySelector('#tiDenoisingStrength').value}`)
  })

// document.getElementById('btnPopulate').addEventListener('click', showLayerNames)

document
  .getElementById('btnSnapAndFill')
  .addEventListener('click', async () => {

    const isSelectionAreaValid = await psapi.checkIfSelectionAreaIsActive()
    if(isSelectionAreaValid){
      
      
      // clear the layers related to the last mask operation.
      g_last_snap_and_fill_layers = await psapi.cleanSnapAndFill(g_last_snap_and_fill_layers)
      // create new layers related to the current mask operation.
      g_last_snap_and_fill_layers = await outpaint.snapAndFillExe(random_session_id)
      console.log ("outpaint.snapAndFillExe(random_session_id):, g_last_snap_and_fill_layers: ",g_last_snap_and_fill_layers)
    }else{
      psapi.promptForMarqueeTool()

    }
  })



document
  .getElementById('btnInitOutpaint')
  .addEventListener('click', async () => {
    try{

      const isSelectionAreaValid = await psapi.checkIfSelectionAreaIsActive()
      
      if (isSelectionAreaValid){
        // clear the layers related to the last mask operation.
        g_last_outpaint_layers = await psapi.cleanLayers(g_last_outpaint_layers)
        // create new layers related to the current mask operation.
        
        g_last_outpaint_layers = await outpaint.outpaintFasterExe(random_session_id)
        console.log ("outpaint.outpaintFasterExe(random_session_id):, g_last_outpaint_layers: ",g_last_outpaint_layers)
      }
      else{
        psapi.promptForMarqueeTool()        

        console.log("please use the rectangular marquee tool and select an area")
      }
    }
    catch(e){
      console.warn("selection area is not valid, please use the rectangular marquee tool",e)
    }
  })

document
  .getElementById('btnInitInpaint')
  .addEventListener('click', async () => {
    const isSelectionAreaValid = await psapi.checkIfSelectionAreaIsActive()
    if(isSelectionAreaValid){

    // delete the layers of the previous mask operation
    g_last_inpaint_layers = await psapi.cleanLayers(g_last_inpaint_layers)
    // store the layer of the current mask operation
    g_last_inpaint_layers =  await outpaint.inpaintFasterExe(random_session_id)
    
    console.log ("outpaint.inpaintFasterExe(random_session_id):, g_last_inpaint_layers: ",g_last_inpaint_layers)
  }
  else{
    psapi.promptForMarqueeTool()
  }
  })

function toggleGenerateInterruptButton (defaultVal) {
  if (defaultVal) {

    document.querySelector('#btnGenerate').style.display = 'none' // hide generate button
    document.querySelector('#btnInterrupt').style.display = 'inline-block' // show interrupt button
    g_can_request_progress = true
  } else {
    document.querySelector('#btnGenerate').style.display = 'inline-block' // hide generate button
    document.querySelector('#btnInterrupt').style.display = 'none' // show interrupt button
    g_can_request_progress = false
  }
}

// document.getElementById('btnSelectTool').addEventListener('click', selectTool)

document.getElementById('btnRandomSeed').addEventListener('click', async () => {
  document.querySelector('#tiSeed').value = '-1'
})
document.getElementById('btnLastSeed').addEventListener('click', async () => {
  try{
    console.log("click on Last seed")
    let seed = '-1'

    console.log("g_metadatas.length: ",g_metadatas.length)
    if (g_metadatas.length > 0){
      seed = g_metadatas[0].Seed
    } 
    console.log("seed:",seed)
    document.querySelector('#tiSeed').value = seed
  }
  catch(e){
    console.warn(e)
  }
})

document.getElementById('btnCleanLayers').addEventListener('click', async () => {
  console.log("click on btnCleanLayers,  g_last_outpaint_layers:",g_last_outpaint_layers)
  console.log("click on btnCleanLayers,  g_last_inpaint_layers:",g_last_inpaint_layers)
  
  console.log("click on btnCleanLayers,  g_last_snap_and_fill_layers:",g_last_snap_and_fill_layers)

  
  console.log("g_last_snap_and_fill_layers")
  g_last_snap_and_fill_layers = await psapi.cleanSnapAndFill(g_last_snap_and_fill_layers)

  if (g_last_outpaint_layers.length > 0){
    g_last_outpaint_layers = await psapi.cleanLayers(g_last_outpaint_layers)
    console.log("g_last_outpaint_layers has 1 layers")

  }
  if (g_last_inpaint_layers.length> 0 ){
    g_last_inpaint_layers = await psapi.cleanLayers(g_last_inpaint_layers)

  }
})
document.getElementById('btnInterrupt').addEventListener('click', async () => {
  try{

    // g_can_request_progress = false
    json = await sdapi.requestInterrupt()
    
    toggleGenerateInterruptButton(false)
  }catch(e)
  {
    toggleGenerateInterruptButton(false)
    console.warn(e)
  }
})


//store active layers only if they are not stored.
async function storeActiveLayers () {
  setTimeout(async () => {
    
    const layers = await app.activeDocument.activeLayers
    console.log("storeActiveLayers: ", layers.length)
    
    if(layers.length > 0){
      g_saved_active_layers = layers
      await psapi.unselectActiveLayersExe()
    }
  }, 200);

  // if (g_is_active_layers_stored == false) {
  //   g_saved_active_layers = await app.activeDocument.activeLayers
  //   g_is_active_layers_stored = true
  //   await psapi.unselectActiveLayersExe()
  // } else {
  // }
}
async function restoreActiveLayers () {
  
  const layers = await app.activeDocument.activeLayers
  console.log("restoreActiveLayers: ", layers.length)
  if(layers.length == 0){
    await psapi.selectLayersExe(g_saved_active_layers)
    g_saved_active_layers = []
  }
  // if (g_is_active_layers_stored == true) {
  //   // g_saved_active_layers = await app.activeDocument.activeLayers
  //   await psapi.selectLayersExe(g_saved_active_layers)
  //   g_is_active_layers_stored = false
  //   g_saved_active_layers = []
  // } 
}
document.getElementById('btnSdUrl').addEventListener('click', async () => {
  //change the sdUrl in server in proxy server
  console.log("you clicked btnSdUrl")
  let new_sd_url = document.getElementById('tiSdUrl').value
  console.log("new_sd_url: ", new_sd_url)

  new_sd_url = new_sd_url.trim()
  console.log("new_sd_url.trim(): ", new_sd_url)

  if (new_sd_url.length > 0) {
    await sdapi.changeSdUrl(new_sd_url)
  }
})


document.querySelector('#taPrompt').addEventListener('focus', async () => {
  console.log("taPrompt focus")
  // console.log('we are in prompt textarea')
  // console.log("g_is_active_layers_stored: ",g_is_active_layers_stored)
  await storeActiveLayers()
  // await psapi.unselectActiveLayersExe()

})
document.querySelector('#taPrompt').addEventListener('blur', async () => {
  console.log("taPrompt blur")
  // console.log('we are out of prompt textarea')
  // await psapi.unselectActiveLayersExe()
  // console.log("g_is_active_layers_stored: ",g_is_active_layers_stored)
  await restoreActiveLayers()
})

document
  .querySelector('#taNegativePrompt')
  .addEventListener('focus', async () => {
    console.log("taNegativePrompt focus")
    // console.log('we are in prompt textarea')

    await storeActiveLayers()
    // await psapi.unselectActiveLayersExe()
  })
document
  .querySelector('#taNegativePrompt')
  .addEventListener('blur', async () => {
    console.log("taNegativePrompt blur")
    // console.log('we are out of prompt textarea')
    // await psapi.unselectActiveLayersExe()
    await restoreActiveLayers()
  })


document.getElementById('btnGenerate').addEventListener('click', async () => {
  
  toggleGenerateInterruptButton(true)
  numberOfImages = document.querySelector('#tiNumberOfImages').value
  numberOfSteps = document.querySelector('#tiNumberOfSteps').value

  const prompt = document.querySelector('#taPrompt').value
  const negative_prompt = document.querySelector('#taNegativePrompt').value
  // console.log("prompt:",prompt)
  // console.log("negative_prompt:",negative_prompt)
  const model_index = document.querySelector('#mModelsMenu').selectedIndex
  const cfg_scale = document.querySelector('#slCfgScale').value
  //  const model_index = document.querySelector("#")
  const seed = document.querySelector('#tiSeed').value
  const mask_blur = document.querySelector('#slMaskBlur').value
  const inpaint_full_res_padding = document.querySelector('#slInpaintPadding').value

  console.dir(numberOfImages)
  const bUsePromptShortcut = document.getElementById('chUsePromptShortcut').checked
  let prompt_shortcut_ui_dict = {}
  try {
    let prompt_shortcut_string = document.getElementById('taPromptShortcut').value
    prompt_shortcut_ui_dict = JSON.parse(prompt_shortcut_string)
  } catch (e) {
    console.warn(`warning prompt_shortcut_ui_dict is not valid Json obj: ${e}`)
    prompt_shortcut_ui_dict = {}
  }


  const uniqueDocumentId = await getUniqueDocumentId()
  payload = {
    prompt: prompt,

    negative_prompt: negative_prompt,
    steps: numberOfSteps,
    // n_iter: numberOfImages,
    sampler_index: g_sd_sampler,
    width: gWidth,
    height: gHeight,
    batch_size: numberOfImages,
    cfg_scale: cfg_scale,
    seed: seed,
    mask_blur: mask_blur,
    use_prompt_shortcut: bUsePromptShortcut,
    prompt_shortcut_ui_dict: prompt_shortcut_ui_dict,
    uniqueDocumentId: uniqueDocumentId
  }

  console.log({ payload })

  //wait 2 seconds till you check for progress
  // setInterval(function(){progressRecursive()},2000);
  setTimeout(function () {
    progressRecursive()

  }, 2000)
  if (g_sd_mode == 'txt2img') {
    json = await sdapi.requestTxt2Img(payload)
  
  } else if (g_sd_mode == 'img2img' || g_sd_mode == 'inpaint') {
    // g_denoising_strength = document.querySelector("#tiDenoisingStrength").value
    if (g_sd_mode == 'inpaint') {
      g_use_mask_image = true
      payload['inpaint_full_res'] =
        document.getElementById('chInpaintFullRes').checked
        payload['inpaint_full_res_padding'] = inpaint_full_res_padding *4
    } else {
      g_use_mask_image = false
      delete payload['inpaint_full_res'] //  inpaint full res is not available in img2img mode
      delete payload['inpaint_full_res_padding']
    }
    console.log(`g_use_mask_image:? ${g_use_mask_image}`)

    payload['denoising_strength'] = g_denoising_strength
    payload['init_image_name'] = g_init_image_name

    if (g_use_mask_image == true) {
      console.log('in true block')

      console.log('g_use_mask_image is ', g_use_mask_image)
      console.log('g_init_image_mask_name is ', g_init_image_mask_name)
      payload['init_image_mask_name'] = g_init_image_mask_name
      payload['inpainting_fill'] = g_inpainting_fill
      // payload['inpainting_fill'] = document.getElementById('slInpainting_fill').value // original
    } else {
      // console.log("g_init_image_mask_name is ",g_init_image_mask_name)
      // payload['init_image_mask_name'] = ""
      delete payload['init_image_mask_name']
      delete payload['inpainting_fill']
    }
    json = await sdapi.requestImg2Img(payload)
  }
  try {
  g_metadatas = []

  for (metadata of json.metadata) {
    metadata_json = JSON.parse(metadata)
    g_metadatas.push(metadata_json)
    console.log('metadata_json:', metadata_json)
  }
} catch (e) {
  console.warn(e)
}

  
  toggleGenerateInterruptButton(false)
  gImage_paths = json.image_paths
  await ImagesToLayersExe()
})

document
  .getElementById('btnRefreshModels')
  .addEventListener('click', (e)=>{
    refreshUI()
    tempDisableElement(e.target,3000)
  }
    )

document.querySelector('#mModelsMenu').addEventListener('change', evt => {
  const model_index = evt.target.selectedIndex
  console.log(`Selected item: ${evt.target.selectedIndex}`)
  let model = g_models[0]
  if (model_index < g_models.length) {
    model = g_models[model_index]
  }
  // g_model_name = `${model.model_name}.ckpt`
  g_model_title = model.title
  console.log('g_model_title: ', g_model_title)
  sdapi.requestSwapModel(g_model_title)
})

// document.getElementById('btnGetActiveLayer').addEventListener('click', getActiveLayer)
// document.getElementById('btnScaleDown').addEventListener('click', scaleDownLayer)
// document.getElementById('btnSelectionInfo').addEventListener('click', getSelectionInfo)
document
  .getElementById('btnLayerToSelection')
  .addEventListener('click', helper.layerToSelection)

// document.getElementById('bGetInitImage').addEventListener('click', () => {
//   sdapi.getInitImage(g_init_image_name)
// })

async function setInitImage () {
  // await exportHelper.exportPng()
  try {
    const layer = await app.activeDocument.activeLayers[0]
    old_name = layer.name 
    await psapi.exportPng(random_session_id)
    
    image_name = psapi.layerNameToFileName(old_name,layer.id,random_session_id)
    
    // image_name = psapi.layerToFileName(layer,random_session_id)

    image_name = `${image_name}.png`
    g_init_image_name = image_name
    console.log(image_name)
    const image_src = await sdapi.getInitImage(g_init_image_name)
    let ini_image_element = document.getElementById('init_image')
    ini_image_element.src = image_src
  } catch (e) {
    console.log('setInitImage error:', e)
  }
}
document.getElementById('bSetInitImage').addEventListener('click', setInitImage)

async function setInitImageMask () {
  try {
    const layer = await app.activeDocument.activeLayers[0]
    old_name = layer.name 
    await psapi.exportPng(random_session_id)
    image_name = psapi.layerNameToFileName(old_name,layer.id,random_session_id)
    
    //get the active layer name
    // const layer = await app.activeDocument.activeLayers[0]
    // image_name = psapi.layerToFileName(layer,random_session_id)
    image_name = `${image_name}.png`
    
    g_init_image_mask_name = image_name
    console.log(image_name)
    
    const image_src = await sdapi.getInitImage(g_init_image_mask_name)
    const ini_image_mask_element = document.getElementById('init_image_mask')
    ini_image_mask_element.src = image_src
  } catch (e) {
    
    console.error(`setInitImageMask error: ${e}`)
  }
}
document
  .getElementById('bSetInitImageMask')
  .addEventListener('click', setInitImageMask)

async function progressRecursive () {
  let json = await sdapi.requestProgress()
  document.querySelector('#pProgressBar').value = json.progress * 100
  if (json.progress > 0 && g_can_request_progress == true) {
    setTimeout(async ()=>{
      await progressRecursive()
    },500)
    
  }
}

// document
//   .getElementById('bGetProgress')
//   .addEventListener('click', progressRecursive)

function changeImage () {
  let img = document.getElementById('img1')
  img.src = 'https://source.unsplash.com/random'
}

// document.getElementById('btnChangeImage').addEventListener('click', changeImage)

async function imageToSmartObject () {
  const { batchPlay } = require('photoshop').action
  const { executeAsModal } = require('photoshop').core

  try {
    // const file = await fs.getFileForOpening()
    // token = await fs.getEntryForPersistentToken(file);
    // const entry = await fs.getEntryForPersistentToken(token);
    // const session_token = await fs.createSessionToken(entry);
    // // let token = await fs.createSessionToken(entry)
    await executeAsModal(
      async () => {
        console.log('imageToSmartObject():')
        const storage = require('uxp').storage
        const fs = storage.localFileSystem
        let pluginFolder = await fs.getPluginFolder()
        let img = await pluginFolder.getEntry('image1.png')
        const result = await batchPlay(
          [
            {
              _obj: 'placeEvent',
              ID: 95,
              null: {
                _path: img,
                _kind: 'local'
              },
              freeTransformCenterState: {
                _enum: 'quadCenterState',
                _value: 'QCSAverage'
              },
              offset: {
                _obj: 'offset',
                horizontal: {
                  _unit: 'pixelsUnit',
                  _value: 0
                },
                vertical: {
                  _unit: 'pixelsUnit',
                  _value: 0
                }
              },
              replaceLayer: {
                _obj: 'placeEvent',
                from: {
                  _ref: 'layer',
                  _id: 56
                },
                to: {
                  _ref: 'layer',
                  _id: 70
                }
              },
              _options: {
                dialogOptions: 'dontDisplay'
              }
            }
          ],
          {
            synchronousExecution: false,
            modalBehavior: 'fail'
          }
        )
      },
      {
        commandName: 'Create Label'
      }
    )
  } catch (e) {
    console.log('imageToSmartObject() => error: ')
    console.log(e)
  }
}

// document.getElementById('btnNewLayer').addEventListener('click', imageToSmartObject )

async function fillLayer () {
  // User picks an image file
  const storage = require('uxp').storage
  const fs = storage.localFileSystem
  let imageFile = await fs.getFileForOpening({
    types: storage.fileTypes.images
  })

  // Create ImageFill for this image
  const ImageFill = require('scenegraph').ImageFill
  let fill = new ImageFill(imageFile)

  let layer = getActiveLayer()
  layer.fillImage()
}
//fillLayer()

// Set fill of first selected item
// selection.items[0].fill = fill;

async function placeEmbedded () {
  console.log('placeEmbedded():')

  const { batchPlay } = require('photoshop').action
  const { executeAsModal } = require('photoshop').core

  const storage = require('uxp').storage
  const fs = storage.localFileSystem
  entry = await fs.getFileForOpening()
  let token = fs.createSessionToken(entry)

  try {
    executeAsModal(async () => {
      const result = await batchPlay(
        [
          {
            _obj: 'placeEvent',
            ID: 6,
            null: {
              _path: token,
              _kind: 'local'
            },
            freeTransformCenterState: {
              _enum: 'quadCenterState',
              _value: 'QCSAverage'
            },
            offset: {
              _obj: 'offset',
              horizontal: {
                _unit: 'pixelsUnit',
                _value: 0
              },
              vertical: {
                _unit: 'pixelsUnit',
                _value: 0
              }
            },
            _isCommand: false,
            _options: {
              dialogOptions: 'dontDisplay'
            }
          }
        ],
        {
          synchronousExecution: false,
          modalBehavior: 'fail'
        }
      )
    })
  } catch (e) {
    console.log(e)
  }
}

// document.getElementById('btnImageFileToLayer').addEventListener('click', placeEmbedded)

// open an image in the plugin folder as new document
async function openImageAction () {
  const storage = require('uxp').storage
  const fs = storage.localFileSystem
  try {
    let pluginFolder = await fs.getPluginFolder()
    // let theTemplate = await pluginFolder.getEntry("/image1.png");
    //directory where all image's request folders are. one folder for each request
    const relative_dir_path = `./server/python_server/`

    image_path = `${relative_dir_path}/${gCurrentImagePath}`
    // 'C:/Users/abdul/Desktop/photoshop_plugins/my_plugin_1/server/python_server/output- 1670544300.95411.png'
    let theTemplate = await pluginFolder.getEntry(image_path)

    await app.open(theTemplate)
  } catch (e) {
    console.log("couldn't open image ", e)
  }
}

async function openImageExe () {
  await require('photoshop').core.executeAsModal(openImageAction)
}
// document.getElementById('btnImagesToLayers').addEventListener('click',openImageExe)

// convert a layer to a smart object
async function convertToSmartObjectAction () {
  const batchPlay = require('photoshop').action.batchPlay
  const result = await batchPlay(
    [
      {
        _obj: 'newPlacedLayer',
        _isCommand: true,
        _options: {
          dialogOptions: 'dontDisplay'
        }
      }
    ],
    {}
  )
}
async function convertToSmartObjectExe () {
  await require('photoshop').core.executeAsModal(convertToSmartObjectAction)
}

async function ImagesToLayersExe () {
  for (image_path of gImage_paths) {
    gCurrentImagePath = image_path
    console.log(gCurrentImagePath)
    await openImageExe() //local image to new document
    await convertToSmartObjectExe() //convert the current image to smart object
    await stackLayers() // move the smart object to the original/old document
    await helper.layerToSelection() //transform the new smart object layer to fit selection area
    // await reselect(selectionInfo)
  }
}

// document.getElementById('btnLoadImages').addEventListener('click',ImagesToLayersExe)

//stack layer to original document
async function stackLayers () {
  //workingDoc is the project you are using stable diffusion in
  const workingDoc = app.documents[0]
  //you should not open two multiple projects this script assume there is only one project opened
  const docsToStack = app.documents.filter(doc => doc._id !== workingDoc._id)
  let docCounter = 0

  // execute as modal is required for functions that change the state of Photoshop or documents
  // think of it as a function that 'wraps' yours and tells Photoshop to go into a modal state and not allow anything to interrupt it from doing whatever is contained in the executeAsModal
  // we also call it with the await keyword to tell JS that we want to wait for it to complete before moving on to later code (in this case there isn't any though)
  await require('photoshop').core.executeAsModal(() => {
    // increment counter
    docCounter++

    // loop through other open docs
    for (const doc of docsToStack) {
      // flatten
      // doc.flatten();

      // rename layer with counter
      doc.layers[0].name = `Layer ${docCounter}`

      // increment counter
      docCounter++

      // duplicate layer to docZero
      doc.layers[0].duplicate(workingDoc)

      // close doc
      doc.closeWithoutSaving()
    }
  })
}
document.getElementById('collapsible').addEventListener('click', function () {
  this.classList.toggle('active')
  var content = this.nextElementSibling
  console.log('content:', content)
  if (content.style.display === 'block') {
    content.style.display = 'none'
    this.textContent = 'Show Samplers'
  } else {
    content.style.display = 'block'
    this.textContent = 'Hide Samplers'
  }
  // this.textContent = `${g_sd_sampler}: ${this.textContent}`
})

document.getElementById('btnLoadHistory').addEventListener('click',async function(){
  try{

    const output_dir_relative = "./server/python_server/"
    const container = document.getElementById("divHistoryImagesContainer")
    const uniqueDocumentId = await getUniqueDocumentId()
    const [image_paths, metadata_jsons] = await sdapi.loadHistory(uniqueDocumentId)
    
    while(container.firstChild){
    container.removeChild(container.firstChild);
    }
    
    let i = 0
    for (image_path of image_paths){
      
      const img = document.createElement('img')
      img.src = `${output_dir_relative}/${image_path}`
      img.className = "history-image"
      img.dataset.metadata_json_string = JSON.stringify(metadata_jsons[i])
      container.appendChild(img)
      img.addEventListener('click',(e)=>{
        const metadata_json = JSON.parse(e.target.dataset.metadata_json_string)
        console.log("metadata_json: ",metadata_json)
        document.querySelector('#tiSeed').value = metadata_json.Seed
        document.querySelector('#historySeedLabel').textContent = metadata_json.Seed
      })
      i++
    }
    
  }catch(e){
    console.warn(`loadHistory warning: ${e}`)
  }

}) 
document.getElementById('btnLoadPromptShortcut').addEventListener('click',async function(){
  try{

    prompt_shortcut = await sdapi.loadPromptShortcut()
    var JSONInPrettyFormat = JSON.stringify(prompt_shortcut, undefined, 4);
    document.getElementById('taPromptShortcut').value = JSONInPrettyFormat
  }catch(e){
    console.warn(`loadPromptShortcut warning: ${e}`)
  }

}) 


document
  .getElementById('btnSavePromptShortcut')
  .addEventListener('click', async function () {
    try {
      

      const r1 = await dialog_box.prompt(
        'Are you sure you want to save prompt shortcut?',
        "This will override your old prompt shortcut file, you can't undo this operation",
        ['Cancel', 'Save']
      )
      if ((r1 || 'Save') !== 'Save') {
        /* cancelled or No */
        console.log("cancel")
      } else {
        /* Yes */
        console.log("Save")
        
      
      
        prompt_shortcut_string = document.getElementById('taPromptShortcut').value
        let prompt_shortcut =  JSON.parse(prompt_shortcut_string)
  
  
        prompt_shortcut = await sdapi.savePromptShortcut(prompt_shortcut)
        // var JSONInPrettyFormat = JSON.stringify(prompt_shortcut, undefined, 4);
        console.log('prompt_shortcut was saved: ', prompt_shortcut)
      
      }
    } catch (e) {
      console.warn(`savePromptShortcut warning: ${e}`)
    }



     
  })