From ac7a8eac01922b981aae30c56a9206d44c1111f4 Mon Sep 17 00:00:00 2001
From: snogcel <ledniknoj@gmail.com>
Date: Thu, 15 Dec 2016 22:14:18 -0700
Subject: [PATCH] bugfix and refactor total amount calculation

---
 index.html | 25 +++++++++++++------------
 1 file changed, 13 insertions(+), 12 deletions(-)

diff --git a/index.html b/index.html
index 279752e..6747f18 100755
--- a/index.html
+++ b/index.html
@@ -61,6 +61,15 @@
             return gov;
         };
 
+        var updateTotal = function() {
+            paymentCycle.selectedPeriods = $('#end_epoch').find(':selected').data('index');
+
+            var payment_amount = parseInt($('#payment_amount').val());
+            var periods = parseInt((paymentCycle.selectedPeriods+1)-paymentCycle.selectedStartIndex);
+
+            $('#total_amount').text(payment_amount * periods);
+        };
+
         $(document).ready(function() {
 
             var gov = init('testnet', testnetProvider); // default network;
@@ -71,24 +80,16 @@
             $('#start_epoch').change(function() {
                 paymentCycle.selectedStartIndex = $('#start_epoch').find(':selected').data('index');
                 paymentCycle.updateEndEpoch();
+
+                updateTotal();
             });
 
             $('#end_epoch').change(function() {
-                paymentCycle.selectedPeriods = $('#end_epoch').find(':selected').data('index');
-
-                var payment_amount = parseInt($('#payment_amount').val());
-                var periods = parseInt(paymentCycle.selectedPeriods+1);
-
-                $('#total_amount').text(payment_amount * periods);
+                updateTotal();
             });
 
             $('#payment_amount').change(function() {
-                paymentCycle.selectedPeriods = $('#end_epoch').find(':selected').data('index');
-                
-                var payment_amount = parseInt($('#payment_amount').val());
-                var periods = parseInt(paymentCycle.selectedPeriods+1);
-
-                $('#total_amount').text(payment_amount * periods);
+                updateTotal();
             });
 
             $("#time").val(Math.floor((new Date).getTime() / 1000));