Compare commits

...

3 Commits

Author SHA1 Message Date
Nathan Fisher 888cc3272b Fix `bootstrap` for all commands 2023-02-13 10:38:02 -05:00
Nathan Fisher 423592973f Revert "Fix bootstrap for all three binaries"
This reverts commit 0450cdfee6.
2023-02-13 10:32:36 -05:00
Nathan Fisher 0450cdfee6 Fix bootstrap for all three binaries 2023-02-13 10:30:46 -05:00
5 changed files with 850 additions and 28 deletions

View File

@ -162,7 +162,7 @@ impl Cmd for Bootstrap {
fs::create_dir_all(&outpath)?;
println!(" mkdir: {}", outpath.display());
}
outpath.push(env!("CARGO_PKG_NAME"));
outpath.push(&shitbox::progname().unwrap());
fs::copy(&progpath, &outpath)?;
println!(
" install: {} -> {}",
@ -175,15 +175,15 @@ impl Cmd for Bootstrap {
links(prefix, usr, matches)?;
}
Some(("manpages", _matches)) => {
manpages(prefix)?;
manpages(prefix, usr)?;
}
Some(("completions", matches)) => {
completions(prefix, matches)?;
completions(prefix, matches, usr)?;
}
Some(("all", matches)) => {
links(prefix, usr, matches)?;
manpages(prefix)?;
completions(prefix, matches)?;
manpages(prefix, usr)?;
completions(prefix, matches, usr)?;
}
_ => {}
}
@ -201,7 +201,7 @@ pub trait BootstrapCmd {
fn completion(&self, outdir: &Path, gen: &str) -> Result<(), io::Error>;
fn linkpath(&self, prefix: &str, usr: bool) -> Option<PathBuf>;
fn link(&self, prefix: &str, usr: bool, soft: bool) -> Result<(), Box<dyn Error>>;
fn manpage(&self, prefix: &str) -> Result<(), io::Error>;
fn manpage(&self, prefix: &str, usr: bool) -> Result<(), io::Error>;
}
impl BootstrapCmd for dyn Cmd {
@ -237,31 +237,33 @@ impl BootstrapCmd for dyn Cmd {
fn link(&self, prefix: &str, usr: bool, soft: bool) -> Result<(), Box<dyn Error>> {
if let Some(linkpath) = self.linkpath(prefix, usr) {
let progname = shitbox::progname().unwrap();
if soft {
let binpath = match self.path().unwrap() {
shitbox::Path::Bin => "shitbox",
shitbox::Path::Sbin => "../bin/shitbox",
shitbox::Path::Bin => progname,
shitbox::Path::Sbin => format!("../bin/{progname}"),
shitbox::Path::UsrBin => {
if usr {
"../../bin/shitbox"
format!("../../bin/{progname}")
} else {
"shitbox"
progname
}
}
shitbox::Path::UsrSbin => {
if usr {
"../../bin/shitbox"
format!("../../bin/{progname}")
} else {
"../bin/shitbox"
format!("../bin/{progname}")
}
}
};
symlink(binpath, &linkpath)?;
symlink(&binpath, &linkpath)?;
println!(" symlink: {binpath} -> {}", linkpath.display());
} else {
let mut binpath = PathBuf::from(prefix);
binpath.push("bin");
binpath.push(env!("CARGO_PKG_NAME"));
println!("Generating link for {}", self.cli().get_name());
fs::hard_link(&binpath, &linkpath)?;
println!(" link: {} -> {}", binpath.display(), linkpath.display());
}
@ -269,13 +271,17 @@ impl BootstrapCmd for dyn Cmd {
Ok(())
}
fn manpage(&self, prefix: &str) -> Result<(), io::Error> {
fn manpage(&self, prefix: &str, usr: bool) -> Result<(), io::Error> {
let command = self.cli();
let fname = match command.get_name() {
"bootstrap" => "shitbox-bootstrap.1".to_string(),
"bootstrap" => format!("{}-bootstrap.1", shitbox::progname().unwrap()),
s => format!("{s}.1"),
};
let outdir: PathBuf = [prefix, "usr", "share", "man", "man1"].iter().collect();
let outdir: PathBuf = if usr {
[prefix, "usr", "share", "man", "man1"].iter().collect()
} else {
[prefix, "share", "man", "man1"].iter().collect()
};
if !outdir.exists() {
fs::create_dir_all(&outdir)?;
}
@ -290,48 +296,61 @@ impl BootstrapCmd for dyn Cmd {
}
}
fn manpages(prefix: &str) -> Result<(), io::Error> {
fn manpages(prefix: &str, usr: bool) -> Result<(), io::Error> {
println!("Generating Unix man pages:");
COMMANDS
.iter()
.try_for_each(|cmd| super::get(cmd).unwrap().manpage(prefix))?;
.try_for_each(|cmd| super::get(cmd).unwrap().manpage(prefix, usr))?;
Ok(())
}
fn completions(prefix: &str, matches: &ArgMatches) -> Result<(), io::Error> {
fn completions(prefix: &str, matches: &ArgMatches, usr: bool) -> Result<(), io::Error> {
println!("Generating completions:");
let basedir: PathBuf = if usr {
[prefix, "usr", "share"].iter().collect()
} else {
[prefix, "share"].iter().collect()
};
if matches.get_flag("bash") || matches.get_flag("all") {
let outdir: PathBuf = [prefix, "share", "bash-completion", "completion"]
.iter()
.collect();
let mut outdir = basedir.clone();
outdir.push("bash-completion");
outdir.push("completion");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "bash")
})?;
}
if matches.get_flag("fish") || matches.get_flag("all") {
let outdir: PathBuf = [prefix, "share", "fish", "completions"].iter().collect();
let mut outdir = basedir.clone();
outdir.push("fish");
outdir.push("completions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "fish")
})?;
}
if matches.get_flag("nu") || matches.get_flag("all") {
let outdir: PathBuf = [prefix, "share", "nu", "completions"].iter().collect();
let mut outdir = basedir.clone();
outdir.push("nu");
outdir.push("completions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "nu")
})?;
}
if matches.get_flag("pwsh") || matches.get_flag("all") {
let outdir: PathBuf = [prefix, "share", "pwsh", "completions"].iter().collect();
let mut outdir = basedir.clone();
outdir.push("pwsh");
outdir.push("completions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "pwsh")
})?;
}
if matches.get_flag("zsh") || matches.get_flag("all") {
let outdir: PathBuf = [prefix, "share", "zsh", "site-functions"].iter().collect();
let mut outdir = basedir.clone();
outdir.push("zsh");
outdir.push("site-functions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "zsh")
@ -376,3 +395,4 @@ fn links(prefix: &str, usr: bool, cmd: &ArgMatches) -> Result<(), Box<dyn Error>
})?;
Ok(())
}

View File

@ -0,0 +1,398 @@
use super::{Cmd, COMMANDS};
use clap::{Arg, ArgAction, ArgMatches, Command};
use clap_complete::{generate_to, shells};
use clap_complete_nushell::Nushell;
use clap_mangen::Man;
use std::{
error::Error,
fs, io,
os::unix::fs::symlink,
path::{Path, PathBuf},
};
#[derive(Debug, Default)]
pub struct Bootstrap;
impl Bootstrap {
fn all() -> clap::Command {
Command::new("all").about("Install everything").args([
Arg::new("soft")
.help("Install soft links instead of hardlinks")
.short('s')
.long("soft")
.action(ArgAction::SetTrue),
Arg::new("all")
.help("Install completions for all supported shells")
.short('a')
.long("all")
.conflicts_with_all(["bash", "fish", "nu", "pwsh", "zsh"])
.action(ArgAction::SetTrue),
Arg::new("bash")
.help("Bash shell completions")
.short('b')
.long("bash")
.action(ArgAction::SetTrue),
Arg::new("fish")
.help("Fish shell completions")
.short('f')
.long("fish")
.action(ArgAction::SetTrue),
Arg::new("nu")
.help("Nushell completions")
.short('n')
.long("nu")
.action(ArgAction::SetTrue),
Arg::new("pwsh")
.help("PowerShell completions")
.short('p')
.long("pwsh")
.action(ArgAction::SetTrue),
Arg::new("zsh")
.help("Zshell completions")
.short('z')
.long("zsh")
.action(ArgAction::SetTrue),
])
}
fn links() -> clap::Command {
Command::new("links")
.about("Install links for each applet")
.arg(
Arg::new("soft")
.help("Install soft links instead of hardlinks")
.short('s')
.long("soft")
.action(ArgAction::SetTrue),
)
}
fn manpages() -> clap::Command {
Command::new("manpages")
.about("Install Unix man pages")
.alias("man")
}
fn completions() -> clap::Command {
Command::new("completions")
.about("Install shell completions")
.alias("comp")
.args([
Arg::new("all")
.help("Install completions for all supported shells")
.short('a')
.long("all")
.exclusive(true)
.action(ArgAction::SetTrue),
Arg::new("bash")
.help("Bash shell completions")
.short('b')
.long("bash")
.action(ArgAction::SetTrue),
Arg::new("fish")
.help("Fish shell completions")
.short('f')
.long("fish")
.action(ArgAction::SetTrue),
Arg::new("nu")
.help("Nushell completions")
.short('n')
.long("nu")
.action(ArgAction::SetTrue),
Arg::new("pwsh")
.help("PowerShell completions")
.short('p')
.long("pwsh")
.action(ArgAction::SetTrue),
Arg::new("zsh")
.help("Zshell completions")
.short('z')
.long("zsh")
.action(ArgAction::SetTrue),
])
}
}
impl Cmd for Bootstrap {
fn cli(&self) -> clap::Command {
Command::new("bootstrap")
.version(env!("CARGO_PKG_VERSION"))
.author("Nathan Fisher")
.about("Install shitbox into the filesystem")
.long_about("Install symlinks, manpages and shell completions")
.args([
Arg::new("prefix")
.help("The directory path under which to install")
.short('p')
.long("prefix")
.num_args(1)
.default_value("/")
.required(false),
Arg::new("usr")
.help("Use /usr")
.long_help(
"Split the installation so that some applets go into /bin | /sbin\n\
while others are placed into /usr/bin | /usr/sbin",
)
.short('u')
.long("usr")
.action(ArgAction::SetTrue),
Arg::new("soft")
.help("Install soft links instead of hardlinks")
.short('s')
.long("soft")
.action(ArgAction::SetTrue),
])
.subcommands([
Self::all(),
Self::links(),
Self::manpages(),
Self::completions(),
])
}
fn run(&self, matches: &ArgMatches) -> Result<(), Box<dyn Error>> {
if let Some(prefix) = matches.get_one::<String>("prefix") {
let usr = matches.get_flag("usr");
if let Some(progpath) = shitbox::progpath() {
let mut outpath = PathBuf::from(prefix);
outpath.push("bin");
println!("Installing binary:");
if !outpath.exists() {
fs::create_dir_all(&outpath)?;
println!(" mkdir: {}", outpath.display());
}
outpath.push(&shitbox::progname().unwrap());
fs::copy(&progpath, &outpath)?;
println!(
" install: {} -> {}",
progpath.display(),
outpath.display()
);
}
match matches.subcommand() {
Some(("links", matches)) => {
links(prefix, usr, matches)?;
}
Some(("manpages", _matches)) => {
manpages(prefix, usr)?;
}
Some(("completions", matches)) => {
completions(prefix, matches, usr)?;
}
Some(("all", matches)) => {
links(prefix, usr, matches)?;
manpages(prefix, usr)?;
completions(prefix, matches, usr)?;
}
_ => {}
}
}
Ok(())
}
fn path(&self) -> Option<shitbox::Path> {
None
}
}
#[allow(clippy::module_name_repetitions)]
pub trait BootstrapCmd {
fn completion(&self, outdir: &Path, gen: &str) -> Result<(), io::Error>;
fn linkpath(&self, prefix: &str, usr: bool) -> Option<PathBuf>;
fn link(&self, prefix: &str, usr: bool, soft: bool) -> Result<(), Box<dyn Error>>;
fn manpage(&self, prefix: &str, usr: bool) -> Result<(), io::Error>;
}
impl BootstrapCmd for dyn Cmd {
fn completion(&self, outdir: &Path, gen: &str) -> Result<(), io::Error> {
let cmd = self.cli();
let name = cmd.get_name();
let mut cmd = self.cli();
if !outdir.exists() {
fs::create_dir_all(outdir)?;
}
let path = match gen {
"bash" => generate_to(shells::Bash, &mut cmd, name, outdir)?,
"fish" => generate_to(shells::Fish, &mut cmd, name, outdir)?,
"nu" => generate_to(Nushell, &mut cmd, name, outdir)?,
"pwsh" => generate_to(shells::PowerShell, &mut cmd, name, outdir)?,
"zsh" => generate_to(shells::Zsh, &mut cmd, name, outdir)?,
_ => unimplemented!(),
};
println!(" {}", path.display());
Ok(())
}
fn linkpath(&self, prefix: &str, usr: bool) -> Option<PathBuf> {
let mut path = PathBuf::from(prefix);
let binpath = self.path();
match binpath {
Some(p) => path.push(p.to_str(usr)),
None => return None,
}
path.push(self.cli().get_name());
Some(path)
}
fn link(&self, prefix: &str, usr: bool, soft: bool) -> Result<(), Box<dyn Error>> {
if let Some(linkpath) = self.linkpath(prefix, usr) {
let progname = shitbox::progname().unwrap();
if soft {
let binpath = match self.path().unwrap() {
shitbox::Path::Bin => progname,
shitbox::Path::Sbin => format!("../bin/{progname}"),
shitbox::Path::UsrBin => {
if usr {
format!("../../bin/{progname}")
} else {
progname
}
}
shitbox::Path::UsrSbin => {
if usr {
format!("../../bin/{progname}")
} else {
format!("../bin/{progname}")
}
}
};
symlink(&binpath, &linkpath)?;
println!(" symlink: {binpath} -> {}", linkpath.display());
} else {
let mut binpath = PathBuf::from(prefix);
binpath.push("bin");
binpath.push(env!("CARGO_PKG_NAME"));
println!("Generating link for {}", self.cli().get_name());
fs::hard_link(&binpath, &linkpath)?;
println!(" link: {} -> {}", binpath.display(), linkpath.display());
}
}
Ok(())
}
fn manpage(&self, prefix: &str, usr: bool) -> Result<(), io::Error> {
let command = self.cli();
let fname = match command.get_name() {
"bootstrap" => format!("{}-bootstrap.1", shitbox::progname().unwrap()),
s => format!("{s}.1"),
};
let outdir: PathBuf = if usr {
[prefix, "usr", "share", "man", "man1"].iter().collect()
} else {
[prefix, "share", "man", "man1"].iter().collect()
};
if !outdir.exists() {
fs::create_dir_all(&outdir)?;
}
let mut outfile = outdir;
outfile.push(fname);
let man = Man::new(command);
let mut buffer: Vec<u8> = vec![];
man.render(&mut buffer)?;
fs::write(&outfile, buffer)?;
println!(" {}", outfile.display());
Ok(())
}
}
fn manpages(prefix: &str, usr: bool) -> Result<(), io::Error> {
println!("Generating Unix man pages:");
COMMANDS
.iter()
.try_for_each(|cmd| super::get(cmd).unwrap().manpage(prefix, usr))?;
Ok(())
}
fn completions(prefix: &str, matches: &ArgMatches, usr: bool) -> Result<(), io::Error> {
println!("Generating completions:");
let basedir: PathBuf = if usr {
[prefix, "usr", "share"].iter().collect()
} else {
[prefix, "share"].iter().collect()
};
if matches.get_flag("bash") || matches.get_flag("all") {
let mut outdir = basedir.clone();
outdir.push("bash-completion");
outdir.push("completion");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "bash")
})?;
}
if matches.get_flag("fish") || matches.get_flag("all") {
let mut outdir = basedir.clone();
outdir.push("fish");
outdir.push("completions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "fish")
})?;
}
if matches.get_flag("nu") || matches.get_flag("all") {
let mut outdir = basedir.clone();
outdir.push("nu");
outdir.push("completions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "nu")
})?;
}
if matches.get_flag("pwsh") || matches.get_flag("all") {
let mut outdir = basedir.clone();
outdir.push("pwsh");
outdir.push("completions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "pwsh")
})?;
}
if matches.get_flag("zsh") || matches.get_flag("all") {
let mut outdir = basedir.clone();
outdir.push("zsh");
outdir.push("site-functions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "zsh")
})?;
}
Ok(())
}
fn links(prefix: &str, usr: bool, cmd: &ArgMatches) -> Result<(), Box<dyn Error>> {
println!("Generating links:");
let mut binpath = PathBuf::from(prefix);
binpath.push("bin");
if !binpath.exists() {
fs::create_dir_all(&binpath)?;
println!(" mkdir: {}", binpath.display());
}
binpath.pop();
binpath.push("sbin");
if !binpath.exists() {
fs::create_dir_all(&binpath)?;
println!(" mkdir: {}", binpath.display());
}
if usr {
binpath.pop();
binpath.push("usr");
binpath.push("bin");
if !binpath.exists() {
fs::create_dir_all(&binpath)?;
println!(" mkdir: {}", binpath.display());
}
binpath.pop();
binpath.push("sbin");
if !binpath.exists() {
fs::create_dir_all(&binpath)?;
println!(" mkdir: {}", binpath.display());
}
}
let soft = cmd.get_flag("soft");
COMMANDS.iter().try_for_each(|c| {
let cmd = super::get(c).unwrap();
cmd.link(prefix, usr, soft)
})?;
Ok(())
}

