# a waf tool to add extension based build patterns for Samba

import os, sys, Options
import string, Task, Utils, optparse
from Configure import conf
from Logs import debug
from TaskGen import extension
from samba_utils import *

################################################################################
# a et task which calls out to compile_et to do the work
Task.simple_task_type('et',
		      '../heimdal_build/et_compile_wrapper.sh . ${TGT[0].bld_dir(env)} default/source4/heimdal_build/compile_et ${SRC[0].abspath(env)} ${TGT[0].bldpath(env)}',
                      color='BLUE', ext_out='.c',
                      shell = False)

@extension('.et')
def process_et(self, node):
    c_node = node.change_ext('.c')
    h_node  = node.change_ext('.h')
    self.create_task('et', node, [c_node, h_node])
    self.allnodes.append(c_node)




# this was my (awful) attempt at a ASN1 rule, i plan on moving it to samba_asn1.py,
# like I did for the PIDL rules, and conver to a make-like rule

################################################################################
# a asn1 task which calls out to asn1_compile_wrapper.sh to do the work
Task.simple_task_type('asn1',
		      '''
# this is the rule for now - ie. an embedded shell script!! like
# I showed you previously, the ASN1 compiler generates weird names,
# we need to fix the names, and add config.h header

# shell script to convert ASN1 to C. This could be separated out if we want to
set -e
compiler=${TGT[0].compiler}
destdir=${TGT[0].destdir}
wrapper=${TGT[0].asn1wrapper}
srcfile=${SRC[0].abspath(env)}
asn1name=${TGT[0].asn1name}
options="${TGT[0].asn1options}"

# run the wrapper
$wrapper . $destdir $compiler $srcfile $asn1name ${options} --one-code-file

# that generated 3 files:
#    ${asn1name}.hx
#    asn1_${asn1name}.x
#    ${asn1name}_files


hxfile=$destdir/$asn1name.hx
xfile=$destdir/asn1_$asn1name.x
listfilee=$destdir/"$asn1name"_files

cfile=${TGT[0].abspath(env)}
hfile=${TGT[1].abspath(env)}

cp $hxfile $hfile
echo '#include "config.h"' > $cfile
cat $xfile >> $cfile
rm -f $listfile

''',
                      color='BLUE',
                      ext_out='.c',
                      shell = True)

@extension('.asn1')
def process_asn1(self, node):

    asn1name = string.replace(node.file(), '.', '_')
    c_node  = NEW_NODE(node, 'asn1_%s.c' % asn1name)
    h_node  = NEW_NODE(node, '%s.h' % asn1name)

    c_node.destdir      = "default/source4/heimdal/" + self.asn1directory
    c_node.asn1options  = self.asn1options
    c_node.asn1name     = asn1name
    c_node.asn1wrapper  = "../heimdal_build/asn1_compile_wrapper.sh"
    c_node.compiler     = "default/source4/heimdal_build/asn1_compile"

    self.create_task('asn1', node, [c_node, h_node])
    self.allnodes.append(c_node)