1 #include <u.h>
2 #include <libc.h>
3 #include <draw.h>
4 #include <thread.h>
5 #include <cursor.h>
6 #include <mouse.h>
7 #include <keyboard.h>
8 #include <frame.h>
9 #include <fcall.h>
10 #include <plumb.h>
11 #include "dat.h"
12 #include "edit.h"
13 #include "fns.h"
14 
15 int	Glooping;
16 int	nest;
17 char	Enoname[] = "no file name given";
18 
19 Address	addr;
20 File	*menu;
21 Rangeset	sel;
22 extern	Text*	curtext;
23 Rune	*collection;
24 int	ncollection;
25 
26 int	append(File*, Cmd*, long);
27 int	pdisplay(File*);
28 void	pfilename(File*);
29 void	looper(File*, Cmd*, int);
30 void	filelooper(Cmd*, int);
31 void	linelooper(File*, Cmd*);
32 Address	lineaddr(long, Address, int);
33 int	filematch(File*, String*);
34 File	*tofile(String*);
35 Rune*	cmdname(File *f, String *s, int);
36 void	runpipe(Text*, int, Rune*, int, int);
37 
38 void
39 clearcollection(void)
40 {
41 	free(collection);
42 	collection = nil;
43 	ncollection = 0;
44 }
45 
46 void
47 resetxec(void)
48 {
49 	Glooping = nest = 0;
50 	clearcollection();
51 }
52 
53 void
54 mkaddr(Address *a, File *f)
55 {
56 	a->r.q0 = f->curtext->q0;
57 	a->r.q1 = f->curtext->q1;
58 	a->f = f;
59 }
60 
61 int
62 cmdexec(Text *t, Cmd *cp)
63 {
64 	int i;
65 	Addr *ap;
66 	File *f;
67 	Window *w;
68 	Address dot;
69 
70 	if(t == nil)
71 		w = nil;
72 	else
73 		w = t->w;
74 	if(w==nil && (cp->addr==0 || cp->addr->type!='"') &&
75 	    !utfrune("bBnqUXY!", cp->cmdc) &&
76 	    !(cp->cmdc=='D' && cp->u.text))
77 		editerror("no current window");
78 	i = cmdlookup(cp->cmdc);	/* will be -1 for '{' */
79 	f = nil;
80 	if(t && t->w){
81 		t = &t->w->body;
82 		f = t->file;
83 		f->curtext = t;
84 	}
85 	if(i>=0 && cmdtab[i].defaddr != aNo){
86 		if((ap=cp->addr)==0 && cp->cmdc!='\n'){
87 			cp->addr = ap = newaddr();
88 			ap->type = '.';
89 			if(cmdtab[i].defaddr == aAll)
90 				ap->type = '*';
91 		}else if(ap && ap->type=='"' && ap->next==0 && cp->cmdc!='\n'){
92 			ap->next = newaddr();
93 			ap->next->type = '.';
94 			if(cmdtab[i].defaddr == aAll)
95 				ap->next->type = '*';
96 		}
97 		if(cp->addr){	/* may be false for '\n' (only) */
98 			static Address none = {0,0,nil};
99 			if(f){
100 				mkaddr(&dot, f);
101 				addr = cmdaddress(ap, dot, 0);
102 			}else	/* a " */
103 				addr = cmdaddress(ap, none, 0);
104 			f = addr.f;
105 			t = f->curtext;
106 		}
107 	}
108 	switch(cp->cmdc){
109 	case '{':
110 		mkaddr(&dot, f);
111 		if(cp->addr != nil)
112 			dot = cmdaddress(cp->addr, dot, 0);
113 		for(cp = cp->u.cmd; cp; cp = cp->next){
114 			if(dot.r.q1 > t->file->b.nc)
115 				editerror("dot extends past end of buffer during { command");
116 			t->q0 = dot.r.q0;
117 			t->q1 = dot.r.q1;
118 			cmdexec(t, cp);
119 		}
120 		break;
121 	default:
122 		if(i < 0)
123 			editerror("unknown command %c in cmdexec", cp->cmdc);
124 		i = (*cmdtab[i].fn)(t, cp);
125 		return i;
126 	}
127 	return 1;
128 }
129 
130 char*
131 edittext(Window *w, int q, Rune *r, int nr)
132 {
133 	File *f;
134 
135 	f = w->body.file;
136 	switch(editing){
137 	case Inactive:
138 		return "permission denied";
139 	case Inserting:
140 		eloginsert(f, q, r, nr);
141 		return nil;
142 	case Collecting:
143 		collection = runerealloc(collection, ncollection+nr+1);
144 		runemove(collection+ncollection, r, nr);
145 		ncollection += nr;
146 		collection[ncollection] = '\0';
147 		return nil;
148 	default:
149 		return "unknown state in edittext";
150 	}
151 }
152 
153 /* string is known to be NUL-terminated */
154 Rune*
155 filelist(Text *t, Rune *r, int nr)
156 {
157 	if(nr == 0)
158 		return nil;
159 	r = skipbl(r, nr, &nr);
160 	if(r[0] != '<')
161 		return runestrdup(r);
162 	/* use < command to collect text */
163 	clearcollection();
164 	runpipe(t, '<', r+1, nr-1, Collecting);
165 	return collection;
166 }
167 
168 int
169 a_cmd(Text *t, Cmd *cp)
170 {
171 	return append(t->file, cp, addr.r.q1);
172 }
173 
174 int
175 b_cmd(Text *t, Cmd *cp)
176 {
177 	File *f;
178 
179 	USED(t);
180 	f = tofile(cp->u.text);
181 	if(nest == 0)
182 		pfilename(f);
183 	curtext = f->curtext;
184 	return TRUE;
185 }
186 
187 int
188 B_cmd(Text *t, Cmd *cp)
189 {
190 	Rune *list, *r, *s;
191 	int nr;
192 
193 	list = filelist(t, cp->u.text->r, cp->u.text->n);
194 	if(list == nil)
195 		editerror(Enoname);
196 	r = list;
197 	nr = runestrlen(r);
198 	r = skipbl(r, nr, &nr);
199 	if(nr == 0)
200 		new(t, t, nil, 0, 0, r, 0);
201 	else while(nr > 0){
202 		s = findbl(r, nr, &nr);
203 		*s = '\0';
204 		new(t, t, nil, 0, 0, r, runestrlen(r));
205 		if(nr > 0)
206 			r = skipbl(s+1, nr-1, &nr);
207 	}
208 	clearcollection();
209 	return TRUE;
210 }
211 
212 int
213 c_cmd(Text *t, Cmd *cp)
214 {
215 	elogreplace(t->file, addr.r.q0, addr.r.q1, cp->u.text->r, cp->u.text->n);
216 	t->q0 = addr.r.q0;
217 	t->q1 = addr.r.q1;
218 	return TRUE;
219 }
220 
221 int
222 d_cmd(Text *t, Cmd *cp)
223 {
224 	USED(cp);
225 	if(addr.r.q1 > addr.r.q0)
226 		elogdelete(t->file, addr.r.q0, addr.r.q1);
227 	t->q0 = addr.r.q0;
228 	t->q1 = addr.r.q0;
229 	return TRUE;
230 }
231 
232 void
233 D1(Text *t)
234 {
235 	if(t->w->body.file->ntext>1 || winclean(t->w, FALSE))
236 		colclose(t->col, t->w, TRUE);
237 }
238 
239 int
240 D_cmd(Text *t, Cmd *cp)
241 {
242 	Rune *list, *r, *s, *n;
243 	int nr, nn;
244 	Window *w;
245 	Runestr dir, rs;
246 	char buf[128];
247 
248 	list = filelist(t, cp->u.text->r, cp->u.text->n);
249 	if(list == nil){
250 		D1(t);
251 		return TRUE;
252 	}
253 	dir = dirname(t, nil, 0);
254 	r = list;
255 	nr = runestrlen(r);
256 	r = skipbl(r, nr, &nr);
257 	do{
258 		s = findbl(r, nr, &nr);
259 		*s = '\0';
260 		/* first time through, could be empty string, meaning delete file empty name */
261 		nn = runestrlen(r);
262 		if(r[0]=='/' || nn==0 || dir.nr==0){
263 			rs.r = runestrdup(r);
264 			rs.nr = nn;
265 		}else{
266 			n = runemalloc(dir.nr+1+nn);
267 			runemove(n, dir.r, dir.nr);
268 			n[dir.nr] = '/';
269 			runemove(n+dir.nr+1, r, nn);
270 			rs = cleanrname(runestr(n, dir.nr+1+nn));
271 		}
272 		w = lookfile(rs.r, rs.nr);
273 		if(w == nil){
274 			snprint(buf, sizeof buf, "no such file %.*S", rs.nr, rs.r);
275 			free(rs.r);
276 			editerror(buf);
277 		}
278 		free(rs.r);
279 		D1(&w->body);
280 		if(nr > 0)
281 			r = skipbl(s+1, nr-1, &nr);
282 	}while(nr > 0);
283 	clearcollection();
284 	free(dir.r);
285 	return TRUE;
286 }
287 
288 static int
289 readloader(void *v, uint q0, Rune *r, int nr)
290 {
291 	if(nr > 0)
292 		eloginsert(v, q0, r, nr);
293 	return 0;
294 }
295 
296 int
297 e_cmd(Text *t, Cmd *cp)
298 {
299 	Rune *name;
300 	File *f;
301 	int i, isdir, q0, q1, fd, nulls, samename, allreplaced;
302 	char *s, tmp[128];
303 	Dir *d;
304 
305 	f = t->file;
306 	q0 = addr.r.q0;
307 	q1 = addr.r.q1;
308 	if(cp->cmdc == 'e'){
309 		if(winclean(t->w, TRUE)==FALSE)
310 			editerror("");	/* winclean generated message already */
311 		q0 = 0;
312 		q1 = f->b.nc;
313 	}
314 	allreplaced = (q0==0 && q1==f->b.nc);
315 	name = cmdname(f, cp->u.text, cp->cmdc=='e');
316 	if(name == nil)
317 		editerror(Enoname);
318 	i = runestrlen(name);
319 	samename = runeeq(name, i, t->file->name, t->file->nname);
320 	s = runetobyte(name, i);
321 	free(name);
322 	fd = open(s, OREAD);
323 	if(fd < 0){
324 		snprint(tmp, sizeof tmp, "can't open %s: %r", s);
325 		free(s);
326 		editerror(tmp);
327 	}
328 	d = dirfstat(fd);
329 	isdir = (d!=nil && (d->qid.type&QTDIR));
330 	free(d);
331 	if(isdir){
332 		close(fd);
333 		snprint(tmp, sizeof tmp, "%s is a directory", s);
334 		free(s);
335 		editerror(tmp);
336 	}
337 	elogdelete(f, q0, q1);
338 	nulls = 0;
339 	loadfile(fd, q1, &nulls, readloader, f);
340 	free(s);
341 	close(fd);
342 	if(nulls)
343 		warning(nil, "%s: NUL bytes elided\n", s);
344 	else if(allreplaced && samename)
345 		f->editclean = TRUE;
346 	return TRUE;
347 }
348 
349 static Rune Lempty[] = { 0 };
350 int
351 f_cmd(Text *t, Cmd *cp)
352 {
353 	Rune *name;
354 	String *str;
355 	String empty;
356 
357 	if(cp->u.text == nil){
358 		empty.n = 0;
359 		empty.r = Lempty;
360 		str = &empty;
361 	}else
362 		str = cp->u.text;
363 	name = cmdname(t->file, str, TRUE);
364 	free(name);
365 	pfilename(t->file);
366 	return TRUE;
367 }
368 
369 int
370 g_cmd(Text *t, Cmd *cp)
371 {
372 	if(t->file != addr.f){
373 		warning(nil, "internal error: g_cmd f!=addr.f\n");
374 		return FALSE;
375 	}
376 	if(rxcompile(cp->re->r) == FALSE)
377 		editerror("bad regexp in g command");
378 	if(rxexecute(t, nil, addr.r.q0, addr.r.q1, &sel) ^ cp->cmdc=='v'){
379 		t->q0 = addr.r.q0;
380 		t->q1 = addr.r.q1;
381 		return cmdexec(t, cp->u.cmd);
382 	}
383 	return TRUE;
384 }
385 
386 int
387 i_cmd(Text *t, Cmd *cp)
388 {
389 	return append(t->file, cp, addr.r.q0);
390 }
391 
392 void
393 copy(File *f, Address addr2)
394 {
395 	long p;
396 	int ni;
397 	Rune *buf;
398 
399 	buf = fbufalloc();
400 	for(p=addr.r.q0; p<addr.r.q1; p+=ni){
401 		ni = addr.r.q1-p;
402 		if(ni > RBUFSIZE)
403 			ni = RBUFSIZE;
404 		bufread(&f->b, p, buf, ni);
405 		eloginsert(addr2.f, addr2.r.q1, buf, ni);
406 	}
407 	fbuffree(buf);
408 }
409 
410 void
411 move(File *f, Address addr2)
412 {
413 	if(addr.f!=addr2.f || addr.r.q1<=addr2.r.q0){
414 		elogdelete(f, addr.r.q0, addr.r.q1);
415 		copy(f, addr2);
416 	}else if(addr.r.q0 >= addr2.r.q1){
417 		copy(f, addr2);
418 		elogdelete(f, addr.r.q0, addr.r.q1);
419 	}else
420 		error("move overlaps itself");
421 }
422 
423 int
424 m_cmd(Text *t, Cmd *cp)
425 {
426 	Address dot, addr2;
427 
428 	mkaddr(&dot, t->file);
429 	addr2 = cmdaddress(cp->u.mtaddr, dot, 0);
430 	if(cp->cmdc == 'm')
431 		move(t->file, addr2);
432 	else
433 		copy(t->file, addr2);
434 	return TRUE;
435 }
436 
437 int
438 p_cmd(Text *t, Cmd *cp)
439 {
440 	USED(cp);
441 	return pdisplay(t->file);
442 }
443 
444 int
445 s_cmd(Text *t, Cmd *cp)
446 {
447 	int i, j, k, c, m, n, nrp, didsub;
448 	long p1, op, delta;
449 	String *buf;
450 	Rangeset *rp;
451 	char *err;
452 	Rune *rbuf;
453 
454 	n = cp->num;
455 	op= -1;
456 	if(rxcompile(cp->re->r) == FALSE)
457 		editerror("bad regexp in s command");
458 	nrp = 0;
459 	rp = nil;
460 	delta = 0;
461 	didsub = FALSE;
462 	for(p1 = addr.r.q0; p1<=addr.r.q1 && rxexecute(t, nil, p1, addr.r.q1, &sel); ){
463 		if(sel.r[0].q0 == sel.r[0].q1){	/* empty match? */
464 			if(sel.r[0].q0 == op){
465 				p1++;
466 				continue;
467 			}
468 			p1 = sel.r[0].q1+1;
469 		}else
470 			p1 = sel.r[0].q1;
471 		op = sel.r[0].q1;
472 		if(--n>0)
473 			continue;
474 		nrp++;
475 		rp = erealloc(rp, nrp*sizeof(Rangeset));
476 		rp[nrp-1] = sel;
477 	}
478 	rbuf = fbufalloc();
479 	buf = allocstring(0);
480 	for(m=0; m<nrp; m++){
481 		buf->n = 0;
482 		buf->r[0] = '\0';
483 		sel = rp[m];
484 		for(i = 0; i<cp->u.text->n; i++)
485 			if((c = cp->u.text->r[i])=='\\' && i<cp->u.text->n-1){
486 				c = cp->u.text->r[++i];
487 				if('1'<=c && c<='9') {
488 					j = c-'0';
489 					if(sel.r[j].q1-sel.r[j].q0>RBUFSIZE){
490 						err = "replacement string too long";
491 						goto Err;
492 					}
493 					bufread(&t->file->b, sel.r[j].q0, rbuf, sel.r[j].q1-sel.r[j].q0);
494 					for(k=0; k<sel.r[j].q1-sel.r[j].q0; k++)
495 						Straddc(buf, rbuf[k]);
496 				}else
497 				 	Straddc(buf, c);
498 			}else if(c!='&')
499 				Straddc(buf, c);
500 			else{
501 				if(sel.r[0].q1-sel.r[0].q0>RBUFSIZE){
502 					err = "right hand side too long in substitution";
503 					goto Err;
504 				}
505 				bufread(&t->file->b, sel.r[0].q0, rbuf, sel.r[0].q1-sel.r[0].q0);
506 				for(k=0; k<sel.r[0].q1-sel.r[0].q0; k++)
507 					Straddc(buf, rbuf[k]);
508 			}
509 		elogreplace(t->file, sel.r[0].q0, sel.r[0].q1,  buf->r, buf->n);
510 		delta -= sel.r[0].q1-sel.r[0].q0;
511 		delta += buf->n;
512 		didsub = 1;
513 		if(!cp->flag)
514 			break;
515 	}
516 	free(rp);
517 	freestring(buf);
518 	fbuffree(rbuf);
519 	if(!didsub && nest==0)
520 		editerror("no substitution");
521 	t->q0 = addr.r.q0;
522 	t->q1 = addr.r.q1;
523 	return TRUE;
524 
525 Err:
526 	free(rp);
527 	freestring(buf);
528 	fbuffree(rbuf);
529 	editerror(err);
530 	return FALSE;
531 }
532 
533 int
534 u_cmd(Text *t, Cmd *cp)
535 {
536 	int n, oseq, flag;
537 
538 	n = cp->num;
539 	flag = TRUE;
540 	if(n < 0){
541 		n = -n;
542 		flag = FALSE;
543 	}
544 	oseq = -1;
545 	while(n-->0 && t->file->seq!=0 && t->file->seq!=oseq){
546 		oseq = t->file->seq;
547 		undo(t, nil, nil, flag, 0, nil, 0);
548 	}
549 	return TRUE;
550 }
551 
552 int
553 w_cmd(Text *t, Cmd *cp)
554 {
555 	Rune *r;
556 	File *f;
557 
558 	f = t->file;
559 	if(f->seq == seq)
560 		editerror("can't write file with pending modifications");
561 	r = cmdname(f, cp->u.text, FALSE);
562 	if(r == nil)
563 		editerror("no name specified for 'w' command");
564 	putfile(f, addr.r.q0, addr.r.q1, r, runestrlen(r));
565 	/* r is freed by putfile */
566 	return TRUE;
567 }
568 
569 int
570 x_cmd(Text *t, Cmd *cp)
571 {
572 	if(cp->re)
573 		looper(t->file, cp, cp->cmdc=='x');
574 	else
575 		linelooper(t->file, cp);
576 	return TRUE;
577 }
578 
579 int
580 X_cmd(Text *t, Cmd *cp)
581 {
582 	USED(t);
583 
584 	filelooper(cp, cp->cmdc=='X');
585 	return TRUE;
586 }
587 
588 void
589 runpipe(Text *t, int cmd, Rune *cr, int ncr, int state)
590 {
591 	Rune *r, *s;
592 	int n;
593 	Runestr dir;
594 	Window *w;
595 
596 	r = skipbl(cr, ncr, &n);
597 	if(n == 0)
598 		editerror("no command specified for %c", cmd);
599 	w = nil;
600 	if(state == Inserting){
601 		w = t->w;
602 		t->q0 = addr.r.q0;
603 		t->q1 = addr.r.q1;
604 		if(cmd == '<' || cmd=='|')
605 			elogdelete(t->file, t->q0, t->q1);
606 	}
607 	s = runemalloc(n+2);
608 	s[0] = cmd;
609 	runemove(s+1, r, n);
610 	n++;
611 	dir.r = nil;
612 	dir.nr = 0;
613 	if(t != nil)
614 		dir = dirname(t, nil, 0);
615 	if(dir.nr==1 && dir.r[0]=='.'){	/* sigh */
616 		free(dir.r);
617 		dir.r = nil;
618 		dir.nr = 0;
619 	}
620 	editing = state;
621 	if(t!=nil && t->w!=nil)
622 		incref(&t->w->ref);	/* run will decref */
623 	run(w, runetobyte(s, n), dir.r, dir.nr, TRUE, nil, nil, TRUE);
624 	free(s);
625 	if(t!=nil && t->w!=nil)
626 		winunlock(t->w);
627 	qunlock(&row.lk);
628 	recvul(cedit);
629 	qlock(&row.lk);
630 	editing = Inactive;
631 	if(t!=nil && t->w!=nil)
632 		winlock(t->w, 'M');
633 }
634 
635 int
636 pipe_cmd(Text *t, Cmd *cp)
637 {
638 	runpipe(t, cp->cmdc, cp->u.text->r, cp->u.text->n, Inserting);
639 	return TRUE;
640 }
641 
642 long
643 nlcount(Text *t, long q0, long q1)
644 {
645 	long nl;
646 	Rune *buf;
647 	int i, nbuf;
648 
649 	buf = fbufalloc();
650 	nbuf = 0;
651 	i = nl = 0;
652 	while(q0 < q1){
653 		if(i == nbuf){
654 			nbuf = q1-q0;
655 			if(nbuf > RBUFSIZE)
656 				nbuf = RBUFSIZE;
657 			bufread(&t->file->b, q0, buf, nbuf);
658 			i = 0;
659 		}
660 		if(buf[i++] == '\n')
661 			nl++;
662 		q0++;
663 	}
664 	fbuffree(buf);
665 	return nl;
666 }
667 
668 void
669 printposn(Text *t, int charsonly)
670 {
671 	long l1, l2;
672 
673 	if (t != nil && t->file != nil && t->file->name != nil)
674 		warning(nil, "%.*S:", t->file->nname, t->file->name);
675 	if(!charsonly){
676 		l1 = 1+nlcount(t, 0, addr.r.q0);
677 		l2 = l1+nlcount(t, addr.r.q0, addr.r.q1);
678 		/* check if addr ends with '\n' */
679 		if(addr.r.q1>0 && addr.r.q1>addr.r.q0 && textreadc(t, addr.r.q1-1)=='\n')
680 			--l2;
681 		warning(nil, "%lud", l1);
682 		if(l2 != l1)
683 			warning(nil, ",%lud", l2);
684 		warning(nil, "\n");
685 		return;
686 	}
687 	warning(nil, "#%d", addr.r.q0);
688 	if(addr.r.q1 != addr.r.q0)
689 		warning(nil, ",#%d", addr.r.q1);
690 	warning(nil, "\n");
691 }
692 
693 int
694 eq_cmd(Text *t, Cmd *cp)
695 {
696 	int charsonly;
697 
698 	switch(cp->u.text->n){
699 	case 0:
700 		charsonly = FALSE;
701 		break;
702 	case 1:
703 		if(cp->u.text->r[0] == '#'){
704 			charsonly = TRUE;
705 			break;
706 		}
707 	default:
708 		SET(charsonly);
709 		editerror("newline expected");
710 	}
711 	printposn(t, charsonly);
712 	return TRUE;
713 }
714 
715 int
716 nl_cmd(Text *t, Cmd *cp)
717 {
718 	Address a;
719 	File *f;
720 
721 	f = t->file;
722 	if(cp->addr == 0){
723 		/* First put it on newline boundaries */
724 		mkaddr(&a, f);
725 		addr = lineaddr(0, a, -1);
726 		a = lineaddr(0, a, 1);
727 		addr.r.q1 = a.r.q1;
728 		if(addr.r.q0==t->q0 && addr.r.q1==t->q1){
729 			mkaddr(&a, f);
730 			addr = lineaddr(1, a, 1);
731 		}
732 	}
733 	textshow(t, addr.r.q0, addr.r.q1, 1);
734 	return TRUE;
735 }
736 
737 int
738 append(File *f, Cmd *cp, long p)
739 {
740 	if(cp->u.text->n > 0)
741 		eloginsert(f, p, cp->u.text->r, cp->u.text->n);
742 	f->curtext->q0 = p;
743 	f->curtext->q1 = p;
744 	return TRUE;
745 }
746 
747 int
748 pdisplay(File *f)
749 {
750 	long p1, p2;
751 	int np;
752 	Rune *buf;
753 
754 	p1 = addr.r.q0;
755 	p2 = addr.r.q1;
756 	if(p2 > f->b.nc)
757 		p2 = f->b.nc;
758 	buf = fbufalloc();
759 	while(p1 < p2){
760 		np = p2-p1;
761 		if(np>RBUFSIZE-1)
762 			np = RBUFSIZE-1;
763 		bufread(&f->b, p1, buf, np);
764 		buf[np] = '\0';
765 		warning(nil, "%S", buf);
766 		p1 += np;
767 	}
768 	fbuffree(buf);
769 	f->curtext->q0 = addr.r.q0;
770 	f->curtext->q1 = addr.r.q1;
771 	return TRUE;
772 }
773 
774 void
775 pfilename(File *f)
776 {
777 	int dirty;
778 	Window *w;
779 
780 	w = f->curtext->w;
781 	/* same check for dirty as in settag, but we know ncache==0 */
782 	dirty = !w->isdir && !w->isscratch && f->mod;
783 	warning(nil, "%c%c%c %.*S\n", " '"[dirty],
784 		'+', " ."[curtext!=nil && curtext->file==f], f->nname, f->name);
785 }
786 
787 void
788 loopcmd(File *f, Cmd *cp, Range *rp, long nrp)
789 {
790 	long i;
791 
792 	for(i=0; i<nrp; i++){
793 		f->curtext->q0 = rp[i].q0;
794 		f->curtext->q1 = rp[i].q1;
795 		cmdexec(f->curtext, cp);
796 	}
797 }
798 
799 void
800 looper(File *f, Cmd *cp, int xy)
801 {
802 	long p, op, nrp;
803 	Range r, tr;
804 	Range *rp;
805 
806 	r = addr.r;
807 	op= xy? -1 : r.q0;
808 	nest++;
809 	if(rxcompile(cp->re->r) == FALSE)
810 		editerror("bad regexp in %c command", cp->cmdc);
811 	nrp = 0;
812 	rp = nil;
813 	for(p = r.q0; p<=r.q1; ){
814 		if(!rxexecute(f->curtext, nil, p, r.q1, &sel)){ /* no match, but y should still run */
815 			if(xy || op>r.q1)
816 				break;
817 			tr.q0 = op, tr.q1 = r.q1;
818 			p = r.q1+1;	/* exit next loop */
819 		}else{
820 			if(sel.r[0].q0==sel.r[0].q1){	/* empty match? */
821 				if(sel.r[0].q0==op){
822 					p++;
823 					continue;
824 				}
825 				p = sel.r[0].q1+1;
826 			}else
827 				p = sel.r[0].q1;
828 			if(xy)
829 				tr = sel.r[0];
830 			else
831 				tr.q0 = op, tr.q1 = sel.r[0].q0;
832 		}
833 		op = sel.r[0].q1;
834 		nrp++;
835 		rp = erealloc(rp, nrp*sizeof(Range));
836 		rp[nrp-1] = tr;
837 	}
838 	loopcmd(f, cp->u.cmd, rp, nrp);
839 	free(rp);
840 	--nest;
841 }
842 
843 void
844 linelooper(File *f, Cmd *cp)
845 {
846 	long nrp, p;
847 	Range r, linesel;
848 	Address a, a3;
849 	Range *rp;
850 
851 	nest++;
852 	nrp = 0;
853 	rp = nil;
854 	r = addr.r;
855 	a3.f = f;
856 	a3.r.q0 = a3.r.q1 = r.q0;
857 	a = lineaddr(0, a3, 1);
858 	linesel = a.r;
859 	for(p = r.q0; p<r.q1; p = a3.r.q1){
860 		a3.r.q0 = a3.r.q1;
861 		if(p!=r.q0 || linesel.q1==p){
862 			a = lineaddr(1, a3, 1);
863 			linesel = a.r;
864 		}
865 		if(linesel.q0 >= r.q1)
866 			break;
867 		if(linesel.q1 >= r.q1)
868 			linesel.q1 = r.q1;
869 		if(linesel.q1 > linesel.q0)
870 			if(linesel.q0>=a3.r.q1 && linesel.q1>a3.r.q1){
871 				a3.r = linesel;
872 				nrp++;
873 				rp = erealloc(rp, nrp*sizeof(Range));
874 				rp[nrp-1] = linesel;
875 				continue;
876 			}
877 		break;
878 	}
879 	loopcmd(f, cp->u.cmd, rp, nrp);
880 	free(rp);
881 	--nest;
882 }
883 
884 struct Looper
885 {
886 	Cmd *cp;
887 	int	XY;
888 	Window	**w;
889 	int	nw;
890 } loopstruct;	/* only one; X and Y can't nest */
891 
892 void
893 alllooper(Window *w, void *v)
894 {
895 	Text *t;
896 	struct Looper *lp;
897 	Cmd *cp;
898 
899 	lp = v;
900 	cp = lp->cp;
901 //	if(w->isscratch || w->isdir)
902 //		return;
903 	t = &w->body;
904 	/* only use this window if it's the current window for the file */
905 	if(t->file->curtext != t)
906 		return;
907 //	if(w->nopen[QWevent] > 0)
908 //		return;
909 	/* no auto-execute on files without names */
910 	if(cp->re==nil && t->file->nname==0)
911 		return;
912 	if(cp->re==nil || filematch(t->file, cp->re)==lp->XY){
913 		lp->w = erealloc(lp->w, (lp->nw+1)*sizeof(Window*));
914 		lp->w[lp->nw++] = w;
915 	}
916 }
917 
918 void
919 alllocker(Window *w, void *v)
920 {
921 	if(v)
922 		incref(&w->ref);
923 	else
924 		winclose(w);
925 }
926 
927 void
928 filelooper(Cmd *cp, int XY)
929 {
930 	int i;
931 
932 	if(Glooping++)
933 		editerror("can't nest %c command", "YX"[XY]);
934 	nest++;
935 
936 	loopstruct.cp = cp;
937 	loopstruct.XY = XY;
938 	if(loopstruct.w)	/* error'ed out last time */
939 		free(loopstruct.w);
940 	loopstruct.w = nil;
941 	loopstruct.nw = 0;
942 	allwindows(alllooper, &loopstruct);
943 	/*
944 	 * add a ref to all windows to keep safe windows accessed by X
945 	 * that would not otherwise have a ref to hold them up during
946 	 * the shenanigans.  note this with globalincref so that any
947 	 * newly created windows start with an extra reference.
948 	 */
949 	allwindows(alllocker, (void*)1);
950 	globalincref = 1;
951 	for(i=0; i<loopstruct.nw; i++)
952 		cmdexec(&loopstruct.w[i]->body, cp->u.cmd);
953 	allwindows(alllocker, (void*)0);
954 	globalincref = 0;
955 	free(loopstruct.w);
956 	loopstruct.w = nil;
957 
958 	--Glooping;
959 	--nest;
960 }
961 
962 void
963 nextmatch(File *f, String *r, long p, int sign)
964 {
965 	if(rxcompile(r->r) == FALSE)
966 		editerror("bad regexp in command address");
967 	if(sign >= 0){
968 		if(!rxexecute(f->curtext, nil, p, 0x7FFFFFFFL, &sel))
969 			editerror("no match for regexp");
970 		if(sel.r[0].q0==sel.r[0].q1 && sel.r[0].q0==p){
971 			if(++p>f->b.nc)
972 				p = 0;
973 			if(!rxexecute(f->curtext, nil, p, 0x7FFFFFFFL, &sel))
974 				editerror("address");
975 		}
976 	}else{
977 		if(!rxbexecute(f->curtext, p, &sel))
978 			editerror("no match for regexp");
979 		if(sel.r[0].q0==sel.r[0].q1 && sel.r[0].q1==p){
980 			if(--p<0)
981 				p = f->b.nc;
982 			if(!rxbexecute(f->curtext, p, &sel))
983 				editerror("address");
984 		}
985 	}
986 }
987 
988 File	*matchfile(String*);
989 Address	charaddr(long, Address, int);
990 Address	lineaddr(long, Address, int);
991 
992 Address
993 cmdaddress(Addr *ap, Address a, int sign)
994 {
995 	File *f = a.f;
996 	Address a1, a2;
997 
998 	do{
999 		switch(ap->type){
1000 		case 'l':
1001 		case '#':
1002 			a = (*(ap->type=='#'?charaddr:lineaddr))(ap->num, a, sign);
1003 			break;
1004 
1005 		case '.':
1006 			mkaddr(&a, f);
1007 			break;
1008 
1009 		case '$':
1010 			a.r.q0 = a.r.q1 = f->b.nc;
1011 			break;
1012 
1013 		case '\'':
1014 editerror("can't handle '");
1015 //			a.r = f->mark;
1016 			break;
1017 
1018 		case '?':
1019 			sign = -sign;
1020 			if(sign == 0)
1021 				sign = -1;
1022 			/* fall through */
1023 		case '/':
1024 			nextmatch(f, ap->u.re, sign>=0? a.r.q1 : a.r.q0, sign);
1025 			a.r = sel.r[0];
1026 			break;
1027 
1028 		case '"':
1029 			f = matchfile(ap->u.re);
1030 			mkaddr(&a, f);
1031 			break;
1032 
1033 		case '*':
1034 			a.r.q0 = 0, a.r.q1 = f->b.nc;
1035 			return a;
1036 
1037 		case ',':
1038 		case ';':
1039 			if(ap->u.left)
1040 				a1 = cmdaddress(ap->u.left, a, 0);
1041 			else
1042 				a1.f = a.f, a1.r.q0 = a1.r.q1 = 0;
1043 			if(ap->type == ';'){
1044 				f = a1.f;
1045 				a = a1;
1046 				f->curtext->q0 = a1.r.q0;
1047 				f->curtext->q1 = a1.r.q1;
1048 			}
1049 			if(ap->next)
1050 				a2 = cmdaddress(ap->next, a, 0);
1051 			else
1052 				a2.f = a.f, a2.r.q0 = a2.r.q1 = f->b.nc;
1053 			if(a1.f != a2.f)
1054 				editerror("addresses in different files");
1055 			a.f = a1.f, a.r.q0 = a1.r.q0, a.r.q1 = a2.r.q1;
1056 			if(a.r.q1 < a.r.q0)
1057 				editerror("addresses out of order");
1058 			return a;
1059 
1060 		case '+':
1061 		case '-':
1062 			sign = 1;
1063 			if(ap->type == '-')
1064 				sign = -1;
1065 			if(ap->next==0 || ap->next->type=='+' || ap->next->type=='-')
1066 				a = lineaddr(1L, a, sign);
1067 			break;
1068 		default:
1069 			error("cmdaddress");
1070 			return a;
1071 		}
1072 	}while(ap = ap->next);	/* assign = */
1073 	return a;
1074 }
1075 
1076 struct Tofile{
1077 	File		*f;
1078 	String	*r;
1079 };
1080 
1081 void
1082 alltofile(Window *w, void *v)
1083 {
1084 	Text *t;
1085 	struct Tofile *tp;
1086 
1087 	tp = v;
1088 	if(tp->f != nil)
1089 		return;
1090 	if(w->isscratch || w->isdir)
1091 		return;
1092 	t = &w->body;
1093 	/* only use this window if it's the current window for the file */
1094 	if(t->file->curtext != t)
1095 		return;
1096 //	if(w->nopen[QWevent] > 0)
1097 //		return;
1098 	if(runeeq(tp->r->r, tp->r->n, t->file->name, t->file->nname))
1099 		tp->f = t->file;
1100 }
1101 
1102 File*
1103 tofile(String *r)
1104 {
1105 	struct Tofile t;
1106 	String rr;
1107 
1108 	rr.r = skipbl(r->r, r->n, &rr.n);
1109 	t.f = nil;
1110 	t.r = &rr;
1111 	allwindows(alltofile, &t);
1112 	if(t.f == nil)
1113 		editerror("no such file\"%S\"", rr.r);
1114 	return t.f;
1115 }
1116 
1117 void
1118 allmatchfile(Window *w, void *v)
1119 {
1120 	struct Tofile *tp;
1121 	Text *t;
1122 
1123 	tp = v;
1124 	if(w->isscratch || w->isdir)
1125 		return;
1126 	t = &w->body;
1127 	/* only use this window if it's the current window for the file */
1128 	if(t->file->curtext != t)
1129 		return;
1130 //	if(w->nopen[QWevent] > 0)
1131 //		return;
1132 	if(filematch(w->body.file, tp->r)){
1133 		if(tp->f != nil)
1134 			editerror("too many files match \"%S\"", tp->r->r);
1135 		tp->f = w->body.file;
1136 	}
1137 }
1138 
1139 File*
1140 matchfile(String *r)
1141 {
1142 	struct Tofile tf;
1143 
1144 	tf.f = nil;
1145 	tf.r = r;
1146 	allwindows(allmatchfile, &tf);
1147 
1148 	if(tf.f == nil)
1149 		editerror("no file matches \"%S\"", r->r);
1150 	return tf.f;
1151 }
1152 
1153 int
1154 filematch(File *f, String *r)
1155 {
1156 	char *buf;
1157 	Rune *rbuf;
1158 	Window *w;
1159 	int match, i, dirty;
1160 	Rangeset s;
1161 
1162 	/* compile expr first so if we get an error, we haven't allocated anything */
1163 	if(rxcompile(r->r) == FALSE)
1164 		editerror("bad regexp in file match");
1165 	buf = fbufalloc();
1166 	w = f->curtext->w;
1167 	/* same check for dirty as in settag, but we know ncache==0 */
1168 	dirty = !w->isdir && !w->isscratch && f->mod;
1169 	snprint(buf, BUFSIZE, "%c%c%c %.*S\n", " '"[dirty],
1170 		'+', " ."[curtext!=nil && curtext->file==f], f->nname, f->name);
1171 	rbuf = bytetorune(buf, &i);
1172 	fbuffree(buf);
1173 	match = rxexecute(nil, rbuf, 0, i, &s);
1174 	free(rbuf);
1175 	return match;
1176 }
1177 
1178 Address
1179 charaddr(long l, Address addr, int sign)
1180 {
1181 	if(sign == 0)
1182 		addr.r.q0 = addr.r.q1 = l;
1183 	else if(sign < 0)
1184 		addr.r.q1 = addr.r.q0 -= l;
1185 	else if(sign > 0)
1186 		addr.r.q0 = addr.r.q1 += l;
1187 	if(addr.r.q0<0 || addr.r.q1>addr.f->b.nc)
1188 		editerror("address out of range");
1189 	return addr;
1190 }
1191 
1192 Address
1193 lineaddr(long l, Address addr, int sign)
1194 {
1195 	int n;
1196 	int c;
1197 	File *f = addr.f;
1198 	Address a;
1199 	long p;
1200 
1201 	a.f = f;
1202 	if(sign >= 0){
1203 		if(l == 0){
1204 			if(sign==0 || addr.r.q1==0){
1205 				a.r.q0 = a.r.q1 = 0;
1206 				return a;
1207 			}
1208 			a.r.q0 = addr.r.q1;
1209 			p = addr.r.q1-1;
1210 		}else{
1211 			if(sign==0 || addr.r.q1==0){
1212 				p = 0;
1213 				n = 1;
1214 			}else{
1215 				p = addr.r.q1-1;
1216 				n = textreadc(f->curtext, p++)=='\n';
1217 			}
1218 			while(n < l){
1219 				if(p >= f->b.nc)
1220 					editerror("address out of range");
1221 				if(textreadc(f->curtext, p++) == '\n')
1222 					n++;
1223 			}
1224 			a.r.q0 = p;
1225 		}
1226 		while(p < f->b.nc && textreadc(f->curtext, p++)!='\n')
1227 			;
1228 		a.r.q1 = p;
1229 	}else{
1230 		p = addr.r.q0;
1231 		if(l == 0)
1232 			a.r.q1 = addr.r.q0;
1233 		else{
1234 			for(n = 0; n<l; ){	/* always runs once */
1235 				if(p == 0){
1236 					if(++n != l)
1237 						editerror("address out of range");
1238 				}else{
1239 					c = textreadc(f->curtext, p-1);
1240 					if(c != '\n' || ++n != l)
1241 						p--;
1242 				}
1243 			}
1244 			a.r.q1 = p;
1245 			if(p > 0)
1246 				p--;
1247 		}
1248 		while(p > 0 && textreadc(f->curtext, p-1)!='\n')	/* lines start after a newline */
1249 			p--;
1250 		a.r.q0 = p;
1251 	}
1252 	return a;
1253 }
1254 
1255 struct Filecheck
1256 {
1257 	File	*f;
1258 	Rune	*r;
1259 	int nr;
1260 };
1261 
1262 void
1263 allfilecheck(Window *w, void *v)
1264 {
1265 	struct Filecheck *fp;
1266 	File *f;
1267 
1268 	fp = v;
1269 	f = w->body.file;
1270 	if(w->body.file == fp->f)
1271 		return;
1272 	if(runeeq(fp->r, fp->nr, f->name, f->nname))
1273 		warning(nil, "warning: duplicate file name \"%.*S\"\n", fp->nr, fp->r);
1274 }
1275 
1276 Rune*
1277 cmdname(File *f, String *str, int set)
1278 {
1279 	Rune *r, *s;
1280 	int n;
1281 	struct Filecheck fc;
1282 	Runestr newname;
1283 
1284 	r = nil;
1285 	n = str->n;
1286 	s = str->r;
1287 	if(n == 0){
1288 		/* no name; use existing */
1289 		if(f->nname == 0)
1290 			return nil;
1291 		r = runemalloc(f->nname+1);
1292 		runemove(r, f->name, f->nname);
1293 		return r;
1294 	}
1295 	s = skipbl(s, n, &n);
1296 	if(n == 0)
1297 		goto Return;
1298 
1299 	if(s[0] == '/'){
1300 		r = runemalloc(n+1);
1301 		runemove(r, s, n);
1302 	}else{
1303 		newname = dirname(f->curtext, runestrdup(s), n);
1304 		n = newname.nr;
1305 		r = runemalloc(n+1);
1306 		runemove(r, newname.r, n);
1307 		free(newname.r);
1308 	}
1309 	fc.f = f;
1310 	fc.r = r;
1311 	fc.nr = n;
1312 	allwindows(allfilecheck, &fc);
1313 	if(f->nname == 0)
1314 		set = TRUE;
1315 
1316     Return:
1317 	if(set && !runeeq(r, n, f->name, f->nname)){
1318 		filemark(f);
1319 		f->mod = TRUE;
1320 		f->curtext->w->dirty = TRUE;
1321 		winsetname(f->curtext->w, r, n);
1322 	}
1323 	return r;
1324 }