/* ltexiolib.c

  Copyright 2006-2010 Taco Hoekwater <[email protected]>

  This file is part of LuaTeX.

  LuaTeX is free software; you can redistribute it and/or modify it under
  the terms of the GNU General Public License as published by the Free
  Software Foundation; either version 2 of the License, or (at your
  option) any later version.

  LuaTeX is distributed in the hope that it will be useful, but WITHOUT
  ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  License for more details.

  You should have received a copy of the GNU General Public License along
  with LuaTeX; if not, see <http://www.gnu.org/licenses/>. */

#include "lua/luatex-api.h"
#include "ptexlib.h"

static const char _svn_version[] =
   "$Id: ltexiolib.c 3550 2010-03-26 14:37:09Z taco $ $URL: http://foundry.supelec.fr/svn/luatex/tags/beta-0.70.1/source/texk/web2c/luatexdir/lua/ltexiolib.c $";

typedef void (*texio_printer) (const char *);

static char *loggable_info = NULL;

static boolean get_selector_value(lua_State * L, int i, int *l)
{
   boolean r = false;
   if (lua_isstring(L, i)) {
       const char *s = lua_tostring(L, i);
       if (strcmp(s, "log") == 0) {
           *l = log_only;
           r = true;
       } else if (strcmp(s, "term") == 0) {
           *l = term_only;
           r = true;
       } else if (strcmp(s, "term and log") == 0) {
           *l = term_and_log;
           r = true;
       }
   } else {
       luaL_error(L, "first argument is not a string");
   }
   return r;
}

static int do_texio_print(lua_State * L, texio_printer printfunction)
{
   const char *s;
   int i = 1;
   int save_selector = selector;
   int n = lua_gettop(L);
   if (n == 0 || !lua_isstring(L, -1)) {
       luaL_error(L, "no string to print");
   }
   if (n > 1) {
       if (get_selector_value(L, i, &selector))
           i++;
   }
   if (selector != log_only && selector != term_only
       && selector != term_and_log) {
       normalize_selector();   /* sets selector */
   }
   for (; i <= n; i++) {
       if (lua_isstring(L, i)) {
           s = lua_tostring(L, i);
           printfunction(s);
       } else {
           luaL_error(L, "argument is not a string");
       }
   }
   selector = save_selector;
   return 0;
}

static void do_texio_ini_print(lua_State * L, const char *extra)
{
   const char *s;
   int i = 1;
   int l = term_and_log;
   int n = lua_gettop(L);
   if (n > 1) {
       if (get_selector_value(L, i, &l))
           i++;
   }
   for (; i <= n; i++) {
       if (lua_isstring(L, i)) {
           s = lua_tostring(L, i);
           if (l == term_and_log || l == term_only)
               fprintf(stdout, "%s%s", extra, s);
           if (l == log_only || l == term_and_log) {
               if (loggable_info == NULL) {
                   loggable_info = strdup(s);
               } else {
                   char *v = concat3(loggable_info, extra, s);
                   free(loggable_info);
                   loggable_info = v;
               }
           }
       }
   }
}

static int texio_print(lua_State * L)
{
   if (ready_already != 314159 || job_name == 0) {
       do_texio_ini_print(L, "");
       return 0;
   }
   return do_texio_print(L, tprint);
}

static int texio_printnl(lua_State * L)
{
   if (ready_already != 314159 || job_name == 0) {
       do_texio_ini_print(L, "\n");
       return 0;
   }
   return do_texio_print(L, tprint_nl);
}

/* at the point this function is called, the selector is log_only */
void flush_loggable_info(void)
{
   if (loggable_info != NULL) {
       fprintf(log_file, "%s\n", loggable_info);
       free(loggable_info);
       loggable_info = NULL;
   }
}


static const struct luaL_reg texiolib[] = {
   {"write", texio_print},
   {"write_nl", texio_printnl},
   {NULL, NULL}                /* sentinel */
};

int luaopen_texio(lua_State * L)
{
   luaL_register(L, "texio", texiolib);
   return 1;
}