132 lines
4.7 KiB
Nim
132 lines
4.7 KiB
Nim
# SPDX-License-Identifier: AGPL-3.0-only
|
|
import asyncdispatch, httpclient, uri, strutils, sequtils, sugar
|
|
import packedjson
|
|
import types, query, formatters, consts, apiutils, parser
|
|
import experimental/parser as newParser
|
|
|
|
proc getGraphUser*(id: string): Future[User] {.async.} =
|
|
if id.len == 0 or id.any(c => not c.isDigit): return
|
|
let
|
|
variables = %*{"userId": id, "withSuperFollowsUserFields": true}
|
|
js = await fetchRaw(graphUser ? {"variables": $variables}, Api.userRestId)
|
|
result = parseGraphUser(js)
|
|
|
|
proc getGraphListBySlug*(name, list: string): Future[List] {.async.} =
|
|
let
|
|
variables = %*{"screenName": name, "listSlug": list, "withHighlightedLabel": false}
|
|
url = graphListBySlug ? {"variables": $variables}
|
|
result = parseGraphList(await fetch(url, Api.listBySlug))
|
|
|
|
proc getGraphList*(id: string): Future[List] {.async.} =
|
|
let
|
|
variables = %*{"listId": id, "withHighlightedLabel": false}
|
|
url = graphList ? {"variables": $variables}
|
|
result = parseGraphList(await fetch(url, Api.list))
|
|
|
|
proc getGraphListMembers*(list: List; after=""): Future[Result[User]] {.async.} =
|
|
if list.id.len == 0: return
|
|
let
|
|
variables = %*{
|
|
"listId": list.id,
|
|
"cursor": after,
|
|
"withSuperFollowsUserFields": false,
|
|
"withBirdwatchPivots": false,
|
|
"withDownvotePerspective": false,
|
|
"withReactionsMetadata": false,
|
|
"withReactionsPerspective": false,
|
|
"withSuperFollowsTweetFields": false
|
|
}
|
|
url = graphListMembers ? {"variables": $variables}
|
|
result = parseGraphListMembers(await fetchRaw(url, Api.listMembers), after)
|
|
|
|
proc getListTimeline*(id: string; after=""): Future[Timeline] {.async.} =
|
|
if id.len == 0: return
|
|
let
|
|
ps = genParams({"list_id": id, "ranking_mode": "reverse_chronological"}, after)
|
|
url = listTimeline ? ps
|
|
result = parseTimeline(await fetch(url, Api.timeline), after)
|
|
|
|
proc getUser*(username: string): Future[User] {.async.} =
|
|
if username.len == 0: return
|
|
let
|
|
ps = genParams({"screen_name": username})
|
|
json = await fetchRaw(userShow ? ps, Api.userShow)
|
|
result = parseUser(json, username)
|
|
|
|
proc getUserById*(userId: string): Future[User] {.async.} =
|
|
if userId.len == 0: return
|
|
let
|
|
ps = genParams({"user_id": userId})
|
|
json = await fetchRaw(userShow ? ps, Api.userShow)
|
|
result = parseUser(json)
|
|
|
|
proc getTimeline*(id: string; after=""; replies=false): Future[Timeline] {.async.} =
|
|
if id.len == 0: return
|
|
let
|
|
ps = genParams({"userId": id, "include_tweet_replies": $replies}, after)
|
|
url = timeline / (id & ".json") ? ps
|
|
result = parseTimeline(await fetch(url, Api.timeline), after)
|
|
|
|
proc getMediaTimeline*(id: string; after=""): Future[Timeline] {.async.} =
|
|
if id.len == 0: return
|
|
let url = mediaTimeline / (id & ".json") ? genParams(cursor=after)
|
|
result = parseTimeline(await fetch(url, Api.timeline), after)
|
|
|
|
proc getPhotoRail*(name: string): Future[PhotoRail] {.async.} =
|
|
if name.len == 0: return
|
|
let
|
|
ps = genParams({"screen_name": name, "trim_user": "true"},
|
|
count="18", ext=false)
|
|
url = photoRail ? ps
|
|
result = parsePhotoRail(await fetch(url, Api.timeline))
|
|
|
|
proc getSearch*[T](query: Query; after=""): Future[Result[T]] {.async.} =
|
|
when T is User:
|
|
const
|
|
searchMode = ("result_filter", "user")
|
|
parse = parseUsers
|
|
fetchFunc = fetchRaw
|
|
else:
|
|
const
|
|
searchMode = ("tweet_search_mode", "live")
|
|
parse = parseTimeline
|
|
fetchFunc = fetch
|
|
|
|
let q = genQueryParam(query)
|
|
if q.len == 0 or q == emptyQuery:
|
|
return Result[T](beginning: true, query: query)
|
|
|
|
let url = search ? genParams(searchParams & @[("q", q), searchMode], after)
|
|
try:
|
|
result = parse(await fetchFunc(url, Api.search), after)
|
|
result.query = query
|
|
except InternalError:
|
|
return Result[T](beginning: true, query: query)
|
|
|
|
proc getTweetImpl(id: string; after=""): Future[Conversation] {.async.} =
|
|
let url = tweet / (id & ".json") ? genParams(cursor=after)
|
|
result = parseConversation(await fetch(url, Api.tweet), id)
|
|
|
|
proc getReplies*(id, after: string): Future[Result[Chain]] {.async.} =
|
|
result = (await getTweetImpl(id, after)).replies
|
|
result.beginning = after.len == 0
|
|
|
|
proc getTweet*(id: string; after=""): Future[Conversation] {.async.} =
|
|
result = await getTweetImpl(id)
|
|
if after.len > 0:
|
|
result.replies = await getReplies(id, after)
|
|
|
|
proc getStatus*(id: string): Future[Tweet] {.async.} =
|
|
let url = status / (id & ".json") ? genParams()
|
|
result = parseStatus(await fetch(url, Api.status))
|
|
|
|
proc resolve*(url: string; prefs: Prefs): Future[string] {.async.} =
|
|
let client = newAsyncHttpClient(maxRedirects=0)
|
|
try:
|
|
let resp = await client.request(url, HttpHead)
|
|
result = resp.headers["location"].replaceUrls(prefs)
|
|
except:
|
|
discard
|
|
finally:
|
|
client.close()
|