#include <stdio.h>
#include <string.h>
#include <tlmplib.h>
#include "tlmpsql.h"

static void sqlgen_print (const char *table, const SSTRINGS &tb, int primkey)
{
	printf ("class C_%s: public NSQL_RECORD{\n",table);
	printf ("public:\n");
	for (int i=0; i<tb.getnb(); i++){
		printf ("\tNSQL_FIELD %s;\n",tb.getitem(i)->get());
	}
	printf ("\tC_%s()\n",table);
	printf ("\t:\n");
	for (int i=0; i<tb.getnb(); i++){
		const char *field = tb.getitem(i)->get();
		printf ("\t\t%s(\"%s\")%s\n",field,field,i==tb.getnb()-1 ? "" : ",");
	}
	printf ("\t{\n");
	for (int i=0; i<tb.getnb(); i++){
		const char *field = tb.getitem(i)->get();
		printf ("\t\ttb[%d] = &%s;\n",i,field);
	}
	printf ("\t\tnbfield=%d;\n",tb.getnb());
	printf ("\t\tprimkey=%d;\n",primkey);
	printf ("\t}\n");
	#if 0
		printf ("\tvoid setempty()\n");
		printf ("\t{\n");
		for (int i=0; i<tb.getnb(); i++){
			const char *field = tb.getitem(i)->get();
			printf ("\t\t%s.setempty();\n",field);
		}
		printf ("\t}\n");	
	#endif
	printf ("};\n");
}


<mod>
int main (int argc, char *argv[])
{
	<glocal>
		const char *database;
	</glocal>
	glocal.database = NULL;
	int ret = <call tlmpprogram>(argc,argv,"tlmpsql");
	<f usage>
		printf ("sqlgen_interface [--prefix ] table-name fields ...\n");
		printf ("sqlgen_interface [--prefix ] --database db tables ...\n");
	</f>
	<f onearg>
		int ret = -1;
		if (strcmp(opt,"--prefix")==0){
			printf ("#ifndef NSQL_H\n");
			printf ("\t#include <nsql.h>\n");
			printf ("#endif\n");
			ret = 0;
		}else if (strcmp(opt,"--database")==0){
			glocal.database = val;
			ret = 1;
		}
		return ret;
	</f>
	<f main>
		int ret = -1;
		if (glocal.database != NULL){
			query_setdefaultdb ("localhost",glocal.database);
			for (int i=0; i<argc; i++){
				glocal SSTRINGS tb;
				glocal int primkey = -1;
				<call sql_query>("describe %s",argv[i]);
				<f empty>
					fprintf (stderr,"No field in this table ???\n");
				</f>
				<f onerow>
					if (stricmp(row[3],"PRI")==0) glocal.primkey = rownum;
					glocal.tb.add (new SSTRING(row[0]));
				</f>
				</call>
				sqlgen_print (argv[i],glocal.tb,glocal.primkey);
			}
			ret = 0;
		}else{
			SSTRINGS tb;
			for (int i=1; i<argc; i++){
				tb.add (new SSTRING(argv[i]));
			}
			sqlgen_print (argv[0],tb,-1);
			ret = 0;
		}
		return ret;
	</f>
	</call>
	return ret;
}
</mod>