void
emitdi(void)
{
flushdi();
runefmtstrinit(&difmt);
difmtinit = 1;
fmtrune(&difmt, Uformatted);
}
/* .di, .da */
void
r_di(int argc, Rune **argv)
{
br();
if(argc > 2)
warn("extra arguments to %C%S", dot, argv[0]);
if(argc == 1){
/* end diversion */
if(ndi <= 0){
/* warn("unmatched %C%S", dot, argv[0]); */
return;
}
flushdi();
if(--ndi == 0){
_nr(L(".z"), nil);
outcb = nil;
}else{
_nr(L(".z"), di[ndi-1]);
runefmtstrinit(&difmt);
fmtrune(&difmt, Uformatted);
difmtinit = 1;
}
return;
}
/* start diversion */
/* various register state should be saved, but it's all useless to us */
flushdi();
if(ndi >= nelem(di))
sysfatal("%Cdi overflow", dot);
if(argv[0][1] == 'i')
ds(argv[1], nil);
_nr(L(".z"), argv[1]);
runestrcpy(di[ndi++], argv[1]);
runefmtstrinit(&difmt);
fmtrune(&difmt, Uformatted);
difmtinit = 1;
outcb = outdi;
}
/*
* 18. Insertions from the standard input
*/
void
r_rd(int argc, Rune **argv)
{
char buf[100];
char *s;
Rune *p;
Fmt fmt;
static int didstdin;
static Biobuf bstdin;
/*
* print prompt, then read until double newline,
* then run the text just read as though it were
* a macro body, using the remaining arguments.
*/
if(fd2path(0, buf, sizeof buf) >= 0 && strstr(buf, "/dev/cons")){
if(argc > 1)
fprint(2, "%S", argv[1]);
else
fprint(2, "%c", 7/*BEL*/);
}
if(!didstdin){
Binit(&bstdin, 0, OREAD);
didstdin = 1;
}
runefmtstrinit(&fmt);
while((s = Brdstr(&bstdin, '\n', 0)) != nil){
if(s[0] == '\n'){
free(s);
break;
}
fmtprint(&fmt, "%s", s);
free(s);
}
p = runefmtstrflush(&fmt);
if(p == nil)
warn("out of memory in %Crd", dot);
ds(L(".rd"), p);
argc--;
argv++;
argv[0] = L(".rd");
runmacro('.', argc, argv);
ds(L(".rd"), nil);
}
/*
* print into an allocated string buffer
*/
Rune*
runevsmprint(const char *fmt, va_list args)
{
Fmt f;
int n;
if(runefmtstrinit(&f) < 0)
return nil;
va_copy(f.args,args);
n = dofmt(&f, fmt);
va_end(f.args);
if(f.start == nil)
return nil;
if(n < 0){
free(f.start);
return nil;
}
*(Rune*)f.to = '\0';
return (Rune*)f.start;
}
/* define macro - .de, .am, .ig */
void
r_de(int argc, Rune **argv)
{
Rune *end, *p;
Fmt fmt;
int ignore, len;
delreq(argv[1]);
delraw(argv[1]);
ignore = runestrcmp(argv[0], L("ig")) == 0;
if(!ignore)
runefmtstrinit(&fmt);
end = L("..");
if(argc >= 3)
end = argv[2];
if(runestrcmp(argv[0], L("am")) == 0 && (p=getds(argv[1])) != nil)
fmtrunestrcpy(&fmt, p);
len = runestrlen(end);
while((p = readline(CopyMode)) != nil){
if(runestrncmp(p, end, len) == 0
&& (p[len]==' ' || p[len]==0 || p[len]=='\t'
|| (p[len]=='\\' && p[len+1]=='}'))){
free(p);
goto done;
}
if(!ignore)
fmtprint(&fmt, "%S\n", p);
free(p);
}
warn("eof in %C%S %S - looking for %#Q", dot, argv[0], argv[1], end);
done:
if(ignore)
return;
p = runefmtstrflush(&fmt);
if(p == nil)
sysfatal("out of memory");
ds(argv[1], p);
free(p);
}
