Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -2,6 +2,7 @@ import scratchcommunication
 import time
 import requests
 import string
+import traceback
 
 PROJECT_ID = "1293030706"
 USERNAME = "server"
@@ -13,10 +14,6 @@ tw = scratchcommunication.TwCloudConnection(
     contact_info=CONTACT
 )
 
-# ---------------------
-# 文字<->数字マッピング
-# 0-9: 0-9, a-z: 10-35
-# ---------------------
 CHARS = list(string.digits + string.ascii_lowercase)
 CHAR_TO_NUM = {c: i for i, c in enumerate(CHARS)}
 NUM_TO_CHAR = {i: c for i, c in enumerate(CHARS)}
@@ -27,100 +24,140 @@ NUM_TO_CHAR = {i: c for i, c in enumerate(CHARS)}
 def get_var(name):
     try:
         return tw.get_variable(name=name, name_literal=False)
-    except Exception:
+    except Exception as e:
+        print(f"[ERROR] get_var('{name}') failed: {e}")
+        traceback.print_exc()
         return None
 
 def set_var(name, value):
     try:
         return tw.set_variable(name=name, value=value, name_literal=False)
-    except Exception:
+    except Exception as e:
+        print(f"[ERROR] set_var('{name}', '{value}') failed: {e}")
+        traceback.print_exc()
         return None
 
 # ---------------------
-# n0管理（使用中フラグ）
+# n0管理
 # ---------------------
 def n0_is_free():
-    v = get_var("n0")
-    return v == "0" or v is None
+    try:
+        v = get_var("n0")
+        return v == "0" or v is None
+    except Exception as e:
+        print(f"[ERROR] n0_is_free() check failed: {e}")
+        traceback.print_exc()
+        return False
 
 def set_n0(value):
-    set_var("n0", str(value))
+    try:
+        set_var("n0", str(value))
+    except Exception as e:
+        print(f"[ERROR] set_n0({value}) failed: {e}")
+        traceback.print_exc()
 
 # ---------------------
 # IDエンコード / デコード
 # ---------------------
 def encode_id(sid):
-    """a-z0-9を00-35に変換して結合"""
-    out = ""
-    for c in sid:
-        num = CHAR_TO_NUM[c]
-        out += f"{num:02d}"
-    return out
+    try:
+        out = ""
+        for c in sid:
+            num = CHAR_TO_NUM[c]
+            out += f"{num:02d}"
+        return out
+    except Exception as e:
+        print(f"[ERROR] encode_id('{sid}') failed: {e}")
+        traceback.print_exc()
+        return "00"  # デフォルトの無効値
 
 def decode_id(data):
-    out = ""
-    for i in range(0, len(data), 2):
-        num = int(data[i:i+2])
-        out += NUM_TO_CHAR[num]
-    return out
+    try:
+        out = ""
+        for i in range(0, len(data), 2):
+            num = int(data[i:i+2])
+            out += NUM_TO_CHAR[num]
+        return out
+    except Exception as e:
+        print(f"[ERROR] decode_id('{data}') failed: {e}")
+        traceback.print_exc()
+        return "0"
 
 # ---------------------
 # n1の監視・送受信
 # ---------------------
 def monitor_n1():
-    v = get_var("n1")
-    if not v or len(v) < 3:
+    try:
+        v = get_var("n1")
+        if not v or len(v) < 3:
+            return None
+
+        status = v[0]
+        read_flag = v[1]
+        content = v[2:]
+
+        return {"status": status, "read": read_flag, "content": content}
+    except Exception as e:
+        print(f"[ERROR] monitor_n1() failed: {e}")
+        traceback.print_exc()
         return None
 
-    status = v[0]  # 0: ID送信, 1: レスポンス返送
-    read_flag = v[1]  # 0:未読, 1:既読
-    content = v[2:]
-
-    return {"status": status, "read": read_flag, "content": content}
-
 def mark_n1_read():
-    v = get_var("n1")
-    if v and len(v) >= 2:
-        newv = v[0] + "1" + v[2:]
-        set_var("n1", newv)
+    try:
+        v = get_var("n1")
+        if v and len(v) >= 2:
+            newv = v[0] + "1" + v[2:]
+            set_var("n1", newv)
+    except Exception as e:
+        print(f"[ERROR] mark_n1_read() failed: {e}")
+        traceback.print_exc()
 
 def send_response(encoded_id):
-    # 先頭1は1 (レスポンス送信), 2文字目は0 (未読)
-    payload = "1" + "0" + encoded_id
-    set_var("n1", payload)
+    try:
+        payload = "1" + "0" + encoded_id
+        set_var("n1", payload)
+    except Exception as e:
+        print(f"[ERROR] send_response('{encoded_id}') failed: {e}")
+        traceback.print_exc()
 
 # ---------------------
 # メインループ
 # ---------------------
 while True:
-    n1_data = monitor_n1()
-    if n1_data:
-        # ID受信（送信元Scratchから）
-        if n1_data["status"] == "0" and n1_data["read"] == "0" and n0_is_free():
-            # 使用中フラグ立てる
-            set_n0(1)
-
-            received_id = n1_data["content"]
-            mark_n1_read()  # 既読にする
-
-            # サーバーにID送信してレスポンス取得
-            try:
-                url = f"https://20.rf.gd/set.php?project_id={received_id}"
-                r = requests.get(url, timeout=10, cookies={"__test": "872039d283902860e895efb6018d2632"},)
-                r.raise_for_status()
-                response_json = r.json()
-                short_id = response_json.get("short_id", "0")  # short_idがなければ"0"
-                encoded_short = encode_id(short_id)
-
-                # ScratchにレスポンスID送信
-                send_response(encoded_short)
-
-            except Exception:
-                # エラー時は0で返す
-                send_response(encode_id("0"))
-
-        # レスポンス既読を検知
-        elif n1_data["status"] == "1" and n1_data["read"] == "1":
-            set_n0(0)  # 使用中フラグを下ろす
+    try:
+        n1_data = monitor_n1()
+        if n1_data:
+            # ID受信
+            if n1_data["status"] == "0" and n1_data["read"] == "0" and n0_is_free():
+                set_n0(1)
+                received_id = n1_data["content"]
+                mark_n1_read()
+
+                try:
+                    url = f"https://20.rf.gd/set.php?project_id={received_id}"
+                    r = requests.get(
+                        url,
+                        timeout=10,
+                        cookies={"__test": "872039d283902860e895efb6018d2632"},
+                    )
+                    r.raise_for_status()
+                    response_json = r.json()
+                    short_id = response_json.get("short_id", "0")
+                    encoded_short = encode_id(short_id)
+
+                    send_response(encoded_short)
+
+                except Exception as e:
+                    print(f"[ERROR] Server request for ID '{received_id}' failed: {e}")
+                    traceback.print_exc()
+                    send_response(encode_id("0"))
+
+            # レスポンス既読
+            elif n1_data["status"] == "1" and n1_data["read"] == "1":
+                set_n0(0)
+
+    except Exception as e:
+        print(f"[ERROR] Main loop iteration failed: {e}")
+        traceback.print_exc()
 
     time.sleep(0.2)