/***********************************************************************/
/*                                                                     */
/*                           Objective Caml                            */
/*                                                                     */
/*  Developed by Jacob Navia.                                          */
/*  Copyright 2001 Institut National de Recherche en Informatique et   */
/*  en Automatique.  All rights reserved.  This file is distributed    */
/*  under the terms of the GNU Library General Public License, with    */
/*  the special exception on linking described in file ../LICENSE.     */
/*                                                                     */
/***********************************************************************/

/* $Id$ */

#include <stdio.h>
#include <windows.h>
#include <Richedit.h>
#include "inria.h"
#include "inriares.h"

LOGFONT CurrentFont;
int CurrentFontFamily = (FIXED_PITCH | FF_MODERN);
int CurrentFontStyle;
char CurrentFontName[64] = "Courier";
/*------------------------------------------------------------------------
 Procedure:     OpenMlFile ID:1
 Purpose:       Opens a file, either a source file (*.ml) or an *.cmo
                file.
 Input:         A buffer where the name will be stored, and its
                length
 Output:        The user's choice will be stored in the buffer.
 Errors:        None
------------------------------------------------------------------------*/
int OpenMlFile(char *fname,int lenbuf)
{
	OPENFILENAME ofn;
	int r;
	char *p,defext[5],tmp[512];

	memset(&ofn,0,sizeof(OPENFILENAME));
	memset(tmp,0,sizeof(tmp));
	fname[0] = 0;
	strcpy(tmp,"ocaml sources|*.ml|bytecode object files|*.cmo|All files|*.*");
	p = tmp;
	while (*p) {
		if (*p == '|')
			*p = 0;
		p++;
	}
	strcpy(defext,"ml");
	ofn.lStructSize = sizeof(OPENFILENAME);
	ofn.hwndOwner = hwndMain;
	ofn.lpstrFilter = tmp;
	ofn.nFilterIndex = 1;
	ofn.hInstance = hInst;
	ofn.lpstrFile = fname;
	ofn.lpstrTitle = "Open file";
	ofn.lpstrInitialDir = LibDir;
	ofn.nMaxFile = lenbuf;
	ofn.Flags = OFN_PATHMUSTEXIST | OFN_NOCHANGEDIR | OFN_LONGNAMES |
		OFN_HIDEREADONLY |OFN_EXPLORER;
	r = GetOpenFileName(&ofn);
	if (r == 0)
		return 0;
	else return 1;
}
/*------------------------------------------------------------------------
 Procedure:     GetSaveName ID:1
 Purpose:       Get a name to save the current session (Save as menu
                item)
 Input:         A buffer where the name of the file will be stored,
                and its length
 Output:        The name of the file choosen by the user will be
                stored in the buffer
 Errors:        none
------------------------------------------------------------------------*/
int GetSaveName(char *fname,int lenbuf)
{
	OPENFILENAME ofn;
	int r;
	char *p,defext[5],tmp[512];

	memset(&ofn,0,sizeof(OPENFILENAME));
	memset(tmp,0,sizeof(tmp));
	fname[0] = 0;
	strcpy(tmp,"Text files|*.txt");
	p = tmp;
	while (*p) {
		if (*p == '|')
			*p = 0;
		p++;
	}
	strcpy(defext,"txt");
	ofn.lStructSize = sizeof(OPENFILENAME);
	ofn.hwndOwner = hwndMain;
	ofn.lpstrFilter = tmp;
	ofn.nFilterIndex = 1;
	ofn.hInstance = hInst;
	ofn.lpstrFile = fname;
	ofn.lpstrTitle = "Save as";
	ofn.lpstrInitialDir = LibDir;
	ofn.nMaxFile = lenbuf;
	ofn.Flags =  OFN_NOCHANGEDIR | OFN_LONGNAMES |
		OFN_HIDEREADONLY |OFN_EXPLORER;
	r = GetSaveFileName(&ofn);
	if (r == 0)
		return 0;
	else return 1;
}
/*------------------------------------------------------------------------
 Procedure:     BrowseForFile ID:1
 Purpose:       Let's the user browse for a certain kind of file.
                Currently this is only used when browsing for
                ocaml.exe.
 Input:         The name of the file to browse for, and the path
                where the user's choice will be stored.
 Output:        1 if user choosed a path, zero otherwise
 Errors:        None
------------------------------------------------------------------------*/
int BrowseForFile(char *fname,char *path)
{
	OPENFILENAME ofn;
	char *p,tmp[512],browsefor[512];
	int r;

	memset(tmp,0,sizeof(tmp));
	strncpy(tmp,fname,sizeof(tmp)-1);
	p = tmp;
	while (*p) {
		if (*p == '|')
			*p = 0;
		p++;
	}
	memset(&ofn,0,sizeof(OPENFILENAME));
	ofn.lpstrFilter = tmp;
	ofn.nFilterIndex = 1;
	ofn.lStructSize = sizeof(OPENFILENAME);
	ofn.hwndOwner = hwndMain;
	ofn.hInstance = hInst;
	ofn.lpstrFilter = tmp;
	ofn.lpstrFile = path;
	wsprintf(browsefor,"Open %s",fname);
	ofn.lpstrTitle = browsefor;
	ofn.lpstrInitialDir = "c:\\";
	ofn.nMaxFile = MAX_PATH;
	ofn.Flags = OFN_PATHMUSTEXIST | OFN_NOCHANGEDIR | OFN_LONGNAMES |
		OFN_HIDEREADONLY |OFN_EXPLORER;
	r = GetOpenFileName(&ofn);
	if (r == 0)
		return 0;
	else return 1;
}

