1 ae03a48b 2009-08-15 rsc #include <u.h>
 2 ae03a48b 2009-08-15 rsc #include <libc.h>
 3 ae03a48b 2009-08-15 rsc 
 4 ae03a48b 2009-08-15 rsc void
 5 ae03a48b 2009-08-15 rsc usage(void)
 6 ae03a48b 2009-08-15 rsc {
 7 ae03a48b 2009-08-15 rsc 	print("status=usage\n");
 8 ae03a48b 2009-08-15 rsc 	exits(0);
 9 ae03a48b 2009-08-15 rsc }
10 ae03a48b 2009-08-15 rsc 
11 ae03a48b 2009-08-15 rsc char*
12 ae03a48b 2009-08-15 rsc findarg(char *flags, Rune r)
13 ae03a48b 2009-08-15 rsc {
14 ae03a48b 2009-08-15 rsc 	char *p;
15 ae03a48b 2009-08-15 rsc 	Rune rr;
16 fa325e9b 2020-01-10 cross 
17 ae03a48b 2009-08-15 rsc 	for(p=flags; p!=(char*)1; p=strchr(p, ',')+1){
18 ae03a48b 2009-08-15 rsc 		chartorune(&rr, p);
19 ae03a48b 2009-08-15 rsc 		if(rr == r)
20 ae03a48b 2009-08-15 rsc 			return p;
21 ae03a48b 2009-08-15 rsc 	}
22 fa325e9b 2020-01-10 cross 	return nil;
23 ae03a48b 2009-08-15 rsc }
24 ae03a48b 2009-08-15 rsc 
25 ae03a48b 2009-08-15 rsc int
26 ae03a48b 2009-08-15 rsc countargs(char *p)
27 ae03a48b 2009-08-15 rsc {
28 ae03a48b 2009-08-15 rsc 	int n;
29 ae03a48b 2009-08-15 rsc 
30 ae03a48b 2009-08-15 rsc 	n = 1;
31 ae03a48b 2009-08-15 rsc 	while(*p == ' ')
32 ae03a48b 2009-08-15 rsc 		p++;
33 ae03a48b 2009-08-15 rsc 	for(; *p && *p != ','; p++)
34 ae03a48b 2009-08-15 rsc 		if(*p == ' ' && *(p-1) != ' ')
35 ae03a48b 2009-08-15 rsc 			n++;
36 ae03a48b 2009-08-15 rsc 	return n;
37 ae03a48b 2009-08-15 rsc }
38 ae03a48b 2009-08-15 rsc 
39 ae03a48b 2009-08-15 rsc void
40 ae03a48b 2009-08-15 rsc main(int argc, char *argv[])
41 ae03a48b 2009-08-15 rsc {
42 ae03a48b 2009-08-15 rsc 	char *flags, *p, buf[512];
43 ae03a48b 2009-08-15 rsc 	int i, n;
44 ae03a48b 2009-08-15 rsc 	Fmt fmt;
45 fa325e9b 2020-01-10 cross 
46 fa325e9b 2020-01-10 cross 	doquote = needsrcquote;
47 ae03a48b 2009-08-15 rsc 	quotefmtinstall();
48 ae03a48b 2009-08-15 rsc 	argv0 = argv[0];	/* for sysfatal */
49 fa325e9b 2020-01-10 cross 
50 ae03a48b 2009-08-15 rsc 	flags = getenv("flagfmt");
51 ae03a48b 2009-08-15 rsc 	if(flags == nil){
52 ae03a48b 2009-08-15 rsc 		fprint(2, "$flagfmt not set\n");
53 ae03a48b 2009-08-15 rsc 		print("exit 'missing flagfmt'");
54 ae03a48b 2009-08-15 rsc 		exits(0);
55 ae03a48b 2009-08-15 rsc 	}
56 ae03a48b 2009-08-15 rsc 
57 ae03a48b 2009-08-15 rsc 	fmtfdinit(&fmt, 1, buf, sizeof buf);
58 ae03a48b 2009-08-15 rsc 	for(p=flags; p!=(char*)1; p=strchr(p, ',')+1)
59 ae03a48b 2009-08-15 rsc 		fmtprint(&fmt, "flag%.1s=()\n", p);
60 ae03a48b 2009-08-15 rsc 	ARGBEGIN{
61 ae03a48b 2009-08-15 rsc 	default:
62 ae03a48b 2009-08-15 rsc 		if((p = findarg(flags, ARGC())) == nil)
63 ae03a48b 2009-08-15 rsc 			usage();
64 ae03a48b 2009-08-15 rsc 		p += runelen(ARGC());
65 ae03a48b 2009-08-15 rsc 		if(*p == ',' || *p == 0){
66 ae03a48b 2009-08-15 rsc 			fmtprint(&fmt, "flag%C=1\n", ARGC());
67 ae03a48b 2009-08-15 rsc 			break;
68 ae03a48b 2009-08-15 rsc 		}
69 ae03a48b 2009-08-15 rsc 		n = countargs(p);
70 ae03a48b 2009-08-15 rsc 		fmtprint(&fmt, "flag%C=(", ARGC());
71 ae03a48b 2009-08-15 rsc 		for(i=0; i<n; i++)
72 ae03a48b 2009-08-15 rsc 			fmtprint(&fmt, "%s%q", i ? " " : "", EARGF(usage()));
73 ae03a48b 2009-08-15 rsc 		fmtprint(&fmt, ")\n");
74 ae03a48b 2009-08-15 rsc 	}ARGEND
75 fa325e9b 2020-01-10 cross 
76 ae03a48b 2009-08-15 rsc 	fmtprint(&fmt, "*=(");
77 ae03a48b 2009-08-15 rsc 	for(i=0; i<argc; i++)
78 ae03a48b 2009-08-15 rsc 		fmtprint(&fmt, "%s%q", i ? " " : "", argv[i]);
79 ae03a48b 2009-08-15 rsc 	fmtprint(&fmt, ")\n");
80 ae03a48b 2009-08-15 rsc 	fmtprint(&fmt, "status=''\n");
81 ae03a48b 2009-08-15 rsc 	fmtfdflush(&fmt);
82 ae03a48b 2009-08-15 rsc 	exits(0);
83 ae03a48b 2009-08-15 rsc }