1#!/usr/bin/env python 2# encoding: utf-8 3# Thomas Nagy, 2010 (ita) 4# Przemyslaw Rzepecki, 2016 5 6""" 7Erlang support 8""" 9 10import re 11from waflib import Task, TaskGen 12from waflib.TaskGen import feature, after_method, before_method 13# to load the method "to_incnodes" below 14from waflib.Tools import ccroot 15 16# Those flags are required by the Erlang VM to execute/evaluate code in 17# non-interactive mode. It is used in this tool to create Erlang modules 18# documentation and run unit tests. The user can pass additional arguments to the 19# 'erl' command with ERL_FLAGS environment variable. 20EXEC_NON_INTERACTIVE = ['-noshell', '-noinput', '-eval'] 21 22def configure(conf): 23 conf.find_program('erlc', var='ERLC') 24 conf.find_program('erl', var='ERL') 25 conf.add_os_flags('ERLC_FLAGS') 26 conf.add_os_flags('ERL_FLAGS') 27 conf.env.ERLC_DEF_PATTERN = '-D%s' 28 conf.env.ERLC_INC_PATTERN = '-I%s' 29 30@TaskGen.extension('.erl') 31def process_erl_node(self, node): 32 tsk = self.create_task('erl', node, node.change_ext('.beam')) 33 tsk.erlc_incnodes = [tsk.outputs[0].parent] + self.to_incnodes(self.includes) 34 tsk.env.append_value('ERLC_INCPATHS', [x.abspath() for x in tsk.erlc_incnodes]) 35 tsk.env.append_value('ERLC_DEFINES', self.to_list(getattr(self, 'defines', []))) 36 tsk.env.append_value('ERLC_FLAGS', self.to_list(getattr(self, 'flags', []))) 37 tsk.cwd = tsk.outputs[0].parent 38 39class erl(Task.Task): 40 color = 'GREEN' 41 run_str = '${ERLC} ${ERL_FLAGS} ${ERLC_INC_PATTERN:ERLC_INCPATHS} ${ERLC_DEF_PATTERN:ERLC_DEFINES} ${SRC}' 42 43 def scan(task): 44 node = task.inputs[0] 45 46 deps = [] 47 scanned = set([]) 48 nodes_to_scan = [node] 49 50 for n in nodes_to_scan: 51 if n.abspath() in scanned: 52 continue 53 54 for i in re.findall(r'-include\("(.*)"\)\.', n.read()): 55 for d in task.erlc_incnodes: 56 r = d.find_node(i) 57 if r: 58 deps.append(r) 59 nodes_to_scan.append(r) 60 break 61 scanned.add(n.abspath()) 62 63 return (deps, []) 64 65@TaskGen.extension('.beam') 66def process(self, node): 67 pass 68 69 70class erl_test(Task.Task): 71 color = 'BLUE' 72 run_str = '${ERL} ${ERL_FLAGS} ${ERL_TEST_FLAGS}' 73 74@feature('eunit') 75@after_method('process_source') 76def add_erl_test_run(self): 77 test_modules = [t.outputs[0] for t in self.tasks] 78 test_task = self.create_task('erl_test') 79 test_task.set_inputs(self.source + test_modules) 80 test_task.cwd = test_modules[0].parent 81 82 test_task.env.append_value('ERL_FLAGS', self.to_list(getattr(self, 'flags', []))) 83 84 test_list = ", ".join([m.change_ext("").path_from(test_task.cwd)+":test()" for m in test_modules]) 85 test_flag = 'halt(case lists:all(fun(Elem) -> Elem == ok end, [%s]) of true -> 0; false -> 1 end).' % test_list 86 test_task.env.append_value('ERL_TEST_FLAGS', EXEC_NON_INTERACTIVE) 87 test_task.env.append_value('ERL_TEST_FLAGS', test_flag) 88 89 90class edoc(Task.Task): 91 color = 'BLUE' 92 run_str = "${ERL} ${ERL_FLAGS} ${ERL_DOC_FLAGS}" 93 def keyword(self): 94 return 'Generating edoc' 95 96@feature('edoc') 97@before_method('process_source') 98def add_edoc_task(self): 99 # do not process source, it would create double erl->beam task 100 self.meths.remove('process_source') 101 e = self.path.find_resource(self.source) 102 t = e.change_ext('.html') 103 png = t.parent.make_node('erlang.png') 104 css = t.parent.make_node('stylesheet.css') 105 tsk = self.create_task('edoc', e, [t, png, css]) 106 tsk.cwd = tsk.outputs[0].parent 107 tsk.env.append_value('ERL_DOC_FLAGS', EXEC_NON_INTERACTIVE) 108 tsk.env.append_value('ERL_DOC_FLAGS', 'edoc:files(["%s"]), halt(0).' % tsk.inputs[0].abspath()) 109 # TODO the above can break if a file path contains '"' 110 111