SQLite

/*
** 2001 September 15
**
** The author disclaims copyright to this source code.  In place of
** a legal notice, here is a blessing:
**
**    May you do good and not evil.
**    May you find forgiveness for yourself and forgive others.
**    May you share freely, never taking more than you give.
**
*************************************************************************
** This file contains code to implement the "sqlite" command line
** utility for accessing SQLite databases.
ToDo: The -utf8 option may not play well with line editing input libraries.
*/
#if (defined(_WIN32) || defined(WIN32)) && !defined(_CRT_SECURE_NO_WARNINGS)
/* This needs to come before any includes for MSVC compiler */
#define _CRT_SECURE_NO_WARNINGS
#endif
typedef unsigned int u32;
typedef unsigned short int u16;

/*
** optionally #include a user-defined header, whereby compilation options
** may be set prior to where they take effect, but after platform setup.
** If SQLITE_CUSTOM_INCLUDE=? is defined, its value names the #include
** file. Note that this macro has a like effect on sqlite3.c compilation.
*/
# define SHELL_STRINGIFY_(f) #f
# define SHELL_STRINGIFY(f) SHELL_STRINGIFY_(f)
#ifdef SQLITE_CUSTOM_INCLUDE

  }
  return dynPrompt.dynamicPrompt;
}
#endif /* !defined(SQLITE_OMIT_DYNAPROMPT) */

#if (defined(_WIN32) || defined(WIN32)) && !defined(SQLITE_SHELL_FIDDLE)
static int infsMode;
static UINT codePage = 0;
static HANDLE hConsoleIn;
static DWORD consoleMode;

/*
** Prepare input stream, (if known to be a WIN32 console), for UTF-8
** input (from either typing or suitable paste operations) and for
** UTF-8 rendering. This may "fail" with a message to stderr, where
** the preparation is not done and common "code page" issues occur.
*/
static void instream_prepare(void){
  if( isatty(0) ){
    if( !IsValidCodePage(CP_UTF8) ){
      fprintf(stderr, "Cannot use UTF-8 code page.\n");
      console_utf8 = 0;
      return;
    }
    codePage = GetConsoleCP();

    console_utf8 = 0;
  }
}

/*
** Undo the effects of instream_prepare(), if any.
*/
static void SQLITE_CDECL instream_restore(void){
  if( console_utf8 && codePage!=0 ){
    _setmode(_fileno(stdin), infsMode);
    SetConsoleCP(codePage);
    SetConsoleMode( hConsoleIn, consoleMode );
    console_utf8 = 0; /* Avoid multiple calls. */
  }
}

/*
** Collect input like fgets(...) with special provisions for input
** from the Windows console to get around its strange coding issues.
** Defers to plain fgets() when input is not interactive or when the
** startup option, -utf8, has not been provided or taken effect.



*/
static char* utf8_fgets(char *buf, int ncmax, FILE *fin){
  #define SQLITE_IA_LINE_LEN 150
  if( fin==stdin && stdin_is_interactive && console_utf8 ){

    wchar_t wbuf[SQLITE_IA_LINE_LEN];
    int noc = 0;
    if( ncmax == 0 ) return 0;
    buf[0] = 0;
    while( noc < ncmax-7-1 ){
      /* There is room for at least 2 more characters and the 0-terminator. */
      int na = (ncmax-1 - noc)/4;
      wchar_t * wz;
      if( na > SQLITE_IA_LINE_LEN ) na = SQLITE_IA_LINE_LEN;
      wz = fgetws(wbuf, na, stdin);
      if( wz != 0 ){
        int nmb = WideCharToMultiByte(CP_UTF8, 0, wz, -1, 0, 0, 0, 0);
        if( nmb !=0 && noc+nmb <= ncmax ){
          WideCharToMultiByte(CP_UTF8, 0, wz, -1, buf+noc, nmb, 0, 0);
          noc += nmb-1; /* Strip 0-terminator added by above call. */
          if( buf[noc-1] == '\n' ) break;
        }else break; /* Drop apparent garbage in. (Could assert.) */
      }else break;
    }


    if( noc == 0 ) return 0;
    return buf;
  }else{
    return fgets(buf, ncmax, fin);
  }
}

# define fgets(b,n,f) utf8_fgets(b,n,f)

#if SQLITE_SHELL_IS_UTF8
int SQLITE_CDECL main(int argc, char **argv){
#else
int SQLITE_CDECL wmain(int argc, wchar_t **wargv){
  char **argv;
#endif
#ifdef SQLITE_DEBUG
  sqlite3_int64 mem_main_enter = 0;
#endif
  char *zErrMsg = 0;
#ifdef SQLITE_SHELL_FIDDLE
#  define data shellState
#else
  ShellState data;
#endif

  stdin_is_interactive = 0;
  stdout_is_console = 1;
  data.wasm.zDefaultDbName = "/fiddle.sqlite3";
#else
  stdin_is_interactive = isatty(0);
  stdout_is_console = isatty(1);
#endif
#if (defined(_WIN32) || defined(WIN32)) && !defined(SQLITE_SHELL_FIDDLE)
  atexit(instream_restore); /* Needs revision for CLI as library call */
#endif
#ifdef SQLITE_DEBUG
  mem_main_enter = sqlite3_memory_used();
#endif
#if !defined(_WIN32_WCE)
  if( getenv("SQLITE_DEBUG_BREAK") ){
    if( isatty(0) && isatty(2) ){
      fprintf(stderr,
          "attach debugger to process %d and press any key to continue.\n",
          GETPID());
      fgetc(stdin);

        free(zHistory);
      }
    }else{
      data.in = stdin;
      rc = process_input(&data);
    }
  }





#ifndef SQLITE_SHELL_FIDDLE
  /* In WASM mode we have to leave the db state in place so that
  ** client code can "push" SQL into it after this call returns. */
  free(azCmd);
  set_table_name(&data, 0);
  if( data.db ){
    session_close_all(&data, -1);