feat(eval): authority settings (#110)

koishijs · Oct 14, 2020 · ed03ed1 · ed03ed1
1 parent 9b7f3eb
commit ed03ed1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/packages/plugin-eval/src/index.ts b/packages/plugin-eval/src/index.ts
@@ -25,6 +25,7 @@ declare module 'koishi-core/dist/session' {
 
 const defaultConfig: EvalConfig = {
   prefix: '>',
+  authority: 2,
   timeout: 1000,
   setupFiles: {},
   maxLogs: Infinity,
@@ -38,7 +39,7 @@ const logger = new Logger('eval')
 export const name = 'eval'
 
 export function apply(ctx: Context, config: Config = {}) {
-  const { prefix } = config = { ...defaultConfig, ...config }
+  const { prefix, authority } = config = { ...defaultConfig, ...config }
   const { app } = ctx
   const worker = new EvalWorker(app, config)
   defineProperty(app, 'worker', worker)
@@ -59,7 +60,7 @@ export function apply(ctx: Context, config: Config = {}) {
     .option('slient', '-s  不输出最后的结果')
     .option('restart', '-r  重启子线程', { authority: 3 })
     .before((session) => {
-      if (!session['_redirected'] && session.$user?.authority < 2) return '权限不足。'
+      if (!session['_redirected'] && session.$user?.authority < authority) return '权限不足。'
     })
 
   attachTraps(cmd, config, async ({ session, options, ctxOptions }, expr) => {

diff --git a/packages/plugin-eval/src/main.ts b/packages/plugin-eval/src/main.ts
@@ -17,6 +17,7 @@ const logger = new Logger('eval')
 
 export interface MainConfig extends FieldOptions {
   prefix?: string
+  authority?: number
   timeout?: number
   maxLogs?: number
   resourceLimits?: ResourceLimits