Use a user-defined type for search results

[nimcoon.git] / cli_tube.nim

import htmlparser
import httpClient
import os
import osproc
import sequtils, sugar
import strtabs
import strutils
import uri
import xmltree

# TODO Pretty colors in terminal
# import terminal

# Supported video players in order of preference
# TODO Should go into a config file
let supportedPlayers = @["mpv", "mplayer", "vlc"]

type SearchResult = tuple[title: string, url: string]

proc findSupportedPlayer(): string =
  for player in supported_players:
    let playerBin = execProcess("which " & player)
    if playerBin.len != 0:
      return strip(playerBin)

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"]))

proc presentVideoOptions(searchResults: seq[SearchResult]) =
  for index, (title, url) in searchResults:
    echo index, ". ", title, "\n", url, "\n"

presentVideoOptions(
  extractTitlesAndUrls(
    getYoutubePage(
      paramStr(1))))