import strutils, strformat, times, uri, tables import xmltree, htmlparser import regex import types, utils, query from unicode import Rune, `$` const ytRegex = re"([A-z.]+\.)?youtu(be\.com|\.be)" twRegex = re"(www\.|mobile\.)?twitter\.com" igRegex = re"(www\.)?instagram.com" cards = "cards.twitter.com/cards" tco = "https://t.co" nbsp = $Rune(0x000A0) wwwRegex = re"https?://(www[0-9]?\.)?" manifestRegex = re"(.+(.ts|.m3u8|.vmap))" userpicRegex = re"_(normal|bigger|mini|200x200|400x400)(\.[A-z]+)$" extRegex = re"(\.[A-z]+)$" tombstoneRegex = re"\n* *Learn more" proc stripText*(text: string): string = text.replace(nbsp, " ").strip() proc stripHtml*(text: string): string = var html = parseHtml(text) for el in html.findAll("a"): let link = el.attr("href") if "http" in link: if el.len == 0: continue el[0].text = link html.innerText() proc shortLink*(text: string; length=28): string = result = text.replace(wwwRegex, "") if result.len > length: result = result[0 ..< length] & "…" proc replaceUrl*(url: string; prefs: Prefs; absolute=""): string = result = url if prefs.replaceYouTube.len > 0: result = result.replace(ytRegex, prefs.replaceYouTube) if prefs.replaceYouTube in result: result = result.replace("/c/", "/") if prefs.replaceInstagram.len > 0: result = result.replace(igRegex, prefs.replaceInstagram) if prefs.replaceTwitter.len > 0: result = result.replace(tco, "https://" & prefs.replaceTwitter & "/t.co") result = result.replace(cards, prefs.replaceTwitter & "/cards") result = result.replace(twRegex, prefs.replaceTwitter) if absolute.len > 0: result = result.replace("href=\"/", "href=\"https://" & absolute & "/") proc proxifyVideo*(manifest: string; proxy: bool): string = proc cb(m: RegexMatch; s: string): string = result = "https://video.twimg.com" & s[m.group(0)[0]] if proxy: result = getVidUrl(result) result = manifest.replace(manifestRegex, cb) proc getUserpic*(userpic: string; style=""): string = let pic = userpic.replace(userpicRegex, "$2") pic.replace(extRegex, style & "$1") proc getUserpic*(profile: Profile; style=""): string = getUserPic(profile.userpic, style) proc getVideoEmbed*(cfg: Config; id: int64): string = &"https://{cfg.hostname}/i/videos/{id}" proc pageTitle*(profile: Profile): string = &"{profile.fullname} (@{profile.username})" proc pageTitle*(tweet: Tweet): string = &"{pageTitle(tweet.profile)}: \"{stripHtml(tweet.text)}\"" proc pageDesc*(profile: Profile): string = if profile.bio.len > 0: stripHtml(profile.bio) else: "The latest tweets from " & profile.fullname proc getJoinDate*(profile: Profile): string = profile.joinDate.format("'Joined' MMMM YYYY") proc getJoinDateFull*(profile: Profile): string = profile.joinDate.format("h:mm tt - d MMM YYYY") proc getTime*(tweet: Tweet): string = tweet.time.format("d/M/yyyy', 'HH:mm:ss") proc getRfc822Time*(tweet: Tweet): string = tweet.time.format("ddd', 'd MMM yyyy HH:mm:ss 'GMT'") proc getTweetTime*(tweet: Tweet): string = tweet.time.format("h:mm tt' · 'MMM d', 'YYYY") proc getShortTime*(tweet: Tweet): string = let now = now().utc then = tweet.time.utc since = now - then if now.year != then.year: result = tweet.time.format("d MMM yyyy") elif since.inDays >= 1: result = tweet.time.format("MMM d") elif since.inHours >= 1: result = $since.inHours & "h" elif since.inMinutes >= 1: result = $since.inMinutes & "m" elif since.inSeconds > 1: result = $since.inSeconds & "s" else: # this shouldn't happen, but just in case result = "now" proc getLink*(tweet: Tweet; focus=true): string = if tweet.id == 0: return var username = tweet.profile.username if username.len == 0: username = "i" result = &"/{username}/status/{tweet.id}" if focus: result &= "#m" proc getTombstone*(text: string): string = text.replace(tombstoneRegex, "").stripText().strip(chars={' ', '\n'}) proc getTwitterLink*(path: string; params: Table[string, string]): string = let twitter = parseUri("https://twitter.com") username = params.getOrDefault("name") query = initQuery(params, username) if "/search" notin path: return $(twitter / path ? filterParams(params)) let p = { "f": $query.kind, "q": genQueryParam(query), "src": "typed_query" } result = $(parseUri("https://twitter.com") / path ? p) if username.len > 0: result = result.replace("/" & username, "") proc getLocation*(u: Profile | Tweet): (string, string) = if "://" in u.location: return (u.location, "") let loc = u.location.split(":") let url = if loc.len > 1: "/search?q=place:" & loc[1] else: "" (loc[0], url) proc getSuspended*(username: string): string = &"User \"{username}\" has been suspended"