78 lines
1.4 KiB
Go
78 lines
1.4 KiB
Go
package bot
|
|
|
|
import (
|
|
"sort"
|
|
|
|
"worktimeBot/internal/db"
|
|
)
|
|
|
|
type State int
|
|
|
|
const (
|
|
StateIdle State = iota
|
|
StateWorking
|
|
StateOnBreak
|
|
)
|
|
|
|
type DailyTotals struct {
|
|
TotalSeconds int64
|
|
BreakSeconds int64
|
|
PairCount int
|
|
}
|
|
|
|
func ComputeDailyTotals(events []db.Event, minBreakThreshold 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
|
|
})
|
|
|
|
var workTotal, breakTotal int64
|
|
var workStart, breakStart int64
|
|
pairs := 0
|
|
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
|
|
pairs++
|
|
breakStart = e.OccurredAt
|
|
state = StateOnBreak
|
|
case StateIdle:
|
|
breakStart = e.OccurredAt
|
|
state = StateOnBreak
|
|
case StateOnBreak:
|
|
breakStart = e.OccurredAt
|
|
}
|
|
}
|
|
}
|
|
|
|
return DailyTotals{
|
|
TotalSeconds: workTotal,
|
|
BreakSeconds: breakTotal,
|
|
PairCount: pairs,
|
|
}
|
|
}
|