#include <windows.h>

#include <windows.h>
#include <stdio.h>
#include <string.h>
#include <time.h>
#include <wsqllib.h>

#include "..\sqlbase.h"
#include "..\dbinit.h"
#include "logbook.h"

HINSTANCE hInst;

MENUITEM extraMenuItems[] = {
{"Insert Start\tCtrl+S", ID_INSERT_START},
{"Insert End\tCtrl+E", ID_INSERT_END},
{"Lookup Country\tCtrl+P", ID_LOOKUP_COUNTRY},
{"Check Country\tCtrl+K", ID_CHECK_COUNTRY},
{"Show Country Data\tCtrl+R", ID_SHOW_COUNTRY_DATA},
{NULL, 0}
};

ACCEL accelTable[] = {
{FVIRTKEY | FCONTROL, (WORD)'I', ID_INIT_FIELDS},
{FVIRTKEY | FCONTROL, (WORD)'S', ID_INSERT_START},
{FVIRTKEY | FCONTROL, (WORD)'E', ID_INSERT_END},
{FVIRTKEY | FCONTROL, (WORD)'P', ID_LOOKUP_COUNTRY},
{FVIRTKEY | FCONTROL, (WORD)'K', ID_CHECK_COUNTRY},
{FVIRTKEY | FCONTROL, (WORD)'R', ID_SHOW_COUNTRY_DATA},
{FVIRTKEY | FCONTROL, VK_RETURN, ID_CTRL_RETURN}
};

int GetAccelTableSize(void)
{
return sizeof(accelTable);
}

BOOL CALLBACK DlgProc(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
static HWND listBox;
int i, n;
static DLG_DATA dd, *pdd;

switch (msg) {
case WM_INITDIALOG:
pdd = (DLG_DATA*)lParam;
dd = *pdd;
listBox = GetDlgItem(hDlg, ID_LISTBOX);
SetWindowText(hDlg, dd.lbTitle);
for (i = 0; dd.lbData[i]; i++) {
SendMessage(listBox, LB_ADDSTRING, (WPARAM)0,
(LPARAM)dd.lbData[i]);
}
SetFocus(listBox);
return FALSE;

case WM_COMMAND :
switch (LOWORD (wParam)) {
case IDOK:
n = SendMessage(listBox, LB_GETCURSEL, (WPARAM)0,
(LPARAM)0);
EndDialog(hDlg, n + 1);
return TRUE;
case IDCANCEL:
EndDialog(hDlg, 0);
return TRUE ;
}
break;
}
return FALSE;
}

void InitFields(VARS *v)
{
int i;

for (i = 0; i < v->nFields; i++)
SetWindowText(v->hwndEdit[i], "");

SetWindowText(v->hwndEdit[8], "CW");
SetWindowText(v->hwndEdit[9], "150");

SetWindowText(v->hwndEdit[12], "No");
SetWindowText(v->hwndEdit[13], "No");
}

void InsertStart(VARS *v)
{
time_t t;
struct tm *gmt;
char s[80];

t = time(NULL);
gmt = gmtime(&t);

strftime(s, 80, "%Y-%m-%d", gmt);
SetWindowText(v->hwndEdit[0], s);

strftime(s, 80, "%H%M", gmt);
SetWindowText(v->hwndEdit[1], s);
}

void InsertEnd(VARS *v)
{
time_t t;
struct tm *gmt;
char s[80];

t = time(NULL);
gmt = gmtime(&t);

strftime(s, 80, "%H%M", gmt);
SetWindowText(v->hwndEdit[2], s);
}

void LookupCountry(VARS *v)
{
DLG_DATA dd;
char *countries[20];
char buf[4096];
char callsign[50];
MYSQL_ROW row;
int i, n, end;

memset(callsign, 0, 50);
GetWindowText(v->hwndEdit[3], callsign, 50);
if (! (end = strlen(callsign)))
return;

while (1) {
sprintf(buf, "SELECT * FROM prefix WHERE prefix LIKE '%%:%s:%%'",
callsign);

if (! SQLQuery(0, buf))
return;

for (i = 0; row = SQLFetchRow(0); i++) {
countries[i] = (char*)malloc(strlen(row[0]) + 1);
strcpy(countries[i], row[0]);
}
if (i)
break;
else
SQLEndQuery(0);

*(callsign + (--end)) = '\0';
if (! (*callsign)) {
MessageBox(NULL, "No match found", "SQLBase",
MB_OK | MB_ICONINFORMATION);
return;
}
}

SQLEndQuery(0);
countries[i] = NULL;

n = 0;
if (i > 1) {
dd.lbTitle = "Select Country";
dd.lbData = countries;
if (! (n = DialogBoxParam(hInst, "ListBox", v->hwndMain,
DlgProc, (LPARAM)&dd)))
goto end;
n--;
}

SetWindowText(v->hwndEdit[6], countries[n]);
end:
for (i = 0; countries[i]; i++)
free(countries[i]);
}

void CheckCountry(VARS *v)
{
MYSQL_ROW row;
char country[50];
char buf[4096];
int i;
BOOL found = FALSE, confirmed = FALSE;

memset(country, 0, 50);
GetWindowText(v->hwndEdit[6], country, 50);
if (! strlen(country))
return;

sprintf(buf, "SELECT * FROM %s WHERE country='%s'",
v->table, country);

if (! SQLQuery(0, buf))
return;

for (i = 0; row = SQLFetchRow(0); i++) {
found = TRUE;
if (strcmp(row[13], "Yes")) {
confirmed = TRUE;
break;
}
}

SQLEndQuery(0);

if (! found) {
sprintf(buf, "'%s' is a new country", country);
} else if (! confirmed) {
sprintf(buf, "'%s' worked but not confirmed", country);
} else {
sprintf(buf, "'%s' worked and confirmed", country);
}

MessageBox(v->hwndMain, buf, "SQLBase",
MB_OK | MB_ICONINFORMATION);
}

void ShowCountryData(VARS *v)
{

}

void CheckCtrlReturn(VARS *v)
{
DLG_DATA dd;
char *modes[] = {"CW", "LSB", "USB", "AM", "FM", "RY",
"AMT", "PTR", "PKT", "STV", NULL};
int n;

if (GetFocus() == v->hwndEdit[8]) {
dd.lbTitle = "Select Mode";
dd.lbData = modes;
if ((n = DialogBoxParam(hInst, "ListBox", v->hwndMain,
DlgProc, (LPARAM)&dd)))
SetWindowText(v->hwndEdit[8], modes[n - 1]);
}
}

void CheckExtraAccel(WORD id, VARS *v)
{
switch (id ) {
case ID_INIT_FIELDS:
InitFields(v);
break;

case ID_INSERT_START:
InsertStart(v);
break;

case ID_INSERT_END:
InsertEnd(v);
break;

case ID_LOOKUP_COUNTRY:
LookupCountry(v);
break;

case ID_CHECK_COUNTRY:
CheckCountry(v);
break;

case ID_SHOW_COUNTRY_DATA:
ShowCountryData(v);
break;

case ID_CTRL_RETURN:
CheckCtrlReturn(v);
break;
}
}

BOOL WINAPI DllEntryPoint(HINSTANCE hInstance, DWORD fdwReason,
PVOID pvReserved)
{
hInst = hInstance;
return TRUE;
}