Update join.py

join.py

@@ -1,215 +1,408 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-
-import os
 import time
 import requests
-from typing import List, Dict, Any, Optional
-
-# ===== 設定 =====
-X_ACCOUNT = os.getenv("dmsendertoken")
-if not X_ACCOUNT:
-    raise RuntimeError("環境変数 dmsendertoken が設定されていません")
-
-BASE = "https://desk-api.channel.io/desk/channels/240996"
-GROUP_ID = 574317
-CHECK_PERSON_ID = "655217"
-
-# accountIdベースのブラックリスト
-BLACKLIST_ACCOUNT_IDS = ["486092", "471246"]
+import os
+import json
+from datetime import datetime, timedelta
+from zoneinfo import ZoneInfo
+import websocket
+import threading
+import rel
+
+BASE_URL = "https://desk-api.channel.io/desk/channels/240996"
+API_URL = "https://api.channel.works/desk/channels/240996"
+TARGET_GROUP_CHAT_ID = "574628"
+WEBSOCKET_URL = "wss://desk-ws.channel.io/socket.io/?platform=web&EIO=4&transport=websocket"
 
 HEADERS = {
-    "accept": "application/json",
     "accept-language": "ja",
-    "content-type": "application/json",
-    "x-account": X_ACCOUNT,
+    "x-account": os.getenv("dmsendertoken")
 }
 
