monobank-firefly3-bot/main.go

217 lines
6.7 KiB
Go
Raw Normal View History

2024-03-25 11:55:01 +02:00
package main
import (
"context"
"encoding/json"
"fmt"
"github.com/antihax/optional"
"github.com/joho/godotenv"
"io"
"log"
"main/firefly3"
"main/monobank/api/webhook/models"
2024-03-26 11:26:35 +02:00
"math"
2024-03-25 11:55:01 +02:00
"net/http"
"os"
2024-03-26 12:39:04 +02:00
"slices"
2024-03-25 11:55:01 +02:00
"strconv"
"time"
)
// https://api.monobank.ua/docs/index.html#tag/Kliyentski-personalni-dani/paths/~1personal~1statement~1{account}~1{from}~1{to}/get
// https://api-docs.firefly-iii.org/#/accounts/listAccount
2024-03-26 19:06:31 +02:00
// curl -X POST https://api.monobank.ua/personal/webhook -H 'Content-TransactionType: application/json' -H 'X-Token: ' -d '{"webHookUrl":"https://monobank-firefly3.stuzer.link/webhook"}'
2024-03-25 11:55:01 +02:00
2024-03-26 19:06:31 +02:00
// curl -X POST https://monobank-firefly3.stuzer.link/webhook -H 'Content-TransactionType: application/json' -d '{"test":123}'
2024-03-25 11:55:01 +02:00
2024-03-26 19:06:31 +02:00
var ShopConfig []ShopConfigItem
2024-03-26 11:26:35 +02:00
2024-03-25 11:55:01 +02:00
func handleWebhook(w http.ResponseWriter, r *http.Request) {
2024-03-26 19:06:31 +02:00
LogString("-----------------\nwebhook received!")
2024-03-25 11:55:01 +02:00
// read body bytes
body, err := io.ReadAll(r.Body)
if err != nil {
2024-03-26 19:06:31 +02:00
LogString(err.Error())
2024-03-25 11:55:01 +02:00
w.WriteHeader(http.StatusOK)
return
}
//fmt.Println(string(body))
//w.WriteHeader(http.StatusOK)
//return
2024-03-26 19:06:31 +02:00
body = []byte("{\"type\":\"StatementItem\",\"data\":{\"account\":\"4723djMLsLOCzhoeYjxqRw\",\"statementItem\":{\"id\":\"pPnTnerhs6R3kEw2sQ\",\"time\":1711462716,\"description\":\"Сільпо\",\"mcc\":5411,\"originalMcc\":5411,\"amount\":-60300,\"operationAmount\":-60300,\"currencyCode\":980,\"commissionRate\":0,\"cashbackAmount\":452,\"balance\":9143537,\"hold\":true,\"receiptId\":\"C1H6-8520-8T85-B17M\"}}}")
LogString(string(body))
2024-03-25 11:55:01 +02:00
2024-03-26 12:39:04 +02:00
if len(string(body)) == 0 {
2024-03-26 19:06:31 +02:00
LogString("empty body")
2024-03-26 12:39:04 +02:00
w.WriteHeader(http.StatusOK)
return
}
2024-03-25 11:55:01 +02:00
// parse body
var transaction models.Transaction
err = json.Unmarshal(body, &transaction)
if err != nil {
2024-03-26 19:06:31 +02:00
LogString(err.Error())
2024-03-25 11:55:01 +02:00
w.WriteHeader(http.StatusOK)
return
}
// init firefly3 client
clientConf := firefly3.NewConfiguration()
2024-03-26 11:26:35 +02:00
clientConf.BasePath = os.Getenv("FIREFLY3_API_URL")
2024-03-25 11:55:01 +02:00
clientConf.AddDefaultHeader("Authorization", "Bearer "+os.Getenv("FIREFLY3_TOKEN"))
client := firefly3.NewAPIClient(clientConf)
ctx := context.Background()
2024-03-26 11:26:35 +02:00
// get firefly3 account
2024-03-25 11:55:01 +02:00
listOpts := firefly3.AccountsApiListAccountOpts{
Type_: optional.NewInterface("asset"),
}
accounts, _, err := client.AccountsApi.ListAccount(ctx, &listOpts)
if err != nil {
2024-03-26 19:06:31 +02:00
LogString(err.Error())
2024-03-26 11:26:35 +02:00
w.WriteHeader(http.StatusOK)
return
2024-03-25 11:55:01 +02:00
}
var account firefly3.AccountRead
for _, row := range accounts.Data {
if row.Attributes.Notes == transaction.Data.Account {
account = row
}
}
if len(account.Id) == 0 {
2024-03-26 19:06:31 +02:00
LogString("unable to find account " + transaction.Data.Account + " in firefly3")
2024-03-25 11:55:01 +02:00
w.WriteHeader(http.StatusOK)
return
}
2024-03-26 12:39:04 +02:00
// create transaction
2024-03-26 19:06:31 +02:00
transactionTypeWithdrawal := firefly3.WITHDRAWAL_TransactionTypeProperty
transactionTypeTransfer := firefly3.TRANSFER_TransactionTypeProperty
2024-03-26 12:40:23 +02:00
firefly3Transaction := firefly3.TransactionSplitStore{
2024-03-26 11:26:35 +02:00
Type_: &transactionTypeWithdrawal,
2024-03-26 12:39:04 +02:00
Date: time.Unix(int64(transaction.Data.StatementItem.Time), 0).Add(time.Hour * 2),
2024-03-26 11:26:35 +02:00
Notes: string(body),
2024-03-26 12:39:04 +02:00
Amount: strconv.Itoa(int(math.Abs(math.Round(float64(transaction.Data.StatementItem.Amount / 100))))),
2024-03-26 11:26:35 +02:00
SourceId: account.Id,
}
2024-03-26 12:39:04 +02:00
if slices.Contains([]string{"З чорної картки"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Type_ = &transactionTypeTransfer
firefly3Transaction.Description = "Transfer between accounts"
firefly3Transaction.DestinationId = "60"
2024-03-26 12:39:04 +02:00
} else if slices.Contains([]string{"З білої картки"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Type_ = &transactionTypeTransfer
firefly3Transaction.Description = "Transfer between accounts"
firefly3Transaction.DestinationId = "1"
2024-03-26 19:06:31 +02:00
} else {
for _, row := range ShopConfig {
if slices.Contains(row.Names, transaction.Data.StatementItem.Description) || slices.Contains(row.MCCCodes, transaction.Data.StatementItem.Mcc) {
firefly3Transaction.Description = row.TransactionDescription
break
}
}
2024-03-25 11:55:01 +02:00
}
2024-03-26 12:40:23 +02:00
if len(firefly3Transaction.Description) > 0 {
2024-03-25 11:55:01 +02:00
transactionOpts := firefly3.TransactionsApiStoreTransactionOpts{}
_, _, err = client.TransactionsApi.StoreTransaction(ctx, firefly3.TransactionStore{
ApplyRules: true,
2024-03-26 12:40:23 +02:00
Transactions: []firefly3.TransactionSplitStore{firefly3Transaction},
2024-03-25 11:55:01 +02:00
}, &transactionOpts)
if err != nil {
2024-03-26 19:06:31 +02:00
LogString(err.Error())
2024-03-25 11:55:01 +02:00
w.WriteHeader(http.StatusOK)
return
}
}
w.WriteHeader(http.StatusOK)
}
func main() {
err := godotenv.Load(".env")
if err != nil {
log.Fatalf("Error loading .env file")
}
2024-03-26 19:06:31 +02:00
/**
* Bills
*/
ShopConfig = append(ShopConfig, ShopConfigItem{
Names: []string{"Hetzner"},
TransactionDescription: "Hetzner: vps2",
})
ShopConfig = append(ShopConfig, ShopConfigItem{
Names: []string{"YouTube"},
TransactionDescription: "YouTube membership: Latte ASMR",
})
ShopConfig = append(ShopConfig, ShopConfigItem{
Names: []string{"Київстар +380672463500"},
TransactionDescription: "Kyivstar: +380672463500",
})
ShopConfig = append(ShopConfig, ShopConfigItem{
Names: []string{"Lifecell +380732463500"},
TransactionDescription: "Lifecell: +380732463500",
})
ShopConfig = append(ShopConfig, ShopConfigItem{
Names: []string{"JetBrains"},
TransactionDescription: "JetBrains: GoLand",
})
/**
* Other
*/
ShopConfig = append(ShopConfig, ShopConfigItem{
MCCCodes: []int{5411, 5499, 5451, 5422, 5412, 5921},
Names: []string{"АТБ", "Велмарт", "Novus", "Glovo", "zakaz.ua", "Мегамаркет", "Сільпо"},
TransactionDescription: "Groceries",
})
ShopConfig = append(ShopConfig, ShopConfigItem{
Names: []string{"Аптека Доброго Дня", "Аптека оптових цін", "Аптека Копійка", "Аптека Гала", "Аптека АНЦ", "APTEKA 7", "vidshkod ekv apt12"},
TransactionDescription: "Medications",
})
ShopConfig = append(ShopConfig, ShopConfigItem{
MCCCodes: []int{4131, 4111, 4112},
Names: []string{"Київ Цифровий", "Київпастранс"},
TransactionDescription: "Medications",
})
ShopConfig = append(ShopConfig, ShopConfigItem{
Names: []string{"McDonalds"},
TransactionDescription: "McDonalds",
})
ShopConfig = append(ShopConfig, ShopConfigItem{
Names: []string{"LeoCafe"},
TransactionDescription: "Cafe",
})
ShopConfig = append(ShopConfig, ShopConfigItem{
Names: []string{"Lumberjack Barberhouse"},
TransactionDescription: "Lumberjack: haircut",
})
2024-03-25 11:55:01 +02:00
http.HandleFunc("/webhook", handleWebhook)
2024-03-26 19:06:31 +02:00
2024-03-25 11:55:01 +02:00
fmt.Println("Webhook server listening on :3021")
2024-03-26 19:06:31 +02:00
err = http.ListenAndServe(":3021", nil)
if err != nil {
panic(err.Error())
}
2024-03-25 11:55:01 +02:00
}