Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -18,7 +18,7 @@ print("[INFO] Connected successfully.")
 def get_var(name):
     try:
         value = tw.get_variable(name=name, name_literal=False)
-        print(f"[GET] Variable '{name}' = {value}")
+        print(f"[GET] Variable '{name}' = {value[:50]}..." if value and len(value) > 50 else f"[GET] Variable '{name}' = {value}")
         return value
     except Exception as e:
         print(f"[ERROR] Failed to get variable '{name}': {e}")
@@ -27,11 +27,11 @@ def get_var(name):
 def set_var(name, value):
     try:
         tw.set_variable(name=name, value=value, name_literal=False)
-        print(f"[SET] Variable '{name}' = {value}")
+        print(f"[SET] Variable '{name}' = {value[:50]}..." if len(value) > 50 else f"[SET] Variable '{name}' = {value}")
         return True
     except Exception as e:
         print(f"[ERROR] Failed to set variable '{name}': {e}")
-        return False
+        return None
 
 # --- n-chars.txt の読み込み ---
 print("[INFO] Loading n-chars.txt...")
@@ -47,7 +47,7 @@ def decode_prompt(encoded_str):
         if idx < len(n_chars):
             chars.append(n_chars[idx])
     decoded = "".join(chars)
-    print(f"[DECODE] Encoded prompt '{encoded_str[:20]}...' -> '{decoded[:30]}...'")
+    print(f"[DECODE] Encoded prompt -> '{decoded[:50]}...'")
     return decoded
 
 def rgb_to_scratch_number(rgb):
@@ -99,35 +99,99 @@ def split_packets(data, max_length=9998):
 
 # --- メインループ ---
 print("[INFO] Starting main loop...")
+last_processed_id = ""  # 最後に処理したIDを記録
+
 while True:
     try:
         n1 = get_var("n1")
-        if n1 and len(n1) >= 3 and n1[0] == "0":  # 未処理フラグ
-            print("[INFO] Received new request from n1.")
-            if get_var("n0") == "0":  # 他ユーザー未使用
+        n0 = get_var("n0")
+        
+        # n1が存在し、長さが3以上で、先頭が"0"（新規リクエスト）
+        if n1 and len(n1) >= 3 and n1[0] == "0":
+            user_id = n1[1:3]  # ユーザーID（2桁）
+            request_data = n1  # 完全なリクエストデータ
+            
+            print(f"[INFO] Received request from user {user_id}")
+            print(f"[INFO] Request data length: {len(request_data)}")
+            
+            # 同じリクエストを繰り返し処理しないようにチェック
+            if request_data == last_processed_id:
+                print("[INFO] Already processed this request, marking as read...")
+                # 既読マークをつける（先頭を"01"に変更）
+                marked_as_read = "01" + request_data[2:]
+                set_var("n1", marked_as_read)
+                time.sleep(0.1)
+                continue
+            
+            # n0が利用可能かチェック
+            if n0 == "0" or n0 is None:
+                print("[INFO] n0 is free, starting processing...")
+                
+                # 処理中フラグをセット
                 set_var("n0", "1")
-                user_id = n1[1:3]  # 実際のIDはここを調整
-                encoded_prompt = n1[3:]
+                
+                # リクエストを既読にする（先頭を"01"に変更）
+                marked_as_read = "01" + request_data[2:]
+                set_var("n1", marked_as_read)
+                
+                # プロンプトをデコード（先頭の"0" + ユーザーID(2桁)を除去）
+                encoded_prompt = request_data[3:]
                 prompt = decode_prompt(encoded_prompt)
-        
+                
                 # 画像生成
                 img = generate_image(prompt)
                 scratch_data = resize_and_encode(img)
-        
-                # パケット分割・送信
+                
+                # パケット分割
                 packets = split_packets(scratch_data)
+                
+                # 最初のパケットを送信
                 for i, pkt in enumerate(packets):
+                    print(f"[INFO] Sending packet {i+1}/{len(packets)}")
                     set_var("n1", pkt)
-                    time.sleep(0.2)
-                    while get_var("n1") != "11":
-                        time.sleep(0.1)
-        
-                # --- ここで既読フラグに変更 ---
-                set_var("n1", "01" + n1[2:])
-                print("[INFO] Marked n1 as processed.")
-        
-                # 処理完了後リセット
+                    
+                    if i < len(packets) - 1:
+                        # 次のパケットのACKを待つ
+                        print("[INFO] Waiting for ACK '11'...")
+                        ack_timeout = 0
+                        while ack_timeout < 50:  # 5秒タイムアウト
+                            current_n1 = get_var("n1")
+                            if current_n1 == "11":
+                                print("[INFO] Received ACK")
+                                break
+                            time.sleep(0.1)
+                            ack_timeout += 1
+                        
+                        if ack_timeout >= 50:
+                            print("[ERROR] ACK timeout")
+                            set_var("n0", "0")
+                            break
+                    else:
+                        # 最後のパケットは完了を待つ
+                        print("[INFO] Waiting for completion '99'...")
+                        complete_timeout = 0
+                        while complete_timeout < 50:
+                            current_n1 = get_var("n1")
+                            if current_n1 == "99":
+                                print("[INFO] Received completion signal")
+                                break
+                            time.sleep(0.1)
+                            complete_timeout += 1
+                
+                # 完了後リセット
+                print("[INFO] Transmission complete. Resetting n0.")
                 set_var("n0", "0")
+                last_processed_id = request_data  # 処理済みとして記録
+                
+            else:
+                print("[INFO] n0 is busy, waiting...")
+        else:
+            # 何もしない
+            pass
+            
     except Exception as e:
         print(f"[ERROR] Exception in main loop: {e}")
+        import traceback
+        traceback.print_exc()
+    
     time.sleep(0.2)