Subversion Repositories planix.SVN

#include "a.h"

/*
 * 19. Input/output file switching.
 */

/* .so - push new source file */
void
r_so(int argc, Rune **argv)
{
        USED(argc);
        pushinputfile(argv[1]);
}

/* .nx - end this file, switch to arg */
void
r_nx(int argc, Rune **argv)
{
        int n;
        
        if(argc == 1){
                while(popinput())
                        ;
        }else{
                if(argc > 2)
                        warn("too many arguments for .nx");
                while((n=popinput()) && n != 2)
                        ;
                pushinputfile(argv[1]);
        }
}

/* .sy - system: run string */
void
r_sy(Rune *name)
{
        USED(name);
        warn(".sy not implemented");
}

/* .pi - pipe output to string */
void
r_pi(Rune *name)
{
        USED(name);
        warn(".pi not implemented");
}

/* .cf - copy contents of filename to output */
void
r_cf(int argc, Rune **argv)
{
        int c;
        char *p;
        Biobuf *b;

        USED(argc);
        p = esmprint("%S", argv[1]);
        if((b = Bopen(p, OREAD)) == nil){
                fprint(2, "%L: open %s: %r\n", p);
                free(p);
                return;
        }
        free(p);

        while((c = Bgetrune(b)) >= 0)
                outrune(c);
        Bterm(b);
}

void
r_inputpipe(Rune *name)
{
        Rune *cmd, *stop, *line;
        int n, pid, p[2], len;
        Waitmsg *w;
        
        USED(name);
        if(pipe(p) < 0){
                warn("pipe: %r");
                return;
        }
        stop = copyarg();
        cmd = readline(CopyMode);
        pid = fork();
        switch(pid){
        case 0:
                if(p[0] != 0){
                        dup(p[0], 0);
                        close(p[0]);
                }
                close(p[1]);
                execl("/bin/rc", "rc", "-c", esmprint("%S", cmd), nil);
                warn("%Cdp %S: %r", dot, cmd);
                _exits(nil);
        case -1:
                warn("fork: %r");
        default:
                close(p[0]);
                len = runestrlen(stop);
                fprint(p[1], ".ps %d\n", getnr(L(".s")));
                fprint(p[1], ".vs %du\n", getnr(L(".v")));
                fprint(p[1], ".ft %d\n", getnr(L(".f")));
                fprint(p[1], ".ll 8i\n");
                fprint(p[1], ".pl 30i\n");
                while((line = readline(~0)) != nil){
                        if(runestrncmp(line, stop, len) == 0 
                        && (line[len]==' ' || line[len]==0 || line[len]=='\t'
                                || (line[len]=='\\' && line[len+1]=='}')))
                                break;
                        n = runestrlen(line);
                        line[n] = '\n';
                        fprint(p[1], "%.*S", n+1, line);
                        free(line);
                }
                free(stop);
                close(p[1]);
                w = wait();
                if(w == nil){
                        warn("wait: %r");
                        return;
                }
                if(w->msg[0])
                        sysfatal("%C%S %S: %s", dot, name, cmd, w->msg);
                free(cmd);
                free(w);
        }
}       

void
t19init(void)
{
        addreq(L("so"), r_so, 1);
        addreq(L("nx"), r_nx, -1);
        addraw(L("sy"), r_sy);
        addraw(L("inputpipe"), r_inputpipe);
        addraw(L("pi"), r_pi);
        addreq(L("cf"), r_cf, 1);
        
        nr(L("$$"), getpid());
}