#include <windows.h>

#include <commctrl.h>

#include "exdll.h"

#include <Python.h>

HINSTANCE g_hInstance;

HWND g_hwndParent;

HWND g_hwndList, g_hwndProgress;

static void LogMessage(const char *pStr);

static void callPython(int eval_type);

static void SetProgressRange(int iLowLim, int iHighLim);

static void SetProgressPos(int iPos);

const struct {char *name; int enumcode;} string_to_enum[] = {

{"0", INST_0},

{"1", INST_1},

{"2", INST_2},

{"3", INST_3},

{"4", INST_4},

{"5", INST_5},

{"6", INST_6},

{"7", INST_7},

{"8", INST_8},

{"9", INST_9},

{"R0", INST_R0},

{"R1", INST_R1},

{"R2", INST_R2},

{"R3", INST_R3},

{"R4", INST_R4},

{"R5", INST_R5},

{"R6", INST_R6},

{"R7", INST_R7},

{"R8", INST_R8},

{"R9", INST_R9},

{"CMDLINE", INST_CMDLINE},

{"INSTDIR", INST_INSTDIR},

{"OUTDIR", INST_OUTDIR},

{"EXEDIR", INST_EXEDIR},

{"LANGUAGE", INST_LANG},

};

#define __doc__log "log(string)\n"\

"Write Messages to the NSIS log window."

static PyObject*

py_log(PyObject *self, PyObject *args)

{

char *logtext;

if (!PyArg_ParseTuple(args, "s", &logtext))

return 0;

LogMessage(logtext);

Py_INCREF(Py_None);

return Py_None;

}

#define __doc__messagebox "messagebox(string, title='NSIS Python')\n"\

"Pop up a message box."

static PyObject*

py_messagebox(PyObject *self, PyObject *args)

{

char *text;

char *title = "NSIS Python";

if (!PyArg_ParseTuple(args, "s|s", &text, &title))

return 0;

MessageBox(g_hwndParent, text, title, MB_OK);

Py_INCREF(Py_None);

return Py_None;

}

my_strcmp(const char *s1, const char *s2)

{

while (*s1 == *s2++)

if (*s1++ == 0)

return (0);

return (*(const unsigned char *)s1 - *(const unsigned char *)(s2 - 1));

}

#define __doc__getvar "getvar(varname_string)\n"\

"Get a variable fom NSIS. The contents of a variable is always a string."

static PyObject*

py_getvar(PyObject *self, PyObject *args)

{

int i;

char *varname;

if (!PyArg_ParseTuple(args, "s", &varname))

return 0;

if (varname[0] == '$') varname++; //vars may start with "$" but not required

for( i=0; i<26; i++) {

if (my_strcmp(varname, string_to_enum[i].name) == 0) {

return PyString_FromString(getuservariable(string_to_enum[i].enumcode));

}

}

PyErr_Format(PyExc_NameError, "There is no NSIS variable named '$%s'.", varname);

return NULL;

}

#define __doc__setvar "setvar(varname_string, value)\n"\

"Set a variable fom NSIS. The contents of a variable is always a string."

static PyObject*

py_setvar(PyObject *self, PyObject *args)

{

int i;

char *varname;

char *value;

if (!PyArg_ParseTuple(args, "ss", &varname, &value))

return 0;

if (varname[0] == '$') varname++; //vars may start with "$" but not required

for( i=0; i<26; i++) {

if (my_strcmp(varname, string_to_enum[i].name) == 0) {

setuservariable(string_to_enum[i].enumcode, value);

Py_INCREF(Py_None);

return Py_None;

}

}

PyErr_Format(PyExc_NameError, "There is no NSIS variable named '$%s'.", varname);

return NULL;

}

#define __doc__getParent "getParent()\n"\

"Get the parent's handle of the installer."

static PyObject*

py_getParent(PyObject *self, PyObject *args)

{

if (!PyArg_ParseTuple(args, ""))

return 0;

return PyLong_FromLong((long)g_hwndParent);

}

#define __doc__setProgressRange "setProgressRange()\n"\

"Set the low and high limit of the progress bar."

static PyObject*

py_setProgressRange(PyObject *self, PyObject *args)

{

int low, high;

if (!PyArg_ParseTuple(args, "ii", &low, &high))

return 0;

SetProgressRange(low, high);

Py_INCREF(Py_None);

return Py_None;

}

