refa: move browser loader to @koishijs/play

packages/koishi/package.json

@@ -51,7 +51,7 @@
   },
   "dependencies": {
     "@koishijs/core": "4.11.6",
-    "@koishijs/loader": "2.5.6",
+    "@koishijs/loader": "2.6.1",
     "@koishijs/utils": "^7.0.0",
     "@satorijs/satori": "^2.1.2",
     "cac": "^6.7.14",

packages/loader/package.json

@@ -1,21 +1,10 @@
 {
   "name": "@koishijs/loader",
   "description": "Config Loader for Koishi",
-  "version": "2.5.6",
-  "main": "lib/node.js",
+  "version": "2.6.1",
+  "main": "lib/index.js",
+  "module": "lib/shared.mjs",
   "typings": "lib/index.d.ts",
-  "exports": {
-    ".": {
-      "node": "./lib/node.js",
-      "browser": "./lib/browser.mjs"
-    },
-    "./shared": {
-      "require": "./lib/shared.js",
-      "import": "./lib/shared.mjs"
-    },
-    "./src/*": "./src/*",
-    "./package.json": "./package.json"
-  },
   "files": [
     "lib"
   ],

packages/loader/src/index.ts

@@ -1,5 +1,128 @@
-// placeholder file, do not modify
-import Loader from './node'
+import { accessSync, constants, readdirSync, readFileSync, statSync, writeFileSync } from 'fs'
+import { dirname, extname, resolve } from 'path'
+import { Context, Logger } from '@koishijs/core'
+import { Loader, unwrapExports } from './shared'
+import * as dotenv from 'dotenv'
+import * as yaml from 'js-yaml'
+import ns from 'ns-require'
 
-export default Loader
-export * from './node'
+export * from './shared'
+
+const logger = new Logger('app')
+
+const writableExts = ['.json', '.yml', '.yaml']
+const supportedExts = ['.js', '.json', '.ts', '.coffee', '.yaml', '.yml']
+
+export default class NodeLoader extends Loader {
+  public envData = JSON.parse(process.env.KOISHI_SHARED || '{}')
+  public baseDir = process.cwd()
+  public extname: string
+  public scope: ns.Scope
+  public ctxData = {
+    env: process.env,
+  }
+
+  constructor(filename?: string) {
+    super()
+    if (filename) {
+      filename = resolve(this.baseDir, filename)
+      const stats = statSync(filename)
+      if (stats.isFile()) {
+        this.filename = filename
+        this.baseDir = dirname(filename)
+        this.extname = extname(filename)
+        if (!supportedExts.includes(this.extname)) {
+          throw new Error('extension not supported')
+        }
+      } else {
+        this.baseDir = filename
+        this.findConfig()
+      }
+    } else {
+      this.findConfig()
+    }
+    this.writable = this.checkWritable()
+    this.envfile = resolve(this.baseDir, '.env')
+    this.scope = ns({
+      namespace: 'koishi',
+      prefix: 'plugin',
+      official: 'koishijs',
+      dirname: this.baseDir,
+    })
+  }
+
+  private checkWritable() {
+    if (!writableExts.includes(this.extname)) return false
+    try {
+      accessSync(this.filename, constants.W_OK)
+      return true
+    } catch {
+      return false
+    }
+  }
+
+  private findConfig() {
+    const files = readdirSync(this.baseDir)
+    for (const basename of ['koishi.config', 'koishi']) {
+      for (const extname of supportedExts) {
+        if (files.includes(basename + extname)) {
+          this.extname = extname
+          this.filename = resolve(this.baseDir, basename + extname)
+          return
+        }
+      }
+    }
+    throw new Error('config file not found')
+  }
+
+  readConfig() {
+    // load .env file into process.env
+    dotenv.config({ path: this.envfile })
+
+    if (['.yaml', '.yml'].includes(this.extname)) {
+      this.config = yaml.load(readFileSync(this.filename, 'utf8')) as any
+    } else if (['.json'].includes(this.extname)) {
+      // we do not use require here because it will pollute require.cache
+      this.config = JSON.parse(readFileSync(this.filename, 'utf8')) as any
+    } else {
+      const module = require(this.filename)
+      this.config = module.default || module
+    }
+
+    return new Context.Config(this.interpolate(this.config))
+  }
+
+  writeConfig() {
+    this.app.emit('config')
+    this.suspend = true
+    if (!this.writable) throw new Error('cannot overwrite readonly config')
+    if (this.extname === '.json') {
+      writeFileSync(this.filename, JSON.stringify(this.config, null, 2))
+    } else {
+      writeFileSync(this.filename, yaml.dump(this.config))
+    }
+  }
+
+  async resolve(name: string) {
+    return this.scope.resolve(name)
+  }
+
+  async resolvePlugin(name: string) {
+    try {
+      this.cache[name] ||= this.scope.resolve(name)
+    } catch (err) {
+      logger.error(err.message)
+      return
+    }
+    return unwrapExports(require(this.cache[name]))
+  }
+
+  fullReload(code = Loader.exitCode) {
+    const body = JSON.stringify(this.envData)
+    process.send({ type: 'shared', body }, (err: any) => {
+      if (err) logger.error('failed to send shared data')
+      logger.info('trigger full reload')
+      process.exit(code)
+    })
+  }
+}