#include "fitz-internal.h"
#include "mupdf-internal.h"

typedef struct pdf_js_event_s
{
	pdf_obj *target;
	char *value;
} pdf_js_event;

struct pdf_js_s
{
	pdf_document *doc;
	pdf_obj *form;
	pdf_js_event event;
	pdf_jsimp *imp;
	pdf_jsimp_type *doctype;
	pdf_jsimp_type *eventtype;
	pdf_jsimp_type *fieldtype;
};

static pdf_obj *load_color(fz_context *ctx, pdf_jsimp *imp, pdf_jsimp_obj *val)
{
	pdf_obj *col = NULL;

	if (pdf_jsimp_array_len(imp, val) == 4)
	{
		pdf_obj *comp = NULL;
		pdf_jsimp_obj *jscomp = NULL;
		int i;

		col = pdf_new_array(ctx, 3);

		fz_var(comp);
		fz_var(jscomp);
		fz_try(ctx)
		{
			for (i = 0; i < 3; i++)
			{
				jscomp = pdf_jsimp_array_item(imp, val, i+1);
				comp = pdf_new_real(ctx, pdf_jsimp_toNumber(imp, jscomp));
				pdf_array_push(col, comp);
				pdf_jsimp_drop_obj(imp, jscomp);
				jscomp = NULL;
				pdf_drop_obj(comp);
				comp = NULL;
			}
		}
		fz_catch(ctx)
		{
			pdf_jsimp_drop_obj(imp, jscomp);
			pdf_drop_obj(comp);
			pdf_drop_obj(col);
			fz_rethrow(ctx);
		}
	}

	return col;
}

static pdf_jsimp_obj *field_buttonSetCaption(void *jsctx, void *obj, int argc, pdf_jsimp_obj *args[])
{
	pdf_js  *js = (pdf_js *)jsctx;
	pdf_obj *field = (pdf_obj *)obj;
	char    *name;

	if (argc != 1)
		return NULL;

	name = pdf_jsimp_toString(js->imp, args[0]);
	pdf_field_buttonSetCaption(js->doc, field, name);

	return NULL;
}

static pdf_jsimp_obj *field_getFillColor(void *jsctx, void *obj)
{
	return NULL;
}

static void field_setFillColor(void *jsctx, void *obj, pdf_jsimp_obj *val)
{
	pdf_js *js = (pdf_js *)jsctx;
	fz_context *ctx = js->doc->ctx;
	pdf_obj *field = (pdf_obj *)obj;
	pdf_obj *col = load_color(js->doc->ctx, js->imp, val);

	fz_try(ctx)
	{
		pdf_field_setFillColor(js->doc, field, col);
	}
	fz_always(ctx)
	{
		pdf_drop_obj(col);
	}
	fz_catch(ctx)
	{
		fz_rethrow(ctx);
	}
}

static pdf_jsimp_obj *field_getTextColor(void *jsctx, void *obj)
{
	return NULL;
}

static void field_setTextColor(void *jsctx, void *obj, pdf_jsimp_obj *val)
{
	pdf_js *js = (pdf_js *)jsctx;
	fz_context *ctx = js->doc->ctx;
	pdf_obj *field = (pdf_obj *)obj;
	pdf_obj *col = load_color(js->doc->ctx, js->imp, val);

	fz_try(ctx)
	{
		pdf_field_setTextColor(js->doc, field, col);
	}
	fz_always(ctx)
	{
		pdf_drop_obj(col);
	}
	fz_catch(ctx)
	{
		fz_rethrow(ctx);
	}
}

static pdf_jsimp_obj *field_getBorderStyle(void *jsctx, void *obj)
{
	pdf_js *js = (pdf_js *)jsctx;
	pdf_obj *field = (pdf_obj *)obj;

	return pdf_jsimp_fromString(js->imp, pdf_field_getBorderStyle(js->doc, field));
}

static void field_setBorderStyle(void *jsctx, void *obj, pdf_jsimp_obj *val)
{
	pdf_js *js = (pdf_js *)jsctx;
	pdf_obj *field = (pdf_obj *)obj;

	pdf_field_setBorderStyle(js->doc, field, pdf_jsimp_toString(js->imp, val));
}

static pdf_jsimp_obj *field_getValue(void *jsctx, void *obj)
{
	pdf_js *js = (pdf_js *)jsctx;
	pdf_obj *field = (pdf_obj *)obj;
	char *fval = pdf_field_getValue(js->doc, field);

	return pdf_jsimp_fromString(js->imp, fval?fval:"");
}

static void field_setValue(void *jsctx, void *obj, pdf_jsimp_obj *val)
{
	pdf_js *js = (pdf_js *)jsctx;
	pdf_obj *field = (pdf_obj *)obj;

	pdf_field_setValue(js->doc, field, pdf_jsimp_toString(js->imp, val));
}

static pdf_jsimp_obj *event_getTarget(void *jsctx, void *obj)
{
	pdf_js *js = (pdf_js *)jsctx;
	pdf_js_event *e = (pdf_js_event *)obj;

	return pdf_jsimp_new_obj(js->imp, js->fieldtype, e->target);
}

static void event_setTarget(void *jsctx, void *obj, pdf_jsimp_obj *val)
{
	pdf_js *js = (pdf_js *)jsctx;
	fz_warn(js->doc->ctx, "Unexpected call to event_setTarget");
}

static pdf_jsimp_obj *event_getValue(void *jsctx, void *obj)
{
	pdf_js *js = (pdf_js *)jsctx;
	pdf_js_event *e = (pdf_js_event *)obj;

	return pdf_jsimp_fromString(js->imp, js->event.value);
}

static void event_setValue(void *jsctx, void *obj, pdf_jsimp_obj *val)
{
	pdf_js *js = (pdf_js *)jsctx;
	fz_context *ctx = js->doc->ctx;
	fz_free(ctx, js->event.value);
	js->event.value = NULL;
	js->event.value = fz_strdup(ctx, pdf_jsimp_toString(js->imp, val));
}

static pdf_jsimp_obj *doc_getEvent(void *jsctx, void *obj)
{
	pdf_js *js = (pdf_js *)jsctx;

	return pdf_jsimp_new_obj(js->imp, js->eventtype, &js->event);
}

static void doc_setEvent(void *jsctx, void *obj, pdf_jsimp_obj *val)
{
	pdf_js *js = (pdf_js *)jsctx;
	fz_warn(js->doc->ctx, "Unexpected call to doc_setEvent");
}

static pdf_obj *find_field(pdf_obj *dict, char *name, int len)
{
	pdf_obj *field;

	int i, n = pdf_array_len(dict);

	for (i = 0; i < n; i++)
	{
		char *part;

		field = pdf_array_get(dict, i);
		part = pdf_to_str_buf(pdf_dict_gets(field, "T"));
		if (strlen(part) == len && !memcmp(part, name, len))
			break;
	}

	return i < n ? field : NULL;
}

static pdf_jsimp_obj *doc_getField(void *jsctx, void *obj, int argc, pdf_jsimp_obj *args[])
{
	pdf_js  *js = (pdf_js *)jsctx;
	pdf_obj *arr = js->form;
	pdf_obj *dict = NULL;
	int      len;
	char    *name, *dot;

	if (argc != 1)
		return NULL;

	/* Process the fully qualified field name which has
	 * the partial names delimited by '.' */
	name = pdf_jsimp_toString(js->imp, args[0]);
	/* Pretend there was a preceding '.' to simplify the loop */
	dot = name - 1;

	while (dot && arr)
	{
		name = dot + 1;
		dot = strchr(name, '.');
		len = dot ? dot - name : strlen(name);
		dict = find_field(arr, name, len);
		if (dot)
			arr = pdf_dict_gets(dict, "Kids");
	}

	return dict ? pdf_jsimp_new_obj(js->imp, js->fieldtype, dict) : NULL;
}

static void declare_dom(pdf_js *js)
{
	pdf_jsimp      *imp       = js->imp;

	/* Create the document type */
	js->doctype = pdf_jsimp_new_type(imp, NULL);
	pdf_jsimp_addmethod(imp, js->doctype, "getField", doc_getField);
	pdf_jsimp_addproperty(imp, js->doctype, "event", doc_getEvent, doc_setEvent);

	/* Create the event type */
	js->eventtype = pdf_jsimp_new_type(imp, NULL);
	pdf_jsimp_addproperty(imp, js->eventtype, "target", event_getTarget, event_setTarget);
	pdf_jsimp_addproperty(imp, js->eventtype, "value", event_getValue, event_setValue);

	/* Create the field type */
	js->fieldtype = pdf_jsimp_new_type(imp, NULL);
	pdf_jsimp_addproperty(imp, js->fieldtype, "value", field_getValue, field_setValue);
	pdf_jsimp_addproperty(imp, js->fieldtype, "borderStyle", field_getBorderStyle, field_setBorderStyle);
	pdf_jsimp_addproperty(imp, js->fieldtype, "textColor", field_getTextColor, field_setTextColor);
	pdf_jsimp_addproperty(imp, js->fieldtype, "fillColor", field_getFillColor, field_setFillColor);
	pdf_jsimp_addmethod(imp, js->fieldtype, "buttonSetCaption", field_buttonSetCaption);

	/* Create the document object and tell the engine to use */
	pdf_jsimp_set_global_type(js->imp, js->doctype);
}

static void preload_helpers(pdf_js *js)
{
	pdf_jsimp_execute(js->imp,
		"var border = new Array();\n"
		"border.s = \"Solid\";\n"
		"border.d = \"Dashed\";\n"
		"border.b = \"Beveled\";\n"
		"border.i = \"Inset\";\n"
		"border.u = \"Underline\";\n"
		"var color = new Array();\n"
		"color.transparent = [ \"T\" ];\n"
		"color.black = [ \"G\", 0];\n"
		"color.white = [ \"G\", 1];\n"
		"color.red = [ \"RGB\", 1,0,0 ];\n"
		"color.green = [ \"RGB\", 0,1,0 ];\n"
		"color.blue = [ \"RGB\", 0,0,1 ];\n"
		"color.cyan = [ \"CMYK\", 1,0,0,0 ];\n"
		"color.magenta = [ \"CMYK\", 0,1,0,0 ];\n"
		"color.yellow = [ \"CMYK\", 0,0,1,0 ];\n"
		"color.dkGray = [ \"G\", 0.25];\n"
		"color.gray = [ \"G\", 0.5];\n"
		"color.ltGray = [ \"G\", 0.75];\n"
		"\n"
		"function AFNumber_Format(nDec,sepStyle,negStyle,currStyle,strCurrency,bCurrencyPrepend)\n"
		"{\n"
		"	var val = event.value;\n"
		"	var fracpart;\n"
		"	var intpart;\n"
		"	var point = sepStyle&2 ? ',' : '.';\n"
		"	var separator = sepStyle&2 ? '.' : ',';\n"
		"\n"
		"	if (/^\\D*\\./.test(val))\n"
		"		val = '0'+val;\n"
		"\n"
		"	var groups = val.match(/\\d+/g);\n"
		"\n"
		"	switch (groups.length)\n"
		"	{\n"
		"	case 0:\n"
		"		return;\n"
		"	case 1:\n"
		"		fracpart = '';\n"
		"		intpart = groups[0];\n"
		"		break;\n"
		"	default:\n"
		"		fracpart = groups.pop();\n"
		"		intpart = groups.join('');\n"
		"		break;\n"
		"	}\n"
		"\n"
		"	// Remove leading zeros\n"
		"	intpart = intpart.replace(/^0*/,'');\n"
		"	if (!intpart)\n"
		"		intpart = '0';\n"
		"\n"
		"	if ((sepStyle & 1) == 0)\n"
		"	{\n"
		"		// Add the thousands sepearators: pad to length multiple of 3 with zeros,\n"
		"		// split into 3s, join with separator, and remove the leading zeros\n"
		"		intpart = new Array(2-(intpart.length+2)%3+1).join('0') + intpart;\n"
		"		intpart = intpart.match(/.../g).join(separator).replace(/^0*/,'');\n"
		"	}\n"
		"\n"
		"	if (!intpart)\n"
		"		intpart = '0';\n"
		"\n"
		"	// Adjust fractional part to correct number of decimal places\n"
		"	fracpart += new Array(nDec+1).join('0');\n"
		"	fracpart = fracpart.substr(0,nDec);\n"
		"\n"
		"	if (fracpart)\n"
		"		intpart += point+fracpart;\n"
		"\n"
		"	if (bCurrencyPrepend)\n"
		"		intpart = strCurrency+intpart;\n"
		"	else\n"
		"		intpart += strCurrency;\n"
		"\n"
		"	if (/-/.test(val))\n"
		"	{\n"
		"		switch (negStyle)\n"
		"		{\n"
		"		case 0:\n"
		"			intpart = '-'+intpart;\n"
		"			break;\n"
		"		case 1:\n"
		"			break;\n"
		"		case 2:\n"
		"		case 3:\n"
		"			intpart = '('+intpart+')';\n"
		"			break;\n"
		"		}\n"
		"	}\n"
		"\n"
		"	if (negStyle&1)\n"
		"		event.target.textColor = /-/.text(val) ? color.red : color.black;\n"
		"\n"
		"	event.value = intpart;\n"
		"}\n"
		"\n"
		"function AFSimple_Calculate(op, list)\n"
		"{\n"
		"	var res;\n"
		"\n"
		"	switch (op)\n"
		"	{\n"
		"		case 'SUM':\n"
		"			res = 0;\n"
		"			break;\n"
		"		case 'PRD':\n"
		"			res = 1;\n"
		"			break;\n"
		"		case 'AVG':\n"
		"			res = 0;\n"
		"			break;\n"
		"	}\n"
		"\n"
		"	if (typeof list == 'string')\n"
		"		list = list.split(/ *, */);\n"
		"\n"
		"	for (var i = 0; i < list.length; i++)\n"
		"	{\n"
		"		var field = getField(list[i]);\n"
		"		var value = Number(field.value);\n"
		"\n"
		"		switch (op)\n"
		"		{\n"
		"			case 'SUM':\n"
		"				res += value;\n"
		"				break;\n"
		"			case 'PRD':\n"
		"				res *= value;\n"
		"				break;\n"
		"			case 'AVG':\n"
		"				res += value;\n"
		"				break;\n"
		"			case 'MIN':\n"
		"				if (i == 0 || value < res)\n"
		"					res = value;\n"
		"				break;\n"
		"			case 'MAX':\n"
		"				if (i == 0 || value > res)\n"
		"					res = value;\n"
		"				break;\n"
		"		}\n"
		"	}\n"
		"\n"
		"	if (op == 'AVG')\n"
		"		res /= list.length;\n"
		"\n"
		"	event.value = res;\n"
		"}\n");
}

