bitwarden-browser/src/app/vault/vaultController.js

angular
    .module('bit.vault')

    .controller('vaultController', function ($scope, $uibModal, apiService, $filter, cryptoService, authService, toastr, cipherService) {
        $scope.sites = [];
        $scope.folders = [];

        $scope.loadingSites = true;
        apiService.sites.list({}, function (sites) {
            $scope.loadingSites = false;

            var decSites = [];
            for (var i = 0; i < sites.Data.length; i++) {
                var decSite = {
                    id: sites.Data[i].Id,
                    folderId: sites.Data[i].FolderId,
                    favorite: sites.Data[i].Favorite
                };

                try { decSite.name = cryptoService.decrypt(sites.Data[i].Name); }
                catch (err) { decSite.name = '[error: cannot decrypt]'; }

                if (sites.Data[i].Username) {
                    try { decSite.username = cryptoService.decrypt(sites.Data[i].Username); }
                    catch (err) { decSite.username = '[error: cannot decrypt]'; }
                }

                decSites.push(decSite);
            }

            $scope.sites = decSites;
        }, function () {
            $scope.loadingSites = false;
        });

        $scope.loadingFolders = true;
        apiService.folders.list({}, function (folders) {
            $scope.loadingFolders = false;

            var decFolders = [{
                id: null,
                name: '(none)'
            }];

            for (var i = 0; i < folders.Data.length; i++) {
                var decFolder = {
                    id: folders.Data[i].Id
                };

                try { decFolder.name = cryptoService.decrypt(folders.Data[i].Name); }
                catch (err) { decFolder.name = '[error: cannot decrypt]'; }

                decFolders.push(decFolder);
            }

            $scope.folders = decFolders;
        }, function () {
            $scope.loadingFolders = false;
        });

        $scope.folderSort = function (item) {
            if (!item.id) {
                return '';
            }

            return item.name.toLowerCase();
        };

        $scope.editSite = function (site) {
            var editModel = $uibModal.open({
                animation: true,
                templateUrl: 'app/vault/views/vaultEditSite.html',
                controller: 'vaultEditSiteController',
                resolve: {
                    siteId: function () { return site.id; },
                    folders: function () { return $scope.folders; }
                }
            });

            editModel.result.then(function (returnVal) {
                if (returnVal.action === 'edit') {
                    var siteToUpdate = $filter('filter')($scope.sites, { id: returnVal.data.id }, true);

                    if (siteToUpdate && siteToUpdate.length > 0) {
                        siteToUpdate[0].folderId = returnVal.data.folderId;
                        siteToUpdate[0].name = returnVal.data.name;
                        siteToUpdate[0].username = returnVal.data.username;
                        siteToUpdate[0].favorite = returnVal.data.favorite;
                    }
                }
                else if (returnVal.action === 'delete') {
                    var siteToDelete = $filter('filter')($scope.sites, { id: returnVal.data }, true);
                    if (siteToDelete && siteToDelete.length > 0) {
                        var index = $scope.sites.indexOf(siteToDelete[0]);
                        if (index > -1) {
                            $scope.sites.splice(index, 1);
                        }
                    }
                }
            });
        };

        $scope.$on('vaultAddSite', function (event, args) {
            $scope.addSite();
        });

        $scope.addSite = function (folder) {
            var addModel = $uibModal.open({
                animation: true,
                templateUrl: 'app/vault/views/vaultAddSite.html',
                controller: 'vaultAddSiteController',
                resolve: {
                    folders: function () { return $scope.folders; },
                    selectedFolder: function () { return folder; }
                }
            });

            addModel.result.then(function (addedSite) {
                $scope.sites.push(addedSite);
            });
        };

        $scope.deleteSite = function (site) {
            if (!confirm('Are you sure you want to delete this site (' + site.name + ')?')) {
                return;
            }

            apiService.sites.del({ id: site.id }, function () {
                var index = $scope.sites.indexOf(site);
                if (index > -1) {
                    $scope.sites.splice(index, 1);
                }
            });
        };

        $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) {
                var folder = $filter('filter')($scope.folders, { id: editedFolder.id }, true);
                if (folder && folder.length > 0) {
                    folder[0].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) {
                $scope.folders.push(addedFolder);
            });
        };

        $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 () {
                var index = $scope.folders.indexOf(folder);
                if (index > -1) {
                    $scope.folders.splice(index, 1);
                }
            });
        };

        $scope.canDeleteFolder = function (folder) {
            if (!folder || !folder.id) {
                return false;
            }

            var sites = $filter('filter')($scope.sites, { folderId: folder.id });
            return sites.length === 0;
        };
    });