AISDK/AIDiagnosisSystem/License/AILicenseGenerator/MainWindow.xaml.cs

using AILicenseGenerator.Helper;
using AILicenseGenerator.Models;
using Microsoft.Win32;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text.Json;
using System.Windows;
using Vinno.AI.CommonSDK.Enums;

namespace AILicenseGenerator
{
    /// <summary>
    /// MainWindow.xaml 的交互逻辑
    /// </summary>
    public partial class MainWindow : Window
    {
        private const string _logFolderName = "AILicenseGeneratorLogs";
        private const string _generateFolderName = "GenerateFolder";
        private const string _functionSettingFile = "FunctionInfo.ini";
        private const string _licenseFileNameWithWildcard = "*****License.lic";
        private readonly string _version = Assembly.GetExecutingAssembly().GetName().Version.ToString();
        private readonly string _generateFolderPath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, _generateFolderName);
        private List<AIFunctionInfoDTO> _enabledFunctionList = new List<AIFunctionInfoDTO>();
        private List<AIFunctionInfoDTO> _functionList = new List<AIFunctionInfoDTO>();
        private List<AIFunctionInfoDTO> _functionListForAllEnginesVersion = new List<AIFunctionInfoDTO>();

        public MainWindow()
        {
            InitializeComponent();
            Title = "AILicenseGenerator_" + _version;
            Logger.LogDir = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, _logFolderName);
            TranslateHelper.Init(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "Resources", "Chinese.json"));
            FunctionInit();
            AllFunctionListForAllEnginesVersionInit();
            if (AllFunctionEnableCheckBox.IsChecked == false)
            {
                FunctionDataGrid.ItemsSource = _functionList;
            }
            else
            {
                FunctionDataGrid.ItemsSource = _functionListForAllEnginesVersion;
            }
            this.DataContext = this;
        }

        private void FunctionInit()
        {
            var functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AIDiagnosisSDK.AIDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(MyocardialDiagnosisSDK.MyocardialDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(ThyroidClassificationSDK.ThyroidClassification.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AutoEFDiagnosisSDK.AutoEFDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AutoIVCDiagnosisSDK.AutoIVCDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AutoVTIDiagnosisSDK.AutoVTIDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(HepatoRenalRatioDiagnosisSDK.HepatoRenalRatioDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(VetHeartDiagnosisSDK.VetHeartDiagnosis.AllFunctions);
            if(functions!=null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(CarotidClassificationSDK.CarotidClassification.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AmnioticFluidDiagnosisSDK.AmnioticFluidDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AutoDFRDiagnosisSDK.AutoDFRDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(NerveIdentificationSDK.NerveIdentification.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AutoBLineDiagnosisSDK.AutoBLineDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(FatLayerIdentificationSDK.FatLayerIdentification.AllFunctions);
            if (functions != null)
            {
                _functionList.AddRange(functions);
            }
        }

        private void AllFunctionListForAllEnginesVersionInit()
        {
            var functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AIDiagnosisSDK.AIDiagnosis.AllFunctions);
            if(functions!=null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(MyocardialDiagnosisSDK.MyocardialDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(ThyroidClassificationSDK.ThyroidClassification.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AutoEFDiagnosisSDK.AutoEFDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AutoIVCDiagnosisSDK.AutoIVCDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AutoVTIDiagnosisSDK.AutoVTIDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(HepatoRenalRatioDiagnosisSDK.HepatoRenalRatioDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(VetHeartDiagnosisSDK.VetHeartDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(CarotidClassificationSDK.CarotidClassification.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AmnioticFluidDiagnosisSDK.AmnioticFluidDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AutoDFRDiagnosisSDK.AutoDFRDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(NerveIdentificationSDK.NerveIdentification.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(AutoBLineDiagnosisSDK.AutoBLineDiagnosis.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            functions = AIConvertHelper.ConvertListFunctionInfoToListAIFunctionInfoDTO(FatLayerIdentificationSDK.FatLayerIdentification.AllFunctions);
            if (functions != null)
            {
                _functionListForAllEnginesVersion.AddRange(functions);
            }
            foreach (var function in _functionListForAllEnginesVersion)
            {
                function.Version = "All";
                foreach (var enigne in function.Engines)
                {
                    enigne.Version = "All";
                }
            }
        }

        private void FunctionSettingButton_Click(object sender, RoutedEventArgs e)
        {
            try
            {
                PreGeneration();
                var jsonString = JsonSerializer.Serialize(_enabledFunctionList, new JsonSerializerOptions
                {
                    WriteIndented = true,
                });
                var filePath = Path.Combine(_generateFolderPath, _functionSettingFile);
                File.WriteAllText(filePath, jsonString);
                PostGeneration(_generateFolderPath);
                MessageBox.Show($"配置文件生成成功", "提示");
            }
            catch (Exception ex)
            {
                MessageBox.Show($"配置文件生成出错，{ex}", "提示");
            }
        }

        private void PreGeneration()
        {
            DirectoryHelper.DeleteDirectory(_generateFolderPath);
            DirectoryHelper.CreateDirectory(_generateFolderPath);
            GetEnabledFunctions();
        }

        private void GetEnabledFunctions()
        {
            _enabledFunctionList.Clear();
            if (AllFunctionEnableCheckBox.IsChecked == true)
            {
                var enableFunctions = _functionListForAllEnginesVersion.Where(x => x.IsEnabled).ToList();
                foreach (var function in enableFunctions)
                {
                    var enabledEngines = function.Engines.Where(x => x.IsEnabled).ToList();
                    _enabledFunctionList.Add(new AIFunctionInfoDTO(function.FunctionName, function.IsEnabled, function.Version, function.ExpiredTime, enabledEngines));
                }
            }
            else
            {
                var enableFunctions = _functionList.Where(x => x.IsEnabled).ToList();
                foreach (var function in enableFunctions)
                {
                    var enabledEngines = function.Engines.Where(x => x.IsEnabled).ToList();
                    _enabledFunctionList.Add(new AIFunctionInfoDTO(function.FunctionName, function.IsEnabled, function.Version, function.ExpiredTime, enabledEngines));
                }
            }
        }

        private void PostGeneration(string path)
        {
            using (Process process = new Process())
            {
                ProcessStartInfo psi = new ProcessStartInfo("Explorer.exe");
                psi.Arguments = path;
                process.StartInfo = psi;
                process.Start();
            }
        }

        private void LicenseButton_Click(object sender, RoutedEventArgs e)
        {
            try
            {
                PreGeneration();
                if (_enabledFunctionList != null)
                {
                    foreach (var function in _enabledFunctionList)
                    {
                        if (Enum.TryParse(function.FunctionName, out AIEnumType type))
                        {
                            var licenseFileName = _licenseFileNameWithWildcard.Replace("*****", type.ToString());
                            var jsonString = JsonSerializer.Serialize(function, new JsonSerializerOptions
                            {
                                WriteIndented = true,
                            });
                            var desString = DesBuilder.Encrypt(jsonString);
                            var filePath = Path.Combine(_generateFolderPath, licenseFileName);
                            File.WriteAllText(filePath, desString);
                        }
                    }
                }
                PostGeneration(_generateFolderPath);
                MessageBox.Show($"License生成成功", "提示");
            }
            catch (Exception ex)
            {
                MessageBox.Show($"License生成出错，{ex}", "提示");
            }
        }

        private void GenerateButton_Click(object sender, RoutedEventArgs e)
        {
            try
            {
                PreGeneration();
                var jsonString = JsonSerializer.Serialize(_enabledFunctionList, new JsonSerializerOptions
                {
                    WriteIndented = true,
                });
                var filePath = Path.Combine(_generateFolderPath, _functionSettingFile);
                File.WriteAllText(filePath, jsonString);
                if (_enabledFunctionList != null)
                {
                    foreach (var function in _enabledFunctionList)
                    {
                        if (Enum.TryParse(function.FunctionName, out AIEnumType type))
                        {
                            var licenseFileName = _licenseFileNameWithWildcard.Replace("*****", type.ToString());
                            jsonString = JsonSerializer.Serialize(function, new JsonSerializerOptions
                            {
                                WriteIndented = true,
                            });
                            var desString = DesBuilder.Encrypt(jsonString);
                            filePath = Path.Combine(_generateFolderPath, licenseFileName);
                            File.WriteAllText(filePath, desString);
                        }
                    }
                }
                PostGeneration(_generateFolderPath);
                MessageBox.Show($"全部生成成功", "提示");
            }
            catch (Exception ex)
            {
                MessageBox.Show($"全部生成出错，{ex}", "提示");
            }
        }

        private void AllFunctionEnable_Click(object sender, RoutedEventArgs e)
        {
            if (AllFunctionEnableCheckBox.IsChecked == false)
            {
                FunctionDataGrid.ItemsSource = _functionList;
            }
            else
            {
                FunctionDataGrid.ItemsSource = _functionListForAllEnginesVersion;
            }
        }

        private void LicenseDecryptButton_Click(object sender, RoutedEventArgs e)
        {
            OpenFileDialog openFileDialog = new OpenFileDialog
            {
                Filter = "License文件|*.lic",
                Multiselect = true,
                Title = "选择License文件"
            };
            if (openFileDialog.ShowDialog() ?? false)
            {
                var files = openFileDialog.FileNames;
                if (files != null && files.Length > 0)
                {
                    var filePath = Path.GetDirectoryName(files[0]);
                    foreach (var file in files)
                    {
                        try
                        {
                            var fileName = Path.GetFileNameWithoutExtension(file);
                            var newFileName = $"Decrpt{fileName}.txt";
                            var newFilePath = Path.Combine(filePath, newFileName);
                            var licenseInfo = File.ReadAllText(file);
                            var decryptInfo = DesBuilder.Decrypt(licenseInfo);
                            File.WriteAllText(newFilePath, decryptInfo);
                        }
                        catch(Exception ex)
                        {
                            MessageBox.Show($"解密出错:{ex}", "提示");
                        }
                    }
                    PostGeneration(filePath);
                    MessageBox.Show($"解密完成！", "提示");
                }
            }
        }
    }
}