tnew - plan9port - [fork] Plan 9 from user space
HTML git clone git://src.adamsgaard.dk/plan9port
DIR Log
DIR Files
DIR Refs
DIR README
DIR LICENSE
---
DIR commit 61b8c11b6659ec7eabbe9249fa9d0088f4c1b883
DIR parent acd47706189b842db4894473168a4dded897e245
HTML Author: rsc <devnull@localhost>
Date: Wed, 10 Aug 2005 18:11:58 +0000
new
Diffstat:
M src/cmd/getflags/getflags.c | 199 +++++++++++++++++++++++++++++++
1 file changed, 199 insertions(+), 0 deletions(-)
---
DIR diff --git a/src/cmd/getflags/getflags.c b/src/cmd/getflags/getflags.c
t@@ -0,0 +1,199 @@
+/*% cyntax % && cc -go # %
+ * getflags: process flags for command files
+ * Usage: ifs='' eval `{getflags [-s] flagfmt [arg ...]} # rc
+ * Usage: IFS= eval `getflags -b [-s] flagfmt [arg...]` # Bourne shell
+ * -b means give Bourne-shell compatible output
+ */
+#include <u.h>
+#include <libc.h>
+#include "getflags.h"
+
+/* predefine functions */
+void bourneprint(int, char *[]);
+void bournearg(char *);
+void rcprint(int, char *[]);
+void usmsg(char *);
+int count(int, char *);
+void rcarg(char *);
+
+void
+main(int argc, char *argv[])
+{
+ int bourne;
+ argc=getflags(argc, argv, "b");
+ if(argc<2) usage("flagfmt [arg ...]");
+ bourne=flag['b']!=0;
+ flag['b']=0;
+ if((argc=getflags(argc-1, argv+1, argv[1]))<0){
+ usmsg(argv[1]);
+ exits(0);
+ }
+ if(bourne) bourneprint(argc, argv);
+ else rcprint(argc, argv);
+ exits(0);
+}
+void
+bourneprint(int argc, char *argv[])
+{
+ register int c, i, n;
+ for(c=0;c!=NFLAG;c++) if(flag[c]){
+ print("FLAG%c=", c); /* bug -- c could be a bad char */
+ n=count(c, argv[1]);
+ if(n==0)
+ print("1\n");
+ else{
+ print("'");
+ bournearg(flag[c][0]);
+ for(i=1;i!=n;i++){
+ print(" ");
+ bournearg(flag[c][i]);
+ }
+ print("'\n");
+ }
+ }
+ print("set --");
+ for(c=1;c!=argc;c++){
+ print(" ");
+ bournearg(argv[c+1]);
+ }
+ print("\n");
+}
+void
+bournearg(char *s)
+{
+ for(;*s;s++)
+ if(*s=='\'')
+ print("'\\''");
+ else
+ print("%c", *s);
+}
+void
+rcprint(int argc, char *argv[])
+{
+ register c, i, n;
+ for(c=0;c!=NFLAG;c++) if(flag[c]){
+ print("FLAG%c=", c); /* bug -- c could be a bad char */
+ n=count(c, argv[1]);
+ if(n==0)
+ print("''");
+ else if(n==1)
+ rcarg(flag[c][0]);
+ else{
+ print("(");
+ rcarg(flag[c][0]);
+ for(i=1;i!=n;i++){
+ print(" ");
+ rcarg(flag[c][i]);
+ }
+ print(")");
+ }
+ print("\n");
+ }
+ print("*=");
+ if(argc==1) print("()");
+ else if(argc==2) rcarg(argv[2]);
+ else{
+ print("(");
+ rcarg(argv[2]);
+ for(c=2;c!=argc;c++){
+ print(" ");
+ rcarg(argv[c+1]);
+ }
+ print(")");
+ }
+ print("\n");
+}
+void
+usmsg(char *flagarg)
+{
+ register char *s, *t, c;
+ register count, nflag=0;
+ print("echo Usage: $0'");
+ for(s=flagarg;*s;){
+ c=*s;
+ if(*s++==' ') continue;
+ if(*s==':')
+ count = strtol(++s, &s, 10);
+ else count=0;
+ if(count==0){
+ if(nflag==0) print(" [-");
+ nflag++;
+ print("%c", c);
+ }
+ if(*s=='['){
+ int depth=1;
+ s++;
+ for(;*s!='\0' && depth>0; s++)
+ if (*s==']') depth--;
+ else if (*s=='[') depth++;
+ }
+ }
+ if(nflag) print("]");
+ for(s=flagarg;*s;){
+ c=*s;
+ if(*s++==' ') continue;
+ if(*s==':')
+ count = strtol(++s, &s, 10);
+ else count=0;
+ if(count!=0){
+ print(" [-");
+ print("%c", c);
+ if(*s=='['){
+ int depth=1;
+ s++;
+ t=s;
+ for(;*s!='\0' && depth>0; s++)
+ if (*s==']') depth--;
+ else if (*s=='[') depth++;
+ print(" ");
+ write(1, t, s - t);
+ }
+ else
+ while(count--) print(" arg");
+ print("]");
+ }
+ else if(*s=='['){
+ int depth=1;
+ s++;
+ for(;*s!='\0' && depth>0; s++)
+ if (*s==']') depth--;
+ else if (*s=='[') depth++;
+ }
+ }
+ print("' $usage;\n");
+ print("exit 'usage'\n");
+}
+int
+count(int flag, char *flagarg)
+{
+ register char *s, c;
+ register n;
+ for(s=flagarg;*s;){
+ c=*s;
+ if(*s++==' ') continue;
+ if(*s==':')
+ n = strtol(++s, &s, 10);
+ else n=0;
+ if(*s=='['){
+ int depth=1;
+ s++;
+ for(;*s!='\0' && depth>0; s++)
+ if (*s==']') depth--;
+ else if (*s=='[') depth++;
+ }
+ if(c==flag) return n;
+ }
+ return -1; /* never happens */
+}
+void
+rcarg(char *s)
+{
+ if(*s=='\0' || strpbrk(s, "\n \t#;&|^$=`'{}()<>?")){
+ print("\'");
+ for(;*s;s++)
+ if(*s=='\'') print("''");
+ else print("%c", *s);
+ print("\'");
+ }
+ else print("%s", s);
+}