module cli import term import strings const ( base_indent_len = 2 min_description_indent_len = 20 spacing = 2 ) fn help_flag(with_abbrev bool) Flag { sabbrev := if with_abbrev { 'h' } else { '' } return Flag{ flag: .bool name: 'help' abbrev: sabbrev description: 'Prints help information.' } } fn help_cmd() Command { return Command{ name: 'help' usage: '' description: 'Prints help information.' execute: print_help_for_command } } fn print_help_for_command(help_cmd Command) ? { if help_cmd.args.len > 0 { mut cmd := help_cmd.parent for arg in help_cmd.args { mut found := false for sub_cmd in cmd.commands { if sub_cmd.name == arg { cmd = &sub_cmd found = true break } } if !found { args := help_cmd.args.join(' ') println('Invalid command: $args') return } } print(cmd.help_message()) } else { if help_cmd.parent != 0 { print(help_cmd.parent.help_message()) } } } fn (cmd Command) help_message() string { mut help := '' help += 'Usage: $cmd.full_name()' if cmd.flags.len > 0 { help += ' [flags]' } if cmd.commands.len > 0 { help += ' [commands]' } if cmd.usage.len > 0 { help += ' $cmd.usage' } else { for i in 0 .. cmd.required_args { help += ' ' } } help += '\n' if cmd.description != '' { help += '\n$cmd.description\n' } mut abbrev_len := 0 mut name_len := min_description_indent_len if cmd.flags.have_abbrev() { for flag in cmd.flags { abbrev_len = max(abbrev_len, flag.abbrev.len + spacing + 1) // + 1 for '-' in front name_len = max(name_len, abbrev_len + flag.name.len + spacing + 2) // + 2 for '--' in front } for command in cmd.commands { name_len = max(name_len, command.name.len + spacing) } } else { for flag in cmd.flags { name_len = max(name_len, abbrev_len + flag.name.len + spacing + 1) // + 1 for '-' in front } for command in cmd.commands { name_len = max(name_len, command.name.len + spacing) } } if cmd.flags.len > 0 { help += '\nFlags:\n' for flag in cmd.flags { mut flag_name := '' if flag.abbrev != '' && cmd.flags.have_abbrev() { abbrev_indent := ' '.repeat(abbrev_len - flag.abbrev.len - 1) // - 1 for '-' in front flag_name = '-$flag.abbrev$abbrev_indent--$flag.name' } else if cmd.flags.have_abbrev() { abbrev_indent := ' '.repeat(abbrev_len) flag_name = '$abbrev_indent--$flag.name' } else { flag_name = '-$flag.name' } mut required := '' if flag.required { required = ' (required)' } base_indent := ' '.repeat(base_indent_len) description_indent := ' '.repeat(name_len - flag_name.len) help += '$base_indent$flag_name$description_indent' + pretty_description(flag.description + required, base_indent_len + name_len) + '\n' } } if cmd.commands.len > 0 { help += '\nCommands:\n' for command in cmd.commands { base_indent := ' '.repeat(base_indent_len) description_indent := ' '.repeat(name_len - command.name.len) help += '$base_indent$command.name$description_indent' + pretty_description(command.description, name_len) + '\n' } } return help } // pretty_description resizes description text depending on terminal width. // Essentially, smart wrap-around fn pretty_description(s string, indent_len int) string { width, _ := term.get_terminal_size() // Don't prettify if the terminal is that small, it won't be pretty anyway. if indent_len > width { return s } indent := ' '.repeat(indent_len) chars_per_line := width - indent_len // Give us enough room, better a little bigger than smaller mut acc := strings.new_builder(((s.len / chars_per_line) + 1) * (width + 1)) for k, line in s.split('\n') { if k != 0 { acc.write('\n$indent') } mut i := chars_per_line - 2 mut j := 0 for ; i < line.len; i += chars_per_line - 2 { for line[i] != ` ` { i-- } // indent was already done the first iteration if j != 0 { acc.write(indent) } acc.writeln(line[j..i].trim_space()) j = i } // We need this even though it should never happen if j != 0 { acc.write(indent) } acc.write(line[j..].trim_space()) } return acc.str() } fn max(a int, b int) int { res := if a > b { a } else { b } return res }