import
- os,
parseopt,
- std/[terminal],
strformat,
strutils,
tables
of "d", "download": options["download"] = true
of cmdEnd: discard
- return (searchQuery, options)
+ (searchQuery, options)
proc isValidOptions*(options: Options): bool =
# Check for invalid combinations of options
- var invalidCombinations = [("musicOnly", "fullScreen")]
+ var invalidCombinations = [("musicOnly", "fullScreen"), ("download", "fullScreen")]
+ result = true
for combination in invalidCombinations:
if options[combination[0]] and options[combination[1]]:
stderr.writeLine fmt"Incompatible options provided: {combination[0]} and {combination[1]}"
- return false
- return true
-
+ result = false
proc main() =
let
if(not isValidOptions(options)):
quit(1)
- if searchQuery.startswith("https:") or searchQuery.startswith("magnet:"):
- directPlay(searchQuery, player)
- quit(0)
-
- let searchResults = extractTitlesAndUrls(getYoutubePage(searchQuery))
-
- proc getUserInput(): string =
- if options["feelingLucky"]: "0"
- else:
- presentVideoOptions(searchResults[..(limit-1)])
- stdout.styledWrite(fgYellow, "Choose video number: ")
- readLine(stdin)
-
- # This is a pure function with no side effects
- func buildPlayerArgs(number: int): seq[string] =
- var args = @[searchResults[number].url]
- if options["musicOnly"]: args.add("--no-video")
- if options["fullScreen"]: args.add("--fullscreen")
- return args
-
- func buildMusicDownloadArgs(number: int): seq[string] =
- {.noSideEffect.}:
- var args = @["--ignore-errors", "-f", "bestaudio", "--extract-audio", "--audio-format", "mp3", "--audio-quality", "0", "-o"]
- let downloadLocation = &"'{expandTilde(musicDownloadDirectory)}/{searchResults[number].title}.mp3'"
- args.add(downloadLocation)
- args.add(searchResults[number].url)
- return args
-
- func buildVideoDownloadArgs(number: int): seq[string] =
- {.noSideEffect.}:
- var args = @["-f", "best", "-o"]
- let downloadLocation = &"'{expandTilde(videoDownloadDirectory)}/%(title)s.%(ext)s'"
- args.add(downloadLocation)
- args.add(searchResults[number].url)
- return args
-
- proc handleUserInput(number: int) =
+ if searchQuery.startswith("http") or searchQuery.startswith("magnet"):
if options["download"]:
- if options["musicOnly"]:
- download(buildMusicDownloadArgs(number), searchResults[number].title)
- else:
- download(buildVideoDownloadArgs(number), searchResults[number].title)
+ directDownload(sanitizeURL(searchQuery), options["musicOnly"])
else:
- play(player, buildPlayerArgs(number), searchResults[number].title)
-
-
- while(true):
- let userInput = getUserInput()
-
- if userInput == "all":
- for number in 0..(len(searchResults)):
- # TODO `spawn` this?
- handleUserInput(number)
+ directPlay(sanitizeURL(searchQuery), player, options)
+ quit(0)
- if userInput == "q":
- break
+ let searchResults = extractTitlesAndUrls(getYoutubePage(searchQuery))
- # Play the video using the preferred/available media player
- let videoNumber = parseInt(userInput)
- handleUserInput(videoNumber)
- if options["feelingLucky"]:
- break
+ present(searchResults, options, (0, limit-1), player)
when isMainModule: