1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
use std::io::Write;
use crate::Generator;
use clap::*;
pub struct Fish;
impl Generator for Fish {
fn file_name(name: &str) -> String {
format!("{}.fish", name)
}
fn generate(app: &App, buf: &mut dyn Write) {
let command = app.get_bin_name().unwrap();
let mut buffer = String::new();
gen_fish_inner(command, app, &mut buffer);
w!(buf, buffer.as_bytes());
}
}
fn escape_string(string: &str) -> String {
string.replace("\\", "\\\\").replace("'", "\\'")
}
fn gen_fish_inner(root_command: &str, app: &App, buffer: &mut String) {
debug!("gen_fish_inner");
let mut basic_template = format!("complete -c {} -n ", root_command);
let mut bin_name = app.get_bin_name().unwrap();
if root_command == bin_name {
basic_template.push_str("\"__fish_use_subcommand\"");
} else {
bin_name = &app.get_name();
basic_template.push_str(format!("\"__fish_seen_subcommand_from {}\"", bin_name).as_str());
}
debug!("gen_fish_inner: bin_name={}", bin_name);
for option in opts!(app) {
let mut template = basic_template.clone();
if let Some(data) = option.get_short() {
template.push_str(format!(" -s {}", data).as_str());
}
if let Some(data) = option.get_long() {
template.push_str(format!(" -l {}", data).as_str());
}
if let Some(data) = option.get_about() {
template.push_str(format!(" -d '{}'", escape_string(data)).as_str());
}
if let Some(ref data) = option.get_possible_values() {
template.push_str(format!(" -r -f -a \"{}\"", data.join(" ")).as_str());
}
buffer.push_str(template.as_str());
buffer.push_str("\n");
}
for flag in Fish::flags(app) {
let mut template = basic_template.clone();
if let Some(data) = flag.get_short() {
template.push_str(format!(" -s {}", data).as_str());
}
if let Some(data) = flag.get_long() {
template.push_str(format!(" -l {}", data).as_str());
}
if let Some(data) = flag.get_about() {
template.push_str(format!(" -d '{}'", escape_string(data)).as_str());
}
buffer.push_str(template.as_str());
buffer.push_str("\n");
}
for subcommand in app.get_subcommands() {
let mut template = basic_template.clone();
template.push_str(" -f");
template.push_str(format!(" -a \"{}\"", &subcommand.get_name()).as_str());
if let Some(data) = subcommand.get_about() {
template.push_str(format!(" -d '{}'", escape_string(data)).as_str())
}
buffer.push_str(template.as_str());
buffer.push_str("\n");
}
for subcommand in app.get_subcommands() {
gen_fish_inner(root_command, subcommand, buffer);
}
}