mirror of
https://github.com/ammaraskar/pyCraft.git
synced 2024-11-24 19:25:33 +01:00
148 lines
3.9 KiB
Python
148 lines
3.9 KiB
Python
import struct
|
|
import types
|
|
from io import BytesIO
|
|
from pynbt import NBTFile
|
|
|
|
|
|
def readBoolean(FileObject):
|
|
return struct.unpack('?', FileObject.read(1))[0]
|
|
|
|
|
|
def readByte(FileObject):
|
|
return struct.unpack('>b', FileObject.read(1))[0]
|
|
|
|
|
|
def readUnsignedByte(FileObject):
|
|
return struct.unpack('>B', FileObject.read(1))[0]
|
|
|
|
|
|
def readShort(FileObject):
|
|
return struct.unpack('>h', FileObject.read(2))[0]
|
|
|
|
|
|
def readUnsignedShort(FileObject):
|
|
return struct.unpack('>H', FileObject.read(2))[0]
|
|
|
|
|
|
def readInt(FileObject):
|
|
return struct.unpack('>i', FileObject.read(4))[0]
|
|
|
|
|
|
def readFloat(FileObject):
|
|
return struct.unpack('>f', FileObject.read(4))[0]
|
|
|
|
|
|
def readLong(FileObject):
|
|
return struct.unpack('>q', FileObject.read(8))[0]
|
|
|
|
|
|
def readDouble(FileObject):
|
|
return struct.unpack('>d', FileObject.read(8))[0]
|
|
|
|
|
|
def readByteArray(FileObject, length):
|
|
return struct.unpack(str(length) + "s", FileObject.read(length))[0]
|
|
|
|
|
|
def readString(FileObject):
|
|
length = readShort(FileObject) * 2
|
|
return unicode(FileObject.read(length), "utf-16be")
|
|
|
|
|
|
def sendBoolean(socket, value):
|
|
assert type(value) is types.BooleanType, "value is not a boolean: %r" % value
|
|
socket.send(struct.pack('?', value))
|
|
|
|
|
|
def sendByte(socket, value):
|
|
socket.send(struct.pack('>b', value))
|
|
|
|
|
|
def sendUnsignedByte(socket, value):
|
|
socket.send(struct.pack('>B', value))
|
|
|
|
|
|
def sendShort(socket, value):
|
|
socket.send(struct.pack('>h', value))
|
|
|
|
|
|
def sendUnsignedShort(socket, value):
|
|
socket.send(struct.pack('>H', value))
|
|
|
|
|
|
def sendInt(socket, value):
|
|
assert type(value) is types.IntType, "value is not an integer: %r" % value
|
|
socket.send(struct.pack('>i', value))
|
|
|
|
|
|
def sendFloat(socket, value):
|
|
socket.send(struct.pack('>f', value))
|
|
|
|
|
|
def sendLong(socket, value):
|
|
socket.send(struct.pack('>q', value))
|
|
|
|
|
|
def sendDouble(socket, value):
|
|
socket.send(struct.pack('>d', value))
|
|
|
|
|
|
def sendString(socket, value):
|
|
value = unicode(value).encode('utf-16be')
|
|
socket.send(struct.pack('>h', len(value) / 2))
|
|
socket.send(value)
|
|
|
|
|
|
def readEntityMetadata(FileObject):
|
|
metadata = {}
|
|
byte = readUnsignedByte(FileObject)
|
|
while byte != 127:
|
|
index = byte & 0x1F # Lower 5 bits
|
|
ty = byte >> 5 # Upper 3 bits
|
|
if ty == 0: val = readByte(FileObject)
|
|
if ty == 1: val = readShort(FileObject)
|
|
if ty == 2: val = readInt(FileObject)
|
|
if ty == 3: val = readFloat(FileObject)
|
|
if ty == 4:
|
|
val = readString(FileObject)
|
|
if ty == 5:
|
|
val = {}
|
|
val["id"] = readShort(FileObject)
|
|
if (val["id"] != -1):
|
|
val["count"] = readByte(FileObject)
|
|
val["damage"] = readShort(FileObject)
|
|
nbtDataLength = readShort(FileObject)
|
|
if (nbtDataLength != -1):
|
|
val["NBT"] = NBTFile(BytesIO(readByteArray(FileObject, nbtDataLength)),
|
|
compression=NBTFile.Compression.GZIP)
|
|
if ty == 6:
|
|
val = []
|
|
for i in range(3):
|
|
val.append(readInt(FileObject))
|
|
metadata[index] = (ty, val)
|
|
byte = readUnsignedByte(FileObject)
|
|
return metadata
|
|
|
|
|
|
def readSlotData(FileObject):
|
|
BlockID = readShort(FileObject)
|
|
if (BlockID != -1):
|
|
ItemCount = readByte(FileObject)
|
|
Damage = readShort(FileObject)
|
|
MetadataLength = readShort(FileObject)
|
|
if (MetadataLength != -1):
|
|
ByteArray = readByteArray(FileObject, MetadataLength)
|
|
NBTData = NBTFile(BytesIO(ByteArray), compression=NBTFile.Compression.GZIP)
|
|
return {'BlockID': BlockID,
|
|
'ItemCount': ItemCount,
|
|
'Damage': Damage,
|
|
'Data': NBTData
|
|
}
|
|
return {'BlockID': BlockID,
|
|
'ItemCount': ItemCount,
|
|
'Damage': Damage
|
|
}
|
|
return {'BlockID': -1,
|
|
'ItemCount': 0
|
|
}
|