256 lines
12 KiB
Python
256 lines
12 KiB
Python
import discord
|
||
from discord import app_commands
|
||
from discord.ext import commands, tasks
|
||
import datetime
|
||
import os
|
||
import asyncio
|
||
from typing import Optional, List, Dict, Any
|
||
from database import Database
|
||
from dotenv import load_dotenv
|
||
|
||
load_dotenv()
|
||
|
||
# --- Views & Modals ---
|
||
|
||
class TaskModal(discord.ui.Modal, title="新規タスク登録"):
|
||
task_title = discord.ui.TextInput(label="タイトル", placeholder="タスク名", required=True, max_length=100)
|
||
priority = discord.ui.TextInput(label="優先順位 (SSS-D)", placeholder="S", default="A", min_length=1, max_length=3)
|
||
deadline = discord.ui.TextInput(label="締め切り (YYYY-MM-DD HH:MM)", placeholder="2024-12-31 23:59", required=True)
|
||
details = discord.ui.TextInput(label="詳細", style=discord.TextStyle.long, required=False, max_length=1000)
|
||
|
||
def __init__(self, db: Database):
|
||
super().__init__()
|
||
self.db = db
|
||
|
||
async def on_submit(self, interaction: discord.Interaction):
|
||
prio = self.priority.value.upper()
|
||
if prio not in ["SSS", "SS", "S", "A", "B", "C", "D"]:
|
||
await interaction.response.send_message(f"❌ 優先順位が無効です。", ephemeral=True)
|
||
return
|
||
|
||
try:
|
||
deadline_dt = datetime.datetime.strptime(self.deadline.value, "%Y-%m-%d %H:%M")
|
||
except ValueError:
|
||
await interaction.response.send_message("❌ 日時の形式が正しくありません。", ephemeral=True)
|
||
return
|
||
|
||
tid = await self.db.add_task(
|
||
user_id=interaction.user.id,
|
||
user_name=interaction.user.display_name,
|
||
guild_id=interaction.guild_id,
|
||
channel_id=interaction.channel_id,
|
||
title=self.task_title.value,
|
||
description=self.details.value,
|
||
priority=prio,
|
||
deadline=deadline_dt.isoformat()
|
||
)
|
||
await interaction.response.send_message(f"✅ タスク『{self.task_title.value}』(ID: {tid}) を登録しました！", ephemeral=True)
|
||
|
||
class EditTaskModal(discord.ui.Modal):
|
||
def __init__(self, db: Database, task: dict):
|
||
super().__init__(title=f"タスク編集 (ID: {task['id']})")
|
||
self.db = db
|
||
self.task_id = task['id']
|
||
deadline_dt = datetime.datetime.fromisoformat(task['deadline'])
|
||
|
||
self.task_title = discord.ui.TextInput(label="タイトル", default=task['title'], required=True)
|
||
self.priority = discord.ui.TextInput(label="優先順位 (SSS-D)", default=task['priority'], min_length=1, max_length=3)
|
||
self.deadline = discord.ui.TextInput(label="締め切り (YYYY-MM-DD HH:MM)", default=deadline_dt.strftime("%Y-%m-%d %H:%M"), required=True)
|
||
self.details = discord.ui.TextInput(label="詳細", default=task['description'] or "", style=discord.TextStyle.long, required=False)
|
||
|
||
self.add_item(self.task_title)
|
||
self.add_item(self.priority)
|
||
self.add_item(self.deadline)
|
||
self.add_item(self.details)
|
||
|
||
async def on_submit(self, interaction: discord.Interaction):
|
||
prio = self.priority.value.upper()
|
||
if prio not in ["SSS", "SS", "S", "A", "B", "C", "D"]:
|
||
await interaction.response.send_message("❌ 優先順位が無効です。", ephemeral=True)
|
||
return
|
||
try:
|
||
deadline_dt = datetime.datetime.strptime(self.deadline.value, "%Y-%m-%d %H:%M")
|
||
except ValueError:
|
||
await interaction.response.send_message("❌ 日時の形式が正しくありません。", ephemeral=True)
|
||
return
|
||
|
||
await self.db.update_task(self.task_id, self.task_title.value, self.details.value, prio, deadline_dt.isoformat())
|
||
await interaction.response.send_message(f"✨ タスクを更新しました！", ephemeral=True)
|
||
|
||
class TaskActionView(discord.ui.View):
|
||
"""通知やリストに表示する「完了」ボタン"""
|
||
def __init__(self, db: Database, task_id: str):
|
||
super().__init__(timeout=None)
|
||
self.db = db
|
||
self.task_id = task_id
|
||
|
||
@discord.ui.button(label="✅ 完了にする", style=discord.ButtonStyle.success)
|
||
async def complete(self, interaction: discord.Interaction, button: discord.ui.Button):
|
||
await self.db.complete_task(self.task_id)
|
||
# メッセージを「完了済み」に更新
|
||
embed = interaction.message.embeds[0]
|
||
embed.title = f"🏁 【完了】 {embed.title}"
|
||
embed.color = discord.Color.light_grey()
|
||
await interaction.response.edit_message(content="お疲れ様でした！", embed=embed, view=None)
|
||
|
||
class ControlPanel(discord.ui.View):
|
||
def __init__(self, db: Database):
|
||
super().__init__(timeout=None)
|
||
self.db = db
|
||
|
||
@discord.ui.button(label="➕ タスク追加", style=discord.ButtonStyle.success, custom_id="panel_add")
|
||
async def add_btn(self, interaction: discord.Interaction, button: discord.ui.Button):
|
||
await interaction.response.send_modal(TaskModal(self.db))
|
||
|
||
@discord.ui.button(label="📋 タスク一覧", style=discord.ButtonStyle.primary, custom_id="panel_list")
|
||
async def list_btn(self, interaction: discord.Interaction, button: discord.ui.Button):
|
||
tasks_list = await self.db.get_user_tasks(interaction.user.id, interaction.channel_id)
|
||
if not tasks_list:
|
||
await interaction.response.send_message("📋 未完了のタスクはありません。", ephemeral=True)
|
||
return
|
||
|
||
await interaction.response.defer(ephemeral=True)
|
||
for t in tasks_list:
|
||
deadline_fmt = datetime.datetime.fromisoformat(t['deadline']).strftime("%m/%d %H:%M")
|
||
embed = discord.Embed(title=t['title'], color=discord.Color.green())
|
||
embed.add_field(name="優先度", value=t['priority'], inline=True)
|
||
embed.add_field(name="締切", value=deadline_fmt, inline=True)
|
||
if t['description']:
|
||
embed.description = t['description']
|
||
await interaction.followup.send(embed=embed, view=TaskActionView(self.db, t['id']), ephemeral=True)
|
||
|
||
# --- Bot Core ---
|
||
|
||
class TaskBot(commands.Bot):
|
||
def __init__(self, db: Database):
|
||
intents = discord.Intents.default()
|
||
intents.message_content = True
|
||
intents.members = True
|
||
super().__init__(command_prefix="!", intents=intents)
|
||
self.db = db
|
||
|
||
async def setup_hook(self):
|
||
await self.db.initialize()
|
||
self.daily_notification.start()
|
||
self.deadline_monitor.start()
|
||
await self.add_cog(TaskCommands(self))
|
||
print("Setup hook completed.")
|
||
|
||
async def on_ready(self):
|
||
print(f"Logged in as {self.user} (ID: {self.user.id})")
|
||
|
||
async def on_message(self, message: discord.Message):
|
||
if message.author.bot: return
|
||
if self.user.mentioned_in(message):
|
||
await message.reply("📋 **タスク管理パネル**", view=ControlPanel(self.db))
|
||
await self.process_commands(message)
|
||
|
||
@tasks.loop(time=datetime.time(hour=7, minute=0))
|
||
async def daily_notification(self):
|
||
print("Running daily notification...")
|
||
tasks_list = await self.db.get_all_active_tasks()
|
||
now = datetime.datetime.now()
|
||
for t in tasks_list:
|
||
deadline = datetime.datetime.fromisoformat(t["deadline"])
|
||
diff = deadline - now
|
||
new_prio = t["priority"]
|
||
if diff.days <= 1: new_prio = "SSS"
|
||
elif diff.days <= 3: new_prio = "SS"
|
||
elif diff.days <= 7: new_prio = "S"
|
||
if new_prio != t["priority"]:
|
||
await self.db.update_priority(t["id"], new_prio)
|
||
|
||
# チャンネルごとの通知
|
||
channel_tasks = {}
|
||
for t in tasks_list:
|
||
cid = t["channel_id"]
|
||
if cid not in channel_tasks: channel_tasks[cid] = {}
|
||
uid = t["user_id"]
|
||
if uid not in channel_tasks[cid]: channel_tasks[cid][uid] = []
|
||
channel_tasks[cid][uid].append(t)
|
||
|
||
for cid, u_dict in channel_tasks.items():
|
||
channel = self.get_channel(cid)
|
||
if not channel: continue
|
||
for uid, t_list in u_dict.items():
|
||
embed = discord.Embed(title="☀️ 本日のタスクリマインド", color=discord.Color.gold())
|
||
desc = "\n".join([f"• **[{t['priority']}] {t['title']}** ({datetime.datetime.fromisoformat(t['deadline']).strftime('%m/%d %H:%M')})" for t in t_list])
|
||
embed.description = desc
|
||
await channel.send(content=f"<@{uid}> さん、今日の予定です！", embed=embed)
|
||
|
||
@tasks.loop(minutes=1)
|
||
async def deadline_monitor(self):
|
||
now = datetime.datetime.now()
|
||
tasks_list = await self.db.get_all_active_tasks()
|
||
for t in tasks_list:
|
||
deadline = datetime.datetime.fromisoformat(t["deadline"])
|
||
diff = deadline - now
|
||
seconds_left = diff.total_seconds()
|
||
|
||
channel = self.get_channel(t["channel_id"])
|
||
if not channel: continue
|
||
|
||
# 1時間前
|
||
if 0 < seconds_left <= 3600 and t['last_reminded_type'] != '1h':
|
||
embed = discord.Embed(title=f"⏳ 締切1時間前: {t['title']}", color=discord.Color.orange())
|
||
await channel.send(content=f"<@{t['user_id']}> 準備はいいですか？あと1時間です！", embed=embed, view=TaskActionView(self.db, t['id']))
|
||
await self.db.update_reminded_type(t['id'], '1h')
|
||
|
||
# 締切
|
||
elif seconds_left <= 0 and t['last_reminded_type'] != '0h':
|
||
embed = discord.Embed(title=f"⏰ 締切時刻です！: {t['title']}", color=discord.Color.red())
|
||
await channel.send(content=f"<@{t['user_id']}> 締切時間になりました！完了しましたか？", embed=embed, view=TaskActionView(self.db, t['id']))
|
||
await self.db.update_reminded_type(t['id'], '0h')
|
||
|
||
class TaskCommands(commands.Cog):
|
||
def __init__(self, bot: TaskBot):
|
||
self.bot = bot
|
||
self.db = bot.db
|
||
|
||
async def task_autocomplete(self, interaction: discord.Interaction, current: str):
|
||
tasks = await self.db.get_user_tasks(interaction.user.id, interaction.channel_id)
|
||
return [app_commands.Choice(name=f"[{t['priority']}] {t['title']}", value=t['id']) for t in tasks if current.lower() in t['title'].lower()][:25]
|
||
|
||
@commands.command(name="sync")
|
||
@commands.is_owner()
|
||
async def sync_commands(self, ctx: commands.Context, guild_only: bool = True):
|
||
try:
|
||
if guild_only:
|
||
self.bot.tree.copy_global_to(guild=ctx.guild)
|
||
await self.bot.tree.sync(guild=ctx.guild)
|
||
await ctx.send("✅ 同期完了(サーバー)")
|
||
else:
|
||
await self.bot.tree.sync()
|
||
await ctx.send("✅ 同期完了(グローバル)")
|
||
except Exception as e: await ctx.send(f"❌ エラー: {e}")
|
||
|
||
@app_commands.command(name="add")
|
||
async def add_slash(self, interaction: discord.Interaction):
|
||
await interaction.response.send_modal(TaskModal(self.db))
|
||
|
||
@app_commands.command(name="list")
|
||
async def list_tasks_slash(self, interaction: discord.Interaction):
|
||
tasks_list = await self.db.get_user_tasks(interaction.user.id, interaction.channel_id)
|
||
if not tasks_list:
|
||
await interaction.response.send_message("📋 未完了のタスクはありません。", ephemeral=True)
|
||
return
|
||
await interaction.response.defer(ephemeral=True)
|
||
for t in tasks_list:
|
||
deadline_fmt = datetime.datetime.fromisoformat(t['deadline']).strftime("%m/%d %H:%M")
|
||
embed = discord.Embed(title=f"[{t['priority']}] {t['title']}", color=discord.Color.green())
|
||
embed.description = f"📅 締切: {deadline_fmt}\n{t['description'] or ''}"
|
||
await interaction.followup.send(embed=embed, view=TaskActionView(self.db, t['id']), ephemeral=True)
|
||
|
||
@app_commands.command(name="done")
|
||
@app_commands.autocomplete(task_id=task_autocomplete)
|
||
async def done_slash(self, interaction: discord.Interaction, task_id: str):
|
||
await self.db.complete_task(task_id)
|
||
await interaction.response.send_message(f"✅ 完了にしました！", ephemeral=True)
|
||
|
||
@app_commands.command(name="edit")
|
||
@app_commands.autocomplete(task_id=task_autocomplete)
|
||
async def edit_slash(self, interaction: discord.Interaction, task_id: str):
|
||
task = await self.db.get_task_by_id(task_id)
|
||
if task: await interaction.response.send_modal(EditTaskModal(self.db, task))
|
||
else: await interaction.response.send_message("❌ 見つかりません。", ephemeral=True)
|