Merge pull request #377 from dezgeg/add_tool

nologin: Add tool

Author: cakebaker

Cargo.lock

@@ -39,6 +39,7 @@ feat_common_core = [
   "mcookie",
   "mesg",
   "mountpoint",
+  "nologin",
   "renice",
   "rev",
   "setsid",
@@ -102,6 +103,7 @@ lsmem = { optional = true, version = "0.0.1", package = "uu_lsmem", path = "src/
 mcookie = { optional = true, version = "0.0.1", package = "uu_mcookie", path = "src/uu/mcookie" }
 mesg = { optional = true, version = "0.0.1", package = "uu_mesg", path = "src/uu/mesg" }
 mountpoint = { optional = true, version = "0.0.1", package = "uu_mountpoint", path = "src/uu/mountpoint" }
+nologin = { optional = true, version = "0.0.1", package = "uu_nologin", path = "src/uu/nologin" }
 renice = { optional = true, version = "0.0.1", package = "uu_renice", path = "src/uu/renice" }
 rev = { optional = true, version = "0.0.1", package = "uu_rev", path = "src/uu/rev" }
 setsid = { optional = true, version = "0.0.1", package = "uu_setsid", path ="src/uu/setsid" }

Cargo.toml

@@ -0,0 +1,15 @@
+[package]
+name = "uu_nologin"
+version = "0.0.1"
+edition = "2021"
+
+[lib]
+path = "src/nologin.rs"
+
+[[bin]]
+name = "nologin"
+path = "src/main.rs"
+
+[dependencies]
+uucore = { workspace = true }
+clap = { workspace = true }

src/uu/nologin/Cargo.toml

@@ -0,0 +1,7 @@
+# nologin
+
+```
+nologin [options]
+```
+
+Politely refuse a login.

src/uu/nologin/nologin.md

@@ -0,0 +1 @@
+uucore::bin!(uu_nologin);

src/uu/nologin/src/main.rs

@@ -0,0 +1,91 @@
+// This file is part of the uutils util-linux package.
+//
+// For the full copyright and license information, please view the LICENSE
+// file that was distributed with this source code.
+
+use clap::{crate_version, Arg, ArgAction, Command};
+use std::fs;
+use uucore::{error::UResult, format_usage, help_about, help_usage};
+
+const ABOUT: &str = help_about!("nologin.md");
+const USAGE: &str = help_usage!("nologin.md");
+
+#[uucore::main]
+pub fn uumain(args: impl uucore::Args) -> UResult<()> {
+    let _matches = uu_app().try_get_matches_from(args)?;
+
+    // Try to read custom message from /etc/nologin.txt
+    let message = match fs::read_to_string("/etc/nologin.txt") {
+        Ok(content) => content.trim().to_string(),
+        Err(_) => "This account is currently not available.".to_string(),
+    };
+
+    println!("{}", message);
+    std::process::exit(1);
+}
+
+pub fn uu_app() -> Command {
+    Command::new(uucore::util_name())
+        .version(crate_version!())
+        .about(ABOUT)
+        .override_usage(format_usage(USAGE))
+        .infer_long_args(true)
+        .arg(
+            Arg::new("command")
+                .short('c')
+                .long("command")
+                .value_name("command")
+                .help("does nothing (for compatibility)"),
+        )
+        .arg(
+            Arg::new("init-file")
+                .long("init-file")
+                .value_name("file")
+                .help("does nothing (for compatibility)"),
+        )
+        .arg(
+            Arg::new("interactive")
+                .short('i')
+                .long("interactive")
+                .help("does nothing (for compatibility)")
+                .action(ArgAction::SetTrue),
+        )
+        .arg(
+            Arg::new("login")
+                .short('l')
+                .long("login")
+                .help("does nothing (for compatibility)")
+                .action(ArgAction::SetTrue),
+        )
+        .arg(
+            Arg::new("noprofile")
+                .long("noprofile")
+                .help("does nothing (for compatibility)")
+                .action(ArgAction::SetTrue),
+        )
+        .arg(
+            Arg::new("norc")
+                .long("norc")
+                .help("does nothing (for compatibility)")
+                .action(ArgAction::SetTrue),
+        )
+        .arg(
+            Arg::new("posix")
+                .long("posix")
+                .help("does nothing (for compatibility)")
+                .action(ArgAction::SetTrue),
+        )
+        .arg(
+            Arg::new("rcfile")
+                .long("rcfile")
+                .value_name("file")
+                .help("does nothing (for compatibility)"),
+        )
+        .arg(
+            Arg::new("restricted")
+                .short('r')
+                .long("restricted")
+                .help("does nothing (for compatibility)")
+                .action(ArgAction::SetTrue),
+        )
+}

src/uu/nologin/src/nologin.rs

@@ -0,0 +1,37 @@
+// This file is part of the uutils util-linux package.
+//
+// For the full copyright and license information, please view the LICENSE
+// file that was distributed with this source code.
+
+use uutests::{new_ucmd, util::TestScenario, util_name};
+
+#[test]
+fn test_invalid_arg() {
+    new_ucmd!().arg("--definitely-invalid").fails().code_is(1);
+}
+
+#[test]
+fn test_nologin_args() {
+    let args_to_try: &[&[&str]] = &[
+        &[],
+        &["-c", "command"],
+        &["--init-file", "file"],
+        &["-i"],
+        &["--interactive"],
+        &["-l"],
+        &["--login"],
+        &["--noprofile"],
+        &["--norc"],
+        &["--posix"],
+        &["--rcfile", "file"],
+        &["--restricted"],
+        &["-r"],
+    ];
+    for args in args_to_try {
+        new_ucmd!()
+            .args(args)
+            .fails()
+            .code_is(1)
+            .stdout_contains("This account is currently not available.");
+    }
+}

tests/by-util/test_nologin.rs

@@ -35,6 +35,10 @@ mod test_mesg;
 #[path = "by-util/test_mountpoint.rs"]
 mod test_mountpoint;
 
+#[cfg(feature = "nologin")]
+#[path = "by-util/test_nologin.rs"]
+mod test_nologin;
+
 #[cfg(feature = "blockdev")]
 #[path = "by-util/test_blockdev.rs"]
 mod test_blockdev;