beanbot/bot/txn.go

package bot

import (
	"fmt"
	"sync"
	"time"

	"git.s8k.top/SeraphJACK/beanbot/repo"
	"git.s8k.top/SeraphJACK/beanbot/syntax"
	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)

type transaction struct {
	ctx        *messageHandleContext
	raw        string
	txn        *syntax.Transaction
	confirmMsg tgbotapi.Message
	commitTime time.Time
}

var lock sync.Mutex
var aboutToCommitTxn = make(map[string]*transaction)

func aboutToCommit(id string, txn *transaction) {
	lock.Lock()
	defer lock.Unlock()

	aboutToCommitTxn[id] = txn
}

func cancel(id string) {
	lock.Lock()
	defer lock.Unlock()

	if v, ok := aboutToCommitTxn[id]; ok {
		delete(aboutToCommitTxn, id)
		// delete transaction confirmation message
		go v.ctx.bot.Send(tgbotapi.NewDeleteMessage(v.ctx.chat.ID, v.confirmMsg.MessageID))
	}
}

func commitAll() {
	lock.Lock()
	defer lock.Unlock()

	for k, v := range aboutToCommitTxn {
		// txn not due to commit, skip
		if time.Now().Before(v.commitTime) {
			continue
		}

		updateRecentCommand(v.raw)

		err := repo.CommitTransaction(v.txn)
		if err != nil {
			go v.ctx.bot.Send(tgbotapi.NewMessage(v.ctx.chat.ID, fmt.Sprintf("Failed to commit txn: %v", err)))
		}

		delete(aboutToCommitTxn, k)

		if err == nil {
			// delete transaction message
			go v.ctx.bot.Send(tgbotapi.NewDeleteMessage(v.ctx.chat.ID, v.ctx.msg.MessageID))
		}
		// delete transaction confirmation message
		go v.ctx.bot.Send(tgbotapi.NewDeleteMessage(v.ctx.chat.ID, v.confirmMsg.MessageID))

		go updateRecentKeyboard(v.ctx)
	}
}