-GET_PARAMS = {
-    "sortOrder": "desc",
-    "limit": 34,
-    "logFolded": "false",
+JST = ZoneInfo("Asia/Tokyo")
+
+# WebSocket接続管理用
+ws = None
+ws_connected = False
+
+# ---------------- 天気コード ----------------
+
+WEATHER_CODE_MAP = {
+  0: "雲の発達なしまたは観測なし",
+  1: "雲が消散または弱まる",
+  2: "変化なし",
+  3: "雲が発生または発達",
+  4: "煙による視程低下",
+  5: "煙霧",
+  6: "広範囲の砂塵",
+  7: "風で砂や塵が舞う",
+  8: "砂塵旋風",
+  9: "砂嵐または塵嵐",
+  10: "霧",
+  11: "🌫️（斑状）",
+  12: "🌫️（連続）",
+  13: "⚡",
+  14: "⛆",
+  15: "⛆",
+  16: "⛆",
+  17: "⛈（降水なし）",
+  18: "スコール",
+  19: "漏斗雲",
+  20: "霧雨または氷晶",
+  21: "⛆",
+  22: "❄",
+  23: "雨と雪または氷あられ",
+  24: "凍る雨または霧雨",
+  25: "にわか雨",
+  26: "にわか雪または雨雪",
+  27: "にわか雹",
+  28: "🌫️",
+  29: "⛈",
+  30: "弱い砂嵐（減少）",
+  31: "弱い砂嵐（変化なし）",
+  32: "弱い砂嵐（増加）",
+  33: "激しい砂嵐（減少）",
+  34: "激しい砂嵐（変化なし）",
+  35: "激しい砂嵐（増加）",
+  36: "弱い地吹雪（低）",
+  37: "激しい地吹雪（低）",
+  38: "弱い地吹雪（高）",
+  39: "激しい地吹雪（高）",
+  40: "遠方の霧",
+  41: "斑状の霧",
+  42: "霧（薄くなった・空見える）",
+  43: "霧（薄くなった・空見えない）",
+  44: "霧（変化なし・空見える）",
+  45: "霧（変化なし・空見えない）",
+  46: "霧（濃くなった・空見える）",
+  47: "霧（濃くなった・空見えない）",
+  48: "着氷性の霧（空見える）",
+  49: "着氷性の霧（空見えない）",
+  50: "弱い霧雨（断続）",
+  51: "弱い霧雨（連続）",
+  52: "中程度の霧雨（断続）",
+  53: "中程度の霧雨（連続）",
+  54: "強い霧雨（断続）",
+  55: "強い霧雨（連続）",
+  56: "凍る霧雨（弱い）",
+  57: "凍る霧雨（強い）",
+  58: "霧雨と雨（弱い）",
+  59: "霧雨と雨（強い）",
+  60: "弱い雨（断続）",
+  61: "弱い雨（連続）",
+  62: "中程度の雨（断続）",
+  63: "中程度の雨（連続）",
+  64: "強い雨（断続）",
+  65: "強い雨（連続）",
+  66: "凍る雨（弱い）",
+  67: "凍る雨（強い）",
+  68: "雨または霧雨と雪（弱い）",
+  69: "雨または霧雨と雪（強い）",
+  70: "弱い雪（断続）",
+  71: "弱い雪（連続）",
+  72: "中程度の雪（断続）",
+  73: "中程度の雪（連続）",
+  74: "強い雪（断続）",
+  75: "強い雪（連続）",
+  76: "ダイヤモンドダスト",
+  77: "氷晶",
+  78: "星形雪結晶",
+  79: "氷あられ",
+  80: "弱いにわか雨",
+  81: "強いにわか雨",
+  82: "猛烈なにわか雨",
+  83: "弱い雨雪にわか",
+  84: "強い雨雪にわか",
+  85: "弱いにわか雪",
+  86: "強いにわか雪",
+  87: "弱い氷あられにわか",
+  88: "強い氷あられにわか",
+  89: "弱い雹にわか",
+  90: "強い雹にわか",
+  91: "弱い雨＋直前雷雨",
+  92: "強い雨＋直前雷雨",
+  93: "弱い雪等＋直前雷雨",
+  94: "強い雪等＋直前雷雨",
+  95: "雷雨（雹なし）",
+  96: "雷雨（雹あり）",
+  97: "激しい雷雨（雹なし）",
+  98: "雷雨＋砂嵐",
+  99: "激しい雷雨（雹あり）"
 }
 
-WELCOME_TEMPLATE = """
-こんにちは。ITRSAにようこそ。
-自動でいくつかの部屋に招待しています。
-<b>アイコンや名前の設定方法：</b>
-左下の自身のアイコン画像をクリックし、「個人設定」から変更できます。
-<b>使い方：</b>
-左側のメニューから、部屋(グループ)を開くことができます。
-・「あいてぃーあーるえすえい」は本部で、雑���などをする部屋です。
-・「Youtubeダウンローダー」は、URLを送ると自動でYoutubeをダウンロードしてくれる部屋です。
-・「チャッピーくん」は、ChatGPTと話せる部屋です。
-・「幹部と裁判所からの報告」はここを管理する幹部や裁判所から通知が来る部屋です。この部屋は基本的に抜けないでください。
-"""
-
-INVITE_GROUPS = [
-    583390,
-    574628,
-    574793,
-    574744,
-    574717,
-]
-
-# ===== 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 format_weather_code(code: int) -> str:
+    if code < 10:
+        return WEATHER_CODE_MAP.get(code, str(code))
 
+    main = code // 10
+    sub = code % 10
 
-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()
+    main_str = WEATHER_CODE_MAP.get(main)
+    sub_str = WEATHER_CODE_MAP.get(sub)
 
-    groups = response.json().get("groups", [])
-    return {
-        group.get("id"): group.get("title")
-        for group in groups
-        if "id" in group and "title" in group
-    }
+    if main_str and sub_str:
+        return f"{main_str}、{sub_str}"
 
+    return str(code)
 
-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
-    }
+# ---------------- WebSocket処理 ----------------
 
