waf/waflib/extras/wix.py

88 lines
2.6 KiB
Python

#!/usr/bin/python
# encoding: utf-8
# vim: tabstop=4 noexpandtab
"""
Windows Installer XML Tool (WiX)
.wxs --- candle ---> .wxobj --- light ---> .msi
bld(features='wix', some.wxs, gen='some.msi', candleflags=[..], lightflags=[..])
bld(features='wix', source=['bundle.wxs','WixBalExtension'], gen='setup.exe', candleflags=[..])
"""
import os, copy
from waflib import TaskGen
from waflib import Task
from waflib.Utils import winreg
class candle(Task.Task):
run_str = '${CANDLE} -nologo ${CANDLEFLAGS} -out ${TGT} ${SRC[0].abspath()}',
class light(Task.Task):
run_str = "${LIGHT} -nologo -b ${SRC[0].parent.abspath()} ${LIGHTFLAGS} -out ${TGT} ${SRC[0].abspath()}"
@TaskGen.feature('wix')
@TaskGen.before_method('process_source')
def wix(self):
#X.wxs -> ${SRC} for CANDLE
#X.wxobj -> ${SRC} for LIGHT
#X.dll -> -ext X in ${LIGHTFLAGS}
#X.wxl -> wixui.wixlib -loc X.wxl in ${LIGHTFLAGS}
wxobj = []
wxs = []
exts = []
wxl = []
rest = []
for x in self.source:
if x.endswith('.wxobj'):
wxobj.append(x)
elif x.endswith('.wxs'):
wxobj.append(self.path.find_or_declare(x[:-4]+'.wxobj'))
wxs.append(x)
elif x.endswith('.dll'):
exts.append(x[:-4])
elif '.' not in x:
exts.append(x)
elif x.endswith('.wxl'):
wxl.append(x)
else:
rest.append(x)
self.source = self.to_nodes(rest) #.wxs
cndl = self.create_task('candle', self.to_nodes(wxs), self.to_nodes(wxobj))
lght = self.create_task('light', self.to_nodes(wxobj), self.path.find_or_declare(self.gen))
cndl.env.CANDLEFLAGS = copy.copy(getattr(self,'candleflags',[]))
lght.env.LIGHTFLAGS = copy.copy(getattr(self,'lightflags',[]))
for x in wxl:
lght.env.append_value('LIGHTFLAGS','wixui.wixlib')
lght.env.append_value('LIGHTFLAGS','-loc')
lght.env.append_value('LIGHTFLAGS',x)
for x in exts:
cndl.env.append_value('CANDLEFLAGS','-ext')
cndl.env.append_value('CANDLEFLAGS',x)
lght.env.append_value('LIGHTFLAGS','-ext')
lght.env.append_value('LIGHTFLAGS',x)
#wix_bin_path()
def wix_bin_path():
basekey = r"SOFTWARE\Microsoft\.NETFramework\AssemblyFolders"
query = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, basekey)
cnt=winreg.QueryInfoKey(query)[0]
thiskey = r'C:\Program Files (x86)\WiX Toolset v3.10\SDK'
for i in range(cnt-1,-1,-1):
thiskey = winreg.EnumKey(query,i)
if 'WiX' in thiskey:
break
winreg.CloseKey(query)
return os.path.normpath(winreg.QueryValue(winreg.HKEY_LOCAL_MACHINE, basekey+r'\\'+thiskey)+'..\\bin')
def configure(ctx):
path_list=[wix_bin_path()]
ctx.find_program('candle', var='CANDLE', mandatory=True, path_list = path_list)
ctx.find_program('light', var='LIGHT', mandatory=True, path_list = path_list)