shithub: libmujs

ref: 32c0c6c4103e8753ac8ad8ca581bec9c87362ca0
dir: /jsbfunction.c/

#include "js.h"
#include "jscompile.h"
#include "jsobject.h"
#include "jsrun.h"
#include "jsstate.h"

static int jsB_new_Function(js_State *J, int n) { return 0; }
static int jsB_Function(js_State *J, int n) { return 0; }

static int jsB_Function_prototype(js_State *J, int n)
{
	js_pushundefined(J);
	return 1;
}

static int Fp_toString(js_State *J, int nargs)
{
	js_Object *self = js_toobject(J, 0);
	char *s;
	int i, n;

	if (!js_iscallable(J, 0))
		jsR_error(J, "TypeError");

	if (self->type == JS_CFUNCTION || self->type == JS_CSCRIPT) {
		js_Function *F = self->u.f.function;
		n = strlen("function () { ... }");
		n += strlen(F->name);
		for (i = 0; i < F->numparams; i++)
			n += strlen(F->params[i]) + 1;
		s = malloc(n);
		strcpy(s, "function ");
		strcat(s, F->name);
		strcat(s, "(");
		for (i = 0; i < F->numparams; i++) {
			if (i > 0) strcat(s, ",");
			strcat(s, F->params[i]);
		}
		strcat(s, ") { ... }");
		js_pushstring(J, s);
		free(s);
	} else {
		js_pushliteral(J, "function () { ... }");
	}

	return 1;
}

static int Fp_apply(js_State *J, int n)
{
	int i, argc;
	char name[20];

	if (!js_iscallable(J, 0))
		jsR_error(J, "TypeError");
	js_copy(J, 0);

	if (js_isundefined(J, 1) || js_isnull(J, 1))
		js_pushglobal(J);
	else
		js_copy(J, 1);

	js_getproperty(J, 2, "length");
	argc = js_tonumber(J, -1);
	js_pop(J, 1);

	for (i = 0; i < argc; ++i) {
		sprintf(name, "%d", i);
		js_getproperty(J, 2, name);
	}

	js_call(J, argc);
	return 1;
}

static int Fp_call(js_State *J, int n)
{
	int i;

	if (!js_iscallable(J, 0))
		jsR_error(J, "TypeError");
	js_copy(J, 0);

	if (js_isundefined(J, 1) || js_isnull(J, 1))
		js_pushglobal(J);
	else
		js_copy(J, 1);

	for (i = 1; i < n; ++i)
		js_copy(J, i + 1);

	js_call(J, n - 1);
	return 1;
}

void jsB_initfunction(js_State *J)
{
	J->Function_prototype->u.c.function = jsB_Function_prototype;
	J->Function_prototype->u.c.constructor = NULL;

	js_pushobject(J, jsR_newcconstructor(J, jsB_Function, jsB_new_Function));
	{
		jsB_propn(J, "length", 1);

		js_pushobject(J, J->Function_prototype);
		{
			js_copy(J, -2);
			js_setproperty(J, -2, "constructor");
			jsB_propf(J, "toString", Fp_toString, 2);
			jsB_propf(J, "apply", Fp_apply, 2);
			jsB_propf(J, "call", Fp_call, 1);
		}
		js_setproperty(J, -2, "prototype");
	}
	js_setglobal(J, "Function");
}