extension.js

const vscode = require('vscode')
const axios = require('axios').default
const md5 = require('md5')
const { Case } = require('change-case-all')

const switchId = 'banjiao.switch'
const batchId = 'banjiao.batch'
const keymapsId = 'banjiao.keymaps'
const setNameId = 'banjiao.setName'
const translateAppidId = 'banjiao.translateAppid'
const translateSecretId = 'banjiao.translateSecret'

let switchConfig
let batchConfig
let keymapsConfig
let translateAppidConfig
let translateSecretConfig

function setConfig () {
  const config = vscode.workspace.getConfiguration()
  switchConfig = config.get(switchId)
  batchConfig = config.get(batchId)
  keymapsConfig = config.get(keymapsId)
  translateAppidConfig = config.get(translateAppidId)
  translateSecretConfig = config.get(translateSecretId)
}

function getHalfWidthChar (char) {
  for (let i = 0; i < keymapsConfig.length; i++) {
    if (keymapsConfig[i].full === char) {
      return keymapsConfig[i].half
    }
  }

  return char
}

function convertToHalfWidthChar (event) {
  if (vscode.window.activeTextEditor === undefined) return

  vscode.window.activeTextEditor.edit(
    editBuilder => {
      event.contentChanges.forEach(content => {
        const currentPosition = content.range.start
        const contentText = content.text
        let contentTextRange

        if (contentText.length === 1) {
          // 只处理单符号
          let halfChar, prevChar

          if (currentPosition.character > 0) {
            prevChar = event.document.getText(new vscode.Range(currentPosition.translate(0, -1), currentPosition))
          }

          // ^ 全角 …… 会触发两次change事件
          if (contentText === '…' && contentText === prevChar) {
            halfChar = getHalfWidthChar(contentText)
            contentTextRange = new vscode.Range(currentPosition.translate(0, -1), currentPosition.translate(0, 1))
            editBuilder.replace(contentTextRange, halfChar)
          }

          // _ 全角 —— 会触发两次change事件
          if (contentText === '—' && contentText === prevChar) {
            halfChar = getHalfWidthChar(contentText)
            contentTextRange = new vscode.Range(currentPosition.translate(0, -1), currentPosition.translate(0, 1))
            editBuilder.replace(contentTextRange, halfChar)
          }

          // 其他全角字符
          if (contentText !== '…' && contentText !== '—') {
            halfChar = getHalfWidthChar(contentText)

            if (halfChar !== contentText) {
              contentTextRange = new vscode.Range(currentPosition, currentPosition.translate(0, 1))
              editBuilder.replace(contentTextRange, halfChar)
            }
          }
        } else if (batchConfig && contentText.length > 1) {
          // 处理粘贴长文本
          const replacedText = contentText.split('').reduce(
            (previousChar, currentChar) => {
              return previousChar + getHalfWidthChar(currentChar)
            },
            ''
          )

          if (contentText !== replacedText) {
            if (contentText.includes('\n')) { // 多行文本
              const lines = contentText.split('\n')
              const endLineNumber = currentPosition.line + lines.length - 1
              const endLineText = lines[lines.length - 1]
              const endLineDocumentText = vscode.window.activeTextEditor.document.lineAt(endLineNumber).text
              const endLineCharacter = endLineDocumentText.indexOf(endLineText) + endLineText.length
              const endPosition = new vscode.Position(endLineNumber, endLineCharacter)
              contentTextRange = new vscode.Range(currentPosition, endPosition)
            } else if (currentPosition.character === 0) { // 单行第一个位置
              const lineText = vscode.window.activeTextEditor.document.lineAt(currentPosition.line).text
              const startPosition = new vscode.Position(currentPosition.line, lineText.indexOf(contentText))
              const endPosition = new vscode.Position(currentPosition.line, lineText.indexOf(contentText) + contentText.length)
              contentTextRange = new vscode.Range(startPosition, endPosition)
            } else { // 单行其他位置
              const endPosition = new vscode.Position(currentPosition.line, currentPosition.character + content.text.length)
              contentTextRange = new vscode.Range(currentPosition, endPosition)
            }

            editBuilder.replace(contentTextRange, replacedText)
          }
        }
      })
    },
    {
      undoStopAfter: false,
      undoStopBefore: false
    }
  ).then(fulfilled => {
    if (fulfilled === false) {
      convertToHalfWidthChar(event)
    }
  })
}

function getStatusBarItemLabel () {
  return switchConfig ? '半角' : '全角'
}

function getStatusBarItemTooltip () {
  return new vscode.MarkdownString(switchConfig ? '点击关半角 `Alt+B`' : '点击开半角 `Alt+B`')
}

async function fetchTranslateResult (originText, from, to) {
  const domain = 'it'
  const salt = Math.random()

  try {
    const response = await axios({
      method: 'get',
      url: 'https://fanyi-api.baidu.com/api/trans/vip/fieldtranslate',
      params: {
        q: originText,
        from,
        to,
        appid: translateAppidConfig,
        salt,
        domain,
        sign: md5(translateAppidConfig + originText + salt + domain + translateSecretConfig)
      }
    })

    if (response.data.error_code) {
      throw new Error(response.data.error_msg)
    } else {
      return {
        translatedText: response.data.trans_result[0].dst
      }
    }
  } catch (error) {
    return {
      errorMessage: error.message
    }
  }
}

