from rich.console import Console
#from rich import inspect
import platform
from pathlib import Path
import os
import json
import argparse
import youtube_dl
from get_channels import retrieve_youtube_subscriptions
from print_logo import print_logo

if platform.system() == 'Windows':
	import msvcrt
else:
	from getch import getch

parser = argparse.ArgumentParser()
parser.add_argument('-i', '--input', action='store', type=str, 
	help='Path to json file generated by previously running this program.')
parser.add_argument('-o', '--output', action='store', type=str, 
	help='Output folder.')
parser.add_argument('-a', '--all', action='store_true', 
	help='Download all subscriptions.')

args = parser.parse_args()

json_input = args.input
output_dir = args.output or 'output'
download_all = args.all

c = Console()
ydl_opts = {
	'format': 'best'
}

print_logo()

if json_input:
	f = open(json_input, "r", encoding='utf-8')
	all_channels = json.loads(f.read())
	f.close()
else:
	all_channels = retrieve_youtube_subscriptions()
	curr_channel = 0
	c.print(f'You will be prompted if you want to download a channel for each of your subscriptions. (total {len(all_channels)})', style='bold')
	for ch in all_channels:
		if download_all:
			ch['download'] = True
		else:
			curr_channel += 1
			c.print(f'[dim][{curr_channel}/{len(all_channels)}]:[/dim] {ch["title"]} [cyan]\[y/n]')
			while True:
				if platform.system() == 'Windows':
					key = str(msvcrt.getch())
					if key == "b'y'":
						ch['download'] = True
						break
					elif key == "b'n'":
						ch['download'] = False
						break
					else:
						c.print('Press "y" or "n"', style='yellow')
				else:
					key = getch()
					if key == "y":
						ch['download'] = True
						break
					elif key == "n":
						ch['download'] = False
						break
					else:
						c.print('Press "y" or "n"', style='yellow')

	c.print('All done! 🎉')
	c.print('Saving to download_list.json...', style='italic')
	f = open("download_list.json", "w", encoding='utf-8')
	f.write(json.dumps(all_channels))
	f.close()

for ch in all_channels:
	if ch['download']:
		ydl_opts['outtmpl'] = '{}/{}/%(title)s.%(ext)s'.format(output_dir, ch['title'])
		with youtube_dl.YoutubeDL(ydl_opts) as ydl:
			Path(os.path.join(output_dir, ch['title'])).mkdir(parents=True, exist_ok=True)
			ydl.download(['https://www.youtube.com/channel/{}'.format(ch["id"])])