xref: /plan9-contrib/sys/src/cmd/mk/shprint.c (revision 99eb86a7eded05d02373a1dc66f01c7d211105e0)
13e12c5d1SDavid du Colombier #include	"mk.h"
23e12c5d1SDavid du Colombier 
33e12c5d1SDavid du Colombier static char *vexpand(char*, Envy*, Bufblock*);
4219b2ee8SDavid du Colombier static char *shquote(char*, Rune, Bufblock*);
53e12c5d1SDavid du Colombier static char *shbquote(char*, Bufblock*);
63e12c5d1SDavid du Colombier 
73e12c5d1SDavid du Colombier void
shprint(char * s,Envy * env,Bufblock * buf)83e12c5d1SDavid du Colombier shprint(char *s, Envy *env, Bufblock *buf)
93e12c5d1SDavid du Colombier {
103e12c5d1SDavid du Colombier 	int n;
113e12c5d1SDavid du Colombier 	Rune r;
123e12c5d1SDavid du Colombier 
133e12c5d1SDavid du Colombier 	while(*s) {
143e12c5d1SDavid du Colombier 		n = chartorune(&r, s);
153e12c5d1SDavid du Colombier 		if (r == '$')
163e12c5d1SDavid du Colombier 			s = vexpand(s, env, buf);
173e12c5d1SDavid du Colombier 		else {
183e12c5d1SDavid du Colombier 			rinsert(buf, r);
193e12c5d1SDavid du Colombier 			s += n;
207dd7cddfSDavid du Colombier 			s = copyq(s, r, buf);	/*handle quoted strings*/
213e12c5d1SDavid du Colombier 		}
223e12c5d1SDavid du Colombier 	}
233e12c5d1SDavid du Colombier 	insert(buf, 0);
243e12c5d1SDavid du Colombier }
253e12c5d1SDavid du Colombier 
263e12c5d1SDavid du Colombier static char *
mygetenv(char * name,Envy * env)273e12c5d1SDavid du Colombier mygetenv(char *name, Envy *env)
283e12c5d1SDavid du Colombier {
293e12c5d1SDavid du Colombier 	if (!env)
303e12c5d1SDavid du Colombier 		return 0;
317dd7cddfSDavid du Colombier 	if (symlook(name, S_WESET, 0) == 0 && symlook(name, S_INTERNAL, 0) == 0)
323e12c5d1SDavid du Colombier 		return 0;
337dd7cddfSDavid du Colombier 		/* only resolve internal variables and variables we've set */
343e12c5d1SDavid du Colombier 	for(; env->name; env++){
353e12c5d1SDavid du Colombier 		if (strcmp(env->name, name) == 0)
367dd7cddfSDavid du Colombier 			return wtos(env->values, ' ');
373e12c5d1SDavid du Colombier 	}
383e12c5d1SDavid du Colombier 	return 0;
393e12c5d1SDavid du Colombier }
403e12c5d1SDavid du Colombier 
413e12c5d1SDavid du Colombier static char *
vexpand(char * w,Envy * env,Bufblock * buf)423e12c5d1SDavid du Colombier vexpand(char *w, Envy *env, Bufblock *buf)
433e12c5d1SDavid du Colombier {
443e12c5d1SDavid du Colombier 	char *s, carry, *p, *q;
453e12c5d1SDavid du Colombier 
467dd7cddfSDavid du Colombier 	assert(/*vexpand no $*/ *w == '$');
473e12c5d1SDavid du Colombier 	p = w+1;	/* skip dollar sign */
483e12c5d1SDavid du Colombier 	if(*p == '{') {
493e12c5d1SDavid du Colombier 		p++;
503e12c5d1SDavid du Colombier 		q = utfrune(p, '}');
513e12c5d1SDavid du Colombier 		if (!q)
523e12c5d1SDavid du Colombier 			q = strchr(p, 0);
533e12c5d1SDavid du Colombier 	} else
543e12c5d1SDavid du Colombier 		q = shname(p);
553e12c5d1SDavid du Colombier 	carry = *q;
563e12c5d1SDavid du Colombier 	*q = 0;
573e12c5d1SDavid du Colombier 	s = mygetenv(p, env);
583e12c5d1SDavid du Colombier 	*q = carry;
593e12c5d1SDavid du Colombier 	if (carry == '}')
603e12c5d1SDavid du Colombier 		q++;
613e12c5d1SDavid du Colombier 	if (s) {
623e12c5d1SDavid du Colombier 		bufcpy(buf, s, strlen(s));
633e12c5d1SDavid du Colombier 		free(s);
643e12c5d1SDavid du Colombier 	} else 		/* copy name intact*/
653e12c5d1SDavid du Colombier 		bufcpy(buf, w, q-w);
663e12c5d1SDavid du Colombier 	return(q);
673e12c5d1SDavid du Colombier }
683e12c5d1SDavid du Colombier 
693e12c5d1SDavid du Colombier void
front(char * s)703e12c5d1SDavid du Colombier front(char *s)
713e12c5d1SDavid du Colombier {
72276e7d6dSDavid du Colombier 	char *t, *q;
73*99eb86a7SDavid du Colombier 	int i, j;
743e12c5d1SDavid du Colombier 	char *flds[512];
753e12c5d1SDavid du Colombier 
763e12c5d1SDavid du Colombier 	q = strdup(s);
77276e7d6dSDavid du Colombier 	i = getfields(q, flds, nelem(flds), 0, " \t\n");
783e12c5d1SDavid du Colombier 	if(i > 5){
793e12c5d1SDavid du Colombier 		flds[4] = flds[i-1];
803e12c5d1SDavid du Colombier 		flds[3] = "...";
813e12c5d1SDavid du Colombier 		i = 5;
823e12c5d1SDavid du Colombier 	}
833e12c5d1SDavid du Colombier 	t = s;
843e12c5d1SDavid du Colombier 	for(j = 0; j < i; j++){
85*99eb86a7SDavid du Colombier 		for(s = flds[j]; *s; *t++ = *s++);
863e12c5d1SDavid du Colombier 		*t++ = ' ';
873e12c5d1SDavid du Colombier 	}
883e12c5d1SDavid du Colombier 	*t = 0;
893e12c5d1SDavid du Colombier 	free(q);
903e12c5d1SDavid du Colombier }
91