/*
 *	 Copyright 1999, Dmitry Kovalev <mit@openldap.org>, 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 "ac/string.h"
#include "slap.h"
#include "lber_pvt.h"
#include "ldap_pvt.h"
#include "back-sql.h"
#include "sql-wrap.h"
#include "schema-map.h"
#include "util.h"

/*
 * Uses the pointer to the ObjectClass structure
 */
static int
backsql_cmp_oc( backsql_oc_map_rec *m1, backsql_oc_map_rec *m2 )
{
	return ( m1->oc < m2->oc ? -1 : ( m1->oc > m2->oc ? 1 : 0 ) );
}

static int
backsql_cmp_oc_id( backsql_oc_map_rec *m1, backsql_oc_map_rec *m2 )
{
	return ( m1->id < m2->id ? -1 : ( m1->id > m2->id ? 1 : 0 ) );
}

/*
 * Uses the pointer to the AttributeDescription structure
 */
static int
backsql_cmp_attr( backsql_at_map_rec *m1, backsql_at_map_rec *m2 )
{
	return ( m1->ad < m2->ad ? -1 : ( m1->ad > m2->ad ? 1 : 0 ) );
}

static int
backsql_make_attr_query( 
	backsql_oc_map_rec 	*oc_map,
	backsql_at_map_rec 	*at_map )
{
	struct berval	tmps = BER_BVNULL;
	ber_len_t	tmpslen = 0;

	backsql_strfcat( &tmps, &tmpslen, "lblblblbcbl", 
			(ber_len_t)sizeof( "SELECT " ) - 1, "SELECT ", 
			&at_map->sel_expr, 
			(ber_len_t)sizeof( " AS " ) - 1, " AS ", 
			&at_map->ad->ad_cname,
			(ber_len_t)sizeof( " FROM " ) - 1, " FROM ", 
			&at_map->from_tbls, 
			(ber_len_t)sizeof( " WHERE " ) - 1, " WHERE ", 
			&oc_map->keytbl,
			'.', 
			&oc_map->keycol,
			(ber_len_t)sizeof( "=?" ) - 1, "=?" );

	if ( at_map->join_where.bv_val != NULL ) {
		backsql_strfcat( &tmps, &tmpslen, "lb",
				(ber_len_t)sizeof( " AND ") - 1, " AND ", 
				&at_map->join_where );
	}

	at_map->query = tmps.bv_val;
	
	return 0;
}

static int
backsql_add_sysmaps( backsql_oc_map_rec *oc_map )
{
	backsql_at_map_rec	*at_map;
	char			s[ 30 ];
	ber_len_t		len, slen;
	

	snprintf( s, sizeof( s ), "%ld", oc_map->id );
	slen = strlen( s );

	at_map = (backsql_at_map_rec *)ch_calloc(1, 
			sizeof( backsql_at_map_rec ) );
	at_map->ad = slap_schema.si_ad_objectClass;
	ber_str2bv( "ldap_entry_objclasses.oc_name", 0, 1, &at_map->sel_expr );
	ber_str2bv( "ldap_entry_objclasses,ldap_entries", 0, 1, 
			&at_map->from_tbls );
	len = at_map->from_tbls.bv_len + 1;
	backsql_merge_from_clause( &at_map->from_tbls, &len, &oc_map->keytbl );

	len = 0;
	at_map->join_where.bv_val = NULL;
	at_map->join_where.bv_len = 0;
	backsql_strfcat( &at_map->join_where, &len, "lbcbll",
			(ber_len_t)sizeof( "ldap_entries.id=ldap_entry_objclasses.entry_id and ldap_entries.keyval=" ) - 1,
				"ldap_entries.id=ldap_entry_objclasses.entry_id and ldap_entries.keyval=",
			&oc_map->keytbl, 
			'.', 
			&oc_map->keycol,
			(ber_len_t)sizeof( " and ldap_entries.oc_map_id=" ) - 1, 
				" and ldap_entries.oc_map_id=", 
			slen, s );

	at_map->add_proc = NULL;
	at_map->delete_proc = NULL;
	at_map->param_order = 0;
	at_map->expect_return = 0;
	backsql_make_attr_query( oc_map, at_map );
	avl_insert( &oc_map->attrs, at_map, 
			(AVL_CMP)backsql_cmp_attr, NULL );

	at_map = (backsql_at_map_rec *)ch_calloc( 1, 
			sizeof( backsql_at_map_rec ) );
	at_map->ad = slap_schema.si_ad_ref;
	ber_str2bv( "ldap_referrals.url", 0, 1, &at_map->sel_expr );
	ber_str2bv( "ldap_referrals,ldap_entries", 0, 1, &at_map->from_tbls );
	len = at_map->from_tbls.bv_len + 1;
	backsql_merge_from_clause( &at_map->from_tbls, &len, &oc_map->keytbl );

	len = 0;
	at_map->join_where.bv_val = NULL;
	at_map->join_where.bv_len = 0;
	backsql_strfcat( &at_map->join_where, &len, "lbcbll",
			(ber_len_t)sizeof( "ldap_entries.id=ldap_referrals.entry_id and ldap_entries.keyval=" ) - 1,
				"ldap_entries.id=ldap_referrals.entry_id and ldap_entries.keyval=",
			&oc_map->keytbl, 
			'.', 
			&oc_map->keycol,
			(ber_len_t)sizeof( " and ldap_entries.oc_map_id=" ) - 1, 
				" and ldap_entries.oc_map_id=", 
			slen, s );

	at_map->add_proc = NULL;
	at_map->delete_proc = NULL;
	at_map->param_order = 0;
	at_map->expect_return = 0;
	backsql_make_attr_query( oc_map, at_map );
	avl_insert( &oc_map->attrs, at_map, 
			(AVL_CMP)backsql_cmp_attr, NULL );

	return 1;
}

