mirror of
https://github.com/ChuckNorrison/LightningTipBot.git
synced 2026-08-17 13:07:16 +02:00
100 lines
3 KiB
Go
100 lines
3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/LightningTipBot/LightningTipBot/internal/lnbits"
|
|
tb "gopkg.in/tucnak/telebot.v2"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func migration() (db *gorm.DB, txLogger *gorm.DB) {
|
|
txLogger, err := gorm.Open(sqlite.Open(Configuration.Database.TransactionsPath), &gorm.Config{DisableForeignKeyConstraintWhenMigrating: true, FullSaveAssociations: true})
|
|
if err != nil {
|
|
panic("Initialize orm failed.")
|
|
}
|
|
|
|
orm, err := gorm.Open(sqlite.Open(Configuration.Database.DbPath), &gorm.Config{DisableForeignKeyConstraintWhenMigrating: true, FullSaveAssociations: true})
|
|
if err != nil {
|
|
panic("Initialize orm failed.")
|
|
}
|
|
|
|
err = orm.AutoMigrate(&lnbits.User{})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
err = txLogger.AutoMigrate(&Transaction{})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return orm, txLogger
|
|
}
|
|
|
|
func GetUserByTelegramUsername(toUserStrWithoutAt string, bot TipBot) (*lnbits.User, error) {
|
|
toUserDb := &lnbits.User{}
|
|
tx := bot.database.Where("telegram_username = ?", strings.ToLower(toUserStrWithoutAt)).First(toUserDb)
|
|
if tx.Error != nil || toUserDb.Wallet == nil {
|
|
err := tx.Error
|
|
if toUserDb.Wallet == nil {
|
|
err = fmt.Errorf("%s | user @%s has no wallet", tx.Error, toUserStrWithoutAt)
|
|
}
|
|
return nil, err
|
|
}
|
|
return toUserDb, nil
|
|
}
|
|
|
|
// GetLnbitsUser will not update the user in database.
|
|
// this is required, because fetching lnbits.User from a incomplete tb.User
|
|
// will update the incomplete (partial) user in storage.
|
|
// this function will accept users like this:
|
|
// &tb.User{ID: toId, Username: username}
|
|
// without updating the user in storage.
|
|
func GetLnbitsUser(u *tb.User, bot TipBot) (*lnbits.User, error) {
|
|
user := &lnbits.User{Name: strconv.Itoa(u.ID)}
|
|
tx := bot.database.First(user)
|
|
if tx.Error != nil {
|
|
errmsg := fmt.Sprintf("[GetUser] Couldn't fetch %s from database: %s", GetUserStr(u), tx.Error.Error())
|
|
log.Warnln(errmsg)
|
|
user.Telegram = u
|
|
return user, tx.Error
|
|
}
|
|
return user, nil
|
|
}
|
|
|
|
// GetUser from telegram user. Update the user if user information changed.
|
|
func GetUser(u *tb.User, bot TipBot) (*lnbits.User, error) {
|
|
user, err := GetLnbitsUser(u, bot)
|
|
if err != nil {
|
|
return user, err
|
|
}
|
|
go func() {
|
|
userCopy := bot.copyLowercaseUser(u)
|
|
if !reflect.DeepEqual(userCopy, user.Telegram) {
|
|
// update possibly changed user details in database
|
|
user.Telegram = userCopy
|
|
err = UpdateUserRecord(user, bot)
|
|
if err != nil {
|
|
log.Warnln(fmt.Sprintf("[UpdateUserRecord] %s", err.Error()))
|
|
}
|
|
}
|
|
}()
|
|
return user, err
|
|
}
|
|
|
|
func UpdateUserRecord(user *lnbits.User, bot TipBot) error {
|
|
user.Telegram = bot.copyLowercaseUser(user.Telegram)
|
|
tx := bot.database.Save(user)
|
|
if tx.Error != nil {
|
|
errmsg := fmt.Sprintf("[UpdateUserRecord] Error: Couldn't update %s's info in database.", GetUserStr(user.Telegram))
|
|
log.Errorln(errmsg)
|
|
return tx.Error
|
|
}
|
|
log.Debugf("[UpdateUserRecord] Records of user %s updated.", GetUserStr(user.Telegram))
|
|
return nil
|
|
}
|