From d86eb8c99f6d2c9fbc055976df90b16fea060b1c Mon Sep 17 00:00:00 2001
From: Can Yumusak <cyumusak@squareup.com>
Date: Thu, 21 Dec 2023 17:57:43 +0100
Subject: [PATCH] Added Notifications

---
 config.go                   |  3 +++
 reader/nordigen/auth.go     | 31 +++++++++++++++++++++++--------
 reader/nordigen/nordigen.go |  6 +++---
 3 files changed, 29 insertions(+), 11 deletions(-)

diff --git a/config.go b/config.go
index 097e40a..70c327d 100644
--- a/config.go
+++ b/config.go
@@ -52,6 +52,9 @@ type Config struct {
 	// PayeeStrip is depreciated please use Nordigen.PayeeStrip instead
 	PayeeStrip []string `envconfig:"YNABBER_PAYEE_STRIP"`
 
+	// SecretID is used to create requisition
+	NotificationScript string `envconfig:"NOTIFICATION_SCRIPT"`
+
 	// Reader and/or writer specific settings
 	Nordigen Nordigen
 	YNAB     YNAB
diff --git a/reader/nordigen/auth.go b/reader/nordigen/auth.go
index 4481519..eea5fd7 100644
--- a/reader/nordigen/auth.go
+++ b/reader/nordigen/auth.go
@@ -4,8 +4,10 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"github.com/martinohansen/ynabber"
 	"log"
 	"os"
+	"os/exec"
 	"path"
 	"strconv"
 	"time"
@@ -26,11 +28,11 @@ func (auth Authorization) Store() string {
 
 // AuthorizationWrapper tries to get requisition from disk, if it fails it will
 // create a new and store that one to disk.
-func (auth Authorization) Wrapper() (nordigen.Requisition, error) {
+func (auth Authorization) Wrapper(cfg ynabber.Config) (nordigen.Requisition, error) {
 	requisitionFile, err := os.ReadFile(auth.Store())
 	if errors.Is(err, os.ErrNotExist) {
 		log.Print("Requisition is not found")
-		return auth.CreateAndSave()
+		return auth.CreateAndSave(cfg)
 	} else if err != nil {
 		return nordigen.Requisition{}, fmt.Errorf("ReadFile: %w", err)
 	}
@@ -39,24 +41,24 @@ func (auth Authorization) Wrapper() (nordigen.Requisition, error) {
 	err = json.Unmarshal(requisitionFile, &requisition)
 	if err != nil {
 		log.Print("Failed to parse requisition file")
-		return auth.CreateAndSave()
+		return auth.CreateAndSave(cfg)
 	}
 
 	switch requisition.Status {
 	case "EX":
 		log.Printf("Requisition is expired")
-		return auth.CreateAndSave()
+		return auth.CreateAndSave(cfg)
 	case "LN":
 		return requisition, nil
 	default:
 		log.Printf("Unsupported requisition status: %s", requisition.Status)
-		return auth.CreateAndSave()
+		return auth.CreateAndSave(cfg)
 	}
 }
 
-func (auth Authorization) CreateAndSave() (nordigen.Requisition, error) {
+func (auth Authorization) CreateAndSave(cfg ynabber.Config) (nordigen.Requisition, error) {
 	log.Print("Creating new requisition...")
-	requisition, err := auth.Create()
+	requisition, err := auth.Create(cfg)
 	if err != nil {
 		return nordigen.Requisition{}, fmt.Errorf("AuthorizationCreate: %w", err)
 	}
@@ -81,7 +83,7 @@ func (auth Authorization) Save(requisition nordigen.Requisition) error {
 	return nil
 }
 
-func (auth Authorization) Create() (nordigen.Requisition, error) {
+func (auth Authorization) Create(cfg ynabber.Config) (nordigen.Requisition, error) {
 	requisition := nordigen.Requisition{
 		Redirect:      "https://raw.githubusercontent.com/martinohansen/ynabber/main/ok.html",
 		Reference:     strconv.Itoa(int(time.Now().Unix())),
@@ -94,6 +96,7 @@ func (auth Authorization) Create() (nordigen.Requisition, error) {
 		return nordigen.Requisition{}, fmt.Errorf("CreateRequisition: %w", err)
 	}
 
+	auth.Notify(cfg, r)
 	log.Printf("Initiate requisition by going to: %s", r.Link)
 
 	// Keep waiting for the user to accept the requisition
@@ -108,3 +111,15 @@ func (auth Authorization) Create() (nordigen.Requisition, error) {
 
 	return r, nil
 }
+
+func (auth Authorization) Notify(cfg ynabber.Config, r nordigen.Requisition) {
+	if cfg.NotificationScript != "" {
+		cmd := exec.Command(cfg.NotificationScript, r.Link)
+		_, err := cmd.Output()
+		if err != nil {
+			log.Println("Could not notify user about new requisition: ", err)
+		}
+	} else {
+		log.Println("No Notification Script set")
+	}
+}
diff --git a/reader/nordigen/nordigen.go b/reader/nordigen/nordigen.go
index 79c07bc..2cc17ac 100644
--- a/reader/nordigen/nordigen.go
+++ b/reader/nordigen/nordigen.go
@@ -92,7 +92,7 @@ func BulkReader(cfg ynabber.Config) (t []ynabber.Transaction, err error) {
 		BankID: cfg.Nordigen.BankID,
 		File:   dataFile(cfg),
 	}
-	r, err := Authorization.Wrapper()
+	r, err := Authorization.Wrapper(cfg)
 	if err != nil {
 		return nil, fmt.Errorf("failed to authorize: %w", err)
 	}
@@ -107,13 +107,13 @@ func BulkReader(cfg ynabber.Config) (t []ynabber.Transaction, err error) {
 		// Handle expired, or suspended accounts by recreating the
 		// requisition.
 		switch accountMetadata.Status {
-		case "EXPIRED", "SUSPENDED":
+		case "EXPIRED", "SUSPENDED ":
 			log.Printf(
 				"Account: %s is %s. Going to recreate the requisition...",
 				account,
 				accountMetadata.Status,
 			)
-			Authorization.CreateAndSave()
+			Authorization.CreateAndSave(cfg)
 		}
 
 		account := ynabber.Account{