From 697747ade1d0280439d6cd9b12be6b304c795a58 Mon Sep 17 00:00:00 2001 From: stuzer05 Date: Wed, 27 Mar 2024 22:28:01 +0200 Subject: [PATCH] Refactor --- http.go | 37 ++++++++++++ main.go | 158 ------------------------------------------------- shop_config.go | 9 --- webhook.go | 134 +++++++++++++++++++++++++++++++++++++++++ 4 files changed, 171 insertions(+), 167 deletions(-) create mode 100644 http.go delete mode 100644 shop_config.go create mode 100644 webhook.go diff --git a/http.go b/http.go new file mode 100644 index 0000000..6892f5f --- /dev/null +++ b/http.go @@ -0,0 +1,37 @@ +package main + +import ( + "encoding/json" + "io" + monobank "main/monobank/api/webhook/models" + "net/http" +) + +func readResponseBody(r *http.Request) (monobank.Transaction, error) { + // read body bytes + body, err := io.ReadAll(r.Body) + if err != nil { + return monobank.Transaction{}, err + } + + //fmt.Println(string(body)) + //w.WriteHeader(http.StatusOK) + //return + + //body = []byte("{\"type\":\"StatementItem\",\"data\":{\"account\":\"4723djMLsLOCzhoeYjxqRw\",\"statementItem\":{\"id\":\"5_NQ0arGAmp2pyNzvA\",\"time\":1711544958,\"description\":\"З чорної картки\",\"mcc\":4829,\"originalMcc\":4829,\"amount\":-572000,\"operationAmount\":-572000,\"currencyCode\":980,\"commissionRate\":22000,\"cashbackAmount\":0,\"balance\":8101246,\"hold\":true,\"receiptId\":\"EMXC-P266-90PC-EB8C\"}}}") + LogString(string(body)) + + // check empty body + if len(string(body)) == 0 { + return monobank.Transaction{}, err + } + + // parse body + var transaction monobank.Transaction + err = json.Unmarshal(body, &transaction) + if err != nil { + return monobank.Transaction{}, err + } + + return transaction, nil +} diff --git a/main.go b/main.go index ffa1b2c..436b337 100644 --- a/main.go +++ b/main.go @@ -1,20 +1,11 @@ package main import ( - "context" - "encoding/json" "fmt" "github.com/joho/godotenv" - "io" "log" - firefly3 "main/firefly3" - monobank "main/monobank/api/webhook/models" - "math" "net/http" "os" - "slices" - "strconv" - "time" ) // https://api.monobank.ua/docs/index.html#tag/Kliyentski-personalni-dani/paths/~1personal~1statement~1{account}~1{from}~1{to}/get @@ -24,155 +15,6 @@ import ( // curl -X POST https://monobank-firefly3.stuzer.link/webhook -H 'Content-TransactionType: application/json' -d '{"test":123}' -func readResponseBody(r *http.Request) (monobank.Transaction, error) { - // read body bytes - body, err := io.ReadAll(r.Body) - if err != nil { - return monobank.Transaction{}, err - } - - //fmt.Println(string(body)) - //w.WriteHeader(http.StatusOK) - //return - - //body = []byte("{\"type\":\"StatementItem\",\"data\":{\"account\":\"4723djMLsLOCzhoeYjxqRw\",\"statementItem\":{\"id\":\"5_NQ0arGAmp2pyNzvA\",\"time\":1711544958,\"description\":\"З чорної картки\",\"mcc\":4829,\"originalMcc\":4829,\"amount\":-572000,\"operationAmount\":-572000,\"currencyCode\":980,\"commissionRate\":22000,\"cashbackAmount\":0,\"balance\":8101246,\"hold\":true,\"receiptId\":\"EMXC-P266-90PC-EB8C\"}}}") - LogString(string(body)) - - // check empty body - if len(string(body)) == 0 { - return monobank.Transaction{}, err - } - - // parse body - var transaction monobank.Transaction - err = json.Unmarshal(body, &transaction) - if err != nil { - return monobank.Transaction{}, err - } - - return transaction, nil -} - -func handleWebhook(w http.ResponseWriter, r *http.Request) { - var err error - - firefly3TransactionTypeWithdrawal := firefly3.WITHDRAWAL_TransactionTypeProperty - firefly3TransactionTypeTransfer := firefly3.TRANSFER_TransactionTypeProperty - - // read request - var monobankTransaction monobank.Transaction - monobankTransaction, err = readResponseBody(r) - if err != nil { - LogString(err.Error()) - w.WriteHeader(http.StatusOK) - return - } - - // get body json string (for logging) - monobankTransactionJson, err := json.Marshal(monobankTransaction) - if err != nil { - LogString(err.Error()) - w.WriteHeader(http.StatusOK) - return - } - - // read config - var config Config - config, err = ReadConfig(os.Getenv("CONFIG_PATH")) - if err != nil { - LogString(err.Error()) - w.WriteHeader(http.StatusOK) - return - } - - // find accounts - account := ConfigGetAccountByMonobankId(config, monobankTransaction.Data.Account) - - // cancel if one of account ids is empty - if len(account.Firefly3Id) == 0 || len(account.MonobankId) == 0 { - LogString("cannot find firefly3 or monobank ids (" + monobankTransaction.Data.Account + ")") - w.WriteHeader(http.StatusOK) - return - } - - // init firefly3 client - clientConf := firefly3.NewConfiguration() - clientConf.BasePath = os.Getenv("FIREFLY3_API_URL") - clientConf.AddDefaultHeader("Authorization", "Bearer "+os.Getenv("FIREFLY3_TOKEN")) - firefly3Client := firefly3.NewAPIClient(clientConf) - - // create firefly3 transaction - var firefly3Transactions []firefly3.TransactionSplitStore - - firefly3Transaction := firefly3.TransactionSplitStore{ - Date: time.Unix(int64(monobankTransaction.Data.StatementItem.Time), 0).Add(time.Hour * 2), - Notes: string(monobankTransactionJson), - Amount: strconv.Itoa(int(math.Abs(math.Round(float64(monobankTransaction.Data.StatementItem.Amount/100)))) - int(math.Abs(math.Round(float64(monobankTransaction.Data.StatementItem.CommissionRate/100))))), - SourceId: account.Firefly3Id, - } - - // Special transaction cases - if slices.Contains([]string{"Термінал EasyPay", "City24", "Термінал City24"}, monobankTransaction.Data.StatementItem.Description) { - firefly3Transaction.Type_ = &firefly3TransactionTypeTransfer - firefly3Transaction.Description = "Transfer between accounts" - firefly3Transaction.SourceName = "Wallet cash (UAH)" // test - firefly3Transaction.DestinationId = account.Firefly3Id // test - firefly3Transactions = append(firefly3Transactions, firefly3Transaction) - } else { - for _, row := range config.TransactionTypes { - if slices.Contains(row.Names, monobankTransaction.Data.StatementItem.Description) || slices.Contains(row.MccCodes, monobankTransaction.Data.StatementItem.Mcc) { - switch row.Firefly3.Type { - case "withdrawal": - firefly3Transaction.Type_ = &firefly3TransactionTypeWithdrawal - break - case "transfer": - firefly3Transaction.Type_ = &firefly3TransactionTypeTransfer - break - default: - firefly3Transaction.Type_ = &firefly3TransactionTypeWithdrawal - } - - firefly3Transaction.Description = row.Firefly3.Description - firefly3Transaction.DestinationName = row.Firefly3.Destination - firefly3Transaction.CategoryName = row.Firefly3.Category - firefly3Transactions = append(firefly3Transactions, firefly3Transaction) - break - } - } - } - - // record transfer fee - if monobankTransaction.Data.StatementItem.CommissionRate > 0 { - firefly3Transactions = append(firefly3Transactions, firefly3.TransactionSplitStore{ - Type_: &firefly3TransactionTypeWithdrawal, - Date: time.Unix(int64(monobankTransaction.Data.StatementItem.Time), 0).Add(time.Hour * 2), - Notes: string(monobankTransactionJson), - Description: "Transfer fee", - Amount: strconv.Itoa(int(math.Abs(math.Round(float64(monobankTransaction.Data.StatementItem.CommissionRate / 100))))), - SourceId: account.Firefly3Id, - }) - } - - // log firefly3 transactions - if len(firefly3Transactions) > 0 { - transactionOpts := firefly3.TransactionsApiStoreTransactionOpts{} - - for _, transaction := range firefly3Transactions { - _, _, err = firefly3Client.TransactionsApi.StoreTransaction(context.Background(), firefly3.TransactionStore{ - ApplyRules: true, - Transactions: []firefly3.TransactionSplitStore{transaction}, - }, &transactionOpts) - if err != nil { - LogString(err.Error()) - w.WriteHeader(http.StatusOK) - return - } - } - } - - w.WriteHeader(http.StatusOK) -} - func main() { // load .env err := godotenv.Load(".env") diff --git a/shop_config.go b/shop_config.go deleted file mode 100644 index d866b96..0000000 --- a/shop_config.go +++ /dev/null @@ -1,9 +0,0 @@ -package main - -type ShopConfigItem struct { - MCCCodes []int - Names []string - TransactionDescription string - TransactionDestination string - TransactionCategory string -} diff --git a/webhook.go b/webhook.go new file mode 100644 index 0000000..e1d8a01 --- /dev/null +++ b/webhook.go @@ -0,0 +1,134 @@ +package main + +import ( + "context" + "encoding/json" + "main/firefly3" + monobank "main/monobank/api/webhook/models" + "math" + "net/http" + "os" + "slices" + "strconv" + "time" +) + +func handleWebhook(w http.ResponseWriter, r *http.Request) { + var err error + + firefly3TransactionTypeWithdrawal := firefly3.WITHDRAWAL_TransactionTypeProperty + firefly3TransactionTypeTransfer := firefly3.TRANSFER_TransactionTypeProperty + + // read request + var monobankTransaction monobank.Transaction + monobankTransaction, err = readResponseBody(r) + if err != nil { + LogString(err.Error()) + w.WriteHeader(http.StatusOK) + return + } + + // get body json string (for logging) + monobankTransactionJson, err := json.Marshal(monobankTransaction) + if err != nil { + LogString(err.Error()) + w.WriteHeader(http.StatusOK) + return + } + + // read config + var config Config + config, err = ReadConfig(os.Getenv("CONFIG_PATH")) + if err != nil { + LogString(err.Error()) + w.WriteHeader(http.StatusOK) + return + } + + // find accounts + account := ConfigGetAccountByMonobankId(config, monobankTransaction.Data.Account) + + // cancel if one of account ids is empty + if len(account.Firefly3Id) == 0 || len(account.MonobankId) == 0 { + LogString("cannot find firefly3 or monobank ids (" + monobankTransaction.Data.Account + ")") + w.WriteHeader(http.StatusOK) + return + } + + // init firefly3 client + clientConf := firefly3.NewConfiguration() + clientConf.BasePath = os.Getenv("FIREFLY3_API_URL") + clientConf.AddDefaultHeader("Authorization", "Bearer "+os.Getenv("FIREFLY3_TOKEN")) + firefly3Client := firefly3.NewAPIClient(clientConf) + + // create firefly3 transaction + var firefly3Transactions []firefly3.TransactionSplitStore + + firefly3Transaction := firefly3.TransactionSplitStore{ + Date: time.Unix(int64(monobankTransaction.Data.StatementItem.Time), 0).Add(time.Hour * 2), + Notes: string(monobankTransactionJson), + Amount: strconv.Itoa(int(math.Abs(math.Round(float64(monobankTransaction.Data.StatementItem.Amount/100)))) - int(math.Abs(math.Round(float64(monobankTransaction.Data.StatementItem.CommissionRate/100))))), + SourceId: account.Firefly3Id, + } + + // Special transaction cases + if slices.Contains([]string{"Термінал EasyPay", "City24", "Термінал City24"}, monobankTransaction.Data.StatementItem.Description) { + firefly3Transaction.Type_ = &firefly3TransactionTypeTransfer + firefly3Transaction.Description = "Transfer between accounts" + firefly3Transaction.SourceName = "Wallet cash (UAH)" // test + firefly3Transaction.DestinationId = account.Firefly3Id // test + firefly3Transactions = append(firefly3Transactions, firefly3Transaction) + } else { + for _, row := range config.TransactionTypes { + if slices.Contains(row.Names, monobankTransaction.Data.StatementItem.Description) || slices.Contains(row.MccCodes, monobankTransaction.Data.StatementItem.Mcc) { + switch row.Firefly3.Type { + case "withdrawal": + firefly3Transaction.Type_ = &firefly3TransactionTypeWithdrawal + break + case "transfer": + firefly3Transaction.Type_ = &firefly3TransactionTypeTransfer + break + default: + firefly3Transaction.Type_ = &firefly3TransactionTypeWithdrawal + } + + firefly3Transaction.Description = row.Firefly3.Description + firefly3Transaction.DestinationName = row.Firefly3.Destination + firefly3Transaction.CategoryName = row.Firefly3.Category + firefly3Transactions = append(firefly3Transactions, firefly3Transaction) + break + } + } + } + + // record transfer fee + if monobankTransaction.Data.StatementItem.CommissionRate > 0 { + firefly3Transactions = append(firefly3Transactions, firefly3.TransactionSplitStore{ + Type_: &firefly3TransactionTypeWithdrawal, + Date: time.Unix(int64(monobankTransaction.Data.StatementItem.Time), 0).Add(time.Hour * 2), + Notes: string(monobankTransactionJson), + Description: "Transfer fee", + Amount: strconv.Itoa(int(math.Abs(math.Round(float64(monobankTransaction.Data.StatementItem.CommissionRate / 100))))), + SourceId: account.Firefly3Id, + }) + } + + // log firefly3 transactions + if len(firefly3Transactions) > 0 { + transactionOpts := firefly3.TransactionsApiStoreTransactionOpts{} + + for _, transaction := range firefly3Transactions { + _, _, err = firefly3Client.TransactionsApi.StoreTransaction(context.Background(), firefly3.TransactionStore{ + ApplyRules: true, + Transactions: []firefly3.TransactionSplitStore{transaction}, + }, &transactionOpts) + if err != nil { + LogString(err.Error()) + w.WriteHeader(http.StatusOK) + return + } + } + } + + w.WriteHeader(http.StatusOK) +}