View File

@ -1,6 +1,7 @@
use shitbox::Cmd;
mod b2sum;
mod bootstrap;
mod hashbox;
mod md5sum;
mod sha1sum;
@ -15,6 +16,7 @@ mod sha512sum;
pub fn get(name: &str) -> Option<Box<dyn Cmd>> {
match name {
"b2sum" => Some(Box::new(b2sum::B2sum::default())),
"bootstrap" => Some(Box::new(bootstrap::Bootstrap::default())),
"hashbox" => Some(Box::new(hashbox::Hashbox::default())),
"md5sum" => Some(Box::new(md5sum::Md5sum::default())),
"sha1sum" => Some(Box::new(sha1sum::Sha1sum::default())),
@ -26,8 +28,9 @@ pub fn get(name: &str) -> Option<Box<dyn Cmd>> {
}
}
pub static COMMANDS: [&str; 7] = [
pub static COMMANDS: [&str; 8] = [
"b2sum",
"bootstrap",
"md5sum",
"sha1sum",
"sha224sum",

View File

@ -0,0 +1,398 @@
use super::{Cmd, COMMANDS};
use clap::{Arg, ArgAction, ArgMatches, Command};
use clap_complete::{generate_to, shells};
use clap_complete_nushell::Nushell;
use clap_mangen::Man;
use std::{
error::Error,
fs, io,
os::unix::fs::symlink,
path::{Path, PathBuf},
};
#[derive(Debug, Default)]
pub struct Bootstrap;
impl Bootstrap {
fn all() -> clap::Command {
Command::new("all").about("Install everything").args([
Arg::new("soft")
.help("Install soft links instead of hardlinks")
.short('s')
.long("soft")
.action(ArgAction::SetTrue),
Arg::new("all")
.help("Install completions for all supported shells")
.short('a')
.long("all")
.conflicts_with_all(["bash", "fish", "nu", "pwsh", "zsh"])
.action(ArgAction::SetTrue),
Arg::new("bash")
.help("Bash shell completions")
.short('b')
.long("bash")
.action(ArgAction::SetTrue),
Arg::new("fish")
.help("Fish shell completions")
.short('f')
.long("fish")
.action(ArgAction::SetTrue),
Arg::new("nu")
.help("Nushell completions")
.short('n')
.long("nu")
.action(ArgAction::SetTrue),
Arg::new("pwsh")
.help("PowerShell completions")
.short('p')
.long("pwsh")
.action(ArgAction::SetTrue),
Arg::new("zsh")
.help("Zshell completions")
.short('z')
.long("zsh")
.action(ArgAction::SetTrue),
])
}
fn links() -> clap::Command {
Command::new("links")
.about("Install links for each applet")
.arg(
Arg::new("soft")
.help("Install soft links instead of hardlinks")
.short('s')
.long("soft")
.action(ArgAction::SetTrue),
)
}
fn manpages() -> clap::Command {
Command::new("manpages")
.about("Install Unix man pages")
.alias("man")
}
fn completions() -> clap::Command {
Command::new("completions")
.about("Install shell completions")
.alias("comp")
.args([
Arg::new("all")
.help("Install completions for all supported shells")
.short('a')
.long("all")
.exclusive(true)
.action(ArgAction::SetTrue),
Arg::new("bash")
.help("Bash shell completions")
.short('b')
.long("bash")
.action(ArgAction::SetTrue),
Arg::new("fish")
.help("Fish shell completions")
.short('f')
.long("fish")
.action(ArgAction::SetTrue),
Arg::new("nu")
.help("Nushell completions")
.short('n')
.long("nu")
.action(ArgAction::SetTrue),
Arg::new("pwsh")
.help("PowerShell completions")
.short('p')
.long("pwsh")
.action(ArgAction::SetTrue),
Arg::new("zsh")
.help("Zshell completions")
.short('z')
.long("zsh")
.action(ArgAction::SetTrue),
])
}
}
impl Cmd for Bootstrap {
fn cli(&self) -> clap::Command {
Command::new("bootstrap")
.version(env!("CARGO_PKG_VERSION"))
.author("Nathan Fisher")
.about("Install shitbox into the filesystem")
.long_about("Install symlinks, manpages and shell completions")
.args([
Arg::new("prefix")
.help("The directory path under which to install")
.short('p')
.long("prefix")
.num_args(1)
.default_value("/")
.required(false),
Arg::new("usr")
.help("Use /usr")
.long_help(
"Split the installation so that some applets go into /bin | /sbin\n\
while others are placed into /usr/bin | /usr/sbin",
)
.short('u')
.long("usr")
.action(ArgAction::SetTrue),
Arg::new("soft")
.help("Install soft links instead of hardlinks")
.short('s')
.long("soft")
.action(ArgAction::SetTrue),
])
.subcommands([
Self::all(),
Self::links(),
Self::manpages(),
Self::completions(),
])
}
fn run(&self, matches: &ArgMatches) -> Result<(), Box<dyn Error>> {
if let Some(prefix) = matches.get_one::<String>("prefix") {
let usr = matches.get_flag("usr");
if let Some(progpath) = shitbox::progpath() {
let mut outpath = PathBuf::from(prefix);
outpath.push("bin");
println!("Installing binary:");
if !outpath.exists() {
fs::create_dir_all(&outpath)?;
println!(" mkdir: {}", outpath.display());
}
outpath.push(&shitbox::progname().unwrap());
fs::copy(&progpath, &outpath)?;
println!(
" install: {} -> {}",
progpath.display(),
outpath.display()
);
}
match matches.subcommand() {
Some(("links", matches)) => {
links(prefix, usr, matches)?;
}
Some(("manpages", _matches)) => {
manpages(prefix, usr)?;
}
Some(("completions", matches)) => {
completions(prefix, matches, usr)?;
}
Some(("all", matches)) => {
links(prefix, usr, matches)?;
manpages(prefix, usr)?;
completions(prefix, matches, usr)?;
}
_ => {}
}
}
Ok(())
}
fn path(&self) -> Option<shitbox::Path> {
None
}
}
#[allow(clippy::module_name_repetitions)]
pub trait BootstrapCmd {
fn completion(&self, outdir: &Path, gen: &str) -> Result<(), io::Error>;
fn linkpath(&self, prefix: &str, usr: bool) -> Option<PathBuf>;
fn link(&self, prefix: &str, usr: bool, soft: bool) -> Result<(), Box<dyn Error>>;
fn manpage(&self, prefix: &str, usr: bool) -> Result<(), io::Error>;
}
impl BootstrapCmd for dyn Cmd {
fn completion(&self, outdir: &Path, gen: &str) -> Result<(), io::Error> {
let cmd = self.cli();
let name = cmd.get_name();
let mut cmd = self.cli();
if !outdir.exists() {
fs::create_dir_all(outdir)?;
}
let path = match gen {
"bash" => generate_to(shells::Bash, &mut cmd, name, outdir)?,
"fish" => generate_to(shells::Fish, &mut cmd, name, outdir)?,
"nu" => generate_to(Nushell, &mut cmd, name, outdir)?,
"pwsh" => generate_to(shells::PowerShell, &mut cmd, name, outdir)?,
"zsh" => generate_to(shells::Zsh, &mut cmd, name, outdir)?,
_ => unimplemented!(),
};
println!(" {}", path.display());
Ok(())
}
fn linkpath(&self, prefix: &str, usr: bool) -> Option<PathBuf> {
let mut path = PathBuf::from(prefix);
let binpath = self.path();
match binpath {
Some(p) => path.push(p.to_str(usr)),
None => return None,
}
path.push(self.cli().get_name());
Some(path)
}
fn link(&self, prefix: &str, usr: bool, soft: bool) -> Result<(), Box<dyn Error>> {
if let Some(linkpath) = self.linkpath(prefix, usr) {
let progname = shitbox::progname().unwrap();
if soft {
let binpath = match self.path().unwrap() {
shitbox::Path::Bin => progname,
shitbox::Path::Sbin => format!("../bin/{progname}"),
shitbox::Path::UsrBin => {
if usr {
format!("../../bin/{progname}")
} else {
progname
}
}
shitbox::Path::UsrSbin => {
if usr {
format!("../../bin/{progname}")
} else {
format!("../bin/{progname}")
}
}
};
symlink(&binpath, &linkpath)?;
println!(" symlink: {binpath} -> {}", linkpath.display());
} else {
let mut binpath = PathBuf::from(prefix);
binpath.push("bin");
binpath.push(env!("CARGO_PKG_NAME"));
println!("Generating link for {}", self.cli().get_name());
fs::hard_link(&binpath, &linkpath)?;
println!(" link: {} -> {}", binpath.display(), linkpath.display());
}
}
Ok(())
}
fn manpage(&self, prefix: &str, usr: bool) -> Result<(), io::Error> {
let command = self.cli();
let fname = match command.get_name() {
"bootstrap" => format!("{}-bootstrap.1", shitbox::progname().unwrap()),
s => format!("{s}.1"),
};
let outdir: PathBuf = if usr {
[prefix, "usr", "share", "man", "man1"].iter().collect()
} else {
[prefix, "share", "man", "man1"].iter().collect()
};
if !outdir.exists() {
fs::create_dir_all(&outdir)?;
}
let mut outfile = outdir;
outfile.push(fname);
let man = Man::new(command);
let mut buffer: Vec<u8> = vec![];
man.render(&mut buffer)?;
fs::write(&outfile, buffer)?;
println!(" {}", outfile.display());
Ok(())
}
}
fn manpages(prefix: &str, usr: bool) -> Result<(), io::Error> {
println!("Generating Unix man pages:");
COMMANDS
.iter()
.try_for_each(|cmd| super::get(cmd).unwrap().manpage(prefix, usr))?;
Ok(())
}
fn completions(prefix: &str, matches: &ArgMatches, usr: bool) -> Result<(), io::Error> {
println!("Generating completions:");
let basedir: PathBuf = if usr {
[prefix, "usr", "share"].iter().collect()
} else {
[prefix, "share"].iter().collect()
};
if matches.get_flag("bash") || matches.get_flag("all") {
let mut outdir = basedir.clone();
outdir.push("bash-completion");
outdir.push("completion");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "bash")
})?;
}
if matches.get_flag("fish") || matches.get_flag("all") {
let mut outdir = basedir.clone();
outdir.push("fish");
outdir.push("completions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "fish")
})?;
}
if matches.get_flag("nu") || matches.get_flag("all") {
let mut outdir = basedir.clone();
outdir.push("nu");
outdir.push("completions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "nu")
})?;
}
if matches.get_flag("pwsh") || matches.get_flag("all") {
let mut outdir = basedir.clone();
outdir.push("pwsh");
outdir.push("completions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "pwsh")
})?;
}
if matches.get_flag("zsh") || matches.get_flag("all") {
let mut outdir = basedir.clone();
outdir.push("zsh");
outdir.push("site-functions");
COMMANDS.iter().try_for_each(|cmd| {
let cmd = super::get(cmd).unwrap();
cmd.completion(&outdir, "zsh")
})?;
}
Ok(())
}
fn links(prefix: &str, usr: bool, cmd: &ArgMatches) -> Result<(), Box<dyn Error>> {
println!("Generating links:");
let mut binpath = PathBuf::from(prefix);
binpath.push("bin");
if !binpath.exists() {
fs::create_dir_all(&binpath)?;
println!(" mkdir: {}", binpath.display());
}
binpath.pop();
binpath.push("sbin");
if !binpath.exists() {
fs::create_dir_all(&binpath)?;
println!(" mkdir: {}", binpath.display());
}
if usr {
binpath.pop();
binpath.push("usr");
binpath.push("bin");
if !binpath.exists() {
fs::create_dir_all(&binpath)?;
println!(" mkdir: {}", binpath.display());
}
binpath.pop();
binpath.push("sbin");
if !binpath.exists() {
fs::create_dir_all(&binpath)?;
println!(" mkdir: {}", binpath.display());
}
}
let soft = cmd.get_flag("soft");
COMMANDS.iter().try_for_each(|c| {
let cmd = super::get(c).unwrap();
cmd.link(prefix, usr, soft)
})?;
Ok(())
}

View File

@ -1,6 +1,7 @@
use shitbox::Cmd;
mod blkid;
mod bootstrap;
mod clear;
mod mount;
mod mountpoint;
@ -15,6 +16,7 @@ mod utilbox;
#[allow(clippy::box_default)]
pub fn get(name: &str) -> Option<Box<dyn Cmd>> {
match name {
"bootstrap" => Some(Box::new(bootstrap::Bootstrap::default())),
"clear" => Some(Box::new(clear::Clear::default())),
"mountpoint" => Some(Box::new(mountpoint::Mountpoint::default())),
"swaplabel" => Some(Box::new(swaplabel::Swaplabel::default())),
@ -25,7 +27,8 @@ pub fn get(name: &str) -> Option<Box<dyn Cmd>> {
}
}
pub static COMMANDS: [&str; 6] = [
pub static COMMANDS: [&str; 7] = [
"bootstrap",
"clear",
"mountpoint",
"swaplabel",