feat: register hook supports

package.json

@@ -80,6 +80,9 @@
     "dirname": "ai",
     "commands": "./dist/oclif/commands",
     "helpClass": "./dist/oclif/lib/help",
+    "hooks": {
+      "init_tools": "./dist/oclif/hooks/init-tools"
+    },
     "plugins": [
       "@oclif/plugin-help",
       "@oclif/plugin-plugins",

src/lib/run-script.ts

@@ -10,7 +10,7 @@ import { ConfigFile, getMultiLevelExtname, parseJsJson, wait } from '@isdk/ai-to
 import { AIScriptServer, LogLevel, LogLevelMap } from '@isdk/ai-tool-agent'
 import { detectTextLanguage as detectLang } from '@isdk/detect-text-language'
 import { prompt, setHistoryStore, HistoryStore } from './prompt.js'
-import { initTools } from './init-tools.js'
+// import { initTools } from './init-tools.js'
 import { ux } from '@oclif/core'
 
 class AIScriptEx extends AIScriptServer {
@@ -74,7 +74,7 @@ function renameOldFile(filename: string) {
   }
 }
 export async function runScript(filename: string, options: IRunScriptOptions) {
-  initTools(options)
+  // initTools(options)
 
   const { logLevel: level, interactive, stream } = options
 

src/oclif/commands/brain/download.ts

@@ -1,7 +1,6 @@
 import path from 'path'
 import enquier from 'enquirer'
 import { Args, Flags } from '@oclif/core'
-import { initTools } from '../../../lib/init-tools.js'
 import { AICommand } from '../../lib/ai-command.js'
 import { showBanner } from '../../lib/help.js'
 import { downloadBrain, getQuantsFromBrain, listBrains } from '../../../lib/brain.js'
@@ -55,7 +54,7 @@ export default class DownloadBrainCommand extends AICommand {
     const isJson = this.jsonEnabled()
     const {args, flags} = opts
     const userConfig = this.loadConfig(flags.config, opts)
-    initTools(userConfig)
+    this.config.runHook('init_tools', {id: 'brain:download', userConfig})
 
     process.on('SIGINT', ()=>{
       process.exit(0)

src/oclif/commands/brain/index.ts

@@ -1,5 +1,4 @@
 import { Args, Flags } from '@oclif/core'
-import { initTools } from '../../../lib/init-tools.js'
 import { AICommand } from '../../lib/ai-command.js'
 import { showBanner } from '../../lib/help.js'
 import { parseJsJson } from '@isdk/ai-tool'
@@ -53,7 +52,8 @@ export default class Brain extends AICommand {
     const isJson = this.jsonEnabled()
     const {args, flags} = opts
     const userConfig = this.loadConfig(flags.config, opts)
-    initTools(userConfig)
+    this.config.runHook('init_tools', {id: 'brain', userConfig})
+
     if (flags.refresh) {
       upgradeBrains()
     }

src/oclif/commands/brain/list.ts

@@ -1,7 +1,6 @@
 // import util from 'util'
 import { parseJsJson } from '@isdk/ai-tool'
 import { listBrains, printBrains, upgradeBrains } from '../../../lib/brain.js'
-import { initTools } from '../../../lib/init-tools.js'
 import { AICommand } from '../../lib/ai-command.js'
 import { showBanner } from '../../lib/help.js'
 import { Args, Flags } from '@oclif/core'
@@ -58,7 +57,7 @@ export default class AIBrainListCommand extends AICommand {
     const isJson = this.jsonEnabled()
     const {args, flags} = opts
     const userConfig = this.loadConfig(flags.config, opts)
-    initTools(userConfig)
+    this.config.runHook('init_tools', {id: 'brain', userConfig})
 
     if (flags.refresh) {
       upgradeBrains()

src/oclif/commands/run/index.ts

@@ -27,7 +27,7 @@ export default class RunScript extends AICommand {
   static description = 'Execute ai-agent script file and return result. with `-i` to interactive.'
 
   static examples = [
-    `<%= config.bin %> <%= command.id %> -f ./script.yaml "{content: 'hello world'}"
+    `<%= config.bin %> <%= command.id %> -f ./script.yaml "{content: 'hello world'}" -l info
 ┌────────────────────
 │[info]:Start Script: ...
 `,
@@ -65,6 +65,7 @@ export default class RunScript extends AICommand {
     }
 
     try {
+      this.config.runHook('init_tools', {id: 'run', userConfig})
       let result = await runScript(script, userConfig)
       if (LogLevelMap[userConfig.logLevel] >= LogLevelMap.info && result?.content) {
         result = result.content

src/oclif/hooks/init-tools.ts

@@ -0,0 +1,10 @@
+import type { Hook, Config } from '@oclif/core'
+
+import { initTools as _initTools } from '../../lib/init-tools.js'
+
+export async function init_tools(this: Hook.Context, options: {userConfig: any, config: Config}) {
+  _initTools(options.userConfig)
+  this.config.runHook('register', {...options})
+}
+
+export default init_tools