node-karin 0.0.3 → 0.1.0

package/lib/renderer/server.js

@@ -1,112 +1,112 @@
-import Renderer from './app.js'
-import RenderBase from './base.js'
-import { randomUUID } from 'crypto'
-import logger from '../utils/logger.js'
-import listener from '../core/listener.js'
-class Puppeteer extends RenderBase {
-  socket
-  id
-  type
-  host
-  url
-  index
-  constructor () {
-    super()
-    this.id = 0
-    this.index = 0
-    this.type = ''
-    this.host = ''
-    this.url = ''
+import Renderer from './app.js'
+import RenderBase from './base.js'
+import { randomUUID } from 'crypto'
+import logger from '../utils/logger.js'
+import listener from '../core/listener.js'
+class Puppeteer extends RenderBase {
+  socket
+  id
+  type
+  host
+  url
+  index
+  constructor () {
+    super()
+    this.id = 0
+    this.index = 0
+    this.type = ''
+    this.host = ''
+    this.url = ''
   }
-
-  async server (socket, request) {
-    this.socket = socket
-    this.id = request.headers['renderer-id']
-    this.type = request.headers['renderer-type']
-    /** 注册渲染器 */
-    this.host = request.headers.host
-    this.url = `ws://${this.host + request.url}`
-    logger.info(`[渲染器:${this.id}] 收到新的连接请求：` + logger.green(this.url))
-    /** 监听上报事件 */
-    this.socket.on('message', data => {
-      const json = JSON.parse(data.toString())
-      if (json.echo) {
-        listener.emit(json.echo, json)
-      } else if (json.action === 'heartbeat') {
-        logger.debug(`[渲染器:${this.id}] 收到心跳：${this.url}`)
-      } else {
-        logger.warn(`[渲染器:${this.id}] 收到未知数据：`, data)
-      }
-    })
-    /** 监听断开 */
-    this.socket.on('close', () => {
-      logger.warn(`[渲染器:${this.id}] 连接断开：${this.url}`)
-      /** 卸载渲染器 */
-      this.index && Renderer.unapp(this.index)
-      this.index = 0
-    })
-    /** 注册渲染器 */
-    try {
-      const index = Renderer.app({
-        id: this.id,
-        type: this.type,
-        render: this.render.bind(this),
-      })
-      this.index = index
-    } catch (error) {
-      logger.error(`[渲染器:${this.id}] 注册渲染器失败：`, error)
-      /** 断开连接 */
-      this.socket.close()
-    }
+
+  async server (socket, request) {
+    this.socket = socket
+    this.id = request.headers['renderer-id']
+    this.type = request.headers['renderer-type']
+    /** 注册渲染器 */
+    this.host = request.headers.host
+    this.url = `ws://${this.host + request.url}`
+    logger.info(`[渲染器:${this.id}] 收到新的连接请求：` + logger.green(this.url))
+    /** 监听上报事件 */
+    this.socket.on('message', data => {
+      const json = JSON.parse(data.toString())
+      if (json.echo) {
+        listener.emit(json.echo, json)
+      } else if (json.action === 'heartbeat') {
+        logger.debug(`[渲染器:${this.id}] 收到心跳：${this.url}`)
+      } else {
+        logger.warn(`[渲染器:${this.id}] 收到未知数据：`, data)
+      }
+    })
+    /** 监听断开 */
+    this.socket.on('close', () => {
+      logger.warn(`[渲染器:${this.id}] 连接断开：${this.url}`)
+      /** 卸载渲染器 */
+      this.index && Renderer.unapp(this.index)
+      this.index = 0
+    })
+    /** 注册渲染器 */
+    try {
+      const index = Renderer.app({
+        id: this.id,
+        type: this.type,
+        render: this.render.bind(this),
+      })
+      this.index = index
+    } catch (error) {
+      logger.error(`[渲染器:${this.id}] 注册渲染器失败：`, error)
+      /** 断开连接 */
+      this.socket.close()
+    }
   }
-
-  /**
-     * 渲染模板
-     * @param {object} options 渲染参数
-     * @param {string} options.file http地址或本地文件路径
-     * @param {string} [options.name] 模板名称
-     * @param {string} [options.fileID] art-template后的文件名
-     * @param {object} [options.data] 传递给模板的数据 template.render(data)
-     * @param {'png'|'jpeg'|'webp'} [options.type] 截图类型 默认'webp'
-     * @param {number} [options.quality] 截图质量 默认90 1-100
-     * @param {boolean} options.omitBackground 是否隐藏背景 默认false
-     * @param {object} [options.setViewport] 设置视窗大小和设备像素比 默认1920*1080、1
-     * @param {number} [options.setViewport.width] 视窗宽度
-     * @param {number} [options.setViewport.height] 视窗高度
-     * @param {string} [options.setViewport.deviceScaleFactor] 设备像素比
-     * @param {number|boolean} [options.multiPage] 分页截图 传递数字则视为视窗高度 返回数组
-     * @param {object} [options.pageGotoParams] 页面goto时的参数
-     * @param {number} [options.pageGotoParams.timeout] 页面加载超时时间
-     * @param {'load'|'domcontentloaded'|'networkidle0'|'networkidle2'} [options.pageGotoParams.waitUntil] 页面加载状态
-     * @returns {Promise<string|string[]>} 返回图片base64或数组
-     */
-  async render (options) {
-    /** 渲染模板 */
-    let file = ''
-    if (options.file.includes('http') || options.vue) {
-      file = options.file
-    } else {
-      file = 'file://' + this.dealTpl(options)
-    }
-    if (!file) { throw new Error(`[渲染器:${this.id}] 模板文件不存在：${options.name}`) }
-    const echo = randomUUID()
-    const action = 'render'
-    const data = options
-    /** 移除掉模板参数 */
-    if (data.data) { delete data.data }
-    data.file = file
-    logger.debug(`[渲染器:${this.id}][反向WS] \n请求:${this.url} \nhtml: ${options.file} \ndata: ${JSON.stringify(data)}`)
-    this.socket.send(JSON.stringify({ echo, action, data }))
-    return new Promise((resolve, reject) => {
-      listener.once(echo, data => {
-        if (data.ok) { return resolve(data.data) }
-        reject(new Error(JSON.stringify(data)))
-      })
-    })
-  }
-}
-export default {
-  type: 'render',
-  path: '/puppeteer',
-  adapter: Puppeteer,
-}
+
+  /**
+     * 渲染模板
+     * @param {object} options 渲染参数
+     * @param {string} options.file http地址或本地文件路径
+     * @param {string} [options.name] 模板名称
+     * @param {string} [options.fileID] art-template后的文件名
+     * @param {object} [options.data] 传递给模板的数据 template.render(data)
+     * @param {'png'|'jpeg'|'webp'} [options.type] 截图类型 默认'webp'
+     * @param {number} [options.quality] 截图质量 默认90 1-100
+     * @param {boolean} options.omitBackground 是否隐藏背景 默认false
+     * @param {object} [options.setViewport] 设置视窗大小和设备像素比 默认1920*1080、1
+     * @param {number} [options.setViewport.width] 视窗宽度
+     * @param {number} [options.setViewport.height] 视窗高度
+     * @param {string} [options.setViewport.deviceScaleFactor] 设备像素比
+     * @param {number|boolean} [options.multiPage] 分页截图 传递数字则视为视窗高度 返回数组
+     * @param {object} [options.pageGotoParams] 页面goto时的参数
+     * @param {number} [options.pageGotoParams.timeout] 页面加载超时时间
+     * @param {'load'|'domcontentloaded'|'networkidle0'|'networkidle2'} [options.pageGotoParams.waitUntil] 页面加载状态
+     * @returns {Promise<string|string[]>} 返回图片base64或数组
+     */
+  async render (options) {
+    /** 渲染模板 */
+    let file = ''
+    if (options.file.includes('http') || options.vue) {
+      file = options.file
+    } else {
+      file = 'file://' + this.dealTpl(options)
+    }
+    if (!file) { throw new Error(`[渲染器:${this.id}] 模板文件不存在：${options.name}`) }
+    const echo = randomUUID()
+    const action = 'render'
+    const data = options
+    /** 移除掉模板参数 */
+    if (data.data) { delete data.data }
+    data.file = file
+    logger.debug(`[渲染器:${this.id}][反向WS] \n请求:${this.url} \nhtml: ${options.file} \ndata: ${JSON.stringify(data)}`)
+    this.socket.send(JSON.stringify({ echo, action, data }))
+    return new Promise((resolve, reject) => {
+      listener.once(echo, data => {
+        if (data.ok) { return resolve(data.data) }
+        reject(new Error(JSON.stringify(data)))
+      })
+    })
+  }
+}
+export default {
+  type: 'render',
+  path: '/puppeteer',
+  adapter: Puppeteer,
+}

package/lib/renderer/wormhole.d.ts

@@ -1 +1 @@
-export default function connect(): void;
+export default function connect(): void;

package/lib/renderer/wormhole.js

@@ -1,154 +1,154 @@
-import WebSocket from 'ws'
-import fs from 'fs'
-import path from 'path'
-import { URL } from 'url'
-import HttpRenderer from './http.js'
-import config from '../utils/config.js'
-import { render } from '../index.js'
-let ws
-let reConnect
-const chunkSize = 1024 * 1024 * 3 // 文件分片大小
-export default function connect () {
-  let heartbeat
-  let index = 0
-  reConnect = undefined
-  const wsUrl = config.Server.HttpRender.WormholeClient
-  ws = new WebSocket(wsUrl)
-  ws.on('open', function open () {
-    logger.info('连接到wormhole服务器' + wsUrl)
-    // 发送心跳
-    heartbeat = setInterval(() => {
-      ws.send(JSON.stringify({ type: 'heartbeat', date: new Date() }))
-    }, 30000) // 每30秒发送一次心跳
-  })
-  ws.on('message', msg => {
-    let data = ''
-    try {
-      data = JSON.parse(msg.toString())
-    } catch (error) {
-      logger.warn(`收到非法消息${data}`)
-    }
-    const echo = data.echo
-    switch (data.type) {
-      case 'msg': {
-        const { post, token, WormholeClient } = config.Server.HttpRender
-        const parsedUrl = new URL(WormholeClient)
-        const { hostname, port } = parsedUrl
-        const ishttps = WormholeClient.includes('wss://')
-        const host = `${ishttps ? 'https' : 'http'}://${hostname}${port ? `:${port}` : ''}/web/${data.date}`
-        logger.mark(`web渲染器已连接，地址：${host}`)
-        /** 注册渲染器 */
-        const rd = new HttpRenderer(host, post, token)
-        index = render.app({ id: 'puppeteer', type: 'image', render: rd.render.bind(rd) })
-        break
-      }
-      case 'web':
-        if (data.path) {
-          const filePath = data.path
-          const query = data.query
-          if (query.html) {
-            ws.send(JSON.stringify({ type: 'web', command: 'redirect', path: filePath, target: query.html.startsWith('/') ? query.html.slice(1) : query.html, echo }))
-            return
-          }
-          const list = ['.css', '.html', '.ttf', '.jpg', '.jpeg', '.png', '.gif', '.bmp', '.ico', '.woff', '.woff2']
-          if (!list.some(ext => path.extname(filePath).endsWith(ext))) {
-            logger.warn(`拦截非资源文件${filePath}`)
-            ws.send(JSON.stringify({ type: 'web', state: 'error', error: '非资源文件', echo }))
-            return
-          }
-          /** 判断一下是否为html 如果是需要特殊处理 */
-          if (path.extname(filePath) === '.html') {
-            let html = filePath
-            /** 获取文件路径 对路径进行处理，去掉../、./ */
-            html = `./${html.replace(/\\/g, '/').replace(/(\.\/|\.\.\/)/g, '')}`
-            /** 判断是否为html文件且路径存在 */
-            if (!fs.existsSync(html)) {
-              return ws.send(JSON.stringify({ type: 'web', state: 'error', error: '文件不存在', echo }))
-            }
-            let content = fs.readFileSync(html, 'utf-8')
-            /** 处理所有绝对路径、相对路径 */
-            content = content.replace(new RegExp(`(${process.cwd()}|${process.cwd().replace(/\\/g, '/')})`, 'g'), '')
-            // 保存到本地
-            // filePath = './1.html'
-            // fs.writeFileSync(filePath, content, 'utf-8')
-            return ws.send(JSON.stringify({
-              type: 'web',
-              path: data.path,
-              command: 'resource',
-              data: Buffer.from(content),
-              state: 'complete',
-              part: 0,
-              echo,
-            }))
-          }
-          logger.info(`获取网页文件数据:${filePath}`)
-          // 获取文件
-          const stream = fs.createReadStream(filePath, { highWaterMark: chunkSize })
-          let part = 0
-          stream.on('data', chunk => {
-            part++
-            const message = {
-              type: 'web',
-              path: data.path,
-              command: 'resource',
-              data: chunk,
-              state: 'part',
-              part,
-              echo,
-            }
-            ws.send(JSON.stringify(message))
-          })
-          stream.on('end', () => {
-            part++
-            // 如果是最后一片段，则更新状态为 'complete'
-            if (stream.readableEnded) {
-              ws.send(JSON.stringify({
-                type: 'web',
-                path: data.path,
-                command: 'resource',
-                data: '',
-                state: 'complete',
-                part,
-                echo,
-              }))
-            }
-          })
-          stream.on('error', err => {
-            ws.send(JSON.stringify({ type: 'web', state: 'error', error: err.message, echo }))
-          })
-        } else {
-          ws.send(JSON.stringify({ type: 'web', state: 'error', error: '错误的文件路径', echo }))
-        }
-        break
-      default:
-        logger.warn(`未知消息类型${JSON.stringify(data)}`)
-        break
-    }
-  })
-  ws.on('close', function close () {
-    /** 卸载渲染器 */
-    index && render.unapp(index)
-    index = 0
-    if (heartbeat) {
-      clearInterval(heartbeat)
-      heartbeat = null
-    }
-    logger.warn('连接关闭，10秒后尝试重新连接')
-    if (!reConnect) {
-      reConnect = setTimeout(connect, 10000)
-    }
-  })
-  ws.on('error', function error () {
-    /** 卸载渲染器 */
-    index && render.unapp(index)
-    index = 0
-    if (heartbeat) {
-      clearInterval(heartbeat)
-      heartbeat = null
-    }
-    logger.warn('连接错误，10秒后尝试重新连接')
-    if (!reConnect) {
-      reConnect = setTimeout(connect, 10000)
-    }
-  })
-}
+import WebSocket from 'ws'
+import fs from 'fs'
+import path from 'path'
+import { URL } from 'url'
+import HttpRenderer from './http.js'
+import config from '../utils/config.js'
+import { render } from '../index.js'
+let ws
+let reConnect
+const chunkSize = 1024 * 1024 * 3 // 文件分片大小
+export default function connect () {
+  let heartbeat
+  let index = 0
+  reConnect = undefined
+  const wsUrl = config.Server.HttpRender.WormholeClient
+  ws = new WebSocket(wsUrl)
+  ws.on('open', function open () {
+    logger.info('连接到wormhole服务器' + wsUrl)
+    // 发送心跳
+    heartbeat = setInterval(() => {
+      ws.send(JSON.stringify({ type: 'heartbeat', date: new Date() }))
+    }, 30000) // 每30秒发送一次心跳
+  })
+  ws.on('message', msg => {
+    let data = ''
+    try {
+      data = JSON.parse(msg.toString())
+    } catch (error) {
+      logger.warn(`收到非法消息${data}`)
+    }
+    const echo = data.echo
+    switch (data.type) {
+      case 'msg': {
+        const { post, token, WormholeClient } = config.Server.HttpRender
+        const parsedUrl = new URL(WormholeClient)
+        const { hostname, port } = parsedUrl
+        const ishttps = WormholeClient.includes('wss://')
+        const host = `${ishttps ? 'https' : 'http'}://${hostname}${port ? `:${port}` : ''}/web/${data.date}`
+        logger.mark(`web渲染器已连接，地址：${host}`)
+        /** 注册渲染器 */
+        const rd = new HttpRenderer(host, post, token)
+        index = render.app({ id: 'puppeteer', type: 'image', render: rd.render.bind(rd) })
+        break
+      }
+      case 'web':
+        if (data.path) {
+          const filePath = data.path
+          const query = data.query
+          if (query.html) {
+            ws.send(JSON.stringify({ type: 'web', command: 'redirect', path: filePath, target: query.html.startsWith('/') ? query.html.slice(1) : query.html, echo }))
+            return
+          }
+          const list = ['.css', '.html', '.ttf', '.jpg', '.jpeg', '.png', '.gif', '.bmp', '.ico', '.woff', '.woff2']
+          if (!list.some(ext => path.extname(filePath).endsWith(ext))) {
+            logger.warn(`拦截非资源文件${filePath}`)
+            ws.send(JSON.stringify({ type: 'web', state: 'error', error: '非资源文件', echo }))
+            return
+          }
+          /** 判断一下是否为html 如果是需要特殊处理 */
+          if (path.extname(filePath) === '.html') {
+            let html = filePath
+            /** 获取文件路径 对路径进行处理，去掉../、./ */
+            html = `./${html.replace(/\\/g, '/').replace(/(\.\/|\.\.\/)/g, '')}`
+            /** 判断是否为html文件且路径存在 */
+            if (!fs.existsSync(html)) {
+              return ws.send(JSON.stringify({ type: 'web', state: 'error', error: '文件不存在', echo }))
+            }
+            let content = fs.readFileSync(html, 'utf-8')
+            /** 处理所有绝对路径、相对路径 */
+            content = content.replace(new RegExp(`(${process.cwd()}|${process.cwd().replace(/\\/g, '/')})`, 'g'), '')
+            // 保存到本地
+            // filePath = './1.html'
+            // fs.writeFileSync(filePath, content, 'utf-8')
+            return ws.send(JSON.stringify({
+              type: 'web',
+              path: data.path,
+              command: 'resource',
+              data: Buffer.from(content),
+              state: 'complete',
+              part: 0,
+              echo,
+            }))
+          }
+          logger.info(`获取网页文件数据:${filePath}`)
+          // 获取文件
+          const stream = fs.createReadStream(filePath, { highWaterMark: chunkSize })
+          let part = 0
+          stream.on('data', chunk => {
+            part++
+            const message = {
+              type: 'web',
+              path: data.path,
+              command: 'resource',
+              data: chunk,
+              state: 'part',
+              part,
+              echo,
+            }
+            ws.send(JSON.stringify(message))
+          })
+          stream.on('end', () => {
+            part++
+            // 如果是最后一片段，则更新状态为 'complete'
+            if (stream.readableEnded) {
+              ws.send(JSON.stringify({
+                type: 'web',
+                path: data.path,
+                command: 'resource',
+                data: '',
+                state: 'complete',
+                part,
+                echo,
+              }))
+            }
+          })
+          stream.on('error', err => {
+            ws.send(JSON.stringify({ type: 'web', state: 'error', error: err.message, echo }))
+          })
+        } else {
+          ws.send(JSON.stringify({ type: 'web', state: 'error', error: '错误的文件路径', echo }))
+        }
+        break
+      default:
+        logger.warn(`未知消息类型${JSON.stringify(data)}`)
+        break
+    }
+  })
+  ws.on('close', function close () {
+    /** 卸载渲染器 */
+    index && render.unapp(index)
+    index = 0
+    if (heartbeat) {
+      clearInterval(heartbeat)
+      heartbeat = null
+    }
+    logger.warn('连接关闭，10秒后尝试重新连接')
+    if (!reConnect) {
+      reConnect = setTimeout(connect, 10000)
+    }
+  })
+  ws.on('error', function error () {
+    /** 卸载渲染器 */
+    index && render.unapp(index)
+    index = 0
+    if (heartbeat) {
+      clearInterval(heartbeat)
+      heartbeat = null
+    }
+    logger.warn('连接错误，10秒后尝试重新连接')
+    if (!reConnect) {
+      reConnect = setTimeout(connect, 10000)
+    }
+  })
+}