int
backsql_load_schema_map( backsql_info *si, SQLHDBC dbh )
{
	SQLHSTMT 		oc_sth, at_sth;
	RETCODE			rc;
	BACKSQL_ROW_NTS		oc_row, at_row;
	unsigned long		oc_id;
	backsql_oc_map_rec	*oc_map;
	backsql_at_map_rec	*at_map;

	Debug( LDAP_DEBUG_TRACE, "==>load_schema_map()\n", 0, 0, 0 );

	/* 
	 * TimesTen : See if the ldap_entries.dn_ru field exists in the schema
	 */
	if ( !BACKSQL_DONTCHECK_LDAPINFO_DN_RU( si ) ) {
		rc = backsql_Prepare( dbh, &oc_sth, 
				backsql_check_dn_ru_query, 0 );
		if ( rc == SQL_SUCCESS ) {
			/* Yes, the field exists */
			si->bsql_flags |= BSQLF_HAS_LDAPINFO_DN_RU;
   			Debug( LDAP_DEBUG_TRACE, "ldapinfo.dn_ru field exists "
				"in the schema\n", 0, 0, 0 );
		} else {
			/* No such field exists */
			si->bsql_flags &= ~BSQLF_HAS_LDAPINFO_DN_RU;
		}

		SQLFreeStmt( oc_sth, SQL_DROP );
	}


	rc = backsql_Prepare( dbh, &oc_sth, si->oc_query, 0 );
	if ( rc != SQL_SUCCESS ) {
		Debug( LDAP_DEBUG_TRACE, "load_schema_map(): "
			"error preparing oc_query: '%s'\n", 
			si->oc_query, 0, 0 );
		backsql_PrintErrors( si->db_env, dbh, oc_sth, rc );
		return LDAP_OTHER;
	}
	Debug( LDAP_DEBUG_TRACE, "load_schema_map(): at_query '%s'\n", 
			si->at_query, 0, 0 );

	rc = backsql_Prepare( dbh, &at_sth, si->at_query, 0 );
	if ( rc != SQL_SUCCESS ) {
		Debug( LDAP_DEBUG_TRACE, "load_schema_map(): "
			"error preparing at_query: '%s'\n", 
			si->at_query, 0, 0 );
		backsql_PrintErrors( si->db_env, dbh, at_sth, rc );
		return LDAP_OTHER;
	}

	rc = backsql_BindParamID( at_sth, 1, &oc_id );
	if ( rc != SQL_SUCCESS ) {
		Debug( LDAP_DEBUG_TRACE, "load_schema_map(): "
			"error binding param for at_query: \n", 0, 0, 0 );
		backsql_PrintErrors( si->db_env, dbh, at_sth, rc );
		return LDAP_OTHER;
	}

	rc = SQLExecute( oc_sth );
	if ( rc != SQL_SUCCESS ) {
		Debug( LDAP_DEBUG_TRACE, "load_schema_map(): "
			"error executing oc_query: \n", 0, 0, 0 );
		backsql_PrintErrors( si->db_env, dbh, oc_sth, rc );
		return LDAP_OTHER;
	}

	backsql_BindRowAsStrings( oc_sth, &oc_row );
	rc = SQLFetch( oc_sth );
	for ( ; BACKSQL_SUCCESS( rc ); rc = SQLFetch( oc_sth ) ) {
		int	colnum;

		oc_map = (backsql_oc_map_rec *)ch_calloc( 1,
				sizeof( backsql_oc_map_rec ) );

		oc_map->id = strtol( oc_row.cols[ 0 ], NULL, 0 );

		oc_map->oc = oc_find( oc_row.cols[ 1 ] );
		if ( oc_map->oc == NULL ) {
			Debug( LDAP_DEBUG_TRACE, "load_schema_map(): "
				"objectClass '%s' is not defined in schema\n", 
				oc_row.cols[ 1 ], 0, 0 );
			return LDAP_OTHER;	/* undefined objectClass ? */
		}
		
		ber_str2bv( oc_row.cols[ 2 ], 0, 1, &oc_map->keytbl );
		ber_str2bv( oc_row.cols[ 3 ], 0, 1, &oc_map->keycol );
		oc_map->create_proc = ( oc_row.value_len[ 4 ] < 0 ) ? NULL 
			: ch_strdup( oc_row.cols[ 4 ] );

		colnum = 5;
		if ( BACKSQL_CREATE_NEEDS_SELECT( si ) ) {
			colnum = 6;
			oc_map->create_keyval = ( oc_row.value_len[ 5 ] < 0 ) 
				? NULL : ch_strdup( oc_row.cols[ 5 ] );
		}
		oc_map->delete_proc = ( oc_row.value_len[ colnum ] < 0 ) ? NULL 
			: ch_strdup( oc_row.cols[ colnum ] );
		oc_map->expect_return = strtol( oc_row.cols[ colnum + 1 ], 
				NULL, 0 );

		/*
		 * FIXME: first attempt to check for offending
		 * instructions in {create|delete}_proc
		 */

		oc_map->attrs = NULL;
		avl_insert( &si->oc_by_oc, oc_map,
				(AVL_CMP)backsql_cmp_oc, NULL );
		avl_insert( &si->oc_by_id, oc_map,
				(AVL_CMP)backsql_cmp_oc_id, NULL );
		oc_id = oc_map->id;
		Debug( LDAP_DEBUG_TRACE, "load_schema_map(): "
			"objectClass '%s': keytbl='%s' keycol='%s'\n",
			BACKSQL_OC_NAME( oc_map ),
			oc_map->keytbl.bv_val, oc_map->keycol.bv_val );
		if ( oc_map->create_proc ) {
			Debug( LDAP_DEBUG_TRACE, "create_proc='%s'\n",
				oc_map->create_proc, 0, 0 );
		}
		if ( oc_map->create_keyval ) {
			Debug( LDAP_DEBUG_TRACE, "create_keyval='%s'\n",
				oc_map->create_keyval, 0, 0 );
		}
		if ( oc_map->delete_proc ) {
			Debug( LDAP_DEBUG_TRACE, "delete_proc='%s'\n", 
				oc_map->delete_proc, 0, 0 );
		}
		Debug( LDAP_DEBUG_TRACE, "expect_return: "
			"add=%d, del=%d; attributes:\n",
			BACKSQL_IS_ADD( oc_map->expect_return ), 
			BACKSQL_IS_DEL( oc_map->expect_return ), 0 );

		Debug( LDAP_DEBUG_TRACE, "load_schema_map(): "
			"autoadding 'objectClass' and 'ref' mappings\n",
			0, 0, 0 );
		backsql_add_sysmaps( oc_map );
		rc = SQLExecute( at_sth );
		if ( rc != SQL_SUCCESS ) {
			Debug( LDAP_DEBUG_TRACE, "load_schema_map(): "
				"error executing at_query: \n", 0, 0, 0 );
			backsql_PrintErrors( SQL_NULL_HENV, dbh, at_sth, rc );
			return LDAP_OTHER;
		}

		backsql_BindRowAsStrings( at_sth, &at_row );
		rc = SQLFetch( at_sth );
		for ( ; BACKSQL_SUCCESS(rc); rc = SQLFetch( at_sth ) ) {
			const char	*text = NULL;
			struct berval	bv;
			ber_len_t	tmpslen;

			Debug( LDAP_DEBUG_TRACE, "********'%s'\n",
				at_row.cols[ 0 ], 0, 0 );
			Debug( LDAP_DEBUG_TRACE, 
				"name='%s',sel_expr='%s' from='%s'",
				at_row.cols[ 0 ], at_row.cols[ 1 ],
				at_row.cols[ 2 ] );
			Debug( LDAP_DEBUG_TRACE, 
				"join_where='%s',add_proc='%s'",
				at_row.cols[ 3 ], at_row.cols[ 4 ], 0 );
			Debug( LDAP_DEBUG_TRACE, "delete_proc='%s'\n",
					at_row.cols[ 5 ], 0, 0 );
			/* TimesTen */
			Debug( LDAP_DEBUG_TRACE, "sel_expr_u='%s'\n",
					at_row.cols[ 8 ], 0, 0 );
			at_map = (backsql_at_map_rec *)ch_calloc( 1,
					sizeof( backsql_at_map_rec ) );
			rc = slap_str2ad( at_row.cols[ 0 ], 
					&at_map->ad, &text );
			if ( rc != LDAP_SUCCESS ) {
				Debug( LDAP_DEBUG_TRACE, "load_schema_map(): "
					"attribute '%s' for objectClass '%s' "
					"is not defined in schema: %s\n", 
					at_row.cols[ 0 ],
					BACKSQL_OC_NAME( oc_map ), text );
				return LDAP_CONSTRAINT_VIOLATION;
			}
				
			ber_str2bv( at_row.cols[ 1 ], 0, 1, &at_map->sel_expr );
			if ( at_row.value_len[ 8 ] < 0 ) {
				at_map->sel_expr_u.bv_val = NULL;
				at_map->sel_expr_u.bv_len = 0;
			} else {
				ber_str2bv( at_row.cols[ 8 ], 0, 1, 
						&at_map->sel_expr_u );
			}
			tmpslen = 0;
			ber_str2bv( at_row.cols[ 2 ], 0, 0, &bv );
			backsql_merge_from_clause( &at_map->from_tbls, 
					&tmpslen, &bv );
			if ( at_row.value_len[ 3 ] < 0 ) {
				at_map->join_where.bv_val = NULL;
				at_map->join_where.bv_len = 0;
			} else {
				ber_str2bv( at_row.cols[ 3 ], 0, 1, 
						&at_map->join_where );
			}
			at_map->add_proc = NULL;
			if ( at_row.value_len[ 4 ] > 0 ) {
				at_map->add_proc = ch_strdup( at_row.cols[4] );
			}
			at_map->delete_proc = NULL;
			if ( at_row.value_len[ 5 ] > 0 ) {
				at_map->delete_proc
					= ch_strdup( at_row.cols[ 5 ] );
			}
			at_map->param_order = strtol( at_row.cols[ 6 ], 
					NULL, 0 );
			at_map->expect_return = strtol( at_row.cols[ 7 ],
					NULL, 0 );
			backsql_make_attr_query( oc_map, at_map );
			Debug( LDAP_DEBUG_TRACE, "load_schema_map(): "
				"preconstructed query '%s'\n",
				at_map->query, 0, 0 );
			avl_insert( &oc_map->attrs, at_map, 
					(AVL_CMP)backsql_cmp_attr, NULL );
		}
		backsql_FreeRow( &at_row );
		SQLFreeStmt( at_sth, SQL_CLOSE );
	}
	backsql_FreeRow( &oc_row );
	SQLFreeStmt( at_sth, SQL_DROP );
	SQLFreeStmt( oc_sth, SQL_DROP );
	si->bsql_flags |= BSQLF_SCHEMA_LOADED;
	Debug( LDAP_DEBUG_TRACE, "<==load_schema_map()\n", 0, 0, 0 );
	return LDAP_SUCCESS;
}

