Update turbowarp-server/qr-converter.py

turbowarp-server/qr-converter.py

@@ -14,146 +14,147 @@ tw = scratchcommunication.TwCloudConnection(
     contact_info=CONTACT
 )
 
+# ---------------------
+# 文字テーブル
+# ---------------------
 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)}
 
+# ---------------------
+# 文字列エンコード / デコード
+# ---------------------
+def encode(text):
+    try:
+        out = ""
+        for c in text:
+            out += f"{CHAR_TO_NUM[c]:02d}"
+        return out
+    except Exception as e:
+        print(f"[ERROR] encode('{text}') failed: {e}")
+        traceback.print_exc()
+        return "00"
+
+def decode(data):
+    try:
+        out = ""
+        for i in range(0, len(data), 2):
+            num = int(data[i:i+2])
+            out += NUM_TO_CHAR.get(num, "0")
+        return out
+    except Exception as e:
+        print(f"[ERROR] decode('{data}') failed: {e}")
+        traceback.print_exc()
+        return "0"
+
 # ---------------------
 # クラウド変数操作
 # ---------------------
 def get_var(name):
     try:
         return tw.get_variable(name=name, name_literal=False)
-    except Exception as e:
-        print(f"[ERROR] get_var('{name}') failed: {e}")
-        traceback.print_exc()
+    except Exception:
         return None
 
 def set_var(name, value):
     try:
-        return tw.set_variable(name=name, value=value, name_literal=False)
+        tw.set_variable(name=name, value=value, name_literal=False)
     except Exception as e:
         print(f"[ERROR] set_var('{name}', '{value}') failed: {e}")
         traceback.print_exc()
-        return None
 
 # ---------------------
 # n0管理
 # ---------------------
 def n0_is_free():
-    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
+    v = get_var("n0")
+    return v in [None, "0"]
 
 def set_n0(value):
-    try:
-        set_var("n0", str(value))
-    except Exception as e:
-        print(f"[ERROR] set_n0({value}) failed: {e}")
-        traceback.print_exc()
+    set_var("n0", str(value))
 
 # ---------------------
-# IDエンコード / デコード
+# データ送受信
 # ---------------------
-def encode_id(sid):
-    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"  # デフォルトの無効値
+SLOT = "n1"
+buffers = {}
+
+def send(slot, text):
+    encoded = encode(text)
+    size = 99996
+    packets = [encoded[i:i+size] for i in range(0, len(encoded), size)]
+    total = len(packets)
+
+    for p in packets:
+        packet = f"1{total}0{p}"
+        start = time.time()
+        set_var(slot, packet)
+
+        while True:
+            v = get_var(slot)
+            if v and len(v) > 2 and v[2] == "1":
+                break
+            if time.time() - start > 10:
+                print(f"[WARN] send timeout for slot '{slot}'")
+                break
+            time.sleep(0.1)
+
+def receive(slot):
+    v = get_var(slot)
+    if not v or len(v) < 3:
+        return None
+
+    if v[0] != "0" or v[2] != "0":
+        return None
 
-def decode_id(data):
     try:
-        out = ""
-        for i in range(0, len(data), 2):
-            num = int(data[i:i+2])
-            out += NUM_TO_CHAR[num]
-        return out
+        total = int(v[1])
     except Exception as e:
-        print(f"[ERROR] decode_id('{data}') failed: {e}")
+        print(f"[ERROR] Invalid total in slot '{slot}': {v}")
         traceback.print_exc()
-        return "0"
+        return None
 
-# ---------------------
-# n1の監視・送受信
-# ---------------------
-def monitor_n1():
-    try:
-        v = get_var("n1")
-        if not v or len(v) < 3:
-            return None
+    data = v[3:]
+    newv = v[:2] + "1" + v[3:]
+    set_var(slot, newv)
 
-        status = v[0]
-        read_flag = v[1]
-        content = v[2:]
+    if slot not in buffers:
+        buffers[slot] = []
 
-        return {"status": status, "read": read_flag, "content": content}
-    except Exception as e:
-        print(f"[ERROR] monitor_n1() failed: {e}")
-        traceback.print_exc()
-        return None
+    buffers[slot].append(data)
 
-def mark_n1_read():
-    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()
+    if len(buffers[slot]) < total:
+        return None
 
-def send_response(encoded_id):
-    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()
+    joined = "".join(buffers[slot])
+    buffers[slot] = []
+    return decode(joined)
 
 # ---------------------
 # メインループ
 # ---------------------
 while True:
     try:
-        n1_data = monitor_n1()
-        if n1_data:
-            # ID受信
-            if n1_data["status"] == "0" and n1_data["read"] == "0" and n0_is_free():
+        if n0_is_free():
+            received_id = receive(SLOT)
+            if received_id:
+                print(f"[INFO] Received ID: {received_id}")
                 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 = requests.get(url, timeout=10)
                     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)
+                    print(f"[INFO] Server response short_id: {short_id}")
+                    send(SLOT, short_id)
 
                 except Exception as e:
-                    print(f"[ERROR] Server request for ID '{received_id}' failed: {e}")
+                    print(f"[ERROR] Server request failed for ID '{received_id}': {e}")
                     traceback.print_exc()
-                    send_response(encode_id("0"))
+                    send(SLOT, "0")
 
-            # レスポンス既読
-            elif n1_data["status"] == "1" and n1_data["read"] == "1":
                 set_n0(0)
 
     except Exception as e: