def __init__(self, pos, etype = "unknown", content = ""):

self.pos = pos

self.content = content

self.type = etype

def getPos(self):

return self.pos

def getContent(self):

return self.content

def addContent(self, newcontent):

self.content += newcontent

def getType(self):

return self.type

# get the leading comment for this element

def getComment(self):

rValue = ""

for l in self.content.split("\n"):

if l[0] == "#" or l == "":

rValue += l+"\n"

return rValue

# get the name of the function or variable, return "" for statements or unknown

def getName(self):

if self.type == "unknown" or self.type == "statement":

return ""

elif self.type == "variable":

for l in self.content.split("\n"):

vartest = re.search("^[A-Z][_A-Za-z0-9]*=", l)

if vartest:

return l.split("=")[0]

return ""

elif self.type == "function":

for l in self.content.split("\n"):

functest = re.search('^[a-zA-Z][A-Za-z0-9_-]*[A-Za-z0-9] ?\(\) ?\{', l)

if functest:

return l.replace("()", "").replace("{", "").strip()

return ""

else:

return ""

# get the body of a function, the value of a variable or the code of a statement

def getBody(self):

header_passed = False

rValue = ""

for l in self.content.split("\n"):

header_passed = header_passed or (l != "" and l[0] != "#" )

if header_passed and (rValue != "" or l.strip() != ""):

if self.type == "unknown" or self.type == "statement":

rValue += l+"\n"

elif self.type == "variable":

if l[:len(self.getName())] == self.getName():

rValue += l[len(self.getName())+1:]+"\n"

else:

rValue += l+"\n"

elif self.type == "function" and l.strip() != "}":

rValue += l+"\n"

if self.type == "function":

rValue = reduce(lambda a,b: a+"\n"+b, rValue.split("\n")[1:-1]) # drop first and last line

# kind of default constructor to set all fields to sane values

def __localinit__(self):

self.filename = ""

self.location = ""

self.category = ""

self.package = ""

self.version = ""

self.elements = { "functions": [], "variables": [], "statements": [] }

self.header = ""

self.content = "" # only for internal use, call getContent() if you want the content

# constructor to get the ebuild by filename

def __init__(self, filename):

self.__localinit__()

# check the location

allportdirs = portdir + overlays

for d in allportdirs:

if filename[:len(d)] == d:

self.location = d

if len(self.location) <= 1:

raise Exception("invalid ebuild directory")

if self.location[-1] == "/":

self.location = self.location[:-1]

s = string.split(filename, "/")

self.filename = s[len(s)-1]

self.package = s[len(s)-2]

self.category = s[len(s)-3]

# use portage to get the version part

self.version = portage.catpkgsplit(self.category+"/"+self.filename.replace(".ebuild", ""))

if self.version[3] == "r0":

self.version = self.version[2]

else:

self.version = self.version[2]+"-"+self.version[3]

self.path = self.location+"/"+self.category+"/"+self.package+"/"+self.filename

# constructor to get the ebuild by location, category, package and version

def __init__(self, category, package, version, location=portdir):

self.__localinit__()

self.location = location

if self.location[-1] == "/":

self.location = self.location[:-1]

self.category = category

self.package = package

self.version = version

self.filename = package+"-"+version+".ebuild"

self.path = self.location+"/"+self.category+"/"+self.package+"/"+self.filename

# return the complete content of the ebuild

def getContent(self):

myelements = []

for l in self.elements.values():

myelements += l

myelements.sort(lambda x,y: x.getPos() - y.getPos())

mycontent = self.header+"\n"

for e in myelements:

mycontent += e.getContent()+"\n"

self.content = mycontent

return mycontent

# read and parse the ebuild

def readEbuild(self):

if not os.path.exists(self.path):

return ERR_FILE_NOT_FOUND

# check for bash syntax errors

if os.system("/bin/bash -n " + self.path) != 0:

return ERR_SYNTAX

# read the complete content

myfile = open(self.path, "r")

self.content = myfile.readlines()

myfile.close()

# now parse the whole thing

ln = 0 # linenumber

l = "" # current line

pos = 0 # current element position

header_finished = False

myelement = None

tmpcontent = ""

while ln < len(self.content):

l = self.content[ln]

ln += 1

# check for header

if not header_finished and l[0] == "#":

self.header += l

continue

header_finished = True

# now read the different elements (functions, variables, statements)

if myelement == None and l[0] == "#": # no current element and line is a comment

tmpcontent +=l # add line to temporary content

continue

elif myelement != None and l[0] == "#": # we have a element and line is a comment

myelement.addContent(l) # add line to current element

continue

vartest = re.search('^[A-Z]+.*= ?', l)

functest = re.search('^[a-zA-Z]*[_a-zA-Z0-9]* ?\(\) ?{', l)

# variable here

if vartest:

v = l

#Multi-line variables

if (l.count('"') - l.count('\\\"')) == 1:

while (self.content[ln].count('"') - self.content[ln].count('\\\"')) != 1:

l = self.content[ln]

ln += 1

v += l

l = self.content[ln]

ln += 1

v += l

myelement = element(pos, "variable", string.strip(tmpcontent+"\n"+v+"\n"))

pos += 1

self.elements['variables'].append(myelement)

myelement = None

tmpcontent = ""

continue

# function here

if functest:

myelement = element(pos, "function", tmpcontent+"\n")

pos += 1

while l.strip() != "}":

myelement.addContent(l)

l = self.content[ln]

ln += 1

myelement.addContent(l)

self.elements['functions'].append(myelement)

myelement = None

tmpcontent = ""

continue

# something else

if l.strip() == "":

continue

if myelement == None:

myelement = element(pos, "statement", tmpcontent+"\n"+l)

pos += 1

else:

myelement.addcontent(l)

self.elements['statements'].append(myelement)

myelement = None

# write the ebuild to disk

def writeEbuild(self):

f = open(self.path, "w")

f.write(self.getContent())