backsql_oc_map_rec *
backsql_oc2oc( backsql_info *si, ObjectClass *oc )
{
	backsql_oc_map_rec	tmp, *res;

#ifdef BACKSQL_TRACE
	Debug( LDAP_DEBUG_TRACE, "==>backsql_oc2oc(): "
		"searching for objectclass with name='%s'\n",
		objclass, 0, 0 );
#endif /* BACKSQL_TRACE */

	tmp.oc = oc;
	res = (backsql_oc_map_rec *)avl_find( si->oc_by_oc, &tmp,
			(AVL_CMP)backsql_cmp_oc );
#ifdef BACKSQL_TRACE
	if ( res != NULL ) {
		Debug( LDAP_DEBUG_TRACE, "<==backsql_oc2oc(): "
			"found name='%s', id=%d\n", 
			BACKSQL_OC_NAME( res ), res->id, 0 );
	} else {
		Debug( LDAP_DEBUG_TRACE, "<==backsql_oc2oc(): "
			"not found\n", 0, 0, 0 );
	}
#endif /* BACKSQL_TRACE */
 
	return res;
}

backsql_oc_map_rec *
backsql_name2oc( backsql_info *si, struct berval *oc_name )
{
	backsql_oc_map_rec	tmp, *res;

#ifdef BACKSQL_TRACE
	Debug( LDAP_DEBUG_TRACE, "==>oc_with_name(): "
		"searching for objectclass with name='%s'\n",
		objclass, 0, 0 );
#endif /* BACKSQL_TRACE */

	tmp.oc = oc_bvfind( oc_name );
	if ( tmp.oc == NULL ) {
		return NULL;
	}

	res = (backsql_oc_map_rec *)avl_find( si->oc_by_oc, &tmp,
			(AVL_CMP)backsql_cmp_oc );
#ifdef BACKSQL_TRACE
	if ( res != NULL ) {
		Debug( LDAP_DEBUG_TRACE, "<==oc_with_name(): "
			"found name='%s', id=%d\n", 
			BACKSQL_OC_NAME( res ), res->id, 0 );
	} else {
		Debug( LDAP_DEBUG_TRACE, "<==oc_with_name(): "
			"not found\n", 0, 0, 0 );
	}
#endif /* BACKSQL_TRACE */
 
	return res;
}

