**    May you share freely, never taking more than you give.
**
*************************************************************************
** Code for testing the virtual table interfaces.  This code
** is not included in the SQLite library.  It is used for automated
** testing of the SQLite library.
**
** $Id: test8.c,v 1.21 2006/06/15 07:29:01 danielk1977 Exp $
*/
#include "sqliteInt.h"
#include "tcl.h"
#include "os.h"
#include <stdlib.h>
#include <string.h>

................................................................................
    pConstraint = &pIdxInfo->aConstraint[ii];
    pUsage = &pIdxInfo->aConstraintUsage[ii];

    int iCol = pConstraint->iColumn;
    if( pVtab->aIndex[iCol] ){
      char *zCol = pVtab->aCol[iCol];
      char *zOp = 0;



      switch( pConstraint->op ){
        case SQLITE_INDEX_CONSTRAINT_EQ:
          zOp = "="; break;
        case SQLITE_INDEX_CONSTRAINT_LT:
          zOp = "<"; break;
        case SQLITE_INDEX_CONSTRAINT_GT:
          zOp = ">"; break;
................................................................................
  pIdxInfo->idxNum = hashString(zQuery);
  pIdxInfo->idxStr = zQuery;
  pIdxInfo->needToFreeIdxStr = 1;
  pIdxInfo->estimatedCost = 1.0;
  return SQLITE_OK;
}




























int echoUpdate(sqlite3_vtab *tab, int nData, sqlite3_value **apData){
  echo_vtab *pVtab = (echo_vtab *)tab;
  sqlite3 *db = pVtab->db;
  int rc = SQLITE_OK;







  assert( nData==pVtab->nCol+2 || nData==1 );

  /* If apData[0] is an integer and nData>1 then do an UPDATE */
  if( nData>1 && sqlite3_value_type(apData[0])==SQLITE_INTEGER ){
    char *zUpdate = sqlite3_mprintf("UPDATE %Q", pVtab->zTableName);
    char *zSep = " SET";
    char *zTemp;
    int i, j;
    sqlite3_stmt *pStmt;
    
    if( apData[1] && sqlite3_value_type(apData[1])  &&



           sqlite3_value_int64(apData[0])!=sqlite3_value_int64(apData[1]) ){
       zTemp = sqlite3_mprintf("%s SET rowid=%lld", zUpdate, zSep,
                                sqlite3_value_int64(apData[1]));
       sqlite3_free(zUpdate);
       zUpdate = zTemp;
       zSep = ",";
    }
    for(i=2; i<nData; i++){
      if( apData[i]==0 ) continue;

      zTemp = sqlite3_mprintf("%s%s %Q=?", zUpdate, zSep, pVtab->aCol[i-2]);
      sqlite3_free(zUpdate);
      zUpdate = zTemp;
      zSep = ",";
    }
    zTemp = sqlite3_mprintf("%s WHERE rowid=%lld", zUpdate,
                             sqlite3_value_int64(apData[0]));
    sqlite3_free(zUpdate);
    zUpdate = zTemp;
    rc = sqlite3_prepare(db, zUpdate, -1, &pStmt, 0);
    assert( rc!=SQLITE_OK || pStmt );
    if( rc ) return rc;
    for(i=2, j=1; i<nData; i++){
      if( apData[i]==0 ) continue;
      switch( sqlite3_value_type(apData[i]) ){
        case SQLITE_INTEGER: {
          sqlite3_bind_int64(pStmt, j, sqlite3_value_int64(apData[i]));
          break;

        }
        case SQLITE_FLOAT: {
          sqlite3_bind_double(pStmt, j, sqlite3_value_double(apData[i]));
          break;
        }
        case SQLITE_NULL: {
          sqlite3_bind_null(pStmt, j);
          break;
        }
        case SQLITE_TEXT: {
          sqlite3_bind_text(pStmt, j, sqlite3_value_text(apData[i]),
                            sqlite3_value_bytes(apData[i]), SQLITE_TRANSIENT);
          break;
        }
        case SQLITE_BLOB: {
          sqlite3_bind_blob(pStmt, j, sqlite3_value_blob(apData[i]),
                            sqlite3_value_bytes(apData[i]), SQLITE_TRANSIENT);
          break;
        }
      }
      j++;
    }
    sqlite3_step(pStmt);
    rc = sqlite3_finalize(pStmt);
    return rc;
  }

  /* If apData[0] is an integer, delete the identified row */
  if( sqlite3_value_type(apData[0])==SQLITE_INTEGER ){
    const char *zFormat = "DELETE FROM %Q WHERE rowid = ?";
    char *zDelete = sqlite3_mprintf(zFormat, pVtab->zTableName);
    if( !zDelete ){
      rc = SQLITE_NOMEM;
    }else{
      sqlite3_stmt *pStmt = 0;
      rc = sqlite3_prepare(db, zDelete, -1, &pStmt, 0);
      assert( rc!=SQLITE_OK || pStmt );
      if( rc==SQLITE_OK ){
        sqlite3_bind_value(pStmt, 1, apData[0]);
        sqlite3_step(pStmt);
        rc = sqlite3_finalize(pStmt);

      }
    }
  }

  /* If there is more than a single argument, INSERT a new row */
  if( rc==SQLITE_OK && nData>1 ){


    int ii;
    char *zInsert = 0;
    char *zValues = 0;
    char *zQuery = 0;
    const char *zTab = pVtab->zTableName;

    zInsert = sqlite3_mprintf("INSERT OR REPLACE INTO %Q (rowid", zTab);


    zValues = sqlite3_mprintf("?");


    for(ii=0; ii<pVtab->nCol && zInsert && zValues; ii++){
      char *zNew = sqlite3_mprintf("%s, %Q", zInsert, pVtab->aCol[ii]);
      sqlite3_free(zInsert);
      zInsert = zNew;




      zNew = sqlite3_mprintf("%s, ?", zValues);
      sqlite3_free(zValues);
      zValues = zNew;

    }

    if( zInsert && zValues ){
      zQuery = sqlite3_mprintf("%s) VALUES(%s)", zInsert, zValues);




    }
    if( zQuery ){
      sqlite3_stmt *pStmt = 0;
      rc = sqlite3_prepare(db, zQuery, -1, &pStmt, 0);
      for(ii=1; rc==SQLITE_OK && ii<nData; ii++){
        rc = sqlite3_bind_value(pStmt, ii, apData[ii]);
      }
      if( rc==SQLITE_OK ){
        sqlite3_step(pStmt);
        rc = sqlite3_finalize(pStmt);

      }else{
        sqlite3_finalize(pStmt);


      }
    }

    sqlite3_free(zValues);
    sqlite3_free(zInsert);


    sqlite3_free(zQuery);
    if( rc==SQLITE_OK && (!zValues || !zInsert || !zQuery) ){
      sqlite3FailedMalloc();
      rc = SQLITE_NOMEM;


    }








  }

  return rc;
}

/*
** A virtual table module that merely echos method calls into TCL

**    May you share freely, never taking more than you give.
**
*************************************************************************
** Code for testing the virtual table interfaces.  This code
** is not included in the SQLite library.  It is used for automated
** testing of the SQLite library.
**
** $Id: test8.c,v 1.22 2006/06/15 10:41:16 danielk1977 Exp $
*/
#include "sqliteInt.h"
#include "tcl.h"
#include "os.h"
#include <stdlib.h>
#include <string.h>

................................................................................
    pConstraint = &pIdxInfo->aConstraint[ii];
    pUsage = &pIdxInfo->aConstraintUsage[ii];

    int iCol = pConstraint->iColumn;
    if( pVtab->aIndex[iCol] ){
      char *zCol = pVtab->aCol[iCol];
      char *zOp = 0;
      if( iCol<0 ){
        zCol = "rowid";
      }
      switch( pConstraint->op ){
        case SQLITE_INDEX_CONSTRAINT_EQ:
          zOp = "="; break;
        case SQLITE_INDEX_CONSTRAINT_LT:
          zOp = "<"; break;
        case SQLITE_INDEX_CONSTRAINT_GT:
          zOp = ">"; break;
................................................................................
  pIdxInfo->idxNum = hashString(zQuery);
  pIdxInfo->idxStr = zQuery;
  pIdxInfo->needToFreeIdxStr = 1;
  pIdxInfo->estimatedCost = 1.0;
  return SQLITE_OK;
}

static void string_concat(char **pzStr, char *zAppend, int doFree){
  char *zIn = *pzStr;
  if( zIn ){
    char *zTemp = zIn;
    zIn = sqlite3_mprintf("%s%s", zIn, zAppend);
    sqlite3_free(zTemp);
  }else{
    zIn = sqlite3_mprintf("%s", zAppend);
  }
  *pzStr = zIn;
  if( doFree ){
    sqlite3_free(zAppend);
  }
}

/*
**    apData[0]  apData[1]  apData[2..]
**
**    INTEGER                              DELETE            
**
**    INTEGER    NULL       (nCol args)    UPDATE (do not set rowid)
**    INTEGER    INTEGER    (nCol args)    UPDATE (with SET rowid = <arg1>)
**
**    NULL       NULL       (nCol args)    INSERT INTO (automatic rowid value)
**    NULL       INTEGER    (nCol args)    INSERT (incl. rowid value)
**
*/
int echoUpdate(sqlite3_vtab *tab, int nData, sqlite3_value **apData){
  echo_vtab *pVtab = (echo_vtab *)tab;
  sqlite3 *db = pVtab->db;
  int rc = SQLITE_OK;

  sqlite3_stmt *pStmt;
  char *z = 0;               /* SQL statement to execute */
  int bindArgZero = 0;       /* True to bind apData[0] to sql var no. nData */
  int bindArgOne = 0;        /* True to bind apData[1] to sql var no. 1 */
  int i;                     /* Counter variable used by for loops */

  assert( nData==pVtab->nCol+2 || nData==1 );

  /* If apData[0] is an integer and nData>1 then do an UPDATE */
  if( nData>1 && sqlite3_value_type(apData[0])==SQLITE_INTEGER ){
    z = sqlite3_mprintf("UPDATE %Q", pVtab->zTableName);
    char *zSep = " SET";




    bindArgOne = (apData[1] && sqlite3_value_type(apData[1])==SQLITE_INTEGER);
    bindArgZero = 1;

    if( bindArgOne ){
       string_concat(&z, " SET rowid=?1 ", 0);




       zSep = ",";
    }
    for(i=2; i<nData; i++){
      if( apData[i]==0 ) continue;
      string_concat(&z, sqlite3_mprintf(
          "%s %Q=?%d", zSep, pVtab->aCol[i-2], i), 1);


      zSep = ",";
    }













    string_concat(&z, sqlite3_mprintf(" WHERE rowid=?%d", nData), 0);
  }


























  /* If apData[0] is an integer and nData==1 then do a DELETE */
  else if( nData==1 && sqlite3_value_type(apData[0])==SQLITE_INTEGER ){
    z = sqlite3_mprintf("DELETE FROM %Q WHERE rowid = ?1", pVtab->zTableName);











    bindArgZero = 1;
  }





  /* If the first argument is NULL and there are more than two args, INSERT */
  else if( nData>2 && sqlite3_value_type(apData[0])==SQLITE_NULL ){
    int ii;
    char *zInsert = 0;
    char *zValues = 0;



    zInsert = sqlite3_mprintf("INSERT OR REPLACE INTO %Q (", pVtab->zTableName);
    if( sqlite3_value_type(apData[1])==SQLITE_INTEGER ){
      bindArgOne = 1;
      zValues = sqlite3_mprintf("?");
      string_concat(&zInsert, "rowid", 0);
    }





    assert((pVtab->nCol+2)==nData);
    for(ii=2; ii<nData; ii++){
      string_concat(&zInsert, 
          sqlite3_mprintf("%s%Q", zValues?", ":"", pVtab->aCol[ii-2]), 1);

      string_concat(&zValues, 
          sqlite3_mprintf("%s?%d", zValues?", ":"", ii), 1);
    }



    string_concat(&z, zInsert, 1);
    string_concat(&z, ") VALUES(", 0);
    string_concat(&z, zValues, 1);
    string_concat(&z, ")", 0);
  }









  /* Anything else is an error */
  else{

    assert(0);
    return SQLITE_ERROR;
  }




  rc = sqlite3_prepare(db, z, -1, &pStmt, 0);
  assert( rc!=SQLITE_OK || pStmt );
  sqlite3_free(z);
  if( rc==SQLITE_OK ) {


    if( bindArgZero ){
      sqlite3_bind_value(pStmt, nData, apData[0]);
    }
    if( bindArgOne ){
      sqlite3_bind_value(pStmt, 1, apData[1]);
    }
    for(i=2; i<nData; i++){
      if( apData[i] ) sqlite3_bind_value(pStmt, i, apData[i]);
    }
    sqlite3_step(pStmt);
    rc = sqlite3_finalize(pStmt);
  }

  return rc;
}

/*
** A virtual table module that merely echos method calls into TCL

#    May you find forgiveness for yourself and forgive others.
#    May you share freely, never taking more than you give.
#
#***********************************************************************
# This file implements regression tests for SQLite library.  The
# focus of this file is creating and dropping virtual tables.
#
# $Id: vtab1.test,v 1.16 2006/06/15 07:29:01 danielk1977 Exp $

set testdir [file dirname $argv0]
source $testdir/tester.tcl

ifcapable !vtab {
  finish_test
  return
................................................................................
} {1 2 3}
do_test vtab1-6-4 {
  execsql {
    UPDATE techo SET a = 5;
    SELECT * FROM techo;
  }
} {5 2 3}



do_test vtab1-6-5 {










  execsql {







    DELETE FROM techo;
    SELECT * FROM techo;
  }
} {}



























# check_echo_table vtab1-6.3 treal techo




finish_test

#    May you find forgiveness for yourself and forgive others.
#    May you share freely, never taking more than you give.
#
#***********************************************************************
# This file implements regression tests for SQLite library.  The
# focus of this file is creating and dropping virtual tables.
#
# $Id: vtab1.test,v 1.17 2006/06/15 10:41:16 danielk1977 Exp $

set testdir [file dirname $argv0]
source $testdir/tester.tcl

ifcapable !vtab {
  finish_test
  return
................................................................................
} {1 2 3}
do_test vtab1-6-4 {
  execsql {
    UPDATE techo SET a = 5;
    SELECT * FROM techo;
  }
} {5 2 3}

# TODO: This causes a crash at present.
#
# do_test vtab1-6-5 {
# explain {
#    UPDATE techo set a = a||b||c;
#  }
#  execsql {
#    UPDATE techo set a = a||b||c;
#    SELECT * FROM techo;
#  }
#} {523 2 3}

do_test vtab1-6-6 {
  execsql {
    UPDATE techo set rowid = 10;
    SELECT rowid FROM techo;
  }
} {10}

do_test vtab1-6-7 {
  execsql {
    DELETE FROM techo;
    SELECT * FROM techo;
  }
} {}

file delete -force test2.db
file delete -force test2.db-journal
sqlite3 db2 test2.db
execsql {
  CREATE TABLE techo(a PRIMARY KEY, b, c);
} db2
proc check_echo_table {tn} {
  set ::data1 [execsql {SELECT rowid, * FROM techo}]
  set ::data2 [execsql {SELECT rowid, * FROM techo} db2]
  do_test $tn {
    string equal $::data1 $::data2
  } 1
}
set tn 0
foreach stmt [list \
  {INSERT INTO techo VALUES('abc', 'def', 'ghi')}                        \
  {INSERT INTO techo SELECT a||'.'||rowid, b, c FROM techo}              \
  {INSERT INTO techo SELECT a||'x'||rowid, b, c FROM techo}              \
  {INSERT INTO techo SELECT a||'y'||rowid, b, c FROM techo}              \
  {DELETE FROM techo WHERE (oid % 3) = 0}                                \
  {UPDATE techo set rowid = 100 WHERE rowid = 1}                         \
  {INSERT INTO techo(a, b) VALUES('hello', 'world')}                     \
  {DELETE FROM techo}                                                    \
] {
  execsql $stmt
  execsql $stmt db2
  check_echo_table vtab1-6.6.[incr tn]
}


db2 close
finish_test