Update join.py

join.py

@@ -1,201 +1,118 @@
 #!/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, Optional, Set
+from typing import List, Dict, Any, Set
 
-# --- 設定 ---
+# ===== 設定 =====
 X_ACCOUNT = os.getenv("dmsendertoken")
 if not X_ACCOUNT:
     raise RuntimeError("環境変数 dmsendertoken が設定されていません。")
 
-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}"
+BASE_MESSAGES = "https://desk-api.channel.io/desk/channels/200605/groups/463667/messages"
+INVITE_GROUPS = [519217, 536194, 534868]
 
-HEADERS_GET = {
-    "accept": "application/json",
-    "accept-language": "ja",
-    "x-account": X_ACCOUNT,
-}
-HEADERS_POST = {
+WELCOME_TEXT = "こんにちは。ITRSAにようこそ。自動でいくつかの部屋に招待しています。"
+
+CHECK_INTERVAL = 10
+BOT_PERSON_ID = "599642"
+
+HEADERS = {
     "accept": "application/json",
     "accept-language": "ja",
     "content-type": "application/json",
     "x-account": X_ACCOUNT,
 }
 
-TARGET_PERSON_ID = "604730"  # チェック対象の personId（文字列比較）
-INVITE_GROUP_IDS = [463667]
-
-WELCOME_MESSAGE_TEXT = "こんにちは。ITRSAにようこそ。自動でいくつかの部屋に招待しています。"
-
-# --- ヘルパー ---
-
+PARAMS = {
+    "sortOrder": "desc",
+    "limit": 34,
+    "logFolded": "false",
+}
 
-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)
+# ===== API ユーティリティ =====
+def get_messages() -> List[Dict[str, Any]]:
+    r = requests.get(BASE_MESSAGES, headers=HEADERS, params=PARAMS, timeout=20)
     r.raise_for_status()
-    data = r.json()
-    msgs = data.get("messages", [])
-    if not isinstance(msgs, list):
-        return []
-    return msgs
+    return r.json().get("messages", [])
 
-
-def send_welcome_message(group_id: int, text: str) -> None:
-    url = BASE_POST.format(group_id=group_id)
+def post_welcome_message() -> None:
     payload = {
         "requestId": f"desk-web-{int(time.time() * 1000)}",
-        "blocks": [{"type": "text", "value": text}],
+        "blocks": [{"type": "text", "value": WELCOME_TEXT}],
     }
-    r = requests.post(url, headers=HEADERS_POST, json=payload, timeout=30)
+    r = requests.post(BASE_MESSAGES, headers=HEADERS, json=payload, timeout=20)
     r.raise_for_status()
-
-
-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)
+    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"
+    r = requests.post(
+        url,
+        headers=HEADERS,
+        params={"managerIds": manager_id},
+        timeout=20,
+    )
     r.raise_for_status()
+    print(f"[INFO] group {group_id} に personId={manager_id} を招待")
 
+# ===== メイン処理 =====
+def process():
+    messages = get_messages()
 
-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:
-        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
+    # 新しい順で処理されている前提
+    join_candidates: List[Dict[str, Any]] = []
 
-    invited_manager_ids: Set[str] = set()
-    welcome_sent = False
+    # 599642 の最新発言時刻
+    latest_bot_created_at = None
+    for m in messages:
+        if str(m.get("personId")) == BOT_PERSON_ID:
+            latest_bot_created_at = int(m.get("createdAt") or 0)
+            break
 
-    # messages は API から降順 (新しい順) で来るはずだが、安全のためそのまま扱う。
     for msg in messages:
-        try:
-            created_at = int(msg.get("createdAt") or 0)
-        except Exception:
-            created_at = 0
+        created_at = int(msg.get("createdAt") or 0)
 
-        join_person_ids = extract_join_person_ids_from_message(msg)
-        if not join_person_ids:
+        # bot発言より古い join は対象外
+        if latest_bot_created_at and created_at <= latest_bot_created_at:
             continue
 
-        # この join が起きたメッセージ以降に personId==599642 のメッセージがあるか確認
-        if any_person599642_after(created_at, messages):
-            # あるならこの join はスキップ（歓迎メッセージが既にある/今後来る想定のため）
-            continue
+        log = msg.get("log") or {}
+        if log.get("action") == "join":
+            join_candidates.append(msg)
 
-        # 招待対象 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:
-            messages = fetch_messages(GROUP_ID_TO_WATCH)
-        except Exception as e:
-            print(f"[ERROR] メッセージ取得に失敗しました: {e}")
-            # エラーでも sleep して再試行
-            time.sleep(10)
-            continue
+    if not join_candidates:
+        return
 
-        try:
-            process_once(messages)
-        except Exception as e:
-            print(f"[ERROR] 処理中に例外が発生しました: {e}")
+    # join した personId を重複なしで集める
+    join_person_ids: Set[str] = {
+        str(m.get("personId"))
+        for m in join_candidates
+        if m.get("personId")
+    }
+
+    if not join_person_ids:
+        return
 
-        time.sleep(10)
+    # === 招待 ===
+    for pid in join_person_ids:
+        for gid in INVITE_GROUPS:
+            invite_to_group(gid, pid)
 
+    # === 歓迎メッセージ��1回のみ） ===
+    post_welcome_message()
+
+# ===== ループ =====
+def main():
+    print("[INFO] Join監視Bot 起動")
+    while True:
+        try:
+            process()
+        except Exception as e:
+            print(f"[ERROR] 処理中に例外: {e}")
+        time.sleep(CHECK_INTERVAL)
 
 if __name__ == "__main__":
-    main_loop()
+    main()