From 05c36b300db460925e834204b8e7c668b57e3948 Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Mon, 25 Nov 2019 16:10:51 -0500
Subject: [PATCH] codebook csv importer

---
 src/importers/codebookCsvImporter.ts | 46 ++++++++++++++++++++++++++++
 src/services/import.service.ts       |  4 +++
 2 files changed, 50 insertions(+)
 create mode 100644 src/importers/codebookCsvImporter.ts

diff --git a/src/importers/codebookCsvImporter.ts b/src/importers/codebookCsvImporter.ts
new file mode 100644
index 00000000000..59ebdd4a8fe
--- /dev/null
+++ b/src/importers/codebookCsvImporter.ts
@@ -0,0 +1,46 @@
+import { BaseImporter } from './baseImporter';
+import { Importer } from './importer';
+
+import { ImportResult } from '../models/domain/importResult';
+
+export class CodebookCsvImporter extends BaseImporter implements Importer {
+    parse(data: string): ImportResult {
+        const result = new ImportResult();
+        const results = this.parseCsv(data, true);
+        if (results == null) {
+            result.success = false;
+            return result;
+        }
+
+        results.forEach((value) => {
+            this.processFolder(result, this.getValueOrDefault(value.Category));
+
+            const cipher = this.initLoginCipher();
+            cipher.favorite = this.getValueOrDefault(value.Favorite) === 'True';
+            cipher.name = this.getValueOrDefault(value.Entry, '--');
+            cipher.notes = this.getValueOrDefault(value.Note);
+            cipher.login.username = this.getValueOrDefault(value.Username, value.Email);
+            cipher.login.password = this.getValueOrDefault(value.Password);
+            cipher.login.totp = this.getValueOrDefault(value.TOTP);
+            cipher.login.uris = this.makeUriArray(value.Website);
+
+            if (!this.isNullOrWhitespace(value.Username)) {
+                this.processKvp(cipher, 'Email', value.Email);
+            }
+            this.processKvp(cipher, 'Phone', value.Phone);
+            this.processKvp(cipher, 'PIN', value.PIN);
+            this.processKvp(cipher, 'Account', value.Account);
+            this.processKvp(cipher, 'Date', value.Date);
+
+            this.cleanupCipher(cipher);
+            result.ciphers.push(cipher);
+        });
+
+        if (this.organization) {
+            this.moveFoldersToCollections(result);
+        }
+
+        result.success = true;
+        return result;
+    }
+}
diff --git a/src/services/import.service.ts b/src/services/import.service.ts
index 2526485bc9b..df6735bd472 100644
--- a/src/services/import.service.ts
+++ b/src/services/import.service.ts
@@ -33,6 +33,7 @@ import { BlurCsvImporter } from '../importers/blurCsvImporter';
 import { ButtercupCsvImporter } from '../importers/buttercupCsvImporter';
 import { ChromeCsvImporter } from '../importers/chromeCsvImporter';
 import { ClipperzHtmlImporter } from '../importers/clipperzHtmlImporter';
+import { CodebookCsvImporter } from '../importers/codebookCsvImporter';
 import { DashlaneJsonImporter } from '../importers/dashlaneJsonImporter';
 import { EnpassCsvImporter } from '../importers/enpassCsvImporter';
 import { EnpassJsonImporter } from '../importers/enpassJsonImporter';
@@ -125,6 +126,7 @@ export class ImportService implements ImportServiceAbstraction {
         { id: 'logmeoncecsv', name: 'LogMeOnce (csv)' },
         { id: 'blackberrycsv', name: 'BlackBerry Password Keeper (csv)' },
         { id: 'buttercupcsv', name: 'Buttercup (csv)' },
+        { id: 'codebookcsv', name: 'Codebook (csv)' },
     ];
 
     constructor(private cipherService: CipherService, private folderService: FolderService,
@@ -267,6 +269,8 @@ export class ImportService implements ImportServiceAbstraction {
                 return new BlackBerryCsvImporter();
             case 'buttercupcsv':
                 return new ButtercupCsvImporter();
+            case 'codebookcsv':
+                return new CodebookCsvImporter();
             default:
                 return null;
         }