/*------------------------------------------------------------------------
 Procedure:     CallChangeFont ID:1
 Purpose:       Calls the standard windows font change dialog. If the
                user validates a font, it will destroy the current
                font, and recreate a new font with the given
                parameters.
 Input:         The calling window handle
 Output:        Zero if the user cancelled, 1 otherwise.
 Errors:        None
------------------------------------------------------------------------*/
static int CallChangeFont(HWND hwnd)
{
	LOGFONT lf;
	CHOOSEFONT cf;
	int r;
	HWND hwndChild;

	memset(&cf, 0, sizeof(CHOOSEFONT));
	memcpy(&lf, &CurrentFont, sizeof(LOGFONT));
	cf.lStructSize = sizeof(CHOOSEFONT);
	cf.hwndOwner = hwnd;
	cf.lpLogFont = &lf;
	cf.Flags = CF_SCREENFONTS | CF_EFFECTS | CF_APPLY | CF_INITTOLOGFONTSTRUCT;
	cf.nFontType = SCREEN_FONTTYPE;
	r = ChooseFont(&cf);
	if (!r)
		return (0);
	DeleteObject(ProgramParams.hFont);
	memcpy(&CurrentFont, &lf, sizeof(LOGFONT));
	ProgramParams.hFont = CreateFontIndirect(&CurrentFont);
	strcpy(CurrentFontName, CurrentFont.lfFaceName);
	CurrentFontFamily = lf.lfPitchAndFamily;
	CurrentFontStyle = lf.lfWeight;
    hwndChild = (HWND) GetWindowLong(hwndSession, DWL_USER);
	SendMessage(hwndChild,WM_SETFONT,(WPARAM)ProgramParams.hFont,0);
	ForceRepaint();
	return (1);
}

/*------------------------------------------------------------------------
 Procedure:     CallDlgProc ID:1
 Purpose:       Calls a dialog box procedure
 Input:         The function to call, and the numerical ID of the
                resource where the dialog box is stored
 Output:        Returns the result of the dialog box.
 Errors:        None
------------------------------------------------------------------------*/
int CallDlgProc(BOOL (CALLBACK *fn)(HWND,UINT,WPARAM,LPARAM), int id)
{
   int result;

   result = DialogBoxParam(hInst, MAKEINTRESOURCE(id), GetActiveWindow(),
                        fn, 0);
   return result;
}


