/*

 ** 2006 August 23

 **

 ** 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.

 **

 *************************************************************************

 ** Test extension for testing the sqlite3_auto_extension() function.

 **

 ** $Id: test_autoext.c,v 1.5 2008/07/08 02:12:37 drh Exp $

 */

 #include "tcl.h"

 #include "sqlite3ext.h"

 #ifndef SQLITE_OMIT_LOAD_EXTENSION

 static SQLITE_EXTENSION_INIT1

 /*

 ** The sqr() SQL function returns the square of its input value.

 */

 static void sqrFunc(

   sqlite3_context *context,

   int argc,

   sqlite3_value **argv

 ){

   double r = sqlite3_value_double(argv[0]);

   sqlite3_result_double(context, r*r);

 }

 /*

 ** This is the entry point to register the extension for the sqr() function.

 */

 static int sqr_init(

   sqlite3 *db, 

   char **pzErrMsg, 

   const sqlite3_api_routines *pApi

 ){

   SQLITE_EXTENSION_INIT2(pApi);

   sqlite3_create_function(db, "sqr", 1, SQLITE_ANY, 0, sqrFunc, 0, 0);

   return 0;

 }

 /*

 ** The cube() SQL function returns the cube of its input value.

 */

 static void cubeFunc(

   sqlite3_context *context,

   int argc,

   sqlite3_value **argv

 ){

   double r = sqlite3_value_double(argv[0]);

   sqlite3_result_double(context, r*r*r);

 }

 /*

 ** This is the entry point to register the extension for the cube() function.

 */

 static int cube_init(

   sqlite3 *db, 

   char **pzErrMsg, 

   const sqlite3_api_routines *pApi

 ){

   SQLITE_EXTENSION_INIT2(pApi);

   sqlite3_create_function(db, "cube", 1, SQLITE_ANY, 0, cubeFunc, 0, 0);

   return 0;

 }

 /*

 ** This is a broken extension entry point

 */

 static int broken_init(

   sqlite3 *db, 

   char **pzErrMsg, 

   const sqlite3_api_routines *pApi

 ){

   char *zErr;

   SQLITE_EXTENSION_INIT2(pApi);

   zErr = sqlite3_mprintf("broken autoext!");

   *pzErrMsg = zErr;

   return 1;

 }

 /*

 ** tclcmd:   sqlite3_auto_extension_sqr

 **

 ** Register the "sqr" extension to be loaded automatically.

 */

 static int autoExtSqrObjCmd(

   void * clientData,

   Tcl_Interp *interp,

   int objc,

   Tcl_Obj *CONST objv[]

 ){

   int rc = sqlite3_auto_extension((void*)sqr_init);

   Tcl_SetObjResult(interp, Tcl_NewIntObj(rc));

   return SQLITE_OK;

 }

 /*

 ** tclcmd:   sqlite3_auto_extension_cube

 **

 ** Register the "cube" extension to be loaded automatically.

 */

 static int autoExtCubeObjCmd(

   void * clientData,

   Tcl_Interp *interp,

   int objc,

   Tcl_Obj *CONST objv[]

 ){

   int rc = sqlite3_auto_extension((void*)cube_init);

   Tcl_SetObjResult(interp, Tcl_NewIntObj(rc));

   return SQLITE_OK;

 }

 /*

 ** tclcmd:   sqlite3_auto_extension_broken

 **

 ** Register the broken extension to be loaded automatically.

 */

 static int autoExtBrokenObjCmd(

   void * clientData,

   Tcl_Interp *interp,

   int objc,

   Tcl_Obj *CONST objv[]

 ){

   int rc = sqlite3_auto_extension((void*)broken_init);

   Tcl_SetObjResult(interp, Tcl_NewIntObj(rc));

   return SQLITE_OK;

 }

 #endif /* SQLITE_OMIT_LOAD_EXTENSION */

 /*

 ** tclcmd:   sqlite3_reset_auto_extension

 **

 ** Reset all auto-extensions

 */

 static int resetAutoExtObjCmd(

   void * clientData,

   Tcl_Interp *interp,

   int objc,

   Tcl_Obj *CONST objv[]

 ){

   sqlite3_reset_auto_extension();

   return SQLITE_OK;

 }

 /*

 ** This procedure registers the TCL procs defined in this file.

 */

 int Sqlitetest_autoext_Init(Tcl_Interp *interp){

 #ifndef SQLITE_OMIT_LOAD_EXTENSION

   Tcl_CreateObjCommand(interp, "sqlite3_auto_extension_sqr",

           autoExtSqrObjCmd, 0, 0);

   Tcl_CreateObjCommand(interp, "sqlite3_auto_extension_cube",

           autoExtCubeObjCmd, 0, 0);

   Tcl_CreateObjCommand(interp, "sqlite3_auto_extension_broken",

           autoExtBrokenObjCmd, 0, 0);

 #endif

   Tcl_CreateObjCommand(interp, "sqlite3_reset_auto_extension",

           resetAutoExtObjCmd, 0, 0);

   return TCL_OK;

 }