X-Git-Url: https://njoseph.me/gitweb/nimcoon.git/blobdiff_plain/c8812b680092eea171bcbd3a0c025467031b74e2..d65a1dcf24e9c17cea89385c78b1c39c314f0f97:/lib.nim

diff --git a/lib.nim b/lib.nim
new file mode 100644
index 0000000..25669a3
--- /dev/null
+++ b/lib.nim
@@ -0,0 +1,62 @@
+import
+  htmlparser,
+  httpClient,
+  osproc,
+  sequtils,
+  sugar,
+  strformat,
+  std/[terminal],
+  strtabs,
+  strutils,
+  uri,
+  xmltree
+
+import config
+
+type
+  SearchResult* = tuple[title: string, url: string]
+  CommandLineOptions* = tuple[searchQuery: string, musicOnly: bool, feelingLucky: bool, fullScreen: bool]
+
+proc selectMediaPlayer*(): string =
+  let availablePlayers = filterIt(supportedPlayers, execProcess("which " & it).len != 0)
+  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)
+  let client = newHttpClient()
+  let response = get(client, &"https://www.youtube.com/results?hl=en&search_query={queryParam}")
+  return $response.body
+
+proc extractTitlesAndUrls*(html: string): seq[SearchResult] =
+  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"]))[..(limit-1)]
+
+proc presentVideoOptions*(searchResults: seq[SearchResult]) =
+  echo ""
+  for index, (title, url) in searchResults:
+    styledEcho $index, ". ", styleBright, fgMagenta, title, "\n", resetStyle, fgCyan, url, "\n"
+
+proc play*(player: string, args: openArray[string]) =
+  # poEchoCmd can be added to options for debugging
+  discard execProcess(player, args=args, options={poStdErrToStdOut, poUsePath})
+  quit(0)
+
+func urlLongen(url: string): string =
+  url.replace("youtu.be/", "www.youtube.com/watch?v=")
+
+func stripZshEscaping(url: string): string =
+  url.replace("\\", "")
+
+func sanitizeURL*(url: string): string =
+  urlLongen(stripZshEscaping(url))
+
+proc directPlay*(searchQuery: string, player: string) =
+  if "watch?" in searchQuery or "videos/watch" in searchQuery or "soundcloud.com" in searchQuery:
+    play(player, args=[sanitizeURL(searchQuery)])
+  elif searchQuery.startswith("magnet:"):
+    play("peerflix", args=[searchQuery, &"--{player}"])