/*************
 * streams.c
 ************/

#include "ngspice/ngspice.h"
#include "ngspice/wordlist.h"
#include "ngspice/bool.h"

#include "variable.h"
#include "terminal.h"
#include "quote.h"
#include "ngspice/cpextern.h"
#include "streams.h"


bool cp_debug = FALSE;
char cp_gt = '>';
char cp_lt = '<';
char cp_amp = '&';

FILE *cp_in = NULL;
FILE *cp_out = NULL;
FILE *cp_err = NULL;

/* These are the fps that cp_ioreset resets the cp_* to.  They are
 * changed by the source routines.  */
FILE *cp_curin = NULL;
FILE *cp_curout = NULL;
FILE *cp_curerr = NULL;


static bool
fileexists(char *name)
{
#ifdef HAVE_ACCESS
    if (access(name, 0) == 0)
        return (TRUE);
#endif
    return (FALSE);
}


/* This routine sets the cp_{in,out,err} pointers and takes the io
 * directions out of the command line.  */
wordlist *
cp_redirect(wordlist *wl)
{
    int gotinput = 0, gotoutput = 0, goterror = 0, append = 0;
    wordlist *w;
    char *fname;
    FILE *fp;

    w = wl->wl_next;    /* Don't consider empty commands. */

    while (w) {
        if (*w->wl_word == cp_lt) {

            wordlist *beg = w;

            if (gotinput++) {
                fprintf(cp_err, "Error: ambiguous input redirect.\n");
                goto error;
            }
            w = w->wl_next;

            if (w && *w->wl_word == cp_lt) {
                fprintf(cp_err, "Error: `<<' redirection is not implemented.\n");
                goto error;
            }

            if (!w) {
                fprintf(cp_err, "Error: missing name for input.\n");
                return (NULL);
            }

            fname = cp_unquote(w->wl_word);
            w = w->wl_next;

#ifdef CPDEBUG
            if (cp_debug)
                fprintf(cp_err, "Input file is %s...\n", fname);
#endif

            fp = fopen(fname, "r");
            tfree(fname);

            if (!fp) {
                perror(fname);
                goto error;
            }

            cp_in = fp;

            wl_delete_slice(beg, w);

        } else if (*w->wl_word == cp_gt) {

            wordlist *beg = w;

            if (gotoutput++) {
                fprintf(cp_err, "Error: ambiguous output redirect.\n");
                goto error;
            }
            w = w->wl_next;

            if (w && *w->wl_word == cp_gt) {
                append++;
                w = w->wl_next;
            }

            if (w && *w->wl_word == cp_amp) {
                if (goterror++) {
                    fprintf(cp_err, "Error: ambiguous error redirect.\n");
                    return (NULL);
                }
                w = w->wl_next;
            }

            if (!w) {
                fprintf(cp_err, "Error: missing name for output.\n");
                return (NULL);
            }

            fname = cp_unquote(w->wl_word);
            w = w->wl_next;

#ifdef CPDEBUG
            if (cp_debug)
                fprintf(cp_err, "Output file is %s... %s\n", fname,
                        append ? "(append)" : "");
#endif

            if (cp_noclobber && fileexists(fname)) {
                fprintf(stderr, "Error: %s: file exists\n", fname);
                goto error;
            }

            fp = fopen(fname, append ? "a" : "w+");
            tfree(fname);

            if (!fp) {
                perror(fname);
                goto error;
            }

            cp_out = fp;
            if (goterror)
                cp_err = fp;

            out_isatty = FALSE;

            wl_delete_slice(beg, w);

        } else {
            w = w->wl_next;
        }
    }
    return (wl);

error:
    wl_free(wl);                /* FIXME, Ouch !! */
    return (NULL);
}


/* Reset the cp_* FILE pointers to the standard ones.  This is tricky,
 * since if we are sourcing a command file, and io has been redirected
 * from inside the file, we have to reset it back to what it was for
 * the source, not for the top level.  That way if you type "foo >
 * bar" where foo is a script, and it has redirections of its own
 * inside of it, none of the output from foo will get sent to
 * stdout...  */

void
cp_ioreset(void)
{
    if (cp_in != cp_curin)
        if (cp_in)
            fclose(cp_in);
    if (cp_out != cp_curout)
        if (cp_out)
            fclose(cp_out);
    if (cp_err != cp_curerr)
        if (cp_err  &&  cp_err != cp_out)
            fclose(cp_err);

    cp_in  = cp_curin;
    cp_out = cp_curout;
    cp_err = cp_curerr;

    /*** Minor bug here... */
    out_isatty = TRUE;
}


/* Do this only right before an exec, since we lose the old std*'s. */

void
fixdescriptors(void)
{
    if (cp_in != stdin)
        dup2(fileno(cp_in), fileno(stdin));
    if (cp_out != stdout)
        dup2(fileno(cp_out), fileno(stdout));
    if (cp_err != stderr)
        dup2(fileno(cp_err), fileno(stderr));
}