From db3cf3e42d93112278f5e86cca2c886627ef48b2 Mon Sep 17 00:00:00 2001
From: Ava Hahn <a.hahn@f5.com>
Date: Mon, 22 Apr 2024 13:26:34 +0100
Subject: tools: Add unitctl CLI

* Pull in entire unit-rust-sdk project
  * not included: CLA, COC, License
  * not included: duplicate openapi spec
  * not included: CI workflows
  * not included: changelog tooling
  * not included: commitsar tooling
  * not included: OpenAPI Web UI feature
* update links in unitctl manpage
* remove IDE configuration from .gitignore
* rename Containerfile.debian to Dockerfile
* simplify call to uname
* keep Readmes and Makefiles to 80 character lines
* outline specifically how to build unitctl
  for any desired target, and where to then
  find the binary for use
* remove a section on the vision of the CLI
  which was superfluous given the state of
  completeness of the code and its use in
  unit
* remove out of date feature proposals from readme
* makefile: do not run when Rustup is not present
* bump mio version to latest
* generate openapi client library on demand
  * generate-openapi only runs when not present
  * generate-openapi now a dependency of binary build targets
  * deleted autogenerated code
  * reverted readme and Cargo document to autogenerated state
  * add additional build requirement to Readme

Co-developed-by: Elijah Zupancic <e.zupancic@f5.com>
Signed-off-by: Elijah Zupancic <e.zupancic@f5.com>
Signed-off-by: Ava Hahn <a.hahn@f5.com>
Reviewed-by: Andrew Clayton <a.clayton@nginx.com> # non rust stuff
[ tools/cli => tools/unitctl and subject tweak - Andrew ]
Signed-off-by: Andrew Clayton <a.clayton@nginx.com>
---
 tools/unitctl/unit-client-rs/src/unitd_cmd.rs | 85 +++++++++++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 tools/unitctl/unit-client-rs/src/unitd_cmd.rs

(limited to 'tools/unitctl/unit-client-rs/src/unitd_cmd.rs')

diff --git a/tools/unitctl/unit-client-rs/src/unitd_cmd.rs b/tools/unitctl/unit-client-rs/src/unitd_cmd.rs
new file mode 100644
index 00000000..c4883ed5
--- /dev/null
+++ b/tools/unitctl/unit-client-rs/src/unitd_cmd.rs
@@ -0,0 +1,85 @@
+use std::error::Error as StdError;
+use std::io::{Error as IoError, ErrorKind};
+
+use crate::runtime_flags::RuntimeFlags;
+use std::path::{Path, PathBuf};
+
+#[derive(Debug, Clone)]
+pub struct UnitdCmd {
+    pub(crate) process_executable_path: Option<Box<Path>>,
+    pub version: Option<String>,
+    pub flags: Option<RuntimeFlags>,
+}
+
+impl UnitdCmd {
+    pub(crate) fn new<S>(full_cmd: S, binary_name: &str) -> Result<UnitdCmd, Box<dyn StdError>>
+    where
+        S: Into<String>,
+    {
+        let process_cmd: String = full_cmd.into();
+        let parsable = process_cmd
+            .strip_prefix("unit: main v")
+            .and_then(|s| s.strip_suffix(']'));
+        if parsable.is_none() {
+            let msg = format!("cmd does not have the expected format: {}", process_cmd);
+            return Err(IoError::new(ErrorKind::InvalidInput, msg).into());
+        }
+        let parts = parsable
+            .expect("Unable to parse cmd")
+            .splitn(2, " [")
+            .collect::<Vec<&str>>();
+        if parts.len() != 2 {
+            let msg = format!("cmd does not have the expected format: {}", process_cmd);
+            return Err(IoError::new(ErrorKind::InvalidInput, msg).into());
+        }
+        let version: Option<String> = Some(parts[0].to_string());
+        let executable_path = UnitdCmd::parse_executable_path_from_cmd(parts[1], binary_name);
+        let flags = UnitdCmd::parse_runtime_flags_from_cmd(parts[1]);
+
+        Ok(UnitdCmd {
+            process_executable_path: executable_path,
+            version,
+            flags,
+        })
+    }
+
+    fn parse_executable_path_from_cmd<S>(full_cmd: S, binary_name: &str) -> Option<Box<Path>>
+    where
+        S: Into<String>,
+    {
+        let cmd = full_cmd.into();
+        if cmd.is_empty() {
+            return None;
+        }
+
+        let split = cmd.splitn(2, binary_name).collect::<Vec<&str>>();
+        if split.is_empty() {
+            return None;
+        }
+
+        let path = format!("{}{}", split[0], binary_name);
+        Some(PathBuf::from(path).into_boxed_path())
+    }
+
+    fn parse_runtime_flags_from_cmd<S>(full_cmd: S) -> Option<RuntimeFlags>
+    where
+        S: Into<String>,
+    {
+        let cmd = full_cmd.into();
+        if cmd.is_empty() {
+            return None;
+        }
+        // Split out everything in between the brackets [ and ]
+        let split = cmd.trim_end_matches(']').splitn(2, '[').collect::<Vec<&str>>();
+        if split.is_empty() {
+            return None;
+        }
+        /* Now we need to parse a string like this:
+         * ./sbin/unitd --no-daemon --tmp /tmp
+         * and only return what is after the invoking command */
+        split[0]
+            .find("--")
+            .map(|index| cmd[index..].to_string())
+            .map(RuntimeFlags::new)
+    }
+}
-- 
cgit