Create orion-server/join.py

orion-server/join.py

@@ -0,0 +1,205 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import os
+import time
+import requests
+from typing import List, Dict, Any, Optional
+
+# ===== 設定 =====
+X_ACCOUNT = os.getenv("orionchanneltalktoken")
+if not X_ACCOUNT:
+    raise RuntimeError("環境変数 orionchanneltalktoken が設定されていません")
+
+BASE = "https://desk-api.channel.io/desk/channels/232082"
+GROUP_ID = 548829
+CHECK_PERSON_ID = "619110"
+
+# accountIdベースのブラックリスト
+BLACKLIST_ACCOUNT_IDS = []
+
+HEADERS = {
+    "accept": "application/json",
+    "accept-language": "ja",
+    "content-type": "application/json",
+    "x-account": X_ACCOUNT,
+}
+
+GET_PARAMS = {
+    "sortOrder": "desc",
+    "limit": 34,
+    "logFolded": "false",
+}
+
+WELCOME_TEMPLATE = """
+こんにちは。おりおんるーむにようこそ。
+自動でいくつかの部屋に招待しています。
+"""
+
+INVITE_GROUPS = [
+    548834,
+    548832,
+    548833
+]
+
+# ===== API =====
+
+def get_messages() -> List[Dict[str, Any]]:
+    url = f"{BASE}/groups/{GROUP_ID}/messages"
+    r = requests.get(url, headers=HEADERS, params=GET_PARAMS, timeout=20)
+    r.raise_for_status()
+    return r.json().get("messages", [])
+
+
+def get_group_titles() -> Dict[int, str]:
+    url = f"{BASE}/groups?limit=1000"
+    response = requests.get(url, headers=HEADERS, timeout=20)
+    response.raise_for_status()
+
+    groups = response.json().get("groups", [])
+    return {
+        group.get("id"): group.get("title")
+        for group in groups
+        if "id" in group and "title" in group
+    }
+
+
+def get_manager_account_id(person_id: str) -> Optional[str]:
+    """
+    personId を使って managers API を取得し、
+    managers[0].accountId を返す
+    """
+    url = f"{BASE}/managers"
+    params = {
+        "limit": 1,
+        "since": person_id
+    }
+
+    r = requests.get(url, headers=HEADERS, params=params, timeout=20)
+    r.raise_for_status()
+
+    data = r.json()
+    managers = data.get("managers", [])
+
+    if not managers:
+        return None
+
+    return str(managers[0].get("accountId"))
+
+
+def blacklist_manager(person_id: str) -> None:
+    """
+    personId を使って manager を削除
+    """
+    url = f"{BASE}/managers/{person_id}"
+    requests.delete(url, headers=HEADERS, timeout=20).raise_for_status()
+    payload = {
+        "requestId": f"desk-web-{int(time.time() * 1000)}",
+        "blocks": [
+            {"type": "text", "value": "さようなら。"}
+        ]
+    }
+    url = f"{BASE}/groups/{GROUP_ID}/messages"
+    requests.post(url, headers=HEADERS, json=payload, timeout=20).raise_for_status()
+    print(f"[INFO] personId={person_id} を削除しました")
+
+
+def post_welcome_message() -> None:
+    TITLES = get_group_titles()
+
+    welcome_text = WELCOME_TEMPLATE.format(
+        hq=TITLES.get(532214, "本部")
+    )
+
+    url = f"{BASE}/groups/{GROUP_ID}/messages"
+    payload = {
+        "requestId": f"desk-web-{int(time.time() * 1000)}",
+        "blocks": [
+            {"type": "text", "value": welcome_text}
+        ]
+    }
+
+    requests.post(url, headers=HEADERS, json=payload, timeout=20).raise_for_status()
+    print("[INFO] 歓迎メッセージを送信しました")
+
+
+def invite_person(group_id: int, person_id: str) -> None:
+    url = f"{BASE}/groups/{group_id}/invite"
+    params = {"managerIds": person_id}
+    requests.post(url, headers=HEADERS, params=params, timeout=20).raise_for_status()
+    print(f"[INFO] personId={person_id} を group {group_id} に招待しました")
+
+
+# ===== ロジック =====
+
+def process():
+    messages = get_messages()
+
+    # CHECK_PERSON_ID の最新発言時刻
+    latest_check_person = max(
+        (int(m.get("createdAt", 0))
+         for m in messages
+         if str(m.get("personId")) == CHECK_PERSON_ID),
+        default=0
+    )
+
+    join_targets = []
+    
+    for m in messages:
+        log = m.get("log") or {}
+        if log.get("action") != "join":
+            continue
+    
+        created_at = int(m.get("createdAt", 0))
+        person_id = str(m.get("personId", ""))
+    
+        # ===== ここを追加 =====
+        # CHECK_PERSON_ID の最新発言より前なら無視
+        if created_at <= latest_check_person:
+            continue
+    
+        # ===== ブラックリスト判定（accountIdベース）=====
+        try:
+            account_id = get_manager_account_id(person_id)
+        except Exception as e:
+            print(f"[ERROR] manager取得失敗 personId={person_id} : {e}")
+            continue
+    
+        if account_id and account_id in BLACKLIST_ACCOUNT_IDS:
+            try:
+                blacklist_manager(person_id)
+            except Exception as e:
+                print(f"[ERROR] 削除失敗 personId={person_id} : {e}")
+            continue
+    
+        # ===== 通常処理 =====
+        join_targets.append(person_id)
+    if not join_targets:
+        return
+
+    join_targets = list(set(join_targets))
+
+    # 歓迎メッセージは1回���け
+    post_welcome_message()
+
+    # 招待処理
+    for pid in join_targets:
+        for gid in INVITE_GROUPS:
+            try:
+                invite_person(gid, pid)
+            except Exception as e:
+                print(f"[ERROR] 招待失敗 personId={pid} group={gid} : {e}")
+
+
+def main():
+    print("[INFO] Bot 起動（10秒間隔）")
+    while True:
+        try:
+            process()
+        except Exception as e:
+            print(f"[ERROR] {e}")
+        time.sleep(10)
+
+
+if __name__ == "__main__":
+    main()