#include <grass/dbmi.h>
#include "odbc.h"
#include "globals.h"
#include "proto.h"

int db__driver_fetch(cn, position, more)
     dbCursor *cn;
     int position;
     int *more;
{
    cursor *c;
    dbToken token;
    dbTable *table;
    dbColumn *column;
    dbValue *value;
    int col, ncols;
    SQLINTEGER len;
    int htype, sqltype, ctype;
    SQLRETURN ret;
    DATE_STRUCT date;
    TIME_STRUCT time;
    TIMESTAMP_STRUCT timestamp;

    /* get cursor token */
    token = db_get_cursor_token(cn);

    /* get the cursor by its token */
    if (!(c = (cursor *) db_find_token(token))) {
	db_error("cursor not found");
	return DB_FAILED;
    }

    /* fetch on position */
    switch (position) {
    case DB_NEXT:
	ret = SQLFetchScroll(c->stmt, SQL_FETCH_NEXT, 0);
	break;
    case DB_CURRENT:
	ret = SQLFetchScroll(c->stmt, SQL_FETCH_RELATIVE, 0);
	break;
    case DB_PREVIOUS:
	ret = SQLFetchScroll(c->stmt, SQL_FETCH_PRIOR, 0);
	break;
    case DB_FIRST:
	ret = SQLFetchScroll(c->stmt, SQL_FETCH_FIRST, 0);
	break;
    case DB_LAST:
	ret = SQLFetchScroll(c->stmt, SQL_FETCH_LAST, 0);
	break;
    default:
	return DB_FAILED;
    };

    if (ret == SQL_NO_DATA) {
	*more = 0;
	return DB_OK;
    }
    if (!SQL_SUCCEEDED(ret)) {
	return DB_FAILED;
    }
    *more = 1;

    /* get the data out of the descriptor into the table */
    table = db_get_cursor_table(cn);
    ncols = db_get_table_number_of_columns(table);
    for (col = 1; col <= ncols; col++) {
	column = db_get_table_column(table, col - 1);
	value = db_get_column_value(column);
	db_free_string(&value->s);

	/* Is null? */
	SQLGetData(c->stmt, col, SQL_C_CHAR, NULL, 0, &len);
	if (len == SQL_NULL_DATA) {
	    value->isNull = 1;
	    continue;
	}
	else
	    value->isNull = 0;

	sqltype = db_get_column_sqltype(column);
	ctype = db_sqltype_to_Ctype(sqltype);
	htype = db_get_column_host_type(column);

	switch (ctype) {
	case DB_C_TYPE_STRING:
	    if (htype == SQL_CHAR) {
		len = db_get_column_length(column);
		db_enlarge_string(&value->s, len + 1);
		ret =
		    SQLGetData(c->stmt, col, SQL_C_CHAR, value->s.string,
			       len + 1, NULL);
	    }
	    else if (htype == SQL_VARCHAR) {
		ret = SQLGetData(c->stmt, col, SQL_C_CHAR, NULL, 0, &len);
		db_enlarge_string(&value->s, len + 1);
		ret =
		    SQLGetData(c->stmt, col, SQL_C_CHAR, value->s.string,
			       len + 1, NULL);
	    }
	    else {		/* now the same as SQL_VARCHAR, could differ for other htype ?  */
		ret = SQLGetData(c->stmt, col, SQL_C_CHAR, NULL, 0, &len);
		db_enlarge_string(&value->s, len + 1);
		ret =
		    SQLGetData(c->stmt, col, SQL_C_CHAR, value->s.string,
			       len + 1, NULL);
	    }
	    break;
	case DB_C_TYPE_INT:
	    ret =
		SQLGetData(c->stmt, col, SQL_C_LONG, &value->i,
			   sizeof(value->i), NULL);
	    break;
	case DB_C_TYPE_DOUBLE:
	    ret =
		SQLGetData(c->stmt, col, SQL_C_DOUBLE, &value->d,
			   sizeof(value->d), NULL);
	    break;

	case DB_C_TYPE_DATETIME:
	    switch (sqltype) {
	    case DB_SQL_TYPE_DATE:
		ret =
		    SQLGetData(c->stmt, col, SQL_C_TYPE_DATE, &date,
			       sizeof(date), NULL);
		value->t.year = date.year;
		value->t.month = date.month;
		value->t.day = date.day;
		value->t.hour = 0;
		value->t.minute = 0;
		value->t.seconds = 0.0;
		break;
	    case DB_SQL_TYPE_TIME:
		ret =
		    SQLGetData(c->stmt, col, SQL_C_TYPE_TIME, &time,
			       sizeof(time), NULL);
		value->t.year = 0;
		value->t.month = 0;
		value->t.day = 0;
		value->t.hour = time.hour;
		value->t.minute = time.minute;
		value->t.seconds = time.second;
		break;
	    case DB_SQL_TYPE_TIMESTAMP:
		ret =
		    SQLGetData(c->stmt, col, SQL_C_TYPE_TIMESTAMP, &timestamp,
			       sizeof(timestamp), NULL);
		value->t.year = timestamp.year;
		value->t.month = timestamp.month;
		value->t.day = timestamp.day;
		value->t.hour = timestamp.hour;
		value->t.minute = timestamp.minute;
		value->t.seconds = timestamp.second;
		break;
		/*  
		   case DB_SQL_TYPE_INTERVAL:
		   break;
		   default: 
		   break;
		 */
	    }

	default:
	    len = db_get_column_length(column);
	    db_enlarge_string(&value->s, len + 1);
	    ret =
		SQLGetData(c->stmt, col, SQL_C_CHAR, value->s.string, len + 1,
			   NULL);
	    break;
	}
    }
    return DB_OK;
}

int db__driver_get_num_rows(cn)
     dbCursor *cn;
{
    cursor *c;
    dbToken token;

    /* get cursor token */
    token = db_get_cursor_token(cn);

    /* get the cursor by its token */
    if (!(c = (cursor *) db_find_token(token))) {
	db_error("cursor not found");
	return DB_FAILED;
    }

    return (c->nrows);
}