X-Git-Url: https://njoseph.me/gitweb/nimcoon.git/blobdiff_plain/2d9a92ad0d51695e6a4e7ea457d297b7abe9d4ac..68fd8100bc36465b29ae76b9f41d9eb7fa052e7a:/cli_tube.nim diff --git a/cli_tube.nim b/cli_tube.nim deleted file mode 100644 index a1ba868..0000000 --- a/cli_tube.nim +++ /dev/null @@ -1,51 +0,0 @@ -import htmlparser -import httpClient -import os -import osproc -import sequtils, sugar -import strtabs -import strutils -import uri -import xmltree - -# Supported video players in order of preference -let supportedPlayers = @["mpv", "mplayer", "vlc"] - -# Hard-coded terminal colors -proc reset(): string {.procvar.} = "\e[0m" -proc bold*(s: string): string {.procvar.} = "\e[1m" & s & reset() - -proc fgMagenta*(s: string): string {.procvar.} = "\e[35m" & s & reset() -proc fgCyan*(s: string): string {.procvar.} = "\e[36m" & s & reset() - -type SearchResult = tuple[title: string, url: string] - -proc selectMediaPlayer(): string = - let availablePlayers = filterIt(supportedPlayers, execProcess("which " & it).len != 0) - if len(availablePlayers) == 0: - echo "Please install one of the supported media players: ", supportedPlayers - raise newException(OSError, "No supported media player found") - else: - return availablePlayers[0] - -proc getYoutubePage(searchQuery: string): string = - let queryParam = encodeUrl(searchQuery) - var client = newHttpClient() - let response = get(client, "https://www.youtube.com/results?hl=en&search_query=" & queryParam) - # TODO Get rid of temp file or make one temp file per user - writeFile("/tmp/cli-tube-page.html", response.body) - return "/tmp/cli-tube-page.html" - -proc extractTitlesAndUrls(htmlFile: string): seq[SearchResult] = - loadHtml(htmlFile).findAll("a"). - filter(a => "watch" in a.attrs["href"] and a.attrs.hasKey "title"). - map(a => (a.attrs["title"], "https://www.youtube.com" & a.attrs["href"])) - -proc presentVideoOptions(searchResults: seq[SearchResult]) = - for index, (title, url) in searchResults: - echo index, ". ", title.bold.fgMagenta, "\n", url.fgCyan, "\n" - -presentVideoOptions( - extractTitlesAndUrls( - getYoutubePage( - paramStr(1))))