using DataDomain;
using Helpers;
using OpenQA.Selenium;
using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace BrowserHelper.Extensions
{
    public static class HtmlTableExtension
    {
        //Read of Table
        private static List<TableDataCollection> ReadTable(IWebElement table)
        {
            var tableDataCollection = new List<TableDataCollection>();
            var columns = table.FindElements(By.TagName("th"));
            var rows = table.FindElements(By.TagName("tr"));
            int rowIndex = 0;
            foreach (var row in rows)
            {
                int colIndex = 0;
                var colDatas = row.FindElements(By.TagName("td"));
                if (colDatas.Count != 0)
                    foreach (var colValue in colDatas)
                    {
                        tableDataCollection.Add(new TableDataCollection
                        {
                            RowNumber = rowIndex,
                            ColumnName = columns[colIndex].Text != "" ?
                            columns[colIndex].Text : colIndex.ToString(),
                            ColumnValue = colValue.Text,
                            ColumnSpecialValue = GetControl(colValue)
                        });
                        colIndex++;
                    }
                rowIndex++;
            }
            return tableDataCollection;
        }

        private static List<TableDataCollection> ReadTablePartly(IWebElement table, string[] shares, string[] shareHeaders)
        {
            var tableDataCollection = new List<TableDataCollection>();
            var columns = table.FindElements(By.TagName("th"));
            var rows = table.FindElements(By.TagName("tr"));
            var antal = shares.Length;
            int rowIndex = 0;
            foreach (var row in rows)
            {
                int colIndex = 0;
                var colDatas = row.FindElements(By.TagName("td"));
                if (colDatas.Count != 0 && shares.Contains(colDatas[0].Text))
                {
                    if (colDatas.Count != 0)
                        foreach (var colValue in colDatas)
                        {
                            if (shareHeaders.Contains(columns[colIndex].Text))
                            {
                                tableDataCollection.Add(new TableDataCollection
                                {
                                    RowNumber = rowIndex,
                                    ColumnName = columns[colIndex].Text != "" ?
                                    columns[colIndex].Text : colIndex.ToString(),
                                    ColumnValue = colValue.Text,
                                    //ColumnSpecialValue = GetControl(colValue)
                                });
                            }
                            colIndex++;
                        }
                    antal--;
                }

                rowIndex++;
                if (antal < 1) break;
            }
            return tableDataCollection;
        }

        public static List<DiTraderStockRow> SimpleReadTablePartly(this IWebElement table, string[] shares, string[] shareHeaders)
        {
            var shareCollection = new List<DiTraderStockRow>();
            var columns = table.FindElements(By.TagName("th"));
            IWebElement tblBody = table.FindElement(By.XPath(".//tbody"));
            var rows = tblBody.FindElements(By.TagName("tr"));
            Dictionary<string, int> columnPos = new Dictionary<string, int>();
            GlobalStopwatch.PrintSecs("After allocations");
            for (int i = 0; i < columns.Count; i++)
            {
                if (shareHeaders.Contains(columns[i].Text)) columnPos.Add(columns[i].Text, i);
            }
            var antal = shares.Length;
            int rowIndex = 0;
            GlobalStopwatch.PrintSecs("Before scraplooping");
            foreach (var row in rows)
            {
                var element = row.FindElement(By.XPath(".//td"));
                if (!shares.Contains(element.Text)) { }
                else
                {
                    var colDatas = row.FindElements(By.TagName("td"));
                    if (colDatas.Count != 0 && shares.Contains(colDatas[0].Text))
                    {
                        shareCollection.Add(new DiTraderStockRow
                        {
                            StockName = colDatas[columnPos["Aktie"]].Text,
                            LatestPrice = decimal.Parse(colDatas[columnPos["Senast"]].Text),
                            TimeOfDay = TimeSpan.Parse(colDatas[columnPos["Tid"]].Text)
                        });

                        rowIndex++;
                        antal--;
                    }
                }
                if (antal < 1) break;
            }
            GlobalStopwatch.PrintSecs("After scraplooping");
            return shareCollection;
        }

        private static ColumnSpecialValue GetControl(IWebElement columnValue)
        {
            ColumnSpecialValue? columnSpecialValue = null;
            if (columnValue.FindElements(By.TagName("a")).Count > 0)
            {
                columnSpecialValue = new ColumnSpecialValue
                {
                    ElementCollection = columnValue.FindElements(By.TagName("a")),
                    ControlType = ControlType.hyperlink
                };
            }
            if (columnValue.FindElements(By.TagName("input")).Count > 0)
            {
                columnSpecialValue = new ColumnSpecialValue
                {
                    ElementCollection = columnValue.FindElements(By.TagName("input")),
                    ControlType = ControlType.input
                };
            }
            return columnSpecialValue;
        }

        public static List<TableDataCollection> ReadHandledStock(this IWebElement table)
        {
            var tableDataCollection = new List<TableDataCollection>();
            var columns = table.FindElements(By.TagName("th"));
            var rows = table.FindElements(By.TagName("tr"));
            var antal = rows.Count;
            int rowIndex = 0;
            foreach (var row in rows)
            {
                var colDatas = row.FindElements(By.TagName("td"));
                if (colDatas.Count != 0)
                {
                    if (colDatas.Count != 0)
                        tableDataCollection.Add(new TableDataCollection
                        {
                            RowNumber = rowIndex,
                            ColumnName = columns[0].Text,
                            ColumnValue = colDatas[0].Text,
                            ColumnSpecialValue = null
                        });
                }
                rowIndex++;
                antal--;
                if (antal < 0) break;
            }

            return tableDataCollection;
        }


        //public static List<DiTraderStockRow> GetCertainStocks(this IWebElement element, string[] shares, string[] shareHeaders)
        //{
        //    List<DiTraderStockRow> shareList = new List<DiTraderStockRow>();
        //    var reducedList = ReadTablePartly(element, shares, shareHeaders);
        //    int rNr = -1;
        //    DiTraderStockRow Aktie = null;
        //    foreach (var shareRow in reducedList)
        //    {
        //        if (shareRow.RowNumber != rNr)
        //        {
        //            if (Aktie != null)
        //            {
        //                shareList.Add(Aktie);
        //            }
        //            Aktie = new();
        //            rNr = shareRow.RowNumber;
        //        }
        //        switch (shareRow.ColumnName)
        //        {
        //            case "Aktie":
        //                {
        //                    Aktie.StockName = shareRow.ColumnValue;
        //                    break;
        //                }
        //            case "Senast":
        //                {
        //                    Aktie.LatestPrice = Convert.ToDecimal(shareRow.ColumnValue);
        //                    break;
        //                }
        //            case "Tid":
        //                {
        //                    Aktie.TimeOfDay = TimeSpan.Parse(shareRow.ColumnValue);
        //                    break;
        //                }
        //        }
        //    }
        //    shareList.Add(Aktie);
        //    return shareList;
        //}


        public static void PerformActionOnCell(this IWebElement element, string targetColumnIndex, string refColumnName, string refColumnValue, string controlToOperate = null)
        {
            var table = ReadTable(element);

            foreach (int rowNumber in GetDynamicRowNumber(table, refColumnName, refColumnValue))
            {
                var cell = (from e in table
                            where e.ColumnName == targetColumnIndex && e.RowNumber == rowNumber
                            select e.ColumnSpecialValue).SingleOrDefault();

                if (controlToOperate != null && cell != null)
                {
                    IWebElement elementToClick = null;

                    if (cell.ControlType == ControlType.hyperlink)
                    {
                        elementToClick = (from c in cell.ElementCollection
                                          where c.Text == controlToOperate.ToString()
                                          select c).SingleOrDefault();
                    }
                    if (cell.ControlType == ControlType.input)
                    {
                        elementToClick = (from c in cell.ElementCollection
                                          where c.GetAttribute("value") == controlToOperate.ToString()
                                          select c).SingleOrDefault();

                    }
                    elementToClick?.Click();
                }
                else
                {
                    cell.ElementCollection?.First().Click();
                }
            }

        }

        private static IEnumerable GetDynamicRowNumber(List<TableDataCollection> tableCollection, string columnName, string columnValue)
        {
            foreach (var table in tableCollection)
            {
                if (table.ColumnName == columnName && table.ColumnValue == columnValue)
                    yield return table.RowNumber;
            }
        }
    }

    public class TableDataCollection
    {
        public int RowNumber { get; set; }
        public string? ColumnName { get; set; }
        public string? ColumnValue { get; set; }
        public ColumnSpecialValue? ColumnSpecialValue { get; set; }
    }

    public class ColumnSpecialValue
    {
        public IEnumerable<IWebElement>? ElementCollection { get; set; }
        public ControlType? ControlType { get; set; }
    }

    public enum ControlType
    {
        hyperlink,
        input,
        option,
        select
    }

    public class Share
    {
        public string AktieNamn { get; set; }
        public decimal SenastePris { get; set; }
        public TimeOnly RegTime { get; set; }
    }
}