function activate ({ subscriptions }) {
  setConfig()

  const switchCommand = vscode.commands.registerCommand(switchId, () => {
    const config = vscode.workspace.getConfiguration()
    const triggerSwitch = !switchConfig
    config.update(switchId, triggerSwitch, vscode.ConfigurationTarget.Global)
  })
  subscriptions.push(switchCommand)

  const setNameCommand = vscode.commands.registerCommand(setNameId, (variableNames, currentSelection) => {
    const editor = vscode.window.activeTextEditor
    if (!editor) return

    const selection = new vscode.Selection(currentSelection.start, currentSelection.end)
    editor.selection = selection

    vscode.window.showQuickPick(
      variableNames,
      {
        placeHolder: '请选择半角风格'
      }
    )
      .then(selected => {
        if (selected) {
          editor.edit(editBuilder => {
            editBuilder.replace(selection, selected)
          })
        }
      })
  })
  subscriptions.push(setNameCommand)

  const hoverProvider = vscode.languages.registerHoverProvider('*', {
    async provideHover (document, position) {
      if (
        vscode.window.activeTextEditor === undefined ||
        !(switchConfig && translateAppidConfig && translateSecretConfig)
      ) return

      const selection = vscode.window.activeTextEditor.selection
      const hoverRange = document.getWordRangeAtPosition(position)
      let currentText, currentSelection

      if (!hoverRange || hoverRange.isEmpty) return

      if (selection.isEmpty) {
        // 鼠标悬停且无选区
        currentText = document.getText(hoverRange)
        currentSelection = new vscode.Selection(hoverRange.start, hoverRange.end)
      } else if (
        selection.start.line === selection.end.line &&
        position.line === selection.start.line &&
        position.character >= selection.start.character &&
        position.character <= selection.end.character
      ) {
        // 鼠标悬停在当前选区范围内
        currentText = document.getText(selection)
        currentSelection = selection
      } else {
        return
      }

      const containsChinese = /[\u4e00-\u9fa5]/.test(currentText)
      const sourceText = containsChinese ? currentText : Case.capital(currentText)

      if (!sourceText || sourceText.length > 64) return

      const sourceLanguage = containsChinese ? 'zh' : 'en'
      const targetLanguage = containsChinese ? 'en' : 'zh'
      const { translatedText, errorMessage } = await fetchTranslateResult(sourceText, sourceLanguage, targetLanguage)

      if (errorMessage) {
        vscode.window.showErrorMessage(errorMessage)
        return
      }

      const capitalText = Case.capital(containsChinese ? translatedText : currentText)
      const cleanedText = capitalText.replace(/The\s*|\s+/g, '') // 使用正则表达式删除所有的 The 和空格
      const variableNames = [
        Case.camel(cleanedText),
        Case.pascal(cleanedText),
        Case.kebab(cleanedText),
        Case.snake(cleanedText),
        Case.constant(cleanedText)
      ]
      const command = vscode.Uri.parse(`command:banjiao.setName?${encodeURIComponent(JSON.stringify([variableNames, currentSelection]))}`)

      const MarkdownString = new vscode.MarkdownString()
      MarkdownString.isTrusted = true
      MarkdownString.appendMarkdown(translatedText)
      MarkdownString.appendMarkdown(` [半角](${command} "点击选择半角风格")`)

      return new vscode.Hover(MarkdownString)
    }
  })
  subscriptions.push(hoverProvider)

  const statusBarItem = vscode.window.createStatusBarItem(vscode.StatusBarAlignment.Right, 100)
  statusBarItem.command = switchId
  statusBarItem.text = getStatusBarItemLabel()
  statusBarItem.tooltip = getStatusBarItemTooltip()
  subscriptions.push(statusBarItem)

  if (vscode.workspace.textDocuments.length > 0) {
    statusBarItem.show()
  }

  vscode.window.onDidChangeActiveTextEditor(textEditor => {
    if (textEditor) {
      statusBarItem.show()
    } else {
      statusBarItem.hide()
    }
  })

  vscode.workspace.onDidChangeConfiguration(() => {
    setConfig()

    statusBarItem.text = getStatusBarItemLabel()
    statusBarItem.tooltip = getStatusBarItemTooltip()
  })

  vscode.workspace.onDidChangeTextDocument(event => {
    // [undo 和 redo 时忽略转换](https://code.visualstudio.com/api/references/vscode-api#TextDocumentChangeReason)
    if (event.reason === 1 || event.reason === 2) return

    // 只处理代码文件修改
    // 如 event.document.uri.scheme === 'vscode-scm' 是 vscode git 客户端 commit-msg 输入框会被禁止
    if (switchConfig && event.document.uri.scheme !== 'vscode-scm') {
      convertToHalfWidthChar(event)
    }
  })

  console.info('已激活半角插件')
}

function deactivate () {}

module.exports = {
  activate,
  deactivate
}