#!/usr/bin/env python3

# programmes expire, so i have to update the times in the tests every time i run them
# but thats a massive ballache, so i end up just not running them, which leads to cockups
# so, this script has the tests automatically use the latest episode as you run it, by setting dynamically generated time values
# everything else is always the same so it should be fine lol


import datetime
import os
import sys
import unittest

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
sys.path.insert(0, "/home/g/Downloads/yt-dlp/")  # TODO: un-hardcode. has to be the source/git repo because pip doesnt carry the tests

from yt_dlp_plugins.extractor import radiko_time as rtime

MON, TUE, WED, THU, FRI, SAT, SUN = range(7)
weekdays = {0: "MON", 1: "TUE", 2: "WED", 3: "THU", 4: "FRI", 5: "SAT", 6: "SUN"}

now = rtime.RadikoTime.now(tz = rtime.JST)
UTC = datetime.timezone.utc

def get_latest_airtimes(now, weekday, hour, minute, duration):
	days_after_weekday = (7 - (now.weekday() - weekday)) % 7
	latest_airdate = (now + datetime.timedelta(days=days_after_weekday)).replace(hour=hour, minute=minute, second=0, microsecond=0)
	if (latest_airdate + duration) > now:
		latest_airdate -= datetime.timedelta(days=7)
	return latest_airdate, latest_airdate + duration

def get_test_timefields(airtime, release_time):
	return {
		"timestamp": airtime.timestamp(),
		"release_timestamp": release_time.timestamp(),
		"upload_date": airtime.astimezone(UTC).strftime("%Y%m%d"),
		"release_date": release_time.astimezone(UTC).strftime("%Y%m%d"),

		"duration": (release_time - airtime).total_seconds(),
	}




from yt_dlp_plugins.extractor.radiko import RadikoTimeFreeIE, RadikoShareIE, RadikoLiveIE, RadikoShareIE, RadikoPersonIE, RadikoStationButtonIE

RadikoTimeFreeIE._TESTS = []



# TOKYO MOON - interfm - EVERY FRI 2300
airtime, release_time = get_latest_airtimes(now, FRI, 23, 0, datetime.timedelta(hours=1))
RadikoTimeFreeIE._TESTS.append({
	"url": f"https://radiko.jp/#!/ts/INT/{airtime.timestring()}",
	"info_dict": {
		"ext": "m4a",
		"id": f"INT-{airtime.timestring()}",

		**get_test_timefields(airtime, release_time),

		'title': 'TOKYO MOON',
		'description': 'md5:4185349a530cfc4d0580e6996a511273',
		'uploader': 'interfm',
		'uploader_id': 'INT',
		'uploader_url': 'https://www.interfm.co.jp/',
		'channel': 'interfm',
		'channel_id': 'INT',
		'channel_url': 'https://www.interfm.co.jp/',
		'thumbnail': 'https://program-static.cf.radiko.jp/ehwtw6mcvy.jpg',
		'chapters': list,
		'tags': ['松浦俊夫', 'ジャズの魅力を楽しめる'],
		'cast': ['松浦\u3000俊夫'],
		'series': 'Tokyo Moon',
		'live_status': 'was_live',
	}
})


# late-night/v. early morning show, to test broadcast day handling
# this should be monday 27:00 / tuesday 03:00
airtime, release_time = get_latest_airtimes(now, TUE, 3, 0, datetime.timedelta(hours=2))
RadikoTimeFreeIE._TESTS.append({
	"url": f"https://radiko.jp/#!/ts/TBS/{airtime.timestring()}",
	"info_dict": {
		"ext": "m4a",
		"id": f"TBS-{airtime.timestring()}",

		**get_test_timefields(airtime, release_time),
		'title': 'CITY CHILL CLUB',
		'description': r"re:^目を閉じて…リラックスして[\S\s]+chill@tbs.co.jp$",
		'uploader': 'TBSラジオ',
		'uploader_id': 'TBS',
		'uploader_url': 'https://www.tbsradio.jp/',
		'channel': 'TBSラジオ',
		'channel_id': 'TBS',
		'channel_url': 'https://www.tbsradio.jp/',
		'thumbnail': 'https://program-static.cf.radiko.jp/nrf8fowbjo.jpg',
		'chapters': list,
		'tags': ['CCC905', '音楽との出会いが楽しめる', '人気アーティストトーク', '音楽プロデューサー出演', 'ドライブ中におすすめ', '寝る前におすすめ', '学生におすすめ'],
		'cast': list,
		'series': 'CITY CHILL CLUB',
		'live_status': 'was_live',
	},
})


# testing 29-hour clock handling
airtime, release_time = get_latest_airtimes(now, WED, 0, 0, datetime.timedelta(minutes=55))
share_timestring = (airtime - datetime.timedelta(days=1)).strftime("%Y%m%d") + "240000"

RadikoShareIE._TESTS = [{
		"url": f"http://radiko.jp/share/?sid=FMT&t={share_timestring}",
		"info_dict": {
			"live_status": "was_live",
			"ext": "m4a",
			"id": f"FMT-{airtime.timestring()}",

			**get_test_timefields(airtime, release_time),

			"title": "JET STREAM",
			"series": "JET STREAM",
			"description": r"re:^JET STREAM・・・作家が描く世界への旅。[\s\S]+https://www.tfm.co.jp/f/jetstream/message$",
			"chapters": list,
			"thumbnail": "https://program-static.cf.radiko.jp/greinlrspi.jpg",

			"channel": "TOKYO FM",
			"channel_id": "FMT",
			"channel_url": "https://www.tfm.co.jp/",
			"uploader": "TOKYO FM",
			"uploader_id": "FMT",
			"uploader_url": "https://www.tfm.co.jp/",

			"cast": ["福山雅治"],
			"tags": ["福山雅治", "夜間飛行", "音楽との出会いが楽しめる", "朗読を楽しめる", "寝る前に聴きたい"],
		},
	}]



IEs = [
	RadikoTimeFreeIE, RadikoShareIE,
	RadikoLiveIE, RadikoShareIE, RadikoPersonIE, RadikoStationButtonIE
]

import test.helper as th

# override to only get testcases from our IEs

def _new_gettestcases(include_onlymatching=False):
	import yt_dlp.plugins as plugins
	plugins.load_all_plugins()

	for ie in IEs:
		yield from ie.get_testcases(include_onlymatching)

def _new_getwebpagetestcases():
	import yt_dlp.plugins as plugins
	plugins.load_all_plugins()

	for ie in IEs:
		for tc in ie.get_webpage_testcases():
			tc.setdefault('add_ie', []).append('Generic')
			yield tc

th.gettestcases = _new_gettestcases
th.getwebpagetestcases = _new_getwebpagetestcases

import test.test_download as td

class TestDownload(td.TestDownload):
	pass

if __name__ == "__main__":
	unittest.main()