X-Git-Url: http://njoseph.me/gitweb/nimcoon.git/blobdiff_plain/13a4017d99baaf7919263edaf05b13063e91cb8d..5f9cdfeff96bb2848e3dc72bab410859d78b9c65:/src/lib.nim diff --git a/src/lib.nim b/src/lib.nim index 3d3f515..204c328 100644 --- a/src/lib.nim +++ b/src/lib.nim @@ -1,109 +1,132 @@ import - htmlparser, httpClient, + json, os, osproc, + re, sequtils, - sugar, - strformat, std/[terminal], strformat, - strtabs, strutils, - tables, - uri, - xmltree + tables + +import + config, + types, + youtube + + +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}" -import config -type - Options* = Table[string, bool] - SearchResult* = tuple[title: string, url: string] - SearchResults* = seq[tuple[title: string, url: string]] - CommandLineOptions* = tuple[searchQuery: string, options: Options] - SelectionRange* = tuple[begin: int, until: int] +proc isInstalled(program: string): bool = + execProcess("which " & program).len != 0 -# poEchoCmd can be added to options for debugging -let processOptions = {poStdErrToStdOut, poUsePath} 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): SearchResults = - {.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: 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] = + +func buildMusicDownloadArgs(url: string): seq[string] = {.noSideEffect.}: - var args = @["--ignore-errors", "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o"] let downloadLocation = &"'{expandTilde(musicDownloadDirectory)}/%(title)s.%(ext)s'" - args.add(downloadLocation) - args.add(url) - return args + @["--ignore-errors", "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", + "--audio-quality", "0", "-o", downloadLocation, url] + -func buildVideoDownloadArgs*(url: string): seq[string] = +func buildVideoDownloadArgs(url: string): seq[string] = {.noSideEffect.}: - var args = @["-f", "best", "-o"] let downloadLocation = &"'{expandTilde(videoDownloadDirectory)}/%(title)s.%(ext)s'" - args.add(downloadLocation) - args.add(url) - return args + @["-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, musicOnly: bool) = - if url.startswith("magnet:"): - if musicOnly: +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: - discard execProcess("peerflix", args=[url, &"--{player}"], options=processOptions) + # WebTorrent is so much faster! + discard execProcess("webtorrent", args=[url, &"--{player}"], options=processOptions) else: - if musicOnly: - discard execShellCmd(&"{player} --no-video {url}") - else: - discard execProcess(player, args=[url], options=processOptions) + play(player, options, url) + proc directDownload*(url: string, musicOnly: bool) = let args = if musicOnly: buildMusicDownloadArgs(url) else: buildVideoDownloadArgs(url) - discard execShellCmd(&"youtube-dl {args.join(\" \")}") + 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" @@ -112,26 +135,58 @@ proc offerSelection(searchResults: SearchResults, options: Table[string, bool], stdout.styledWrite(fgYellow, "Choose video number: ") readLine(stdin) -# This is a pure function with no side effects -func buildPlayerArgs(searchResult: SearchResult, options: Table[string, bool]): seq[string] = - var args = @[searchResult.url] - if options["musicOnly"]: args.add("--no-video") - if options["fullScreen"]: args.add("--fullscreen") - return args proc handleUserInput(searchResult: SearchResult, options: Table[string, bool], player: string) = - if options["download"]: + 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, buildPlayerArgs(searchResult, options), searchResult.title) + 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 - ]# + ##[ 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) @@ -144,14 +199,25 @@ proc present*(searchResults: SearchResults, options: Table[string, bool], select 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: - handleUserInput(searchResults[parseInt(userInput)], options, player) + 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: