#include "WriteMainFile.h"
#include "Tools.h"
#include "MB_protos.h"
#include "MB_pragmas.h"
#include "mb.h"

#include <clib/exec_protos.h>
#include <clib/dos_protos.h>
#include <stdlib.h>
#include <string.h>

extern void Quit(void); 

#define HEADER_END_MSG 				"/* GenCodeC header end */"
#define	HEADER_END					HEADER_END_MSG "\n"
#define	LIBRARY_DECLARATION_MSG		"/* Declarations for libraries (inserted by GenCodeC) */"
#define	LIBRARY_DECLARATION			LIBRARY_DECLARATION_MSG "\n"
#define	LIBRARY_MUI_MSG				"struct Library * MUIMasterBase"
#define	LIBRARY_MUI					LIBRARY_MUI_MSG ";\n"
#define	LIBRARY_LOCALE_MSG			"struct Library * LocaleBase"
#define	LIBRARY_LOCALE				LIBRARY_LOCALE_MSG ";\n"
#define OPEN_LOCALE_LIBRARY_MSG		"if (!(LocaleBase = OpenLibrary(\"locale.library\",38)))"
#define OPEN_LOCALE_LIBRARY			"\t" OPEN_LOCALE_LIBRARY_MSG "\n"
#define CLOSE_LOCALE_LIBRARY_MSG	"CloseLibrary(LocaleBase)"
#define CLOSE_LOCALE_LIBRARY		CLOSE_LOCALE_LIBRARY_MSG ";\n"
#define	INIT_BEGIN_MSG				"/* Init() function */"
#define	INIT_BEGIN					"\n" INIT_BEGIN_MSG "\n"
#define	INIT_END_MSG				"/* GenCodeC init() end */"
#define	INIT_END					"}\n" INIT_END_MSG "\n"
#define	END_BEGIN_MSG				"/* End() function */"
#define	END_BEGIN					"\n" END_BEGIN_MSG "\n"
#define END_FUNCTION_MSG			"void end( void )"
#define END_FUNCTION				END_FUNCTION_MSG "\n{\n"
#define	END_END_MSG					"/* GenCodeC end() end */"
#define	END_END						"}\n" END_END_MSG "\n"
#define	OPEN_CATALOG_MSG			"OpenAppCatalog(NULL,NULL)"
#define	OPEN_CATALOG				OPEN_CATALOG_MSG ";\n"
#define	CLOSE_CATALOG_MSG			"CloseAppCatalog()"
#define	CLOSE_CATALOG				"\t" CLOSE_CATALOG_MSG ";\n"
#define	CASE_BEGIN_MSG				"/* Insert your code between the \"case\" statement and comment \"end of case ...\" */"
#define	CASE_BEGIN					CASE_BEGIN_MSG "\n"
#define	CASE_END_MSG				"/* End computing of IDCMP */"
#define	CASE_END					CASE_END_MSG "\n"

/****************************************************************************************************************/
/*****																										*****/
/**								 				WriteIDCMP													   **/
/*****																										*****/
/****************************************************************************************************************/

static void WriteIDCMP(FILE *file,char *adr_file,char *buf,ULONG varnb)
{
	int		i;
	char 	*name;
	char	*tmp;
	ULONG	type;
	char 	*str;
	char	*str1;
	char	ctmp;

	/* Write IDCMPs */
	for(i=0;i<varnb;i++)
	{
		MB_GetVarInfo(i,
					  MUIB_VarType,&type,
					  MUIB_VarName,&name,
					  TAG_END);
		if (!(tmp = AllocMemory(strlen(name)+19)))
		{
			fclose(file);
			FreeMemory(adr_file);
			Quit();
		}
		strcpy(tmp,"case ");
		strcat(tmp,name);
		strcat(tmp,":");
		if (type == TYPEVAR_IDENT)
		{
			if (str = strstr(buf,tmp))
			{
				/* Write old IDCMP */
				strcpy(tmp,"/* end of case ");
				strcat(tmp,name);
				strcat(tmp," */");
				if (!(str1=strstr(str,tmp)))
				{
					char *msg;

					if (!(msg = (char *)AllocMemory(60+strlen(tmp))))
					{
						fclose(file);
						FreeMemory(buf);
						Quit();
					}
					sprintf(msg,"Can't find  \"%s\"\n in MainFile generated by GenCodeC\n\n",tmp);
					DisplayMsg(msg);
					FreeMemory(msg);
					fclose(file);
					FreeMemory(buf);
					Quit();
				}
				str1+=strlen(tmp);
				ctmp=*str1;
				*str1='\0';
				fprintf(file,"\t\t\t%s\n\n",str);
				*str1=ctmp;
			}
			else 
			{
				/* Write NEW IDCMP */
				fprintf(file,"\t\t\tcase %s:\n",name);
				fprintf(file,"\t\t\t/* end of case %s */\n\n",name);
			}
		}
		FreeMemory(tmp);
	}
}

