Add option to play music only
[nimcoon.git] / clitube.nim
CommitLineData
44978125
JN
1import htmlparser
2import httpClient
4827df7a 3import parseopt
44978125 4import osproc
2f3c8875 5import sequtils, sugar
b40b7243 6import strformat
3d5bf3fd 7import std/[terminal]
44978125
JN
8import strtabs
9import strutils
10import uri
11import xmltree
44978125 12
2f0b8428 13import preferences
c760b5d9 14
2f3c8875
JN
15type SearchResult = tuple[title: string, url: string]
16
933e339b
JN
17proc selectMediaPlayer(): string =
18 let availablePlayers = filterIt(supportedPlayers, execProcess("which " & it).len != 0)
19 if len(availablePlayers) == 0:
b40b7243 20 stderr.writeLine &"Please install one of the supported media players: {supportedPlayers}"
933e339b
JN
21 raise newException(OSError, "No supported media player found")
22 else:
23 return availablePlayers[0]
44978125 24
39a495a9
JN
25proc getYoutubePage(searchQuery: string): string =
26 let queryParam = encodeUrl(searchQuery)
485b1053 27 let client = newHttpClient()
b40b7243 28 let response = get(client, &"https://www.youtube.com/results?hl=en&search_query={queryParam}")
485b1053 29 return $response.body
44978125 30
2f3c8875 31proc extractTitlesAndUrls(htmlFile: string): seq[SearchResult] =
485b1053 32 parseHtml(htmlFile).findAll("a").
44978125 33 filter(a => "watch" in a.attrs["href"] and a.attrs.hasKey "title").
c760b5d9 34 map(a => (a.attrs["title"], "https://www.youtube.com" & a.attrs["href"]))[..(limit-1)]
44978125 35
2f3c8875 36proc presentVideoOptions(searchResults: seq[SearchResult]) =
efcc0441 37 echo ""
2f3c8875 38 for index, (title, url) in searchResults:
3d5bf3fd 39 styledEcho $index, ". ", styleBright, fgMagenta, title, "\n", resetStyle, fgCyan, url, "\n"
44978125 40
84d84780
JN
41let player = selectMediaPlayer()
42
4827df7a
JN
43var searchQuery = ""
44var musicOnly = false
45
46for kind, key, value in getopt():
47 case kind
48 of cmdArgument:
49 searchQuery = key
50 of cmdShortOption, cmdLongOption:
51 case key
52 of "m", "music": musicOnly = true
53 of cmdEnd:
54 discard
55
56let noVideo = if musicOnly: "--no-video" else: ""
57
58if "https://www.youtube.com" in searchQuery:
59 discard execProcess(&"{player} {searchQuery}")
84d84780
JN
60 quit(0)
61
4827df7a
JN
62
63let searchResults = extractTitlesAndUrls(getYoutubePage(searchQuery))
efcc0441
JN
64
65presentVideoOptions(searchResults)
66
3d5bf3fd 67stdout.styledWrite(fgYellow, "Choose video number: ")
2f0b8428 68let number = parseInt(readLine(stdin))
efcc0441 69
3d5bf3fd 70styledEcho "\n", fgGreen, "Playing ", styleBright, fgMagenta, searchResults[number].title
efcc0441 71
3d5bf3fd 72# Play the video using the preferred/available media player
4827df7a 73discard execProcess(&"{player} {noVideo} {searchResults[number].url}")