nitter/src/prefs.nim

53 lines
1.2 KiB
Nim

import strutils, sequtils, macros
import prefs_impl, types
export genUpdatePrefs
static:
var pFields: seq[string]
for id in getTypeImpl(Prefs)[2]:
if $id[0] == "id": continue
pFields.add $id[0]
let pDefs = toSeq(allPrefs()).mapIt(it.name)
let missing = pDefs.filterIt(it notin pFields)
if missing.len > 0:
raiseAssert("{$1} missing from the Prefs type" % missing.join(", "))
dbFromTypes("prefs.db", "", "", "", [Prefs])
withDb:
try:
createTables()
except DbError:
discard
proc getDefaultPrefs(hostname: string): Prefs =
result = genDefaultPrefs()
result.replaceTwitter = hostname
proc cache*(prefs: var Prefs) =
withDb:
try:
doAssert prefs.id != 0
discard Prefs.getOne("id = ?", prefs.id)
prefs.update()
except AssertionError, KeyError:
prefs.insert()
proc getPrefs*(id, hostname: string): Prefs =
if id.len == 0:
return getDefaultPrefs(hostname)
withDb:
try:
result.getOne("id = ?", id)
except KeyError:
result = getDefaultPrefs(hostname)
proc resetPrefs*(prefs: var Prefs; hostname: string) =
var defPrefs = getDefaultPrefs(hostname)
defPrefs.id = prefs.id
cache(defPrefs)
prefs = defPrefs