authy importer

Author: replydev

converters/authy.py

@@ -0,0 +1,34 @@
+import sys
+import xml.etree.ElementTree as ET
+import json
+
+def fetch_json(filename):
+    tree = ET.parse(filename)
+    j = tree.find("string").text.replace(""","\"")
+    return json.loads(j)
+    
+def object_to_cotp_json(d):
+    final = []
+    for element in d:
+        final.append(
+            {
+                'label': element['name'],
+                'secret': element['decryptedSecret'],
+                'issuer': "",
+                'digits': int(element['digits'])
+            }
+        )
+    return final
+
+def main():
+    if len(sys.argv) != 3:
+        print("Usage: python authy.py [INPUT_FILE] [OUTPUT_FILE]")
+        return
+
+    data = fetch_json(sys.argv[1])
+    output_file = open(sys.argv[2],'w')
+    output_file.write(json.dumps(object_to_cotp_json(data)))
+    output_file.close()
+
+
+main()

converters/gauth_to_json.py renamed to converters/gauth.py

@@ -12,6 +12,7 @@ def get_accounts(filename):
                 'label': row[0],
                 'secret': row[1],
                 'issuer': row[2],
+                'digits': 6,
             }
         )
     c.close()

src/argument_functions.rs

@@ -27,7 +27,9 @@ pub fn import(args: Vec<String>){
         match &args[2][..]{
             "cotp" | "andotp" => result = importers::and_otp::import(&args[3]),
             "aegis" => result = importers::aegis::import(&args[3]),
-            "gauth" | "google_authenticator" => result = importers::gauth::import(&args[3]),
+            "gauth" | 
+            "google_authenticator" |
+            "authy" => result = importers::converted::import(&args[3]),
             _=> {
                 println!("Invalid argument: {}", &args[2]);
                 return;
@@ -57,6 +59,8 @@ pub fn import(args: Vec<String>){
         println!("\"cotp\"");
         println!("\"aegis\"");
         println!("\"andotp\"");
+        println!("\"gauth\" or \"google_authenticator\"");
+        println!("\"authy\"");
     }
 }
 

src/importers/gauth.rs renamed to src/importers/converted.rs

@@ -4,33 +4,35 @@ use serde_json;
 use serde::Deserialize;
 
 #[derive(Deserialize)]
-struct GAuthJson{
+struct ConvertedJson{
     label: Option<String>,
     secret: String,
     issuer: Option<String>,
+    digits: u64,
 }
 
 pub fn import(filepath: &str) -> Result<Vec<OTPElement>,String> {
     let file_to_import_contents = read_to_string(filepath).unwrap();
-    let result: Result<Vec<GAuthJson>,serde_json::Error> = serde_json::from_str(&file_to_import_contents);
-    let gauth: Vec<GAuthJson>;
+    let result: Result<Vec<ConvertedJson>,serde_json::Error> = serde_json::from_str(&file_to_import_contents);
+    let vector: Vec<ConvertedJson>;
 
     match result{
-        Ok(r) => gauth = r,
+        Ok(r) => vector = r,
         Err(e) => return Err(format!("{}",e)),
     }
 
     let mut elements: Vec<OTPElement> = Vec::new();
 
-    for i in 0..gauth.len(){
-        let secret = gauth[i].secret.to_owned();
-        let issuer = gauth[i].issuer.to_owned().unwrap_or_default();
-        let label = gauth[i].label.to_owned().unwrap_or_default();
+    for i in 0..vector.len(){
+        let secret = vector[i].secret.to_owned();
+        let issuer = vector[i].issuer.to_owned().unwrap_or_default();
+        let label = vector[i].label.to_owned().unwrap_or_default();
+        let digits = vector[i].digits;
         elements.push(OTPElement::new(
             secret,
             issuer,
             label,
-            6,
+            digits,
             String::from("TOTP"), 
             String::from("SHA1"), 
             String::from(""), 

src/importers/mod.rs

@@ -1,3 +1,3 @@
 pub mod and_otp;
 pub mod aegis;
-pub mod gauth;
+pub mod converted;