/*
	Build TLMP protocol analyser component from a proto file.
	A proto file contains one line per protocol command. The line start with the protocol verb
	followed by the name of all parameters.

	Special case:
	The parameter * means that a variable number of words is accepted. They will be passed
	to the application as a string (const char *) called extra.

	The parameter [ means that the following parameters are optional. They will be empty
	if missing.
*/
#include <stdio.h>
#include <ctype.h>
#include <tlmplib.h>
#include <misc.h>
#include <string.h>
#include <string>
#include <vector>

using namespace std;

struct PROTO{
	string command;		// Name of the protocol command
	vector<string> prms;	// Name of the parameters
	unsigned min_prms;	// Minimum number of parameters
	unsigned max_prms;	// Maximum number
	bool any_extra;		// There is not max_prm. Anything after min_prm
				// goes in the extra
};


static void printargs (bool first_done, vector<string> &args)
{
	for (unsigned i=0; i<args.size(); i++){
		if (first_done) printf (",");
		const char *argname = args[i].c_str();
		const char *argend = argname + strlen(argname);
		// Find the name of the variable, so go from the back until we get an invalid character for
		// a C++ name
		while (argend > argname){
			char prev = argend[-1];
			if (isalpha(prev) || isdigit(prev) || prev == '_'){
				argend--;
			}else{
				break;
			}
		}
		printf ("%s",argend);
		first_done = true;
	}
}

/*
	Convert the name into a valid C++ name.
	Change - for _
*/
static void buildp_cnvtoc (const char *func, string &funcc)
{
	while (*func != '\0'){
		if (*func == '-'){
			funcc += '_';
		}else{
			funcc += *func;
		}
		func++;
	}
}

