All files / src extension.ts

0% Statements 0/165
0% Branches 0/1
0% Functions 0/1
0% Lines 0/165

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222                                                                                                                                                                                                                                                                                                                                                                                                                                                           
import * as vscode from 'vscode'
import { CodeDetector } from './codeDetector'
import { CodeEditorProvider } from './codeEditorProvider'
import { DecorationProvider } from './decorationProvider'
import { shouldProcessFile } from './fileUtils'
import { logger } from './logger'
import { getTempDirectoryUri } from './tempUtils'
 
export function activate(context: vscode.ExtensionContext) {
  logger.info('JSON String Code Editor extension is now active!')
 
  const detector = new CodeDetector()
  detector.updateConfiguration() // 初始化配置
 
  // 监听配置变化
  const configChangeListener = vscode.workspace.onDidChangeConfiguration((e: vscode.ConfigurationChangeEvent) => {
    if (e.affectsConfiguration('vscode-json-string-code-editor')) {
      detector.updateConfiguration()
      logger.onConfigurationChanged()
    }
  })
 
  const editorProvider = new CodeEditorProvider(context)
  const decorationProvider = new DecorationProvider(detector)
 
  // 注册命令:编辑代码
  const editCodeCommand = vscode.commands.registerCommand(
    'vscode-json-string-code-editor.editCode',
    async () => {
      const editor = vscode.window.activeTextEditor
      if (!editor) {
        logger.error('No active editor found')
        return
      }
 
      const document = editor.document
      const selection = editor.selection
      const position = selection.active
 
      // 检查是否启用了自动检测
      const config = vscode.workspace.getConfiguration('vscode-json-string-code-editor')
      const enableAutoDetection = config.get('enableAutoDetection', true)
      if (!enableAutoDetection) {
        vscode.window.showInformationMessage('自动检测功能已禁用,请在设置中启用 enableAutoDetection')
        return
      }
 
      // 检测当前位置是否包含代码
      const codeInfo = detector.detectCodeAtPosition(document, position)
      if (!codeInfo) {
        logger.info('No code detected at current position')
        return
      }
 
      // 打开临时编辑器
      await editorProvider.openCodeEditor(codeInfo, document, editor)
    },
  )
 
  // 注册代码镜头提供器
  const codeLensProvider = vscode.languages.registerCodeLensProvider(
    ['json', 'jsonc'],
    {
      provideCodeLenses(document: vscode.TextDocument): vscode.CodeLens[] {
        // 检查是否启用了自动检测
        const config = vscode.workspace.getConfiguration('vscode-json-string-code-editor')
        const enableAutoDetection = config.get('enableAutoDetection', true)
        if (!enableAutoDetection) {
          return []
        }
 
        const codeLenses: vscode.CodeLens[] = []
        const codeBlocks = detector.detectAllCodeBlocks(document)
 
        for (const block of codeBlocks) {
          const range = new vscode.Range(
            document.positionAt(block.start),
            document.positionAt(block.end),
          )
 
          const codeLens = new vscode.CodeLens(range, {
            title: `✏️ Edit ${block.language.charAt(0).toUpperCase() + block.language.slice(1)}`,
            command: 'vscode-json-string-code-editor.editCodeAtRange',
            arguments: [document.uri.toString(), block],
          })
 
          codeLenses.push(codeLens)
        }
 
        return codeLenses
      },
    },
  )
 
  // 注册范围编辑命令
  const editCodeAtRangeCommand = vscode.commands.registerCommand(
    'vscode-json-string-code-editor.editCodeAtRange',
    async (documentUri: string, blockInfo: any) => {
      const editor = vscode.window.activeTextEditor
      if (!editor) {
        return
      }
 
      // 检查文件是否应该被处理(包括文件类型和include配置)
      if (!shouldProcessFile(editor.document)) {
        return
      }
 
      // 检查是否启用了自动检测
      const config = vscode.workspace.getConfiguration('vscode-json-string-code-editor')
      const enableAutoDetection = config.get('enableAutoDetection', true)
      if (!enableAutoDetection) {
        vscode.window.showInformationMessage('自动检测功能已禁用,请在设置中启用 enableAutoDetection')
        return
      }
 
      // blockInfo 已经是 CodeBlockInfo 格式,直接使用
      await editorProvider.openCodeEditor(blockInfo, editor.document, editor)
    },
  )
 
  // 注册清理临时文件命令
  const cleanupTempFilesCommand = vscode.commands.registerCommand(
    'vscode-json-string-code-editor.cleanupTempFiles',
    async () => {
      try {
        // 获取临时文件目录路径
        const tmpDirUri = getTempDirectoryUri()
 
        // 检查临时目录是否存在
        try {
          await vscode.workspace.fs.stat(tmpDirUri)
        }
        catch {
          logger.info('No temporary files to clean up')
          return
        }
 
        // 递归删除临时目录中的所有文件
        const deleteCount = await deleteTempFiles(tmpDirUri)
 
        logger.info(`Cleaned up ${deleteCount} temporary files`)
      }
      catch (error) {
        const errorMessage = error instanceof Error ? error.message : 'Unknown error'
        logger.error(`Failed to clean up temporary files: ${errorMessage}`)
      }
    },
  )
 
  // 清理临时文件的辅助函数
  async function deleteTempFiles(dirUri: vscode.Uri): Promise<number> {
    let deleteCount = 0
 
    try {
      // 读取目录内容
      const entries = await vscode.workspace.fs.readDirectory(dirUri)
 
      for (const [name, type] of entries) {
        const entryUri = vscode.Uri.joinPath(dirUri, name)
 
        if (type === vscode.FileType.Directory) {
          // 递归删除子目录
          deleteCount += await deleteTempFiles(entryUri)
          await vscode.workspace.fs.delete(entryUri, { recursive: false })
        }
        else {
          // 删除文件
          await vscode.workspace.fs.delete(entryUri)
          deleteCount++
        }
      }
 
      return deleteCount
    }
    catch (error) {
      logger.error(`Failed to delete files in directory ${dirUri.fsPath}: ${error}`)
      return deleteCount
    }
  }
 
  // 注册点击处理器
  const clickHandler = vscode.window.onDidChangeTextEditorSelection(async (event) => {
    const editor = event.textEditor
    if (!editor || !shouldProcessFile(editor.document)) {
      return
    }
 
    // 检查是否点击了代码
    const selection = event.selections[0]
    if (selection && selection.isEmpty) {
      const position = selection.active
      const codeInfo = detector.detectCodeAtPosition(editor.document, position)
 
      if (codeInfo) {
        // 检查是否是单击(而不是拖拽选择)
        const timeSinceLastClick = Date.now() - (clickHandler as any).lastClickTime || 0
        if (timeSinceLastClick > 100) { // 防止重复触发
          (clickHandler as any).lastClickTime = Date.now()
          await editorProvider.openCodeEditor(codeInfo, editor.document, editor)
        }
      }
    }
  })
 
  context.subscriptions.push(
    editCodeCommand,
    editCodeAtRangeCommand,
    cleanupTempFilesCommand,
    codeLensProvider,
    decorationProvider,
    clickHandler,
    configChangeListener,
  )
}
 
export function deactivate() {
  // 清理日志服务
  logger.dispose()
  logger.info('JSON String Code Editor extension deactivated')
}