db123
a8b849f8fd
- Split 1571-line handlers.go into: handlers.go (core), clock.go, settings.go, calendar.go, report.go - Redesigned export month picker: year navigation + 12-month grid instead of prev/next month - Fixed calendar last row: pad remaining cells with empty buttons to ensure 7 columns - Added Go doc comments across all files (dateutil.go, totals.go, store.go, main.go, webhook.go)
93 lines
2.4 KiB
Go
93 lines
2.4 KiB
Go
// Package bot implements the Telegram bot logic: message routing, time tracking,
|
|
// calendar rendering, Excel export, and inline editing of events.
|
|
package bot
|
|
|
|
import (
|
|
"sort"
|
|
|
|
"worktimeBot/internal/db"
|
|
)
|
|
|
|
// State represents the current tracking state during daily totals computation.
|
|
type State int
|
|
|
|
const (
|
|
// StateIdle indicates no active work session.
|
|
StateIdle State = iota
|
|
// StateWorking indicates an active work session.
|
|
StateWorking
|
|
// StateOnBreak indicates an active break session.
|
|
StateOnBreak
|
|
)
|
|
|
|
// DailyTotals holds the computed total work and break seconds for a single day.
|
|
type DailyTotals struct {
|
|
TotalSeconds int64
|
|
BreakSeconds int64
|
|
}
|
|
|
|
// ComputeDailyTotals calculates the total work and break time from a list of events.
|
|
// It sorts events chronologically, inserts synthetic start/end events when needed,
|
|
// and tracks state transitions between idle, working, and on-break.
|
|
func ComputeDailyTotals(events []db.Event, minBreakThreshold int64, dayStart, dayEnd int64) DailyTotals {
|
|
if len(events) == 0 {
|
|
return DailyTotals{}
|
|
}
|
|
sorted := make([]db.Event, len(events))
|
|
copy(sorted, events)
|
|
sort.Slice(sorted, func(i, j int) bool {
|
|
return sorted[i].OccurredAt < sorted[j].OccurredAt
|
|
})
|
|
|
|
if dayStart > 0 && len(sorted) > 0 && sorted[0].EventType == "out" {
|
|
first := db.Event{EventType: "in", OccurredAt: dayStart}
|
|
sorted = append([]db.Event{first}, sorted...)
|
|
}
|
|
if dayEnd > 0 && len(sorted) > 0 && sorted[len(sorted)-1].EventType == "in" {
|
|
last := db.Event{EventType: "out", OccurredAt: dayEnd}
|
|
sorted = append(sorted, last)
|
|
}
|
|
|
|
var workTotal, breakTotal int64
|
|
var workStart, breakStart int64
|
|
state := StateIdle
|
|
|
|
for _, e := range sorted {
|
|
switch e.EventType {
|
|
case "in":
|
|
switch state {
|
|
case StateIdle:
|
|
workStart = e.OccurredAt
|
|
state = StateWorking
|
|
case StateOnBreak:
|
|
breakDuration := e.OccurredAt - breakStart
|
|
if minBreakThreshold > 0 && breakDuration <= minBreakThreshold {
|
|
workStart = breakStart
|
|
} else {
|
|
breakTotal += breakDuration
|
|
workStart = e.OccurredAt
|
|
}
|
|
state = StateWorking
|
|
}
|
|
case "out":
|
|
switch state {
|
|
case StateWorking:
|
|
workTotal += e.OccurredAt - workStart
|
|
workStart = 0
|
|
breakStart = e.OccurredAt
|
|
state = StateOnBreak
|
|
case StateIdle:
|
|
breakStart = e.OccurredAt
|
|
state = StateOnBreak
|
|
case StateOnBreak:
|
|
breakStart = e.OccurredAt
|
|
}
|
|
}
|
|
}
|
|
|
|
return DailyTotals{
|
|
TotalSeconds: workTotal,
|
|
BreakSeconds: breakTotal,
|
|
}
|
|
}
|