summaryrefslogtreecommitdiff
path: root/lib/fileutil/walk.myr
blob: 75ed5e22ba6cda676b2e93cb008adb48cd110040 (plain)
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
use std

pkg fileutil =
	type walkiter = struct
		dirstk	: std.dir#[:]
		curdir	: byte[:][:]
		iterdir	: bool
	;;

	impl iterable walkiter -> byte[:]
	const bywalk	: (dir : std.dir# -> walkiter)
;;

const bywalk = {d
	-> [.dirstk = std.sldup([d][:]), .curdir = std.sldup([""][:])]
}

impl iterable walkiter -> byte[:] =
	__iternext__ = {itp, valp
		var cur, p

:nextfile
		cur = itp.dirstk[itp.dirstk.len - 1]
		match std.dirread(cur)
		| `std.Some ".":	goto nextfile
		| `std.Some "..":	goto nextfile
		| `std.Some ent:
			p = std.pathcat(itp.curdir[itp.curdir.len - 1], ent)
			if std.fisdir(p)
				match std.diropen(p)
				| `std.Ok d:	std.slpush(&itp.dirstk, d)
				| `std.Err e:	/* ? */
				;;
				std.slpush(&itp.curdir, p)
				goto nextfile
			else
				valp# = p
			;;
			-> true
		| `std.None:
			/* don't close the directory given to us by the user */
			if itp.dirstk.len > 1
				std.dirclose(itp.dirstk[itp.dirstk.len - 1])
				std.slfree(itp.curdir[itp.curdir.len - 1])
				std.slpop(&itp.curdir)
				std.slpop(&itp.dirstk)
				goto nextfile
			else
				-> false
			;;
		;;
	}

	__iterfin__ = {itp, valp
		std.slfree(valp#)
	}
;;

generic last = {sl : @a[:]
	-> sl[sl.len - 1]
}