/****************************************************************************************************************/
/*****																										*****/
/**								 				WriteIncludes												   **/
/*****																										*****/
/****************************************************************************************************************/

static char *WriteIncludes(FILE *file,char *adr_file,char *HeaderFile,char *MainFile,BOOL OldMainExists) 
{
	if (!OldMainExists)
	{
		fprintf(file,"\n/* Include generated by GenCodeC */\n");
		fprintf(file,"#include \"%s\"\n\n",FilePart(HeaderFile));

		return NULL;
	}
	else
	{
		return CopyBlock(file,adr_file,adr_file,HEADER_END,LIBRARY_DECLARATION,
						HEADER_END_MSG,LIBRARY_DECLARATION_MSG,MainFile);
	}
}

/****************************************************************************************************************/
/*****																										*****/
/**								 				WriteDeclarations											   **/
/*****																										*****/
/****************************************************************************************************************/

static char *WriteDeclarations(FILE *file,char *adr_file,char *str,char *MainFile,BOOL OldMainExists,BOOL Locale) 
{
	if (!OldMainExists)
	{
		fprintf(file,LIBRARY_DECLARATION);
		fprintf(file,LIBRARY_MUI);
		if (Locale)
				fprintf(file,LIBRARY_LOCALE);

		return NULL;
	}
	else
	{
		fprintf(file,"%s",LIBRARY_DECLARATION);
		str = CopyBlock(file,adr_file,str,LIBRARY_DECLARATION,LIBRARY_MUI,
						LIBRARY_DECLARATION_MSG,LIBRARY_MUI_MSG,MainFile);
		fprintf(file,"%s",LIBRARY_MUI);
		if (Locale && (!strstr(str,LIBRARY_LOCALE)))
		{
			fprintf(file,"%s",LIBRARY_LOCALE);
			str = CopyBlock(file,adr_file,str,LIBRARY_MUI,INIT_BEGIN,
							LIBRARY_MUI_MSG,INIT_BEGIN_MSG,MainFile);
		}
		else
		{
			if (!Locale && (strstr(str,LIBRARY_LOCALE)))
				str = CopyBlock(file,adr_file,str,LIBRARY_LOCALE,INIT_BEGIN,
								LIBRARY_LOCALE_MSG,INIT_BEGIN_MSG,MainFile);
			else
				str = CopyBlock(file,adr_file,str,LIBRARY_MUI,INIT_BEGIN,
								LIBRARY_MUI_MSG,INIT_BEGIN_MSG,MainFile);
		}
		return str;
	}
}

/****************************************************************************************************************/
/*****																										*****/
/**								 			WriteInitFunction												   **/
/*****																										*****/
/****************************************************************************************************************/

static char *WriteInitFunction(FILE *file,char *adr_file,char *str,char *MainFile,BOOL OldMainExists,BOOL Locale)
{
	if (!OldMainExists)
	{
		fprintf(file,INIT_BEGIN);
		fprintf(file,"void init( void )\n");
		fprintf(file,"{\n");
		fprintf(file,"\tif (!(MUIMasterBase = OpenLibrary(MUIMASTER_NAME,MUIMASTER_VMIN)))\n");
		fprintf(file,"\t{\n");
		fprintf(file,"\t\tprintf(\"Can't Open MUIMaster Library\\n\");\n");
		fprintf(file,"\t\texit(20);\n");
		fprintf(file,"\t}\n");
		if (Locale)
		{
			fprintf(file,"%s",OPEN_LOCALE_LIBRARY);
			fprintf(file,"\t{\n");
			fprintf(file,"\t\tprintf(\"Can't Open Locale Library\\n\");\n");
			fprintf(file,"\t\tprintf(\"Built-in Language will be used !!!\\n\");\n");
			fprintf(file,"\t}\n");
			fprintf(file,"\tOpenAppCatalog(NULL,NULL);\n");
		}
		fprintf(file,INIT_END);
	
		return NULL;
	}
	else
	{
		fprintf(file,INIT_BEGIN);
		if (Locale && !(strstr(str,OPEN_LOCALE_LIBRARY)))
		{
			str = CopyBlock(file,adr_file,str,INIT_BEGIN,INIT_END,
							INIT_BEGIN_MSG,INIT_END_MSG,MainFile);
			fprintf(file,"%s",OPEN_LOCALE_LIBRARY);
			fprintf(file,"\t{\n");
			fprintf(file,"\t\tprintf(\"Can't Open Locale Library\\n\");\n");
			fprintf(file,"\t\tprintf(\"Built-in Language will be used !!!\\n\");\n");
			fprintf(file,"\t}\n");
			fprintf(file,"\tOpenAppCatalog(NULL,NULL);\n");
		}
		else
		{
			if (!Locale && (strstr(str,OPEN_LOCALE_LIBRARY)))
			{
				str = CopyBlock(file,adr_file,str,INIT_BEGIN,OPEN_LOCALE_LIBRARY,
								INIT_BEGIN_MSG,OPEN_LOCALE_LIBRARY_MSG,MainFile);
				str = CopyBlock(file,adr_file,str,OPEN_CATALOG,INIT_END,
								OPEN_CATALOG_MSG,INIT_END_MSG,MainFile);
			}
			else 
			{
				str = CopyBlock(file,adr_file,str,INIT_BEGIN,INIT_END,
								INIT_BEGIN_MSG,INIT_END_MSG,MainFile);
			}
		}
		fprintf(file,INIT_END);

		return str;
	}
}

