suckit 0.2.0

SuckIT, Suck the InTernet
Documentation 
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

use pathdiff;
use std::fs;

use std::io::Write;
use std::path::PathBuf;

use symlink::symlink_file;

use crate::{error, warn};

///Save content in a file
pub fn save_file(file_name: &str, content: &[u8], path: &Option<PathBuf>) {
    let path = match path {
        Some(path) => path.join(file_name),
        None => PathBuf::from(file_name),
    };

    if let Some(parent) = path.parent() {
        if let Err(err) = fs::create_dir_all(parent) {
            error!("Couldn't create folder {}: {}", parent.display(), err);
        }
    }

    let mut file = match fs::File::create(&path) {
        Err(err) => error!("Couldn't create {}: {}", path.display(), err),
        Ok(file) => file,
    };

    if let Err(err) = file.write_all(content) {
        error!("Couldn't write to {}: {}", path.display(), err);
    }
}

///Create a symlink
pub fn symlink(source: &str, destination: &str, path: &Option<PathBuf>) {
    let source = match path {
        Some(path) => path.join(source),
        None => PathBuf::from(source),
    };

    if let Some(parent) = source.parent() {
        match fs::create_dir_all(parent) {
            Err(err) => {
                error!("Couldn't create folder {}: {}", parent.display(), err);
            }
            Ok(()) => (),
        }
    }

    let destination = match path {
        Some(path) => path.join(destination),
        None => PathBuf::from(destination),
    };

    let target = pathdiff::diff_paths(&destination, &source.parent().unwrap()).unwrap();

    if let Err(err) = symlink_file(&target, &source) {
        warn!(
            "Couldn't create symlink\n{} -> {}:\n{:#?}",
            source.display(),
            target.display(),
            err,
        );
    }
}