/*
 *	 Copyright 1999, Dmitry Kovalev (zmit@mail.ru), All rights reserved.
 *
 *	 Redistribution and use in source and binary forms are permitted only
 *	 as authorized by the OpenLDAP Public License.	A copy of this
 *	 license is available at http://www.OpenLDAP.org/license.html or
 *	 in file LICENSE in the top-level directory of the distribution.
 */

#include "portable.h"

#ifdef SLAPD_SQL

#include <stdio.h>
#include <sys/types.h>
#include <string.h>
#include "slap.h"
#include "back-sql.h"
#include "sql-wrap.h"
#include "schema-map.h"
#include "entry-id.h"
#include "util.h"

int backsql_modify(BackendDB *be,Connection *conn,Operation *op,
	char *dn,char *ndn,LDAPModList *modlist)
{
 backsql_info *bi=(backsql_info*)be->be_private;
 SQLHDBC dbh;
 SQLHSTMT sth;
 RETCODE rc;
 backsql_oc_map_rec *oc=NULL;
 backsql_entryID e_id,*res;
 LDAPModList *c_mod;
 backsql_at_map_rec *at=NULL;
 struct berval *at_val;
 int i;

 dn=dn_validate(dn);
 Debug(LDAP_DEBUG_TRACE,"==>backsql_modify(): changing entry '%s'\n",dn,0,0);
 dbh=backsql_get_db_conn(be,conn);
 if (!dbh)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_modify(): could not get connection handle - exiting\n",0,0,0);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }
 res=backsql_dn2id(&e_id,dbh,dn);
 if (res==NULL)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_modify(): could not lookup entry id\n",0,0,0);
  send_ldap_result(conn,op,LDAP_NO_SUCH_OBJECT,"",NULL,NULL,NULL);
  return 1;
 }

 oc=backsql_oc_with_id(bi,e_id.oc_id);
 if (oc==NULL)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_modify(): cannot determine objectclass of entry -- aborting\n",0,0,0);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }

 SQLAllocStmt(dbh, &sth);

 Debug(LDAP_DEBUG_TRACE,"backsql_modify(): traversing modifications list\n",0,0,0);
 for(c_mod=modlist;c_mod!=NULL;c_mod=c_mod->ml_next)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_modify(): attribute '%s'\n",c_mod->ml_type,0,0);
  at=backsql_at_with_name(oc,c_mod->ml_type);
  if (at==NULL)
  {
   Debug(LDAP_DEBUG_TRACE,"backsql_add(): attribute provided is not registered in this objectclass ('%s')\n",c_mod->ml_type,0,0);
   continue;
  }
  SQLBindParameter(sth,1,SQL_PARAM_INPUT,SQL_C_ULONG,SQL_INTEGER,0,0,&e_id.keyval,0,0);
  switch(c_mod->ml_op)
  {
   case LDAP_MOD_REPLACE:
			{
			 char *query;
			 int qlen;
			 SQLHSTMT asth;
			 BACKSQL_ROW_NTS row;
			 
			 Debug(LDAP_DEBUG_TRACE,"backsql_modify(): replacing values for attribute '%s'\n",at->name,0,0);
             if (at->add_proc==NULL)
			 {
			  Debug(LDAP_DEBUG_TRACE,"backsql_modify(): add procedure is not defined for this attribute ('%s') - unable to perform replacements\n",at->name,0,0);
			  break;
			 }
del_all:
			 query=NULL;
			 qlen=0;
			 query=backsql_strcat(query,&qlen,"SELECT ",at->sel_expr," AS ",at->name,
						" FROM ",at->from_tbls,
						" WHERE ",oc->keytbl,".",oc->keycol,"=?",NULL);
			 if (at->join_where!=NULL && at->join_where[0]!='\0')
			  query=backsql_strcat(query,&qlen," AND ",at->join_where,NULL);

			 Debug(LDAP_DEBUG_TRACE,"backsql_modify() constructed query to get all existing values: %s\n",query,0,0);
			 if ((rc=backsql_Prepare(dbh,&asth,query,0)) != SQL_SUCCESS)
			 {
			  Debug(LDAP_DEBUG_TRACE,"backsql_get_attr_values(): error preparing query\n",0,0,0);
			  backsql_PrintErrors(bi->db_env,dbh,asth,rc);
			  free(query);
			  break;
			 }
			 free(query);

			 if (backsql_BindParamID(asth,1,&e_id.keyval) != SQL_SUCCESS)
			 {
			  Debug(LDAP_DEBUG_TRACE,"backsql_get_attr_values(): error binding key value parameter\n",0,0,0);
			  backsql_PrintErrors(bi->db_env,dbh,asth,rc);
			  SQLFreeStmt(asth,SQL_DROP);
			  break;
			 }

			 if ((rc=SQLExecute(asth)) != SQL_SUCCESS && rc!= SQL_SUCCESS_WITH_INFO)
			 {
			  Debug(LDAP_DEBUG_TRACE,"backsql_get_attr_values(): error executing attribute query\n",0,0,0);
			  backsql_PrintErrors(bi->db_env,dbh,asth,rc);
			  SQLFreeStmt(asth,SQL_DROP);
			  break;
			 }

			 backsql_BindRowAsStrings(asth,&row);
			 while ((rc=SQLFetch(asth)) == SQL_SUCCESS || rc==SQL_SUCCESS_WITH_INFO)
			 {
			  for (i=0;i<row.ncols;i++)
			  {
			   SQLBindParameter(sth,2,SQL_PARAM_INPUT,SQL_C_CHAR,SQL_CHAR,0,0,row.cols[i],strlen(row.cols[i]),0);
			   Debug(LDAP_DEBUG_TRACE,"backsql_modify(): executing '%s'\n",at->delete_proc,0,0);
			   rc=SQLExecDirect(sth,at->delete_proc,SQL_NTS);
			   if (rc!=SQL_SUCCESS)
				{
			     Debug(LDAP_DEBUG_TRACE,"backsql_modify(): delete_proc execution failed\n",0,0,0);
			     backsql_PrintErrors(bi->db_env,dbh,sth,rc);
				}
			  }
			 }
			 backsql_FreeRow(&row);
             SQLFreeStmt(asth,SQL_DROP);
			}
			//PASSTHROUGH - to add new attributes -- do NOT add break
  case LDAP_MOD_ADD:
			if (at->add_proc==NULL)
			{
			 Debug(LDAP_DEBUG_TRACE,"backsql_modify(): add procedure is not defined for this attribute ('%s')\n",at->name,0,0);
			 break;
			}
			if (c_mod->ml_bvalues==NULL)
			{
			 Debug(LDAP_DEBUG_TRACE,"backsql_modify(): no values given to add for attribute '%s'\n",at->name,0,0);
			 break;
			}
			Debug(LDAP_DEBUG_TRACE,"backsql_modify(): adding new values for attribute '%s'\n",at->name,0,0);
			for(i=0,at_val=c_mod->ml_bvalues[0];at_val!=NULL;i++,at_val=c_mod->ml_bvalues[i])
			{
			 //check for syntax here - maybe need binary bind?
			 SQLBindParameter(sth,2,SQL_PARAM_INPUT,SQL_C_CHAR,SQL_CHAR,0,0,at_val->bv_val,at_val->bv_len,0);
			 Debug(LDAP_DEBUG_TRACE,"backsql_modify(): executing '%s'\n",at->add_proc,0,0);
			 rc=SQLExecDirect(sth,at->add_proc,SQL_NTS);
			 if (rc!=SQL_SUCCESS)
			 {
			  Debug(LDAP_DEBUG_TRACE,"backsql_modify(): add_proc execution failed\n",0,0,0);
			  backsql_PrintErrors(bi->db_env,dbh,sth,rc);
			 }
			}
			break;
  case LDAP_MOD_DELETE:
			if (at->delete_proc==NULL)
			{
			 Debug(LDAP_DEBUG_TRACE,"backsql_modify(): delete procedure is not defined for this attribute ('%s')\n",at->name,0,0);
			 break;
			}
			if (c_mod->ml_bvalues==NULL)
			{
			 Debug(LDAP_DEBUG_TRACE,"backsql_modify(): no values given to delete for attribute '%s' -- deleting all values\n",at->name,0,0);
			 goto del_all;
			}
            Debug(LDAP_DEBUG_TRACE,"backsql_modify(): deleting values for attribute '%s'\n",at->name,0,0);
			for(i=0,at_val=c_mod->ml_bvalues[0];at_val!=NULL;i++,at_val=c_mod->ml_bvalues[i])
			{
			 //check for syntax here - maybe need binary bind?
			 SQLBindParameter(sth,2,SQL_PARAM_INPUT,SQL_C_CHAR,SQL_CHAR,0,0,at_val->bv_val,at_val->bv_len,0);
			 Debug(LDAP_DEBUG_TRACE,"backsql_modify(): executing '%s'\n",at->delete_proc,0,0);
			 rc=SQLExecDirect(sth,at->delete_proc,SQL_NTS);
			 if (rc!=SQL_SUCCESS)
			 {
			  Debug(LDAP_DEBUG_TRACE,"backsql_modify(): delete_proc execution failed\n",0,0,0);
			  backsql_PrintErrors(bi->db_env,dbh,sth,rc);
			 }
			}
			break;
  }
  SQLFreeStmt(sth,SQL_RESET_PARAMS);
 }

 SQLFreeStmt(sth,SQL_DROP);
 send_ldap_result(conn,op,LDAP_SUCCESS,"",NULL,NULL,NULL);
 Debug(LDAP_DEBUG_TRACE,"<==backsql_modify()\n",0,0,0);
 return 0;
}

