Update join.py

join.py

@@ -1,104 +1,201 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
+"""
+10秒ごとに指定のグループのメッセージを取得し、
+各メッセージ内の log / logs の action == "join" を検出。
+その join が発生したメッセージの createdAt より新しい (createdAt が大きい)
+personId == 599642 のメッセージがなければ歓迎メッセージを送信し、
+join を起こした人を指定のグループへ invite POST する。
+"""
+
 import os
 import time
+import json
 import requests
-from typing import List, Dict, Any
+from typing import List, Dict, Any, Optional, Set
 
 # --- 設定 ---
 X_ACCOUNT = os.getenv("dmsendertoken")
 if not X_ACCOUNT:
     raise RuntimeError("環境変数 dmsendertoken が設定されていません。")
 
-BASE_MESSAGES = "https://desk-api.channel.io/desk/channels/200605/groups/463667/messages"
-INVITE_GROUPS = [519217, 536194, 534868]
-
-BOT_PERSON_ID = "599642"
-WELCOME_TEXT = "こんにちは。ITRSAにようこそ。自動でいくつかの部屋に招待しています。"
+GROUP_ID_TO_WATCH = 463667
+PARAMS = {"sortOrder": "desc", "limit": 34, "logFolded": "false"}
+BASE_GET = "https://desk-api.channel.io/desk/channels/200605/groups/{group_id}/messages"
+BASE_POST = "https://desk-api.channel.io/desk/channels/200605/groups/{group_id}/messages"
+INVITE_ENDPOINT_TEMPLATE = "https://desk-api.channel.io/desk/channels/200605/groups/{invite_group_id}/invite?managerIds={manager_id}"
 
