You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
2.5 KiB
100 lines
2.5 KiB
import asyncio
|
|
|
|
import discord
|
|
from discord.ext import commands
|
|
|
|
from redbot.core import Config
|
|
|
|
from datetime import datetime, timedelta
|
|
|
|
from .game import Game
|
|
|
|
|
|
class Werewolf:
|
|
"""
|
|
Base to host werewolf on a guild
|
|
"""
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.config = Config.get_conf(self, identifier=87101114101119111108102, force_registration=True)
|
|
default_global = {}
|
|
default_guild = {
|
|
}
|
|
|
|
self.config.register_global(**default_global)
|
|
self.config.register_guild(**default_guild)
|
|
|
|
self.games = {} # Active games stored here, id is per guild
|
|
|
|
@commands.group()
|
|
async def ww(self, ctx: commands.Context):
|
|
"""
|
|
Base command for this cog. Check help for the commands list.
|
|
"""
|
|
if ctx.invoked_subcommand is None:
|
|
await ctx.send_help()
|
|
|
|
@ww.command()
|
|
async def new(self, ctx, game_code):
|
|
"""
|
|
Create and join a new game of Werewolf
|
|
"""
|
|
|
|
game = self._get_game(ctx.guild, game_code)
|
|
|
|
if not game:
|
|
ctx.send("Failed to start a new game")
|
|
|
|
@ww.command()
|
|
async def join(self, ctx):
|
|
"""
|
|
Joins a game of Werewolf
|
|
"""
|
|
|
|
game = self._get_game(ctx.guild)
|
|
|
|
if not game:
|
|
ctx.send("No game to join!\nCreate a new one with `[p]ww new`")
|
|
return
|
|
|
|
await game.join(ctx.author, ctx.channel)
|
|
|
|
@ww.command()
|
|
async def quit(self, ctx):
|
|
"""
|
|
Quit a game of Werewolf
|
|
"""
|
|
|
|
game = self._get_game(ctx.guild)
|
|
|
|
await game.quit(ctx.author, ctx.channel)
|
|
|
|
@ww.command()
|
|
async def vote(self, ctx, id):
|
|
"""
|
|
Vote for a player by ID
|
|
"""
|
|
game = self._get_game(guild)
|
|
if not game:
|
|
ctx.send("No game running, cannot vote")
|
|
|
|
# Game handles response now
|
|
channel = ctx.channel
|
|
if channel is game.village_channel:
|
|
await game.vote(ctx.author, id, channel)
|
|
|
|
if channel in (c["channel"] for c in game.p_channels.values()):
|
|
await game.vote(ctx.author, id, channel)
|
|
|
|
def _get_game(self, guild, game_code=None):
|
|
if guild.id not in self.games:
|
|
if not game_code:
|
|
return None
|
|
self.games[guild.id] = Game(guild, game_code)
|
|
|
|
return self.games[guild.id]
|
|
|
|
async def _game_start(self, game):
|
|
await game.start()
|