259 lines
12 KiB
Python
259 lines
12 KiB
Python
from discord.ext import commands
|
|
import discord
|
|
|
|
from . import utils
|
|
|
|
import re
|
|
import random
|
|
|
|
|
|
class Board:
|
|
def __init__(self, player1, player2):
|
|
# Our board just needs to be a 3x3 grid. To keep formatting nice, each one is going to be a space to start
|
|
self.board = [[' ', ' ', ' '], [' ', ' ', ' '], [' ', ' ', ' ']]
|
|
|
|
# Randomize who goes first when the board is created
|
|
if random.SystemRandom().randint(0, 1):
|
|
self.challengers = {'x': player1, 'o': player2}
|
|
else:
|
|
self.challengers = {'x': player2, 'o': player1}
|
|
|
|
# X's always go first
|
|
self.X_turn = True
|
|
|
|
def full(self):
|
|
# For this check we just need to see if there is a space anywhere, if there is then we're not full
|
|
for row in self.board:
|
|
if ' ' in row:
|
|
return False
|
|
return True
|
|
|
|
def can_play(self, player):
|
|
# Simple check to see if the player is the one that's up
|
|
if self.X_turn:
|
|
return player == self.challengers['x']
|
|
else:
|
|
return player == self.challengers['o']
|
|
|
|
def update(self, x, y):
|
|
# If it's x's turn, we place an x, otherwise place an o
|
|
letter = 'x' if self.X_turn else 'o'
|
|
# Make sure the place we're trying to update is blank, we can't override something
|
|
if self.board[x][y] == ' ':
|
|
self.board[x][y] = letter
|
|
else:
|
|
return False
|
|
# If we were succesful in placing the piece, we need to switch whose turn it is
|
|
self.X_turn = not self.X_turn
|
|
return True
|
|
|
|
def check(self):
|
|
# Checking all possiblities will be fun...
|
|
# First base off the top-left corner, see if any possiblities with that match
|
|
# We need to also make sure that the place is not blank, so that 3 in a row that are blank doesn't cause a 'win'
|
|
# Top-left, top-middle, top right
|
|
if self.board[0][0] == self.board[0][1] and self.board[0][0] == self.board[0][2] and self.board[0][0] != ' ':
|
|
return self.challengers[self.board[0][0]]
|
|
# Top-left, middle-left, bottom-left
|
|
if self.board[0][0] == self.board[1][0] and self.board[0][0] == self.board[2][0] and self.board[0][0] != ' ':
|
|
return self.challengers[self.board[0][0]]
|
|
# Top-left, middle, bottom-right
|
|
if self.board[0][0] == self.board[1][1] and self.board[0][0] == self.board[2][2] and self.board[0][0] != ' ':
|
|
return self.challengers[self.board[0][0]]
|
|
|
|
# Next check the top-right corner, not re-checking the last possiblity that included it
|
|
# Top-right, middle-right, bottom-right
|
|
if self.board[0][2] == self.board[1][2] and self.board[0][2] == self.board[2][2] and self.board[0][2] != ' ':
|
|
return self.challengers[self.board[0][2]]
|
|
# Top-right, middle, bottom-left
|
|
if self.board[0][2] == self.board[1][1] and self.board[0][2] == self.board[2][0] and self.board[0][2] != ' ':
|
|
return self.challengers[self.board[0][2]]
|
|
|
|
# Next up, bottom-right corner, only one possiblity to check here, other two have been checked
|
|
# Bottom-right, bottom-middle, bottom-left
|
|
if self.board[2][2] == self.board[2][1] and self.board[2][2] == self.board[2][0] and self.board[2][2] != ' ':
|
|
return self.challengers[self.board[2][2]]
|
|
|
|
# No need to check the bottom-left, all posiblities have been checked now
|
|
# Base things off the middle now, as we only need the two 'middle' possiblites that aren't diagonal
|
|
# Top-middle, middle, bottom-middle
|
|
if self.board[1][1] == self.board[0][1] and self.board[1][1] == self.board[2][1] and self.board[1][1] != ' ':
|
|
return self.challengers[self.board[1][1]]
|
|
# Left-middle, middle, right-middle
|
|
if self.board[1][1] == self.board[1][0] and self.board[1][1] == self.board[1][2] and self.board[1][1] != ' ':
|
|
return self.challengers[self.board[1][1]]
|
|
|
|
# Otherwise nothing has been found, return None
|
|
return None
|
|
|
|
def __str__(self):
|
|
# Simple formatting here when you look at it, enough spaces to even out where everything is
|
|
# Place whatever is at the grid in place, whether it's x, o, or blank
|
|
_board = " {} | {} | {}\n".format(self.board[0][0], self.board[0][1], self.board[0][2])
|
|
_board += "———————————————\n"
|
|
_board += " {} | {} | {}\n".format(self.board[1][0], self.board[1][1], self.board[1][2])
|
|
_board += "———————————————\n"
|
|
_board += " {} | {} | {}\n".format(self.board[2][0], self.board[2][1], self.board[2][2])
|
|
return "```\n{}```".format(_board)
|
|
|
|
|
|
class TicTacToe:
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.boards = {}
|
|
|
|
def create(self, server_id, player1, player2):
|
|
self.boards[server_id] = Board(player1, player2)
|
|
|
|
# Return whoever is x's so that we know who is going first
|
|
return self.boards[server_id].challengers['x']
|
|
|
|
@commands.group(aliases=['tic', 'tac', 'toe'], no_pm=True, invoke_without_command=True)
|
|
@utils.custom_perms(send_messages=True)
|
|
async def tictactoe(self, ctx, *, option: str):
|
|
"""Updates the current server's tic-tac-toe board
|
|
You obviously need to be one of the players to use this
|
|
It also needs to be your turn
|
|
Provide top, left, bottom, right, middle as you want to mark where to play on the board
|
|
|
|
EXAMPLE: !tictactoe middle top
|
|
RESULT: Your piece is placed in the very top space, in the middle"""
|
|
player = ctx.message.author
|
|
board = self.boards.get(ctx.message.guild.id)
|
|
# Need to make sure the board exists before allowing someone to play
|
|
if not board:
|
|
await ctx.send("There are currently no Tic-Tac-Toe games setup!")
|
|
return
|
|
# Now just make sure the person can play, this will fail if o's are up and x tries to play
|
|
# Or if someone else entirely tries to play
|
|
if not board.can_play(player):
|
|
await ctx.send("You cannot play right now!")
|
|
return
|
|
|
|
# Search for the positions in the option given, the actual match doesn't matter, just need to check if it exists
|
|
top = re.search('top', option)
|
|
middle = re.search('middle', option)
|
|
bottom = re.search('bottom', option)
|
|
left = re.search('left', option)
|
|
right = re.search('right', option)
|
|
|
|
# Just a bit of logic to ensure nothing that doesn't make sense is given
|
|
if top and bottom:
|
|
await ctx.send("That is not a valid location! Use some logic, come on!")
|
|
return
|
|
if left and right:
|
|
await ctx.send("That is not a valid location! Use some logic, come on!")
|
|
return
|
|
# Make sure at least something was given
|
|
if not top and not bottom and not left and not right and not middle:
|
|
await ctx.send("Please provide a valid location to play!")
|
|
return
|
|
|
|
x = 0
|
|
y = 0
|
|
# Simple assignments
|
|
if top:
|
|
x = 0
|
|
if bottom:
|
|
x = 2
|
|
if left:
|
|
y = 0
|
|
if right:
|
|
y = 2
|
|
# If middle was given and nothing else, we need the exact middle
|
|
if middle and not (top or bottom or left or right):
|
|
x = 1
|
|
y = 1
|
|
# If just top or bottom was given, we assume this means top-middle or bottom-middle
|
|
# We don't need to do anything fancy with top/bottom as it's already assigned, just assign middle
|
|
if (top or bottom) and not (left or right):
|
|
y = 1
|
|
# If just left or right was given, we assume this means left-middle or right-middle
|
|
# We don't need to do anything fancy with left/right as it's already assigned, just assign middle
|
|
elif (left or right) and not (top or bottom):
|
|
x = 1
|
|
|
|
# If all checks have been made, x and y should now be defined
|
|
# Correctly based on the matches, and we can go ahead and update the board
|
|
# We've already checked if the author can play, so there's no need to make any additional checks here
|
|
# board.update will handle which letter is placed
|
|
# If it returns false however, then someone has already played in that spot and nothing was updated
|
|
if not board.update(x, y):
|
|
await ctx.send("Someone has already played there!")
|
|
return
|
|
# Next check if there's a winner
|
|
winner = board.check()
|
|
if winner:
|
|
# Get the loser based on whether or not the winner is x's
|
|
# If the winner is x's, the loser is o's...obviously, and vice-versa
|
|
loser = board.challengers['x'] if board.challengers['x'] != winner else board.challengers['o']
|
|
await ctx.send("{} has won this game of TicTacToe, better luck next time {}".format(winner.display_name,
|
|
loser.display_name))
|
|
# Handle updating ratings based on the winner and loser
|
|
await utils.update_records('tictactoe', winner, loser)
|
|
# This game has ended, delete it so another one can be made
|
|
del self.boards[ctx.message.guild.id]
|
|
else:
|
|
# If no one has won, make sure the game is not full. If it has, delete the board and say it was a tie
|
|
if board.full():
|
|
await ctx.send("This game has ended in a tie!")
|
|
del self.boards[ctx.message.guild.id]
|
|
# If no one has won, and the game has not ended in a tie, print the new updated board
|
|
else:
|
|
player_turn = board.challengers.get('x') if board.X_turn else board.challengers.get('o')
|
|
fmt = str(board) + "\n{} It is now your turn to play!".format(player_turn.display_name)
|
|
await ctx.send(fmt)
|
|
|
|
@tictactoe.command(name='start', aliases=['challenge', 'create'], no_pm=True)
|
|
@utils.custom_perms(send_messages=True)
|
|
async def start_game(self, ctx, player2: discord.Member):
|
|
"""Starts a game of tictactoe with another player
|
|
|
|
EXAMPLE: !tictactoe start @OtherPerson
|
|
RESULT: A new game of tictactoe"""
|
|
player1 = ctx.message.author
|
|
# For simplicities sake, only allow one game on a server at a time.
|
|
# Things can easily get confusing (on the server's end) if we allow more than one
|
|
if self.boards.get(ctx.message.guild.id) is not None:
|
|
await ctx.send("Sorry but only one Tic-Tac-Toe game can be running per server!")
|
|
return
|
|
# Make sure we're not being challenged, I always win anyway
|
|
if player2 == ctx.message.guild.me:
|
|
await ctx.send("You want to play? Alright lets play.\n\nI win, so quick you didn't even notice it.")
|
|
return
|
|
if player2 == player1:
|
|
await ctx.send("You can't play yourself, I won't allow it. Go find some friends")
|
|
return
|
|
|
|
# Create the board and return who has been decided to go first
|
|
x_player = self.create(ctx.message.guild.id, player1, player2)
|
|
fmt = "A tictactoe game has just started between {} and {}".format(player1.display_name, player2.display_name)
|
|
# Print the board too just because
|
|
fmt += str(self.boards[ctx.message.guild.id])
|
|
|
|
# We don't need to do anything weird with assigning x_player to something
|
|
# it is already a member object, just use it
|
|
fmt += "I have decided at random, and {} is going to be x's this game. It is your turn first! " \
|
|
"Use the {}tictactoe command, and a position, to choose where you want to play" \
|
|
.format(x_player.display_name, ctx.prefix)
|
|
await ctx.send(fmt)
|
|
|
|
@tictactoe.command(name='delete', aliases=['stop', 'remove', 'end'], no_pm=True)
|
|
@utils.custom_perms(kick_members=True)
|
|
async def stop_game(self, ctx):
|
|
"""Force stops a game of tictactoe
|
|
This should realistically only be used in a situation like one player leaves
|
|
Hopefully a moderator will not abuse it, but there's not much we can do to avoid that
|
|
|
|
EXAMPLE: !tictactoe stop
|
|
RESULT: No more tictactoe!"""
|
|
if self.boards.get(ctx.message.guild.id) is None:
|
|
await ctx.send("There are no tictactoe games running on this server!")
|
|
return
|
|
|
|
del self.boards[ctx.message.guild.id]
|
|
await ctx.send("I have just stopped the game of TicTacToe, a new should be able to be started now!")
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(TicTacToe(bot))
|