feat: Support completion generating #41

Merged
awiteb merged 2 commits from awiteb/fix-20 into master 2024-05-10 13:02:42 +02:00 AGit
2 changed files with 46 additions and 2 deletions
Showing only changes of commit f022574631 - Show all commits

View file

@ -0,0 +1,37 @@
// Lprs - A local CLI vault manager
// Copyright (C) 2024 Awiteb <a@4rs.nl>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/gpl-3.0.html>.
use std::io;
use clap::{Args, CommandFactory};
use crate::{cli::Cli, vault::Vaults, LprsCommand, LprsResult};
/// Generate shell auto completion
#[derive(Debug, Args)]
pub struct Completion {
/// Shell to generate completion for
shell: clap_complete::Shell,
}
impl LprsCommand for Completion {
fn run(self, _vault_manager: Vaults) -> LprsResult<()> {
clap_complete::generate(self.shell, &mut Cli::command(), "lprs", &mut io::stdout());
Ok(())
}
}

View file

@ -25,6 +25,8 @@ use crate::{impl_commands, utils, vault::Vaults, LprsCommand, LprsResult};
pub mod add_command; pub mod add_command;
/// Clean command, used to clean the vaults file (remove all vaults) /// Clean command, used to clean the vaults file (remove all vaults)
pub mod clean_command; pub mod clean_command;
/// Generate shell completion
pub mod completion_command;
/// Edit command, used to edit the vault content /// Edit command, used to edit the vault content
pub mod edit_command; pub mod edit_command;
/// Export command, used to export the vaults /// Export command, used to export the vaults
@ -65,9 +67,11 @@ pub enum Commands {
Export(export_command::Export), Export(export_command::Export),
/// Import vaults /// Import vaults
Import(import_command::Import), Import(import_command::Import),
/// Generate shell completion
Completion(completion_command::Completion),
} }
impl_commands!(Commands, Add Remove List Clean Edit Gen Get Export Import); impl_commands!(Commands, Add Remove List Clean Edit Gen Get Export Import Completion);
#[derive(Parser, Debug)] #[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)] #[command(author, version, about, long_about = None)]
@ -116,7 +120,10 @@ impl Cli {
self.command.validate_args()?; self.command.validate_args()?;
let vault_manager = if matches!(self.command, Commands::Clean(..) | Commands::Gen(..)) { let vault_manager = if matches!(
self.command,
Commands::Clean(..) | Commands::Gen(..) | Commands::Completion(..)
) {
log::info!("Running command that don't need the vault manager"); log::info!("Running command that don't need the vault manager");
// Returns empty vault manager for those commands don't need it // Returns empty vault manager for those commands don't need it
Vaults { Vaults {