backsql_oc_map_rec *
backsql_id2oc( backsql_info *si, unsigned long id )
{
	backsql_oc_map_rec	tmp, *res;
 
#ifdef BACKSQL_TRACE
	Debug( LDAP_DEBUG_TRACE, "==>oc_with_id(): "
		"searching for objectclass with id='%d'\n", id, 0, 0 );
#endif /* BACKSQL_TRACE */

	tmp.id = id;
	res = (backsql_oc_map_rec *)avl_find( si->oc_by_id, &tmp,
			(AVL_CMP)backsql_cmp_oc_id );

#ifdef BACKSQL_TRACE
	if ( res != NULL ) {
		Debug( LDAP_DEBUG_TRACE, "<==oc_with_name(): "
			"found name='%s', id=%d\n",
			BACKSQL_OC_NAME( res ), res->id, 0 );
	} else {
		Debug( LDAP_DEBUG_TRACE, "<==oc_with_name(): "
			"not found\n", 0, 0, 0 );
	}
#endif /* BACKSQL_TRACE */
	
	return res;
}

backsql_at_map_rec *
backsql_ad2at( backsql_oc_map_rec* objclass, AttributeDescription *ad )
{
	backsql_at_map_rec	tmp, *res;
 
#ifdef BACKSQL_TRACE
	Debug( LDAP_DEBUG_TRACE, "==>backsql_ad2at(): "
		"searching for attribute '%s' for objectclass '%s'\n",
		attr, BACKSQL_OC_NAME( objclass ), 0 );
#endif /* BACKSQL_TRACE */

	tmp.ad = ad;
	res = (backsql_at_map_rec *)avl_find( objclass->attrs, &tmp,
			(AVL_CMP)backsql_cmp_attr );

#ifdef BACKSQL_TRACE
	if ( res != NULL ) {
		Debug( LDAP_DEBUG_TRACE, "<==backsql_ad2at(): "
			"found name='%s', sel_expr='%s'\n",
			res->ad->ad_cname.bv_val, res->sel_expr.bv_val, 0 );
	} else {
		Debug( LDAP_DEBUG_TRACE, "<==backsql_ad2at(): "
			"not found\n", 0, 0, 0 );
	}
#endif /* BACKSQL_TRACE */

	return res;
}

