Add Linux support for KRNL automation with subprocess and xdotool

Co-authored-by: oicne4 <oicne4@gmail.com>

Author: cursoragent

krnl_automation.py

@@ -1,5 +1,6 @@
 import re
 import time
+import subprocess
 import pyautogui
 import pyperclip
 from selenium import webdriver
@@ -9,12 +10,13 @@
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 
-# キー形式チェック
+# Linux用のKRNLパス（実際のパスに変更してください）
+KRNL_PATH = "/path/to/krnl"  # ここに実際のKRNL実行ファイルパスを入れてください
+
 def check_krnl_key(key: str):
     pattern = re.compile(r'^[0-9a-fA-F]{4}(-[0-9a-fA-F]{4}){3}$')
     return bool(pattern.match(key))
 
-# KRNLキー自動取得
 def auto_get_key_full():
     print("\n[+] KRNLキー取得自動化を開始します...")
     driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()))
@@ -24,19 +26,16 @@ def auto_get_key_full():
     krnl_key = None
 
     try:
-        # 「次へ」クリック
         next_btn = wait.until(EC.element_to_be_clickable((By.XPATH, "//button[contains(text(), '次へ')]")))
         time.sleep(1)
         next_btn.click()
         print("[+] '次へ'ボタンをクリックしました")
 
-        # 「続行」クリック
         time.sleep(5)
         continue_btn = wait.until(EC.element_to_be_clickable((By.XPATH, "//button[contains(text(), '続行')]")))
         continue_btn.click()
         print("[+] '続行'ボタンをクリックしました")
 
-        # キー取得
         key_element = wait.until(EC.presence_of_element_located((By.XPATH, "//code")))
         krnl_key = key_element.text.strip()
         pyperclip.copy(krnl_key)
@@ -49,26 +48,50 @@ def auto_get_key_full():
 
     return krnl_key
 
-# KRNLにキーを入力してInject
+def focus_krnl_window():
+    """Linux用のウィンドウフォーカス関数"""
+    try:
+        # xdotoolを使用してKRNLウィンドウをアクティブにする
+        subprocess.run(["xdotool", "search", "--name", "KRNL", "windowactivate"], check=True)
+        time.sleep(1)
+        print("[+] KRNLウィンドウをアクティブにしました")
+        return True
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        print("[-] xdotoolが見つからないか、KRNLウィンドウが見つかりません。")
+        print("[-] 手動でKRNLウィンドウをアクティブにしてください。")
+        time.sleep(5)  # ユーザーに手動でアクティブにしてもらう時間
+        return True
+
 def inject_krnl(key):
     print("\n[+] KRNLにキーを入力してInjectします...")
-    time.sleep(5)  # ユーザーにKRNL画面をアクティブにしてもらう時間
-
-    # キー入力欄クリック（座標は環境に合わせて調整）
-    pyautogui.click(500, 400)  # 仮の座標
+    if not focus_krnl_window():
+        return
+    
+    # キー入力欄クリック（位置は環境に合わせて調整）
+    pyautogui.click(500, 400)
     time.sleep(0.5)
-
+    
     # キー貼り付け
     pyautogui.hotkey("ctrl", "v")
     time.sleep(0.5)
-
-    # Submitクリック（座標も調整必要）
+    
+    # Injectボタンクリック（位置は調整が必要）
     pyautogui.click(600, 500)
     time.sleep(1)
-
-    print("[+] キー送信完了、Injectが開始されます。")
+    
+    print("[+] Inject処理完了しました。")
 
 if __name__ == "__main__":
+    print("[+] KRNLを起動します...")
+    try:
+        subprocess.Popen([KRNL_PATH])
+        time.sleep(10)  # 起動待機（PC環境によって調整してください）
+    except FileNotFoundError:
+        print("[-] KRNL実行ファイルが見つかりません。")
+        print("[-] KRNL_PATHを正しいパスに設定してください。")
+        print("[-] 手動でKRNLを起動してください。")
+        time.sleep(5)
+
     user_key = input("既にKRNLキーを持っていますか？（なければEnter）: ").strip()
 
     if user_key and check_krnl_key(user_key):