/*------------------------------------------------------------------------
 Procedure:     CallChangeColor ID:1
 Purpose:       Calls the standard color dialog of windows, starting
                with the given color reference. The result is the
                same as the input if the user cancels, or another
                color if the user validates another one.
 Input:         The starting color
 Output:        The color the user has choosen.
 Errors:        None
------------------------------------------------------------------------*/
static COLORREF CallChangeColor(COLORREF InitialColor)
{
	CHOOSECOLOR CC;
	COLORREF CustColors[16];
	int r, g, b, i;
	memset(&CC, 0, sizeof(CHOOSECOLOR));
	r = g = b = 0;
	for (i = 0; i < 16; i++) {
		CustColors[i] = RGB(r, g, b);
		if (r < 255)
			r += 127;
		else if (g < 255)
			g += 127;
		else if (b < 255)
			g += 127;
	}
	CC.lStructSize = sizeof(CHOOSECOLOR);
	CC.hwndOwner = hwndMain;
	CC.hInstance = hInst;
	CC.rgbResult = InitialColor;
	CC.lpCustColors = CustColors;
	CC.Flags = CC_RGBINIT;
	if (!ChooseColor(&CC))
		return (InitialColor);
	return (CC.rgbResult);
}

/*------------------------------------------------------------------------
 Procedure:     CallPrintSetup ID:1
 Purpose:       Calls the printer setup dialog. Currently it is not
                connected to the rest of the software, since printing
                is not done yet
 Input:         None
 Output:        1 if OK, 0, user cancelled
 Errors:        None
------------------------------------------------------------------------*/
static int CallPrintSetup(void)
{
	PAGESETUPDLG sd;
	int r;

	memset(&sd,0,sizeof(sd));
	sd.lStructSize = sizeof(sd);
	sd.Flags = PSD_RETURNDEFAULT;
	r = PageSetupDlg(&sd);
	if (!r)
		return 0;
	sd.Flags = 0;
	r = PageSetupDlg(&sd);
	return r;
}


/*------------------------------------------------------------------------
 Procedure:     Undo ID:1
 Purpose:       Send an UNDO command to the edit field.
 Input:         The parent window of the control
 Output:        None
 Errors:        None
------------------------------------------------------------------------*/
void Undo(HWND hwnd)
{
	HWND hEdit;

	hEdit = (HWND)GetWindowLong(hwnd,DWL_USER);
	SendMessage(hEdit,EM_UNDO,0,0);
}

/*------------------------------------------------------------------------
 Procedure:     ForceRepaint ID:1
 Purpose:       Forces a complete redraw of the edit control of the
                current session.
 Input:         None
 Output:        None
 Errors:        None
------------------------------------------------------------------------*/
void ForceRepaint(void)
{
	HWND hwndEdit = (HWND)GetWindowLong(hwndSession,DWL_USER);
	InvalidateRect(hwndEdit,NULL,1);
}

static void Add_Char_To_Queue(int c)
{
	HWND hwndEdit = (HWND)GetWindowLong(hwndSession,DWL_USER);
	SendMessage(hwndEdit,WM_CHAR,c,1);
}

/*------------------------------------------------------------------------
 Procedure:     AddLineToControl ID:1
 Purpose:       It will ad the given text at the end of the edit
                control, then it will send a return character to it.
                This simulates user input. The history will not be
                modified by this procedure.
 Input:         The text to be added
 Output:        None
 Errors:        If the line is empty, nothing will be done
------------------------------------------------------------------------*/
void AddLineToControl(char *buf)
{
	HWND hEditCtrl;

	if (*buf == 0)
		return;
	hEditCtrl = (HWND)GetWindowLong(hwndSession,DWL_USER);
	GotoEOF();
	SendMessage(hEditCtrl,EM_REPLACESEL,0,(LPARAM)buf);
	SendMessage(hEditCtrl,WM_CHAR,'\r',0);
}

