Marty-v58/serverfiles/main/stop.py

#!/usr/local/bin/python3
#### @martysama0134 start scripts ####
### TODO fix -s -w for srvr and not only channels
from subprocess import check_output as sp_co, call as sp_call, CalledProcessError as sp_CalledProcessError
from os import getcwd as os_getcwd, chdir as os_chdir

from platform import system as p_system
v_system = p_system()

class M2TYPE:
	SERVER, DB, AUTH, CHANFOLDER, CHANNEL, CORE = list(range(6))
	NOCHAN = 0

def fShell(szCmd, bRet=False):
	try:
		if bRet:
			return sp_co(szCmd, shell=True)[:-1]	# remove final \n
		else:
			return sp_call(szCmd, shell=True)
	except sp_CalledProcessError:
		return -1

def keyCheck(dict, key, elem={}):
	try:
		dict[key]
	except KeyError:
		dict[key]=elem

proclist={}
whichlist={"serv":[], "chan":[]}
def staInit():
	global proclist
	## base
	from json import load as j_loads
	with open(".start.json", "r") as fList:
		mList = j_loads(fList)
	proclist.clear()
	for dic1 in mList:
		keyCheck(proclist, dic1["serv"])
		if dic1["type"]==M2TYPE.DB:
			keyCheck(proclist[dic1["serv"]], "db", [])
			proclist[dic1["serv"]]["db"].append(dic1)
		elif dic1["type"]==M2TYPE.AUTH or dic1["type"]==M2TYPE.CORE:
			keyCheck(proclist[dic1["serv"]], "core", [])
			proclist[dic1["serv"]]["core"].append(dic1)
			if dic1["type"]==M2TYPE.CORE:
				keyCheck(proclist[dic1["serv"]], "chan", set())
				proclist[dic1["serv"]]["chan"].add(dic1["chan"])

def staStart(serv=(), chan={}, level=1):
	# easy workaround for windows (only for debug purposes)
	if v_system=="Windows":
		levelFlag = ("/F" if level==9 else "")
		fShell("taskkill %s /IM game.exe /T"%(levelFlag))
		fShell("taskkill %s /IM db.exe /T"%(levelFlag))
		return
	from time import sleep as t_sleep
	# szPWD=os_getcwd()
	global whichlist
	whichlist["serv"]=serv
	whichlist["chan"]=chan
	keyCheck(whichlist["chan"], "all", ())
	def RunInMe(tmpProcList, bSkipCheck=False):
		for dic1 in tmpProcList:
			# skip not requested servers
			if (whichlist["serv"]) and (dic1["serv"] not in whichlist["serv"]):
				continue
			if not bSkipCheck and dic1["type"]==M2TYPE.CORE:
				# skip not requested channels
				keyCheck(whichlist["chan"], dic1["serv"], ())
				if whichlist["chan"]["all"]:
					# print("all",whichlist["chan"]["all"])
					if not dic1["chan"] in whichlist["chan"]["all"]:
						continue
				if whichlist["chan"][dic1["serv"]]:
					# print(dic1["serv"])
					if not dic1["chan"] in whichlist["chan"][dic1["serv"]]:
						continue
			# print(dic1)
			szPID=fShell("""ps axf | fgrep "./%s" | fgrep -v grep | awk '{print $1}'"""%dic1["name"], True)
			if szPID:
				dwPID=int(szPID)
				# kill process
				fShell("kill -%u %u"%(level, dwPID))
				print(("%s -> %u"%(dic1["name"], dwPID)))
			else: # do nothing
				print(("%s -> not found"%dic1["name"]))
	def CheckInMe(tmpProcList, bSkipCheck=False):
		for dic1 in tmpProcList:
			if (whichlist["serv"]) and (dic1["serv"] not in whichlist["serv"]):
				continue
			if not bSkipCheck and dic1["type"]==M2TYPE.CORE:
				keyCheck(whichlist["chan"], dic1["serv"], ())
				if whichlist["chan"]["all"]:
					if not dic1["chan"] in whichlist["chan"]["all"]:
						continue
				if whichlist["chan"][dic1["serv"]]:
					if not dic1["chan"] in whichlist["chan"][dic1["serv"]]:
						continue
			szPID=fShell("""ps axf | fgrep "./%s" | fgrep -v grep | awk '{print $1}'"""%dic1["name"], True)
			if szPID:
				dwPID=int(szPID)
				fShell("kill -%u %u"%(level, dwPID))
				print(("still waiting for... %s -> %u"%(dic1["name"], dwPID)))
				return False
		return True
	for k1 in proclist.keys():
		RunInMe(proclist[k1]["core"])
		while not CheckInMe(proclist[k1]["core"]):
			t_sleep(3)
		t_sleep(3)
		RunInMe(proclist[k1]["db"])
		t_sleep(1)
		while not CheckInMe(proclist[k1]["db"]):
			t_sleep(3)

if __name__ == "__main__":
	from getopt import getopt as g_getopt, GetoptError as g_GetoptError
	from sys import exit as s_exit, argv as s_argv
	try:
		bIsAll = True
		bHasPrompt = False
		baseChan = []
		listWhich = []
		szWhichServ = ""
		szWhichChan = {}
		dwLevel=1
		optlist, args = g_getopt(s_argv[1:],"l:ps",('level=','prompt','selective','whichserv=','whichchan='))
		for o, a in optlist:
			if o in ('-s', '--selective'):
				bIsAll=False
			elif o in ('-p', '--prompt'):
				bHasPrompt=True
			elif o in ('-l', '--level'):
				dwTmpLevel=int(a)
				if 1<=dwTmpLevel and dwTmpLevel<=9:
					dwLevel=dwTmpLevel
			elif o in ('-ws', '--whichserv'):
				szWhichServ=a
				if szWhichServ:
					bIsAll=False
			elif o in ('-wc', '--whichchan'):
				szWhichChan["all"]=a
				if szWhichChan:
					bIsAll=False
		staInit()
		if bHasPrompt:
			bIsAll=False
			print(("Servers available to run:", " ".join(iter(proclist.keys()))))
			szTmp1 = input("Enter which servers you want to stop: (nothing=all)\ne.g. srv1 srv2 srv3\n>>> ")
			if szTmp1.strip():
				szWhichServ=szTmp1
				for iChan in szTmp1.strip().split():
					if iChan not in iter(proclist.keys()):
						print(("Server not found:", iChan))
						continue
					print(("Channels available to run for %s:"%iChan, " ".join([str(i) for i in proclist[iChan]["chan"]])))
					szTmp2 = input("Enter which additional channels you want to stop: (nothing=all)\ne.g. 1 2 99\n>>> ")
					if szTmp2.strip():
						# print(szTmp1, szTmp2)
						szWhichChan[iChan]=szTmp2
		# print(proclist,whichlist)
		if bIsAll:
			staStart(level=dwLevel)
		else:
			tmpWhichServ,tmpWhichChan=[],{}
			if szWhichServ:
				tmpWhichServ=szWhichServ.split()
			if szWhichChan:
				for iKey in szWhichChan.keys():
					tmpWhichChan[iKey]=set([int(i) for i in szWhichChan[iKey].split()])
			# print(tmpWhichServ, "\n", tmpWhichChan)
			staStart(serv=tmpWhichServ, chan=tmpWhichChan, level=dwLevel)
	except g_GetoptError as err:
		s_exit(err)
#