using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Formats.Nrbf;
using System.IO;
using System.Linq;
using System.Net;
using System.Threading.Tasks;
using Avalonia.Threading;
using DownKyi.Core.BiliApi.BiliUtils;
using DownKyi.Core.BiliApi.Login;
using DownKyi.Core.Storage;
using DownKyi.Core.Storage.Database;
using DownKyi.Models;
using Prism.Commands;
using Prism.Services.Dialogs;
using SqlSugar;

namespace DownKyi.ViewModels.Dialogs;

public class ViewUpgradingDialogViewModel : BaseDialogViewModel
{
    public const string Tag = "DialogLoading";
    private readonly ISqlSugarClient _sqlSugarClient;

    #region 页面属性申明

    private double _percent;

    public double Percent
    {
        get => _percent;
        set => SetProperty(ref _percent, value);
    }

    private string? _message;

    public string? Message
    {
        get => _message;
        set => SetProperty(ref _message, value);
    }

    private bool _restartedVisible;

    public bool RestartVisible
    {
        get => _restartedVisible;
        set => SetProperty(ref _restartedVisible, value);
    }

    #endregion

    #region 命令申明

    private DelegateCommand? _restartCommand;

    public DelegateCommand RestartCommand => _restartCommand ??= new DelegateCommand(ExecuteRestart);

    private void ExecuteRestart()
    {
        var executablePath = Process.GetCurrentProcess().MainModule?.FileName;
        if (executablePath != null)
        {
            Process.Start(executablePath);
            App.Current.AppLife?.Shutdown();
        }
    }

    #endregion

    public ViewUpgradingDialogViewModel(ISqlSugarClient sqlSugarClient)
    {
        _sqlSugarClient = sqlSugarClient;
        Message = "数据迁移中、请不要关闭软件";
    }

    private void SetMessage(string message)
    {
        Dispatcher.UIThread.InvokeAsync(() => Message = message);
    }

    public override void OnDialogOpened(IDialogParameters parameters)
    {
        Upgrade();
    }

    private void Upgrade()
    {
        Task.Run(() => { Upgrade1_0_20To1_0_21(); });
    }

