X-Git-Url: http://njoseph.me/gitweb/nimcoon.git/blobdiff_plain/e71f26f3b64d2bbe6c7b8d355a82c33f985aadf0..5f9cdfeff96bb2848e3dc72bab410859d78b9c65:/src/lib.nim diff --git a/src/lib.nim b/src/lib.nim index f5ee8a5..204c328 100644 --- a/src/lib.nim +++ b/src/lib.nim @@ -1,77 +1,224 @@ import - htmlparser, httpClient, + json, os, osproc, + re, sequtils, - sugar, - strformat, std/[terminal], strformat, - strtabs, strutils, - tables, - uri, - xmltree + tables + +import + config, + types, + youtube -import config -type - Options* = Table[string, bool] - SearchResult* = tuple[title: string, url: string] - CommandLineOptions* = tuple[searchQuery: string, options: Options] +let + processOptions = {poStdErrToStdOut, poUsePath, poEchoCmd} + PEERTUBE_REGEX = re"videos\/watch\/[0-9a-f]{8}\b-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-\b[0-9a-f]{12}" + + +proc isInstalled(program: string): bool = + execProcess("which " & program).len != 0 -# poEchoCmd can be added to options for debugging -let processOptions = {poStdErrToStdOut, poUsePath, poEchoCmd} proc selectMediaPlayer*(): string = - let availablePlayers = filterIt(supportedPlayers, execProcess("which " & it).len != 0) + let availablePlayers = supportedPlayers.filter(isInstalled) if len(availablePlayers) == 0: stderr.writeLine &"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) + +proc getPeerTubeMagnetLink(url: string): string = + ## Gets the magnet link of the best possible resolution from PeerTube + let uuid = url.substr(find(url, PEERTUBE_REGEX) + "videos/watch/".len) + let domainName = url.substr(8, find(url, '/', start=8) - 1) + let apiURL = &"https://{domainName}/api/v1/videos/{uuid}" let client = newHttpClient() - let response = get(client, &"https://www.youtube.com/results?hl=en&search_query={queryParam}") - return $response.body + let response = get(client, apiURL) + let jsonNode = parseJson($response.body) + jsonNode["files"][0]["magnetUri"].getStr() -func extractTitlesAndUrls*(html: string): seq[SearchResult] = - {.noSideEffect.}: - parseHtml(html).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]) = +proc presentVideoOptions*(searchResults: SearchResults) = eraseScreen() for index, (title, url) in searchResults: - styledEcho $index, ". ", styleBright, fgMagenta, title, "\n", resetStyle, fgCyan, url, "\n" + styledEcho $index, ". ", styleBright, fgMagenta, title, "\n", resetStyle, fgCyan, " ", url, "\n" + + +func buildPlayerArgs(url: string, options: Table[string, bool], player: string): seq[string] = + let musicOnly = if options["musicOnly"]: "--no-video" else: "" + let fullScreen = if options["fullScreen"]: "--fullscreen" else: "" + filterIt([url, musicOnly, fullScreen], it != "") -proc play*(player: string, args: openArray[string], title: string) = - styledEcho "\n", fgGreen, "Playing ", styleBright, fgMagenta, title + +proc play*(player: string, options: Table[string, bool], url: string, title: string = "") = + let args = buildPlayerArgs(url, options, player) + if title != "": + styledEcho "\n", fgGreen, "Playing ", styleBright, fgMagenta, title if "--no-video" in args: discard execShellCmd(&"{player} {args.join(\" \")}") else: discard execProcess(player, args=args, options=processOptions) + +func buildMusicDownloadArgs(url: string): seq[string] = + {.noSideEffect.}: + let downloadLocation = &"'{expandTilde(musicDownloadDirectory)}/%(title)s.%(ext)s'" + @["--ignore-errors", "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", + "--audio-quality", "0", "-o", downloadLocation, url] + + +func buildVideoDownloadArgs(url: string): seq[string] = + {.noSideEffect.}: + let downloadLocation = &"'{expandTilde(videoDownloadDirectory)}/%(title)s.%(ext)s'" + @["-f", "best", "-o", downloadLocation, url] + + proc download*(args: openArray[string], title: string) = styledEcho "\n", fgGreen, "Downloading ", styleBright, fgMagenta, title discard execShellCmd(&"youtube-dl {args.join(\" \")}") -func urlLongen(url: string): string = - url.replace("youtu.be/", "www.youtube.com/watch?v=") -func stripZshEscaping(url: string): string = - url.replace("\\", "") +func urlLongen(url: string): string = url.replace("youtu.be/", "www.youtube.com/watch?v=") + + +func rewriteInvidiousToYouTube*(url: string): string = + {.noSideEffect.}: + if rewriteInvidiousURLs and url.replace(".", "").contains("invidious"): + &"https://www.youtube.com/watch?v={url.split(\"=\")[1]}" + else: url + + +func stripZshEscaping(url: string): string = url.replace("\\", "") + func sanitizeURL*(url: string): string = - urlLongen(stripZshEscaping(url)) + rewriteInvidiousToYouTube(urlLongen(stripZshEscaping(url))) + + +proc directPlay*(url: string, player: string, options: Table[string, bool]) = + let url = + if find(url, PEERTUBE_REGEX) != -1 and isInstalled("webtorrent"): + getPeerTubeMagnetLink(url) + else: url + if url.startswith("magnet:") or url.endswith(".torrent"): + if options["musicOnly"]: + # TODO Replace with WebTorrent once it supports media player options + discard execShellCmd(&"peerflix '{url}' -a --{player} -- --no-video") + else: + # WebTorrent is so much faster! + discard execProcess("webtorrent", args=[url, &"--{player}"], options=processOptions) + else: + play(player, options, url) + + +proc directDownload*(url: string, musicOnly: bool) = + let args = + if musicOnly: buildMusicDownloadArgs(url) + else: buildVideoDownloadArgs(url) + if isInstalled("aria2c"): + discard execShellCmd(&"youtube-dl {args.join(\" \")} --external-downloader aria2c --external-downloader-args '-x 16 -s 16 -k 2M'") + else: + discard execShellCmd(&"youtube-dl {args.join(\" \")}") + + +proc offerSelection(searchResults: SearchResults, options: Table[string, bool], selectionRange: SelectionRange): string = + if options["feelingLucky"]: "0" + else: + presentVideoOptions(searchResults[selectionRange.begin .. selectionRange.until]) + stdout.styledWrite(fgYellow, "Choose video number: ") + readLine(stdin) + + +proc handleUserInput(searchResult: SearchResult, options: Table[string, bool], player: string) = + if options["autoPlay"]: + play(player, options, searchResult.url, searchResult.title) + let nextResult = getAutoPlayVideo(searchResult) + handleUserInput(nextResult, options, player) # inifinite playlist till user quits + elif options["download"]: + if options["musicOnly"]: + download(buildMusicDownloadArgs(searchResult.url), searchResult.title) + else: + download(buildVideoDownloadArgs(searchResult.url), searchResult.title) + else: + play(player, options, searchResult.url, searchResult.title) + + +proc isValidOptions*(options: Options): bool = + # Check for invalid combinations of options + var invalidCombinations = [("musicOnly", "fullScreen"), ("download", "fullScreen"), ("download", "autoPlay")] + result = true + for combination in invalidCombinations: + if options[combination[0]] and options[combination[1]]: + stderr.writeLine fmt"Incompatible options provided: {combination[0]} and {combination[1]}" + result = false + # TODO Make this overridable in configuration + if options["autoPlay"] and not options["musicOnly"]: + stderr.writeLine "--music-only must be provided with --auto-play. This is to prevent binge-watching." + result = false + + +proc updateOptions(options: Options, newOptions: string): Options = + result = options + + # Interactive options + for option in newOptions: + case option + of 'm': result["musicOnly"] = true + of 'f': result["fullScreen"] = true + of 'd': result["download"] = true + of 'a': result["autoPlay"] = true + else: + stderr.writeLine "Invalid option provided!" + quit(2) + + if(not isValidOptions(result)): + quit(2) + + +proc present*(searchResults: SearchResults, options: Table[string, bool], selectionRange: SelectionRange, player: string) = + ##[ Continuously present options till the user quits the application + + selectionRange: Currently available range to choose from depending on pagination + ]## + + let userInput = offerSelection(searchResults, options, selectionRange) -proc directPlay*(searchQuery: string, player: string) = - let url = sanitizeURL(searchQuery) - if searchQuery.startswith("magnet:"): - discard execProcess("peerflix", args=[url, &"--{player}"], options=processOptions) + case userInput + of "all": + for selection in selectionRange.begin .. selectionRange.until: + handleUserInput(searchResults[selection], options, player) + quit(0) + of "n": + if selectionRange.until + 1 < len(searchResults): + let newSelectionRange = (selectionRange.until + 1, min(len(searchResults) - 1, selectionRange.until + limit)) + present(searchResults, options, newSelectionRange, player) + else: + present(searchResults, options, selectionRange, player) + of "p": + if selectionRange.begin > 0: + let newSelectionRange = (selectionRange.begin - limit, selectionRange.until - limit) + present(searchResults, options, newSelectionRange, player) + else: + present(searchResults, options, selectionRange, player) + of "q": + quit(0) else: - discard execProcess(player, args=[url], options=processOptions) + if " " in userInput: + let selection = parseInt(userInput.split(" ")[0]) + let updatedOptions = updateOptions(options, userInput.split(" ")[1]) + let searchResult = searchResults[selectionRange.begin .. selectionRange.until][selection] + handleUserInput(searchResult, updatedOptions, player) + else: + let searchResult = searchResults[selectionRange.begin .. selectionRange.until][parseInt(userInput)] + handleUserInput(searchResult, options, player) + if options["feelingLucky"]: + quit(0) + else: + present(searchResults, options, selectionRange, player)