using System;

using System.Collections.Generic;

using System.Linq;

using System.Text;

using CsQuery;

using System.Diagnostics;

using System.Net;

namespace SatChanGen

{

    class KingOfSat

    {

        //

        // Summary:

        //     Create a new CQ object wrapping a single element.

        //

        // Parameters:

        //   aUrl:

        //     URL to a KingOfSat channel list. Typically a package list.

        //

        // Return:

        //   List of channels parsed.		

        public static List<Channel> Parse(List<Channel> aChannels, string aUrl, string aOrbitalPosition, bool aUseChannelIdForName=false, string aCategoryOverride="")

        {

            //To avoid duplicated name

            Dictionary<string, int> names = new Dictionary<string, int>();

            string kos = new WebClient().DownloadString(aUrl);

            //Debug.Write(kos);

            CQ dom = kos;

            //Get all the Frequency elements in our page

            CQ sats = dom[".frq"];

            //Create our list of channels

            List<Channel> channels = new List<Channel>();

            foreach (IDomObject frq in sats.ToList())

            {

                Channel common = new Channel();

                //Parse channel details

                common.OrbitalPosition = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td")?.Get(0).InnerText).Trim();

                if (common.OrbitalPosition != aOrbitalPosition)

                {

                    //Wrong sat, skip

                    continue;

                }

                common.Satellite = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(2) > a").Get(0).InnerText);

                common.Frequency = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(3)").Get(0).InnerText);

                common.Polarisation = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(4)").Get(0).InnerText);

                common.Transponder = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(5) > a").Get(0).InnerText);

                common.Beam = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(6) > a").Get(0).InnerText);

                common.Standard = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(7)").Get(0).InnerText);

                common.Modulation = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(8)").Get(0).InnerText);

                common.SymbolRate = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(9) > a").Get(0).InnerText);

                common.FEC = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(9) > a:nth-child(2)").Get(0).InnerText);

                try

                {

                    common.Provider = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(10) > b").Get(0).InnerText);

                }

                catch (Exception)

                {

                }

                common.Bitrate = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(10)").Get(0).InnerText);

                if (common.Bitrate.Substring(0, ", ".Length) == ", ")

                {

                    common.Bitrate = common.Bitrate.Substring(", ".Length, common.Bitrate.Length - ", ".Length);

                }

                //

                common.NetworkID = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(11)").Get(0).InnerText);

                //common.NetworkID = common.NetworkID.Substring("NID:".Length, common.NetworkID.Length - "NID:".Length);

                //

                common.TransponderID = WebUtility.HtmlDecode(frq.Cq().Find("tbody > tr > td:nth-child(12)").Get(0).InnerText);

                //common.TransponderID = common.TransponderID.Substring("TID:".Length, common.TransponderID.Length - "TID:".Length);

                //We got common properties for the coming channels

                //Debug.Write(common.ToString());

                //Now get all the channels for that frequency

                //Channel common = new Channel();

                CQ channelsDiv = frq.Cq().Next("div");

                CQ channelsTableRows = channelsDiv.Find("table.fl > tbody").Children("tr");

                foreach (IDomObject row in channelsTableRows)

                {

                    Channel channel = new Channel();

                    //Initialize this channel with common properties on this frequency

                    channel.Copy(common);

                    //Try and parse channel name

                    CQ cqChannelName = row.Cq().Find("td:nth-child(3) > a");

                    if (cqChannelName.Length == 0)

                    {

                        cqChannelName = row.Cq().Find("td:nth-child(3) > i");

                        if (cqChannelName.Length == 0)

                        {

                            //Can't get channel name

                            Debug.Write("WARNING: Can't find channel name! Skipping this channel");

                            continue;

                        }

                    }

                    string channelName = "";

                    if (cqChannelName.Get(0).HasAttribute("title") && aUseChannelIdForName)

                    {

                        //We want to use the channel ID

                        channelName = cqChannelName.Get(0).GetAttribute("title");

                    }

                    else

                    {

                        channelName = cqChannelName.Get(0).InnerText;

                    }

                    //Decode HTML

                    channel.Name = WebUtility.HtmlDecode(channelName);

                    //Convert from default encoding to UTF8

                    //We spend a lot of time trying to get this right until we found our answer in the following thread.

                    //http://stackoverflow.com/questions/14057434/how-can-i-transform-string-to-utf-8-in-c

                    byte[] bytes = Encoding.Default.GetBytes(channel.Name);

                    channel.Name = Encoding.UTF8.GetString(bytes);

                    if (channel.Name == "Name" || channel.Name == "Sorted by name")

                    {

                        //Skipping header rows

                        continue;

                    }

                    //Make sure our channel name looks descent

                    channel.Name = CleanChannelName(channel.Name);

                    //Make sure the resulting name is unique to avoid having multiple tuning detail for a single channel

                    if (names.ContainsKey(channel.Name))

                    {

                        names[channel.Name]++;

                        channel.Name += " " + names[channel.Name];

                    }

                    else

                    {

                        names.Add(channel.Name, 1);

                    }

                    //

                    //We don't want channels we already have

                    Channel existingChannel = aChannels.Find(c => c.Name == channel.Name);

                    if (existingChannel != null)

                    {

                        continue;

                    }

                    //So we have a channel name get the other properties then

                    channel.Country = WebUtility.HtmlDecode(row.Cq().Find("td:nth-child(4)").Get(0).InnerText).Trim();

                    channel.Category = WebUtility.HtmlDecode(row.Cq().Find("td:nth-child(5)").Get(0).InnerText).Trim();

                    if (channel.Category == "")

                    {

                        channel.Category = "Other";

                    }

                    //Override category if needed

                    if (aCategoryOverride != "")

                    {

                        channel.Category = aCategoryOverride;

                    }

                    //Skip the packages

                    //Skip the encryptions

                    channel.SID = WebUtility.HtmlDecode(row.Cq().Find("td:nth-child(8)").Get(0).InnerText).Trim();

                    channel.VPID = WebUtility.HtmlDecode(row.Cq().Find("td:nth-child(9)").Get(0).InnerText).Trim();

                    //Skip audios

                    channel.PMT = WebUtility.HtmlDecode(row.Cq().Find("td:nth-child(11)").Get(0).InnerText).Trim();

                    channel.PCR = WebUtility.HtmlDecode(row.Cq().Find("td:nth-child(11)").Get(0).InnerText).Trim();

                    channel.TXT = WebUtility.HtmlDecode(row.Cq().Find("td:nth-child(11)").Get(0).InnerText).Trim();

                    //Append that new channel to our list

                    channels.Add(channel);

                    //Show it in debug output

                    Debug.Write(channel);

                } //For each channel

            } //For each frequency

            return channels;

        }

