damastodon/damastodon.py

#!/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+"! Answer \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+" ◽: @"+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(four_engine.drawChequerboard(board,players=[white_knight,black_knight],space=empty,toprow=conn4row),visibility="direct")
			with open(save_position+account+"."+extension,"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)
			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 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 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
		

		
		#Draughts
		if os.path.exists(save_position+account+".draughts"): #We are in a game, so movements are parsed and lobby commands are disabled
			with open(save_position+account+".draughts","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+".draughts")
					logging.warning("%s file corrupted",account)
					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+".draughts")
					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:]+".draughts")
				os.remove(save_position+white[1:]+".draughts")
				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+".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+" ◽: "+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+" ◽: "+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"])