/*------------------------------------------------------------------------
 Procedure:     AboutDlgProc ID:1
 Purpose:       Shows the "About" dialog box
 Input:
 Output:
 Errors:
------------------------------------------------------------------------*/
static BOOL CALLBACK AboutDlgProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
	if (message == WM_CLOSE)
		EndDialog(hDlg,1);
	return 0;
}
/*------------------------------------------------------------------------
 Procedure:     HistoryDlgProc ID:1
 Purpose:       Shows the history of the session. Only input lines
                are shown. A double click in a line will make this
                dialog box procedure return the index of the selected
                line (1 based). If the windows is closed (what is
                equivalent to cancel), the return value is zero.
 Input:         Normal windows callback
 Output:
 Errors:
------------------------------------------------------------------------*/
static BOOL CALLBACK HistoryDlgProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
	HISTORYLINE *rvp;
	int idx;
	RECT rc;

	switch (message) {
		case WM_INITDIALOG:
			SendDlgItemMessage(hDlg,IDLIST,WM_SETFONT,(WPARAM)ProgramParams.hFont,0);
			rvp = History;
			idx = 0;
			while (rvp) {
				SendDlgItemMessage(hDlg,IDLIST,LB_INSERTSTRING,0,(LPARAM)rvp->Text);
				SendDlgItemMessage(hDlg,IDLIST,LB_SETITEMDATA,0,(LPARAM)idx);
				rvp = rvp->Next;
				idx++;
			}
			SendDlgItemMessage(hDlg,IDLIST,LB_SETCURSEL,(LPARAM)idx-1,0);
			return 1;
		case WM_COMMAND:
			switch(LOWORD(wParam)) {
				case IDLIST:
					switch(HIWORD(wParam)) {
						case LBN_DBLCLK:
							idx = SendDlgItemMessage(hDlg,IDLIST,LB_GETCURSEL,0,0);
							if (idx == LB_ERR)
								break;
							idx = SendDlgItemMessage(hDlg,IDLIST,LB_GETITEMDATA,idx,0);
							EndDialog(hDlg,idx+1);
							return 1;
					}
					break;
			}
			break;
		case WM_SIZE:
			GetClientRect(hDlg,&rc);
			MoveWindow(GetDlgItem(hDlg,IDLIST),0,0,rc.right,rc.bottom,1);
			break;

		case WM_CLOSE:
			EndDialog(hDlg,0);
			break;
	}
	return 0;
}
/*------------------------------------------------------------------------
 Procedure:     SaveText ID:1
 Purpose:       Saves the contents of the session transcript. It will
                loop for each line and write it to the specified file
 Input:         The name of the file where the session will be saved
 Output:        The session is saved
 Errors:        If it can't open the file for writing it will show an
                error box
------------------------------------------------------------------------*/
static void SaveText(char *fname)
{
	int i,len;
	HWND hEdit = (HWND)GetWindowLong(hwndSession,DWL_USER);
	int linesCount = SendMessage(hEdit,EM_GETLINECOUNT,0,0);
	FILE *f;
	char *buf = SafeMalloc(8192);

	f = fopen(fname,"wb");
	if (f == NULL) {
		wsprintf("Impossible to open %s for writing",fname);
		ShowDbgMsg(buf);
		return;
	}
	for (i=0; i<linesCount;i++) {
		*(unsigned short *)buf = 8100;
		len = SendMessage(hEdit,EM_GETLINE,i,(LPARAM)buf);
		buf[len] = 0;
		strcat(buf,"\r\n");
		fwrite(buf,1,len+2,f);
	}
	fclose(f);
	free(buf);
}


static void Add_Clipboard_To_Queue(void)
{
    if (IsClipboardFormatAvailable(CF_TEXT) &&
        OpenClipboard(hwndMain))
    {
        HANDLE hClipData = GetClipboardData(CF_TEXT);

        if (hClipData)
        {
            char *str = GlobalLock(hClipData);

            if (str)
                while (*str)
                {
                    if (*str != '\r')
                        Add_Char_To_Queue(*str);
                    str++;
                }
            GlobalUnlock(hClipData);
        }
        CloseClipboard();
    }

}

static void CopyToClipboard(HWND hwnd)
{
	HWND hwndEdit = (HWND)GetWindowLong(hwndSession,DWL_USER);
	SendMessage(hwndEdit,WM_COPY,0,0);
}

