from __future__ import unicode_literals
from kodi_six import xbmc,xbmcgui,xbmcaddon,xbmcvfs,xbmcplugin
import sys
import time,datetime
import sqlite3
import pytz
import tzlocal
import re

try:
    from urllib.parse import quote_plus
except ImportError:
    from urllib import quote_plus

def log(x):
    xbmc.log(repr(x),xbmc.LOGERROR)

def remove_formatting(label):
    label = re.sub(r"\[/?[BI]\]", '', label, flags=re.I)
    label = re.sub(r"\[/?COLOR.*?\]", '', label, flags=re.I)
    return label

#log(sys.argv)

if len(sys.argv) == 1:
    xbmc.executebuiltin("ActivateWindow(videos,plugin://plugin.video.iptv.archive.downloader)")
    quit()

channel = sys.argv[1]
channel = channel.decode("utf8")
channel = channel.encode("utf8")

title = sys.argv[2]
date = sys.argv[3]
duration = sys.argv[4]
plot = sys.argv[5]

dateshort_format = xbmc.getRegion('dateshort')
time_format = xbmc.getRegion('time').replace('%H%H','%H')
format = "%s %s" % (dateshort_format, time_format.replace(':%S',''))
#log((date,dateshort_format,time_format,format))
start_time = datetime.datetime.fromtimestamp(time.mktime(time.strptime(date, format)))
timezone = tzlocal.get_localzone()
start_time = timezone.localize(start_time)
utc = pytz.timezone('utc')
start_time = start_time.astimezone(utc)
start_time = start_time.replace(tzinfo=None)

#log((channel, start_time))

conn = sqlite3.connect(xbmc.translatePath('special://profile/addon_data/plugin.video.iptv.archive.downloader/xmltv.db'), detect_types=sqlite3.PARSE_DECLTYPES|sqlite3.PARSE_COLNAMES)
cursor = conn.cursor()
try:
    #log(channel)
    channel_id = cursor.execute('SELECT tvg_id FROM streams WHERE name=?',(channel,)).fetchone()[0]
    if not channel_id:
        channel_id = cursor.execute('SELECT tvg_id FROM streams WHERE tvg_name=?',(channel,)).fetchone()[0]
    #log(channel_id)
    if channel_id:
        program_id = cursor.execute('SELECT uid FROM programmes WHERE channelid=? AND start=?',(channel_id,start_time)).fetchone()[0]
        #log((channel_id, program_id, start_time))
        if program_id:
            channel_encoded = channel.encode("utf8")
            channel_encoded = quote_plus(channel_encoded)
            xbmc.executebuiltin("ActivateWindow(videos,plugin://plugin.video.iptv.archive.downloader/broadcast/%s/%s,return)" % (program_id,channel_encoded))

except Exception as e:
    #log(e)
    xbmcgui.Dialog().notification("IPTV Recorder","program not found",xbmcgui.NOTIFICATION_WARNING)