import times, sequtils, strutils, options
import norm/sqlite

export sqlite, options

db("cache.db", "", "", ""):
  type
    Profile* = object
      username*: string
      fullname*: string
      bio*: string
      userpic*: string
      banner*: string
      following*: string
      followers*: string
      tweets*: string
      verified* {.
          dbType: "STRING",
          parseIt: parseBool(it.s)
          formatIt: $it
        .}: bool
      protected* {.
          dbType: "STRING",
          parseIt: parseBool(it.s)
          formatIt: $it
        .}: bool
      updated* {.
          dbType: "INTEGER",
          parseIt: it.i.fromUnix(),
          formatIt: getTime().toUnix()
        .}: Time

type
  QueryType* = enum
    replies, media, custom = "search"

  Query* = object
    filter*: seq[string]
    exclude*: seq[string]
    sep*: string
    fromUser*: string
    queryType*: QueryType

  VideoType* = enum
    vmap, m3u8, mp4

  Video* = object
    contentId*: string
    playbackType*: VideoType
    durationMs*: int
    url*: string
    thumb*: string
    views*: string
    available*: bool

  Gif* = object
    url*: string
    thumb*: string

  GalleryPhoto* = object
    url*: string
    tweetId*: string
    color*: string

  Poll* = object
    options*: seq[string]
    values*: seq[int]
    votes*: string
    status*: string
    leader*: int

  Quote* = object
    id*: string
    profile*: Profile
    text*: string
    reply*: seq[string]
    hasThread*: bool
    sensitive*: bool
    thumb*: string
    badge*: string

  Retweet* = object
    by*: string
    id*: string

  TweetStats* = object
    replies*: string
    retweets*: string
    likes*: string

  Tweet* = ref object
    id*: string
    profile*: Profile
    text*: string
    time*: Time
    shortTime*: string
    reply*: seq[string]
    pinned*: bool
    available*: bool
    hasThread*: bool
    stats*: TweetStats
    retweet*: Option[Retweet]
    quote*: Option[Quote]
    gif*: Option[Gif]
    video*: Option[Video]
    photos*: seq[string]
    poll*: Option[Poll]

  Thread* = ref object
    tweets*: seq[Tweet]
    more*: int

  Conversation* = ref object
    tweet*: Tweet
    before*: Thread
    after*: Thread
    replies*: seq[Thread]

  Timeline* = ref object
    tweets*: seq[Tweet]
    minId*: string
    maxId*: string
    hasMore*: bool
    query*: Option[Query]

proc contains*(thread: Thread; tweet: Tweet): bool =
  thread.tweets.anyIt(it.id == tweet.id)