add part install for linux

Author: tyan-boot

src/cert.rs

@@ -23,7 +23,7 @@ use self::openssl::x509::*;
 
 use self::rand::prelude::random;
 
-use super::utils;
+use utils;
 
 pub struct Cert {
     ca: Option<X509>,

src/install.rs

@@ -0,0 +1,45 @@
+extern crate dirs;
+
+use std::fs::copy;
+use std::fs::DirBuilder;
+use std::path::Path;
+use std::process::Command;
+
+use utils;
+
+pub fn install() {
+    if cfg!(unix) {
+        install_linux();
+    }
+}
+
+pub fn install_linux() {
+    let ssl_dirs = vec![
+        (
+            "/etc/ca-certificates/trust-source/anchors",
+            "update-ca-trust",
+        ),
+        ("/etc/pki/ca-trust/source/anchors", "update-ca-trust"),
+    ];
+
+    for (ssl_dir, update_command) in ssl_dirs {
+        if Path::new(ssl_dir).exists() {
+            let target_path = format!("{}/easycert-ca.pem", ssl_dir);
+
+            Command::new("sudo")
+                .args(&["cp", &utils::ca_path(), &target_path])
+                .output()
+                .unwrap();
+
+            Command::new("sudo")
+                .args(&[update_command])
+                .output()
+                .unwrap();
+
+            return;
+        }
+    }
+
+    eprintln!("can't determinate system trusted store path");
+    eprintln!("failed to install certificate, try install to nss store");
+}

src/main.rs

@@ -5,10 +5,15 @@ extern crate gdi32;
 #[cfg(windows)]
 extern crate user32;
 
+use std::fs::File;
+
+use clap::{App, Arg, ArgMatches, SubCommand};
+
 mod cert;
+mod install;
 mod utils;
 
-use clap::{App, Arg, SubCommand};
+use install::install;
 
 fn main() {
     let args = App::new("mkcert")
@@ -63,13 +68,10 @@ fn main() {
                         .help("lengh of key, default 2048")
                         .takes_value(true),
                 ),
-        );
-
-    let matches = args.get_matches();
-
-    let mut c = cert::Cert::new();
+        )
+        .subcommand(SubCommand::with_name("install").about("install ca to trusted store"));
 
-    if let Some(matches) = matches.subcommand_matches("init") {
+    let get_init_args = |matches: &ArgMatches| {
         let length: u32 = matches
             .value_of("length")
             .unwrap_or("2048")
@@ -88,6 +90,16 @@ fn main() {
             false
         };
 
+        return (cn, length, force);
+    };
+
+    let matches = args.get_matches();
+
+    let mut c = cert::Cert::new();
+
+    if let Some(matches) = matches.subcommand_matches("init") {
+        let (cn, length, force) = get_init_args(&matches);
+
         c.init(&cn, length, force);
     }
 
@@ -109,4 +121,18 @@ fn main() {
             eprintln!("must provide a name!");
         }
     }
+
+    if let Some(matches) = matches.subcommand_matches("install") {
+        let ca = File::open(utils::ca_path());
+        let key = File::open(utils::key_path());
+
+        if let (Ok(_ca), Ok(_key)) = (ca, key) {
+        } else {
+            let (cn, length, force) = get_init_args(&matches);
+
+            c.init(&cn, length, force);
+        }
+
+        install();
+    }
 }