using Microsoft.VisualBasic;
using MongoDB.Driver;
using MongoDB.Driver.Core.Servers;
using NPOI.SS.UserModel;
using NPOI.XSSF.UserModel;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Security.Cryptography;
using System.Text;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Media;
using System.Windows.Media.Imaging;
using System.Windows.Threading;

namespace Ramitta.lib
{
    public static class Basic
    {
        #region 全局颜色
        public static SolidColorBrush 经典紫色 = new SolidColorBrush(Color.FromRgb(92, 40, 147));
        public static SolidColorBrush 错误红色 = new SolidColorBrush(Color.FromRgb(134, 27, 45));
        public static SolidColorBrush 警告橙色 = new SolidColorBrush(Color.FromRgb(202, 81, 0));
        public static SolidColorBrush 正常绿色 = new SolidColorBrush(Color.FromRgb(66, 164, 60));
        public static SolidColorBrush 通知蓝色 = new SolidColorBrush(Color.FromRgb(0, 122, 204));
        public static SolidColorBrush 背景黑色 = new SolidColorBrush(Color.FromRgb(30, 30, 30));
        public static SolidColorBrush 默认灰色 = new SolidColorBrush(Color.FromRgb(66, 66, 66));
        #endregion

        #region 调试函数
        public static void DebugBar(Label obj, String? text, SolidColorBrush? color=null)
        {
            if (obj == null) return;
            // 通过Dispatcher确保在UI线程执行
            obj.Dispatcher.Invoke(() =>
            {
                if (text!=null)obj.Content = text;
                if (color != null) obj.Background = color;


                // 强制刷新UI
                obj.InvalidateVisual();
                obj.InvalidateArrange();
                obj.InvalidateMeasure();
                obj.UpdateLayout();
                // 处理Dispatcher队列中的其他挂起的操作，确保UI更新
                Dispatcher.CurrentDispatcher.Invoke(DispatcherPriority.Background, new Action(delegate { }));
            });
        }
        public static SolidColorBrush GenerateRandomColor(SolidColorBrush color = null)
        {
            // 如果 color 参数为 null，生成一个新的随机颜色
            if (color == null)
            {
                Random random = new Random();
                byte r = (byte)random.Next(0, 256);  // 红色分量 0 - 255
                byte g = (byte)random.Next(0, 256);  // 绿色分量 0 - 255
                byte b = (byte)random.Next(0, 256);  // 蓝色分量 0 - 255

                color = new SolidColorBrush(Color.FromRgb(r, g, b));
            }

            return color;
        }

        // 定义自定义异常类
        public class DebugbarException : Exception
        {
            public string Message { get; }
            public SolidColorBrush Brush { get; }

            public DebugbarException(string message, SolidColorBrush brush)
                : base(message)
            {
                Message = message;
                Brush = brush;
            }
        }
        #endregion

        #region 参数解析
        public static Dictionary<string, string>? Startupe;
        public static Dictionary<string, string> ParseCommandLineArgs(string[] args)
        {
            var arguments = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);

            for (int i = 0; i < args.Length; i++)
            {
                string arg = args[i];

                // 检查是否是参数名（以-或/开头）
                if (arg.StartsWith("-") || arg.StartsWith("/"))
                {
                    string key = arg.TrimStart('-', '/');
                    string value = "true"; // 默认值为true，表示开关存在

                    // 检查下一个参数是否是值（不以-或/开头）
                    if (i + 1 < args.Length && !args[i + 1].StartsWith("-") && !args[i + 1].StartsWith("/"))
                    {
                        value = args[i + 1];
                        i++; // 跳过下一个参数，因为已经被用作值
                    }

                    arguments[key] = value;
                }
            }

            return arguments;
        }
        #endregion

        #region 文件解析
        public static bool IsPathExist(string databasePath)
        {
            if (string.IsNullOrWhiteSpace(databasePath))
            {
                return false;
            }

            return Directory.Exists(databasePath);
        }
        public static void FileWrite(string outputPath, string outputText, bool createDirectories = true,bool fileAdd = false)
        {
            string directory = Path.GetDirectoryName(outputPath);
            if (createDirectories && !Directory.Exists(directory))
            {
                Directory.CreateDirectory(directory);
            }

            // 使用 StreamWriter 写入文件
            using (StreamWriter writer = new StreamWriter(outputPath, false))
            {
                Debug.WriteLine(outputText);
                writer.WriteLine(outputText);
            }
        }
        public static string FileRead(string outputPath)
        {
            try
            {
                return File.ReadAllText(outputPath); 
            }
            catch
            {
                throw;
            }
        }

