From 8d69ea08702f49bb274beaadfdb46642e4bc63ee Mon Sep 17 00:00:00 2001
From: Errietta Erry Kostala <errietta@users.noreply.github.com>
Date: Mon, 10 Feb 2020 14:07:00 +0100
Subject: [PATCH] 6.8.0-pre (#42)

* add the ability to categorise by description, MVP tho

* 6.8.0-pre
---
 package-lock.json                             |  2 +-
 package.json                                  |  2 +-
 .../CategoriseTransactionComponent.tsx        | 64 +++++++++++++++++--
 src/components/Report/ReportComponent.tsx     | 49 ++++++++++----
 src/lib/Util/Util.ts                          |  5 ++
 5 files changed, 101 insertions(+), 21 deletions(-)

diff --git a/package-lock.json b/package-lock.json
index 32a16b2..8081c05 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,6 +1,6 @@
 {
   "name": "hyperbudget-frontend",
-  "version": "6.7.0",
+  "version": "6.8.0-pre",
   "lockfileVersion": 1,
   "requires": true,
   "dependencies": {
diff --git a/package.json b/package.json
index 44e614c..be0becb 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "hyperbudget-frontend",
-  "version": "6.7.0",
+  "version": "6.8.0-pre",
   "description": "",
   "main": "src/index.tsx",
   "scripts": {
diff --git a/src/components/Category/CategoriseTransactionComponent.tsx b/src/components/Category/CategoriseTransactionComponent.tsx
index 7bb0311..77468d4 100644
--- a/src/components/Category/CategoriseTransactionComponent.tsx
+++ b/src/components/Category/CategoriseTransactionComponent.tsx
@@ -4,16 +4,24 @@ import {
   Category, FormattedTransaction
 } from '@hyperbudget/hyperbudget-core';
 
+import { CategorisationType } from '../../lib/Util/Util';
+
 interface CategoriseTransactionState {
-  categoriesForceAdd: Set<string>,
-  categoriesForceRemove: Set<string>,
+  categoriesForceAdd: Set<string>;
+  categoriesForceRemove: Set<string>;
+  categorisationType: CategorisationType;
+  txnDescriptionMatch: string;
 };
 
 interface CategoriseTransactionProps {
   categories: Category[];
   transaction: FormattedTransaction;
   onDoneCategorise: () => void;
-  onSaveCustomCategories: (forceAdd: Set<string>, forceRm: Set<string>) => void;
+  onSaveCustomCategories: (
+    forceAdd: Set<string>, forceRm: Set<string>,
+    categorisationType: CategorisationType,
+    txnDescriptionMatch: string,
+  ) => void;
 };
 
 export class CategoriseTransactionComponent extends React.Component
@@ -53,10 +61,12 @@ export class CategoriseTransactionComponent extends React.Component
     this.state = {
       categoriesForceAdd: new Set(),
       categoriesForceRemove: new Set(),
+      categorisationType: CategorisationType.IDENTIFIER,
+      txnDescriptionMatch: this.props.transaction.description,
     };
   }
 
-  inputChanged(evt): void {
+  selectedCategoriesChanged(evt): void {
     const checkbox = evt.target;
     const catId = checkbox.value;
     console.log(checkbox.checked);
@@ -79,6 +89,24 @@ export class CategoriseTransactionComponent extends React.Component
     }
   }
 
+  selectedTypeChanged(evt): void {
+    const radio = evt.target;
+    const selectedType: CategorisationType = Number(radio.value);
+
+    this.setState({
+      categorisationType: selectedType,
+    })
+  }
+
+  txnDescriptionChanged(evt): void {
+    const text = evt.target;
+    const description = text.value;
+
+    this.setState({
+      txnDescriptionMatch: description,
+    })
+  }
+
   render () {
     return (
       <div className='categorise_transaction'>
@@ -95,7 +123,7 @@ export class CategoriseTransactionComponent extends React.Component
           this.shownCategories.map((category, idx) => (
             <label className='categorise_transaction--label' key={category.name+"-"+idx}>
               <input
-                onChange={this.inputChanged.bind(this)}
+                onChange={this.selectedCategoriesChanged.bind(this)}
                 defaultChecked={this.txnIncludesCat(category)}
                 disabled={this.txnIncludesCat(category) && !this.txnHasCustomCategory(category)}
                 value={category.id}
@@ -103,6 +131,30 @@ export class CategoriseTransactionComponent extends React.Component
             </label>
           ))
         )}
+        <h3>Categorise by</h3>
+        <label className="categorise_transaction--label">
+          <input
+            onChange={this.selectedTypeChanged.bind(this)}
+            value={CategorisationType.IDENTIFIER}
+            checked={this.state.categorisationType === CategorisationType.IDENTIFIER}
+            type="radio" /> This transaction only (transaction identifier)
+        </label>
+         <label className="categorise_transaction--label">
+          <input
+            onChange={this.selectedTypeChanged.bind(this)}
+            value={CategorisationType.DESCRIPTION}
+            checked={this.state.categorisationType === CategorisationType.DESCRIPTION}
+            type="radio" />
+          All transactions matching the following description:
+          <input
+            type="text"
+            value={this.state.txnDescriptionMatch}
+            onChange={this.txnDescriptionChanged.bind(this)}
+            disabled={this.state.categorisationType !== CategorisationType.DESCRIPTION}
+          />
+        </label>
+
+        <hr/>
         <button
             className='btn-primary btn'
             role='button' value="Save"
@@ -110,6 +162,8 @@ export class CategoriseTransactionComponent extends React.Component
               () => this.props.onSaveCustomCategories(
                       this.state.categoriesForceAdd,
                       this.state.categoriesForceRemove,
+                      this.state.categorisationType,
+                      this.state.txnDescriptionMatch
                     )
             }
         >
diff --git a/src/components/Report/ReportComponent.tsx b/src/components/Report/ReportComponent.tsx
index 75bb093..2e04a14 100644
--- a/src/components/Report/ReportComponent.tsx
+++ b/src/components/Report/ReportComponent.tsx
@@ -6,8 +6,8 @@ import 'bootstrap/dist/css/bootstrap.min.css';
 
 import {
   Report, ReportFactory,
-  Category, Categoriser,
-  reportManager, FormattedTransaction, Transaction
+  Category, Categoriser, RuleMatchMode,
+  reportManager, FormattedTransaction, Transaction,
 } from '@hyperbudget/hyperbudget-core';
 
 import { StatementUploaderComponent } from '../StatementUploader/StatementUploaderComponent';
@@ -31,7 +31,11 @@ import { set_transactions, set_categories } from '../../lib/User/User';
 
 import { State } from '../../lib/State/State';
 
-import { deResponsifyPage, responsifyPage, disableScroll, enableScroll } from '../../lib/Util/Util';
+import {
+  deResponsifyPage, responsifyPage,
+  disableScroll, enableScroll,
+  CategorisationType,
+} from '../../lib/Util/Util';
 import { LoadingSpinner } from '../LoadingSpinner';
 
 import queryString from 'query-string';
@@ -285,7 +289,8 @@ class ReportComponent extends React.Component<ReportComponentProps, ReportCompon
 
 
   private onSaveCustomCategories (
-    categoriesForceAdd: Set<string>, categoriesForceRemove: Set<string>
+    categoriesForceAdd: Set<string>, categoriesForceRemove: Set<string>,
+    categorisationType: CategorisationType, txnDescriptionMatch: string
   ): void {
     console.log(categoriesForceAdd, categoriesForceRemove);
 
@@ -304,20 +309,36 @@ class ReportComponent extends React.Component<ReportComponentProps, ReportCompon
       // accidentally, this time it's 100% on purpose. I love js.
       currentCat = this.removeCustomCategory(currentCat, txn);
 
-      if (
-        currentCat.category_rules.identifier &&
-        currentCat.category_rules.identifier.rules
-      ) {
-        currentCat.category_rules.identifier.rules.push(['=', txn.identifier]);
-      } else {
-        currentCat.category_rules.identifier = {
-          "rules": [['=', txn.identifier]]
-        };
+      if (categorisationType === CategorisationType.IDENTIFIER) {
+        if (
+          currentCat.category_rules.identifier &&
+          currentCat.category_rules.identifier.rules
+        ) {
+          currentCat.category_rules.identifier.rules.push(['=', txn.identifier]);
+        } else {
+          currentCat.category_rules.identifier = {
+            "rules": [['=', txn.identifier]]
+          };
+        }
+      } else if (categorisationType === CategorisationType.DESCRIPTION) {
+        if (
+          currentCat.category_rules.description &&
+          currentCat.category_rules.description.rules
+        ) {
+          currentCat.category_rules.description.mode = RuleMatchMode.Flex;
+          currentCat.category_rules.description.rules.push(['=~', txnDescriptionMatch]);
+        } else {
+          currentCat.category_rules.description = {
+            "mode": RuleMatchMode.Flex,
+            "rules": [['=~', txnDescriptionMatch]],
+          };
+        }
       }
     });
 
     // we only support removing ones that were
-    // added manually through this same code at the moment
+    // added manually through this same code
+    // and are identifier matches
     categoriesForceRemove.forEach(id => {
       let currentCat = categoriesHash[id];
       currentCat = this.removeCustomCategory(currentCat, txn);
diff --git a/src/lib/Util/Util.ts b/src/lib/Util/Util.ts
index 051b198..d737e67 100644
--- a/src/lib/Util/Util.ts
+++ b/src/lib/Util/Util.ts
@@ -57,3 +57,8 @@ export const enableScroll = (): void => {
     appContainer.style.overflow = "auto";
   }
 }
+
+export enum CategorisationType {
+  IDENTIFIER,
+  DESCRIPTION,
+};