import json
from datetime import datetime, timezone, timedelta
from os.path import exists as file_exists
from urllib.request import Request, urlopen
import xml.etree.ElementTree as ET

from constants import XMLTV, CITY

date = datetime.now(timezone.utc).replace(tzinfo=None) + timedelta(hours=+9)
ymd = date.strftime('%Y%m%d')
guide = XMLTV()


def fix(inp: str) -> str:
    out = inp.replace('&', '&')
    out = out.replace('<', '＜')
    out = out.replace('>', '＞')

    return out


def genre(_: str) -> str:
    genres = json.load(open('data/genre.json', encoding='utf-8'))

    for gen in genres:
        if _ == gen:
            return genres[gen]

    return '未知'


def request_url() -> str:
    request = 'https://tvguide.myjcom.jp/api/getEpgInfo/?channels='
    xmlfile = open('data/channels.xml', 'r', encoding='utf-8').read().splitlines()

    for line in xmlfile:
        tree = ET.ElementTree(ET.fromstring(line))
        root = tree.getroot()
        chid = root.attrib['id']
        request += f'{chid}_{ymd}%2C'

    return f'{request}&rectime=&rec4k='


if not file_exists('data/channels.xml'):
    c_area = CITY.OSAKA1.value
    c_adult = 'true'

    for c_type in [2, 3, 5, 120]:  # GR / BS / BS4K / CS
        req_channels = Request('https://tvguide.myjcom.jp/api/mypage/getEpgChannelList/?channelType={}&area={}&channelGenre=&course=&chart=&is_adult={}'.format(c_type, c_area, c_adult))
        req_channels.add_header('user-agent', 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36')  # This is required, otherwise we get a 403: Forbidden error.
        channel_data = urlopen(req_channels).read()
        channel_data = json.loads(channel_data)

        for ch in channel_data['header']:
            chid = '{}_{}_{}_{}'.format(ch['channel_type'], ch['channel_id'], ch['network_id'], ymd)
            j1 = '200_00010_0_{}'.format(ymd)
            j2 = '200_00055_0_{}'.format(ymd)

            if j1 not in chid and j2 not in chid:
                guide.add_channel(chid[0:chid.rfind('_')], ch['channel_name'], ch['logo_url'])
                print('Added channel:', ch['channel_name'])

    guide.write_channels()
else:
    guide.channels = open('data/channels.xml', mode='r', encoding='UTF-8').read().splitlines()


for x in range(0, 7):
    req = Request(request_url())
    req.add_header('user-agent', 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36')
    data = urlopen(req).read()
    data = json.loads(data)

    for chid in data:
        for epg_data in data[chid]:
            channel = chid[0:chid.rfind('_')]
            title = fix(epg_data['title'])
            comment = fix(epg_data['commentary'])
            start = datetime.fromtimestamp(datetime.strptime(epg_data['programStart'], '%Y%m%d%H%M%S').timestamp() - 32400).strftime('%Y%m%d%H%M%S')
            end = datetime.fromtimestamp(datetime.strptime(epg_data['programEnd'], '%Y%m%d%H%M%S').timestamp() - 32400).strftime('%Y%m%d%H%M%S')
            gen = genre(epg_data['sortGenre'])
            attr = epg_data['attr']
            icon = 'https://tvguide.myjcom.jp{}'.format(epg_data['imgPath']) if int(epg_data['hasImage']) == 1 and epg_data['imgPath'] else None

            guide.add_programme(channel, title, comment, start, end, gen, attr, icon)

            print('Added EPG info for', title)

    date += timedelta(days=1)
    ymd = date.strftime('%Y%m%d')

guide.create()