import xmltree, sequtils, strtabs, strutils, strformat, json, times
import nimquery, regex

import ./types, ./formatters

proc getAttr(node: XmlNode; attr: string; default=""): string =
  if node.isNil or node.attrs.isNil: return default
  return node.attrs.getOrDefault(attr)

proc selectAttr(node: XmlNode; selector: string; attr: string; default=""): string =
  let res = node.querySelector(selector)
  if res == nil: "" else: res.getAttr(attr, default)

proc selectText(node: XmlNode; selector: string): string =
  let res = node.querySelector(selector)
  result = if res == nil: "" else: res.innerText()

proc parsePopupProfile*(node: XmlNode): Profile =
  let profile = node.querySelector(".profile-card")
  if profile.isNil: return

  result = Profile(
     fullname: profile.selectText(".fullname").strip(),
     username: profile.selectText(".username").strip(chars={'@', ' '}),
     description: profile.selectText(".bio"),
     verified: profile.selectText(".Icon.Icon--verified").len > 0,
     protected: profile.selectText(".Icon.Icon--protected").len > 0,
     userpic: profile.selectAttr(".ProfileCard-avatarImage", "src").getUserpic(),
     banner: profile.selectAttr("svg > image", "xlink:href").replace("600x200", "1500x500")
  )

  if result.banner.len == 0:
      result.banner = profile.selectAttr(".ProfileCard-bg", "style")

  let stats = profile.querySelectorAll(".ProfileCardStats-statLink")
  for s in stats:
    let text = s.getAttr("title").split(" ")[0]
    case s.getAttr("href").split("/")[^1]
    of "followers": result.followers = text
    of "following": result.following = text
    else: result.tweets = text

proc parseIntentProfile*(profile: XmlNode): Profile =
  result = Profile(
    fullname: profile.selectText("a.fn.url.alternate-context").strip(),
    username: profile.selectText(".nickname").strip(chars={'@', ' '}),
    userpic: profile.querySelector(".profile.summary").selectAttr("img.photo", "src").getUserPic(),
    description: profile.selectText("p.note").strip(),
    verified: not profile.querySelector("li.verified").isNil,
    protected: not profile.querySelector("li.protected").isNil,
    banner: "background-color: #161616",
    tweets: "?"
  )

  for stat in profile.querySelectorAll("dd.count > a"):
    case stat.getAttr("href").split("/")[^1]
    of "followers": result.followers = stat.innerText()
    of "following": result.following = stat.innerText()

proc parseTweetProfile*(profile: XmlNode): Profile =
  result = Profile(
    fullname: profile.getAttr("data-name"),
    username: profile.getAttr("data-screen-name"),
    userpic: profile.selectAttr(".avatar", "src").getUserpic(),
    verified: profile.selectText(".Icon.Icon--verified").len > 0
  )

proc parseTweet*(tweet: XmlNode): Tweet =
  result.id = tweet.getAttr("data-item-id")
  result.link = tweet.getAttr("data-permalink-path")
  result.text = tweet.selectText(".tweet-text").stripTwitterUrls()
  result.pinned = "pinned" in tweet.getAttr("class")
  result.profile = parseTweetProfile(tweet)

  let time = tweet.querySelector(".js-short-timestamp")
  result.time = fromUnix(parseInt(time.getAttr("data-time", "0")))
  result.shortTime = time.innerText()

  result.replies = "0"
  result.likes = "0"
  result.retweets = "0"

  for action in tweet.querySelectorAll(".ProfileTweet-actionCountForAria"):
    let
      text = action.innerText.split()
      num = text[0]

    case text[1]
    of "replies": result.replies = num
    of "likes": result.likes = num
    of "retweets": result.retweets = num
    else: discard

  let by = tweet.selectText(".js-retweet-text > a > b")
  if by.len > 0:
    result.retweetBy = some(by)

  for photo in tweet.querySelectorAll(".AdaptiveMedia-photoContainer"):
    result.photos.add photo.attrs["data-image-url"]

  let player = tweet.selectAttr(".PlayableMedia-player", "style")
  if player.len > 0:
    let thumb = player.replace(re".+:url\('([^']+)'\)", "$1")
    if "tweet_video" in thumb:
      result.gif = some(thumb.replace(re".+thumb/([^\.']+)\.jpg.*", "$1"))
    else:
      result.videoThumb = some(thumb)

proc parseTweets*(node: XmlNode): Tweets =
  if node.isNil: return
  node.querySelectorAll(".tweet").map(parseTweet)

template selectTweets*(node: XmlNode; class: string): untyped =
  parseTweets(node.querySelector(class))

proc parseConversation*(node: XmlNode): Conversation =
  result.tweet = parseTweet(node.querySelector(".permalink-tweet-container > .tweet"))
  result.before = node.selectTweets(".in-reply-to")

  let replies = node.querySelector(".replies-to")
  if replies.isNil: return

  result.after = replies.selectTweets(".ThreadedConversation--selfThread")

  for reply in replies.querySelectorAll("li > .stream-items"):
    let thread = parseTweets(reply)
    if not thread.anyIt(it in result.after):
      result.replies.add thread