From c760b5d965a7db394f2a5a2b6b03474ae1d4254f Mon Sep 17 00:00:00 2001 From: Joseph Nuthalapati Date: Thu, 28 Nov 2019 23:11:57 +0530 Subject: [PATCH] Add limit option with default set to 10 results Signed-off-by: Joseph Nuthalapati --- TODO.org | 1 + clitube.nim | 7 +++++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/TODO.org b/TODO.org index 0377b89..a34f276 100644 --- a/TODO.org +++ b/TODO.org @@ -5,3 +5,4 @@ * TODO Pretty colors in terminal * TODO Preferences should go into a config file * TODO Terminal color themes +* TODO Get rid of temp file or make one temp file per user diff --git a/clitube.nim b/clitube.nim index a1ba868..b6dda43 100644 --- a/clitube.nim +++ b/clitube.nim @@ -11,6 +11,9 @@ import xmltree # Supported video players in order of preference let supportedPlayers = @["mpv", "mplayer", "vlc"] +# Only show these many results +let limit = 10 + # Hard-coded terminal colors proc reset(): string {.procvar.} = "\e[0m" proc bold*(s: string): string {.procvar.} = "\e[1m" & s & reset() @@ -32,16 +35,16 @@ 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"])) + map(a => (a.attrs["title"], "https://www.youtube.com" & a.attrs["href"]))[..(limit-1)] proc presentVideoOptions(searchResults: seq[SearchResult]) = + echo "\n" for index, (title, url) in searchResults: echo index, ". ", title.bold.fgMagenta, "\n", url.fgCyan, "\n" -- 2.43.0