/*
 * Deprecated
 */
backsql_at_map_rec *
backsql_name2at( backsql_oc_map_rec* objclass, struct berval *attr )
{
	backsql_at_map_rec	tmp, *res;
	const char		*text = NULL;
 
#ifdef BACKSQL_TRACE
	Debug( LDAP_DEBUG_TRACE, "==>backsql_name2at(): "
		"searching for attribute '%s' for objectclass '%s'\n",
		attr, BACKSQL_OC_NAME( objclass ), 0 );
#endif /* BACKSQL_TRACE */

	if ( slap_bv2ad( attr, &tmp.ad, &text ) != LDAP_SUCCESS ) {
		return NULL;
	}

	res = (backsql_at_map_rec *)avl_find( objclass->attrs, &tmp,
			(AVL_CMP)backsql_cmp_attr );

#ifdef BACKSQL_TRACE
	if ( res != NULL ) {
		Debug( LDAP_DEBUG_TRACE, "<==backsql_name2at(): "
			"found name='%s', sel_expr='%s'\n",
			res->name, res->sel_expr.bv_val, 0 );
	} else {
		Debug( LDAP_DEBUG_TRACE, "<==backsql_name2at(): "
			"not found\n", 0, 0, 0 );
	}
#endif /* BACKSQL_TRACE */

	return res;
}

static void
backsql_free_attr( backsql_at_map_rec *at )
{
	Debug( LDAP_DEBUG_TRACE, "==>free_attr(): '%s'\n", 
			at->ad->ad_cname.bv_val, 0, 0 );
	ch_free( at->sel_expr.bv_val );
	if ( at->from_tbls.bv_val != NULL ) {
		ch_free( at->from_tbls.bv_val );
	}
	if ( at->join_where.bv_val != NULL ) {
		ch_free( at->join_where.bv_val );
	}
	if ( at->add_proc != NULL ) {
		ch_free( at->add_proc );
	}
	if ( at->delete_proc != NULL ) {
		ch_free( at->delete_proc );
	}
	if ( at->query ) {
		ch_free( at->query );
	}

	/* TimesTen */
	if ( at->sel_expr_u.bv_val ) {
		ch_free( at->sel_expr_u.bv_val );
	}
	
	ch_free( at );

	Debug( LDAP_DEBUG_TRACE, "<==free_attr()\n", 0, 0, 0 );
}

static void
backsql_free_oc( backsql_oc_map_rec *oc )
{
	Debug( LDAP_DEBUG_TRACE, "==>free_oc(): '%s'\n", 
			BACKSQL_OC_NAME( oc ), 0, 0 );
	avl_free( oc->attrs, (AVL_FREE)backsql_free_attr );
	ch_free( oc->keytbl.bv_val );
	ch_free( oc->keycol.bv_val );
	if ( oc->create_proc != NULL ) {
		ch_free( oc->create_proc );
	}
	if ( oc->create_keyval != NULL ) {
		ch_free( oc->create_keyval );
	}
	if ( oc->delete_proc != NULL ) {
		ch_free( oc->delete_proc );
	}
	ch_free( oc );

	Debug( LDAP_DEBUG_TRACE, "<==free_oc()\n", 0, 0, 0 );
}

int
backsql_destroy_schema_map( backsql_info *si )
{
	Debug( LDAP_DEBUG_TRACE, "==>destroy_schema_map()\n", 0, 0, 0 );
	avl_free( si->oc_by_oc, NULL );
	avl_free( si->oc_by_id, (AVL_FREE)backsql_free_oc );
	Debug( LDAP_DEBUG_TRACE, "<==destroy_schema_map()\n", 0, 0, 0 );
	return 0;
}

#endif /* SLAPD_SQL */