clitube.nim

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