pdf_js *pdf_new_js(pdf_document *doc)
{
	fz_context *ctx = doc->ctx;
	pdf_js     *js = NULL;
	pdf_obj    *javascript = NULL;
	fz_buffer  *fzbuf = NULL;

	fz_var(js);
	fz_var(javascript);
	fz_var(fzbuf);
	fz_try(ctx)
	{
		int len, i;
		pdf_obj *root, *acroform;
		js = fz_malloc_struct(ctx, pdf_js);
		js->doc = doc;

		/* Find the form array */
		root = pdf_dict_gets(doc->trailer, "Root");
		acroform = pdf_dict_gets(root, "AcroForm");
		js->form = pdf_dict_gets(acroform, "Fields");

		/* Initialise the javascript engine, passing the main context
		 * for use in memory allocation and exception handling. Also
		 * pass our js context, for it to pass back to us. */
		js->imp = pdf_new_jsimp(ctx, js);
		declare_dom(js);

		preload_helpers(js);

		javascript = pdf_load_name_tree(doc, "JavaScript");
		len = pdf_dict_len(javascript);

		for (i = 0; i < len; i++)
		{
			pdf_obj *fragment = pdf_dict_get_val(javascript, i);
			pdf_obj *code = pdf_dict_gets(fragment, "JS");

			if (pdf_is_stream(doc, pdf_to_num(code), pdf_to_gen(code)))
			{
				unsigned char *buf;
				int len;

				fz_try(ctx)
				{
					fzbuf = pdf_load_stream(doc, pdf_to_num(code), pdf_to_gen(code));
					len = fz_buffer_storage(ctx, fzbuf, &buf);
					pdf_jsimp_execute_count(js->imp, (char *)buf, len);
				}
				fz_always(ctx)
				{
					fz_drop_buffer(ctx, fzbuf);
					fzbuf = NULL;
				}
				fz_catch(ctx)
				{
					fz_warn(ctx, "Warning: %s", ctx->error->message);
				}
			}
		}
	}
	fz_always(ctx)
	{
		pdf_drop_obj(javascript);
	}
	fz_catch(ctx)
	{
		pdf_drop_js(js);
		js = NULL;
	}

	return js;
}

void pdf_drop_js(pdf_js *js)
{
	if (js)
	{
		fz_context *ctx = js->doc->ctx;
		fz_free(ctx, js->event.value);
		pdf_jsimp_drop_type(js->imp, js->fieldtype);
		pdf_jsimp_drop_type(js->imp, js->doctype);
		pdf_drop_jsimp(js->imp);
		fz_free(ctx, js);
	}
}

void pdf_js_setup_event(pdf_js *js, pdf_obj *target)
{
	if (js)
	{
		fz_context *ctx = js->doc->ctx;
		char *val = pdf_field_getValue(js->doc, target);

		js->event.target = target;

		fz_free(ctx, js->event.value);
		js->event.value = NULL;
		js->event.value = fz_strdup(ctx, val?val:"");
	}
}

char *pdf_js_getEventValue(pdf_js *js)
{
	return js ? js->event.value : NULL;
}

void pdf_js_execute(pdf_js *js, char *code)
{
	if (js)
	{
		fz_context *ctx = js->doc->ctx;
		fz_try(ctx)
		{
			pdf_jsimp_execute(js->imp, code);
		}
		fz_catch(ctx)
		{
		}
	}
}

void pdf_js_execute_count(pdf_js *js, char *code, int count)
{
	if (js)
	{
		fz_context *ctx = js->doc->ctx;
		fz_try(ctx)
		{
			pdf_jsimp_execute_count(js->imp, code, count);
		}
		fz_catch(ctx)
		{
		}
	}
}