nitter/src/parserutils.nim

import xmltree, htmlparser, strtabs, strformat, times
import nimquery, regex

import ./types, ./formatters, ./api

const
  thumbRegex = re".+:url\('([^']+)'\)"
  gifRegex = re".+thumb/([^\.']+)\.jpg.*"

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 getHeader(profile: XmlNode): XmlNode =
  result = profile.querySelector(".permalink-header")
  if result.isNil:
    result = profile.querySelector(".stream-item-header")
  if result.isNil:
    result = profile.querySelector(".ProfileCard-userFields")

proc isVerified*(profile: XmlNode): bool =
  getHeader(profile).selectText(".Icon.Icon--verified").len > 0

proc isProtected*(profile: XmlNode): bool =
  getHeader(profile).selectText(".Icon.Icon--protected").len > 0

proc getName*(profile: XmlNode; selector: string): string =
  profile.selectText(selector).stripText()

proc getUsername*(profile: XmlNode; selector: string): string =
  profile.selectText(selector).strip(chars={'@', ' '})

proc emojify*(node: XmlNode) =
  for i in node.querySelectorAll(".Emoji"):
    i.add newText(i.getAttr("alt"))

proc getQuoteText*(tweet: XmlNode): string =
  let
    text = tweet.querySelector(".QuoteTweet-text")
    hasEmojis = not text.querySelector(".Emoji").isNil

  if hasEmojis:
    emojify(text)

  result = stripText(selectText(text, ".tweet-text"))
  result = stripTwitterUrls(result)

proc getTweetText*(tweet: XmlNode): string =
  let
    selector = ".tweet-text > a.twitter-timeline-link.u-hidden"
    link = tweet.selectAttr(selector, "data-expanded-url")
    quote = tweet.querySelector(".QuoteTweet")
    text = tweet.querySelector(".tweet-text")
    hasEmojis = not text.querySelector(".Emoji").isNil

  if hasEmojis:
    emojify(text)

  result = stripText(selectText(text, ".tweet-text"))

  if not quote.isNil and link.len > 0:
    result = result.replace(link, "")

  result = stripTwitterUrls(result)

proc getTime(tweet: XmlNode): XmlNode =
  tweet.querySelector(".js-short-timestamp")

proc getTimestamp*(tweet: XmlNode): Time =
  let time = getTime(tweet).getAttr("data-time", "0")
  fromUnix(parseInt(time))

proc getShortTime*(tweet: XmlNode): string =
  getTime(tweet).innerText()

proc getBio*(profile: XmlNode; selector: string): string =
  profile.selectText(selector).stripText()

proc getAvatar*(profile: XmlNode; selector: string): string =
  profile.selectAttr(selector, "src").getUserpic()

proc getBanner*(tweet: XmlNode): string =
  let url = tweet.selectAttr("svg > image", "xlink:href")

  if url.len > 0:
    result = url.replace("600x200", "1500x500")
  else:
    result = tweet.selectAttr(".ProfileCard-bg", "style")

  if result.len == 0:
    result = "background-color: #161616"

proc getPopupStats*(profile: var Profile; node: XmlNode) =
  for s in node.querySelectorAll( ".ProfileCardStats-statLink"):
    let text = s.getAttr("title").split(" ")[0]
    case s.getAttr("href").split("/")[^1]
    of "followers": profile.followers = text
    of "following": profile.following = text
    else: profile.tweets = text

proc getIntentStats*(profile: var Profile; node: XmlNode) =
  profile.tweets = "?"
  for s in node.querySelectorAll( "dd.count > a"):
    let text = s.innerText()
    case s.getAttr("href").split("/")[^1]
    of "followers": profile.followers = text
    of "following": profile.following = text

proc getTweetStats*(tweet: Tweet; node: XmlNode) =
  tweet.replies = "0"
  tweet.retweets = "0"
  tweet.likes = "0"
  for action in node.querySelectorAll(".ProfileTweet-actionCountForAria"):
    let text = action.innerText.split()
    case text[1]
    of "replies":  tweet.replies = text[0]
    of "likes":    tweet.likes = text[0]
    of "retweets": tweet.retweets = text[0]

proc getGif(player: XmlNode): Gif =
  let
    thumb = player.getAttr("style").replace(thumbRegex, "$1")
    id = thumb.replace(gifRegex, "$1")
    url = fmt"https://video.twimg.com/tweet_video/{id}.mp4"
  Gif(url: url, thumb: thumb)

proc getTweetMedia*(tweet: Tweet; node: XmlNode) =
  for photo in node.querySelectorAll(".AdaptiveMedia-photoContainer"):
    tweet.photos.add photo.attrs["data-image-url"]

  let player = node.querySelector(".PlayableMedia")
  if player.isNil:
    return

  if "gif" in player.getAttr("class"):
    tweet.gif = some(getGif(player.querySelector(".PlayableMedia-player")))
  elif "video" in player.getAttr("class"):
    tweet.video = some(Video())

proc getQuoteMedia*(quote: var Quote; node: XmlNode) =
  let sensitive = node.querySelector(".QuoteTweet--sensitive")
  if not sensitive.isNil:
    quote.sensitive = true
    return

  let media = node.querySelector(".QuoteMedia")
  if not media.isNil:
    quote.thumb = some(media.selectAttr("img", "src"))

  let badge = node.querySelector(".AdaptiveMedia-badgeText")
  let gifBadge = node.querySelector(".Icon--gifBadge")

  if not badge.isNil:
    quote.badge = some(badge.innerText())
  elif not gifBadge.isNil:
    quote.badge = some("GIF")