坠星书阁

1
// 命令系统，负责管理模块和处理用户输入
2
type CommandSystem struct {
3
  currentModule  string                         //当前激活的模块，类似于操作系统当前运行的进程
4
  modules        map[string]Module              //注册的功能模块，类似于操作系统中的进程表
5
  directCommands map[string]func(ctx *zero.Ctx) //直接执行的命令，类似于操作系统的系统调用
6
}
7

8
// 创建新的命令系统实例
9
func NewCommandSystem() *CommandSystem {
10
  return &CommandSystem{
11
    currentModule:  "",
12
    modules:        make(map[string]Module),
13
    directCommands: make(map[string]func(ctx *zero.Ctx)),
14
  }
15
}
16

17
// 注册功能模块，类似于操作系统注册进程
18
func (cs *CommandSystem) RegisterModule(module Module) {
19
  cs.modules[module.Name()] = module
20
}
21

22
// 处理用户输入，类似于操作系统的中断处理
23
func (cs *CommandSystem) ProcessInput(input string, ctx *zero.Ctx) {
24
  //检查是否是命令（以/开头）
25
  if strings.HasPrefix(input, "/") {
26
    cmd := strings.TrimPrefix(input, "/")
27

28
    //检查是否是直接执行的命令
29
    if handler, exists := cs.directCommands[cmd]; exists {
30
      handler(ctx)
31
      return
32
    }
33

34
    //检查是否是退出命令
35
    if cmd == "exit" {
36
      if cs.currentModule != "" {
37
        cs.modules[cs.currentModule].Exit(ctx)
38
        cs.currentModule = ""
39
        ctx.Send("已退出当前功能系统")
40
      } else {
41
        ctx.Send("当前不在任何功能系统中")
42
      }
43
      return
44
    }
45

46
    //检查是否是帮助命令
47
    if cmd == "help" {
48
      cs.showHelp(ctx)
49
      return
50
    }
51

52
    //检查是否是有效的模块名称
53
    if module, exists := cs.modules[cmd]; exists {
54
      //退出当前模块
55
      if cs.currentModule != "" {
56
        cs.modules[cs.currentModule].Exit(ctx)
57
      }
58

59
      //进入新模块，类似于操作系统的进程切换
60
      cs.currentModule = cmd
61
      module.Enter(ctx)
62
      return
63
    } else {
64
      //不是系统命令，让plugin处理
65
      ctx.Event.RawMessage = cmd
66
      //让ZeroBot继续处理这个消息
67
      return
68
    }
69
  }
70

71
  //在当前模块中处理命令，类似于进程执行用户空间代码
72
  if cs.currentModule != "" {
73
    module := cs.modules[cs.currentModule]
74

75
    //解析输入为命令和参数
76
    parts := strings.Fields(input)
77
    if len(parts) > 0 {
78
      cmd := parts[0]
79
      args := parts[1:]
80

81
      //处理命令，类似于进程处理系统调用
82
      handled := module.HandleCommand(cmd, args, ctx)
83
      if !handled {
84
        ctx.Send("未知的命令，请输入 /help 查看帮助")
85
      }
86
    } else {
87
      ctx.Send("请输入命令，或输入 /exit 退出当前功能系统")
88
    }
89
    return
90
  }
91
}

1
type Module interface {
2
    //返回模块名称，类似于进程的唯一标识符
3
    Name() string
4

5
    //处理模块内的命令，类似于进程处理系统调用
6
    //返回true表示命令已处理，false表示命令未处理
7
    HandleCommand(cmd string, args []string, ctx *zero.Ctx) bool
8

9
    //进入模块时执行的操作，类似于进程的初始化
10
    Enter(ctx *zero.Ctx)
11

12
    //退出模块时执行的操作，类似于进程的清理
13
    Exit(ctx *zero.Ctx)
14
}