feat: multi-user isolation and graceful shutdown
- Add chat_id to events, settings, and days_off for per-user data - Add proper graceful shutdown (signal handling, WaitGroup) - Separate polling and webhook modes with goroutine management - Add schema migration from single-user to multi-user schema - Refactor handlers with shared actionFunc pattern (msg + callback) - Fix schema path for Docker deployment (/app/db/schema.sql) - Remove emoji from output text for cleaner formatting
This commit is contained in:
@@ -22,6 +22,31 @@ func NewHandler(bot *tgbotapi.BotAPI, store *db.Store, allowed map[int64]bool) *
|
||||
return &Handler{Bot: bot, DB: store, AllowedUsers: allowed}
|
||||
}
|
||||
|
||||
type actionFunc func(int64) (string, error)
|
||||
|
||||
func (h *Handler) handleActionMsg(msg *tgbotapi.Message, fn actionFunc) {
|
||||
text, err := fn(msg.Chat.ID)
|
||||
if err != nil {
|
||||
h.sendText(msg.Chat.ID, err.Error())
|
||||
} else {
|
||||
h.sendSuccessWithMenu(msg.Chat.ID, text)
|
||||
}
|
||||
}
|
||||
|
||||
func (h *Handler) handleActionCallback(chatID int64, msgID int, callbackID string, fn actionFunc) {
|
||||
defer h.Bot.Request(tgbotapi.NewCallback(callbackID, ""))
|
||||
text, err := fn(chatID)
|
||||
if err != nil {
|
||||
text = err.Error()
|
||||
}
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, text)
|
||||
kb := backKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
}
|
||||
|
||||
// --- Entry points ---
|
||||
|
||||
func (h *Handler) HandleMessage(update tgbotapi.Update) {
|
||||
msg := update.Message
|
||||
if len(h.AllowedUsers) > 0 && !h.AllowedUsers[msg.Chat.ID] {
|
||||
@@ -33,17 +58,17 @@ func (h *Handler) HandleMessage(update tgbotapi.Update) {
|
||||
case "/start":
|
||||
h.handleStart(msg)
|
||||
case "/clockin", "Clock In":
|
||||
h.handleClockInMsg(msg)
|
||||
h.handleActionMsg(msg, h.clockIn)
|
||||
case "/clockout", "Clock Out":
|
||||
h.handleClockOutMsg(msg)
|
||||
h.handleActionMsg(msg, h.clockOut)
|
||||
case "/remote":
|
||||
h.toggleRemote(msg)
|
||||
h.handleActionMsg(msg, h.toggleRemote)
|
||||
case "/report":
|
||||
h.handleReport(msg)
|
||||
h.handleActionMsg(msg, h.report)
|
||||
case "/export":
|
||||
h.handleExport(msg)
|
||||
case "/dayoff":
|
||||
h.handleDayOff(msg)
|
||||
h.handleActionMsg(msg, h.dayOff)
|
||||
default:
|
||||
h.sendText(msg.Chat.ID, "Unknown command")
|
||||
}
|
||||
@@ -61,33 +86,253 @@ func (h *Handler) HandleCallback(update tgbotapi.Update) {
|
||||
msgID := cb.Message.MessageID
|
||||
switch cb.Data {
|
||||
case "clockin":
|
||||
h.clockInCallback(chatID, msgID, cb.ID)
|
||||
h.handleActionCallback(chatID, msgID, cb.ID, h.clockIn)
|
||||
case "clockout":
|
||||
h.clockOutCallback(chatID, msgID, cb.ID)
|
||||
h.handleActionCallback(chatID, msgID, cb.ID, h.clockOut)
|
||||
case "remote":
|
||||
h.toggleRemoteCallback(chatID, msgID, cb.ID)
|
||||
h.handleActionCallback(chatID, msgID, cb.ID, h.toggleRemote)
|
||||
case "report":
|
||||
h.reportCallback(chatID, msgID, cb.ID)
|
||||
h.handleActionCallback(chatID, msgID, cb.ID, h.report)
|
||||
case "export":
|
||||
h.exportCallback(chatID, msgID, cb.ID)
|
||||
case "dayoff":
|
||||
h.dayOffCallback(chatID, msgID, cb.ID)
|
||||
h.handleActionCallback(chatID, msgID, cb.ID, h.dayOff)
|
||||
case "back_menu":
|
||||
h.backToMenu(chatID, msgID, cb.ID)
|
||||
}
|
||||
}
|
||||
|
||||
// --- Actions (shared by msg + callback) ---
|
||||
|
||||
func (h *Handler) clockIn(chatID int64) (string, error) {
|
||||
today := time.Now().Format("2006-01-02")
|
||||
isOff, err := h.DB.IsDayOff(chatID, today)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if isOff {
|
||||
return "", errors.New("Today is a day off. Use /dayoff to remove it first.")
|
||||
}
|
||||
last, err := h.DB.GetLastEvent(chatID)
|
||||
if err != nil && err != sql.ErrNoRows {
|
||||
return "", err
|
||||
}
|
||||
if last != nil && last.EventType == "in" {
|
||||
return "", errors.New("Already clocked in")
|
||||
}
|
||||
if _, err := h.DB.GetSetting(chatID, "remote_flag"); err != nil {
|
||||
h.DB.SetSetting(chatID, "remote_flag", "onsite")
|
||||
}
|
||||
if err := h.DB.CreateEvent(chatID, "in", time.Now().Unix(), ""); err != nil {
|
||||
return "", err
|
||||
}
|
||||
return "Clocked in at " + time.Now().Format("15:04"), nil
|
||||
}
|
||||
|
||||
func (h *Handler) clockOut(chatID int64) (string, error) {
|
||||
today := time.Now().Format("2006-01-02")
|
||||
isOff, err := h.DB.IsDayOff(chatID, today)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if isOff {
|
||||
return "", errors.New("Today is a day off. Use /dayoff to remove it first.")
|
||||
}
|
||||
last, err := h.DB.GetLastEvent(chatID)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if last == nil {
|
||||
return "", errors.New("No clock-in found. Start with /clockin first")
|
||||
}
|
||||
if last.EventType == "out" {
|
||||
return "", errors.New("Already clocked out")
|
||||
}
|
||||
if err := h.DB.CreateEvent(chatID, "out", time.Now().Unix(), ""); err != nil {
|
||||
return "", err
|
||||
}
|
||||
return "Clocked out at " + time.Now().Format("15:04"), nil
|
||||
}
|
||||
|
||||
func (h *Handler) toggleRemote(chatID int64) (string, error) {
|
||||
current, err := h.DB.GetSetting(chatID, "remote_flag")
|
||||
if err != nil {
|
||||
current = "onsite"
|
||||
}
|
||||
var newMode, text string
|
||||
if current == "remote" {
|
||||
newMode = "onsite"
|
||||
text = "Onsite mode enabled"
|
||||
} else {
|
||||
newMode = "remote"
|
||||
text = "Remote mode enabled"
|
||||
}
|
||||
if err := h.DB.SetSetting(chatID, "remote_flag", newMode); err != nil {
|
||||
return "", err
|
||||
}
|
||||
return text, nil
|
||||
}
|
||||
|
||||
func (h *Handler) dayOff(chatID int64) (string, error) {
|
||||
today := time.Now().Format("2006-01-02")
|
||||
isOff, err := h.DB.IsDayOff(chatID, today)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if isOff {
|
||||
if err := h.DB.RemoveDayOff(chatID, today); err != nil {
|
||||
return "", err
|
||||
}
|
||||
return "Day off removed", nil
|
||||
}
|
||||
if err := h.DB.SetDayOff(chatID, today, ""); err != nil {
|
||||
return "", err
|
||||
}
|
||||
return "Today marked as day off", nil
|
||||
}
|
||||
|
||||
func (h *Handler) report(chatID int64) (string, error) {
|
||||
return h.buildDailyReport(chatID, time.Now()), nil
|
||||
}
|
||||
|
||||
// --- /start ---
|
||||
|
||||
func (h *Handler) handleStart(msg *tgbotapi.Message) {
|
||||
h.DB.SetSetting("chat_id", fmt.Sprintf("%d", msg.Chat.ID))
|
||||
h.DB.SetSetting(0, "report_chat_id", fmt.Sprintf("%d", msg.Chat.ID))
|
||||
slog.Info("start", "chat_id", msg.Chat.ID)
|
||||
text := "Welcome! Choose an action:"
|
||||
reply := tgbotapi.NewMessage(msg.Chat.ID, text)
|
||||
reply.ReplyMarkup = mainKeyboard()
|
||||
reply := tgbotapi.NewMessage(msg.Chat.ID, "We hope you have a happy day working.")
|
||||
reply.ReplyMarkup = quickKeyboard()
|
||||
h.Bot.Send(reply)
|
||||
menu := tgbotapi.NewMessage(msg.Chat.ID, "Welcome. Choose an action:")
|
||||
menu.ReplyMarkup = mainKeyboard()
|
||||
h.Bot.Send(menu)
|
||||
}
|
||||
|
||||
// --- Export (different flow — sends a file) ---
|
||||
|
||||
func (h *Handler) handleExport(msg *tgbotapi.Message) {
|
||||
now := time.Now()
|
||||
slog.Info("export", "chat_id", msg.Chat.ID, "year", now.Year(), "month", now.Month())
|
||||
data, err := GenerateMonthlyReport(h.DB, msg.Chat.ID, now.Year(), now.Month())
|
||||
if err != nil {
|
||||
slog.Error("generate report", "error", err)
|
||||
h.sendText(msg.Chat.ID, "Error generating report")
|
||||
return
|
||||
}
|
||||
slog.Info("report generated", "bytes", len(data))
|
||||
doc := tgbotapi.NewDocument(msg.Chat.ID, tgbotapi.FileBytes{
|
||||
Name: fmt.Sprintf("worktime_%s.xlsx", now.Format("2006_01")),
|
||||
Bytes: data,
|
||||
})
|
||||
if _, err := h.Bot.Send(doc); err != nil {
|
||||
slog.Error("send document", "error", err)
|
||||
h.sendText(msg.Chat.ID, "Error sending file")
|
||||
}
|
||||
}
|
||||
|
||||
func (h *Handler) exportCallback(chatID int64, msgID int, callbackID string) {
|
||||
defer h.Bot.Request(tgbotapi.NewCallback(callbackID, ""))
|
||||
now := time.Now()
|
||||
slog.Info("export", "chat_id", chatID, "year", now.Year(), "month", now.Month())
|
||||
data, err := GenerateMonthlyReport(h.DB, chatID, now.Year(), now.Month())
|
||||
if err != nil {
|
||||
slog.Error("generate report", "error", err)
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, "Error generating report")
|
||||
kb := backKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
return
|
||||
}
|
||||
slog.Info("report generated", "bytes", len(data))
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, "Report ready:")
|
||||
kb := backKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
doc := tgbotapi.NewDocument(chatID, tgbotapi.FileBytes{
|
||||
Name: fmt.Sprintf("worktime_%s.xlsx", now.Format("2006_01")),
|
||||
Bytes: data,
|
||||
})
|
||||
if _, err := h.Bot.Send(doc); err != nil {
|
||||
slog.Error("send document", "error", err)
|
||||
}
|
||||
}
|
||||
|
||||
// --- Back to menu ---
|
||||
|
||||
func (h *Handler) backToMenu(chatID int64, msgID int, callbackID string) {
|
||||
defer h.Bot.Request(tgbotapi.NewCallback(callbackID, ""))
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, "Welcome. Choose an action:")
|
||||
kb := mainKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
}
|
||||
|
||||
// --- Daily report builder ---
|
||||
|
||||
func (h *Handler) buildDailyReport(chatID int64, now time.Time) string {
|
||||
dateStr := now.Format("2006-01-02")
|
||||
|
||||
isOff, _ := h.DB.IsDayOff(chatID, dateStr)
|
||||
|
||||
events, err := h.DB.EventsForDay(chatID, dateStr)
|
||||
if err != nil {
|
||||
return "Error fetching events."
|
||||
}
|
||||
|
||||
if len(events) == 0 {
|
||||
if isOff {
|
||||
return fmt.Sprintf("%s — Day Off", dateStr)
|
||||
}
|
||||
return fmt.Sprintf("%s — No events recorded.", dateStr)
|
||||
}
|
||||
|
||||
mode := "Onsite"
|
||||
current, err := h.DB.GetSetting(chatID, "remote_flag")
|
||||
if err == nil && current == "remote" {
|
||||
mode = "Remote"
|
||||
}
|
||||
|
||||
totals := ComputeDailyTotals(events)
|
||||
if totals.PairCount == 0 {
|
||||
t := time.Unix(events[0].OccurredAt, 0).Format("15:04")
|
||||
return fmt.Sprintf("%s\nMode: %s\nClocked in at %s\n\nStill working — no clock-out yet.", dateStr, mode, t)
|
||||
}
|
||||
|
||||
lines := fmt.Sprintf("Report for %s\nMode: %s", dateStr, mode)
|
||||
if isOff {
|
||||
lines += "\nDay Off: Yes"
|
||||
}
|
||||
lines += "\n\n"
|
||||
|
||||
for _, e := range events {
|
||||
t := time.Unix(e.OccurredAt, 0).Format("15:04")
|
||||
label := "IN"
|
||||
action := "Clock In"
|
||||
if e.EventType == "out" {
|
||||
label = "OUT"
|
||||
action = "Clock Out"
|
||||
}
|
||||
lines += fmt.Sprintf("%s %s — %s\n", label, t, action)
|
||||
}
|
||||
|
||||
workStr := formatDuration(totals.TotalSeconds)
|
||||
breakStr := formatDuration(totals.BreakSeconds)
|
||||
lines += fmt.Sprintf("\nSummary:\n Work: %s\n Break: %s\n", workStr, breakStr)
|
||||
|
||||
return lines
|
||||
}
|
||||
|
||||
// --- Scheduled report ---
|
||||
|
||||
func (h *Handler) SendDailyReport(chatID int64) {
|
||||
text := h.buildDailyReport(chatID, time.Now())
|
||||
reply := tgbotapi.NewMessage(chatID, text)
|
||||
reply.ReplyMarkup = quickKeyboard()
|
||||
h.Bot.Send(reply)
|
||||
}
|
||||
|
||||
// --- Keyboards ---
|
||||
|
||||
func quickKeyboard() tgbotapi.ReplyKeyboardMarkup {
|
||||
return tgbotapi.NewReplyKeyboard(
|
||||
tgbotapi.NewKeyboardButtonRow(
|
||||
@@ -122,315 +367,6 @@ func backKeyboard() tgbotapi.InlineKeyboardMarkup {
|
||||
)
|
||||
}
|
||||
|
||||
// --- Text command handlers ---
|
||||
|
||||
func (h *Handler) handleClockInMsg(msg *tgbotapi.Message) {
|
||||
if err := h.doClockIn(); err != nil {
|
||||
slog.Error("clock in failed", "chat_id", msg.Chat.ID, "error", err)
|
||||
h.sendText(msg.Chat.ID, err.Error())
|
||||
} else {
|
||||
slog.Info("clocked in", "chat_id", msg.Chat.ID)
|
||||
h.sendSuccessWithMenu(msg.Chat.ID, "✅ Clocked in at "+time.Now().Format("15:04"))
|
||||
}
|
||||
}
|
||||
|
||||
func (h *Handler) handleClockOutMsg(msg *tgbotapi.Message) {
|
||||
if err := h.doClockOut(); err != nil {
|
||||
slog.Error("clock out failed", "chat_id", msg.Chat.ID, "error", err)
|
||||
h.sendText(msg.Chat.ID, err.Error())
|
||||
} else {
|
||||
slog.Info("clocked out", "chat_id", msg.Chat.ID)
|
||||
h.sendSuccessWithMenu(msg.Chat.ID, "✅ Clocked out at "+time.Now().Format("15:04"))
|
||||
}
|
||||
}
|
||||
|
||||
func (h *Handler) toggleRemote(msg *tgbotapi.Message) {
|
||||
current, err := h.DB.GetSetting("remote_flag")
|
||||
if err != nil {
|
||||
current = "onsite"
|
||||
}
|
||||
newMode := "remote"
|
||||
text := "🌍 Remote mode enabled"
|
||||
if current == "remote" {
|
||||
newMode = "onsite"
|
||||
text = "🏢 Onsite mode enabled"
|
||||
}
|
||||
if err := h.DB.SetSetting("remote_flag", newMode); err != nil {
|
||||
slog.Error("toggle remote", "error", err)
|
||||
h.sendText(msg.Chat.ID, "Error toggling mode")
|
||||
return
|
||||
}
|
||||
slog.Info("remote toggled", "mode", newMode)
|
||||
h.sendText(msg.Chat.ID, text)
|
||||
}
|
||||
|
||||
func (h *Handler) handleReport(msg *tgbotapi.Message) {
|
||||
text := h.buildDailyReport(time.Now())
|
||||
slog.Info("report", "chat_id", msg.Chat.ID)
|
||||
reply := tgbotapi.NewMessage(msg.Chat.ID, text)
|
||||
h.Bot.Send(reply)
|
||||
}
|
||||
|
||||
func (h *Handler) buildDailyReport(now time.Time) string {
|
||||
dateStr := now.Format("2006-01-02")
|
||||
events, err := h.DB.EventsForDay(dateStr)
|
||||
if err != nil {
|
||||
return "Error fetching events."
|
||||
}
|
||||
if len(events) == 0 {
|
||||
isOff, _ := h.DB.IsDayOff(dateStr)
|
||||
if isOff {
|
||||
return fmt.Sprintf("📋 %s — Day Off 🎉", dateStr)
|
||||
}
|
||||
return fmt.Sprintf("📋 %s — No events recorded.", dateStr)
|
||||
}
|
||||
|
||||
remoteFlag := "Onsite"
|
||||
current, err := h.DB.GetSetting("remote_flag")
|
||||
if err == nil && current == "remote" {
|
||||
remoteFlag = "Remote"
|
||||
}
|
||||
|
||||
totals := ComputeDailyTotals(events)
|
||||
if totals.PairCount == 0 {
|
||||
t := time.Unix(events[0].OccurredAt, 0).Format("15:04")
|
||||
return fmt.Sprintf("📋 %s\n📍 Mode: %s\n⏰ Clocked in at %s\n\nStill working — no clock-out yet.", dateStr, remoteFlag, t)
|
||||
}
|
||||
|
||||
lines := fmt.Sprintf("📋 Report for %s\n📍 Mode: %s\n\n", dateStr, remoteFlag)
|
||||
|
||||
for _, e := range events {
|
||||
t := time.Unix(e.OccurredAt, 0).Format("15:04")
|
||||
icon := "⏰"
|
||||
label := "Clock In"
|
||||
if e.EventType == "out" {
|
||||
icon = "🔴"
|
||||
label = "Clock Out"
|
||||
}
|
||||
lines += fmt.Sprintf("%s %s — %s\n", icon, t, label)
|
||||
}
|
||||
|
||||
workStr := formatDuration(totals.TotalSeconds)
|
||||
breakStr := formatDuration(totals.BreakSeconds)
|
||||
lines += fmt.Sprintf("\n📊 Summary:\n Work: %s\n Break: %s\n", workStr, breakStr)
|
||||
|
||||
return lines
|
||||
}
|
||||
|
||||
func (h *Handler) handleExport(msg *tgbotapi.Message) {
|
||||
now := time.Now()
|
||||
year, month := now.Year(), now.Month()
|
||||
slog.Info("export requested", "year", year, "month", month)
|
||||
data, err := GenerateMonthlyReport(h.DB, year, month)
|
||||
if err != nil {
|
||||
slog.Error("generate report", "error", err)
|
||||
h.sendText(msg.Chat.ID, "Error generating report")
|
||||
return
|
||||
}
|
||||
slog.Info("report generated", "bytes", len(data))
|
||||
doc := tgbotapi.NewDocument(msg.Chat.ID, tgbotapi.FileBytes{
|
||||
Name: fmt.Sprintf("worktime_%s.xlsx", now.Format("2006_01")),
|
||||
Bytes: data,
|
||||
})
|
||||
if _, err := h.Bot.Send(doc); err != nil {
|
||||
slog.Error("send document", "error", err)
|
||||
h.sendText(msg.Chat.ID, "Error sending file")
|
||||
}
|
||||
}
|
||||
|
||||
func (h *Handler) handleDayOff(msg *tgbotapi.Message) {
|
||||
today := time.Now().Format("2006-01-02")
|
||||
isOff, err := h.DB.IsDayOff(today)
|
||||
if err != nil {
|
||||
slog.Error("day off check", "error", err)
|
||||
h.sendText(msg.Chat.ID, "Error checking day off")
|
||||
return
|
||||
}
|
||||
if isOff {
|
||||
if err := h.DB.RemoveDayOff(today); err != nil {
|
||||
slog.Error("remove day off", "error", err)
|
||||
h.sendText(msg.Chat.ID, "Error removing day off")
|
||||
return
|
||||
}
|
||||
slog.Info("day off removed")
|
||||
h.sendText(msg.Chat.ID, "✅ Day off removed")
|
||||
} else {
|
||||
if err := h.DB.SetDayOff(today, ""); err != nil {
|
||||
slog.Error("set day off", "error", err)
|
||||
h.sendText(msg.Chat.ID, "Error setting day off")
|
||||
return
|
||||
}
|
||||
slog.Info("day off set")
|
||||
h.sendText(msg.Chat.ID, "✅ Today marked as day off")
|
||||
}
|
||||
}
|
||||
|
||||
// --- Callback handlers ---
|
||||
|
||||
func (h *Handler) clockInCallback(chatID int64, msgID int, callbackID string) {
|
||||
defer h.Bot.Request(tgbotapi.NewCallback(callbackID, ""))
|
||||
text := "✅ Clocked in at " + time.Now().Format("15:04")
|
||||
if err := h.doClockIn(); err != nil {
|
||||
text = err.Error()
|
||||
slog.Error("clock in failed", "chat_id", chatID, "error", err)
|
||||
} else {
|
||||
slog.Info("clocked in", "chat_id", chatID)
|
||||
}
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, text)
|
||||
kb := backKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
}
|
||||
|
||||
func (h *Handler) clockOutCallback(chatID int64, msgID int, callbackID string) {
|
||||
defer h.Bot.Request(tgbotapi.NewCallback(callbackID, ""))
|
||||
text := "✅ Clocked out at " + time.Now().Format("15:04")
|
||||
if err := h.doClockOut(); err != nil {
|
||||
text = err.Error()
|
||||
slog.Error("clock out failed", "chat_id", chatID, "error", err)
|
||||
} else {
|
||||
slog.Info("clocked out", "chat_id", chatID)
|
||||
}
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, text)
|
||||
kb := backKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
}
|
||||
|
||||
func (h *Handler) toggleRemoteCallback(chatID int64, msgID int, callbackID string) {
|
||||
defer h.Bot.Request(tgbotapi.NewCallback(callbackID, ""))
|
||||
current, err := h.DB.GetSetting("remote_flag")
|
||||
if err != nil {
|
||||
current = "onsite"
|
||||
}
|
||||
newMode := "remote"
|
||||
text := "🌍 Remote mode enabled"
|
||||
if current == "remote" {
|
||||
newMode = "onsite"
|
||||
text = "🏢 Onsite mode enabled"
|
||||
}
|
||||
if err := h.DB.SetSetting("remote_flag", newMode); err != nil {
|
||||
slog.Error("toggle remote", "error", err)
|
||||
return
|
||||
}
|
||||
slog.Info("remote toggled", "mode", newMode)
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, text)
|
||||
kb := backKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
}
|
||||
|
||||
func (h *Handler) reportCallback(chatID int64, msgID int, callbackID string) {
|
||||
defer h.Bot.Request(tgbotapi.NewCallback(callbackID, ""))
|
||||
text := h.buildDailyReport(time.Now())
|
||||
slog.Info("report", "chat_id", chatID)
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, text)
|
||||
kb := backKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
}
|
||||
|
||||
func (h *Handler) exportCallback(chatID int64, msgID int, callbackID string) {
|
||||
defer h.Bot.Request(tgbotapi.NewCallback(callbackID, ""))
|
||||
now := time.Now()
|
||||
year, month := now.Year(), now.Month()
|
||||
slog.Info("export", "chat_id", chatID, "year", year, "month", month)
|
||||
data, err := GenerateMonthlyReport(h.DB, year, month)
|
||||
if err != nil {
|
||||
slog.Error("generate report", "error", err)
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, "Error generating report")
|
||||
kb := backKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
return
|
||||
}
|
||||
slog.Info("report generated", "bytes", len(data))
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, "Here is your report:")
|
||||
kb := backKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
doc := tgbotapi.NewDocument(chatID, tgbotapi.FileBytes{
|
||||
Name: fmt.Sprintf("worktime_%s.xlsx", now.Format("2006_01")),
|
||||
Bytes: data,
|
||||
})
|
||||
if _, err := h.Bot.Send(doc); err != nil {
|
||||
slog.Error("send document", "error", err)
|
||||
}
|
||||
}
|
||||
|
||||
func (h *Handler) dayOffCallback(chatID int64, msgID int, callbackID string) {
|
||||
defer h.Bot.Request(tgbotapi.NewCallback(callbackID, ""))
|
||||
today := time.Now().Format("2006-01-02")
|
||||
isOff, err := h.DB.IsDayOff(today)
|
||||
if err != nil {
|
||||
slog.Error("day off check", "error", err)
|
||||
return
|
||||
}
|
||||
text := "✅ Today marked as day off"
|
||||
if isOff {
|
||||
h.DB.RemoveDayOff(today)
|
||||
text = "✅ Day off removed"
|
||||
slog.Info("day off removed")
|
||||
} else {
|
||||
slog.Info("day off set")
|
||||
}
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, text)
|
||||
kb := backKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
}
|
||||
|
||||
func (h *Handler) backToMenu(chatID int64, msgID int, callbackID string) {
|
||||
defer h.Bot.Request(tgbotapi.NewCallback(callbackID, ""))
|
||||
edit := tgbotapi.NewEditMessageText(chatID, msgID, "Welcome! Choose an action:")
|
||||
kb := mainKeyboard()
|
||||
edit.ReplyMarkup = &kb
|
||||
h.Bot.Send(edit)
|
||||
}
|
||||
|
||||
// --- Scheduled report ---
|
||||
|
||||
func (h *Handler) SendDailyReport(chatID int64) {
|
||||
text := h.buildDailyReport(time.Now())
|
||||
reply := tgbotapi.NewMessage(chatID, text)
|
||||
reply.ReplyMarkup = quickKeyboard()
|
||||
h.Bot.Send(reply)
|
||||
}
|
||||
|
||||
// --- Pure logic ---
|
||||
|
||||
func (h *Handler) doClockIn() error {
|
||||
last, err := h.DB.GetLastEvent()
|
||||
if err != nil && err != sql.ErrNoRows {
|
||||
return err
|
||||
}
|
||||
if last != nil && last.EventType == "in" {
|
||||
return errors.New("⚠️ You are already clocked in")
|
||||
}
|
||||
if _, err := h.DB.GetSetting("remote_flag"); err != nil {
|
||||
h.DB.SetSetting("remote_flag", "onsite")
|
||||
}
|
||||
return h.DB.CreateEvent("in", time.Now().Unix(), "")
|
||||
}
|
||||
|
||||
func (h *Handler) doClockOut() error {
|
||||
last, err := h.DB.GetLastEvent()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if last == nil {
|
||||
return errors.New("⚠️ No clock-in found. Start with /clockin first")
|
||||
}
|
||||
switch last.EventType {
|
||||
case "in":
|
||||
return h.DB.CreateEvent("out", time.Now().Unix(), "")
|
||||
case "out":
|
||||
return errors.New("⚠️ You are already clocked out")
|
||||
default:
|
||||
return errors.New("⚠️ Error processing request")
|
||||
}
|
||||
}
|
||||
|
||||
// --- Helpers ---
|
||||
|
||||
func formatDuration(seconds int64) string {
|
||||
|
||||
Reference in New Issue
Block a user