sam 0.3.1

A compile time instruction assembler.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

use std::process::Command;
use std::env;

#[cfg(not(target_os = "windows"))]
const GIT: &str = "https://github.com/keystone-engine/keystone.git";
#[cfg(target_os = "windows")]
const ZIP: &str = "https://github.com/keystone-engine/keystone/releases/download/0.9.1/keystone-0.9.1-win64.zip";

fn check_cmd(cmd: &str) -> bool {
    match Command::new(cmd).spawn() {
        Ok(_) => true,
        Err(_) => false, 
    }
}

fn check_sudo() {
    let is_sudo = match env::var("USER") {
        Ok(val) => val == "root",
        Err(_) => false,
    };
    if !is_sudo { panic!("You have to run this as root!"); }
}

#[cfg(target_os = "macos")]
fn check_dependencies() {
    if !check_cmd("brew") { panic!("Homebrew not found!"); }
    if !check_cmd("cmake") { 
        Command::new("brew")
            .args(&["install", "cmake"])
            .output()
            .expect("failed to execute process");
    }
    if !check_cmd("git") { 
        Command::new("brew")
            .args(&["install", "git"])
            .output()
            .expect("failed to execute process");
    }
}

#[cfg(target_os = "linux")]
fn check_dependencies() {
    if !check_cmd("cmake") { 
        Command::new("apt-get")
            .args(&["install", "cmake"])
            .output()
            .expect("failed to execute process");
    }
    if !check_cmd("git") { 
        Command::new("apt-get")
            .args(&["install", "git"])
            .output()
            .expect("failed to execute process");
    }
}

#[cfg(target_os = "windows")]
fn download_kstool() {
    let mut output = Command::new("powershell")
            .args(&["-Command", &format!("Invoke-WebRequest {} -OutFile keystone.zip", ZIP)])
            .output()
            .expect("failed to execute process");
    println!("{:?}", String::from_utf8_lossy(&output.stdout));
    output = Command::new("powershell")
            .args(&["-Command", "Expand-Archive -Path keystone.zip -DestinationPath ."])
            .output()
            .expect("failed to execute process");
    println!("{:?}", String::from_utf8_lossy(&output.stdout));
    let path = env::home_dir().unwrap();
    let cargo_bin = &format!("{}\\.cargo\\bin\\", path.display()); // just install it to .cargo/bin for now.
    output = Command::new("powershell")
            .args(&["xcopy", ".\\keystone-0.9.1-win64\\kstool.exe", cargo_bin])
            .output()
            .expect("failed to execute process");
    println!("{:?}", String::from_utf8_lossy(&output.stdout));
    output = Command::new("powershell")
            .args(&["rm", "-r", ".\\keystone-0.9.1-win64\\"])
            .output()
            .expect("failed to execute process");
    println!("{:?}", String::from_utf8_lossy(&output.stdout));
    output = Command::new("powershell")
            .args(&["rm", ".\\keystone.zip"])
            .output()
            .expect("failed to execute process");
    println!("{:?}", String::from_utf8_lossy(&output.stdout));
}

#[cfg(not(target_os = "windows"))]
fn download_kstool() {
    let mut output = Command::new("git")
            .args(&["clone", GIT])
            .output()
            .expect("failed to execute process");
    println!("{:?}", String::from_utf8_lossy(&output.stdout));
    output = Command::new("mkdir")
            .current_dir("keystone/")
            .args(&["build"])
            .output()
            .expect("failed to execute process");
    println!("{:?}", String::from_utf8_lossy(&output.stdout));
    output = Command::new("sh")
            .current_dir("keystone/build/")
            .args(&["../make-share.sh"])
            .output()
            .expect("failed to execute process");
    println!("{:?}", String::from_utf8_lossy(&output.stdout));
    output = Command::new("make")
            .current_dir("keystone/build/")
            .args(&["install"])
            .output()
            .expect("failed to execute process");
    println!("{:?}", String::from_utf8_lossy(&output.stdout));
    output = Command::new("rm")
            .args(&["-r", "-f", "keystone"])
            .output()
            .expect("failed to execute process");
    println!("{:?}", String::from_utf8_lossy(&output.stdout));
}

fn main() {
    if !check_cmd("kstool") {
        if cfg!(target_os = "linux") { check_sudo(); }
        if cfg!(target_os = "macos") || cfg!(target_os = "linux") { check_dependencies(); }
        download_kstool();
    }
}