/****************************************************************************************************************/
/*****																										*****/
/**								 			WriteEndFunction												   **/
/*****																										*****/
/****************************************************************************************************************/

static char *WriteEndFunction(FILE *file,char *adr_file,char *str,char *MainFile,BOOL OldMainExists,BOOL Locale)
{
	if (!OldMainExists)
	{
		fprintf(file,END_BEGIN);
		fprintf(file,END_FUNCTION);
		if (Locale)
		{
			fprintf(file,"%s",CLOSE_CATALOG);
			fprintf(file,"\tif (LocaleBase)\n");
			fprintf(file,"\t\t%s",CLOSE_LOCALE_LIBRARY);
		}
		fprintf(file,"\tCloseLibrary(MUIMasterBase);\n");
		fprintf(file,"\texit(20);\n");
		fprintf(file,END_END);
		fprintf(file,"\n");

		return NULL;
	}
	else
	{
		fprintf(file,END_BEGIN);
		if (Locale && !(strstr(str,CLOSE_CATALOG)))
		{
			fprintf(file,END_FUNCTION);
			fprintf(file,"%s",CLOSE_CATALOG);
			fprintf(file,"\tif (LocaleBase)\n");
			fprintf(file,"\t\t%s",CLOSE_LOCALE_LIBRARY);
			str = CopyBlock(file,adr_file,str,END_FUNCTION,END_END,
							END_FUNCTION_MSG,END_END_MSG,MainFile);
		}
		else
		{
			if (!Locale && (strstr(str,CLOSE_CATALOG)))
			{
				str = CopyBlock(file,adr_file,str,END_BEGIN,CLOSE_CATALOG,
								END_BEGIN_MSG,CLOSE_CATALOG_MSG,MainFile);
				str = CopyBlock(file,adr_file,str,CLOSE_LOCALE_LIBRARY,END_END,
								CLOSE_LOCALE_LIBRARY_MSG,END_END_MSG,MainFile);
			}
			else 
			{
				str = CopyBlock(file,adr_file,str,END_BEGIN,END_END,
								END_BEGIN_MSG,END_END_MSG,MainFile);
			}
		}
		fprintf(file,END_END);

		return str;
	}
}

/****************************************************************************************************************/
/*****																										*****/
/**								 			WriteMainFunction												   **/
/*****																										*****/
/****************************************************************************************************************/

