atac 0.18.0

Arguably a Terminal API Client. Feature-full, free, open-source, offline and account-less.
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

use std::fs;
use std::fs::OpenOptions;
use std::io::{Read, Write};
use std::path::PathBuf;

use tracing::{info, trace, warn};

use crate::app::app::App;
use crate::cli::args::ARGS;
use crate::panic_error;
use crate::models::collection::{Collection, CollectionFileFormat};
use crate::models::collection::CollectionFileFormat::{Json, Yaml};

impl App<'_> {
    /// Set the app request to the requests found in the collection file
    pub fn set_collections_from_file(&mut self, path_buf: PathBuf, file_format: CollectionFileFormat) {
        let mut file_content = String::new();

        trace!("Trying to open \"{}\" collection", path_buf.display());

        let mut collection_file = OpenOptions::new()
            .read(true)
            .write(true)
            .create(true)
            .open(path_buf.clone())
            .expect("\tCould not open collection file");

        collection_file.read_to_string(&mut file_content).expect("\tCould not read collection file");

        let mut collection: Collection = match file_format {
            Json => match serde_json::from_str(&file_content) {
                Ok(collection) => collection,
                Err(e) => panic_error(format!("Could not parse JSON collection \"{}\"\n\t{}", path_buf.display(), e.to_string()))
            },
            Yaml => match serde_yaml::from_str(&file_content) {
                Ok(collection) => collection,
                Err(e) => panic_error(format!("Could not parse YAML collection \"{}\"\n\t{}", path_buf.display(), e))
            }
        };

        collection.path = path_buf;
        collection.file_format = file_format;

        self.collections.push(collection);

        trace!("Collection file parsed!");
    }

    /// Save app collection in the collection file through a temporary file
    pub fn save_collection_to_file(&mut self, collection_index: usize) {
        if !ARGS.should_save {
            warn!("Dry-run, not saving the collection");
            return;
        }

        let collection = &self.collections[collection_index];

        info!("Saving collection \"{}\"", collection.name);

        let temp_file_name = format!("{}_", collection.path.file_name().unwrap().to_str().unwrap());

        let temp_file_path = collection.path.with_file_name(temp_file_name);
        
        let mut temp_file = OpenOptions::new()
            .write(true)
            .create(true)
            .truncate(true)
            .open(&temp_file_path)
            .expect("Could not open temp file");

        let collection_stringed = match collection.file_format {
            Json => serde_json::to_string_pretty(collection).expect("Could not serialize collection to JSON"),
            Yaml => serde_yaml::to_string(collection).expect("Could not serialize collection to YAML")
        };

        temp_file.write_all(collection_stringed.as_bytes()).expect("Could not write to temp file");
        temp_file.flush().unwrap();

        fs::rename(temp_file_path, &collection.path).expect("Could not move temp file to collection file");
        
        trace!("Collection saved");
    }

    /// Delete collection file
    pub fn delete_collection_file(&mut self, collection: Collection) {
        if !ARGS.should_save {
            return;
        }

        fs::remove_file(&collection.path).expect("Could not delete collection file");
    }
}