angular .module('bit.vault') .controller('vaultController', function ($scope, $uibModal, apiService, $filter, cryptoService, authService, toastr, cipherService, $q, $localStorage, $timeout, $rootScope, $state, $analytics, constants) { $scope.loading = true; $scope.ciphers = []; $scope.folderCount = 0; $scope.collectionCount = 0; $scope.firstCollectionId = null; $scope.constants = constants; $scope.favoriteCollapsed = $localStorage.collapsedFolders && 'favorite' in $localStorage.collapsedFolders; $scope.groupingIdFilter = undefined; $scope.typeFilter = undefined; if ($state.params.refreshFromServer) { $rootScope.vaultGroupings = $rootScope.vaultCiphers = null; } $scope.$on('$viewContentLoaded', function () { if ($rootScope.vaultGroupings && $rootScope.vaultCiphers) { $scope.loading = false; loadGroupingData($rootScope.vaultGroupings); loadCipherData($rootScope.vaultCiphers); return; } loadDataFromServer(); }); function loadDataFromServer() { var decGroupings = [{ id: null, name: 'No Folder', folder: true }]; var collectionPromise = apiService.collections.listMe({ writeOnly: false }, function (collections) { for (var i = 0; i < collections.Data.length; i++) { var decCollection = cipherService.decryptCollection(collections.Data[i], null, true); decCollection.collection = true; decGroupings.push(decCollection); } $scope.collectionCount = collections.Data.length; }).$promise; var folderPromise = apiService.folders.list({}, function (folders) { for (var i = 0; i < folders.Data.length; i++) { var decFolder = cipherService.decryptFolderPreview(folders.Data[i]); decFolder.folder = true; decGroupings.push(decFolder); } $scope.folderCount = folders.Data.length; }).$promise; var groupingPromise = $q.all([collectionPromise, folderPromise]).then(function () { loadGroupingData(decGroupings); }); var cipherPromise = apiService.ciphers.list({}, function (ciphers) { var decCiphers = []; for (var i = 0; i < ciphers.Data.length; i++) { var decCipher = cipherService.decryptCipherPreview(ciphers.Data[i]); decCiphers.push(decCipher); } groupingPromise.then(function () { loadCipherData(decCiphers); }); }).$promise; $q.all([cipherPromise, groupingPromise]).then(function () { $scope.loading = false; }); } function loadGroupingData(decGroupings) { $rootScope.vaultGroupings = $filter('orderBy')(decGroupings, ['folder', groupingSort]); var collections = $filter('filter')($rootScope.vaultGroupings, { collection: true }); if (collections && collections.length) { $scope.firstCollectionId = collections[0].id; } } function loadCipherData(decCiphers) { angular.forEach($rootScope.vaultGroupings, function (grouping, groupingIndex) { grouping.collapsed = $localStorage.collapsedFolders && (grouping.id || 'none') in $localStorage.collapsedFolders; angular.forEach(decCiphers, function (cipherValue) { if (cipherValue.favorite) { cipherValue.sort = -1; } else if (grouping.folder && cipherValue.folderId == grouping.id) { cipherValue.sort = groupingIndex; } else if (grouping.collection && cipherValue.collectionIds.indexOf(grouping.id) > -1) { cipherValue.sort = groupingIndex; } }); }); $rootScope.vaultCiphers = $scope.ciphers = $filter('orderBy')(decCiphers, ['sort', 'name', 'subTitle']); var chunks = chunk($rootScope.vaultCiphers, 400); if (chunks.length > 0) { $scope.ciphers = chunks[0]; var delay = 200; angular.forEach(chunks, function (value, index) { delay += 200; // skip the first chuck if (index > 0) { $timeout(function () { Array.prototype.push.apply($scope.ciphers, value); }, delay); } }); } } function sortScopedCipherData() { $rootScope.vaultCiphers = $scope.ciphers = $filter('orderBy')($rootScope.vaultCiphers, ['name', 'subTitle']); } function chunk(arr, len) { var chunks = [], i = 0, n = arr.length; while (i < n) { chunks.push(arr.slice(i, i += len)); } return chunks; } function groupingSort(item) { if (!item.id) { return ''; } return item.name.toLowerCase(); } $scope.clipboardError = function (e) { alert('Your web browser does not support easy clipboard copying. ' + 'Edit the item and copy it manually instead.'); }; $scope.collapseExpand = function (grouping, favorite) { if (!$localStorage.collapsedFolders) { $localStorage.collapsedFolders = {}; } var id = favorite ? 'favorite' : (grouping.id || 'none'); if (id in $localStorage.collapsedFolders) { delete $localStorage.collapsedFolders[id]; } else { $localStorage.collapsedFolders[id] = true; } }; $scope.editCipher = function (cipher) { var editModel = $uibModal.open({ animation: true, templateUrl: 'app/vault/views/vaultEditCipher.html', controller: 'vaultEditCipherController', resolve: { cipherId: function () { return cipher.id; } } }); editModel.result.then(function (returnVal) { if (returnVal.action === 'edit') { var index = $scope.ciphers.indexOf(cipher); if (index > -1) { $rootScope.vaultCiphers[index] = returnVal.data; } sortScopedCipherData(); } else if (returnVal.action === 'partialEdit') { cipher.folderId = returnVal.data.folderId; cipher.favorite = returnVal.data.favorite; } else if (returnVal.action === 'delete') { removeCipherFromScopes(cipher); } }); }; $scope.$on('vaultAddCipher', function (event, args) { $scope.addCipher(); }); $scope.addCipher = function (grouping, favorite) { var addModel = $uibModal.open({ animation: true, templateUrl: 'app/vault/views/vaultAddCipher.html', controller: 'vaultAddCipherController', resolve: { selectedFolder: function () { return grouping.folder ? grouping : null; }, checkedFavorite: function () { return favorite; } } }); addModel.result.then(function (addedCipher) { $rootScope.vaultCiphers.push(addedCipher); sortScopedCipherData(); }); }; $scope.deleteCipher = function (cipher) { if (!confirm('Are you sure you want to delete this item (' + cipher.name + ')?')) { return; } apiService.ciphers.del({ id: cipher.id }, function () { $analytics.eventTrack('Deleted Item'); removeCipherFromScopes(cipher); }); }; $scope.attachments = function (cipher) { authService.getUserProfile().then(function (profile) { return { isPremium: profile.premium, orgUseStorage: cipher.organizationId && !!profile.organizations[cipher.organizationId].maxStorageGb }; }).then(function (perms) { if (!cipher.hasAttachments) { if (cipher.organizationId && !perms.orgUseStorage) { $uibModal.open({ animation: true, templateUrl: 'app/views/paidOrgRequired.html', controller: 'paidOrgRequiredController', resolve: { orgId: function () { return cipher.organizationId; } } }); return; } if (!cipher.organizationId && !perms.isPremium) { $uibModal.open({ animation: true, templateUrl: 'app/views/premiumRequired.html', controller: 'premiumRequiredController' }); return; } } if (!cipher.organizationId && !cryptoService.getEncKey()) { toastr.error('You cannot use this feature until you update your encryption key.', 'Feature Unavailable'); return; } var attachmentModel = $uibModal.open({ animation: true, templateUrl: 'app/vault/views/vaultAttachments.html', controller: 'vaultAttachmentsController', resolve: { cipherId: function () { return cipher.id; } } }); attachmentModel.result.then(function (hasAttachments) { cipher.hasAttachments = hasAttachments; }); }); }; $scope.editFolder = function (folder) { var editModel = $uibModal.open({ animation: true, templateUrl: 'app/vault/views/vaultEditFolder.html', controller: 'vaultEditFolderController', size: 'sm', resolve: { folderId: function () { return folder.id; } } }); editModel.result.then(function (editedFolder) { folder.name = editedFolder.name; }); }; $scope.$on('vaultAddFolder', function (event, args) { $scope.addFolder(); }); $scope.addFolder = function () { var addModel = $uibModal.open({ animation: true, templateUrl: 'app/vault/views/vaultAddFolder.html', controller: 'vaultAddFolderController', size: 'sm' }); addModel.result.then(function (addedFolder) { $rootScope.vaultGroupings.push(addedFolder); loadGroupingData($rootScope.vaultGroupings); }); }; $scope.deleteFolder = function (folder) { if (!confirm('Are you sure you want to delete this folder (' + folder.name + ')?')) { return; } apiService.folders.del({ id: folder.id }, function () { $analytics.eventTrack('Deleted Folder'); var index = $rootScope.vaultGroupings.indexOf(folder); if (index > -1) { $rootScope.vaultGroupings.splice(index, 1); } }); }; $scope.canDeleteFolder = function (folder) { if (!folder || !folder.id || !$rootScope.vaultCiphers) { return false; } var ciphers = $filter('filter')($rootScope.vaultCiphers, { folderId: folder.id }); return ciphers && ciphers.length === 0; }; $scope.share = function (cipher) { var modal = $uibModal.open({ animation: true, templateUrl: 'app/vault/views/vaultShareCipher.html', controller: 'vaultShareCipherController', resolve: { cipherId: function () { return cipher.id; } } }); modal.result.then(function (orgId) { cipher.organizationId = orgId; }); }; $scope.editCollections = function (cipher) { var modal = $uibModal.open({ animation: true, templateUrl: 'app/vault/views/vaultCipherCollections.html', controller: 'vaultCipherCollectionsController', resolve: { cipherId: function () { return cipher.id; } } }); modal.result.then(function (response) { if (response.collectionIds && !response.collectionIds.length) { removeCipherFromScopes(cipher); } else if (response.collectionIds) { cipher.collectionIds = response.collectionIds; } }); }; $scope.filterGrouping = function (grouping) { $scope.groupingIdFilter = grouping.id; if ($.AdminLTE && $.AdminLTE.layout) { $timeout(function () { $.AdminLTE.layout.fix(); }, 0); } }; $scope.filterType = function (type) { $scope.typeFilter = type; if ($.AdminLTE && $.AdminLTE.layout) { $timeout(function () { $.AdminLTE.layout.fix(); }, 0); } }; $scope.clearFilters = function () { $scope.groupingIdFilter = undefined; $scope.typeFilter = undefined; if ($.AdminLTE && $.AdminLTE.layout) { $timeout(function () { $.AdminLTE.layout.fix(); }, 0); } }; $scope.groupingFilter = function (grouping) { return $scope.groupingIdFilter === undefined || grouping.id === $scope.groupingIdFilter; }; $scope.cipherFilter = function (grouping) { return function (cipher) { var matchesGrouping = grouping === null; if (!matchesGrouping && grouping.folder && cipher.folderId === grouping.id) { matchesGrouping = true; } else if (!matchesGrouping && grouping.collection && cipher.collectionIds.indexOf(grouping.id) > -1) { matchesGrouping = true; } return matchesGrouping && ($scope.typeFilter === undefined || cipher.type === $scope.typeFilter); }; }; $scope.unselectAll = function () { selectAll(false); }; $scope.selectFolder = function (folder, $event) { var checkbox = $($event.currentTarget).closest('.box').find('input[name="cipherSelection"]'); checkbox.prop('checked', true); }; $scope.select = function ($event) { var checkbox = $($event.currentTarget).closest('tr').find('input[name="cipherSelection"]'); checkbox.prop('checked', !checkbox.prop('checked')); }; function distinct(value, index, self) { return self.indexOf(value) === index; } function getSelectedCiphers() { return $('input[name="cipherSelection"]:checked').map(function () { return $(this).val(); }).get().filter(distinct); } function selectAll(select) { $('input[name="cipherSelection"]').prop('checked', select); } $scope.bulkMove = function () { var ids = getSelectedCiphers(); if (ids.length === 0) { alert('You have not selected anything.'); return; } var modal = $uibModal.open({ animation: true, templateUrl: 'app/vault/views/vaultMoveCiphers.html', controller: 'vaultMoveCiphersController', size: 'sm', resolve: { ids: function () { return ids; } } }); modal.result.then(function (folderId) { for (var i = 0; i < ids.length; i++) { var cipher = $filter('filter')($rootScope.vaultCiphers, { id: ids[i] }); if (cipher.length) { cipher[0].folderId = folderId; } } selectAll(false); sortScopedCipherData(); toastr.success('Items have been moved!'); }); }; $scope.bulkDelete = function () { var ids = getSelectedCiphers(); if (ids.length === 0) { alert('You have not selected anything.'); return; } if (!confirm('Are you sure you want to delete the selected items (total: ' + ids.length + ')?')) { return; } $scope.bulkActionLoading = true; apiService.ciphers.delMany({ ids: ids }, function () { $analytics.eventTrack('Bulk Deleted Items'); for (var i = 0; i < ids.length; i++) { var cipher = $filter('filter')($rootScope.vaultCiphers, { id: ids[i] }); if (cipher.length && cipher[0].edit) { removeCipherFromScopes(cipher[0]); } } selectAll(false); $scope.bulkActionLoading = false; toastr.success('Items have been deleted!'); }, function () { toastr.error('An error occurred.'); $scope.bulkActionLoading = false; }); }; function removeCipherFromScopes(cipher) { var index = $rootScope.vaultCiphers.indexOf(cipher); if (index > -1) { $rootScope.vaultCiphers.splice(index, 1); } index = $scope.ciphers.indexOf(cipher); if (index > -1) { $scope.ciphers.splice(index, 1); } } });