You've already forked monobank-firefly3-bot
Refactor
This commit is contained in:
46
app/app.go
Normal file
46
app/app.go
Normal file
@ -0,0 +1,46 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"gitea.stuzer.link/stuzer05/go-firefly3/v2"
|
||||
"gitea.stuzer.link/stuzer05/go-monobank"
|
||||
"os"
|
||||
"stuzer.link/monobank-firefly3-bot/config"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
init sync.Once
|
||||
Config config.Config
|
||||
MonobankClient *monobank.APIClient
|
||||
Firefly3Client *firefly3.APIClient
|
||||
}
|
||||
|
||||
var app Config
|
||||
|
||||
func App() *Config {
|
||||
return &app
|
||||
}
|
||||
|
||||
func Init() {
|
||||
app.init.Do(func() {
|
||||
var err error
|
||||
|
||||
// read config
|
||||
app.Config, err = config.Read(os.Getenv("CONFIG_PATH"))
|
||||
if err != nil {
|
||||
fmt.Println("cannot read config: " + err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// init monobank client
|
||||
monobankClientConf := monobank.NewConfiguration()
|
||||
app.MonobankClient = monobank.NewAPIClient(monobankClientConf)
|
||||
|
||||
// init firefly3 client
|
||||
clientConf := firefly3.NewConfiguration()
|
||||
clientConf.BasePath = os.Getenv("FIREFLY3_API_URL")
|
||||
clientConf.AddDefaultHeader("Authorization", "Bearer "+os.Getenv("FIREFLY3_TOKEN"))
|
||||
app.Firefly3Client = firefly3.NewAPIClient(clientConf)
|
||||
})
|
||||
}
|
||||
162
app/import_transaction.go
Normal file
162
app/import_transaction.go
Normal file
@ -0,0 +1,162 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"gitea.stuzer.link/stuzer05/go-firefly3/v2"
|
||||
"gitea.stuzer.link/stuzer05/go-monobank"
|
||||
"github.com/antihax/optional"
|
||||
"math"
|
||||
"slices"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
func ImportTransaction(monobankTransaction monobank.WebHookResponse) error {
|
||||
firefly3TransactionTypeWithdrawal := firefly3.WITHDRAWAL_TransactionTypeProperty
|
||||
firefly3TransactionTypeDeposit := firefly3.DEPOSIT_TransactionTypeProperty
|
||||
firefly3TransactionTypeTransfer := firefly3.TRANSFER_TransactionTypeProperty
|
||||
|
||||
// get body json string (for logging)
|
||||
monobankTransactionJson, err := json.Marshal(monobankTransaction)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// find accounts
|
||||
account := App().Config.GetAccountByMonobankId(monobankTransaction.Data.Account)
|
||||
|
||||
// cancel if one of account ids is empty
|
||||
if len(account.Firefly3Name) == 0 || len(account.MonobankId) == 0 {
|
||||
return errors.New("cannot find firefly3 or monobank ids (" + monobankTransaction.Data.Account + ")")
|
||||
}
|
||||
|
||||
// create firefly3 transactions list
|
||||
var firefly3Transactions []firefly3.TransactionSplitStore
|
||||
|
||||
// match transaction with config
|
||||
for _, row := range App().Config.TransactionTypes {
|
||||
|
||||
// is refund
|
||||
if slices.Contains(row.NamesRefund, *monobankTransaction.Data.StatementItem.Description) {
|
||||
opts := firefly3.TransactionsApiListTransactionOpts{
|
||||
Limit: optional.NewInt32(999),
|
||||
Type_: optional.NewInterface("withdrawal"),
|
||||
Start: optional.NewString(time.Now().AddDate(0, 0, -7).Format("2006-01-02")), // one week before
|
||||
}
|
||||
oldTransactions, _, err := App().Firefly3Client.TransactionsApi.ListTransaction(context.Background(), &opts)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// find matching transaction to delete
|
||||
isDeleted := false
|
||||
for _, tRows := range oldTransactions.Data {
|
||||
if isDeleted {
|
||||
break
|
||||
}
|
||||
|
||||
for _, tRow := range tRows.Attributes.Transactions {
|
||||
// validate notes is json
|
||||
notesBytes := bytes.NewBufferString(tRow.Notes).Bytes()
|
||||
if !json.Valid(notesBytes) {
|
||||
continue
|
||||
}
|
||||
|
||||
// read monobank transaction
|
||||
var monobankTransaction monobank.StatementItemsInner
|
||||
err = json.Unmarshal(notesBytes, &monobankTransaction)
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
// find transaction
|
||||
sum := int(math.Abs(math.Round(float64(*monobankTransaction.Amount/100)))) - int(math.Abs(math.Round(float64(*monobankTransaction.CommissionRate/100))))
|
||||
sum2, _ := strconv.ParseFloat(tRow.Amount, 64)
|
||||
if slices.Contains(row.Names, *monobankTransaction.Description) && sum == int(sum2) {
|
||||
// delete transaction
|
||||
opts := firefly3.TransactionsApiDeleteTransactionOpts{}
|
||||
App().Firefly3Client.TransactionsApi.DeleteTransaction(context.Background(), tRows.Id, &opts)
|
||||
|
||||
isDeleted = true
|
||||
}
|
||||
}
|
||||
}
|
||||
break
|
||||
} else {
|
||||
// check name & mcc
|
||||
if !(slices.Contains(row.Names, *monobankTransaction.Data.StatementItem.Description) || slices.Contains(row.MccCodes, int(*monobankTransaction.Data.StatementItem.Mcc))) {
|
||||
continue
|
||||
}
|
||||
|
||||
// create firefly3 transaction
|
||||
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))))),
|
||||
SourceName: account.Firefly3Name,
|
||||
}
|
||||
|
||||
// check max sum
|
||||
sum, _ := strconv.Atoi(firefly3Transaction.Amount)
|
||||
if row.SumMax > 0 && sum > row.SumMax {
|
||||
continue
|
||||
}
|
||||
// make transaction
|
||||
switch row.Firefly3.Type {
|
||||
case "withdrawal":
|
||||
firefly3Transaction.Type_ = &firefly3TransactionTypeWithdrawal
|
||||
break
|
||||
case "deposit":
|
||||
firefly3Transaction.Type_ = &firefly3TransactionTypeDeposit
|
||||
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
|
||||
|
||||
// swap source and destination
|
||||
if row.Firefly3.IsUseDestinationAsSource {
|
||||
firefly3Transaction.SourceName, firefly3Transaction.DestinationName = firefly3Transaction.DestinationName, firefly3Transaction.SourceName
|
||||
}
|
||||
|
||||
firefly3Transactions = append(firefly3Transactions, firefly3Transaction)
|
||||
break
|
||||
}
|
||||
}
|
||||
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))))),
|
||||
SourceName: account.Firefly3Name,
|
||||
})
|
||||
}
|
||||
|
||||
// log firefly3 transactions
|
||||
if len(firefly3Transactions) > 0 {
|
||||
transactionOpts := firefly3.TransactionsApiStoreTransactionOpts{}
|
||||
|
||||
for _, transaction := range firefly3Transactions {
|
||||
_, _, err = App().Firefly3Client.TransactionsApi.StoreTransaction(context.Background(), firefly3.TransactionStore{
|
||||
ApplyRules: true,
|
||||
Transactions: []firefly3.TransactionSplitStore{transaction},
|
||||
}, &transactionOpts)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
22
app/logger.go
Normal file
22
app/logger.go
Normal file
@ -0,0 +1,22 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
)
|
||||
|
||||
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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user