int backsql_modrdn(BackendDB *be,Connection *conn,Operation *op,
	char *dn,char *ndn,char *newrdn,int deleteoldrdn,char *newSuperior)
{
 Debug(LDAP_DEBUG_TRACE,"==>backsql_modrdn()\n",0,0,0);
 return 0;
}

int backsql_add(BackendDB *be,Connection *conn,Operation *op,Entry *e)
{
 backsql_info *bi=(backsql_info*)be->be_private;
 SQLHDBC dbh;
 SQLHSTMT sth;
 unsigned long new_keyval;
 long i;
 RETCODE rc;
 backsql_oc_map_rec *oc=NULL;
 backsql_at_map_rec *at_rec=NULL;
 backsql_entryID parent_id,*res;
 Attribute *at;
 struct berval *at_val;
 char *pdn;

 Debug(LDAP_DEBUG_TRACE,"==>backsql_add(): adding entry '%s'\n",e->e_dn,0,0);
 if (dn_validate(e->e_dn)==NULL)
 {
  Debug(LDAP_DEBUG_TRACE,"==>backsql_add(): invalid dn '%s' -- aborting\n",e->e_dn,0,0);
 }
 for(at=e->e_attrs;at!=NULL;at=at->a_next)
 {
  //Debug(LDAP_DEBUG_TRACE,"backsql_add(): scanning entry -- %s\n",at->a_type,0,0);
  if (!strcasecmp(at->a_type,"objectclass"))
  {
   oc=backsql_oc_with_name(bi,at->a_vals[0]->bv_val);
   break;
  }
 }

 if (oc==NULL)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_add(): cannot determine objectclass of entry -- aborting\n",0,0,0);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }
 if (oc->create_proc == NULL)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_add(): create procedure is not defined for this objectclass - aborting\n",0,0,0);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }

 dbh=backsql_get_db_conn(be,conn);
 if (!dbh)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_add(): could not get connection handle - exiting\n",0,0,0);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }

 SQLAllocStmt(dbh, &sth);
 SQLBindParameter(sth,1,SQL_PARAM_OUTPUT,SQL_C_ULONG,SQL_INTEGER,0,0,&new_keyval,0,0);
 //SQLBindParameter(sth,2,SQL_PARAM_OUTPUT,SQL_C_SLONG,SQL_INTEGER,0,0,&retcode,0,0);

 Debug(LDAP_DEBUG_TRACE,"backsql_add(): executing '%s'\n",oc->create_proc,0,0);
 rc=SQLExecDirect(sth,oc->create_proc,SQL_NTS);
 if (rc != SQL_SUCCESS)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_add(): create_proc execution failed\n",0,0,0);
  backsql_PrintErrors(bi->db_env,dbh,sth,rc);
  SQLFreeStmt(sth,SQL_DROP);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }
 SQLFreeStmt(sth,SQL_RESET_PARAMS);
 Debug(LDAP_DEBUG_TRACE,"backsql_add(): create_proc returned keyval=%d\n",new_keyval,0,0);

 for(at=e->e_attrs;at!=NULL;at=at->a_next)
 {
  at_rec=backsql_at_with_name(oc,at->a_type);
  if (at_rec==NULL)
  {
   Debug(LDAP_DEBUG_TRACE,"backsql_add(): attribute provided is not registered in this objectclass ('%s')\n",at->a_type,0,0);
   continue;
  }
  if (at_rec->add_proc==NULL)
  {
   Debug(LDAP_DEBUG_TRACE,"backsql_add(): add procedure is not defined for this attribute ('%s')\n",at->a_type,0,0);
   continue;
  }
  SQLBindParameter(sth,1,SQL_PARAM_INPUT,SQL_C_LONG,SQL_INTEGER,0,0,&new_keyval,0,0);
  for(i=0,at_val=at->a_vals[0];at_val!=NULL;i++,at_val=at->a_vals[i])
  {
   //if (at->a_syntax==SYNTAX_BIN)
   // SQLBindParameter(sth,2,SQL_PARAM_INPUT,SQL_C_CHAR,SQL_BINARY,0,0,at_val->bv_val,0,0);
   //else
    SQLBindParameter(sth,2,SQL_PARAM_INPUT,SQL_C_CHAR,SQL_CHAR,0,0,at_val->bv_val,at_val->bv_len,0);
   Debug(LDAP_DEBUG_TRACE,"backsql_add(): executing '%s'\n",at_rec->add_proc,0,0);
   rc=SQLExecDirect(sth,at_rec->add_proc,SQL_NTS);
   if (rc!=SQL_SUCCESS)
   {
	Debug(LDAP_DEBUG_TRACE,"backsql_add(): add_proc execution failed\n",0,0,0);
	backsql_PrintErrors(bi->db_env,dbh,sth,rc);
   }
  }
 }
 SQLFreeStmt(sth,SQL_RESET_PARAMS); 
 pdn=dn_parent(be,e->e_dn);
 res=backsql_dn2id(&parent_id,dbh,pdn);
 if (res==NULL)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_add(): could not lookup parent entry for new record ('%s')\n",
												pdn,0,0);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }
 free(pdn);
 backsql_BindParamStr(sth,1,e->e_dn,BACKSQL_MAX_DN_LEN);
 SQLBindParameter(sth,2,SQL_PARAM_INPUT,SQL_C_LONG,SQL_INTEGER,0,0,&oc->id,0,0);
 SQLBindParameter(sth,3,SQL_PARAM_INPUT,SQL_C_LONG,SQL_INTEGER,0,0,&parent_id.id,0,0);
 SQLBindParameter(sth,4,SQL_PARAM_INPUT,SQL_C_LONG,SQL_INTEGER,0,0,&new_keyval,0,0);
 rc=SQLExecDirect(sth,bi->insentry_query,SQL_NTS);
 if (rc != SQL_SUCCESS)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_add(): could not insert ldap_entries record\n",0,0,0);
  backsql_PrintErrors(bi->db_env,dbh,sth,rc);
  //execute delete_proc to delete data added !!!
  SQLFreeStmt(sth,SQL_DROP);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }
 SQLFreeStmt(sth,SQL_DROP);
 send_ldap_result(conn,op,LDAP_SUCCESS,"",NULL,NULL,NULL);
 return 0;
}

