]>
Commit | Line | Data |
---|---|---|
1 | #!/usr/bin/env node | |
2 | ||
3 | import { Command, InvalidArgumentError } from '@commander-js/extra-typings' | |
4 | import { listRegistered, registerRunner, unregisterRunner } from './register' | |
5 | import { RunnerServer } from './server' | |
6 | import { ConfigManager, logger } from './shared' | |
7 | ||
8 | const packageJSON = require('./package.json') | |
9 | ||
10 | const program = new Command() | |
11 | .version(packageJSON.version) | |
12 | .option( | |
13 | '--id <id>', | |
14 | 'Runner server id, so you can run multiple PeerTube server runners with different configurations on the same machine', | |
15 | 'default' | |
16 | ) | |
17 | .option('--verbose', 'Run in verbose mode') | |
18 | .hook('preAction', thisCommand => { | |
19 | const options = thisCommand.opts() | |
20 | ||
21 | ConfigManager.Instance.init(options.id) | |
22 | ||
23 | if (options.verbose === true) { | |
24 | logger.level = 'debug' | |
25 | } | |
26 | }) | |
27 | ||
28 | program.command('server') | |
29 | .description('Run in server mode, to execute remote jobs of registered PeerTube instances') | |
30 | .action(async () => { | |
31 | try { | |
32 | await RunnerServer.Instance.run() | |
33 | } catch (err) { | |
34 | logger.error(err, 'Cannot run PeerTube runner as server mode') | |
35 | process.exit(-1) | |
36 | } | |
37 | }) | |
38 | ||
39 | program.command('register') | |
40 | .description('Register a new PeerTube instance to process runner jobs') | |
41 | .requiredOption('--url <url>', 'PeerTube instance URL', parseUrl) | |
42 | .requiredOption('--registration-token <token>', 'Runner registration token (can be found in PeerTube instance administration') | |
43 | .requiredOption('--runner-name <name>', 'Runner name') | |
44 | .option('--runner-description <description>', 'Runner description') | |
45 | .action(async options => { | |
46 | try { | |
47 | await registerRunner(options) | |
48 | } catch (err) { | |
49 | console.error('Cannot register this PeerTube runner.') | |
50 | console.error(err) | |
51 | process.exit(-1) | |
52 | } | |
53 | }) | |
54 | ||
55 | program.command('unregister') | |
56 | .description('Unregister the runner from PeerTube instance') | |
57 | .requiredOption('--url <url>', 'PeerTube instance URL', parseUrl) | |
58 | .action(async options => { | |
59 | try { | |
60 | await unregisterRunner(options) | |
61 | } catch (err) { | |
62 | console.error('Cannot unregister this PeerTube runner.') | |
63 | console.error(err) | |
64 | process.exit(-1) | |
65 | } | |
66 | }) | |
67 | ||
68 | program.command('list-registered') | |
69 | .description('List registered PeerTube instances') | |
70 | .action(async () => { | |
71 | try { | |
72 | await listRegistered() | |
73 | } catch (err) { | |
74 | console.error('Cannot list registered PeerTube instances.') | |
75 | console.error(err) | |
76 | process.exit(-1) | |
77 | } | |
78 | }) | |
79 | ||
80 | program.parse() | |
81 | ||
82 | // --------------------------------------------------------------------------- | |
83 | // Private | |
84 | // --------------------------------------------------------------------------- | |
85 | ||
86 | function parseUrl (url: string) { | |
87 | if (url.startsWith('http://') !== true && url.startsWith('https://') !== true) { | |
88 | throw new InvalidArgumentError('URL should start with a http:// or https://') | |
89 | } | |
90 | ||
91 | return url | |
92 | } |