mirror of
https://gitlab.com/oloturia/damastodon.git
synced 2025-01-04 20:47:15 +01:00
242 lines
11 KiB
Python
Executable file
242 lines
11 KiB
Python
Executable file
#!/usr/bin/python3
|
||
|
||
from mastodon import Mastodon
|
||
import draughts_engine as dama
|
||
import four_engine
|
||
import login
|
||
import pickle
|
||
import random
|
||
import os
|
||
import time
|
||
import sys
|
||
import re
|
||
import logging
|
||
|
||
#configuration server
|
||
api_url = sys.argv[1]
|
||
save_position = "/tmp/"
|
||
CLEANR = re.compile('<.*?>')
|
||
#botname = "@damastodon "
|
||
botname = "@dama "
|
||
|
||
available_games = ("draughts","conn4")
|
||
|
||
#board appearence
|
||
frstrow = "🇻1️⃣ 2️⃣ 3️⃣ 4️⃣ 5️⃣ 6️⃣ 7️⃣ 8️⃣\n"
|
||
column = "🇦🇧🇨🇩🇪🇫🇬🇭"
|
||
space="🟥 "
|
||
white_norm="◽ "
|
||
white_knight="⚪ "
|
||
black_norm="◾ "
|
||
black_knight="⚫ "
|
||
empty="🟦 "
|
||
|
||
#conn4
|
||
conn4row = "1️⃣ 2️⃣ 3️⃣ 4️⃣ 5️⃣ 6️⃣ 7️⃣\n"
|
||
|
||
#logging config
|
||
logging.basicConfig(filename="/tmp/dama.log",level=logging.DEBUG)
|
||
|
||
def cleanHTML(raw):
|
||
cleanText = re.sub(CLEANR, '',raw)
|
||
return cleanText
|
||
|
||
def lobby(notification,content,account,extension):
|
||
try:
|
||
challenged = notification["status"]["mentions"][1]["acct"] #If there isn't another account cited, then the request is malformed
|
||
except:
|
||
mastodon.status_post("Hello @"+account+" \n your request is not valid",visibility="direct")
|
||
return
|
||
|
||
if challenged == account: #The user tried to challenge him/herself
|
||
mastodon.status_post("Hello @"+account+"\n You can't challenge yourself",visibility="direct")
|
||
return
|
||
|
||
if content.split(" ")[0].lower() == extension: #Challenge someone, check if a savegame is already existing
|
||
if os.path.exists(save_position+challenged+"."+extension):
|
||
mastodon.status_post("Hello @"+account+" \n the user you challenged is already playing a match",visibility="direct")
|
||
return
|
||
else:
|
||
with open(save_position+account+"."+extension,"wb") as f: #The request is valid, writes a savegame with the first element as False, that marks that the game isn't started yet
|
||
pickle.dump("WAIT",f)
|
||
ident = mastodon.status_post("Hello @"+challenged+" \n@"+account+" challenged you to a match of "+extension+"! Reply \n @"+account+" OK "+extension.upper()+"\n to accept the challenge or \n@"+account+" NO "+extension.upper()+"\n to cancel.",visibility="direct")
|
||
return
|
||
elif content.split(" ")[1].lower() == "ok" and content.split(" ")[2].lower() == extension: #The opponent accepted the match
|
||
try:
|
||
challenger = notification["status"]["mentions"][1]["acct"]
|
||
except:
|
||
mastodon.status_post("Hello @"+account+" \n your request is not valid",visibility="direct") #Somehow, the opponent answered with only one account
|
||
return
|
||
try:
|
||
with open(save_position+challenger+"."+extension,"rb") as f:
|
||
start = pickle.load(f)
|
||
except FileNotFoundError:
|
||
mastodon.status_post("Hello @"+account+" \n unfortunately, your savegame is corrupted or missing",visibility="direct") #The file has moved or corrupted
|
||
logging.warning("%s file not found",account)
|
||
return
|
||
if start != "WAIT":
|
||
mastodon.status_post("Hello @"+account+" \n your request is not valid",visibility="direct") #The user that challenged us is playing a game with someone else
|
||
return
|
||
os.symlink(save_position+challenger+"."+extension,save_position+account+"."+extension) #Linked the two savegames together
|
||
|
||
|
||
if extension == "draughts": #Draughts init
|
||
board = dama.init_board()
|
||
mastodon.status_post("◾: @"+account+" \n◽: @"+challenger+" \nturn ◽\n"+dama.draw_checkerboard(board,space,white_norm,white_knight,black_norm,black_knight,empty,column,frstrow),visibility="direct")
|
||
elif extension == "conn4": #Conn4 init
|
||
board = four_engine.initChequerboard()
|
||
mastodon.status_post("⚪: @"+account+" \n⚫: @"+challenger+" \nturn ⚪\n"+four_engine.drawChequerboard(board,players=[white_knight,black_knight],space=empty,toprow=conn4row),visibility="direct")
|
||
with open(save_position+account+"."+extension,"wb") as f: #Writing the file with the status of the game
|
||
pickle.dump("START",f) #Now the game has started
|
||
pickle.dump("@"+account,f)
|
||
pickle.dump("@"+challenger,f)
|
||
pickle.dump(False,f)
|
||
pickle.dump(board,f)
|
||
return
|
||
elif content.split(" ")[1].lower() == "no" and content.split(" ")[2].lower == extension: #The opponent refused the game
|
||
try:
|
||
challenger = notification["status"]["mentions"][1]["acct"]
|
||
except:
|
||
mastodon.status_post("Hello @"+account+" \n your request is not valid",visibility="direct") #Only one user in the message
|
||
return
|
||
os.remove(save_position+challenger+"."+extension)
|
||
mastodon.status_post("@"+account+" you cancelled the challenge from @"+challenger,visibility="direct") #Game is cancelled, free to challenge someone else
|
||
return
|
||
else:
|
||
mastodon.status_post("Hello @"+account+" \nI can't understand your command or you're not in a match.\nWrite HELP to see the list of available commands.",visibility="direct") #Every other command for the lobby ends here
|
||
return
|
||
|
||
def load_status(account,extension,content):
|
||
with open(save_position+account+"."+extension,"rb") as f: #Open the status file - extension is the type of the game
|
||
try:
|
||
start = pickle.load(f)
|
||
except EOFError: #Something wrong happened
|
||
mastodon.status_post("Hello @"+account+" \n unfortunately your savegame is corrupted or missing. The game is cancelled.",visibility="direct")
|
||
os.remove(save_position+account+"."+extension)
|
||
logging.warning("% file corrupted or missing",account+"."+extension)
|
||
return False
|
||
|
||
player_1 = pickle.load(f) #Read status from file
|
||
player_2 = pickle.load(f)
|
||
turn = pickle.load(f)
|
||
board = pickle.load(f)
|
||
|
||
if (start == "WAIT"): #The game is not started yet
|
||
if "quit" in content.lower(): #Game withdrawn
|
||
os.remove(save_position+account+"."+extension)
|
||
mastodon.status_post("Hello @"+account+" \nthe challenge has been withdrawn.",visibility="direct")
|
||
else: #Lobby is disabled if a challenge request is active
|
||
mastodon.status_post("Hello @"+account+" \nyou have already challenged someone, type QUIT to withdraw,",visibility="direct")
|
||
return False
|
||
if "quit" in content.lower(): #The game is quitted
|
||
os.remove(save_position+player_1[1:]+"."+extension)
|
||
os.remove(save_position+player_2[1:]+"."+extension)
|
||
mastodon.status_post(player_2+" "+player_1+" the match was cancelled.",visibility="direct")
|
||
return False
|
||
|
||
return True,player_1,player_2,turn,board
|
||
|
||
def check_message(notification):
|
||
account = notification["account"]["acct"]
|
||
try:
|
||
content = cleanHTML(notification["status"]["content"])
|
||
except KeyError:
|
||
return
|
||
content = content[len(botname):]
|
||
if "help" in content.lower(): #Ask for help
|
||
mastodon.status_post("Hello @"+account+" \nchallenge an user in a game of draughts by writing to me\nDRAUGHTS <USERNAME>\nEx. \"DRAUGHTS @someone@mastdn.inst.wxyz\"\nThe challenger takes WHITE and begins the match.\nFor movements and jumps, write the coords separated by spaces.\nEx. \"A4 B5\" (normal movement) or \"A4 C6 D8\" (double jump)\nQUIT ends the match.\nCommands are NOT case sensitive.\nTo challenge someone in a game of Connect 4 write CONN4 <USERNAME>.",visibility="direct")
|
||
return
|
||
|
||
|
||
#Conn4
|
||
if os.path.exists(save_position+account+".conn4"):
|
||
start,player_1,player_2,turn,board = load_status(account,"conn4",content)
|
||
if not(start):
|
||
return
|
||
|
||
if (player_2 == "@"+account and turn == 1) or (player_1 == "@"+account and turn == 0):
|
||
board,win = four_engine.dropChip(board,content.lower()[-1],turn+1)
|
||
if not(board):
|
||
mastodon.status_post("@"+account+" \nInvalid move.",visibility="direct")
|
||
return
|
||
else:
|
||
with open(save_position+account+".conn4","wb") as f:
|
||
pickle.dump("START",f)
|
||
turn = not turn
|
||
pickle.dump(player_1,f)
|
||
pickle.dump(player_2,f)
|
||
pickle.dump(turn,f)
|
||
pickle.dump(board,f)
|
||
if turn == 0: #the first is the current turn, the second is the last turn
|
||
colour = (white_knight,black_knight)
|
||
else:
|
||
colour = (black_knight,white_knight)
|
||
if win == 0:
|
||
mastodon.status_post("⚪: "+player_1+" \n⚫: "+player_2+" \nturn "+colour[0]+"\n"+four_engine.drawChequerboard(board,players=[white_knight,black_knight],space=empty,toprow=conn4row),visibility="direct")
|
||
return
|
||
else: #Someone won!
|
||
mastodon.status_post("⚪: "+player_1+" \n⚫: "+player_2+" \n"+colour[1]+" WINS!\n"+four_engine.drawChequerboard(board,players=[white_knight,black_knight],space=empty,toprow=conn4row),visibility="direct")
|
||
os.remove(save_position+player_1[1:]+".conn4")
|
||
os.remove(save_position+player_2[1:]+".conn4")
|
||
return
|
||
else: #We moved in a wrong turn
|
||
mastodon.status_post("@"+account+" \nIt's not your turn.",visibility="direct")
|
||
return
|
||
|
||
#Draughts
|
||
if os.path.exists(save_position+account+".draughts"): #We are in a game, so movements are parsed and lobby commands are disabled
|
||
start,black,white,turn,board = load_status(account,"draughts",content)
|
||
if not(start):
|
||
return
|
||
|
||
|
||
if (black == "@"+account and turn == 1) or (white == "@"+account and turn == 0): #Check if the turn is right
|
||
board = dama.valid_move(content.lower(),turn,board,inversion=True) #Function dama.valid_move parses the input for couples of letter and number
|
||
if board == -1: #We made an invalid move
|
||
mastodon.status_post("@"+account+" \nInvalid move.",visibility="direct")
|
||
return
|
||
else:
|
||
with open(save_position+account+".draughts","wb") as f: #Save the updated board
|
||
pickle.dump("START",f)
|
||
turn = not turn
|
||
pickle.dump(black,f)
|
||
pickle.dump(white,f)
|
||
pickle.dump(turn,f)
|
||
pickle.dump(board,f)
|
||
if turn == 0:
|
||
colour = "◽"
|
||
else:
|
||
colour = "◾"
|
||
winner = dama.checkWin(board) #Check for winner
|
||
if winner == (False,False): #No one is winning yet
|
||
mastodon.status_post("◾: "+black+" \n◽: "+white+" \nturn "+colour+"\n"+dama.draw_checkerboard(board,space,white_norm,white_knight,black_norm,black_knight,empty,column,frstrow),visibility="direct")
|
||
return
|
||
else: #Someone won
|
||
if winner == (True,False):
|
||
winner_t = "WHITE"
|
||
else:
|
||
winner_t = "BLACK"
|
||
os.remove(save_position+black[1:]+".draughts")
|
||
os.remove(save_position+white[1:]+".draughts")
|
||
mastodon.status_post("◾: "+black+" \n◽: "+white+"\n"+winner_t+" WINS!\n"+dama.draw_checkerboard(board,space,white_norm,white_knight,black_norm,black_knight,empty,column,frstrow),visibility="direct")
|
||
return
|
||
else: #We moved in a wrong turn
|
||
mastodon.status_post("@"+account+" \nIt's not your turn.",visibility="direct")
|
||
return
|
||
|
||
#Lobby
|
||
for game in available_games:
|
||
if game in content.lower():
|
||
if not os.path.exists(save_position+account+"."+game): #If there is no a savegame file, then lobby mode is activated
|
||
lobby(notification,content,account,game)
|
||
|
||
if __name__ == "__main__":
|
||
if api_url[:4] != "http":
|
||
logging.error("Invalid address")
|
||
quit()
|
||
mastodon = login.login(api_url)
|
||
while True:
|
||
time.sleep(10)
|
||
for x in mastodon.notifications():
|
||
check_message(x)
|
||
mastodon.notifications_dismiss(x["id"])
|