Json results for GUIs

Author: replydev

src/argument_functions.rs

@@ -3,6 +3,7 @@ use std::io::prelude::*;
 use super::utils::get_db_path;
 use super::database_loader;
 use super::cryptograpy;
+use super::otp_helper;
 
 pub fn import(args: Vec<String>){
     import_database(&args[2]);
@@ -71,6 +72,15 @@ pub fn export(args: Vec<String>){
     }
 }
 
+pub fn json(args: Vec<String>){
+    if args.len() == 2{
+        println!("{}",otp_helper::get_json_results())
+    }
+    else{
+        println!("Invalid argument, type cotp --json");
+    }
+}
+
 fn import_database(filename: &String){
     let mut unencrypted_content = read_to_string(filename).unwrap();
     let encrypted_content = cryptograpy::encrypt_string(&mut unencrypted_content,&cryptograpy::prompt_for_passwords("Insert password for database encryption: "));

src/main.rs

@@ -29,7 +29,7 @@ fn main() {
     print_title(version);
     let init_result = init();
     match init_result {
-        Ok(()) => println!("sodiumoxide init succeeded"),
+        Ok(()) => {},
         Err(()) => { 
             println!("Failed to init sodiumoxide");
             return;
@@ -54,6 +54,7 @@ fn args_parser(args: Vec<String>) -> bool {
         "-r"  | "--remove" => argument_functions::remove(args),
         "-e"  | "--edit" => argument_functions::edit(args),
         "-ex" | "--export" => argument_functions::export(args),
+        "-j"  | "--json" => argument_functions::json(args),
         _=>{
             println!("Invalid argument: {}, type cotp -h to get command options", args[1]);
             return true;

src/otp_helper.rs

@@ -1,6 +1,39 @@
 use otp::make_totp;
 use super::database_loader;
+use serde_json;
+use serde::{Deserialize, Serialize};
 
+#[derive(Serialize, Deserialize)]
+struct JsonResult{
+    index: usize,
+    issuer: String,
+    label: String,
+    otp_code: String,
+}
+
+impl JsonResult {
+    pub fn new(index: usize, issuer: String, label: String,otp_code: String) -> JsonResult {
+        JsonResult{
+            index: index, 
+            issuer: issuer,
+            label: label,
+            otp_code: otp_code
+        }
+    }
+
+    pub fn set_index(&mut self, index: usize) {
+        self.index = index;
+    }
+    pub fn set_issuer(&mut self,issuer: String){
+        self.issuer = issuer;
+    }
+    pub fn set_label(&mut self,label: String){
+        self.label = label;
+    }
+    pub fn set_otp_code(&mut self,otp_code: String){
+        self.otp_code = otp_code;
+    }
+}
 
 pub fn show_codes(){
     let elements: Vec<database_loader::OTPElement> = database_loader::read_from_file();
@@ -27,4 +60,19 @@ fn get_good_otp_code(element: &database_loader::OTPElement) -> String {
         s_otp = String::from("0") + &s_otp;
     }
     s_otp
+}
+
+pub fn get_json_results() -> String{
+    let elements: Vec<database_loader::OTPElement> = database_loader::read_from_file();
+
+    let mut results: Vec<JsonResult> = Vec::new();
+
+    for i in 0..elements.len() {
+        let otp_code = get_good_otp_code(&elements[i]);
+        results.push(JsonResult::new(i+1,elements[i].issuer(),elements[i].label(),otp_code))
+    }
+
+    let json_string: &str = &serde_json::to_string_pretty(&elements).unwrap();
+
+    json_string.to_string()
 }