mesecode/mesecode.py

262 lines
6.9 KiB
Python

import re, sys, os
# How many spaces in a tab.
# -1 disables support for spaces as tabs (recommended)
# There are usually 4 spaces in a tab
SPACES_PER_TAB = -1
# Lua Libraries:
eat_lib = """function mesecode.item_eat(amt)
if minetest.get_modpath("diet") then
return diet.item_eat(amt)
elseif minetest.get_modpath("hud") then
return hud.item_eat(amt)
else
return minetest.item_eat(amt)
end
end"""
def throwParseError(msg):
print("\033[91mParse Error: " + msg + "\033[0m")
sys.exit(-1)
def checkMkDir(directory):
if not os.path.exists(directory):
os.makedirs(directory)
class MeseCodeParser:
class Node:
def __init__(self, parent, name, value, line, lineno):
self.line = line
self.lineno = lineno
self.name = name
self.value = value
self.children = []
def get(self, name):
for item in self.children:
if item.name == name:
return item
return None
def as_list(self):
retval = []
for item in self.value.split(","):
item = item.strip()
if item != "":
retval.append(item)
for item in self.children:
if item.line != "":
retval.append(item.line)
return retval
def parse(self, filename):
file = open(filename, "r")
self.objects = []
lineno = 0
for line in file.readlines():
lineno += 1
# Remove comments
if line.find("--") != -1:
line = line[:line.find("--")]
if line.strip() == "":
continue
# Find indentation level
if SPACES_PER_TAB != -1:
line = line.replace(SPACES_PER_TAB * " ", "\t")
indented = 0
m = re.search('([\\t]+)', line)
if m:
indented = len(m.group(1))
# Strip redundant symbols
line = line.strip()
if indented == 0:
# Is top level
self.objects.append(self.createNode(None, line, lineno))
else:
count = 1
if len(self.objects) == 0:
throwParseError("Unexpected level of indentation on line " + str(lineno))
node = self.objects[len(self.objects) - 1]
while count < indented:
if len(node.children) == 0:
throwParseError("Unexpected level of indentation on line " + str(lineno))
node = node.children[len(node.children) - 1]
count += 1
node.children.append(self.createNode(node, line, lineno))
return self
def __iter__(self):
return self.objects.__iter__()
def printOut(self, l, objs):
for obj in objs:
print((l*" ") + obj.name + ": " + obj.value)
self.printOut(l + 1, obj.children)
def createNode(self, parent, line, lineno):
return self.Node(parent, line.split(" ")[0], line[len(line.split(" ")[0]) + 1:], line, lineno)
class LuaBuilder:
class Node:
def __init__(self, name, value):
self.name = name
self.value = value
def __init__(self):
self.data = []
def set(self, name, value):
for item in self.data:
if item.name == name:
del item
break
self.data.append(self.Node(name, value))
def set_string(self, name, value):
self.set(name, "\"" + value + "\"")
def append(self, name, value):
for item in self.data:
if item.name == name:
item.value.append(value)
return
self.data.append(self.Node(name, [value]))
def build(self, header, indentation):
retval = header + "{\n"
for item in self.data:
retval += (indentation * "\t") + item.name + " = "
if isinstance(item.value, list):
retval += "{"
for i in item.value:
retval += i + ", "
retval += "}"
else:
retval += item.value
retval += ",\n"
retval += "})\n"
return retval
def getNameFromItem(modname, item):
name = item.get("name")
if name is None:
name = item.name.lower().replace(" ", "_")
else:
name = name.value
if name.find(":") == -1:
return modname + ":" + name
else:
return name
def interpretItem(project, item, lua):
lua.set_string("description", item.value)
groups = item.get("is")
if groups is not None:
for group in groups.as_list():
if group == "ground":
lua.set("is_ground_content", "true")
continue
if group.find("=") == -1:
lua.append("groups", group + " = 1")
else:
lua.append("groups", group)
eaten = item.get("eaten")
if eaten is not None:
project.requires_eat = True
lua.set("on_use", "mesecode.item_eat(" + eaten.value + ")")
def interpretNode(project, item, lua):
interpretItem(project, item, lua)
drops = item.get("drops")
if drops is not None:
for drop in drops.as_list():
try:
if drop.find(":") == -1:
lua.append("drops", "\"" + getNameFromItem(project.modname, project.index[drop]) + "\"")
else:
lua.append("drops", "\"" + drop + "\"")
except KeyError:
throwParseError("Unable to find an item called '" + drop + "' on line " + str(drops.lineno) + " (Did you forget a : at the start?)")
class MeseCodeProject:
def __init__(self, filename, directory):
self.parser = MeseCodeParser().parse(filename)
self.modname = None
self.requires_eat = False
self.index = {}
# Open output directory
directory = directory.strip()
if directory[len(directory)-1] != "/":
directory += "/"
checkMkDir(directory)
depends = open(directory + "depends.txt", "w")
# Build index and find mod name
for item in self.parser:
if item.name == "mod":
if self.modname is not None:
throwParseError("Mod namespace was redefined on line " + str(item.lineno))
self.modname = item.value
elif self.modname is None:
throwParseError("Mod namespace was not defined (You missed out 'mod nameofmod' at the beginning of the file)")
if item.name == "node" or item.name == "craftitem":
self.index[item.value] = item
# Check for modname
if self.modname is None:
throwParseError("Mod namespace was not defined (You missed out 'mod nameofmod' at the beginning of the file)")
# Build init.lua
retval = ""
for item in self.parser:
if item.name == "craftitem":
lb = LuaBuilder()
interpretNode(self, item, lb)
retval += lb.build("minetest.register_craftitem(\"" + getNameFromItem(self.modname, item) + "\", ", 1) + "\n"
elif item.name == "node":
lb = LuaBuilder()
interpretNode(self, item, lb)
retval += lb.build("minetest.register_node(\"" + getNameFromItem(self.modname, item) + "\", ", 1) + "\n"
elif item.name == "script":
retval += "dofile(minetest.get_modpath(\"" + self.modname + "\") .. \"" + item.value + "\")\n\n"
elif item.name == "requires":
depends.write(item.value + "\n")
elif item.name == "depends":
depends.write(item.value + "\n")
elif item.name == "uses":
depends.write(item.value + "?\n")
libs = ""
if self.requires_eat:
if libs == "":
libs = "local mesecode = {}\n"
libs += eat_lib + "\n\n"
depends.write("diet?\nhud?\n")
depends.close()
print(libs + retval)
output = open(directory + "init.lua", "w")
output.write(libs + retval)
output.close()
if __name__ == "__main__":
if len(sys.argv) == 2:
Project(sys.argv[1]).write("output")
elif len(sys.argv) == 3:
Project(sys.argv[1]).write(sys.argv[2])
else:
print("Usage: mesecode.py path/to/file.mese output/directory")