import sys, os, which
import process
import koprocessutils
from xpcom import components, COMException, nsError
from xml.dom.minidom import parseString

wwatch = components.classes["@mozilla.org/embedcomp/window-watcher;1"].getService(components.interfaces.nsIWindowWatcher)
prompt = wwatch.getNewPrompter(wwatch.activeWindow)
userEnv = koprocessutils.getUserEnv()
userPath = userEnv["PATH"].split(os.pathsep)

def executeCmd(argv, cwd=None):
    try:
        p = process.ProcessOpen(argv, cwd=cwd, env=userEnv)
        return [p.stdout.read(), p.stderr.read()]
    finally:
        p.close()

def findZipper():
    # look for zip or 7z
    if sys.platform.startswith('win'):
        exts = ['.exe']
    else:
        exts = None
    # get the current valid environment which is a combination of
    # the startup environment and modifications in komodo preferences
    for exe in [['zip','-Dr'], ['7z','a','-tzip']]:
        zip = which.whichall(exe[0], exts=exts, path=userPath)
        if zip:
            return exe
    else:
        prompt.alert("XPI Build Error","Unable to find 'zip' or '7z' on path")
        raise COMException(nsError.NS_ERROR_UNEXPECTED, "Unable to find 'zip' or '7z' on path")

def getExtVars(rdf_file):
    # parse install.rdf for the vars we want
    rdf_obj = parseString(open(rdf_file, 'r').read())
    ext_id = rdf_obj.getElementsByTagName('em:id')[0].firstChild.data
    ext_name = rdf_obj.getElementsByTagName('em:name')[0].firstChild.data
    ext_version = rdf_obj.getElementsByTagName('em:version')[0].firstChild.data
    return {
        'ext_id': ext_id,
        'ext_name': ext_name,
        'ext_version': ext_version
    }

# get extension data from install.rdf
projectFile = komodo.macro.project.getFile()

vars = getExtVars(os.path.join(projectFile.dirName, 'install.rdf'))

# build the xpi

dirs = ['content', 'skin', 'locale', 'components']
cmd = []
zipper = findZipper()
for dir in dirs:
    path = os.path.join(projectFile.dirName, dir);
    if os.path.exists(path):
        cmd = zipper + ['%s.jar' % vars['ext_name'], dir]
        executeCmd(cmd, projectFile.dirName)

cmd = zipper + ["%s-%s.xpi" % (vars['ext_name'], vars['ext_version']), '%s.jar' % vars['ext_name'], "install.rdf", "chrome.manifest"]
executeCmd(cmd, projectFile.dirName)