int ResetText(void)
{
	HWND hwndEdit = (HWND) GetWindowLong(hwndSession,DWL_USER);
	TEXTRANGE cr;
	int len = SendMessage(hwndEdit,WM_GETTEXTLENGTH,0,0);
	char *tmp = malloc(len+10),*p;

	memset(tmp,0,len+10);
	cr.chrg.cpMin = 0;
	cr.chrg.cpMax = -1;
	cr.lpstrText = tmp;
	SendMessage(hwndEdit,EM_GETTEXTRANGE,0,(LPARAM)&cr);
	p = tmp+len/2;
	while (*p && *p != '\r')
		p++;
	SendMessage(hwndEdit,EM_SETSEL,0,(LPARAM)-1);
	SendMessage(hwndEdit,EM_REPLACESEL,0,(LPARAM)p);
	InvalidateRect(hwndEdit,0,1);
	free(tmp);
	return 0;
}

/*------------------------------------------------------------------------
 Procedure:     HandleCommand ID:1
 Purpose:       Handles all menu commands.
 Input:
 Output:
 Errors:
------------------------------------------------------------------------*/
void HandleCommand(HWND hwnd, WPARAM wParam,LPARAM lParam)
{
	char *fname;
	int r;

	switch(LOWORD(wParam)) {
		case IDM_OPEN:
			fname = SafeMalloc(512);
			if (OpenMlFile(fname,512)) {
				char *buf = SafeMalloc(512);
				char *p = strrchr(fname,'.');
				if (p && !stricmp(p,".ml")) {
					wsprintf(buf,"#use \"%s\";;",fname);
					AddLineToControl(buf);
				}
				else if (p && !stricmp(p,".cmo")) {
					wsprintf(buf,"#load \"%s\";;",fname);
					AddLineToControl(buf);
				}
				free(buf);
			}
			free(fname);
			break;
		case IDM_GC:
			AddLineToControl("Gc.full_major();;");
			break;
		case IDCTRLC:
			InterruptOcaml();
			break;
		case IDM_EDITPASTE:
			Add_Clipboard_To_Queue();
			break;
		case IDM_EDITCOPY:
			CopyToClipboard(hwnd);
			break;
		case IDM_SAVE:
			fname = SafeMalloc(512);
			if (GetSaveName(fname,512)) {
				SaveText(fname);
			}
			free(fname);
			break;
		case IDM_HISTORY:
			r = CallDlgProc(HistoryDlgProc,IDD_HISTORY);
			if (r) {
				AddLineToControl(GetHistoryLine(r-1));
			}
			break;
		case IDM_PRINTSU:
			CallPrintSetup();
			break;
		case IDM_FONT:
			CallChangeFont(hwndMain);
			break;
		case IDM_COLORTEXT:
			ProgramParams.TextColor = CallChangeColor(ProgramParams.TextColor);
			ForceRepaint();
			break;
		case IDM_BACKCOLOR:
			BackColor = CallChangeColor(BackColor);
			DeleteObject(BackgroundBrush);
			BackgroundBrush = CreateSolidBrush(BackColor);
			ForceRepaint();
			break;
		case IDM_EDITUNDO:
			Undo(hwnd);
			break;
		case IDM_WINDOWTILE:
			SendMessage(hwndMDIClient,WM_MDITILE,0,0);
			break;
		case IDM_WINDOWCASCADE:
			SendMessage(hwndMDIClient,WM_MDICASCADE,0,0);
			break;
		case IDM_WINDOWICONS:
			SendMessage(hwndMDIClient,WM_MDIICONARRANGE,0,0);
			break;
		case IDM_EXIT:
			PostMessage(hwnd,WM_CLOSE,0,0);
			break;
		case IDM_ABOUT:
			CallDlgProc(AboutDlgProc,IDD_ABOUT);
			break;
		default:
			if (LOWORD(wParam) >= IDEDITCONTROL && LOWORD(wParam) < IDEDITCONTROL+5) {
				switch (HIWORD(wParam)) {
					case EN_ERRSPACE:
						ResetText();
						break;
				}
			}
			break;
	}
}