#define __doc__setProgressPos "setProgressPos()\n"\

"Set the position of the progress bar."

static PyObject*

py_setProgressPos(PyObject *self, PyObject *args)

{

int pos;

if (!PyArg_ParseTuple(args, "i", &pos))

return 0;

SetProgressPos(pos);

Py_INCREF(Py_None);

return Py_None;

}

static PyMethodDef ext_methods[] = {

{"log", py_log, METH_VARARGS, __doc__log},

{"messagebox", py_messagebox, METH_VARARGS, __doc__messagebox},

{"getvar", py_getvar, METH_VARARGS, __doc__getvar},

{"setvar", py_setvar, METH_VARARGS, __doc__setvar},

{"getParent", py_getParent, METH_VARARGS, __doc__getParent},

{"setProgressRange",py_setProgressRange, METH_VARARGS, __doc__setProgressRange},

{"setProgressPos", py_setProgressPos, METH_VARARGS, __doc__setProgressPos},

//End maeker

{0, 0}

};

void __declspec(dllexport) eval(HWND hwndParent, int string_size, char *variables, stack_t **stacktop) {

g_hwndParent = hwndParent;

EXDLL_INIT();

callPython(Py_eval_input);

}

void __declspec(dllexport) exec(HWND hwndParent, int string_size, char *variables, stack_t **stacktop) {

g_hwndParent = hwndParent;

EXDLL_INIT();

callPython(Py_file_input);

}

void __declspec(dllexport) SetOptimized(HWND hwndParent, int string_size, char *variables, stack_t **stacktop) {

g_hwndParent = hwndParent;

EXDLL_INIT();

Py_OptimizeFlag++;

}

void __declspec(dllexport) UnsetOptimized(HWND hwndParent, int string_size, char *variables, stack_t **stacktop) {

g_hwndParent = hwndParent;

EXDLL_INIT();

if (Py_OptimizeFlag > 0)

Py_OptimizeFlag--;

}

BOOL WINAPI _DllMainCRTStartup(HANDLE hInst, ULONG ul_reason_for_call, LPVOID lpReserved) {

g_hInstance = hInst;

return TRUE;

}

static void callPython(int eval_type) {

PyObject *m, *d, *v, *result;

char *command = (char *)GlobalAlloc(GPTR, sizeof(char)*g_stringsize + 1);

g_hwndList = FindWindowEx(FindWindowEx(g_hwndParent, NULL, "#32770", NULL), NULL, "SysListView32", NULL);

g_hwndProgress = FindWindowEx(FindWindowEx(g_hwndParent, NULL, "#32770", NULL), NULL, "msctls_progress32", NULL);

Py_Initialize();

Py_InitModule("nsis", ext_methods);

popstring(command);

m = PyImport_AddModule("__main__");

if (m == NULL) {

pushstring("error");

return;

}

d = PyModule_GetDict(m);

v = PyRun_String(command, eval_type, d, d);

if (v == NULL) {

PyObject *ptype, *pvalue, *ptraceback;

PyErr_Fetch(&ptype, &pvalue, &ptraceback);

if (pvalue) {

LogMessage("Python Exception:");

LogMessage(PyString_AsString(PyObject_Str(pvalue)));

}

pushstring("error");

PyErr_Clear();

return;

}

result = PyObject_Str(v);

pushstring(PyString_AsString(result));

Py_DECREF(result);

Py_DECREF(v);

Py_Finalize();

GlobalFree(command);

}

static void LogMessage(const char *pStr) {

LVITEM item={0};

int nItemCount;

if (!g_hwndList) return;

if (!lstrlen(pStr)) return;

nItemCount=SendMessage(g_hwndList, LVM_GETITEMCOUNT, 0, 0);

item.mask=LVIF_TEXT;

item.pszText=(char *)pStr;

item.cchTextMax=0;

item.iItem=nItemCount;

ListView_InsertItem(g_hwndList, &item);

ListView_EnsureVisible(g_hwndList, item.iItem, 0);

}

static void SetProgressRange(int iLowLim, int iHighLim) {

if (!g_hwndProgress) return;

SendMessage(g_hwndProgress, PBM_SETRANGE32, (WPARAM)iLowLim, (LPARAM)iHighLim);

}

static void SetProgressPos(int iPos) {

if (!g_hwndProgress) return;

SendMessage(g_hwndProgress, PBM_SETPOS, (WPARAM)iPos, 0);

}