/*=========================================================================

 Program: Visualization Toolkit
 Module: vtkSQLQuery.cxx

 Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
 All rights reserved.
 See Copyright.txt or http://www.kitware.com/Copyright.htm for details.

 This software is distributed WITHOUT ANY WARRANTY; without even
 the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 PURPOSE. See the above copyright notice for more information.

=========================================================================*/
/*-------------------------------------------------------------------------
 Copyright 2008 Sandia Corporation.
 Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
 the U.S. Government retains certain rights in this software.
-------------------------------------------------------------------------*/
#include "vtkSQLQuery.h"

#include "vtkObjectFactory.h"
#include "vtkSQLDatabase.h"
#include "vtkVariantArray.h"

#include "vtksys/SystemTools.hxx"


vtkSQLQuery::vtkSQLQuery()
{
 this->Query = nullptr;
 this->Database = nullptr;
 this->Active = false;
}

vtkSQLQuery::~vtkSQLQuery()
{
 this->SetQuery(nullptr);
 if (this->Database)
 {
 this->Database->Delete();
 this->Database = nullptr;
 }
}

vtkCxxSetObjectMacro(vtkSQLQuery, Database, vtkSQLDatabase);

void vtkSQLQuery::PrintSelf(ostream &os, vtkIndent indent)
{
 this->Superclass::PrintSelf(os, indent);
 os << indent << "Query: " << (this->Query ? this->Query : "nullptr") << endl;
 os << indent << "Database: " << (this->Database ? "" : "nullptr") << endl;
 if (this->Database)
 {
 this->Database->PrintSelf(os, indent.GetNextIndent());
 }
}

vtkStdString vtkSQLQuery::EscapeString( vtkStdString s, bool addSurroundingQuotes )
{
 vtkStdString d;
 if ( addSurroundingQuotes )
 {
 d += '\'';
 }

 for ( vtkStdString::iterator it = s.begin(); it != s.end(); ++ it )
 {
 if ( *it == '\'' )
 d += '\''; // Single quotes are escaped by repeating them
 d += *it;
 }

 if ( addSurroundingQuotes )
 {
 d += '\'';
 }
 return d;
}

char* vtkSQLQuery::EscapeString( const char* src, bool addSurroundingQuotes )
{
 vtkStdString sstr( src );
 vtkStdString dstr = this->EscapeString( sstr, addSurroundingQuotes );
 return vtksys::SystemTools::DuplicateString( dstr.c_str() );
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), unsigned char vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), signed char vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}


bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), unsigned short vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), short vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), unsigned int vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), int vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), unsigned long vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), long vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), unsigned long long vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), long long vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), float vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), double vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), const char *vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), const char *vtkNotUsed(value), size_t vtkNotUsed(length))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}


bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), const vtkStdString &vtkNotUsed(value))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int vtkNotUsed(index), const void *vtkNotUsed(value), size_t vtkNotUsed(length))
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::ClearParameterBindings()
{
 vtkErrorMacro(<<"This database driver does not support bound parameters.");
 return false;
}

bool vtkSQLQuery::BindParameter(int index, vtkVariant data)
{
 if (!data.IsValid())
 {
 return true; // binding nulls is a no-op
 }

#define VTK_VARIANT_BIND_PARAMETER(Type,Function) \
 case Type: return this->BindParameter(index, data.Function())

 switch (data.GetType())
 {
 VTK_VARIANT_BIND_PARAMETER(VTK_STRING,ToString);
 VTK_VARIANT_BIND_PARAMETER(VTK_FLOAT,ToFloat);
 VTK_VARIANT_BIND_PARAMETER(VTK_DOUBLE,ToDouble);
 VTK_VARIANT_BIND_PARAMETER(VTK_CHAR,ToChar);
 VTK_VARIANT_BIND_PARAMETER(VTK_UNSIGNED_CHAR,ToUnsignedChar);
 VTK_VARIANT_BIND_PARAMETER(VTK_SIGNED_CHAR,ToSignedChar);
 VTK_VARIANT_BIND_PARAMETER(VTK_SHORT,ToShort);
 VTK_VARIANT_BIND_PARAMETER(VTK_UNSIGNED_SHORT,ToUnsignedShort);
 VTK_VARIANT_BIND_PARAMETER(VTK_INT,ToInt);
 VTK_VARIANT_BIND_PARAMETER(VTK_UNSIGNED_INT,ToUnsignedInt);
 VTK_VARIANT_BIND_PARAMETER(VTK_LONG,ToLong);
 VTK_VARIANT_BIND_PARAMETER(VTK_UNSIGNED_LONG,ToUnsignedLong);
 VTK_VARIANT_BIND_PARAMETER(VTK_LONG_LONG,ToLongLong);
 VTK_VARIANT_BIND_PARAMETER(VTK_UNSIGNED_LONG_LONG,ToUnsignedLongLong);
 case VTK_OBJECT:
 vtkErrorMacro(<<"Variants of type VTK_OBJECT cannot be inserted into a database.");
 return false;
 default:
 vtkErrorMacro(<<"Variants of type "
 << data.GetType() << " are not currently supported by BindParameter.");
 return false;
 }
}

// ----------------------------------------------------------------------

bool vtkSQLQuery::SetQuery(const char *queryString)
{
 // This is just vtkSetStringMacro from vtkSetGet.h

 vtkDebugMacro(<< this->GetClassName()
 << " (" << this << "): setting Query to "
 << (queryString?queryString:"(null)") );

 if ( this->Query == nullptr && queryString == nullptr)
 {
 return true;
 }
 if ( this->Query && queryString && (!strcmp(this->Query,queryString)))
 {
 return true; // query string isn't changing
 }
 delete [] this->Query;
 if (queryString)
 {
 size_t n = strlen(queryString) + 1;
 char *cp1 = new char[n];
 const char *cp2 = (queryString);
 this->Query = cp1;
 do { *cp1++ = *cp2++; } while ( --n ); \
 }
 else
 {
 this->Query = nullptr;
 }
 this->Modified();
 return true;
}

// As above, this is a copy of vtkGetStringMacro from vtkGetSet.h.

const char* vtkSQLQuery::GetQuery()
{
 vtkDebugMacro(<< this->GetClassName()
 << " (" << this << "): returning Query of "
 << (this->Query?this->Query:"(null)"));
 return this->Query;
}