// Changing defaultConfigurations requires changing popup/popup.js
const defaultConfigurations = {
  openai: {
    url: 'https://api.openai.com/v1/chat/completions',
    base: {
      n: 1,
      temperature: 0.5,
      model: 'gpt-4-turbo'
    },
    Complete: {
      max_tokens: 512,
      messages: [{
        role: 'system',
        content: 'You are an assistant in a Latex editor, usually used for writing academic papers. Continues the given text. No need to rewrite the given text.'
      }]
    },
    Improve: {
      messages: [{
        role: 'system',
        content: 'You are an assistant in a Latex editor, usually used for writing academic papers. Style and improves the given text.'
      }]
    },
    Ask: {
      messages: [{
        role: 'system',
        content: 'You are an assistant in a Latex editor, usually used for writing academic papers. Address the given questions/request without introduction/explanations.'
      }]
    }
  }
}

const settings = [
  { key: 'Complete', shortcut: 'Ctrl + Shift + C', status: 'enabled', type: 'Command' },
  { key: 'Improve', shortcut: 'Ctrl + Shift + I', status: 'enabled', type: 'Command' },
  { key: 'Ask', shortcut: 'Ctrl + Shift + A', status: 'enabled', type: 'Command' },
  { key: 'RequestConfiguration', value: defaultConfigurations, type: 'Configuration' }
]

async function sendMessage(message) {
  const [tab] = await chrome.tabs.query({ active: true, lastFocusedWindow: true })
  if (tab == null || tab.url?.startsWith('chrome://')) return undefined
  chrome.tabs.sendMessage(tab.id, message)
  // do something with response here, not outside the function
  // console.log(response)
}

function addListener(commandName) {
  chrome.commands.onCommand.addListener((command) => {
    if (command !== commandName) return
    console.log(`Command ${command} triggered`)
    sendMessage({ command: command })
  })
}

// Only use this function during the initial installation phase. After
// installation the user may have intentionally unassigned commands.
// Example for install commands: [{"description":"","name":"_execute_action","shortcut":""},{"description":"Use the selected text to ask GPT. It adds to the beginning of the selected text: 'In Latex, '","name":"Ask","shortcut":""},{"description":"Complete selected text","name":"Complete","shortcut":""},{"description":"Improve selected text","name":"Improve","shortcut":""}]
async function checkCommandShortcuts() {
  chrome.commands.getAll((commands) => {
    for (let { name, shortcut } of commands) {
      let command =
        settings.filter(({ type, key }) => 'Command' === type && name === key)
      if (command.length > 0) {
        command = command[0]
        if (shortcut === '') {
          command.status = 'error'
        }
        chrome.storage.local.set({ [command.key]: command })
      }
    }
  })
}

chrome.runtime.onInstalled.addListener((reason) => {
  if (reason.reason === chrome.runtime.OnInstalledReason.INSTALL) {
    checkCommandShortcuts()
    chrome.storage.local.set({ RequestConfiguration: defaultConfigurations })
  } else if (reason.reason === chrome.runtime.OnInstalledReason.UPDATE) {
    let newVersion = chrome.runtime.getManifest().version
    let oldVersion = reason.previousVersion
    let oldVersionArray = oldVersion.split('.')
    if (parseInt(oldVersionArray[0]) === 1 && parseInt(oldVersionArray[1]) < 4) {
      chrome.storage.local.set({ RequestConfiguration: defaultConfigurations })
    }
  }
})

async function setup() {
  addListener('Improve')
  addListener('Complete')
  addListener('Ask')
  console.log('LeafLLM service worker installed')
}

setup()