上次我们聊到 CLI 的领域交互模式。在领域交互模式中，可能存在多层次的子命令。在使用过程中如果全评记忆的话，命令少还好，多了真心记不住。频繁 –help 也是个很麻烦的事情。如果每次按 ‘tab’ 键就可以提示或补齐命令是不是很方便呢。这一节我们就来说说 ‘autocommplete’ 如何实现。我们还是以interactcli-rs中的实现来解说实现过程
实现过程
其实，rustyline 已经为我们提供了基本的helper功能框架,其中包括了completer。我们来看代码，文件位置src/interact/cli.rs
#[derive(Helper)]structMyHelper{completer:CommandCompleter,highlighter:MatchingBracketHighlighter,validator:MatchingBracketValidator,hinter:HistoryHinter,colored_prompt:String,}pubfnrun(){letconfig=Config::builder().history_ignore_space(true).completion_type(CompletionType::List).output_stream(OutputStreamType::Stdout).build();leth=MyHelper{completer:get_command_completer(),highlighter:MatchingBracketHighlighter::new(),hinter:HistoryHinter{},colored_prompt:"".to_owned(),validator:MatchingBracketValidator::new(),};letmutrl=Editor::with_config(config);//letmutrl=Editor::::new();rl.set_helper(Some(h));......}首先定义 MyHelper 结构体， 需要实现 Completer + Hinter + Highlighter + Validator trait。然后通过rustyline的set_helper函数加载我们定义好的helper。在MyHelper 结构体中，需要我们自己来实现completer的逻辑。
Sub command autocompleter实现详解
- SubCmd 结构体
#[derive(Debug,Clone)]pubstructSubCmd{publevel:usize,pubcommand_name:String,pubsubcommands:Vec,}SubCmd 结构体包含：命令级别，命令名称，以及该命令包含的子命令信息，以便在实现实现 autocomplete 时定位命令和子命令的范围
- 在程序启动时遍历所有的command，src/cmd/rootcmd.rs 中的all_subcommand函数负责收集所有命令并转换为Vec
pubfnall_subcommand(app:&clap_Command,beginlevel:usize,input:&mutVec){letnextlevel=beginlevel+1;letmutsubcmds=vec![];foriterminapp.get_subcommands(){subcmds.push(iterm.get_name().to_string());ifiterm.has_subcommands(){all_subcommand(iterm,nextlevel,input);}else{ifbeginlevel==0{all_subcommand(iterm,nextlevel,input);}}}letsubcommand=SubCmd{level:beginlevel,command_name:app.get_name().to_string(),subcommands:subcmds,};input.push(subcommand);}- CommandCompleter 子命令自动补充功能的核心部分
#[derive(Debug,Clone)]pubstructCommandCompleter{subcommands:Vec,}implCommandCompleter{pubfnnew(subcmds:Vec)->Self{Self{subcommands:subcmds,}}//获取level下所有可能的子命令pubfnlevel_possible_cmd(&self,level:usize)->Vec{letmutsubcmds=vec![];letcmds=self.subcommands.clone();foritermincmds{ifiterm.level==level{subcmds.push(iterm.command_name.clone());}}returnsubcmds;}//获取level下某字符串开头的子命令pubfnlevel_prefix_possible_cmd(&self,level:usize,prefix:&str)->Vec{letmutsubcmds=vec![];letcmds=self.subcommands.clone();foritermincmds{ifiterm.level==level&&iterm.command_name.starts_with(prefix){subcmds.push(iterm.command_name);}}returnsubcmds;}//获取某level下某subcommand的所有子命令pubfnlevel_cmd_possible_sub_cmd(&self,level:usize,cmd:String)->Vec{letmutsubcmds=vec![];letcmds=self.subcommands.clone();foritermincmds{ifiterm.level==level&&iterm.command_name==cmd{subcmds=iterm.subcommands.clone();}}returnsubcmds;}//获取某level下某subcommand的所有prefix子命令pubfnlevel_cmd_possible_prefix_sub_cmd(&self,level:usize,cmd:String,prefix:&str,)->Vec{letmutsubcmds=vec![];letcmds=self.subcommands.clone();foritermincmds{ifiterm.level==level&&iterm.command_name==cmd{foriiniterm.subcommands{ifi.starts_with(prefix){subcmds.push(i);}}}}returnsubcmds;}pubfncomplete_cmd(&self,line:&str,pos:usize)->Result<(usize,Vec)>{letmutentries:Vec=Vec::new();letd:Vec=line.split('').collect();ifd.len()==1{ifd.last()==Some(&""){forstrinself.level_possible_cmd(1){letmutreplace=str.clone();replace.push_str("");entries.push(Pair{display:str.clone(),replacement:replace,});}returnOk((pos,entries));}ifletSome(last)=d.last(){forstrinself.level_prefix_possible_cmd(1,*last){letmutreplace=str.clone();replace.push_str("");entries.push(Pair{display:str.clone(),replacement:replace,});}returnOk((pos-last.len(),entries));}}ifd.last()==Some(&""){forstrinself.level_cmd_possible_sub_cmd(d.len()-1,d.get(d.len()-2).unwrap().to_string()){letmutreplace=str.clone();replace.push_str("");entries.push(Pair{display:str.clone(),replacement:replace,});}returnOk((pos,entries));}ifletSome(last)=d.last(){forstrinself.level_cmd_possible_prefix_sub_cmd(d.len()-1,d.get(d.len()-2).unwrap().to_string(),*last,){letmutreplace=str.clone();replace.push_str("");entries.push(Pair{display:str.clone(),replacement:replace,});}returnOk((pos-last.len(),entries));}Ok((pos,entries))}}implCompleterforCommandCompleter{typeCandidate=Pair;fncomplete(&self,line:&str,pos:usize,_ctx:&Context)->Result<(usize,Vec)>{self.complete_cmd(line,pos)}}CommandCompleter 的实现部分比较多，大致包括两个部分，前一部分包括：获取某一级别下所有可能的子命令、获取某级别下某字符串开头的子命令、获取某级别下某个命令的所有子命令，等基本功能。这部分代码中有注释就不一一累述。
函数complete_cmd用来计算行中的位置以及在该位置的替换内容。
输入项是命令行的内容以及光标所在位置，输出项为在该位置需要替换的内容。比如，我们在提示符下输入 “root cm” root 下包含 cmd1、cmd2 两个子命令，此时如果按 ‘tab’键，complete_cmd 函数就会返回 (7,[cmd1,cmd2])。
作者：京东科技贾世闻