        // 获取路径下所有文件和文件夹，支持传入判断文件的函数
        public static List<string> GetFilesAndDirectories(string path, bool? isDirectory = null, Func<string, bool> fileFilter = null, Func<string, string> outputFormat = null)
        {
            List<string> result = new List<string>();

            // 获取文件夹
            var directories = Directory.GetDirectories(path);
            foreach (var dir in directories)
            {
                if (isDirectory == null || isDirectory == true)
                {
                    // 如果指定了输出格式函数，则使用格式化函数；否则使用原始路径
                    result.Add(outputFormat == null ? dir : outputFormat(dir));
                }
            }

            // 获取文件
            var files = Directory.GetFiles(path);
            foreach (var file in files)
            {
                // 如果传入了 fileFilter 函数，调用该函数进行筛选
                if (fileFilter == null || fileFilter(file))
                {
                    if (isDirectory == null || isDirectory == false)
                    {
                        // 如果指定了输出格式函数，则使用格式化函数；否则使用原始路径
                        result.Add(outputFormat == null ? file : outputFormat(file));
                    }
                }
            }

            return result;
        }


        //  生成截图
        public static PngBitmapEncoder CaptureWindowScreenshot(Window window, string? filePath = null)
        {
            // 获取当前窗口的 DPI 缩放因子
            var dpiScale = VisualTreeHelper.GetDpi(window);

            // 获取 DPI 缩放因子的 X 和 Y 值，通常 X 和 Y 应该是相等的
            double dpiX = dpiScale.DpiScaleX;
            double dpiY = dpiScale.DpiScaleY;

            // 根据 DPI 缩放因子调整截图的尺寸
            int width = (int)(window.ActualWidth * dpiX);   // 使用 dpiX
            int height = (int)(window.ActualHeight * dpiY); // 使用 dpiY

            // 创建一个 RenderTargetBitmap 对象，并且考虑 DPI 缩放
            RenderTargetBitmap rtb = new RenderTargetBitmap(width, height, 96 * dpiX, 96 * dpiY, System.Windows.Media.PixelFormats.Pbgra32);

            // 使用 VisualTreeHelper 进行渲染
            rtb.Render(window);

            // 创建一个 PngBitmapEncoder 来保存为 PNG 格式
            PngBitmapEncoder pngEncoder = new PngBitmapEncoder();
            pngEncoder.Frames.Add(BitmapFrame.Create(rtb));

            if (!string.IsNullOrEmpty(filePath))
            {
                // 保存图像到文件
                using (FileStream fs = new FileStream(filePath, FileMode.Create))
                {
                    pngEncoder.Save(fs);
                }
            }
            return pngEncoder;
        }
        #endregion

        #region 任务操作
        public static async Task<int> RunExternalCommand(string? applicationPath = "explorer.exe", string arguments="", bool UseShellExecute = false, bool CreateNoWindow = false)
        {
            ProcessStartInfo startInfo = new ProcessStartInfo
            {
                FileName = applicationPath,
                Arguments = arguments,
                UseShellExecute = UseShellExecute,
                CreateNoWindow = CreateNoWindow
            };

            try
            {
                Process process = Process.Start(startInfo) ??
                    throw new InvalidOperationException("Failed to start process.");

                if (UseShellExecute != true)
                {
                    await Task.Run(() => process.WaitForExit());
                    return process.ExitCode; // 返回进程的退出代码
                }
                else
                {
                    return 0;
                }

            }

            catch (Exception ex)
            {
                throw;
            }
        }

        public static async Task<string> RunExternalCommandResult(string applicationPath, string arguments, bool CreateNoWindow = true)
        {
            // 创建一个新的进程启动信息
            ProcessStartInfo startInfo = new ProcessStartInfo
            {
                FileName = applicationPath,
                Arguments = arguments,
                UseShellExecute = false,  // 必须设置为 false 以便能够重定向输入输出
                RedirectStandardOutput = true,  // 重定向标准输出
                RedirectStandardError = true,    // 可选：也重定向标准错误流
                CreateNoWindow = CreateNoWindow            // 不创建窗口
            };

            try
            {
                // 启动进程
                using (Process process = new Process())
                {
                    process.StartInfo = startInfo;

                    // 启动进程
                    process.Start();

                    // 读取输出流
                    string output = await process.StandardOutput.ReadToEndAsync();
                    string error = await process.StandardError.ReadToEndAsync(); // 读取标准错误流

                    // 等待进程结束
                    await process.WaitForExitAsync();
                    /*
                    if (process.ExitCode != 0)
                    {
                        // 处理非零退出代码的情况
                        throw new Exception($"Process exited with code {process.ExitCode}: {error}");
                    }
                    */

                    return output; // 返回标准输出内容
                }
            }
            catch (Exception ex)
            {
                // 可以记录或处理异常
                throw; // 重新抛出异常以便调用者捕获
            }
        }
        #endregion

