From 95e8855c270c5f128fb94e72f401f81f7ec0581d Mon Sep 17 00:00:00 2001
From: suti7yk5032 <yuman1965969@gmail.com>
Date: Fri, 26 Jul 2024 11:45:13 +0900
Subject: [PATCH] =?UTF-8?q?=E3=82=A2=E3=83=97=E3=83=AA=E3=83=87=E3=83=BC?=
 =?UTF-8?q?=E3=82=BF=E3=81=AE=E5=89=8A=E9=99=A4=E5=87=A6=E7=90=86=E3=82=92?=
 =?UTF-8?q?=E8=BF=BD=E5=8A=A0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dislocker_client.py | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/dislocker_client.py b/dislocker_client.py
index ddfde4e..61e2702 100644
--- a/dislocker_client.py
+++ b/dislocker_client.py
@@ -13,6 +13,7 @@ import tkinter
 import threading
 import signal
 import sys
+import shutil
 
 app_name = "Dislocker"
 dislocker_dir = os.path.dirname(os.path.abspath(sys.argv[0]))
@@ -72,6 +73,37 @@ class App(customtkinter.CTk):
         self.toast()
         self.destroy()
 
+    def delete_appdata(**kwargs):
+        process_name = kwargs["process_name"]
+        dir_path = kwargs["dir_path"]
+
+        if not os.path.exists(dir_path):
+            print(f"エラー: 指定されたディレクトリ {dir_path} が存在しません。")
+            return 1
+
+        try:
+            # プロセスの終了
+            subprocess.run(['taskkill', '/f', '/t', '/im', process_name])
+            print(f"{process_name} を終了しました。")
+
+            # ディレクトリの削除
+            shutil.rmtree(dir_path)
+            print(f"{dir_path} を削除しました。")
+
+            return 0
+        except subprocess.CalledProcessError as e:
+            print(f"プロセス終了エラー: {e}")
+            return 1
+        except PermissionError as e:
+            print(f"権限エラー: {e}")
+            return 1
+        except Exception as error:
+            print("エラーが発生しました。\nエラー内容:")
+            print(f"エラータイプ: {error.__class__.__name__}")
+            print(f"エラー引数: {error.args}")
+            print(f"エラーメッセージ: {str(error)}")
+            return 1
+    
     def block_key(self):
         block_keys = ['ctrl', 'alt', 'windows', 'shift', 'delete']
         for i in block_keys:
@@ -295,6 +327,11 @@ class Monitor():
 
     def signal_handler(self):
         print("停止処理を実行。")
+        appdata_local = os.path.expandvars("%LOCALAPPDATA%")
+        appdata_roaming = os.path.expandvars("%APPDATA%")
+        epic_del = app.delete_appdata(process_name="EpicGamesLauncher.exe", dir_path=f"{appdata_local}\\EpicGamesLauncher\\Saved")
+        chrome_del = app.delete_appdata(process_name="chrome.exe", dir_path=f"{appdata_local}\\Google\\Chrome\\User Data")
+        discord_del = app.delete_appdata(process_name="discord.exe", dir_path=f"{appdata_roaming}\\discord")
         stop_url = client_config["auth_host_url"] + "/stop"
         stop_json = {
             "pc_number": int(client_config["pc_number"])