Anki/yamldb.go

93 lines
1.7 KiB
Go

package main
import (
"time"
"github.com/google/uuid"
)
type (
yamlDB struct {
Knowledge knowledge
Users map[string]user
Cadence []duration
}
knowledge struct {
Questions map[string]Question
Answers map[string]Answer
}
user struct {
History map[string][]History
}
)
func (db yamlDB) HistoryOf(user string) map[string][]History {
return db.Users[user].History
}
func (db yamlDB) Next(user, q string) time.Time {
history := db.Users[user].History[q]
progress := 0
for i := range history {
if history[i].Pass {
progress += 1
} else {
progress -= 1
}
}
if progress < 0 {
progress = 0
} else if progress > len(db.Cadence) {
return time.Now().Add(time.Hour * 24 * 365 * 10)
}
return db.lastTS(user, q).Add(time.Duration(db.Cadence[progress]))
}
func (db yamlDB) Question(q string) Question {
return db.Knowledge.Questions[q]
}
func (db yamlDB) LastAnswer(user, q string) (string, Answer) {
maxk := ""
var maxv Answer
for k, v := range db.Knowledge.Answers {
if v.Q == q && v.Author == user {
if maxv.TS < v.TS {
maxk = k
maxv = v
}
}
}
return maxk, maxv
}
func (db yamlDB) Answer(a string) Answer {
return db.Knowledge.Answers[a]
}
func (db yamlDB) PushAnswer(user, q, a string, pass bool) error {
uuid := uuid.New().String()
db.Knowledge.Answers[uuid] = Answer{
Q: q,
A: a,
TS: time.Now().UnixNano(),
Author: user,
}
db.Users[user].History[q] = append(db.Users[user].History[q], History{
A: uuid,
Pass: pass,
TS: db.Knowledge.Answers[uuid].TS,
})
return nil
}
func (db yamlDB) lastTS(user, q string) time.Time {
max := int64(0)
for _, v := range db.Users[user].History[q] {
if v.TS > max {
max = v.TS
}
}
return time.Unix(0, max)
}