    private void Upgrade1_0_20To1_0_21()
    {
        var noMigrate = false;
        var loginInfoPath = StorageManager.GetLogin();
        if (File.Exists(loginInfoPath))
        {
            using Stream stream = File.Open(loginInfoPath, FileMode.Open);
            if (NrbfDecoder.StartsWithPayloadHeader(stream))
            {
                SetMessage("正在迁移登录信息");
                var cookies = new List<DownKyiCookie>();
                var cookieRecord = NrbfDecoder.DecodeClassRecord(stream);
                if (cookieRecord.TypeNameMatches(typeof(CookieContainer)))
                {
                    var domainTable = cookieRecord.GetClassRecord("m_domainTable");
                    var values = domainTable?.GetArrayRecord("Values");
                    var valuesArray = values?.GetArray(typeof(object[])).Cast<ClassRecord>();

                    foreach (var value in valuesArray ?? Array.Empty<ClassRecord>())
                    {
                        var valueObjects = value.GetClassRecord("m_list")?
                            .GetClassRecord("_list")?
                            .GetArrayRecord("values")?
                            .GetArray(typeof(object[]));
                        foreach (var valueObject in valueObjects?.Cast<ClassRecord>() ?? Array.Empty<ClassRecord>())
                        {
                            if (valueObject == null) continue;
                            foreach (var c in valueObject.GetClassRecord("m_list")?.GetArrayRecord("_items")
                                         ?.GetArray(typeof(object[])).Cast<ClassRecord>() ?? Array.Empty<ClassRecord>())
                            {
                                if (c == null) continue;
                                cookies.Add(new DownKyiCookie(c.GetString("m_name") ?? "", c.GetString("m_value") ?? "", c.GetString("m_domain")));
                            }
                        }
                    }
                }

                LoginHelper.SaveLoginInfoCookies(cookies);
                SetMessage("登录信息迁移完成");
            }
        }
        else
        {
            noMigrate = true;
        }


#if DEBUG
        var oldDbPath = StorageManager.GetDownload().Replace(".db", "_debug.db");
#else
        var oldDbPath = StorageManager.GetDownload();
#endif

        if (File.Exists(oldDbPath))
        {
            SetMessage("正在迁移下载信息");
#if DEBUG
            var dbHelper = new DbHelper(oldDbPath);
#else
            var dbHelper = new DbHelper(oldDbPath, "bdb8eb69-3698-4af9-b722-9312d0fba623");
#endif
            var objects = new Dictionary<string, object>();
            dbHelper.ExecuteQuery("select * from downloaded", reader =>
            {
                while (reader.Read())
                {
                    try
                    {
                        // 读取字节数组
                        var array = (byte[])reader["data"];
                        // 定义一个流
                        using var stream = new MemoryStream(array);
                        // 反序列化
                        var record = NrbfDecoder.DecodeClassRecord(stream);
                        if (!record.TypeNameMatches(typeof(Downloaded))) continue;
                        var obj = new Downloaded
                        {
                            MaxSpeedDisplay = record.GetString($"<{nameof(Downloaded.MaxSpeedDisplay)}>k__BackingField"),
                            FinishedTime = record.GetString($"<{nameof(Downloaded.FinishedTime)}>k__BackingField") ?? "",
                            FinishedTimestamp = record.GetInt64($"<{nameof(Downloaded.FinishedTimestamp)}>k__BackingField")
                        };
                        objects.Add((string)reader["id"], obj);
                    }
                    catch (Exception e)
                    {
                        SetMessage(e.Message);
                    }
                }
            });

            var readNeedDownloadContent = new Func<ClassRecord, Dictionary<string, bool>>(record =>
            {
                var keyArrayRecord = record.GetArrayRecord("KeyValuePairs");
                var keys = keyArrayRecord?.GetArray(typeof(KeyValuePair<string, bool>[]));
                var needDownloadContent = new Dictionary<string, bool>();
                foreach (var keyValuePairRecord in keys?.Cast<ClassRecord>() ?? Array.Empty<ClassRecord>())
                {
                    var key = keyValuePairRecord.GetString("key") ?? "";
                    var value = keyValuePairRecord.GetBoolean("value");
                    needDownloadContent.Add(key, value);
                }

                return needDownloadContent;
            });
            var readQuality = new Func<ClassRecord?, Quality>(record =>
            {
                if (record == null) return new Quality();
                var quality = new Quality
                {
                    Id = record.GetInt32($"<{nameof(Quality.Id)}>k__BackingField"),
                    Name = record.GetString($"<{nameof(Quality.Name)}>k__BackingField") ?? ""
                };

                return quality;
            });
            var i = 0;
            var downloadedList = new List<Downloaded>();
            foreach (var item in objects)
            {
                if (item.Value is Downloaded downloaded)
                {
                    dbHelper.ExecuteQuery($"select * from download_base where id = '{item.Key}'", reader =>
                    {
                        while (reader.Read())
                        {
                            var array = (byte[])reader["data"];
                            // 定义一个流
                            using var stream = new MemoryStream(array);
                            var record = NrbfDecoder.DecodeClassRecord(stream);
                            if (!record.TypeNameMatches(typeof(DownloadBase))) continue;
                            var needDownloadContentRecord = record.GetClassRecord($"<{nameof(DownloadBase.NeedDownloadContent)}>k__BackingField");
                            var needDownloadContent = new Dictionary<string, bool>();
                            if (needDownloadContentRecord != null)
                            {
                                needDownloadContent = readNeedDownloadContent(needDownloadContentRecord);
                            }

                            var download = new Downloaded
                            {
                                MaxSpeedDisplay = downloaded.MaxSpeedDisplay,
                                FinishedTime = downloaded.FinishedTime,
                                FinishedTimestamp = downloaded.FinishedTimestamp,
                                DownloadBase = new DownloadBase
                                {
                                    NeedDownloadContent = needDownloadContent,
                                    Bvid = record.GetString($"<{nameof(DownloadBase.Bvid)}>k__BackingField") ?? "",
                                    Avid = record.GetInt64($"<{nameof(DownloadBase.Avid)}>k__BackingField"),
                                    Cid = record.GetInt64($"<{nameof(DownloadBase.Cid)}>k__BackingField"),
                                    EpisodeId = record.GetInt64($"<{nameof(DownloadBase.EpisodeId)}>k__BackingField"),
                                    CoverUrl = record.GetString($"<{nameof(DownloadBase.CoverUrl)}>k__BackingField") ?? "",
                                    PageCoverUrl = record.GetString($"<{nameof(DownloadBase.PageCoverUrl)}>k__BackingField") ?? "",
                                    ZoneId = record.GetInt32($"<{nameof(DownloadBase.ZoneId)}>k__BackingField"),
                                    Order = record.GetInt32($"<{nameof(DownloadBase.Order)}>k__BackingField"),
                                    MainTitle = record.GetString($"<{nameof(DownloadBase.MainTitle)}>k__BackingField") ?? "",
                                    Name = record.GetString($"<{nameof(DownloadBase.Name)}>k__BackingField") ?? "",
                                    Duration = record.GetString($"<{nameof(DownloadBase.Duration)}>k__BackingField") ?? "",
                                    VideoCodecName = record.GetString($"<{nameof(DownloadBase.VideoCodecName)}>k__BackingField") ?? "",
                                    Resolution = readQuality(record.GetClassRecord($"<{nameof(DownloadBase.Resolution)}>k__BackingField")),
                                    AudioCodec = readQuality(record.GetClassRecord($"<{nameof(DownloadBase.AudioCodec)}>k__BackingField")),
                                    FilePath = record.GetString($"<{nameof(DownloadBase.FilePath)}>k__BackingField") ?? "",
                                    FileSize = record.GetString($"<{nameof(DownloadBase.FileSize)}>k__BackingField") ?? "",
                                    Page = record.GetInt32($"<{nameof(DownloadBase.Page)}>k__BackingField")
                                }
                            };
                            downloadedList.Add(download);
                            i++;
                            // 分批插入数据库，避免一次性插入过多数据导致性能问题，最后一批可能小于100条
                            if (i % 200 == 0 || i == objects.Count)
                            {
                                _sqlSugarClient.InsertNav(downloadedList).Include(o1 => o1.DownloadBase).ExecuteCommand();
                                downloadedList.Clear();
                                var percent = i / (double)objects.Count * 100;
                                var percentStr = $"{i}/{objects.Count}";
                                Dispatcher.UIThread.Invoke(() =>
                                {
                                    Percent = percent;
                                    SetMessage($"正在迁移下载信息({percentStr})");
                                });
                            }
                        }
                    });
                }
            }

            dbHelper.Close();
            File.Delete(oldDbPath);
            Dispatcher.UIThread.Invoke(() =>
            {
                RestartVisible = true;
                SetMessage("下载信息迁移完成");
            });
        }
        else
        {
            noMigrate = true;
        }

        if (noMigrate)
        {
            Dispatcher.UIThread.Invoke(() => RaiseRequestClose(new DialogResult()));
        }
    }
}