monobank-firefly3-bot/main.go

178 lines
6.6 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
// curl -X POST https://api.monobank.ua/personal/webhook -H 'Content-Type: application/json' -H 'X-Token: ' -d '{"webHookUrl":"https://monobank-firefly3.stuzer.link/webhook"}'
// curl -X POST https://monobank-firefly3.stuzer.link/webhook -H 'Content-Type: application/json' -d '{"test":123}'
2024-03-26 11:26:35 +02:00
func logString(str string) {
if len(os.Getenv("LOG_FILE")) == 0 {
return
}
f, err := os.OpenFile(os.Getenv("LOG_FILE"), os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil {
fmt.Println(err)
}
defer f.Close()
if _, err := f.WriteString(str + "\n"); err != nil {
fmt.Println(err)
}
}
2024-03-25 11:55:01 +02:00
func handleWebhook(w http.ResponseWriter, r *http.Request) {
2024-03-26 11:26:35 +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 11:26:35 +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 12:39:04 +02:00
//body = []byte("{\"type\":\"StatementItem\",\"data\":{\"account\":\"4723djMLsLOCzhoeYjxqRw\",\"statementItem\":{\"id\":\"AQj_9aV8GxUS9opPJQ\",\"time\":1711448198,\"description\":\"Київ Цифровий\",\"mcc\":4111,\"originalMcc\":4111,\"amount\":-800,\"operationAmount\":-800,\"currencyCode\":980,\"commissionRate\":0,\"cashbackAmount\":0,\"balance\":9528749,\"hold\":true,\"receiptId\":\"ABBT-6020-K1H7-A2ME\"}}}")
2024-03-26 11:26:35 +02:00
logString(string(body))
2024-03-25 11:55:01 +02:00
2024-03-26 12:39:04 +02:00
if len(string(body)) == 0 {
logString("empty body")
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 11:26:35 +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
// process transaction
2024-03-25 11:55:01 +02:00
transactionTypeWithdrawal := firefly3.WITHDRAWAL_TransactionTypeProperty
transactionTypeTransfer := firefly3.TRANSFER_TransactionTypeProperty
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 11:26:35 +02:00
logString(err.Error())
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 11:26:35 +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 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 17:29:09 +02:00
} else if slices.Contains([]int{5411, 5499, 5451, 5422, 5412, 5921}, transaction.Data.StatementItem.Mcc) || slices.Contains([]string{"АТБ", "Велмарт", "Novus", "Glovo", "zakaz.ua", "Мегамаркет", "Сільпо"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Description = "Groceries"
2024-03-26 17:29:09 +02:00
} else if slices.Contains([]string{"Аптека Доброго Дня", "Аптека оптових цін", "Аптека Копійка", "Аптека Гала", "Аптека АНЦ", "APTEKA 7", "vidshkod ekv apt12"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Description = "Medications"
2024-03-26 17:29:09 +02:00
} else if slices.Contains([]int{4131, 4111, 4112}, transaction.Data.StatementItem.Mcc) || slices.Contains([]string{"Київ Цифровий", "Київпастранс"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Description = "Public transport"
2024-03-26 17:29:09 +02:00
} else if slices.Contains([]string{"McDonalds"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Description = "McDonalds"
2024-03-26 12:39:04 +02:00
} else if slices.Contains([]string{"LeoCafe"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Description = "Cafe"
2024-03-26 12:39:04 +02:00
} else if slices.Contains([]string{"Lumberjack Barberhouse"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Description = "Lumberjack: haircut"
2024-03-26 12:39:04 +02:00
} else if slices.Contains([]string{"Hetzner"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Description = "Hetzner: vps2"
2024-03-26 12:39:04 +02:00
} else if slices.Contains([]string{"YouTube"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Description = "YouTube membership: Latte ASMR"
2024-03-26 12:39:04 +02:00
} else if slices.Contains([]string{"Київстар +380672463500"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Description = "Kyivstar: +380672463500"
2024-03-26 12:39:04 +02:00
} else if slices.Contains([]string{"Lifecell +380732463500"}, transaction.Data.StatementItem.Description) {
2024-03-26 12:40:23 +02:00
firefly3Transaction.Description = "Lifecell: +380732463500"
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 11:26:35 +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")
}
http.HandleFunc("/webhook", handleWebhook)
fmt.Println("Webhook server listening on :3021")
http.ListenAndServe(":3021", nil)
}