mirror of
https://gitlab.com/oloturia/damastodon.git
synced 2025-01-07 22:17:15 +01:00
172 lines
7.9 KiB
Python
Executable file
172 lines
7.9 KiB
Python
Executable file
#!/usr/bin/python3
|
||
|
||
from mastodon import Mastodon
|
||
import draughts_engine as dama
|
||
import login
|
||
import pickle
|
||
import random
|
||
import os
|
||
import time
|
||
import sys
|
||
import re
|
||
|
||
#configuration server
|
||
api_url = sys.argv[1]
|
||
save_position = "/tmp/"
|
||
CLEANR = re.compile('<.*?>')
|
||
botname = "@damastodon "
|
||
|
||
#board appearence
|
||
frstrow = "🇻1️⃣ 2️⃣ 3️⃣ 4️⃣ 5️⃣ 6️⃣ 7️⃣ 8️⃣\n"
|
||
column = "🇦🇧🇨🇩🇪🇫🇬🇭"
|
||
space="🟥 "
|
||
white_norm="◽ "
|
||
white_knight="⚪ "
|
||
black_norm="◾ "
|
||
black_knight="⚫ "
|
||
empty="🟦 "
|
||
|
||
|
||
def cleanHTML(raw):
|
||
cleanText = re.sub(CLEANR, '',raw)
|
||
return cleanText
|
||
|
||
def check_message(notification):
|
||
account = notification["account"]["acct"]
|
||
try:
|
||
content = cleanHTML(notification["status"]["content"])
|
||
except KeyError:
|
||
return
|
||
content = content[len(botname):]
|
||
saves = os.listdir(save_position)
|
||
if "help" in content.lower(): #Ask for help
|
||
mastodon.status_post("Hello @"+account+" \nchallenge an user by writing to me\nCHALL <USERNAME>\nEx. \"CHALL @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.",visibility="direct")
|
||
return
|
||
if not os.path.exists(save_position+account): #If there is no a savegame file, then lobby mode is activated
|
||
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 content[:5].lower() == "chall": #Challenge someone, check if a savegame is already existing
|
||
file_save_white = [sv for sv in saves if account in sv]
|
||
file_save_black = [sv for sv in saves if challenged in sv]
|
||
if len(file_save_white) > 0: #We are playing a match
|
||
mastodon.status_post("Hello @"+account+" \n you're already playing a match",visibility="direct")
|
||
return
|
||
elif len(file_save_black) > 0: #Our opponent is already playing with someone
|
||
mastodon.status_post("Hello @"+account+" \n the user you challenged is already playing a match",visibility="direct")
|
||
return
|
||
else:
|
||
with open(save_position+account,"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 draughts! Answer \n@"+account+" OK\n to accept the chellenge or \n@"+account+" NO\n to cancel.",visibility="direct")
|
||
return
|
||
elif content.split(" ")[1].lower() == "ok": #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,"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
|
||
print(account+" file not found")
|
||
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,save_position+account) #Linked the two savegames together
|
||
board = dama.init_board()
|
||
with open(save_position+account,"wb") as f:
|
||
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)
|
||
mastodon.status_post("◾: @"+account+" ◽: @"+challenger+" \nturn ◽\n"+dama.draw_checkerboard(board,space,white_norm,white_knight,black_norm,black_knight,empty,column,frstrow),visibility="direct")
|
||
return
|
||
elif content.split(" ")[1].lower() == "no": #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)
|
||
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
|
||
else: #We are in a game, so movements are parsed and lobby commands are disabled
|
||
with open(save_position+account,"rb") as f:
|
||
try:
|
||
start = pickle.load(f)
|
||
except EOFError: # Something went very wrong, file is corrupt?
|
||
mastodon.status_post("Hello @"+account+" \n unfortunately, your savegame is corrupted or missing",visibility="direct") #The file has been moved or is corrupted
|
||
os.remove(save_position+account)
|
||
print(account+" file corrupted")
|
||
return
|
||
if start: #The game is started, load other parameters
|
||
black = pickle.load(f)
|
||
white = 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)
|
||
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
|
||
if "quit" in content.lower(): #The game is quitted
|
||
os.remove(save_position+black[1:])
|
||
os.remove(save_position+white[1:])
|
||
mastodon.status_post(black+" "+white+" the match was cancelled.",visibility="direct")
|
||
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,"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+" ◽: "+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:])
|
||
os.remove(save_position+white[1:])
|
||
mastodon.status_post("◾: "+black+" ◽: "+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
|
||
|
||
if __name__ == "__main__":
|
||
if api_url[:4] != "http":
|
||
print("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"])
|