From 1dbf831bda372d4362b3fc524de1f39cb350af7a Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Tue, 11 Jul 2017 10:59:49 -0400
Subject: [PATCH] storage adjustment

---
 ...anizationBillingAdjustStorageController.js | 37 +++++++++++++++
 .../organizationBillingController.js          | 17 +++++++
 src/app/services/apiService.js                |  2 +
 .../settingsBillingAdjustStorageController.js | 37 +++++++++++++++
 .../views/settingsBillingAdjustStorage.html   | 46 +++++++++++++++++++
 src/index.html                                |  2 +
 6 files changed, 141 insertions(+)
 create mode 100644 src/app/organization/organizationBillingAdjustStorageController.js
 create mode 100644 src/app/settings/settingsBillingAdjustStorageController.js
 create mode 100644 src/app/settings/views/settingsBillingAdjustStorage.html

diff --git a/src/app/organization/organizationBillingAdjustStorageController.js b/src/app/organization/organizationBillingAdjustStorageController.js
new file mode 100644
index 0000000000..7363da1b84
--- /dev/null
+++ b/src/app/organization/organizationBillingAdjustStorageController.js
@@ -0,0 +1,37 @@
+angular
+    .module('bit.organization')
+
+    .controller('organizationBillingAdjustStorageController', function ($scope, $state, $uibModalInstance, apiService,
+        $analytics, toastr, add) {
+        $analytics.eventTrack('organizationBillingAdjustStorageController', { category: 'Modal' });
+        $scope.add = add;
+        $scope.storageAdjustment = 0;
+
+        $scope.submit = function () {
+            var request = {
+                storageGbAdjustment: $scope.storageAdjustment
+            };
+
+            if (!add) {
+                request.storageGbAdjustment *= -1;
+            }
+
+            $scope.submitPromise = apiService.organizations.putStorage({ id: $state.params.orgId }, request)
+                .$promise.then(function (response) {
+                    if (add) {
+                        $analytics.eventTrack('Added Organization Storage');
+                        toastr.success('You have added ' + $scope.storageAdjustment + ' GB.');
+                    }
+                    else {
+                        $analytics.eventTrack('Removed Organization Storage');
+                        toastr.success('You have removed ' + $scope.storageAdjustment + ' GB.');
+                    }
+
+                    $uibModalInstance.close();
+                });
+        };
+
+        $scope.close = function () {
+            $uibModalInstance.dismiss('cancel');
+        };
+    });
diff --git a/src/app/organization/organizationBillingController.js b/src/app/organization/organizationBillingController.js
index 796e90195a..c45e1e4d86 100644
--- a/src/app/organization/organizationBillingController.js
+++ b/src/app/organization/organizationBillingController.js
@@ -63,6 +63,23 @@
             });
         };
 
+        $scope.adjustStorage = function (add) {
+            var modal = $uibModal.open({
+                animation: true,
+                templateUrl: 'app/settings/views/settingsBillingAdjustStorage.html',
+                controller: 'organizationBillingAdjustStorageController',
+                resolve: {
+                    add: function () {
+                        return add;
+                    }
+                }
+            });
+
+            modal.result.then(function () {
+                load();
+            });
+        };
+
         $scope.cancel = function () {
             if (!confirm('Are you sure you want to cancel? All users will lose access to the organization ' +
                 'at the end of this billing cycle.')) {
diff --git a/src/app/services/apiService.js b/src/app/services/apiService.js
index 1f8cde2ca8..1127a1988f 100644
--- a/src/app/services/apiService.js
+++ b/src/app/services/apiService.js
@@ -64,6 +64,7 @@
             put: { method: 'POST', params: { id: '@id' } },
             putPayment: { url: _apiUri + '/organizations/:id/payment', method: 'POST', params: { id: '@id' } },
             putSeat: { url: _apiUri + '/organizations/:id/seat', method: 'POST', params: { id: '@id' } },
+            putStorage: { url: _apiUri + '/organizations/:id/storage', method: 'POST', params: { id: '@id' } },
             putUpgrade: { url: _apiUri + '/organizations/:id/upgrade', method: 'POST', params: { id: '@id' } },
             putCancel: { url: _apiUri + '/organizations/:id/cancel', method: 'POST', params: { id: '@id' } },
             putReinstate: { url: _apiUri + '/organizations/:id/reinstate', method: 'POST', params: { id: '@id' } },
@@ -125,6 +126,7 @@
             'import': { url: _apiUri + '/accounts/import', method: 'POST', params: {} },
             postDelete: { url: _apiUri + '/accounts/delete', method: 'POST', params: {} },
             postPremium: { url: _apiUri + '/accounts/premium', method: 'POST', params: {} },
+            putStorage: { url: _apiUri + '/accounts/storage', method: 'POST', params: {} },
             putCancelPremium: { url: _apiUri + '/accounts/cancel-premium', method: 'POST', params: {} },
             putReinstatePremium: { url: _apiUri + '/accounts/reinstate-premium', method: 'POST', params: {} },
             getBilling: { url: _apiUri + '/accounts/billing', method: 'GET', params: {} }
diff --git a/src/app/settings/settingsBillingAdjustStorageController.js b/src/app/settings/settingsBillingAdjustStorageController.js
new file mode 100644
index 0000000000..ceb1709c50
--- /dev/null
+++ b/src/app/settings/settingsBillingAdjustStorageController.js
@@ -0,0 +1,37 @@
+angular
+    .module('bit.settings')
+
+    .controller('settingsBillingAdjustStorageController', function ($scope, $state, $uibModalInstance, apiService,
+        $analytics, toastr, add) {
+        $analytics.eventTrack('settingsBillingAdjustStorageController', { category: 'Modal' });
+        $scope.add = add;
+        $scope.storageAdjustment = 0;
+
+        $scope.submit = function () {
+            var request = {
+                storageGbAdjustment: $scope.storageAdjustment
+            };
+
+            if (!add) {
+                request.storageGbAdjustment *= -1;
+            }
+
+            $scope.submitPromise = apiService.accounts.putStorage(null, request)
+                .$promise.then(function (response) {
+                    if (add) {
+                        $analytics.eventTrack('Added Storage');
+                        toastr.success('You have added ' + $scope.storageAdjustment + ' GB.');
+                    }
+                    else {
+                        $analytics.eventTrack('Removed Storage');
+                        toastr.success('You have removed ' + $scope.storageAdjustment + ' GB.');
+                    }
+
+                    $uibModalInstance.close();
+                });
+        };
+
+        $scope.close = function () {
+            $uibModalInstance.dismiss('cancel');
+        };
+    });
diff --git a/src/app/settings/views/settingsBillingAdjustStorage.html b/src/app/settings/views/settingsBillingAdjustStorage.html
new file mode 100644
index 0000000000..7d1db06c11
--- /dev/null
+++ b/src/app/settings/views/settingsBillingAdjustStorage.html
@@ -0,0 +1,46 @@
+<div class="modal-header">
+    <button type="button" class="close" ng-click="close()" aria-label="Close"><span aria-hidden="true">&times;</span></button>
+    <h4 class="modal-title">
+        <i class="fa fa-database"></i>
+        {{add ? 'Add Storage' : 'Remove Storage'}}
+    </h4>
+</div>
+<form name="form" ng-submit="form.$valid && submit()" api-form="submitPromise">
+    <div class="modal-body">
+        <div class="callout callout-default" ng-show="add">
+            <h4><i class="fa fa-dollar"></i> Note About Charges</h4>
+            <p>
+                Adding storage to your plan will result in adjustments to your billing totals and immediately charge your
+                payment method on file. The first charge will be prorated for the remainder of the current billing cycle.
+            </p>
+        </div>
+        <div class="callout callout-default" ng-show="!add">
+            <h4><i class="fa fa-dollar"></i> Note About Charges</h4>
+            <p>
+                Removing storage will result in adjustments to your billing totals that will be prorated as credits
+                to your next billing charge.
+            </p>
+        </div>
+        <div class="callout callout-danger validation-errors" ng-show="form.$errors">
+            <h4>Errors have occurred</h4>
+            <ul>
+                <li ng-repeat="e in form.$errors">{{e}}</li>
+            </ul>
+        </div>
+        <div class="row">
+            <div class="col-md-6">
+                <div class="form-group">
+                    <label for="gb">{{add ? 'GB of Storage To Add' : 'GB of Storage To Remove'}}</label>
+                    <input type="number" id="gb" name="StroageGbAdjustment" ng-model="storageAdjustment" class="form-control"
+                           required min="0" max="99" />
+                </div>
+            </div>
+        </div>
+    </div>
+    <div class="modal-footer">
+        <button type="submit" class="btn btn-primary btn-flat" ng-disabled="form.$loading">
+            <i class="fa fa-refresh fa-spin loading-icon" ng-show="form.$loading"></i>Submit
+        </button>
+        <button type="button" class="btn btn-default btn-flat" ng-click="close()">Close</button>
+    </div>
+</form>
diff --git a/src/index.html b/src/index.html
index 6a63d13269..b2aa36835b 100644
--- a/src/index.html
+++ b/src/index.html
@@ -185,6 +185,7 @@
     <script src="app/organization/organizationBillingController.js"></script>
     <script src="app/organization/organizationBillingChangePaymentController.js"></script>
     <script src="app/organization/organizationBillingAdjustSeatsController.js"></script>
+    <script src="app/organization/organizationBillingAdjustStorageController.js"></script>
     <script src="app/organization/organizationDeleteController.js"></script>
     <script src="app/organization/organizationBillingChangePlanController.js"></script>
     <script src="app/organization/organizationVaultController.js"></script>
@@ -214,6 +215,7 @@
     <script src="app/settings/settingsDeleteController.js"></script>
     <script src="app/settings/settingsCreateOrganizationController.js"></script>
     <script src="app/settings/settingsBillingController.js"></script>
+    <script src="app/settings/settingsBillingAdjustStorageController.js"></script>
     <script src="app/settings/settingsUpdateKeyController.js"></script>
     <script src="app/settings/settingsPremiumController.js"></script>