-    r = requests.get(url, headers=HEADERS, params=params, timeout=20)
-    r.raise_for_status()
+def on_message(ws, message):
+    """WebSocketメッセージ受信時の処理"""
+    global ws_connected
+    
+    try:
+        # Socket.IOプロトコルの処理
+        if message == "2":
+            # ping対応
+            ws.send("3")
+            return
+        
+        if message == "3":
+            # pong
+            return
+        
+        if message.startswith("0{"):
+            # 接続確立時の初期メッセージ
+            # 認証情報を送信
+            jwt_token = os.getenv("dmsendertoken")
+            auth_message = f'40/desk/channel,{{"channelId":"240996","jwt":"{jwt_token}"}}'
+            ws.send(auth_message)
+            ws_connected = True
+            print("WebSocket: 認証情報を送信しました")
+            return
+        
+        if message.startswith("40"):
+            # 認証成功
+            print("WebSocket: 認証成功")
+            return
+        
+        if message.startswith("42"):
+            # メッセージデータの処理
+            # "42"の後にJSON配列が続く
+            json_str = message[2:]
+            data = json.loads(json_str)
+            
+            if isinstance(data, list) and len(data) >= 2:
+                event_type = data[0]
+                event_data = data[1]
+                
+                if event_type == "push":
+                    # メッセージ受信
+                    handle_push_event(event_data)
+                elif event_type == "create":
+                    # ユーザー参加
+                    handle_create_event(event_data)
+                    
+    except Exception as e:
+        print(f"WebSocket メッセージ処理エラー: {e}")
+
+def handle_push_event(event_data):
+    """pushイベントの処理"""
+    try:
+        entity = event_data.get("entity", {})
+        chat_id = entity.get("chatId")
+        chat_type = entity.get("chatType")
+        person_type = entity.get("personType")
+        message_text = entity.get("plainText", "")
+        
+        if chat_type == "group" and chat_id == TARGET_GROUP_CHAT_ID:
+            print(f"グループメッセージ受信: {message_text}")
+            # ここで受信メッセージに対する処理を実装
+            # 例: コマンド処理など
+            
+    except Exception as e:
+        print(f"push イベント処理エラー: {e}")
+
+def handle_create_event(event_data):
+    """createイベントの処理（ユーザー参加など）"""
+    try:
+        entity = event_data.get("entity", {})
+        log = entity.get("log", {})
+        chat_id = entity.get("chatId")
+        person_id = entity.get("personId")
+        
+        if log.get("action") == "join" and chat_id == TARGET_GROUP_CHAT_ID:
+            print(f"ユーザーが参加しました: {person_id}")
+            # 参加メッセージなどを送信する場合はここで処理
+            
+    except Exception as e:
+        print(f"create イベント処理エラー: {e}")
+
+def on_error(ws, error):
+    print(f"WebSocket エラー: {error}")
+
+def on_close(ws, close_status_code, close_msg):
+    global ws_connected
+    ws_connected = False
+    print("WebSocket 接続が閉じられました")
+
+def on_open(ws):
+    print("WebSocket 接続が開かれました")
+
+def start_websocket():
+    """WebSocket接続を開始"""
+    websocket.enableTrace(False)
+    ws = websocket.WebSocketApp(
+        WEBSOCKET_URL,
+        on_open=on_open,
+        on_message=on_message,
+        on_error=on_error,
+        on_close=on_close
+    )
+    
+    # 永続的に接続を維持
+    ws.run_forever(
+        ping_interval=30,
+        ping_timeout=10,
+        reconnect=5
+    )
 
-    data = r.json()
-    managers = data.get("managers", [])
+# ---------------- メッセージ送信 ----------------
 
-    if not managers:
-        return None
+def post_group_message(chat_id, body):
+    """WebSocket APIを使用してグループメッセージを送信"""
+    url = f"{API_URL}/groups/{chat_id}/messages"
+    r = requests.post(url, headers=HEADERS, json=body)
+    r.raise_for_status()
+    return r.json()
 
-    return str(managers[0].get("accountId"))
+# ---------------- 時報 ----------------
 
+def create_time_signal_body():
+    now = datetime.now(JST)
+    hour = now.hour
 
-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} を削除しました")
+    if hour < 12:
+        period = "午前"
+        display_hour = hour if hour != 0 else 12
+    else:
+        period = "午後"
+        display_hour = hour - 12 if hour > 12 else 12
 
+    message = f"<b>時報</b>\n{period}{display_hour}時をお知らせします。"
 
-def post_welcome_message() -> None:
-    TITLES = get_group_titles()
+    return {
+        "requestId": f"desk-web-time-signal-{int(time.time())}",
+        "blocks": [{"type": "text", "value": message}],
+    }
 