int backsql_delete(BackendDB *be,Connection *conn,Operation *op,
	char *dn,char *ndn)
{
 backsql_info *bi=(backsql_info*)be->be_private;
 SQLHDBC dbh;
 SQLHSTMT sth;
 RETCODE rc;
 backsql_oc_map_rec *oc=NULL;
 backsql_entryID e_id,*res;

 dn=dn_validate(dn);
 Debug(LDAP_DEBUG_TRACE,"==>backsql_delete(): deleting entry '%s'\n",dn,0,0);
 dbh=backsql_get_db_conn(be,conn);
 if (!dbh)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_delete(): could not get connection handle - exiting\n",0,0,0);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }
 res=backsql_dn2id(&e_id,dbh,dn);
 if (res==NULL)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_delete(): could not lookup entry id\n",0,0,0);
  send_ldap_result(conn,op,LDAP_NO_SUCH_OBJECT,"",NULL,NULL,NULL);
  return 1;
 }

 oc=backsql_oc_with_id(bi,e_id.oc_id);
 if (oc==NULL)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_delete(): cannot determine objectclass of entry -- aborting\n",0,0,0);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }
 if (oc->delete_proc == NULL)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_delete(): delete procedure is not defined for this objectclass - aborting\n",0,0,0);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }

 SQLAllocStmt(dbh, &sth);
 SQLBindParameter(sth,1,SQL_PARAM_INPUT,SQL_C_ULONG,SQL_INTEGER,0,0,&e_id.keyval,0,0);
 //SQLBindParameter(sth,2,SQL_PARAM_OUTPUT,SQL_C_SLONG,SQL_INTEGER,0,0,&retcode,0,0);

 Debug(LDAP_DEBUG_TRACE,"backsql_delete(): executing '%s'\n",oc->delete_proc,0,0);
 rc=SQLExecDirect(sth,oc->delete_proc,SQL_NTS);
 if (rc != SQL_SUCCESS)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_delete(): delete_proc execution failed\n",0,0,0);
  backsql_PrintErrors(bi->db_env,dbh,sth,rc);
  SQLFreeStmt(sth,SQL_DROP);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }
 SQLFreeStmt(sth,SQL_RESET_PARAMS);

 SQLBindParameter(sth,1,SQL_PARAM_INPUT,SQL_C_ULONG,SQL_INTEGER,0,0,&e_id.id,0,0);
 rc=SQLExecDirect(sth,bi->delentry_query,SQL_NTS);
 if (rc != SQL_SUCCESS)
 {
  Debug(LDAP_DEBUG_TRACE,"backsql_delete(): failed to delete record from ldap_entries\n",0,0,0);
  backsql_PrintErrors(bi->db_env,dbh,sth,rc);
  SQLFreeStmt(sth,SQL_DROP);
  send_ldap_result(conn,op,LDAP_OTHER,"","SQL-backend error",NULL,NULL);
  return 1;
 }
 SQLFreeStmt(sth,SQL_DROP);

 send_ldap_result(conn,op,LDAP_SUCCESS,"",NULL,NULL,NULL);
 Debug(LDAP_DEBUG_TRACE,"<==backsql_delete()\n",0,0,0);
 return 0;
}

#endif /* SLAPD_SQL */