From dae017f99e4835b717e07cab452a76bc7b22e3b3 Mon Sep 17 00:00:00 2001
From: Knut Ahlers <knut@ahlers.me>
Date: Thu, 1 Feb 2024 12:44:56 +0100
Subject: [PATCH] Add account reconcilation

Signed-off-by: Knut Ahlers <knut@ahlers.me>
---
 frontend/components/accountOverview.vue | 23 +++++++++++++++++++++++
 pkg/api/account.go                      | 19 +++++++++++++++++++
 pkg/api/api.go                          |  3 +++
 pkg/database/database.go                | 17 +++++++++++++++++
 pkg/database/schema.go                  |  1 +
 5 files changed, 63 insertions(+)

diff --git a/frontend/components/accountOverview.vue b/frontend/components/accountOverview.vue
index e2bcc81..38f74c3 100644
--- a/frontend/components/accountOverview.vue
+++ b/frontend/components/accountOverview.vue
@@ -72,6 +72,14 @@
               Add Transfer
             </button>
 
+            <button
+              class="btn btn-success"
+              @click="markAccountReconciled"
+            >
+              <i class="fas fa-fw fa-square-check mr-1" />
+              Mark Reconciled
+            </button>
+
             <button
               class="btn btn-secondary"
               :disabled="selectedTx.length !== 1"
@@ -80,6 +88,7 @@
               <i class="fas fa-fw fa-pencil mr-1" />
               Edit Transaction
             </button>
+
             <button
               type="button"
               class="btn btn-secondary dropdown-toggle dropdown-toggle-split"
@@ -178,7 +187,14 @@
                     {{ formatNumber(tx.amount) }} €
                   </td>
                   <td>
+                    <span
+                      v-if="tx.reconciled"
+                      class="text-success"
+                    >
+                      <i class="fas fa-lock" />
+                    </span>
                     <a
+                      v-else
                       href="#"
                       :class="{'text-decoration-none':true, 'text-muted': !tx.cleared, 'text-success': tx.cleared}"
                       @click.prevent="markCleared(tx.id, !tx.cleared)"
@@ -482,6 +498,13 @@ export default {
 
     formatNumber,
 
+    markAccountReconciled() {
+      return fetch(`/api/accounts/${this.accountId}/reconcile`, {
+        method: 'PUT',
+      })
+        .then(() => this.fetchTransactions())
+    },
+
     markCleared(txId, cleared) {
       return fetch(`/api/transactions/${txId}?cleared=${cleared}`, {
         method: 'PATCH',
diff --git a/pkg/api/account.go b/pkg/api/account.go
index 9c6d323..f0b26d9 100644
--- a/pkg/api/account.go
+++ b/pkg/api/account.go
@@ -102,6 +102,25 @@ func (a apiServer) handleListAccounts(w http.ResponseWriter, r *http.Request) {
 	a.jsonResponse(w, http.StatusOK, payload)
 }
 
+func (a apiServer) handleAccountReconcile(w http.ResponseWriter, r *http.Request) {
+	var (
+		acctID uuid.UUID
+		err    error
+	)
+
+	if acctID, err = uuid.Parse(mux.Vars(r)["id"]); err != nil {
+		a.errorResponse(w, err, "parsing id", http.StatusBadRequest)
+		return
+	}
+
+	if err = a.dbc.MarkAccountReconciled(acctID); err != nil {
+		a.errorResponse(w, err, "marking reconciled", http.StatusInternalServerError)
+		return
+	}
+
+	w.WriteHeader(http.StatusNoContent)
+}
+
 func (a apiServer) handleTransferMoney(w http.ResponseWriter, r *http.Request) {
 	var (
 		amount             float64
diff --git a/pkg/api/api.go b/pkg/api/api.go
index 2010847..665e0d0 100644
--- a/pkg/api/api.go
+++ b/pkg/api/api.go
@@ -38,6 +38,9 @@ func RegisterHandler(apiRouter *mux.Router, dbc *database.Client, logger *logrus
 	apiRouter.
 		HandleFunc("/accounts/{id}", as.handleUpdateAccount).
 		Methods(http.MethodPatch)
+	apiRouter.
+		HandleFunc("/accounts/{id}/reconcile", as.handleAccountReconcile).
+		Methods(http.MethodPut)
 	apiRouter.
 		HandleFunc("/accounts/{id}/transactions", as.handleListTransactionsByAccount).
 		Methods(http.MethodGet)
diff --git a/pkg/database/database.go b/pkg/database/database.go
index 34d256b..b7506b7 100644
--- a/pkg/database/database.go
+++ b/pkg/database/database.go
@@ -267,6 +267,23 @@ func (c *Client) ListTransactionsByAccount(acc uuid.UUID, since, until time.Time
 	return txs, nil
 }
 
+// MarkAccountReconciled marks all cleared transactions as reconciled.
+// The account balance is NOT checked in this method.
+func (c *Client) MarkAccountReconciled(acc uuid.UUID) (err error) {
+	if err = c.retryTx(func(db *gorm.DB) error {
+		return db.
+			Model(&Transaction{}).
+			Where("account = ?", acc).
+			Where("cleared = ?", true).
+			Update("reconciled", true).
+			Error
+	}); err != nil {
+		return fmt.Errorf("updating transactions: %w", err)
+	}
+
+	return nil
+}
+
 // TransferMoney creates new Transactions for the given account
 // transfer. The account type of the from and to account must match
 // for this to work.
diff --git a/pkg/database/schema.go b/pkg/database/schema.go
index ed46bdb..e2ed7de 100644
--- a/pkg/database/schema.go
+++ b/pkg/database/schema.go
@@ -39,6 +39,7 @@ type (
 		Account     uuid.NullUUID `gorm:"type:uuid" json:"account"`
 		Category    uuid.NullUUID `gorm:"type:uuid" json:"category"`
 		Cleared     bool          `json:"cleared"`
+		Reconciled  bool          `json:"reconciled"`
 
 		PairKey uuid.NullUUID `gorm:"type:uuid" json:"-"`
 	}