http.server/index.js

/**
 *
 * @author yutent<yutent.io@gmail.com>
 * @date 2022/04/09 18:26:35
 */

const vsc = require('vscode')
const { join } = require('path')
const { parse } = require('url')
const http = require('http')
const fs = require('iofs')
const { WebSocketServer } = require('ws')

const std = vsc.window.createOutputChannel('http.server')
const decode = decodeURIComponent

const MIME_TYPES = {
  html: 'text/html;charset=utf-8',
  txt: 'text/plain;charset=utf-8',
  css: 'text/css;charset=utf-8',
  xml: 'text/xml;charset=utf-8',
  gif: 'image/gif',
  jpg: 'image/jpeg',
  webp: 'image/webp',
  tiff: 'image/tiff',
  png: 'image/png',
  svg: 'image/svg+xml',
  ico: 'image/x-icon',
  bmp: 'image/x-ms-bmp',
  js: 'application/javascript;charset=utf-8',
  json: 'application/json;charset=utf-8',
  mp3: 'audio/mpeg',
  ogg: 'audio/ogg',
  m4a: 'audio/x-m4a',
  mp4: 'video/mp4',
  webm: 'video/webm',
  ttf: 'font/font-ttf',
  woff: 'font/font-woff',
  woff2: 'font/font-woff2',
  wast: 'application/wast',
  wasm: 'application/wasm',
  other: 'application/octet-stream'
}

const COMMON_HEADERS = {
  'Access-Control-Allow-Origin': '*',
  'Access-Control-Allow-Headers': '*',
  'Cache-Control': 'no-store',
  'X-Powered-By': 'VS Code simple.http'
}

MIME_TYPES.htm = MIME_TYPES.html
MIME_TYPES.jpeg = MIME_TYPES.jpg
MIME_TYPES.tif = MIME_TYPES.tiff

let root
let enabled = false
let port = 23333
let baseUrl = 'http://127.0.0.1:' + port
let ws = null

std.out = function (...args) {
  std.appendLine('[simple.http]: ' + args.join(' '))
  // console.log('[simple.http]: ' + args.join(' '))
}

const HMR_SCRIPT = `
<script>
!(function http_hmr(){
  var ws = new WebSocket(\`\${location.protocol === 'https:' ? 'wss' : 'ws'}://\${location.host}/ws-http-hmr\`)
  ws.addEventListener('open', function (r) {
    if(http_hmr.closed){
      delete http_hmr.closed
      location.reload()
    }
    console.log('http.server hmr ready...')
  })
  ws.addEventListener('close', function(){
    http_hmr.closed = true
    setTimeout(http_hmr, 2000)
  })
  
  ws.addEventListener('message', function (ev) {
    location.reload()
  })
})()
</script>
`

class WebSocket {
  #ws = null // ws实例
  #queue = [] // 消息队列

  constructor(server) {
    if (server.listening) {
      let conn = new WebSocketServer({ server, path: '/ws-http-hmr' })
      conn.on('connection', ws => {
        this.#ws = ws
        if (this.#queue.length) {
          this.#queue = []
          this.send()
        }
      })
    }
  }

  send(msg = {}) {
    if (this.#ws) {
      this.#ws.send(JSON.stringify(msg))
    } else {
      this.#queue.push(msg)
    }
  }
}

function createServer() {
  if (port > 65535) {
    std.out('端口超出有效范围0-65535, 当前为: ' + port)
    enabled = false
    return
  }
  std.out(`尝试使用${port}端口...`)
  const server = http
    .createServer(function (req, res) {
      let pathname = parse(req.url).pathname.slice(1)
      let paths = pathname.split('/')

      pathname = decode(pathname) || 'index.html'

      if (pathname.endsWith('/')) {
        pathname += 'index.html'
      }

      let file = join(root, pathname)
      let stat = fs.stat(file)
      let ext = pathname.split('.').pop()
      let code

      for (let k in COMMON_HEADERS) {
        res.setHeader(k, COMMON_HEADERS[k])
      }

      if (stat.isFile()) {
        res.setHeader('accept-ranges', 'bytes')
        res.setHeader('content-type', MIME_TYPES[ext] || MIME_TYPES.other)

        if (ext === 'html') {
          code = fs
            .cat(file)
            .toString()
            .replace('</head>', HMR_SCRIPT + '</head>')

          res.setHeader('content-length', Buffer.byteLength(code))
        } else {
          res.setHeader('content-length', stat.size)
        }

        res.writeHead(200, 'OK')

        if (ext === 'html') {
          res.end(code)
        } else {
          fs.origin.createReadStream(file).pipe(res)
        }
      } else {
        if (paths.length > 1 && paths[0].endsWith('.html')) {
          let rootFile = join(root, paths[0])
          if (fs.isfile(rootFile)) {
            code = fs
              .cat(rootFile)
              .toString()
              .replace('</head>', HMR_SCRIPT + '</head>')

            res.setHeader('accept-ranges', 'bytes')
            res.setHeader('content-type', MIME_TYPES.html)
            res.setHeader('content-length', Buffer.byteLength(code))
            res.writeHead(200, 'OK')

            res.end(code)
            return
          }
        }
        res.setHeader('content-length', 0)
        res.setHeader('x-url', file)
        res.writeHead(404, 'Not Found')
        res.end('')
      }
    })
    .listen(port)
    .on('error', err => {
      std.out(`${port}端口被占用~~~`)
      port++
      createServer()
    })

  server.on('listening', _ => {
    baseUrl = 'http://127.0.0.1:' + port
    ws = new WebSocket(server)
    std.out('启动成功, 请访问', baseUrl)
  })
}

function __init__() {
  let folders = vsc.workspace.workspaceFolders

  if (folders && folders.length) {
    root = folders[0].uri.fsPath
  }
  if (root) {
    let file = join(root, '.httpserver')
    //
    if (fs.isfile(file)) {
      let conf = JSON.parse(fs.cat(file).toString())

      if (conf.root) {
        root = join(root, conf.root)
      }

      if (conf.headers) {
        for (let k in conf.headers) {
          let _k = k.toLowerCase()
          if (['accept-ranges', 'content-type', 'content-length'].includes(_k)) {
            continue
          }
          COMMON_HEADERS[_k] = conf.headers[k]
        }
      }

      if (conf.enabled) {
        enabled = true
        port = conf.port || 23333

        createServer()
      } else {
        std.out('发现配置文件, 但服务为 关闭状态')
      }
    }
  }
}

function deactivate() {}

exports.activate = function (ctx) {
  __init__()

  vsc.workspace.onDidSaveTextDocument(doc => {
    if (enabled) {
      let file = doc.fileName
      if (file.startsWith(root)) {
        ws.send()
      }
    }
  })

  let cmd = vsc.commands.registerCommand('HttpServer.open', _ => {
    if (enabled) {
      let editor = vsc.window.activeTextEditor
      if (editor) {
        let pathname = editor.document.uri.fsPath.slice(root.length)

        vsc.commands.executeCommand('vscode.open', baseUrl + pathname)
      }
    }
  })
  ctx.subscriptions.push(cmd)
}
exports.deactivate = deactivate