cdbk 0.1.0

a way to bundle apps for linux
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::{
    fs::{self, File, remove_file},
    io::Write,
    path::PathBuf,
};

use cmdparsing::define;
use zip::{CompressionMethod, ZipWriter, write::SimpleFileOptions};

use crate::{locations::rename, manifest::Manifest};

const HELP: &str = r#"usage: cdbk package [file path]
===== FLAGS ===== 
-o|--out [path]: the output path
"#;

define! {
    Arguments;
    help: HELP;
    flags {
        ouput: PathBuf = "o"|"out",
    };
    args {
        input: PathBuf,
    };
}

pub fn package(mut v: Vec<String>) {
    let mut bak = false;
    v.remove(0);
    let args = Arguments::from(v);
    let out = args.ouput.unwrap_or_else(|| {
        let mut out = args.input.clone();
        out.set_extension("cdbk");
        return out;
    });
    let mut new = out.clone();
    new.add_extension("bak");
    if out.exists() {
        rename(&out, &new).unwrap();
        bak = true;
    }

    let Ok(mut file) = File::create(&out) else {
        eprintln!("error creating file");
        remove_file(&out).unwrap();
        if bak {
            rename(&new, &out).unwrap();
        }
        return;
    };
    file.write(b"CDBK").unwrap();
    let options = SimpleFileOptions::default().compression_method(CompressionMethod::Stored);
    let mut zip = ZipWriter::new(file);
    let mut path = args.input;
    path.push("manifest.toml");
    let Ok(data) = fs::read_to_string(&path) else {
        eprintln!("file {:?} does not exist", path);
        remove_file(&out).unwrap();
        if bak {
            rename(&new, &out).unwrap();
        }
        return;
    };
    zip.start_file("manifest.toml", options).unwrap();
    zip.write_all(data.as_bytes()).unwrap();
    let Ok(manifest) = toml::from_str::<Manifest>(&data) else {
        eprintln!("error parsing manifest.toml");
        remove_file(&out).unwrap();
        if bak {
            rename(&new, &out).unwrap();
        }
        return;
    };
    let payload = manifest.payload;
    if payload.components().count() != 1 {
        eprintln!("incorrect manifest");
        remove_file(&out).unwrap();
        if bak {
            rename(&new, &out).unwrap();
        }
        return;
    }
    path.pop();
    path.push(&payload);
    let Ok(data) = fs::read(&path) else {
        eprintln!("file {:?} does not exist", path);
        remove_file(&out).unwrap();
        if bak {
            rename(&new, &out).unwrap();
        }
        return;
    };
    zip.start_file(payload.to_str().unwrap(), options).unwrap();
    zip.write_all(data.as_slice()).unwrap();

    if let Some(icon) = manifest.icon {
        if icon.components().count() != 1 {
            eprintln!("incorrect manifest");
            remove_file(&out).unwrap();
            if bak {
                rename(&new, &out).unwrap();
            }
            return;
        }
        path.pop();
        path.push(&icon);
        let Ok(data) = fs::read(&path) else {
            eprintln!("file {:?} does not exist", path);
            remove_file(&out).unwrap();
            if bak {
                rename(&new, &out).unwrap();
            }
            return;
        };

        zip.start_file(icon.to_str().unwrap(), options).unwrap();
        zip.write_all(data.as_slice()).unwrap();
    }

    zip.finish().unwrap();

    if bak {
        remove_file(&new).unwrap();
    }
}