        // 通过 UDP 发送消息
        public static void SendUdpMessage(string _serverIp, int _serverPort,string message)
        {
            try
            {
                using (UdpClient udpClient = new UdpClient())
                {
                    IPEndPoint endPoint = new IPEndPoint(IPAddress.Parse(_serverIp), _serverPort);
                    byte[] messageBytes = Encoding.UTF8.GetBytes(message);
                    udpClient.Send(messageBytes, messageBytes.Length, endPoint);
                }
            }
            catch (Exception ex)
            {
                Debug.WriteLine($"发送 UDP 消息失败: {ex.Message}");
            }
        }

        // 通过 TCP 发送消息
        public static void SendTcpMessage(string _serverIp, int _serverPort, string message)
        {
            try
            {
                using (TcpClient tcpClient = new TcpClient(_serverIp, _serverPort))
                using (NetworkStream networkStream = tcpClient.GetStream())
                {
                    byte[] messageBytes = Encoding.UTF8.GetBytes(message);
                    networkStream.Write(messageBytes, 0, messageBytes.Length);
                    networkStream.Flush();
                }
            }
            catch (Exception ex)
            {
                Debug.WriteLine($"发送 TCP 消息失败: {ex.Message}");
            }
        }

        //  生成哈希
        public static string GenerateHash(string input)
        {
            // 创建SHA256对象
            using (SHA256 sha256 = SHA256.Create())
            {
                // 将输入字符串转换为字节数组并计算哈希值
                byte[] bytes = sha256.ComputeHash(Encoding.UTF8.GetBytes(input));

                // 将哈希值转换为十六进制字符串
                StringBuilder stringBuilder = new StringBuilder();
                foreach (byte b in bytes)
                {
                    stringBuilder.Append(b.ToString("x2"));
                }

                // 返回最终的哈希字符串
                return stringBuilder.ToString();
            }
        }
    }
    public class CryptoHelper
    {
        /// <summary>
        /// 生成RSA密钥对
        /// </summary>
        /// <param name="keySize">密钥长度（通常为2048、4096）</param>
        /// <returns>包含公钥和私钥的元组</returns>
        public static (string publicKey, string privateKey) GenerateKeyPair(int keySize = 2048)
        {
            try
            {
                using (var rsa = RSA.Create(keySize))
                {
                    string publicKey = rsa.ToXmlString(false);
                    string privateKey = rsa.ToXmlString(true);

                    return (publicKey, privateKey);
                }
            }
            catch (Exception ex)
            {
                MessageBox.Show($"密钥生成失败: {ex.Message}");
                return (null, null);
            }
        }

        /// <summary>
        /// 使用私钥对数据进行签名
        /// </summary>
        public static string? SignData(string data, string privateKey)
        {
            try
            {
                using (var rsa = RSA.Create())
                {
                    rsa.FromXmlString(privateKey);

                    byte[] dataBytes = Encoding.UTF8.GetBytes(data);
                    byte[] signature = rsa.SignData(dataBytes, HashAlgorithmName.SHA256, RSASignaturePadding.Pkcs1);

                    return Convert.ToBase64String(signature);
                }
            }
            catch (Exception ex)
            {
                return null;
            }
        }

        /// <summary>
        /// 使用公钥验证签名
        /// </summary>
        public static bool? VerifySignature(string data, string signature, string publicKey)
        {
            try
            {
                using (var rsa = RSA.Create())
                {
                    rsa.FromXmlString(publicKey);

                    byte[] dataBytes = Encoding.UTF8.GetBytes(data);
                    byte[] signatureBytes = Convert.FromBase64String(signature);

                    return rsa.VerifyData(dataBytes, signatureBytes, HashAlgorithmName.SHA256, RSASignaturePadding.Pkcs1);
                }
            }
            catch (Exception ex)
            {
                return false;
            }
        }

        /// <summary>
        /// 使用公钥加密数据
        /// </summary>
        public static string? Encrypt(string data, string publicKey)
        {
            try
            {
                using (var rsa = RSA.Create())
                {
                    rsa.FromXmlString(publicKey);

                    byte[] dataBytes = Encoding.UTF8.GetBytes(data);
                    byte[] encryptedBytes = rsa.Encrypt(dataBytes, RSAEncryptionPadding.OaepSHA256);

                    return Convert.ToBase64String(encryptedBytes);
                }
            }
            catch (Exception ex)
            {
                return null;
            }
        }

        /// <summary>
        /// 使用私钥解密数据
        /// </summary>
        public static string? Decrypt(string encryptedData, string privateKey)
        {
            try
            {
                using (var rsa = RSA.Create())
                {
                    rsa.FromXmlString(privateKey);

                    byte[] encryptedBytes = Convert.FromBase64String(encryptedData);
                    byte[] decryptedBytes = rsa.Decrypt(encryptedBytes, RSAEncryptionPadding.OaepSHA256);

                    return Encoding.UTF8.GetString(decryptedBytes);
                }
            }
            catch (Exception ex)
            {
                return null;
            }
        }
    }

}