any-env-code / join.py
izuemon's picture
Update join.py
12f6947 verified
Raw
History Blame Contribute Delete
37.7 kB
import time
import requests
import os
import json
import sys
import traceback
from datetime import datetime, timedelta
from zoneinfo import ZoneInfo
import websocket
import threading
import logging
import ssl
import base64
import jwt as pyjwt # PyJWTライブラリをインストール: pip install PyJWT
# ロギングの設定
logging.basicConfig(
level=logging.DEBUG,
format='%(asctime)s [%(levelname)s] %(name)s: %(message)s',
handlers=[
logging.FileHandler('bot.log', encoding='utf-8'),
logging.StreamHandler(sys.stdout)
]
)
logger = logging.getLogger(__name__)
BASE_URL = "https://desk-api.channel.io/desk/channels/240996"
API_URL = "https://api.channel.works/desk/channels/240996"
TARGET_GROUP_CHAT_ID = "574628"
TARGET_GROUP_CHAT_ID2 = "574317"
WEBSOCKET_URL = "wss://desk-ws.channel.io/socket.io/?platform=web&EIO=4&transport=websocket"
# 環境変数の確認
DMSENDER_TOKEN = os.getenv("dmsendertoken")
if not DMSENDER_TOKEN:
print("環境変数 'dmsendertoken' が設定されていません")
sys.exit(1)
HEADERS = {
"accept-language": "ja",
"x-account": DMSENDER_TOKEN
}
JST = ZoneInfo("Asia/Tokyo")
# JWT管理クラス
class JWTManager:
"""JWTトークンの管理とリフレッシュを行うクラス"""
def __init__(self, initial_token=None):
self.token = initial_token
self.token_expiry = None
self.refresh_endpoint = "https://api.channel.works/desk/account/auth/refresh"
self.account_id = None
self._parse_token(initial_token)
def _parse_token(self, token):
"""JWTを解析して有効期限を取得"""
if not token:
return
try:
# JWTのペイロード部分をデコード
parts = token.split('.')
if len(parts) != 3:
return
# ペイロードをデコード
payload = parts[1]
# Base64URLデコード用のパディング追加
payload += '=' * (4 - len(payload) % 4)
decoded = base64.urlsafe_b64decode(payload)
payload_data = json.loads(decoded)
# 有効期限(exp)を取得
if 'exp' in payload_data:
self.token_expiry = datetime.fromtimestamp(payload_data['exp'], JST)
# アカウントID(key)を取得
if 'key' in payload_data:
self.account_id = payload_data['key']
print(f"JWT解析: 有効期限={self.token_expiry}, アカウントID={self.account_id}")
except Exception as e:
print(f"JWT解析エラー: {e}")
def is_expired(self):
"""トークンが期限切れかチェック(5分の余裕を持たせる)"""
if not self.token_expiry:
return True
# 5分前を期限として扱う(余裕を持たせる)
buffer_time = timedelta(minutes=5)
return datetime.now(JST) + buffer_time >= self.token_expiry
def should_refresh(self):
"""リフレッシュが必要かチェック(15分以内の期限)"""
if not self.token_expiry:
return True
# 15分以内に期限切れになる場合はリフレッシュ
threshold = timedelta(minutes=15)
return datetime.now(JST) + threshold >= self.token_expiry
def refresh_token(self):
"""トークンをリフレッシュ"""
if not self.token:
print("JWT: リフレッシュに失敗(トークンがありません)")
return None
try:
print(f"JWT: トークンリフレッシュを開始 (有効期限: {self.token_expiry})")
headers = {
"x-account": self.token,
"accept-language": "ja"
}
response = requests.post(
self.refresh_endpoint,
headers=headers,
timeout=10
)
if response.status_code == 200:
data = response.json()
new_token = data.get('jwt') or data.get('token')
if new_token:
print(f"JWT: トークンリフレッシュ成功 (新しい有効期限を解析中...)")
self.token = new_token
self._parse_token(new_token)
return new_token
else:
print(f"JWT: リフレッシュ応答にトークンがありません: {data}")
return None
else:
print(f"JWT: リフレッシュ失敗: status={response.status_code}, body={response.text[:200]}")
return None
except Exception as e:
print(f"JWT: リフレッシュエラー: {e}")
return None
def get_token(self):
"""現在のトークンを返す(必要に応じてリフレッシュ)"""
if self.should_refresh():
print("JWT: リフレッシュが必要なため、更新を試みます")
self.refresh_token()
return self.token
def get_auth_headers(self):
"""認証ヘッダーを返す"""
token = self.get_token()
if token:
return {
"x-account": token,
"accept-language": "ja"
}
return {}
# JWTマネージャーのインスタンスを作成
jwt_manager = JWTManager(DMSENDER_TOKEN)
# WebSocket接続管理用
ws_app = None
ws_connected = False
ws_reconnect_count = 0
MAX_RECONNECT_ATTEMPTS = 10
ws_auth_attempted = False
# 統計情報
stats = {
"messages_sent": 0,
"messages_received": 0,
"errors": 0,
"ws_reconnects": 0,
"token_refreshes": 0,
"start_time": datetime.now(JST)
}
# ウェルカムメッセージのテンプレート
WELCOME_MESSAGE = """🎉 ようこそ!
<b>チャンネルにご参加いただきありがとうございます!</b>
このグループは○○のためのコミュニティスペースです。
ご質問やご相談があれば、いつでもお気軽に投稿してください。
🔹 ルール
- 相互尊重を心がけましょう
- 宣伝やスパムはご遠慮ください
それでは、楽しい時間をお過ごしください!😊"""
# ---------------- JWT関連の関数 ----------------
def refresh_jwt_token():
"""JWTトークンをリフレッシュする(グローバル関数)"""
global HEADERS
print("=== JWTトークンリフレッシュを開始 ===")
new_token = jwt_manager.refresh_token()
if new_token:
HEADERS["x-account"] = new_token
stats["token_refreshes"] += 1
print(f"JWT: トークン更新完了 (リフレッシュ回数: {stats['token_refreshes']})")
return True
else:
print("JWT: トークン更新に失敗しました")
return False
def get_jwt_for_websocket():
"""WebSocket用のJWTを取得(必要な場合はリフレッシュ)"""
token = jwt_manager.get_token()
if not token:
print("JWT: WebSocket用トークン取得失敗")
return None
print(f"JWT: WebSocket用トークン取得成功 (有効期限: {jwt_manager.token_expiry})")
return token
def check_and_refresh_token():
"""定期的にトークンをチェックし、必要ならリフレッシュ"""
if jwt_manager.should_refresh():
print("JWT: 定期チェックでリフレッシュが必要と判断")
return refresh_jwt_token()
return True
# ---------------- ヘッダー更新関数 ----------------
def update_headers_with_new_token():
"""新しいトークンでヘッダーを更新"""
global HEADERS
token = jwt_manager.get_token()
if token:
HEADERS["x-account"] = token
print("ヘッダーを新しいトークンで更新しました")
return True
return False
# ---------------- WebSocket処理 ----------------
def log_stats():
"""統計情報をログ出力"""
uptime = datetime.now(JST) - stats["start_time"]
print(f"=== 統計情報 ===")
print(f"稼働時間: {uptime}")
print(f"送信メッセージ数: {stats['messages_sent']}")
print(f"受信メッセージ数: {stats['messages_received']}")
print(f"エラー数: {stats['errors']}")
print(f"WebSocket再接続回数: {stats['ws_reconnects']}")
print(f"JWTリフレッシュ回数: {stats['token_refreshes']}")
print(f"JWT有効期限: {jwt_manager.token_expiry}")
def on_message(ws, message):
"""WebSocketメッセージ受信時の処理"""
global ws_connected, ws_reconnect_count, ws_auth_attempted
try:
print(f"WebSocket raw message: {message[:200]}")
# Socket.IOプロトコルの処理
if message == "2":
print("WebSocket: ping受信、pong送信")
ws.send("3")
return
if message == "3":
print("WebSocket: pong受信")
return
if message.startswith("0{"):
# 接続確立時の初期メッセージ - Socket.IO v4
print(f"WebSocket: 接続確立、初期メッセージ受信")
try:
init_data = json.loads(message[1:])
ping_interval = init_data.get("pingInterval", 60000)
ping_timeout = init_data.get("pingTimeout", 25000)
print(f"WebSocket: ping間隔={ping_interval}ms, ping timeout={ping_timeout}ms")
# JWTを取得(リフレッシュが必要な場合は自動更新)
jwt_token = get_jwt_for_websocket()
if not jwt_token:
print("WebSocket: JWT tokenが取得できませんでした")
return
auth_message = f'40/desk/channel,{{"channelId":"240996","jwt":"{jwt_token}"}}'
print(f"WebSocket: 認証メッセージ送信: {auth_message[:100]}...")
ws.send(auth_message)
ws_auth_attempted = True
except json.JSONDecodeError as e:
print(f"WebSocket: 初期メッセージ解析エラー: {e}")
stats["errors"] += 1
except Exception as e:
print(f"WebSocket: 認証情報送信エラー: {e}")
stats["errors"] += 1
return
if message.startswith("40"):
print(f"WebSocket: 認証応答受信: {message[:100]}")
ws_connected = True
ws_reconnect_count = 0
ws_auth_attempted = True
print("WebSocket: 認証成功、接続確立")
# 認証成功後にトークンの有効期限をチェック
if jwt_manager.should_refresh():
print("WebSocket: 認証成功後、トークンリフレッシュをスケジュール")
return
if message.startswith("42"):
try:
if "," in message:
json_start = message.find(",") + 1
json_str = message[json_start:]
print(f"WebSocket: Extracted JSON: {json_str[:200]}")
data = json.loads(json_str)
if isinstance(data, list) and len(data) >= 2:
event_type = data[0]
event_data = data[1]
print(f"WebSocket: イベント受信 type={event_type}")
if event_type == "push":
handle_push_event(event_data)
elif event_type == "create":
handle_create_event(event_data)
elif event_type == "update":
handle_update_event(event_data)
elif event_type == "delete":
handle_delete_event(event_data)
elif event_type == "joined":
print(f"WebSocket: 参加確認: {event_data}")
elif event_type == "ready":
print(f"WebSocket: 準備完了: {event_data}")
elif event_type == "refresh":
# サーバーからのリフレッシュ要求
print(f"WebSocket: サーバーからトークンリフレッシュ要求を受信")
handle_token_refresh_request(ws, event_data)
else:
print(f"WebSocket: 未処理のイベントタイプ: {event_type}")
else:
print(f"WebSocket: 予期しないデータ形式: {json_str[:200]}")
except json.JSONDecodeError as e:
print(f"WebSocket: JSON解析エラー: {e}, データ: {message[:200]}")
stats["errors"] += 1
except Exception as e:
print(f"WebSocket: メッセージ処理エラー: {e}")
print(traceback.format_exc())
stats["errors"] += 1
return
if message.startswith("44"):
print(f"WebSocket: エラーメッセージ受信: {message}")
# 認証エラー(トークン期限切れなど)の処理
if "auth" in message.lower() or "token" in message.lower():
print("WebSocket: 認証エラーが発生しました。トークンをリフレッシュします。")
handle_auth_error(ws)
return
print(f"WebSocket: その他のメッセージ: {message[:100]}")
except Exception as e:
print(f"WebSocket on_message エラー: {e}")
print(traceback.format_exc())
stats["errors"] += 1
def handle_token_refresh_request(ws, event_data):
"""サーバーからのトークンリフレッシュ要求を処理"""
try:
print("JWT: サーバーからリフレッシュ要求を受信")
# トークンをリフレッシュ
if refresh_jwt_token():
# 新しいトークンで再認証
new_token = jwt_manager.get_token()
if new_token and ws:
auth_message = f'40/desk/channel,{{"channelId":"240996","jwt":"{new_token}"}}'
print(f"WebSocket: 再認証メッセージ送信")
ws.send(auth_message)
print("JWT: 再認証メッセージ送信完了")
else:
print("JWT: リフレッシュに失敗しました")
except Exception as e:
print(f"JWT: リフレッシュ要求処理エラー: {e}")
stats["errors"] += 1
def handle_auth_error(ws):
"""認証エラー時の処理"""
try:
print("=== 認証エラー処理 ===")
# トークンをリフレッシュ
if refresh_jwt_token():
new_token = jwt_manager.get_token()
if new_token and ws:
# 再接続を試みる
print("WebSocket: 新しいトークンで再接続を試みます")
ws.close()
# 新しいスレッドで再接続
threading.Thread(target=start_websocket, daemon=True).start()
else:
print("認証エラー: トークンリフレッシュに失敗しました")
except Exception as e:
print(f"認証エラー処理中にエラー: {e}")
stats["errors"] += 1
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")
person_id = entity.get("personId")
message_text = entity.get("plainText", "")
message_id = entity.get("id", "unknown")
refers = event_data.get("refers", {})
manager_info = refers.get("manager", {})
manager_name = manager_info.get("name", "不明")
print(f"メッセージ受信: id={message_id}, chatType={chat_type}, chatId={chat_id}, "
f"personType={person_type}, personId={person_id}, from={manager_name}")
stats["messages_received"] += 1
if chat_type == "group" and chat_id == TARGET_GROUP_CHAT_ID2:
print(f"対象グループのメッセージ: '{message_text[:100]}' from {manager_name}")
else:
print(f"対象外のチャットからのメッセージ: chatType={chat_type}, chatId={chat_id}")
except Exception as e:
print(f"push イベント処理エラー: {e}")
print(traceback.format_exc())
stats["errors"] += 1
def handle_create_event(event_data):
"""createイベントの処理(新規ユーザー検出時にウェルカムメッセージ送信)"""
try:
event = event_data.get("event")
event_type = event_data.get("type")
entity = event_data.get("entity", {})
print(f"createイベント: event={event}, type={event_type}")
# 新規マネージャー(ユーザー)が参加した場合
if event_type == "manager":
manager_id = entity.get("managerId") or entity.get("id")
name = entity.get("name")
email = entity.get("email")
print("=== 新しいマネージャーが参加 ===")
print(f"ID : {manager_id}")
print(f"名前 : {name}")
print(f"メール : {email}")
# ウェルカムメッセージを送信
if name:
try:
welcome_body = create_welcome_message(name)
post_group_message(TARGET_GROUP_CHAT_ID, welcome_body)
print(f"ウェルカムメッセージ送信完了: {name} (ID: {manager_id})")
except Exception as e:
print(f"ウェルカムメッセージ送信エラー: {e}")
stats["errors"] += 1
# ユーザーチャット作成(新規ユーザーからのメッセージ)
elif event_type == "userChat":
chat_id = entity.get("id")
user_id = entity.get("userId")
person_type = entity.get("personType", "user")
message = entity.get("message", {})
print("=== 新規ユーザーチャット ===")
print(f"チャットID: {chat_id}")
print(f"ユーザーID: {user_id}")
print(f"メッセージ: {message.get('plainText', '')[:100]}")
# ウェルカムメッセージを送信(ユーザーチャット作成時も)
try:
welcome_body = create_welcome_message("新規ユーザー")
post_group_message(TARGET_GROUP_CHAT_ID, welcome_body)
print(f"ウェルカムメッセージ送信完了: ユーザー {user_id}")
except Exception as e:
print(f"ウェルカムメッセージ送信エラー: {e}")
stats["errors"] += 1
else:
print(f"未処理のcreateイベント: {event_type}")
except Exception as e:
print(f"createイベント処理エラー: {e}")
print(traceback.format_exc())
stats["errors"] += 1
def handle_update_event(event_data):
"""updateイベントの処理(チャットセッションの更新など)"""
try:
event = event_data.get("event")
entity = event_data.get("entity", {})
event_type = event_data.get("type", "unknown")
print(f"update イベント: event={event}, type={event_type}")
if event_type == "chatSession":
chat_id = entity.get("chatId")
chat_key = entity.get("chatKey")
unread = entity.get("unread", 0)
alert = entity.get("alert", 0)
updated_at = entity.get("updatedAt")
print(f" チャットセッション更新: chatId={chat_id}, chatKey={chat_key}, "
f"unread={unread}, alert={alert}")
if chat_id == TARGET_GROUP_CHAT_ID:
print(f" 対象グループのセッションが更新されました (unread={unread})")
elif event_type == "managerBadge":
manager_id = entity.get("managerId")
team_chat_unread = entity.get("teamChatUnread", 0)
user_chat_unread = entity.get("userChatUnread", 0)
print(f" マネージャーバッジ更新: managerId={manager_id}, "
f"teamChatUnread={team_chat_unread}, userChatUnread={user_chat_unread}")
elif event_type == "group":
group_id = entity.get("id")
group_name = entity.get("name", "unknown")
print(f" グループ更新: id={group_id}, name={group_name}")
elif event_type == "manager":
manager_id = entity.get("id")
manager_name = entity.get("name", "unknown")
print(f" マネージャー更新: id={manager_id}, name={manager_name}")
else:
print(f" その他のupdateイベント: type={event_type}, event={event}")
except Exception as e:
print(f"update イベント処理エラー: {e}")
print(traceback.format_exc())
stats["errors"] += 1
def handle_delete_event(event_data):
"""deleteイベントの処理(オンライン状態の削除など)"""
try:
event = event_data.get("event")
entity = event_data.get("entity", {})
event_type = event_data.get("type", "unknown")
print(f"delete イベント: event={event}, type={event_type}")
if event_type == "online":
entity_id = entity.get("id")
person_id = entity.get("personId")
person_type = entity.get("personType")
channel_id = entity.get("channelId")
print(f" オンライン状態削除: id={entity_id}, personId={person_id}, "
f"personType={person_type}, channelId={channel_id}")
if person_type == "manager":
print(f" マネージャーがオフラインになりました: {person_id}")
elif event_type == "manager":
manager_id = entity.get("id")
manager_name = entity.get("name", "unknown")
removed = entity.get("removed", False)
if removed:
print(f" マネージャー削除: id={manager_id}, name={manager_name}")
else:
print(f" マネージャー削除イベント: id={manager_id}, name={manager_name}")
elif event_type == "group":
group_id = entity.get("id")
group_name = entity.get("name", "unknown")
print(f" グループ削除: id={group_id}, name={group_name}")
else:
print(f" その他のdeleteイベント: type={event_type}, event={event}")
except Exception as e:
print(f"delete イベント処理エラー: {e}")
print(traceback.format_exc())
stats["errors"] += 1
def on_error(ws, error):
"""WebSocketエラー時の処理"""
print(f"WebSocket エラー: {error}")
print(f"エラータイプ: {type(error)}")
if hasattr(error, '__class__'):
print(f"エラークラス: {error.__class__.__name__}")
stats["errors"] += 1
error_str = str(error)
if "Connection to remote host was lost" in error_str:
print("WebSocket: リモートホストとの接続が切断されました(再接続されます)")
elif "Connection refused" in error_str:
print("WebSocket: 接続が拒否されました")
elif "handshake" in error_str.lower():
print("WebSocket: ハンドシェイクエラー")
elif "auth" in error_str.lower() or "token" in error_str.lower():
print("WebSocket: 認証エラーが発生しました。トークンをリフレッシュします。")
# 認証エラーの場合はトークンをリフレッシュ
refresh_jwt_token()
else:
print(f"WebSocket: 未分類のエラー: {error_str}")
def on_close(ws, close_status_code, close_msg):
"""WebSocket接続クローズ時の処理"""
global ws_connected, ws_reconnect_count
ws_connected = False
ws_reconnect_count += 1
stats["ws_reconnects"] += 1
print(f"WebSocket 接続が閉じられました: status_code={close_status_code}, "
f"message={close_msg}, reconnect_count={ws_reconnect_count}/{MAX_RECONNECT_ATTEMPTS}")
if close_status_code:
print(f"クローズステータスコード: {close_status_code}")
if close_msg:
print(f"クローズメッセージ: {close_msg}")
# 認証エラーが原因でクローズした場合
if close_msg and ("auth" in close_msg.lower() or "token" in close_msg.lower()):
print("WebSocket: 認証エラーによりクローズしました。トークンをリフレッシュします。")
refresh_jwt_token()
if ws_reconnect_count > MAX_RECONNECT_ATTEMPTS:
print(f"WebSocket: 最大再接続回数({MAX_RECONNECT_ATTEMPTS})を超えました")
def on_open(ws):
"""WebSocket接続開始時の処理"""
global ws_reconnect_count, ws_auth_attempted
print("WebSocket 接続が開かれました")
print(f"接続URL: {WEBSOCKET_URL}")
ws_reconnect_count = 0
ws_auth_attempted = False
# 接続時にトークンの状態を確認
if jwt_manager.should_refresh():
print("WebSocket: 接続時にトークンリフレッシュが必要です")
refresh_jwt_token()
def on_ping(ws, message):
"""WebSocket ping受信時の処理"""
print(f"WebSocket ping受信: {message}")
def on_pong(ws, message):
"""WebSocket pong受信時の処理"""
print(f"WebSocket pong受信: {message}")
def start_websocket():
"""WebSocket接続を開始"""
global ws_app
print(f"WebSocket接続を開始します: {WEBSOCKET_URL}")
try:
websocket.enableTrace(True)
ssl_opt = {
"cert_reqs": ssl.CERT_NONE,
}
ws_app = websocket.WebSocketApp(
WEBSOCKET_URL,
on_open=on_open,
on_message=on_message,
on_error=on_error,
on_close=on_close,
on_ping=on_ping,
on_pong=on_pong
)
ws_app.header = {
"Origin": "https://desk.channel.io",
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36"
}
print("WebSocket: run_foreverを開始します")
ws_app.run_forever(
ping_interval=30,
ping_timeout=10,
ping_payload="2",
sslopt=ssl_opt,
reconnect=5
)
except Exception as e:
print(f"WebSocket接続の開始に失敗: {e}")
print(traceback.format_exc())
stats["errors"] += 1
time.sleep(10)
start_websocket()
# ---------------- メッセージ送信 ----------------
def post_group_message(chat_id, body, retry_count=3):
"""WebSocket APIを使用してグループメッセージを送信(リトライ機能付き)"""
url = f"{API_URL}/groups/{chat_id}/messages"
# 最新のトークンでヘッダーを更新
update_headers_with_new_token()
for attempt in range(retry_count):
try:
print(f"メッセージ送信試行: chat_id={chat_id}, attempt={attempt + 1}/{retry_count}")
print(f"送信URL: {url}")
print(f"送信データ: {json.dumps(body, ensure_ascii=False)[:200]}")
response = requests.post(url, headers=HEADERS, json=body, timeout=10)
print(f"レスポンスステータス: {response.status_code}")
print(f"レスポンスボディ: {response.text[:200]}")
if response.status_code == 200:
response_data = response.json()
print(f"メッセージ送信成功: chat_id={chat_id}")
stats["messages_sent"] += 1
return response_data
elif response.status_code == 401:
print(f"認証エラー: status={response.status_code}")
# トークンをリフレッシュして再試行
if refresh_jwt_token():
update_headers_with_new_token()
if attempt < retry_count - 1:
print("トークンをリフレッシュしました。再試行します。")
continue
raise Exception("認証エラー: APIキーが無効です")
elif response.status_code == 429:
wait_time = (attempt + 1) * 5
print(f"レート制限: {wait_time}秒待機")
if attempt < retry_count - 1:
time.sleep(wait_time)
else:
raise Exception("レート制限により送信失敗")
else:
print(f"送信エラー: status={response.status_code}, body={response.text[:200]}")
if attempt < retry_count - 1:
time.sleep(2)
else:
response.raise_for_status()
except requests.exceptions.Timeout:
print(f"リクエストタイムアウト: attempt={attempt + 1}")
if attempt < retry_count - 1:
time.sleep(2)
except requests.exceptions.ConnectionError as e:
print(f"接続エラー: {e}, attempt={attempt + 1}")
if attempt < retry_count - 1:
time.sleep(5)
except requests.exceptions.RequestException as e:
print(f"リクエストエラー: {e}, attempt={attempt + 1}")
if attempt < retry_count - 1:
time.sleep(2)
except Exception as e:
print(f"予期しない送信エラー: {e}")
print(traceback.format_exc())
stats["errors"] += 1
raise
print("メッセージ送信に失敗しました(全リトライ終了)")
stats["errors"] += 1
raise Exception("メッセージ送信に失敗しました")
# ---------------- ウェルカムメッセージ ----------------
def create_welcome_message(user_name=None):
"""ウェルカムメッセージを作成"""
try:
now = datetime.now(JST)
if user_name:
welcome_text = f"{user_name}さん、ようこそ!🎉\n\n{WELCOME_MESSAGE}"
else:
welcome_text = WELCOME_MESSAGE
message = f"{welcome_text}\n\n({now.strftime('%Y年%m月%d日 %H:%M')}に参加)"
print(f"ウェルカムメッセージ作成: {user_name if user_name else '新規ユーザー'}")
return {
"requestId": f"desk-web-welcome-{int(time.time())}",
"blocks": [{"type": "text", "value": message}],
}
except Exception as e:
print(f"ウェルカムメッセージ作成エラー: {e}")
raise
# ---------------- 時報 ----------------
def create_time_signal_body():
"""時報メッセージを作成"""
try:
now = datetime.now(JST)
hour = now.hour
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}時をお知らせします。"
print(f"時報メッセージ作成: {message}")
return {
"requestId": f"desk-web-time-signal-{int(time.time())}",
"blocks": [{"type": "text", "value": message}],
}
except Exception as e:
print(f"時報メッセージ作成エラー: {e}")
raise
# ---------------- メインループ ----------------
def main_loop():
"""メイン処理ループ"""
print("メインループを開始します")
last_time_signal = None
check_interval = 30
loop_count = 0
token_check_counter = 0
while True:
try:
loop_count += 1
token_check_counter += 1
now = datetime.now(JST)
current_date = now.date()
current_hour = now.hour
current_minute = now.minute
if loop_count % 20 == 0:
print(f"ループ実行中: カウント={loop_count}, 時刻={now.strftime('%Y-%m-%d %H:%M:%S')}, "
f"接続状態={ws_connected}")
# 5分ごとにトークンの状態をチェック
if token_check_counter >= 10: # 30秒 * 10 = 5分
token_check_counter = 0
print(f"JWT: 定期チェック実行 (有効期限: {jwt_manager.token_expiry})")
if jwt_manager.should_refresh():
print("JWT: 定期チェックでリフレッシュが必要です")
refresh_jwt_token()
# WebSocketが接続中なら再認証
if ws_connected and ws_app:
try:
new_token = jwt_manager.get_token()
if new_token:
auth_message = f'40/desk/channel,{{"channelId":"240996","jwt":"{new_token}"}}'
ws_app.send(auth_message)
print("JWT: WebSocketに新しいトークンを送信しました")
except Exception as e:
print(f"JWT: WebSocketへのトークン送信エラー: {e}")
# 毎正時の時報
if current_minute == 0 and last_time_signal != current_hour:
try:
print(f"時報送信: {current_hour}時")
body = create_time_signal_body()
post_group_message(TARGET_GROUP_CHAT_ID, body)
last_time_signal = current_hour
print(f"時報送信完了: {now.strftime('%H:%M')}")
except Exception as e:
print(f"時報送信エラー: {e}")
stats["errors"] += 1
# 1時間ごとに統計情報をログ出力
if current_minute == 0 and current_hour != last_time_signal:
log_stats()
# WebSocket接続状態の監視
if not ws_connected and loop_count % 10 == 0:
print(f"WebSocketが未接続状態です(再接続試行中: {ws_reconnect_count})")
# 再接続を試みる
if ws_reconnect_count < MAX_RECONNECT_ATTEMPTS:
print("WebSocket: 再接続を試みます")
threading.Thread(target=start_websocket, daemon=True).start()
time.sleep(check_interval)
except KeyboardInterrupt:
print("プログラムを終了します(Ctrl+C)")
log_stats()
break
except Exception as e:
print(f"メインループエラー: {e}")
print(traceback.format_exc())
stats["errors"] += 1
time.sleep(10)
if __name__ == "__main__":
try:
print("=== ボット起動 ===")
print(f"開始時刻: {datetime.now(JST).strftime('%Y-%m-%d %H:%M:%S')}")
print(f"ターゲットグループID: {TARGET_GROUP_CHAT_ID}")
print(f"API URL: {API_URL}")
print(f"WebSocket URL: {WEBSOCKET_URL}")
print(f"環境変数 dmsendertoken: {'設定済み' if DMSENDER_TOKEN else '未設定'} ({len(DMSENDER_TOKEN)}文字)")
print(f"JWT有効期限: {jwt_manager.token_expiry}")
print(f"JWTアカウントID: {jwt_manager.account_id}")
# 初期トークンチェック
if jwt_manager.should_refresh():
print("JWT: 初期トークンが期限切れ間近のためリフレッシュします")
refresh_jwt_token()
# WebSocket接続を別スレッドで開始
ws_thread = threading.Thread(target=start_websocket, daemon=True, name="WebSocketThread")
ws_thread.start()
print(f"WebSocketスレッドを開始しました: {ws_thread.name}")
# WebSocketの接続確立を待機
time.sleep(2)
# メインループを実行
main_loop()
except KeyboardInterrupt:
print("プログラムを終了します")
except Exception as e:
print(f"致命的なエラーが発生しました: {e}")
print(traceback.format_exc())
sys.exit(1)
finally:
print("=== ボット終了 ===")
log_stats()