-PARAMS = {
-    "sortOrder": "desc",
-    "limit": 34,
-    "logFolded": "false",
+HEADERS_GET = {
+    "accept": "application/json",
+    "accept-language": "ja",
+    "x-account": X_ACCOUNT,
 }
-
-HEADERS = {
+HEADERS_POST = {
     "accept": "application/json",
     "accept-language": "ja",
     "content-type": "application/json",
     "x-account": X_ACCOUNT,
 }
 
-# --- API ---
-def post_message(text: str) -> None:
+TARGET_PERSON_ID = "599642"  # チェック対象の personId（文字列比較）
+INVITE_GROUP_IDS = [519217, 536194, 534868]
+
+WELCOME_MESSAGE_TEXT = "こんにちは。ITRSAにようこそ。自動でいくつかの部屋に招待しています。"
+
+# --- ヘルパー ---
+
+
+def fetch_messages(group_id: int) -> List[Dict[str, Any]]:
+    url = BASE_GET.format(group_id=group_id)
+    r = requests.get(url, headers=HEADERS_GET, params=PARAMS, timeout=20)
+    r.raise_for_status()
+    data = r.json()
+    msgs = data.get("messages", [])
+    if not isinstance(msgs, list):
+        return []
+    return msgs
+
+
+def send_welcome_message(group_id: int, text: str) -> None:
+    url = BASE_POST.format(group_id=group_id)
     payload = {
         "requestId": f"desk-web-{int(time.time() * 1000)}",
         "blocks": [{"type": "text", "value": text}],
     }
-    r = requests.post(BASE_MESSAGES, headers=HEADERS, json=payload, timeout=30)
+    r = requests.post(url, headers=HEADERS_POST, json=payload, timeout=30)
     r.raise_for_status()
-    print("[INFO] 歓迎メッセージ送信")
 
-def invite_to_group(group_id: int, manager_id: str) -> None:
-    url = f"https://desk-api.channel.io/desk/channels/200605/groups/{group_id}/invite"
-    params = {"managerIds": manager_id}
-    r = requests.post(url, headers=HEADERS, params=params, timeout=30)
-    r.raise_for_status()
-    print(f"[INFO] group {group_id} に {manager_id} を招待")
 
-# --- メイン処理 ---
-def process():
-    r = requests.get(BASE_MESSAGES, headers=HEADERS, params=PARAMS, timeout=30)
+def post_invite(invite_group_id: int, manager_id: str) -> None:
+    url = INVITE_ENDPOINT_TEMPLATE.format(invite_group_id=invite_group_id, manager_id=manager_id)
+    # API に特別な body が必要ならここに入れるが、指定がなかったので空 JSON を送る
+    r = requests.post(url, headers=HEADERS_POST, json={}, timeout=20)
     r.raise_for_status()
-    messages = r.json().get("messages", [])
 
-    # 最新の bot メッセージ時刻
-    bot_created_at = None
-    for m in messages:
-        if str(m.get("personId")) == BOT_PERSON_ID:
-            bot_created_at = int(m.get("createdAt") or 0)
-            break  # desc なので最初でOK
-
-    join_messages: List[Dict[str, Any]] = []
 
+def extract_join_person_ids_from_message(msg: Dict[str, Any]) -> List[str]:
+    """
+    メッセージ内の log / logs を調べ、action == "join" のエントリから personId を取り出す。
+    log エントリ自体に personId がある場合はそれを使い、なければメッセージの personId を使う。
+    戻り値は文字列の personId のリスト（重複はこの段階では除去しない）。
+    """
+    person_ids = []
+    # 支持されているキー名を両方確認
+    log_lists = []
+    for key in ("log", "logs"):
+        if key in msg and isinstance(msg[key], list):
+            log_lists.append(msg[key])
+
+    # もし log/ logs が無ければ return empty
+    if not log_lists:
+        return []
+
+    for logs in log_lists:
+        for entry in logs:
+            if not isinstance(entry, dict):
+                continue
+            action = entry.get("action")
+            if action != "join":
+                continue
+            pid = entry.get("personId") or msg.get("personId")
+            if pid is not None:
+                person_ids.append(str(pid))
+    return person_ids
+
+
+def any_person599642_after(created_at: int, messages: List[Dict[str, Any]]) -> bool:
+    """
+    messages の中に createdAt が created_at より大きく、かつ personId == TARGET_PERSON_ID のものがあれば True
+    """
     for m in messages:
-        created_at = int(m.get("createdAt") or 0)
-        if bot_created_at and created_at <= bot_created_at:
-            continue
+        try:
+            ca = int(m.get("createdAt") or 0)
+        except Exception:
+            ca = 0
+        if ca > created_at and str(m.get("personId", "")) == TARGET_PERSON_ID:
+            return True
+    return False
+
+
+def process_once(messages: List[Dict[str, Any]]) -> None:
+    """
+    1 回分のメッセージ群を処理する。
+    - 各 message の log(s) 内で action == "join" を探す
+    - そのメッセージの createdAt 以降に personId == 599642 のメッセージが無ければ招待処理
+    - 同じ manager_id への招待は一回の処理実行内で重複しないようにする
+    - 歓迎メッセージはその条件で招待を行う最初のケースで一度だけ送る
+    """
+    if not messages:
+        return
 
-        for log in m.get("log", []) or []:
-            if log.get("action") == "join":
-                join_messages.append({
-                    "createdAt": created_at,
-                    "personId": str(m.get("personId")),
-                })
+    invited_manager_ids: Set[str] = set()
+    welcome_sent = False
 
-    if not join_messages:
-        return
+    # messages は API から降順 (新しい順) で来るはずだが、安全のためそのまま扱う。
+    for msg in messages:
+        try:
+            created_at = int(msg.get("createdAt") or 0)
+        except Exception:
+            created_at = 0
 
-    # bot発言以降に join があり、botの発言がまだ無い場合のみ歓迎メッセージ
-    if bot_created_at is None:
-        post_message(WELCOME_TEXT)
+        join_person_ids = extract_join_person_ids_from_message(msg)
+        if not join_person_ids:
+            continue
 
-    # join 全員を invite（複数可）
-    invited = set()
-    for jm in join_messages:
-        pid = jm["personId"]
-        if not pid or pid in invited:
+        # この join が起きたメッセージ以降に personId==599642 のメッセージがあるか確認
+        if any_person599642_after(created_at, messages):
+            # あるならこの join はスキップ（歓迎メッセージが既にある/今後来る想定のため）
             continue
-        for gid in INVITE_GROUPS:
-            invite_to_group(gid, pid)
-        invited.add(pid)
 
-def main():
-    print("[INFO] Bot 起動（10秒ごとにチェック）")
+        # 招待対象 personId を集める（同一処理内で重複しないよう set で管理）
+        for manager_id in join_person_ids:
+            if manager_id in invited_manager_ids:
+                continue
+            # まず歓迎メッセージ（まだ送っていなければ）
+            if not welcome_sent:
+                try:
+                    send_welcome_message(GROUP_ID_TO_WATCH, WELCOME_MESSAGE_TEXT)
+                    print(f"[INFO] Sent welcome message to group {GROUP_ID_TO_WATCH}")
+                except Exception as e:
+                    print(f"[ERROR] Failed to send welcome message: {e}")
+                    # 歓迎送信失敗しても続行して招待は試みる
+                welcome_sent = True
+
+            # 各 invite グループへ POST
+            for invite_gid in INVITE_GROUP_IDS:
+                try:
+                    post_invite(invite_gid, manager_id)
+                    print(f"[INFO] Invited manager {manager_id} to group {invite_gid}")
+                except Exception as e:
+                    print(f"[ERROR] Failed to invite manager {manager_id} to group {invite_gid}: {e}")
+
+            invited_manager_ids.add(manager_id)
+
+    if invited_manager_ids:
+        print(f"[INFO] Invited manager ids in this cycle: {', '.join(sorted(invited_manager_ids))}")
+    else:
+        print("[INFO] No join events required inviting in this cycle.")
+
+
+def main_loop():
+    print("[INFO] Bot 起動。10秒ごとにチェックします。")
     while True:
         try:
-            process()
+            messages = fetch_messages(GROUP_ID_TO_WATCH)
         except Exception as e:
-            print(f"[ERROR] {e}")
+            print(f"[ERROR] メッセージ取得に失敗しました: {e}")
+            # エラーでも sleep して再試行
+            time.sleep(10)
+            continue
+
+        try:
+            process_once(messages)
+        except Exception as e:
+            print(f"[ERROR] 処理中に例外が発生しました: {e}")
+
         time.sleep(10)
 
+
 if __name__ == "__main__":
-    main()
+    main_loop()