• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1/*
2 * Copyright (c) 2023 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16import commander from 'commander';
17import envConfig from './config/env';
18import { getToolConfiguration, ToolConfigType } from './bin/index';
19import { CommandType, OptionObjType, PluginType, PluginOptionsType, toolNameType, toolNameSet } from './bin/config';
20import { LogUtil } from './utils/logUtil';
21import { FileUtils } from './utils/FileUtils';
22
23class ToolBoxCommander {
24  program: commander.Command = new commander.Command();
25  constructor() {}
26  addPluginCommand(plugin: PluginType): void {
27    const pluginOption: PluginOptionsType = plugin.pluginOptions;
28    if (!pluginOption) {
29      return;
30    }
31    const pluginCommand: commander.Command = this.program
32      .name(pluginOption.name)
33      .description(pluginOption.description)
34      .version(pluginOption.version)
35      .action((opts: OptionObjType) => {
36        this.judgeOpts(opts);
37        plugin.start(opts);
38        plugin.stop();
39      });
40    pluginOption.commands.forEach((command: CommandType) => {
41      if (command.isRequiredOption) {
42        pluginCommand.requiredOption(...command.options);
43      } else {
44        pluginCommand.option(...command.options);
45      }
46    });
47  }
48  buildCommands(): void {
49    this.program.parse();
50  }
51  /**
52   * 判断传入命令是否满足工具允许条件，满足正常允许，不满足的时候通过stopRun停止命令行执行
53   *
54   * @param {OptionObjType} opts
55   */
56  judgeOpts(opts: OptionObjType): void {
57    const toolName: string = opts.toolName;
58    if (!toolNameSet.has(toolName)) {
59      this.stopRun(`error toolName "${toolName}",toolName not in \[${[...toolNameSet]}\] `);
60    }
61    switch (toolName) {
62      case toolNameType.COOLECT:
63        const collectPath = opts.collectPath;
64        if (collectPath === '' || !FileUtils.isExists(collectPath)) {
65          this.stopRun(`error collectPath "${collectPath}",collectPath need a exist file path`);
66        }
67        break;
68    }
69  }
70  /**
71   * 停止命令行执行，输出错误信息
72   *
73   * @param {string} text
74   */
75  stopRun(text: string): void {
76    LogUtil.e('commander', text);
77    this.program.help({ error: true });
78  }
79}
80class ToolboxEntry {
81  commandBuilder: ToolBoxCommander;
82  constructor() {
83    this.commandBuilder = new ToolBoxCommander();
84  }
85  runPlugins(): void {
86    const configuration: ToolConfigType = getToolConfiguration();
87    configuration.plugins.forEach((plugin: PluginType) => {
88      this.commandBuilder.addPluginCommand(plugin);
89    });
90    this.commandBuilder.buildCommands();
91  }
92}
93
94function main(): void {
95  Object.assign(process.env, envConfig);
96  const entry = new ToolboxEntry();
97  entry.runPlugins();
98}
99
100main();
101