v/vlib/cli/command.v

289 lines
6.3 KiB
V
Raw Normal View History

2019-11-21 13:03:12 +01:00
module cli
type FnCommandCallback fn (cmd Command)?
pub fn (f FnCommandCallback) str() string {
return 'FnCommandCallback=>' + ptr_str(f)
}
2020-04-16 14:50:04 +02:00
2019-11-21 13:03:12 +01:00
pub struct Command {
pub mut:
name string
usage string
description string
version string
pre_execute FnCommandCallback
execute FnCommandCallback
post_execute FnCommandCallback
disable_help bool
2019-11-21 13:03:12 +01:00
disable_version bool
disable_flags bool
sort_flags bool = true
sort_commands bool = true
parent &Command = 0
commands []Command
flags []Flag
args []string
}
pub fn (cmd Command) str() string {
mut res := []string{}
res << 'Command{'
res << ' name: "$cmd.name"'
res << ' usage: "$cmd.usage"'
res << ' version: "$cmd.version"'
res << ' description: "$cmd.description"'
res << ' disable_help: $cmd.disable_help'
res << ' disable_flags: $cmd.disable_flags'
res << ' disable_version: $cmd.disable_version'
res << ' sort_flags: $cmd.sort_flags'
res << ' sort_commands: $cmd.sort_commands'
res << ' cb execute: $cmd.execute'
res << ' cb pre_execute: $cmd.pre_execute'
res << ' cb post_execute: $cmd.post_execute'
if cmd.parent == 0 {
res << ' parent: &Command(0)'
} else {
res << ' parent: &Command{$cmd.parent.name ...}'
}
res << ' commands: $cmd.commands'
res << ' flags: $cmd.flags'
res << ' args: $cmd.args'
res << '}'
return res.join('\n')
}
2019-11-21 13:03:12 +01:00
pub fn (cmd Command) is_root() bool {
return isnil(cmd.parent)
}
pub fn (cmd Command) root() Command {
if cmd.is_root() {
return cmd
}
return cmd.parent.root()
2019-11-21 13:03:12 +01:00
}
pub fn (cmd Command) full_name() string {
if cmd.is_root() {
2019-11-21 13:03:12 +01:00
return cmd.name
}
return cmd.parent.full_name() + ' $cmd.name'
2019-11-21 13:03:12 +01:00
}
pub fn (mut cmd Command) add_commands(commands []Command) {
for command in commands {
cmd.add_command(command)
2019-11-21 13:03:12 +01:00
}
}
2020-05-17 13:51:18 +02:00
pub fn (mut cmd Command) add_command(command Command) {
for existing_cmd in cmd.commands {
if existing_cmd.name == command.name {
println("command with the same name \'$command.name\' already exists")
exit(1)
}
}
2019-11-21 13:03:12 +01:00
cmd.commands << command
}
pub fn (mut cmd Command) add_flags(flags []Flag) {
for flag in flags {
cmd.add_flag(flag)
}
}
2020-05-17 13:51:18 +02:00
pub fn (mut cmd Command) add_flag(flag Flag) {
for existing_flag in cmd.flags {
if existing_flag.name == flag.name {
println("flag with the same name \'$flag.name\' already exists")
exit(1)
}
}
2019-11-21 13:03:12 +01:00
cmd.flags << flag
}
2020-05-17 13:51:18 +02:00
pub fn (mut cmd Command) parse(args []string) {
if !cmd.disable_flags {
cmd.add_default_flags()
}
2019-11-21 13:03:12 +01:00
cmd.add_default_commands()
if cmd.sort_flags {
cmd.flags.sort(a.name < b.name)
}
if cmd.sort_commands {
cmd.commands.sort(a.name < b.name)
}
2019-11-30 10:37:34 +01:00
cmd.args = args[1..]
for i in 0 .. cmd.commands.len {
2019-11-21 13:03:12 +01:00
cmd.commands[i].parent = cmd
}
if !cmd.disable_flags {
cmd.parse_flags()
}
2019-11-21 13:03:12 +01:00
cmd.parse_commands()
}
2020-05-17 13:51:18 +02:00
fn (mut cmd Command) add_default_flags() {
if !cmd.disable_help && !cmd.flags.contains('help') {
cmd.add_flag(help_flag(!cmd.flags.contains('h') && cmd.has_abbrev_flags()))
2019-11-21 13:03:12 +01:00
}
if cmd.version != '' && !cmd.flags.contains('version') {
cmd.add_flag(version_flag(!cmd.flags.contains('v') && cmd.has_abbrev_flags()))
2019-11-21 13:03:12 +01:00
}
}
2020-05-17 13:51:18 +02:00
fn (mut cmd Command) add_default_commands() {
if !cmd.disable_help && !cmd.commands.contains('help') && cmd.is_root() {
2019-11-21 13:03:12 +01:00
cmd.add_command(help_cmd())
}
if cmd.version != '' && !cmd.commands.contains('version') {
2019-11-21 13:03:12 +01:00
cmd.add_command(version_cmd())
}
}
2020-05-17 13:51:18 +02:00
fn (mut cmd Command) parse_flags() {
2019-11-21 13:03:12 +01:00
for {
if cmd.args.len < 1 || !cmd.args[0].starts_with('-') {
break
}
mut found := false
for i in 0 .. cmd.flags.len {
2019-11-21 13:03:12 +01:00
mut flag := &cmd.flags[i]
if flag.matches(cmd.args, cmd.has_abbrev_flags()) {
2019-11-21 13:03:12 +01:00
found = true
flag.found = true
cmd.args = flag.parse(cmd.args, cmd.has_abbrev_flags()) or {
println('failed to parse flag ${cmd.args[0]}: $err')
2019-11-21 13:03:12 +01:00
exit(1)
}
break
}
}
if !found {
println('invalid flag: ${cmd.args[0]}')
exit(1)
}
}
}
fn (mut cmd Command) parse_commands() {
global_flags := cmd.flags.filter(it.global)
2019-11-21 13:03:12 +01:00
cmd.check_help_flag()
cmd.check_version_flag()
for i in 0 .. cmd.args.len {
2019-11-21 13:03:12 +01:00
arg := cmd.args[i]
for j in 0 .. cmd.commands.len {
2019-11-21 13:03:12 +01:00
mut command := cmd.commands[j]
if command.name == arg {
for flag in global_flags {
command.add_flag(flag)
}
2019-11-30 10:37:34 +01:00
command.parse(cmd.args[i..])
2019-11-21 13:03:12 +01:00
return
}
}
}
// if no further command was found, execute current command
2019-11-21 13:03:12 +01:00
if int(cmd.execute) == 0 {
if !cmd.disable_help {
cmd.execute_help()
2019-11-21 13:03:12 +01:00
}
} else {
cmd.check_required_flags()
2020-03-10 16:11:17 +01:00
if int(cmd.pre_execute) > 0 {
cmd.pre_execute(*cmd) or {
println('cli preexecution error: $err')
exit(1)
}
}
cmd.execute(*cmd) or {
println('cli execution error: $err')
exit(1)
2020-03-10 16:11:17 +01:00
}
if int(cmd.post_execute) > 0 {
cmd.post_execute(*cmd) or {
println('cli postexecution error: $err')
exit(1)
}
2020-03-10 16:11:17 +01:00
}
2019-11-21 13:03:12 +01:00
}
}
fn (cmd Command) has_abbrev_flags() bool {
mut has_abbrev := false
for flag in cmd.flags {
if flag.abbrev != '' {
has_abbrev = true
}
2019-11-21 13:03:12 +01:00
}
return has_abbrev
}
fn (cmd Command) check_help_flag() {
if !cmd.disable_help && cmd.flags.contains('help') {
help_flag := cmd.flags.get_bool('help') or {
return
} // ignore error and handle command normally
2019-11-21 13:03:12 +01:00
if help_flag {
cmd.execute_help()
2019-11-21 13:03:12 +01:00
exit(0)
}
}
}
fn (cmd Command) check_version_flag() {
2019-11-21 13:03:12 +01:00
if cmd.version != '' && cmd.flags.contains('version') {
version_flag := cmd.flags.get_bool('version') or {
return
} // ignore error and handle command normally
2019-11-21 13:03:12 +01:00
if version_flag {
version_cmd := cmd.commands.get('version') or {
return
} // ignore error and handle command normally
version_cmd.execute(version_cmd)
2019-11-21 13:03:12 +01:00
exit(0)
}
}
}
fn (cmd Command) check_required_flags() {
2019-11-21 13:03:12 +01:00
for flag in cmd.flags {
if flag.required && flag.value == '' {
full_name := cmd.full_name()
println("flag \'$flag.name\' is required by \'$full_name\'")
2019-11-21 13:03:12 +01:00
exit(1)
}
}
}
fn (cmd Command) execute_help() {
if cmd.commands.contains('help') {
help_cmd := cmd.commands.get('help') or {
return
} // ignore error and handle command normally
help_cmd.execute(help_cmd)
} else {
print(cmd.help_message())
}
}
fn (cmds []Command) get(name string) ?Command {
2019-11-21 13:03:12 +01:00
for cmd in cmds {
if cmd.name == name {
return cmd
2019-11-21 13:03:12 +01:00
}
}
return error("command \'$name\' not found.")
2019-11-21 13:03:12 +01:00
}
fn (cmds []Command) contains(name string) bool {
2019-11-21 13:03:12 +01:00
for cmd in cmds {
if cmd.name == name {
return true
2019-11-21 13:03:12 +01:00
}
}
return false
}