modm 0.1.1

A git like cli tool for machine learning version control
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
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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151

//! # modm - a library for masterminds
use std::{fs, io::Error, any::Any};
use serde::{Serialize, Deserialize};
use dialoguer::{theme::ColorfulTheme, Input};
use std::path::Path;
use std::env;
use std::path::PathBuf;

mod utils;
use utils::{AuthorConfig, Config, GitHubConfig, ModelConfig};

/// initializes the model repository in the .modm folder
///
/// ```
///  use modm::init_repository;
///  use std::fs;
///  use std::path::Path;
/// 
///  init_repository(); 
/// 
///  let dir_path = Path::new(".modm");
///  
/// // check if the directory exists
///  if dir_path.is_dir() {
///      assert!(true);
///  }  else {
///      match fs::metadata(dir_path) {
///          Ok(_) => println!("File exists, but is not a directory."),
///          Err(_) => println!("Directory does not exist."),
///      }
///     assert!(false);
///  }
///
/// ```
pub fn init_repository() -> Result<(), Error> {
    let current_dir = env::current_dir().unwrap();
    let folder_name = ".modm";
    let full_path = current_dir.join(folder_name);

    if let Err(e) = fs::create_dir(full_path) {
        println!(".modm repository already exists");
        return Err(e);
    } else {
        println!("Initialized modm repository in {}", folder_name);
    }

    return Ok(())
}

/// creates the model config file
pub fn create_config() -> Result<(), Error> {

    let current_dir = env::current_dir().unwrap();

    let model_name: String = Input::with_theme(&ColorfulTheme::default())
        .with_prompt("Model name: ")
        .interact_text()
        .unwrap();

    let model_version: String = Input::with_theme(&ColorfulTheme::default())
        .with_prompt("Model version: ")
        .interact_text()
        .unwrap();

    let folder_name: String = Input::with_theme(&ColorfulTheme::default())
        .with_prompt("Model output directory: ")
        .interact_text()
        .unwrap();

    let github_repo: String = Input::with_theme(&ColorfulTheme::default())
        .with_prompt("GitHub repo: ")
        .interact_text()
        .unwrap();

    let author_name: String = Input::with_theme(&ColorfulTheme::default())
        .with_prompt("Author name: ")
        .allow_empty(true)
        .interact_text()
        .unwrap();

    let author_email: String = Input::with_theme(&ColorfulTheme::default())
        .with_prompt("Author email: ")
        .allow_empty(true)
        .interact_text()
        .unwrap();



    let config = Config {
        model: ModelConfig {
            name: model_name,
            version: model_version,
            directory: folder_name.clone(),
        },
        github: GitHubConfig {
            repo: github_repo,
        },
        author: AuthorConfig {
            author: author_name,
            email: author_email,
        }
    };

    let config_json = serde_json::to_string_pretty(&config).unwrap();
    let file_path = ".modm/config.json";
    let full_path = current_dir.join(file_path);
    let config_file = Path::new(full_path.to_str().unwrap());

    if config_file.is_file() {
        println!("Config file already exists");
        return Ok(())
    }

    fs::write(config_file, config_json)?;

    println!("Initialized modm repository to {:?} successfuly!", folder_name.clone());

    return Ok(())
}


/// creates the model config file
///
/// ```
///  use modm::{init_repository, create_config};
///  use std::fs;
///  use std::path::Path;
/// 
///  init_repository(); 
///  create_config();
/// 
///  let dir_path = Path::new(".modm/config.json");
///  
/// // check if the directory exists
///  if dir_path.is_file() {
///      assert!(true);
///  }  else {
///      match fs::metadata(dir_path) {
///          Ok(_) => println!("File exists, but is not a directory."),
///          Err(_) => println!("Directory does not exist."),
///      }
///     assert!(false);
///  }
///
/// ```
pub fn init_modm() -> Result<(), Error> {
    init_repository()?;
    create_config()?;

    return Ok(())
}