        //

        public static string CleanChannelName(string aName)

        {

            aName = aName.Trim();

            string[] remove = { " Germany", " Deutschland", " (Germany)", " (Deutschland)" };

            foreach (string item in remove)

            {

                //if (aName.EndsWith(item))

                if (aName.Contains(item))

                {

                    aName = aName.Substring(0, aName.LastIndexOf(item));

                    break; //only allow one match at most

                }

            }

            string[] removePrefix = { "Id: " };

            foreach (string item in removePrefix)

            {

                if (aName.StartsWith(item))

                {

                    aName = aName.Substring(item.Length, aName.Length - item.Length);

                    break; //only allow one match at most

                }

            }

            aName = aName.Trim();

            return aName;

        }

        //

        public static List<Channel> CleanChannelList(List<Channel> aChannels)

        {

            //Create our list of channels

            List<Channel> channels = new List<Channel>();

            foreach (Channel channel in aChannels)

            {

                Channel hdChannel = aChannels.Find(c => c.Name == channel.Name + " HD");

                if (hdChannel == null

                    && !(channel.Name.Contains("Bundesliga") && !channel.Name.Contains("HD")) //We don't want non HD bundesliga

                    && !(channel.Name.StartsWith("Sky Sport") && !channel.Name.Contains("HD")) //We don't want non HD Sky Sport

                    )

                {

                    //Patch some missing or bad categories

                    if (channel.Name.Contains("Bundesliga")

                        || channel.Name.Contains("Sport"))

                    {

                        channel.Category = "Sport";

                    }

                    if (channel.Name.Contains("Sky Select"))

                    {

                        channel.Category = "Pay per view";

                    }

                    if (channel.Name.StartsWith("Sky Atlantic")

                        || channel.Name.StartsWith("SyFy")

                        || channel.Name.StartsWith("Fox"))

                    {

                        channel.Category = "Series";

                    }

                    if (channel.Name.StartsWith("Sky 3D"))

                    {

                        channel.Category = "Movies";

                    }

                    //Collapse some categories

                    if (channel.Category == "Entertainment"

                        || channel.Category == "Music"

                        || channel.Name.Contains("Music"))

                    {

                        channel.Category = "General";

                    }

                    if (channel.Category == "Porn")

                    {

                        channel.Category = "Erotic";

                    }

                    if (channel.Category == "Presentations")

                    {

                        channel.Category = "News";

                    }

                    if (channel.Category == "History")

                    {

                        channel.Category = "Documentaries";

                    }

                    if (channel.Category == "Travel")

                    {

                        channel.Category = "Documentaries";

                    }

                    if (channel.Category == "Lifestyle")

                    {

                        channel.Category = "General";

                    }

                    //if (channel.Category == "Regional")

                    //{

                    //    channel.Category = "General";

                    //}

                    if (channel.Category == "Other")

                    {

                        channel.Category = "General";

                    }

                    if (channel.Category == "Cultural")

                    {

                        channel.Category = "General";

                    }

                    if (channel.Category == "Movies")

                    {

                        channel.Category = "Movies & Series";

                    }

                    if (channel.Category == "Series")

                    {

                        channel.Category = "Movies & Series";

                    }

                    //No corresponding HD channel, keep it then

                    channels.Add(channel);

                }

                else

                {

                    Debug.Write("WARNING: Found HD channel for " + channel.Name + ". Discarding it!\n");

                }

            }

            return channels;

        }

    }

}