es/macros.es at master · mveety/es · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#!/usr/bin/env es

if {~ $#es_internal_symcount 0} {
	es_internal_symcount = 0
}

if {~ $#es_conf_tempdir 0} {
	local (set-es_conf_tempdir=) {
		es_conf_tempdir = '/tmp'
	}
}

set-es_conf_tempdir = @ arg _ {
	if {~ $arg */} {
		arg = <={~~ $arg */}
	}
	result $arg
}

fn gensym prefix {
	prefix = <={if {~ $#prefix 0} { result '__es_sym_'^$pid } { result $prefix }}
	es_internal_symcount = <={add $es_internal_symcount 1}
	result $prefix^$es_internal_symcount
}

fn tempfile prefix dir _ {
	let (t=){
		if {~ $prefix */*} {
			t = $dir
			dir = $prefix
			prefix = $t
		}
	}
	if {~ $#prefix 0} { prefix = 'tempfile' }
	if {~ <={%count $:prefix} 0} { prefix = 'tempfile' }
	let (
		filename = <={gensym $prefix^'.es'^$pid^'.'}^'.tmp'
		tmpfiledir = <={if {~ $#dir 0} {
				result $es_conf_tempdir
			} {
				result $dir
			}
		}
	) {
		if {! access -rw -d $tmpfiledir} {
			throw error 'tempfile' $tmpfiledir^' is not writable'
		}
		if {~ $tmpfiledir */} { tmpfiledir = <={~~ $tmpfiledir */} }
		result $tmpfiledir/$filename
	}
}

fn nrfn name argsbody {
	let (
		args = ''
		body = ''
	) {
		if {~ $#argsbody 0} {
			throw error noreturn 'missing body'
		} {~ $#argsbody 1} {
			eval 'fn-'^$name^' = $&noreturn @ '^$argsbody
		} {
			args = $argsbody(1 ... <={sub $#argsbody 1})
			body = $argsbody($#argsbody)
			eval 'fn-'^$name^' = $&noreturn @ '^$^args^' '^$body
		}
	}
}

fn macro name argsbody {
	if {~ $#argsbody 0} {
		throw error macro 'missing body'
	}
	let (
		macname = <={gensym '__es_macro_'}
		macfn = ''
		macevalfn = ''
		args = <={ if {~ $#argsbody 1} { result ''} { result $argsbody(1 ... <={sub $#argsbody 1}) } }
		body = <={ if {~ $#argsbody 1} { result $argsbody } { result $argsbody($#argsbody) } }
		macresult = '@ { echo -n $*; echo '';'' }'
		maclet = ''
		argsvar = ''
		macvarname = 'macro-'^$name
	) {
		$macvarname = 'fn-'^$name 'fn-'^$macname $macname
		argsvar = <={ if {~ $args ''} {result '$*'} {result '$'^$args} }
		maclet = 'let(fn-macresult = '$^macresult^' ; args = ('$^args'))'
		macfn = 'fn-'^$macname^' = @ '^$^args^' { '^$maclet^' '^$^body^' }'
		macevalfn = 'fn-'^$name^' = @ '^$^args^' { eval `{ '^$macname^' '^$^argsvar^' } }'
		eval $macfn
		eval $macevalfn
	}
}

macro macroexpand name rest {
	let (
		macvarname = 'macro-'^$name
		macdef = ''
	) {
		macdef = $$macvarname
		macresult $macdef(3)^' '$^rest
	}
}

options = $options macros