207 lines
11 KiB
Python
207 lines
11 KiB
Python
from objects.rance import automate
|
|
import random as ra
|
|
import asyncio
|
|
import discord
|
|
|
|
excluded=False
|
|
commands={'rance':[], 'rancequit':[], 'rancehack':[], 'rancerank':[], 'azul':[]}
|
|
|
|
rancegames={}
|
|
|
|
async def rance(args, channel, message):
|
|
if channel.id in rancegames and not type(rancegames[channel.id])==tuple:
|
|
await channel.send("Game already in progress, just watch or pick another channel")
|
|
return
|
|
p1=message.author
|
|
if len(args)==1:
|
|
try:
|
|
num=int(args[0])
|
|
except ValueError: num=None
|
|
if num is not None and 0<num<7: rancegames[channel.id]=(p1,num); await channel.send("Game is open! Awaiting other party!")
|
|
return
|
|
elif len(args)==0 and type(rancegames[channel.id])==tuple:
|
|
p1=rancegames[channel.id][0]; p2=message.author; num=rancegames[channel.id][1]
|
|
else:
|
|
p2=u.findobj(args[0],channel.guild.members)
|
|
if p2 is None:
|
|
await channel.send("Challenging players directly doesn't work anymore, thanks to intents.")
|
|
#await channel.send("First argument should name another player (mention, username, or id)")
|
|
return
|
|
try:
|
|
num=int(args[1])
|
|
except ValueError:
|
|
await channel.send("Second argument should be number of units (1-6)")
|
|
return
|
|
if not 1<=num<=6:
|
|
await channel.send("Number of units should be between 1 and 6, inclusive")
|
|
return
|
|
await rancesetup(p1,p2,channel,num)
|
|
await playrance(channel)
|
|
|
|
async def rancerank(args,channel,message):
|
|
if channel.id in rancegames: await channel.send("Game already in progress, just watch or pick another channel"); return
|
|
if len(args)==0: await channel.send("You need to challenge an opponent with this."); return
|
|
p2=u.findobj(' '.join(args),channel.guild.members)
|
|
if p2 is None: await channel.send("Couldn't find opponent; try mentioning them."); return
|
|
p1=message.author
|
|
rancegames[channel.id]=True
|
|
def validateunits(x):
|
|
n=x.split('\n')
|
|
for m in n[1:]:
|
|
if len(m.split(','))!=9: return f"line {m} is invalid somehow, please redo it"
|
|
return True
|
|
units=await getnp("Please enter your name, then the units you wish to bring to battle, in the following format:\npos,class,atk,def,int,spd,troops,flags,name\nPlease enter one unit per line, but the first line should be your name.\nAlso note that this will be logged; this function is intended for tournament play, where you have an existing roster. If cheating is suspected, the logs will be reviewed. Try not to use it frivolously during tournaments.",[p1,p2],[[],[]],[validateunits])
|
|
u.log('',*units) # That just makes the logs look a bit neater.
|
|
units=units[0].split('\n'),units[1].split('\n')
|
|
names=units[0][0],units[1][0]
|
|
units=[n.split(',') for n in units[0][1:]],[n.split(',') for n in units[1][1:]]
|
|
rancegames[channel.id]=board=automate.Board(names[0],names[1],p1.id,p2.id,{int(m[0])-1:automate.unit.BattleUnit(m[1],*[int(n) for n in m[2:-1]],name=m[-1]) for m in units[0]},{int(m[0])-1:automate.unit.BattleUnit(m[1],*[int(n) for n in m[2:-1]],name=m[-1]) for m in units[1]})
|
|
await p1.send(f"Game's ready in <#{channel.id}>")
|
|
await p2.send(f"Game's ready in <#{channel.id}>")
|
|
await playrance(channel)
|
|
|
|
async def rancesetup(p1,p2,channel,num):
|
|
# All checks finally out of the way zzzz
|
|
rancegames[channel.id]=seed=ra.random()
|
|
await channel.send("ゲーム・スタート！")
|
|
# Fire off the first set of listeners; the player names.
|
|
try:
|
|
names=await getnp("Please enter player name.",[p1,p2],[[],[]],[])
|
|
except (discord.Forbidden,discord.HTTPException) as ex:
|
|
await channel.send("Couldn't DM one of the players (can't tell who yet), maybe their DM settings are wrong; perhaps try DMing me first.")
|
|
rancegames.pop(channel.id)
|
|
return
|
|
if not rancestatus(channel.id,seed): return
|
|
berd=automate.blindstart(num,*names)
|
|
classlist=next(berd)
|
|
# https://stackoverflow.com/a/44780467
|
|
nl='\n'
|
|
longth=lambda x:len(x.split(' '))==num or "Wrong number of units provided"
|
|
def incls(x):
|
|
out=[]
|
|
for n in x.split(' '):
|
|
if n not in automate.classlist: out.append(n)
|
|
if out: return f"Invalid class names: {', '.join(out)}. Please note they're case sensitive."
|
|
return True
|
|
classes=await getnp(f"What classes would you like for your **{num}** units? (Space separated)\nList of melee classes (work best in front row):\n{nl.join(classlist[0])}\n\nRanged (work best in back row):\n{nl.join(classlist[1])}",[p1,p2],[[],[]],[longth,incls])
|
|
if not rancestatus(channel.id,seed): return
|
|
units=berd.send(tuple([m.lower() for m in n.split(' ')] for n in classes))
|
|
def isint(x):
|
|
try: int(x); return True
|
|
except ValueError: return f"'{x}' isn't a valid integer"
|
|
rerolls=await getnp("Your units:\n{0}\n Please pick a number (from 1 to {1}) to reroll; you can discard the reroll, don't worry",[p1,p2],[['\n'.join(units[0]),num],['\n'.join(units[1]),num]],[isint,lambda x:int(x) in range(1,num+1) or f"{x} isn't a valid unit number"])
|
|
rerolls=tuple(int(n)-1 for n in rerolls)
|
|
if not rancestatus(channel.id,seed): return
|
|
news=berd.send(rerolls)
|
|
boolaliases={'new':True,'old':False, 'yes':True,'no':False, 'true':True,'false':False, '1':True,'0':False, 'y':True,'n':False}
|
|
confirms=await getnp("Old unit: {0}\nNew unit: {1}\nWhich would you like to keep (old/new)?",[p1,p2],[[units[0][rerolls[0]],news[0]],[units[1][rerolls[1]],news[1]]],[lambda x:x in boolaliases or "Please specify either `old` or `new`"])
|
|
if not rancestatus(channel.id,seed): return
|
|
_=berd.send(tuple(boolaliases[n.lower()] for n in confirms))
|
|
def checknames(x):
|
|
if x in ('-','`-`'): return True
|
|
try: a={int(n.split(':')[0]):n.split(':')[1] for n in x.split(' ')}
|
|
except: return 'The format was weird in some way, please try again.'
|
|
if min(list(a))<1 or max(list(a))>num+1: return f"Numbers should be between 1 and {num} inclusive."
|
|
return True
|
|
unames=await getnp('You may provide names for your units, if you wish. Please provide them as num:name pairs, eg. `2:Ran 3:Rin`\nUnnamed units will be assigned a number as their name. Say `-` if you don\'t want to name any units.',[p1,p2],[[],[]],[checknames])
|
|
if not rancestatus(channel.id,seed): return
|
|
def isints(x):
|
|
for n in x.split(' '):
|
|
try: int(n)
|
|
except ValueError: return f"'{n}' isn't a valid integer"
|
|
if int(n) not in range(1,7): return f"{n} isn't a valid position"
|
|
return True
|
|
units=berd.send(({int(n.split(':')[0])-1:n.split(':')[1] for n in unames[0].split(' ')} if unames[0]!='-' else {},{int(n.split(':')[0])-1:n.split(':')[1] for n in unames[1].split(' ')} if unames[1]!='-' else {}))
|
|
positions=await getnp("Final unit list:\n{0}\nPlease enter the positions for these units (1-3 back row (ranged units should go here), 4-6 front row (melee units should go here), space separated, positions should be unique!)",[p1,p2],[['\n'.join(units[0])],['\n'.join(units[1])]],[lambda x:len(x.split(' '))==num or f"You should provide {num} position numbers", isints, lambda x:len(set(x.split(' ')))==len(x.split(' ')) or "Position numbers must be unique"])
|
|
if not rancestatus(channel.id,seed): return
|
|
_=berd.send(([int(n)-1 for n in positions[0].split(' ')],[int(n)-1 for n in positions[1].split(' ')]))
|
|
try: berd.send((p1.id,p2.id))
|
|
except StopIteration as ex:
|
|
board=ex.value
|
|
# We have a board!
|
|
rancegames[channel.id]=board
|
|
await p1.send(f"Game's ready in <#{channel.id}>")
|
|
await p2.send(f"Game's ready in <#{channel.id}>")
|
|
|
|
async def playrance(channel,board=None):
|
|
nl='\n'
|
|
if board is None: board=rancegames[channel.id]
|
|
controller=board.controller()
|
|
await channel.send("If a unit wasn't named, its unit code became its name.\nファイト！")
|
|
# No auto quit yet.
|
|
while True:
|
|
if not rancestatus(channel.id,board): return
|
|
# We can discard this during charged actions but if we don't get it shit goes wrong.
|
|
try: active=next(controller)
|
|
except StopIteration as ex: end=ex; break
|
|
if active is None:
|
|
# Charged action is happening. Don't wait for action or anything.
|
|
action=None
|
|
else:
|
|
await channel.send(f'```{board}```')
|
|
await channel.send(f"Available actions for {active[0][0]}'s unit {active[1]}:\n{nl.join([', '.join(n) for n in active[2]])}")
|
|
msg=await client.wait_for('message',check=lambda x:x.author.id==active[0][1] and x.channel.id==channel.id and x.content.split(' ')[0] in [n[0] for n in active[2]])
|
|
action=msg.content.split(' ')
|
|
try: res=controller.send(action); await channel.send(res)
|
|
except Exception as ex: await channel.send(str(ex)+'\nThis might have broken the game state; please confirm and quit if so')
|
|
await channel.send(f'```{board}```')
|
|
await channel.send(end)
|
|
rancegames.pop(channel.id)
|
|
|
|
async def getnp(question,players,formats,conditions):
|
|
# Each condition is a function that will return a string if the content failed validation.
|
|
# That string should be fired back to the user, for them to retry.
|
|
# The conditions should accept a string.
|
|
solos=[get1p(question,players[i],formats[i],conditions) for i in range(len(players))]
|
|
return [n.content.replace(' ',' ') for n in await asyncio.gather(*solos)]
|
|
|
|
async def get1p(question,player,forma,conditions):
|
|
# Probably easier to do it like this.
|
|
await player.send(question.format(*forma))
|
|
valid=False
|
|
while not valid:
|
|
msg=await client.wait_for('message',check=lambda x:x.author.id==player.id and x.channel==x.author.dm_channel)
|
|
content=msg.content.replace(' ',' ')
|
|
valid=True
|
|
for con in conditions:
|
|
res=con(content)
|
|
if res!=True: valid=False; await player.send(f"Please try again:\n{res}"); break
|
|
return msg
|
|
|
|
async def rancequit(args, channel, message):
|
|
# Get noobed
|
|
try: del rancegames[channel.id]; rancegames.pop(channel.id)
|
|
except KeyError: pass
|
|
|
|
# I might make this use the playrance function later.
|
|
async def rancehack(args, channel, message):
|
|
rancegames[channel.id]=board=automate.Board('a','b','','',{0:automate.unit.BattleUnit('tactician',1,9,3,9,400,4,{},'Ram'),2:automate.unit.BattleUnit('tactician',1,9,3,3,400,4,{},'Rem')},{4:automate.unit.BattleUnit('tactician',1,9,3,3,400,4,{},'Rom'),
|
|
2:automate.unit.BattleUnit('tactician',1,9,3,3,400,4,{},'Rim')})
|
|
board.turn=2
|
|
controller=board.controller()
|
|
nl='\n'
|
|
while True:
|
|
if not rancestatus(channel.id,board): return
|
|
# We can discard this during charged actions but if we don't get it shit goes wrong.
|
|
try: active=next(controller)
|
|
except StopIteration as ex: end=ex; break
|
|
if active is None:
|
|
# Charged action is happening. Don't wait for action or anything.
|
|
action=None
|
|
else:
|
|
await channel.send(f'```{board}```')
|
|
await channel.send(f"Available actions for {active[0][0]}'s unit {active[1]}:\n{nl.join([', '.join(n) for n in active[2]])}")
|
|
msg=await client.wait_for('message',check=lambda x:x.author!=client.user and x.channel.id==channel.id and x.content.split(' ')[0] in [n[0] for n in active[2]])
|
|
action=msg.content.split(' ')
|
|
res=controller.send(action); await channel.send(res)
|
|
#except Exception as ex: await channel.send(str(ex)+'\nThis might have broken the game state; please confirm and quit if so')
|
|
await channel.send(f'```{board}```')
|
|
await channel.send(end)
|
|
rancegames.pop(channel.id)
|
|
|
|
def rancestatus(cid,seed):
|
|
return cid in rancegames and rancegames[cid]==seed
|
|
|
|
async def azul(args, channel, message):
|
|
pass
|