From 0940f31b7eba1cf6b6a8c1fca72d97cf686e79f2 Mon Sep 17 00:00:00 2001
From: alielie <alielie@git.com>
Date: 星期五, 04 七月 2025 10:43:00 +0800
Subject: [PATCH] 优化插件,完善TreeListView

---
 WPFProject/App.xaml.cs |   77 ++++++++++++++++++++++++++++++++++++--
 1 files changed, 72 insertions(+), 5 deletions(-)

diff --git a/WPFProject/App.xaml.cs b/WPFProject/App.xaml.cs
index 545b7cc..c63a2e2 100644
--- a/WPFProject/App.xaml.cs
+++ b/WPFProject/App.xaml.cs
@@ -2,9 +2,12 @@
 using HMI.Core.Util;
 using HMI.UI.Util;
 using log4net;
+using Microsoft.Win32;
 using Panuon.WPF.UI;
 using System;
+using System.Diagnostics;
 using System.IO;
+using System.Management;
 using System.Reflection;
 using System.Threading;
 using System.Windows;
@@ -18,6 +21,7 @@
     public partial class App : Application
     {
         private ILog log;
+        private LogCleanner logCleanner;
         public App()
         {
             log = LogManager.GetLogger("APP");
@@ -28,19 +32,82 @@
         {
             var ex = e.ExceptionObject as Exception;
             log.Error("UnhandledException.", ex);
-            MsgBoxHelper.ErrorMessage(Current.MainWindow, this.lang("App.AppUnhandledException",ex.Message),
+            MsgBoxHelper.ErrorMessage(Current.MainWindow, this.lang("App.AppUnhandledException", ex.Message),
                     this.lang("App.Error"), MessageBoxButton.OK,
                     MessageBoxIcon.Error, DefaultButton.YesOK, 5);
+        }
+
+        public static Process GetProcess()
+        {
+            Process cur = Process.GetCurrentProcess();
+            string exePath = Assembly.GetExecutingAssembly().Location;
+            string query = $"SELECT ProcessId FROM Win32_Process WHERE ExecutablePath = '{exePath.Replace("\\", "\\\\")}'";
+            using (var searcher = new ManagementObjectSearcher(query))
+            {
+                foreach (var obj in searcher.Get())
+                {
+                    int processId = Convert.ToInt32(obj["ProcessId"]);
+                    if (processId != cur.Id)
+                    {
+                        return Process.GetProcessById(processId);
+                    }
+                }
+            }
+            return null;
+        }
+
+        /// <summary>
+        /// 鍚姩鍚庡線娉ㄥ唽琛ㄥ啓鐐逛笢瑗�
+        /// </summary>
+        private void WriteRegister()
+        {
+            string Company = "JingCe", Program = "HMI";
+            try
+            {
+                string registryPath = @$"SOFTWARE\{Company}\{Program}";
+
+                using RegistryKey key = Registry.CurrentUser.CreateSubKey(registryPath, true)
+                    ?? throw new Exception("鏃犳硶鍒涘缓鎴栨墦寮�娉ㄥ唽琛ㄩ敭");
+                string path = AppDomain.CurrentDomain.BaseDirectory;
+                key.SetValue("Location", path, RegistryValueKind.String);
+            }
+            catch (Exception ex)
+            {
+                log.Error("Write Register Failed.", ex);
+            }
+        }
+
+        private void DoUpdateAction()
+        {
+            using (SqliteHelper helper = new SqliteHelper(PathHelper.DatabasePath))
+            {
+                if (!helper.HasData("SELECT 1 FROM pragma_table_info('user') WHERE name = 'code';"))
+                {
+                    helper.ExecuteSql("ALTER TABLE user ADD code TEXT;");
+                }
+            }
         }
 
         private static Mutex mutex;
         private void Application_Startup(object sender, StartupEventArgs e)
         {
-            mutex = new Mutex(true, "TestPlatform_OnlyRun_CRNS");
-            if (mutex.WaitOne(0, false))
-            {
-                AppDomain.CurrentDomain.AssemblyResolve += CurrentDomain_AssemblyResolve;
+            var process = GetProcess();
 
+            mutex = new Mutex(true, "TestPlatform_OnlyRun_CRNS");
+            if (mutex.WaitOne(0, false))//&& process == null
+            {
+                WriteRegister();
+                try
+                {
+                    DoUpdateAction();
+                }
+                catch (Exception ex) {
+                    log.Error("Update Failed.", ex);
+                }
+
+                logCleanner = new LogCleanner();
+                AppDomain.CurrentDomain.AssemblyResolve += CurrentDomain_AssemblyResolve;
+                LoggerHelper.Register(new CommonLogger());
                 LangUtil.Load();
                 SystemSetting.SystemSettingChanged += SystemSetting_SystemSettingChanged;
                 ToastHelper.Register(new MainToastHandler());

--
Gitblit v1.9.1