150 lines
3.4 KiB
Python
150 lines
3.4 KiB
Python
import pygame
|
|
import sys
|
|
def my_load(path):
|
|
return None
|
|
mod=sys.modules["pygame.image"]
|
|
mod.load=my_load
|
|
sys.modules["pygame.image"]=mod
|
|
import os
|
|
import random
|
|
import select
|
|
import socket
|
|
import _thread
|
|
import pickle
|
|
import signal
|
|
import glob
|
|
import os.path
|
|
import lib.constants as constants
|
|
from lib.gameregistry import GameRegistry
|
|
from lib.map import Map
|
|
from lib.character import Character
|
|
from lib.block import Block
|
|
from lib.player import Player
|
|
from time import sleep
|
|
import blocks
|
|
import pprint
|
|
player_map={}
|
|
pos_map={}
|
|
map_changes={}
|
|
next_x=0
|
|
|
|
def recv_str(sock):
|
|
str=""
|
|
ch=""
|
|
while True:
|
|
ch=sock.recv(1).decode("utf-8")
|
|
if ch=="\n":
|
|
break
|
|
str+=ch
|
|
# print("Got string: "+str)
|
|
return str
|
|
|
|
def send_str(sock,str,print_str=True):
|
|
if print_str:
|
|
pass
|
|
# print("Sending string: "+str)
|
|
sock.send((str+"\n").encode("utf-8"))
|
|
|
|
def send_hash(sock,hash):
|
|
# print("Sending hash: "+pp.pformat(hash))
|
|
send_str(sock,str(len(hash)),False)
|
|
for key,val in hash.items():
|
|
send_str(sock,str(key),False)
|
|
send_str(sock,str(val),False)
|
|
|
|
def recvall(sock):
|
|
BUFF_SIZE=4096
|
|
data=b''
|
|
while True:
|
|
part=sock.recv(BUFF_SIZE)
|
|
data+=part
|
|
if len(part)<BUFF_SIZE:
|
|
break
|
|
# print("Got data: "+pp.pformat(pickle.loads(data)))
|
|
return data
|
|
|
|
def on_new_client(sock):
|
|
global player_map
|
|
global pos_map
|
|
global map
|
|
global map_changes
|
|
global next_x
|
|
username=recv_str(sock)
|
|
player_map[username]=sock
|
|
pos_map[username]=(next_x,0,"down")
|
|
map_changes[username]=[]
|
|
next_x+=1
|
|
while True:
|
|
msg=recv_str(sock)
|
|
if msg=="CLOSE":
|
|
del player_map[username]
|
|
del pos_map[username]
|
|
del map_changes[username]
|
|
elif msg=="GET_POS_MAP":
|
|
send_hash(sock,pos_map)
|
|
elif msg=="GET_POS":
|
|
send_str(sock,str(pos_map[username][0]))
|
|
send_str(sock,str(pos_map[username][1]))
|
|
send_str(sock,pos_map[username][2])
|
|
elif msg=="SET_POS":
|
|
pos_map[username]=(int(recv_str(sock)),int(recv_str(sock)),recv_str(sock))
|
|
elif msg=="BLOCK_AT_POS":
|
|
x=int(recv_str(sock))
|
|
y=int(recv_str(sock))
|
|
sock.send(pickle.dumps(map.tileAt(x,y)))
|
|
def handle_cmds():
|
|
while True:
|
|
if select.select([sys.stdin],[],[],0.0)[0]:
|
|
cmd=input()
|
|
cmd=cmd.split()
|
|
if len(cmd)>0:
|
|
if cmd[0]=="stop":
|
|
s.close()
|
|
f=open("worlds/map_{}.pkl".format(map_name),"wb")
|
|
pickle.dump(map,f)
|
|
f.close()
|
|
exit(1)
|
|
|
|
def exit_cleanup(signal,frame):
|
|
s.close()
|
|
f=open("worlds/map_{}.pkl".format(map_name),"wb")
|
|
pickle.dump(map,f)
|
|
f.close()
|
|
|
|
Block.init()
|
|
files=glob.glob("worlds/map_*.pkl")
|
|
if len(files)==0:
|
|
new="y"
|
|
else:
|
|
new=input("New world?").lower()
|
|
if new=="n" or new=="no":
|
|
i=1
|
|
map_map=[]
|
|
for name in files:
|
|
name=name.split("_")
|
|
name.pop(0)
|
|
name="_".join(name)
|
|
name,_=os.path.splitext(name)
|
|
print("{}. {}".format(i,name))
|
|
map_map.append(name)
|
|
i+=1
|
|
map_name=map_map[int(input("Which world?"))-1]
|
|
f=open("worlds/map_{}.pkl".format(map_name),"rb")
|
|
map=pickle.load(f)
|
|
f.close()
|
|
else:
|
|
map_name=input("World name:")
|
|
map=Map(None)
|
|
|
|
s=socket.socket()
|
|
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|
host="localhost"
|
|
port=30000
|
|
s.bind((host, port))
|
|
s.listen(5)
|
|
_thread.start_new_thread(handle_cmds,())
|
|
signal.signal(signal.SIGINT, exit_cleanup)
|
|
while True:
|
|
c,addr=s.accept()
|
|
_thread.start_new_thread(on_new_client,(c,))
|