Add local password file manager

subdavis · subdavis · commit ae804e09bcb1 · 2017-12-17T18:35:45.000-05:00
diff --git a/background/background.js b/background/background.js
@@ -1,37 +1,11 @@
-/**
-
-The MIT License (MIT)
-
-Copyright (c) 2015 Steven Campbell.
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in
-all copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-THE SOFTWARE.
-
-*/
-
 "use strict";
 
 /*
   This page runs as an Event page, not a Background page, so don't use global variables
   (they will be lost)
 */
 
-import ProtectedMemory from '$services/protectedMemory.js'
+import { ProtectedMemory } from '$services/protectedMemory.js'
 import { Settings } from '$services/settings.js'
 
 function Background(protectedMemory, settings) {
diff --git a/services/localChromePasswordFileManager.js b/services/localChromePasswordFileManager.js
@@ -1,13 +1,12 @@
 "use strict";
-
+const Base64 = require('base64-arraybuffer')
 import { ChromePromiseApi } from '$lib/chrome-api-promise.js'
 
 const chromePromise = ChromePromiseApi()
 
 function LocalChromePasswordFileManager() {
   var exports = {
     key: 'local',
-    routePath: '/drag-drop-file',
     listDatabases: listDatabases,
     getDatabaseChoiceData: getDatabaseChoiceData,
     getChosenDatabaseFile: getChosenDatabaseFile,
@@ -17,12 +16,29 @@ function LocalChromePasswordFileManager() {
     title: 'Chrome Storage',
     icon: 'icon-upload',
     chooseTitle: 'File System',
-    chooseDescription: 'Upload files from your local or remote file-system.  A one-time copy of the file(s) will be saved in Chrome local storage.  If you update the database on your local system then you will have to re-upload it in order to see the changes.'
+    chooseDescription: 'Upload files from your local or remote file-system.  A one-time copy of the file(s) will be saved in Chrome local storage.  If you update the database on your local system then you will have to re-upload it in order to see the changes.',
+    login: enable,
+    logout: disable,
+    isLoggedIn: isEnabled
   };
 
   var backwardCompatibleVersion = 1; //missing version or version less than this is ignored due missing info or bugs in old storage
   var currentVersion = 1; //current version
 
+  function enable() {
+    return chromePromise.storage.local.set({'localPasswordFilesEnabled':true})
+  }
+
+  function disable () {
+    return chromePromise.storage.local.set({'localPasswordFilesEnabled':false})
+  }
+
+  function isEnabled () {
+    return chromePromise.storage.local.get('localPasswordFilesEnabled').then(result => {
+      return result.localPasswordFilesEnabled || false
+    })
+  }
+
   var savingLocks = [];  //prevent reading while an async save is ongoing
   function listDatabases() {
     return Promise.all(savingLocks).then(function() {
diff --git a/services/passwordFileStore.js b/services/passwordFileStore.js
@@ -4,7 +4,7 @@
  * Provides a container for various storage mechanisms (aka FileManagers)that can be injected,
  * so that the rest of the code can be independent of specifics.
 **/
-module.exports = function PasswordFileStoreRegistry() {
+function PasswordFileStoreRegistry() {
   var exports = {
     listFileManagers: listFileManagers,
     getChosenDatabaseFile: getChosenDatabaseFile
@@ -37,3 +37,5 @@ module.exports = function PasswordFileStoreRegistry() {
 
   return exports;
 }
+
+export { PasswordFileStoreRegistry }
diff --git a/services/protectedMemory.js b/services/protectedMemory.js
@@ -8,7 +8,7 @@
 
 let Base64 = require('base64-arraybuffer')
 
-module.exports = function ProtectedMemory() {
+function ProtectedMemory() {
   var my = {
     getData: getData,
     setData: setData,
@@ -137,3 +137,5 @@ module.exports = function ProtectedMemory() {
 
   return my;
 }
+
+export { ProtectedMemory }
diff --git a/services/sampleDatabaseFileManager.js b/services/sampleDatabaseFileManager.js
@@ -8,7 +8,6 @@ const chromePromise = ChromePromiseApi()
 function SampleDatabaseFileManager() {
   var exports = {
     key: 'sample',
-    routePath: '/sample-database',
     listDatabases: listDatabases,
     getDatabaseChoiceData: getDatabaseChoiceData,
     getChosenDatabaseFile: getChosenDatabaseFile,
diff --git a/services/secureCacheDisk.js b/services/secureCacheDisk.js
@@ -11,7 +11,7 @@
  * Max storage time is 40 minutes, which is the expected TTL of the secret.  You
  * can see details of the expiry time in chrome://identity-internals/
  */
-module.exports = function SecureCacheDisk(protectedMemory, secureCacheMemory, settings) {
+function SecureCacheDisk(protectedMemory, secureCacheMemory, settings) {
   var exports = {
     save: set,
     get: get,
@@ -135,3 +135,5 @@ module.exports = function SecureCacheDisk(protectedMemory, secureCacheMemory, se
 
   return exports;
 }
+
+export { SecureCacheDisk }
diff --git a/services/secureCacheMemory.js b/services/secureCacheMemory.js
@@ -3,7 +3,7 @@
 /**
  * Storage in background page memory.
  */
-module.exports = function SecureCacheMemory(protectedMemory) {
+function SecureCacheMemory(protectedMemory) {
   var exports = {}
 
   var awaiting = [];
@@ -84,3 +84,5 @@ module.exports = function SecureCacheMemory(protectedMemory) {
 
   return exports;
 }
+
+export { SecureCacheMemory }
diff --git a/services/sharedUrlFileManager.js b/services/sharedUrlFileManager.js
@@ -8,7 +8,6 @@ const chromePromise = ChromePromiseApi()
 function SharedUrlFileManager() {
   var exports = {
     key: 'shared-url',
-    routePath: '/shared-url',
     listDatabases: listDatabases,
     getDatabaseChoiceData: getDatabaseChoiceData,
     getChosenDatabaseFile: getChosenDatabaseFile,
diff --git a/src/Options.vue b/src/Options.vue
@@ -13,6 +13,7 @@
           v-if="show.manageDatabases.visible"
           :dropbox-file-manager="services.dropboxFileManager"
           :google-drive-manager="services.googleDrivePasswordFileManager"
+          :local-file-manager="services.localChromePasswordFileManager"
           :onedrive-manager="services.oneDriveFileManager"
           :sample-manager="services.sampleDatabaseFileManager"
           :shared-url-manager="services.sharedUrlFileManager"
@@ -32,12 +33,12 @@
 
 <script>
 // Singletons
-import ChromePromiseApi from '$lib/chrome-api-promise.js'
+import { ChromePromiseApi } from '$lib/chrome-api-promise.js'
 import { Settings } from '$services/settings.js'
-import ProtectedMemory from '$services/protectedMemory.js'
-import SecureCacheMemory from '$services/secureCacheMemory.js'
-import SecureCacheDisk from '$services/secureCacheDisk.js'
-import PasswordFileStore from '$services/passwordFileStore.js'
+import { ProtectedMemory } from '$services/protectedMemory.js'
+import { SecureCacheMemory } from '$services/secureCacheMemory.js'
+import { SecureCacheDisk } from '$services/secureCacheDisk.js'
+import { PasswordFileStoreRegistry } from '$services/passwordFileStore.js'
 import { KeyFileParser } from '$services/keyFileParser.js'
 // File Managers
 import { LocalChromePasswordFileManager } from '$services/localChromePasswordFileManager.js'
@@ -68,7 +69,7 @@ const sharedUrlFileManager = new SharedUrlFileManager()
 const oneDriveFileManager = new OneDriveFileManager(settings)
 const sampleDatabaseFileManager = new SampleDatabaseFileManager()
 
-const passwordFileStoreRegistry = new PasswordFileStore(localChromePasswordFileManager, dropboxFileManager, googleDrivePasswordFileManager, sharedUrlFileManager, sampleDatabaseFileManager, oneDriveFileManager)
+const passwordFileStoreRegistry = new PasswordFileStoreRegistry(localChromePasswordFileManager, dropboxFileManager, googleDrivePasswordFileManager, sharedUrlFileManager, sampleDatabaseFileManager, oneDriveFileManager)
 
 export default {
   name: 'app',
@@ -88,6 +89,7 @@ export default {
         settings,
         dropboxFileManager,
         googleDrivePasswordFileManager,
+        localChromePasswordFileManager,
         oneDriveFileManager,
         sampleDatabaseFileManager,
         sharedUrlFileManager,
diff --git a/src/Popup.vue b/src/Popup.vue
@@ -26,14 +26,14 @@
 <script>
 // Singletons
 import { Settings } from '$services/settings.js'
-import ProtectedMemory from '$services/protectedMemory.js'
+import { ProtectedMemory } from '$services/protectedMemory.js'
 import { KeepassHeader } from '$services/keepassHeader.js'
 import { KeepassReference } from '$services/keepassReference.js'
 import { KeepassService } from '$services/keepassService.js'
 import { UnlockedState } from '$services/unlockedState.js'
-import SecureCacheMemory from '$services/secureCacheMemory.js'
-import SecureCacheDisk from '$services/secureCacheDisk.js'
-import PasswordFileStore from '$services/passwordFileStore.js'
+import { SecureCacheMemory } from '$services/secureCacheMemory.js'
+import { SecureCacheDisk } from '$services/secureCacheDisk.js'
+import { PasswordFileStoreRegistry } from '$services/passwordFileStore.js'
 import { Links } from '$services/links.js'
 // File Managers
 import { LocalChromePasswordFileManager } from '$services/localChromePasswordFileManager.js'
@@ -65,7 +65,7 @@ const sharedUrlFileManager = new SharedUrlFileManager()
 const oneDriveFileManager = new OneDriveFileManager(settings)
 const sampleDatabaseFileManager = new SampleDatabaseFileManager()
 
-const passwordFileStoreRegistry = new PasswordFileStore(localChromePasswordFileManager, dropboxFileManager, googleDrivePasswordFileManager, sharedUrlFileManager, sampleDatabaseFileManager, oneDriveFileManager)
+const passwordFileStoreRegistry = new PasswordFileStoreRegistry(localChromePasswordFileManager, dropboxFileManager, googleDrivePasswordFileManager, sharedUrlFileManager, sampleDatabaseFileManager, oneDriveFileManager)
 const keepassService = new KeepassService(keepassHeader, settings, passwordFileStoreRegistry, keepassReference)
 
 export default {
diff --git a/src/components/LocalPasswordFileProvider.vue b/src/components/LocalPasswordFileProvider.vue
@@ -0,0 +1,130 @@
+<!-- 
+	SharedLinkProvider:
+	Simple http provider that can also handle Dropbox Shared Links
+-->
+<template>
+  <div class="box-bar roomy database-manager">
+  	<generic-provider-ui
+  		:busy="busy"
+  		:databases="databases"
+  		:loggedIn="loggedIn"
+  		:error="messages.error"
+  		:provider-manager="providerManager"
+  		:toggle-login="toggleLogin"
+  		:removeable="true"
+  		:remove-function="removePasswordFile"></generic-provider-ui>
+		<div class="url-form shared-link-box" v-if="loggedIn">
+		  <input type="file" style="display:none" id="file-selector" name='file' @change="handleAdd" multiple />
+		  <a class="waves-effect waves-light btn" @click="selectFile">Add URL Source</a>
+		</div>
+  </div>
+</template>
+
+<script>
+const Base64 = require('base64-arraybuffer')
+import GenericProviderUi from '@/components/GenericProviderUi'
+
+export default {
+	data () {
+		return {
+			busy: false,
+			currentUrl: "",
+			currentUrlTitle: "",
+			databases: [],
+			loggedIn: false,
+			messages: {
+				error: ""
+			}
+		}
+	},
+	components: {
+		GenericProviderUi
+	},
+	props: {
+		providerManager: Object,
+		settings: Object
+	},
+	methods: {
+		toggleLogin () {
+			if (this.loggedIn){
+				this.settings.disableDatabaseProvider(this.providerManager)
+				this.providerManager.logout().then(() => {this.loggedIn = false})
+			} else {
+				this.providerManager.login().then(() => {this.loggedIn = true})
+			}
+		},
+		selectFile (event) {
+	    document.getElementById('file-selector').click();
+	  },
+	  removePasswordFile (index) {
+	  	if (index >= this.databases.length || index < 0)
+	  		return // not a valid index...
+	  	let fi = this.databases[index]
+	    this.providerManager.deleteDatabase(fi)
+	    	.then(this.providerManager.listDatabases)
+	    	.then(files => {
+		      this.databases = files
+		    })
+		},
+		handleAdd (event) {
+			let files = event.target.files;
+			this.messages.error = ""
+			for (var i=0; i<files.length; i++){
+				let reader = new FileReader()
+				let fp = files[i]
+				reader.readAsArrayBuffer(fp)
+				reader.onload = (e) => {
+					if (fp.name.indexOf('.kdbx') < 0 || fp.size < 70) {
+	          this.messages.error += fp.name + " is not a valid KeePass v2+ file. "
+	          return;
+	        }
+
+	        var fi = {
+	          title: fp.name,
+	          lastModified: fp.lastModified,
+	          lastModifiedDate: fp.lastModifiedDate,
+	          size: fp.size,
+	          type: fp.type,
+	          data: Base64.encode(e.target.result)
+	        }
+
+					var existingIndex = null;
+	        this.databases.forEach(function(existingFile, index) {
+	          if (existingFile.title == fi.title) {
+	            existingIndex = index;
+	          }
+	        });
+
+	        if (existingIndex == null) {
+	          //add
+	          this.databases.push(fi);
+	        } else {
+	          //replace
+	          this.databases[existingIndex] = fi;
+	        }
+
+	        return this.providerManager.saveDatabase({
+	          title: fi.title,
+	          data: fi.data,
+	          lastModified: fi.lastModified
+	        })
+	      } // end onload
+			}
+		}
+	},
+	mounted () {
+		this.providerManager.isLoggedIn().then(loggedIn => {
+			console.log("WOOOOO", loggedIn)
+			this.loggedIn = loggedIn
+		})
+		this.providerManager.listDatabases().then(databases => {
+			if (databases !== false)
+				this.databases = databases
+		})
+	}
+}
+</script>
+
+<style lang="scss">
+@import "../styles/settings.scss";
+</style>
diff --git a/src/components/ManageDatabases.vue b/src/components/ManageDatabases.vue
diff --git a/src/components/OauthProvider.vue b/src/components/OauthProvider.vue
