Init

1 files changed, 74 insertions, 0 deletions

diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..cb5a926
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,74 @@
+use std::io::{Read, Write};
+use std::path::Path;
+
+use clap::Args;
+use clap::{Parser, Subcommand};
+mod config;
+
+#[derive(Parser)]
+#[command(version, about, long_about=include_str!("../README.md"))]
+struct Cli {
+    #[command(subcommand)]
+    command: Commands,
+}
+
+#[derive(Subcommand)]
+enum Commands {
+    Auth(Auth),
+    Login(Login),
+}
+#[derive(Args)]
+struct Auth {
+    user: String,
+    home: String,
+    keytype: String,
+    keydata: String,
+}
+#[derive(Args)]
+struct Login {
+    keytype: String,
+    keydata: String,
+}
+
+fn main() {
+    systemd_journal_logger::JournalLog::new()
+        .unwrap()
+        .with_extra_fields(vec![("VERSION", env!("CARGO_PKG_VERSION"))])
+        .with_syslog_identifier("fagit".to_string())
+        .install()
+        .unwrap();
+    log::set_max_level(log::LevelFilter::Info);
+    let args = std::env::args().collect::<Vec<_>>();
+    if args.len() == 0 {
+        println!(include_str!("../README.md"));
+    }
+    let cli = Cli::parse();
+    match &cli.command {
+        Commands::Auth(auth) => find_auth_keys(auth),
+        Commands::Login(login) => crate::login::do_login(login),
+    }
+}
+mod login;
+
+fn find_auth_keys(auth: &Auth) {
+    if auth.user != config::ssh_user() {
+        let authorized_keys_path = Path::new(&auth.home).join(".ssh/authorized_keys_path");
+        let mut data = std::fs::File::open(authorized_keys_path).unwrap();
+        let mut vec = vec![];
+        data.read_to_end(&mut vec).unwrap();
+        std::io::stdout().write_all(&vec).unwrap();
+        return;
+    }
+    // TODO: escape shit in here properly
+    let login_command = format!(
+        "{} login {} {}",
+        config::binary_path(),
+        auth.keytype,
+        auth.keydata
+    );
+    println!("command=\"{}\",no-port-forwarding,no-X11-forwarding,no-agent-forwarding,no-pty,no-user-rc,restrict {} {}",
+		login_command,
+		auth.keytype,
+		auth.keydata
+	);
+}