<mod>
int main (int argc, char *argv[])
{
	glocal int ret = -1;
	glocal const char *name = NULL;
	glocal vector<string> args;
	glocal bool secretmode = false;
	glocal bool gomode = false;
	glocal.ret = <call tlmpprogram>(argc,argv);
	<f init>
		setproginfo ("buildp",VERSION,"Build TLMP protocol components");
		setarg ('n',"name","Name of the component",glocal.name,true);
		setarg ('a',"arg","Extra arg passed to the component",glocal.args,false);
		setarg ('s',"secretmode","Extra processing to validate message signature",glocal.secretmode,false);
		setarg ('g',"gomode","Send a Go after every command",glocal.gomode,false);
	</f>
	<f main>
		glocal vector<PROTO> lines;
		<call loadfile>(argv[0],true);
		<f oneline>
			line = str_skip(line);
			if (line[0] != '\0' && line[0] != '#'){
				vector<string> tb;
				int n = str_splitline (line,' ',tb);
				if (n > 0){
					PROTO p;
					p.command = tb[0];
					p.min_prms = tb.size()-1;
					p.max_prms = p.min_prms;
					p.any_extra = false;
					for (unsigned i=1; i<tb.size(); i++){
						const char *tbi = tb[i].c_str();
						if (strcmp(tbi,"*")==0){
							p.min_prms = i-1;
							p.max_prms = p.min_prms;
							p.any_extra = true;
							if (i!=tb.size()-1){
								tlmp_error ("buildp: file %s, line %d: * must be the last parameter\n"
									,info.filename,noline);
								break;
							}
						}else if (strcmp(tbi,"[")==0){
							p.min_prms = i-1;
							p.max_prms--;
						}else{
							p.prms.push_back(tbi);
						}
					}
					glocal.lines.push_back(p);
				}
			}
			return 0;
		</f>
		</call>
		// Print a comment header explaining how the file was produced
		printf ("// Generated by the ./buildp utility with the following command\n");
		printf ("// ./buildp");
		if (glocal.gomode) printf (" --gomode");
		if (glocal.secretmode) printf (" --secretmode");
		for (unsigned i=0; i<glocal.args.size(); i++) printf (" --arg \"%s\"",glocal.args[i].c_str());
		printf (" --name %s %s",glocal.name,argv[0]);
		printf ("\n");
		// Now the component definition
		{
			printf ("#define _TLMP_%s\n",glocal.name);
			printf ("struct _F_%s{\n",glocal.name);
			for (unsigned i=0; i<glocal.lines.size(); i++){
				PROTO &p = glocal.lines[i];
				const char *func = p.command.c_str();
				string commandc;
				buildp_cnvtoc (func,commandc);
				const char *funcc = commandc.c_str();
				vector<string> &tb = p.prms;
				printf ("\t#define _F_%s_%s(x) void x %s(bool &endserver, bool &endclient, const char *line",glocal.name,funcc,funcc);
				for (unsigned i=0; i<tb.size(); i++){
					printf (",const char *%s",tb[i].c_str());
				}
				if (p.any_extra) printf (",const char *extra");
				for (unsigned i=0; i<glocal.args.size(); i++){
					printf (",%s",glocal.args[i].c_str());
				}
				printf (")\n");
				printf ("\tvirtual _F_%s_%s( )=0;\n",glocal.name,funcc);
			}
			printf ("\t#define _F_%s_invalid(x) void x invalid(bool &endserver, bool &endclient, const char *line",glocal.name);
			for (unsigned i=0; i<glocal.args.size(); i++){
				printf (",%s",glocal.args[i].c_str());
			}
			printf (")\n");
			printf ("\tvirtual _F_%s_invalid( )=0;\n",glocal.name);
			printf ("};\n");
		}
		{
			// We put a _ in front of the standard args to avoid conflict with the
			// user supplied argument names.
			printf ("static void %s(_F_%s &_c, _F_TCPSERVER_V1 *_cc, const char *_line, bool &_endserver, bool &_endclient",glocal.name,glocal.name);
			if (glocal.secretmode){
				printf (",const string &_secret, time_t &_intruder");
			}
			for (unsigned i=0; i<glocal.args.size(); i++){
				printf (",%s",glocal.args[i].c_str());
			}
			printf (")\n");
			printf ("{\n");
			printf ("\tstring newline;\n");
			const char *elsestr = "";
			if (glocal.secretmode){
				printf ("\tif (_intruder){\n");
				printf ("\t\t// Ignore everything from an intruder\n");
				elsestr = "}else ";
			}
			printf ("\t%sif (!fdpass_protocheck(_line,newline)){\n",elsestr);
			printf ("\t\t_c.invalid (_endserver,_endclient,newline.c_str()");
			printargs(true,glocal.args);
			printf (");\n");
			printf ("\t}else{\n");
			printf ("\t\tvector<string> _tb;\n");
			printf ("\t\tint _n = str_splitline(_line,' ',_tb);\n");
			printf ("\t\tif(_n==0){\n");
			printf ("\t\t\t_c.invalid (_endserver,_endclient,_line");
			printargs(true,glocal.args);
			printf (");\n");
			if (glocal.secretmode){
				printf ("\t\t}else if (fdpass_valid_secret(_secret,_line,_tb,_n)==-1){\n");
				printf ("\t\t\t_intruder = time(NULL);\n");
			}
			printf ("\t\t}else{\n");
			printf ("\t\t\tconst char *_func = _tb[0].c_str();\n");
			for (unsigned i=0; i<glocal.lines.size(); i++){
				PROTO &p = glocal.lines[i];
				const char *func = p.command.c_str();
				string commandc;
				buildp_cnvtoc (func,commandc);
				const char *funcc = commandc.c_str();
				if (i==0){
					printf ("\t\t\tif");
				}else{
					printf ("\t\t\t}else if");
				}
				if (p.min_prms == p.max_prms && !p.any_extra){
					printf ("(strcmp(_func,\"%s\")==0 && _n==%u){\n",func,p.max_prms+1);
				}else if (p.min_prms < p.max_prms && !p.any_extra){
					printf ("(strcmp(_func,\"%s\")==0 && _n>=%u && _n<=%u){\n",func
						,p.min_prms+1,p.max_prms+1);
				}else if (p.any_extra){
					printf ("(strcmp(_func,\"%s\")==0 && _n>=%u){\n",func,p.max_prms+1);
				}
				for (unsigned j=p.min_prms; j<p.max_prms; j++){
					printf ("\t\t\t\t_tb.push_back(\"\");\n");
				}
				if (p.any_extra){
					printf ("\t\t\t\tstring _extra;\n");
					printf ("\t\t\t\tfor (int j=%u; j<_n; j++){\n",p.min_prms+1);
					printf ("\t\t\t\t\tif(j>%u) _extra += \" \";\n",p.min_prms+1);
					printf ("\t\t\t\t\t_extra += _tb[j];\n");
					printf ("\t\t\t\t}\n");
				}
				printf ("\t\t\t\t_c.%s(_endserver,_endclient,_line",funcc);
				for (unsigned i=1; i<=p.max_prms; i++){
					printf (",_tb[%u].c_str()",i);
				}
				if (p.any_extra){
					printf (",_extra.c_str()");
				}
				printargs (true,glocal.args);
				printf (");\n");
				if (glocal.gomode) printf ("\t\t\t\t_cc->send(\"Ok\\n\");\n");
			}
			printf ("\t\t\t}else{\n");
			printf ("\t\t\t\t_c.invalid (_endserver,_endclient,_line");
			printargs(true,glocal.args);
			printf (");\n");
			printf ("\t\t\t}\n");
			printf ("\t\t}\n");
			printf ("\t}\n");
			printf ("}\n");
		}
		return 0;
	</f>
	</call>
	return glocal.ret;
}
</mod>