mirror of https://github.com/zedeus/nitter
72 lines
1.8 KiB
Nim
72 lines
1.8 KiB
Nim
import asyncdispatch, times, strutils
|
|
import norm/sqlite
|
|
|
|
import types, api/profile
|
|
|
|
template safeAddColumn(field: typedesc): untyped =
|
|
try: field.addColumn
|
|
except DbError: discard
|
|
|
|
dbFromTypes("cache.db", "", "", "", [Profile, Video])
|
|
|
|
withDb:
|
|
try:
|
|
createTables()
|
|
except DbError:
|
|
discard
|
|
Video.title.safeAddColumn
|
|
Video.description.safeAddColumn
|
|
|
|
var profileCacheTime = initDuration(minutes=10)
|
|
|
|
proc isOutdated*(profile: Profile): bool =
|
|
getTime() - profile.updated > profileCacheTime
|
|
|
|
proc cache*(profile: var Profile) =
|
|
withDb:
|
|
try:
|
|
let p = Profile.getOne("lower(username) = ?", toLower(profile.username))
|
|
profile.id = p.id
|
|
profile.update()
|
|
except KeyError:
|
|
if profile.username.len > 0:
|
|
profile.insert()
|
|
|
|
proc hasCachedProfile*(username: string): Option[Profile] =
|
|
withDb:
|
|
try:
|
|
let p = Profile.getOne("lower(username) = ?", toLower(username))
|
|
doAssert not p.isOutdated
|
|
result = some p
|
|
except AssertionError, KeyError:
|
|
result = none Profile
|
|
|
|
proc getCachedProfile*(username, agent: string; force=false): Future[Profile] {.async.} =
|
|
withDb:
|
|
try:
|
|
result.getOne("lower(username) = ?", toLower(username))
|
|
doAssert not result.isOutdated
|
|
except AssertionError, KeyError:
|
|
result = await getProfileFull(username, agent)
|
|
cache(result)
|
|
|
|
proc setProfileCacheTime*(minutes: int) =
|
|
profileCacheTime = initDuration(minutes=minutes)
|
|
|
|
proc cache*(video: var Video) =
|
|
withDb:
|
|
try:
|
|
let v = Video.getOne("videoId = ?", video.videoId)
|
|
video.id = v.id
|
|
video.update()
|
|
except KeyError:
|
|
if video.videoId.len > 0:
|
|
video.insert()
|
|
|
|
proc getCachedVideo*(id: int): Option[Video] =
|
|
withDb:
|
|
try:
|
|
return some Video.getOne("videoId = ?", $id)
|
|
except KeyError:
|
|
return none Video
|