static void WriteMainFunction(FILE *file,char *adr_file,char *str,char *MainFile,char *name,
							  BOOL OldMainExists,ULONG varnb)
{
	if (!OldMainExists)
	{
		fprintf(file,"/* Main Function inserted by GenCodeC */\n");
		fprintf(file,"int main(int argc,char **argv)\n");
		fprintf(file,"{\n");
		fprintf(file,"\tstruct Obj%s * %s = NULL;	/* Object */\n",name,name);
		fprintf(file,"\tBOOL\trunning = TRUE;\n");
		fprintf(file,"\tULONG\tsignal;\n\n");
		fprintf(file,"\t/* Program initialisation : generated by GenCodeC */\n");
		fprintf(file,"\tinit();\n\n");
		fprintf(file,"\t/* Create Object : generated by GenCodeC */\n");
		fprintf(file,"\tif (!(%s = Create%s()))\n",name,name);
		fprintf(file,"\t{\n");
		fprintf(file,"\t\tprintf(\"Can't Create %s\\n\");\n",name);
		fprintf(file,"\t\tend();\n");
		fprintf(file,"\t}\n\n");
		fprintf(file,"\twhile (running)\n");
		fprintf(file,"\t{\n");
		fprintf(file,"\t\tswitch (DoMethod(%s->%s,MUIM_Application_Input,&signal))\n",name,name);
		fprintf(file,"\t\t{\n");
		fprintf(file,"\t\t\tcase MUIV_Application_ReturnID_Quit:\n");
		fprintf(file,"\t\t\t\trunning = FALSE;\n");
		fprintf(file,"\t\t\t\tbreak;\n\n");
		fprintf(file,"\t\t\t%s",CASE_BEGIN);
	}
	else
	{
		str = CopyBlock(file,adr_file,str,END_END,CASE_BEGIN,
						END_END_MSG,CASE_BEGIN_MSG,MainFile);
		fprintf(file,CASE_BEGIN);
	}

	if (!OldMainExists)
		WriteIDCMP(file,adr_file,adr_file,varnb);
	else
		WriteIDCMP(file,adr_file,str,varnb);

	if (!OldMainExists)
	{		
		fprintf(file,"\t\t\t/* End computing of IDCMP */\n"); 
		fprintf(file,"\t\t}\n");
		fprintf(file,"\t\tif (running && signal) Wait(signal);\n");
		fprintf(file,"\t}\n");
		fprintf(file,"\tDispose%s(%s);\n",name,name);
		fprintf(file,"\tend();\n");
		fprintf(file,"}\n");
	}
	else
	{
		fprintf(file,"\t\t\t%s",CASE_END);
		CopyBlock(file,adr_file,str,CASE_END,NULL,
				  CASE_END_MSG,NULL,MainFile);
	}
}

/****************************************************************************************************************/
/*****																										*****/
/**								 			WriteMainFile													   **/
/*****																										*****/
/****************************************************************************************************************/

void WriteMainFile(char *HeaderFile,char *MainFile,char *MainHeaderText,ULONG varnb,BOOL Locale)
{
	ULONG							length;
	BPTR							TMPfile;
	BPTR							BakFile;
	char							*adr_file = NULL;
	char							*str;
	__aligned struct FileInfoBlock  Info;
	char							*buffer	 = NULL;
	char							*name = NULL;
	BOOL							OldMainExists = FALSE;
	FILE							*file;


	/* If the file already exists, we load it in memory and we save it in a .bak file*/
	if (TMPfile = Open(MainFile, MODE_OLDFILE))
	{
		OldMainExists = TRUE;
		ExamineFH(TMPfile, &Info);
		length = Info.fib_Size;
		if (!(adr_file = AllocMemory(length+1)))
		{
			Close(TMPfile);
			Quit();
		}
		Read( TMPfile, adr_file, length);
		adr_file[length] = '\0';
		Close(TMPfile);
		if (!(buffer = AllocMemory(strlen(MainFile)+4+1)))
		{
			Close(TMPfile);
			Quit();
		}
		strcpy(buffer,MainFile);
		strcat(buffer,".bak");
		if (BakFile = Open(buffer,MODE_NEWFILE))
		{
			Write(BakFile,adr_file,length);
			Close(BakFile);
		}
		FreeMemory(buffer);
	}

	MB_GetVarInfo(0, MUIB_VarName, &name, TAG_END);

	if (file = fopen(MainFile,"w"))
	{
		fprintf(file,"/* Main-Header File inserted by GenCodeC */\n");

		/* Copy Main-Header into MainFile */
		fprintf(file,"%s",MainHeaderText);

		fprintf(file,HEADER_END);
		
		str = WriteIncludes(file,adr_file,HeaderFile,MainFile,OldMainExists);

		str = WriteDeclarations(file,adr_file,str,MainFile,OldMainExists,Locale);

		str = WriteInitFunction(file,adr_file,str,MainFile,OldMainExists,Locale);

		if (OldMainExists)
			str = CopyBlock(file,adr_file,str,INIT_END,END_BEGIN,
							INIT_END_MSG,END_BEGIN_MSG,MainFile);

		str = WriteEndFunction(file,adr_file,str,MainFile,OldMainExists,Locale);

		WriteMainFunction(file,adr_file,str,MainFile,name,OldMainExists,varnb);

		fclose(file);
		FreeMemory(adr_file);
	}
	else
		DisplayMsg("Unable to create Main-File !\n");
}