WisdomCity-Web
/
huikaifa


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
							/*
 * @Author: wjc
 * @Date: 2021-12-14 15:05:05
 * @LastEditors: wjc
<<<<<<< Updated upstream
 * @LastEditTime: 2021-12-22 10:03:46
=======
 * @LastEditTime: 2021-12-24 18:03:19
>>>>>>> Stashed changes
 * @Description:
 */
/* eslint-disable */
const vscode = require('vscode')
const fs = require('fs')
const path = require('path')
const open = require('open')

const DB_PATH = path.join(__dirname, './plugin/dbData/db.json')

function getExtensionFileAbsolutePath(context, relativePath) {
  return path.join(context.extensionPath, relativePath)
}

/**
 * 从某个HTML文件读取能被Webview加载的HTML内容
 * @param {*} context 上下文
 * @param {*} templatePath 相对于插件根目录的html文件相对路径
 */
 function getWebViewContent(context, templatePath) {
  const resourcePath = getExtensionFileAbsolutePath(context, templatePath)
  const dirPath = path.dirname(resourcePath)
  let html = fs.readFileSync(resourcePath, 'utf-8')
  // vscode不支持直接加载本地资源，需要替换成其专有路径格式，这里只是简单的将样式和JS的路径替换
  html = html.replace(/(<link.+?href="|<script.+?src="|<img.+?src=")(.+?)"/g, (m, $1, $2) => {
    return $1 + vscode.Uri.file(path.resolve(dirPath, $2)).with({ scheme: 'vscode-resource' }).toString() + '"'
  })
  return html
}

const methods = {
  writeFile(message, vscode, dirPath) {
    const { fileName, code } = message.data
    const filePath = path.join(dirPath, fileName)
    fs.writeFileSync(filePath, code)
    vscode.window.showInformationMessage(`文件${fileName}创建成功`)
  },
  openUrl(message, vscode, dirPath) {
    open(message.data.url)
  },
  setStorageItem(message, vscode, dirPath) {
    const { key, val } = message.data
    const str = fs.readFileSync(DB_PATH).toString()
    let json = {}
    if (str) {
      json = JSON.parse(str)
    }
    json[key] = val
    fs.writeFileSync(DB_PATH, JSON.stringify(json))
  }
}

module.exports = function (context) {
  context.subscriptions.push(vscode.commands.registerCommand('extension.openHuiKaiFa', (uri) => {
    if (!uri) {
      console.log('--url--', uri)
      let dirPath = uri.fsPath,
          stat = fs.lstatSync(dirPath)
      if (stat.isFile()) dirPath = path.dirname(dirPath)

      let pclintBar = vscode.window.createStatusBarItem()
      pclintBar.text = `目标文件夹：${dirPath}`
      pclintBar.show()

      const panel = vscode.window.createWebviewPanel(
        'huikaifa',
        '绘开发',
        vscode.ViewColumn.One,
        {
          enableScripts: true, // 启用JS，默认禁用
          retainContextWhenHidden: true // webview被隐藏时保持状态，避免被重置
        }
      )
      panel.onDidChangeViewState(e => {
        if (panel.visible) {
          pclintBar.show()
        } else {
          pclintBar.hide()
        }
      })
      panel.webview.html = getWebViewContent(context, 'plugin/index.html')
      // panel.webview.html = '<html><body>你好，我是Webview</body></html>'
      panel.webview.postMessage({
        cmd: 'setSrc',
        data: {
          src: vscode.workspace.getConfiguration().get('openHuiKaiFa.src'),
          db: JSON.parse(fs.readFileSync(DB_PATH).toString() || '{}')
        }
      })
      panel.webview.onDidReceiveMessage(message => {
        if (message.cmd && message.data) {
          const method = methods[message.cmd]
          if (method) method(message, vscode, dirPath)
        } else {
          vscode.window.showInformationMessage('没有与消息对应的方法')
        }
      }, undefined, context.subscriptions)
      panel.onDidDispose(e => {
        pclintBar.dispose()
      })
    } else {
      vscode.window.showInformationMessage('无法获取文件夹路径')
    }
  }))
}