-    welcome_text = WELCOME_TEMPLATE.format(
-        hq=TITLES.get(532214, "本部")
+# ---------------- 天気予報 ----------------
+
+def load_todoufuken():
+    with open("todoufuken.json", encoding="utf-8") as f:
+        lines = []
+        for line in f:
+            stripped = line.lstrip()
+            if stripped.startswith("//"):
+                continue
+            lines.append(line)
+
+    json_str = "".join(lines)
+    return json.loads(json_str)
+
+
+def fetch_weather(todoufuken):
+    lats = ",".join(str(p["lat"]) for p in todoufuken)
+    lons = ",".join(str(p["lon"]) for p in todoufuken)
+
+    url = (
+        "https://api.open-meteo.com/v1/forecast"
+        f"?latitude={lats}"
+        f"&longitude={lons}"
+        "&hourly=relative_humidity_2m,apparent_temperature,temperature_2m,"
+        "precipitation_probability,weather_code"
+        "&forecast_days=1"
+        "&timezone=Asia%2FTokyo"
     )
 
-    url = f"{BASE}/groups/{GROUP_ID}/messages"
-    payload = {
-        "requestId": f"desk-web-{int(time.time() * 1000)}",
-        "blocks": [
-            {"type": "text", "value": welcome_text}
-        ]
+    r = requests.get(url)
+    r.raise_for_status()
+    return r.json()
+
+
+def create_weather_body():
+    now = datetime.now(JST)
+    today_7am = now.replace(hour=7, minute=0, second=0, microsecond=0)
+
+    todoufuken = load_todoufuken()
+    data = fetch_weather(todoufuken)
+
+    lines = ["<b>☀️ 本日の天気予報（7時以降・1時間ごと）</b>"]
+
+    for idx, pref in enumerate(todoufuken):
+        hourly = data[idx]["hourly"]
+        times = hourly["time"]
+
+        lines.append(f"\n<b>{pref['name']}：</b>")
+
+        TARGET_HOURS = {7, 10, 13, 16, 21}
+        
+        for i, t in enumerate(times):
+            t_jst = (
+                datetime.fromisoformat(t)
+                .replace(tzinfo=ZoneInfo("UTC"))
+                .astimezone(JST)
+            )
+        
+            if t_jst.hour not in TARGET_HOURS:
+                continue
+
+            wc = hourly["weather_code"][i]
+            at = hourly["apparent_temperature"][i]
+            temp = hourly["temperature_2m"][i]
+            pop = hourly["precipitation_probability"][i]
+            rh = hourly["relative_humidity_2m"][i]
+
+            wc_str = format_weather_code(wc)
+
+            line = (
+                f"　{t_jst.strftime('%H時')}："
+                f"{wc_str} "
+                f"{at}℃ {temp}℃ {pop}% {rh}%"
+            )
+            lines.append(line)
+    message2 = "☀️・・・晴れ、☁️・・・曇り、🌫️・・・霧・霧雨、🌧️・・・雨、🌨️・・・雪、🌦️・・・にわか雨、⛈️・・・雷雨"
+    return {
+        "requestId": f"desk-web-weather-{int(time.time())}",
+        "blocks": [{"type": "text", "value": "\n".join(lines)},{"type": "text", "value": message2}],
     }
 
-    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", ""))
+def main_loop():
+    last_time_signal = None
+    last_weather = None
     
-        # ===== ここを追加 =====
-        # CHECK_PERSON_ID の最新発言より前なら無視
-        if created_at <= latest_check_person:
-            continue
-    
-        # ===== ブラックリスト判定（accountIdベース）=====
+    while True:
         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
+            now = datetime.now(JST)
+            current_date = now.date()
 
-    join_targets = list(set(join_targets))
+            # 毎正時の時報
+            if now.minute == 0 and last_time_signal != now.hour:
+                body = create_time_signal_body()
+                post_group_message(TARGET_GROUP_CHAT_ID, body)
+                last_time_signal = now.hour
+                print(f"時報を送信しました: {now.strftime('%H:%M')}")
 
-    # 歓迎メッセージは1回だけ
-    post_welcome_message()
+            # 午前11時30分に天気予報
+            if now.hour == 11 and now.minute == 30 and last_weather != current_date:
+                body = create_weather_body()
+                post_group_message(TARGET_GROUP_CHAT_ID, body)
+                last_weather = current_date
+                print("天気予報を送信しました")
 
-    # 招待処理
-    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}")
+            time.sleep(30)  # 30秒間隔でチェック
 
-
-def main():
-    print("[INFO] Bot 起動（10秒間隔）")
-    while True:
-        try:
-            process()
         except Exception as e:
-            print(f"[ERROR] {e}")
-        time.sleep(10)
+            print("Error:", e)
+            time.sleep(10)
 
 
 if __name__ == "__main__":
-    main()
+    # WebSocket接続を別スレッドで開始
+    ws_thread = threading.Thread(target=start_websocket, daemon=True)
+    ws_thread.start()
+    
+    # メインループを実行
+    main_loop()