/*
    * DBConnection.cpp
    * Part of SQLMeta, a language to use sql-queries in html pages.
    *
    * Copyright (C) 2001  Daan Vreeken
    *
    * This program 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.
    *
    * This program 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 General Public License for more details.
    *
    * You should have received a copy of the GNU General Public License
    * along with this program; if not, write to the Free Software
    * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
    *
    */
   
   #include "sqlmeta.h"
   
   
   #define NeedReconnect		0
   #define ChangedUser		1
   #define ChangedDB		2
   
   
   
   
   DBConnection::DBConnection(MetaParser *MyParent, DBInfo *MyInfo)
   {
   	Parent=MyParent;
   
   	Info=MyInfo;
   	Connected=0;
   	RowNr=0;
   }
   
   
   
   int DBConnection::TryConnect(void)
   {
   	if (Info->Host==NULL)
   	{
   		Parent->Error(ErrSQLConnect,"No hostname specified");
   		return 0;
   	}
   	if (Info->User==NULL)
   	{
   		Parent->Error(ErrSQLConnect,"No username specified");
   		return 0;
   	}
   	if (Info->Password==NULL)
   	{
   		Parent->Error(ErrSQLConnect,"No password specified");
   		return 0;
   	}
   	if (Info->DB==NULL)
   	{
   		Parent->Error(ErrSQLConnect,"No database specified");
   		return 0;
   	}
   
   	Connection=(MYSQL *)malloc(sizeof(MYSQL));
   	if (Connection==NULL)
   	{
   		Parent->Error(ErrMalloc,"DBConnection::TryConnect");
   		return 0;
   	}
   
   	mysql_init(Connection);
   
   	if (!mysql_real_connect(Connection,Info->Host,Info->User,Info->Password,Info->DB,0,NULL,0))
   	{
   		ShowError(ErrSQLConnect);
   		return 0;
   	}
   
   	Connected=1;
   	return 1;
   }
   
   
   
   int DBConnection::Query(char *Query)
   {
   	if (Connection==NULL)
   	{
   		ShowError(ErrSQLConnectFirst);
   		return 0;
   	}
   
   	if (mysql_query(Connection,Query))
   	{
   		ShowError(ErrSQLQuery);
   		return 0;
   	}
   	return 1;
   }
   
   
   
   int DBConnection::GetResult(void)
   {
   	if (Connection==NULL)
   	{
   		ShowError(ErrSQLConnectFirst);
   		return 0;
   	}
   
   	Result=mysql_use_result(Connection);
   	if (Result==NULL)
   	{
   		ShowError(ErrSQLNoResult);
   		return 0;
   	}
   
   	Fields=mysql_num_fields(Result);
   	FieldName=mysql_fetch_fields(Result);
   
   	return 1;
   }
   
   
   
   int DBConnection::NextRow(void)
   {
   	Row=mysql_fetch_row(Result);
   	RowNr++;
   
   	return (Row!=NULL);
   }
   
   
   
   void DBConnection::FreeResult(void)
   {
   	mysql_free_result(Result);
   	Result=NULL;
   }
   
   
   
   void DBConnection::ShowError(char *Err)
   {
   	Parent->Error(Err,mysql_error(Connection));
   }
   
   
   
   
   DBConnection::~DBConnection(void)
   {
   	if (Connected)
   	{
   		if (Result)
   			FreeResult();
   
   		mysql_close(Connection);
   		free(Connection);
   	}
   }