gecko-dev/dom/tools/JSStubGen.cpp

2111 строки
60 KiB
C++

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
*
* The contents of this file are subject to the Netscape Public
* License Version 1.1 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of
* the License at http://www.mozilla.org/NPL/
*
* Software distributed under the License is distributed on an "AS
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
* implied. See the License for the specific language governing
* rights and limitations under the License.
*
* The Original Code is mozilla.org code.
*
* The Initial Developer of the Original Code is Netscape
* Communications Corporation. Portions created by Netscape are
* Copyright (C) 1998 Netscape Communications Corporation. All
* Rights Reserved.
*
* Contributor(s):
* Pierre Phaneuf <pp@ludusdesign.com>
*/
#ifndef XP_MAC
#include <sys/types.h>
#include <sys/stat.h>
#else
#include <stat.h>
#include <stddef.h>
#endif
#if !defined XP_UNIX && !defined XP_MAC && ! defined XP_BEOS
#include <direct.h>
#endif
#include <fstream.h>
#include <ctype.h>
#include "nsIPtr.h"
#include "plhash.h"
#include "JSStubGen.h"
#include "Exceptions.h"
#include "IdlSpecification.h"
#include "IdlInterface.h"
#include "IdlVariable.h"
#include "IdlAttribute.h"
#include "IdlFunction.h"
#include "IdlParameter.h"
static const char kFilePrefix[] = "nsJS";
static const char kFileSuffix[] = "cpp";
JSStubGen::JSStubGen()
{
}
JSStubGen::~JSStubGen()
{
}
void
JSStubGen::Generate(char *aFileName,
char *aOutputDirName,
IdlSpecification &aSpec,
int aIsGlobal)
{
IdlInterface *iface = aSpec.GetInterfaceAt(0);
if (!OpenFile(iface->GetName(), aOutputDirName, kFilePrefix, kFileSuffix)) {
throw new CantOpenFileException(aFileName);
}
mIsGlobal = aIsGlobal;
GenerateNPL();
GenerateIncludes(aSpec);
GenerateIIDDefinitions(aSpec);
#ifndef USE_COMPTR
GenerateDefPtrs(aSpec);
#endif
GeneratePropertySlots(aSpec);
GeneratePropertyFunc(aSpec, PR_TRUE);
GeneratePropertyFunc(aSpec, PR_FALSE);
GenerateCustomPropertyFuncs(aSpec);
GenerateFinalize(aSpec);
GenerateEnumerate(aSpec);
GenerateResolve(aSpec);
GenerateMethods(aSpec);
GenerateJSClass(aSpec);
GenerateClassProperties(aSpec);
GenerateClassFunctions(aSpec);
GenerateConstructor(aSpec);
GenerateInitClass(aSpec);
GenerateNew(aSpec);
CloseFile();
}
static const char kIncludeDefaultsStr[] = "\n"
"#include \"jsapi.h\"\n"
"#include \"nsJSUtils.h\"\n"
"#include \"nsDOMError.h\"\n"
"#include \"nscore.h\"\n"
"#include \"nsIServiceManager.h\"\n"
"#include \"nsIScriptContext.h\"\n"
"#include \"nsIScriptSecurityManager.h\"\n"
"#include \"nsIJSScriptObject.h\"\n"
"#include \"nsIScriptObjectOwner.h\"\n"
"#include \"nsIScriptGlobalObject.h\"\n"
"#include \"nsCOMPtr.h\"\n"
"#include \"nsDOMPropEnums.h\"\n"
#ifndef USE_COMPTR
"#include \"nsIPtr.h\"\n"
#endif
"#include \"nsString.h\"\n";
static const char kIncludeStr[] = "#include \"nsIDOM%s.h\"\n";
static const char kXPIDLIncludeStr[] = "#include \"%s.h\"\n";
static const char kIncludeConstructorStr[] =
"#include \"nsIScriptNameSpaceManager.h\"\n"
"#include \"nsIComponentManager.h\"\n"
"#include \"nsIJSNativeInitializer.h\"\n"
"#include \"nsDOMCID.h\"\n";
static PRIntn
IncludeEnumerator(PLHashEntry *he, PRIntn i, void *arg)
{
char buf[512];
switch ((Type)(int)(he->value)) {
case TYPE_OBJECT:
case TYPE_FUNC:
sprintf(buf, kIncludeStr, (char *)he->key);
break;
case TYPE_XPIDL_OBJECT:
sprintf(buf, kXPIDLIncludeStr, (char *)he->key);
break;
default:
// uh oh...
break;
}
ofstream *file = (ofstream *)arg;
*file << buf;
return HT_ENUMERATE_NEXT;
}
void
JSStubGen::GenerateIncludes(IdlSpecification &aSpec)
{
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
*file << kIncludeDefaultsStr;
EnumerateAllObjects(aSpec, (PLHashEnumerator)IncludeEnumerator,
file, PR_FALSE);
if (HasConstructor(*primary_iface, NULL)) {
*file << kIncludeConstructorStr;
}
*file << "\n";
}
static const char kIIDDefaultStr[] = "\n"
"static NS_DEFINE_IID(kIScriptObjectOwnerIID, NS_ISCRIPTOBJECTOWNER_IID);\n"
"static NS_DEFINE_IID(kIJSScriptObjectIID, NS_IJSSCRIPTOBJECT_IID);\n"
"static NS_DEFINE_IID(kIScriptGlobalObjectIID, NS_ISCRIPTGLOBALOBJECT_IID);\n";
static const char kIIDStr[] = "static NS_DEFINE_IID(kI%sIID, %s);\n";
PRIntn
JSStubGen_IIDEnumerator(PLHashEntry *he, PRIntn i, void *arg)
{
char buf[512];
char iid_buf[256];
JSStubGen *me = (JSStubGen *)arg;
ofstream *file = me->GetFile();
switch ((Type)(int)(he->value)) {
case TYPE_OBJECT:
case TYPE_FUNC:
me->GetInterfaceIID(iid_buf, (char *)he->key);
sprintf(buf, kIIDStr, (char *)he->key, iid_buf);
break;
case TYPE_XPIDL_OBJECT:
{
// This sucks. I know.
char* p = (char*) he->key;
if (p[0] == 'n' && p[1] == 's' && p[2] == 'I')
p += 3;
me->GetXPIDLInterfaceIID(iid_buf, p);
sprintf(buf, kIIDStr, p, iid_buf);
break;
}
}
*file << buf;
return HT_ENUMERATE_NEXT;
}
void
JSStubGen::GenerateIIDDefinitions(IdlSpecification &aSpec)
{
ofstream *file = GetFile();
*file << kIIDDefaultStr;
EnumerateAllObjects(aSpec, (PLHashEnumerator)JSStubGen_IIDEnumerator,
this, PR_FALSE);
*file << "\n";
}
static const char kDefPtrStr[] =
"NS_DEF_PTR(nsIDOM%s);\n";
static const char kDefXPIDLPtrStr[] =
"NS_DEF_PTR(%s);\n";
PRIntn
JSStubGen_DefPtrEnumerator(PLHashEntry *he, PRIntn i, void *arg)
{
char buf[512];
JSStubGen *me = (JSStubGen *)arg;
ofstream *file = me->GetFile();
switch ((Type)(int)(he->value)) {
case TYPE_OBJECT:
case TYPE_FUNC:
sprintf(buf, kDefPtrStr, (char *)he->key);
break;
case TYPE_XPIDL_OBJECT:
sprintf(buf, kDefXPIDLPtrStr, (char *)he->key);
break;
}
*file << buf;
return HT_ENUMERATE_NEXT;
}
void
JSStubGen::GenerateDefPtrs(IdlSpecification &aSpec)
{
ofstream *file = GetFile();
EnumerateAllObjects(aSpec, (PLHashEnumerator)JSStubGen_DefPtrEnumerator,
this, PR_FALSE);
*file << "\n";
}
static const char kPropEnumStr[] =
"//\n"
"// %s property ids\n"
"//\n"
"enum %s_slots {\n";
static const char kPropSlotStr[] = " %s_%s = -%d";
void
JSStubGen::GeneratePropertySlots(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
int any_props = 0;
int prop_counter = 0;
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
strcpy(iface_name, iface->GetName());
StrUpr(iface_name);
int a, acount = iface->AttributeCount();
for (a = 0; a < acount; a++) {
IdlAttribute *attr = iface->GetAttributeAt(a);
char attr_name[128];
if (attr->GetIsNoScript()) {
continue;
}
// If this is the first time...
if (!any_props) {
sprintf(buf, kPropEnumStr, iface->GetName(), iface->GetName());
*file << buf;
}
// If this is the first time for this interface...
else if (a == 0) {
*file << ",\n";
}
any_props = 1;
strcpy(attr_name, attr->GetName());
StrUpr(attr_name);
sprintf(buf, kPropSlotStr, iface_name, attr_name, ++prop_counter);
*file << buf;
if (a != acount-1) {
*file << ",\n";
}
}
}
if (any_props) {
*file << "\n};\n";
}
}
static const char kPropFuncBeginStr[] = "\n"
"/***********************************************************************/\n"
"//\n"
"// %s Properties %ster\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"%s%sProperty(JSContext *cx, JSObject *obj, jsval id, jsval *vp)\n"
"{\n"
" nsIDOM%s *a = (nsIDOM%s*)nsJSUtils::nsGetNativeThis(cx, obj);\n"
"\n"
" // If there's no private data, this must be the prototype, so ignore\n"
" if (nsnull == a) {\n"
" return JS_TRUE;\n"
" }\n"
"\n";
static const char kIntCaseStr[] =
" nsresult rv = NS_OK;\n"
" if (JSVAL_IS_INT(id)) {\n"
" nsIScriptSecurityManager *secMan = nsJSUtils::nsGetSecurityManager(cx, obj);\n"
" if (!secMan)\n"
" return PR_FALSE;\n"
" switch(JSVAL_TO_INT(id)) {\n";
static const char kIntCaseNamedItemStr[] =
" PRBool checkNamedItem = PR_TRUE;\n"
" nsresult rv = NS_OK;\n"
" if (JSVAL_IS_INT(id)) {\n"
" nsIScriptSecurityManager *secMan = nsJSUtils::nsGetSecurityManager(cx, obj);\n"
" if (!secMan)\n"
" return PR_FALSE;\n"
" checkNamedItem = PR_FALSE;\n"
" switch(JSVAL_TO_INT(id)) {\n";
static const char kPropFuncDefaultStr[] =
" default:\n"
" return nsJSUtils::nsCallJSScriptObject%sProperty(a, cx, obj, id, vp);\n"
" }\n"
" }\n";
static const char kPropFuncDefaultNamedItemStr[] =
" default:\n"
" checkNamedItem = PR_TRUE;\n"
" }\n"
" }\n";
static const char kPropFuncDefaultItemStr[] =
" default:\n"
" {\n"
" %s prop;\n"
" rv = a->Item(JSVAL_TO_INT(id), %sprop);\n"
" if (NS_SUCCEEDED(rv)) {\n"
"%s"
" }\n"
" }\n"
" }\n"
" }\n";
static const char kPropFuncDefaultItemEllipsisStr[] =
" default:\n"
" {\n"
" rv = a->Item(cx, &id, 1, vp);\n"
" }\n"
" }\n"
" }\n";
static const char kPropFuncDefaultItemNonPrimaryStr[] =
" default:\n"
" {\n"
" %s prop;\n"
" nsIDOM%s* b;\n"
" if (NS_OK == a->QueryInterface(kI%sIID, (void **)&b)) {\n"
" nsresult result = NS_OK;\n"
" rv = b->Item(JSVAL_TO_INT(id), %sprop);\n"
" if (NS_SUCCEEDED(rv)) {\n"
"%s"
" }\n"
" NS_RELEASE(b);\n"
" }\n"
" else {\n"
" rv = NS_ERROR_DOM_WRONG_TYPE_ERR;\n"
" }\n"
" }\n"
" }\n"
" }\n";
static const char kPropFuncDefaultItemEllipsisNonPrimaryStr[] =
" default:\n"
" {\n"
" nsIDOM%s* b;\n"
" if (NS_OK == a->QueryInterface(kI%sIID, (void **)&b)) {\n"
" rv = b->Item(cx, &id, 1, vp);\n"
" NS_RELEASE(b);\n"
" }\n"
" else {\n"
" rv = NS_ERROR_DOM_WRONG_TYPE_ERR;\n"
" }\n"
" }\n"
" }\n"
" }\n";
static const char kPropFuncEndStr[] =
" else {\n"
" return nsJSUtils::nsCallJSScriptObject%sProperty(a, cx, obj, id, vp);\n"
" }\n"
"\n"
" if (NS_FAILED(rv))\n"
" return nsJSUtils::nsReportError(cx, obj, rv);\n"
" return PR_TRUE;\n"
"}\n";
static const char kPropFuncNamedItemStr[] =
"\n"
" if (checkNamedItem) {\n"
" %s prop;\n"
" nsAutoString name;\n"
" nsresult result = NS_OK;\n"
"\n"
" JSString *jsstring = JS_ValueToString(cx, id);\n"
" if (nsnull != jsstring) {\n"
" name.Assign(JS_GetStringChars(jsstring));\n"
" }\n"
" else {\n"
" name.SetLength(0);\n"
" }\n"
"\n"
" result = a->NamedItem(name, %sprop);\n"
" if (NS_SUCCEEDED(result)) {\n"
" if (NULL != prop) {\n"
"%s"
" }\n"
" else {\n"
" return nsJSUtils::nsCallJSScriptObject%sProperty(a, cx, obj, id, vp);\n"
" }\n"
" }\n"
" else {\n"
" return nsJSUtils::nsReportError(cx, obj, result);\n"
" }\n"
" }\n";
static const char kPropFuncNamedItemEllipsisStr[] =
"\n"
" if (checkNamedItem) {\n"
" rv = a->NamedItem(cx, &id, 1, vp);\n"
" }\n";
static const char kPropFuncNamedItemNonPrimaryStr[] =
"\n"
" if (checkNamedItem) {\n"
" %s prop;\n"
" nsIDOM%s* b;\n"
" nsAutoString name;\n"
"\n"
" JSString *jsstring = JS_ValueToString(cx, id);\n"
" if (nsnull != jsstring) {\n"
" name.Assign(JS_GetStringChars(jsstring));\n"
" }\n"
" else {\n"
" name.SetLength(0);\n"
" }\n"
"\n"
" if (NS_OK == a->QueryInterface(kI%sIID, (void **)&b)) {\n"
" nsresult result = NS_OK;\n"
" result = b->NamedItem(name, %sprop);\n"
" if (NS_SUCCEEDED(result)) {\n"
" NS_RELEASE(b);\n"
" if (NULL != prop) {\n"
"%s"
" }\n"
" else {\n"
" return nsJSUtils::nsCallJSScriptObject%sProperty(a, cx, obj, id, vp);\n"
" }\n"
" }\n"
" else {\n"
" NS_RELEASE(b);\n"
" return nsJSUtils::nsReportError(cx, obj, result);\n"
" }\n"
" }\n"
" else {\n"
" return nsJSUtils::nsReportError(cx, obj, NS_ERROR_DOM_WRONG_TYPE_ERR);\n"
" }\n"
" }\n";
static const char kPropFuncNamedItemEllipsisNonPrimaryStr[] =
"\n"
" if (checkNamedItem) {\n"
" nsIDOM%s* b;\n"
" nsresult result = NS_OK;\n"
" if (NS_OK == a->QueryInterface(kI%sIID, (void **)&b)) {\n"
" result = b->NamedItem(cx, &id, 1, vp);\n"
" NS_RELEASE(b);\n"
" if (NS_FAILED(result)) {\n"
" return nsJSUtils::nsReportError(cx, obj, result);\n"
" }\n"
" }\n"
" else {\n"
" return nsJSUtils::nsReportError(cx, obj, NS_ERROR_DOM_WRONG_TYPE_ERR);\n"
" }\n"
" }\n";
#define JSGEN_GENERATE_PROPFUNCBEGIN(buffer, op, className) \
sprintf(buffer, kPropFuncBeginStr, className, op, op, className, className, className)
#define JSGEN_GENERATE_PROPFUNCEND(buffer, op) \
sprintf(buffer, kPropFuncEndStr, op)
#define JSGEN_GENERATE_PROPFUNCDEFAULT(buffer, op) \
sprintf(buffer, kPropFuncDefaultStr, op)
static const char kPropCaseBeginStr[] =
" case %s_%s:\n"
" {\n"
" rv = secMan->CheckScriptAccess(cx, obj, NS_DOM_PROP_%s_%s, %s);\n"
" if (NS_SUCCEEDED(rv)) {\n";
static const char kPropCaseEndStr[] =
" }\n"
" break;\n"
" }\n";
static const char kNoAttrStr[] = " case 0:\n";
void
JSStubGen::GeneratePropertyFunc(IdlSpecification &aSpec, PRBool aIsGetter)
{
char buf[1024];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
PRBool any = PR_FALSE;
JSGEN_GENERATE_PROPFUNCBEGIN(buf, aIsGetter ? "Get" : "Set", primary_iface->GetName());
*file << buf;
IdlFunction *item_func = NULL;
IdlFunction *named_item_func = NULL;
IdlInterface *item_iface = NULL;
IdlInterface *named_item_iface = NULL;
PRBool named_item_has_ellipsis = PR_FALSE;
PRBool item_has_ellipsis = PR_FALSE;
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
int m, mcount = iface->FunctionCount();
for (m = 0; m < mcount; m++) {
IdlFunction *func = iface->GetFunctionAt(m);
if (strcmp(func->GetName(), "item") == 0) {
item_func = func;
item_iface = iface;
item_has_ellipsis = func->GetHasEllipsis();
}
else if (strcmp(func->GetName(), "namedItem") == 0) {
named_item_func = func;
named_item_iface = iface;
named_item_has_ellipsis = func->GetHasEllipsis();
}
}
}
if (NULL == named_item_func) {
*file << kIntCaseStr;
}
else {
*file << kIntCaseNamedItemStr;
}
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
strcpy(iface_name, iface->GetName());
StrUpr(iface_name);
int a, acount = iface->AttributeCount();
for (a = 0; a < acount; a++) {
IdlAttribute *attr = iface->GetAttributeAt(a);
char attr_name[128];
strcpy(attr_name, attr->GetName());
StrUpr(attr_name);
if (attr->GetIsNoScript() || attr->GetReplaceable() || (!aIsGetter && (attr->GetReadOnly()))) {
continue;
}
any = PR_TRUE;
char upr_attr_name[128];
strcpy(upr_attr_name, attr_name);
StrUpr(upr_attr_name);
char upr_iface_name[128];
strcpy(upr_iface_name, iface_name);
StrUpr(upr_iface_name);
if (aIsGetter) {
sprintf(buf, kPropCaseBeginStr, iface_name, attr_name, upr_iface_name, upr_attr_name, "PR_FALSE");
}
else {
sprintf(buf, kPropCaseBeginStr, iface_name, attr_name, upr_iface_name, upr_attr_name, "PR_TRUE");
}
*file << buf;
if (aIsGetter) {
GeneratePropGetter(file, *iface, *attr,
iface == primary_iface ? JSSTUBGEN_PRIMARY : JSSTUBGEN_NONPRIMARY);
}
else {
GeneratePropSetter(file, *iface, *attr, iface == primary_iface);
}
sprintf(buf, kPropCaseEndStr);
*file << buf;
}
}
if (!any) {
*file << kNoAttrStr;
}
if (aIsGetter) {
if (NULL != item_func) {
IdlVariable *rval = item_func->GetReturnValue();
if (item_has_ellipsis) {
GeneratePropGetter(file, *item_iface, *rval,
item_iface == primary_iface ?
JSSTUBGEN_DEFAULT_ELLIPSIS : JSSTUBGEN_DEFAULT_NONPRIMARY_ELLIPSIS);
}
else {
GeneratePropGetter(file, *item_iface, *rval,
item_iface == primary_iface ?
JSSTUBGEN_DEFAULT : JSSTUBGEN_DEFAULT_NONPRIMARY);
}
}
else if (NULL != named_item_func) {
*file << kPropFuncDefaultNamedItemStr;
}
else {
JSGEN_GENERATE_PROPFUNCDEFAULT(buf, aIsGetter ? "Get" : "Set");
*file << buf;
}
}
else {
JSGEN_GENERATE_PROPFUNCDEFAULT(buf, aIsGetter ? "Get" : "Set");
*file << buf;
}
if (aIsGetter && (NULL != named_item_func)) {
IdlVariable *rval = named_item_func->GetReturnValue();
if (named_item_has_ellipsis) {
GeneratePropGetter(file, *named_item_iface, *rval,
named_item_iface == primary_iface ?
JSSTUBGEN_NAMED_ITEM_ELLIPSIS : JSSTUBGEN_NAMED_ITEM_NONPRIMARY_ELLIPSIS);
}
else {
GeneratePropGetter(file, *named_item_iface, *rval,
named_item_iface == primary_iface ?
JSSTUBGEN_NAMED_ITEM : JSSTUBGEN_NAMED_ITEM_NONPRIMARY);
}
}
if (aIsGetter) {
JSGEN_GENERATE_PROPFUNCEND(buf, "Get");
}
else {
JSGEN_GENERATE_PROPFUNCEND(buf, "Set");
}
*file << buf;
}
static const char kGetCaseStr[] =
" %s prop;\n"
" rv = a->Get%s(%sprop);\n"
" if (NS_SUCCEEDED(rv)) {\n"
"%s"
" }\n";
static const char kGetCaseNonPrimaryStr[] =
" %s prop;\n"
" nsIDOM%s* b;\n"
" if (NS_OK == a->QueryInterface(kI%sIID, (void **)&b)) {\n"
" rv = b->Get%s(%sprop);\n"
" if(NS_SUCCEEDED(rv)) {\n"
"%s"
" }\n"
" NS_RELEASE(b);\n"
" }\n"
" else {\n"
" rv = NS_ERROR_DOM_WRONG_TYPE_ERR;\n"
" }\n";
static const char kObjectGetCaseStr[] =
" // get the js object\n"
" nsJSUtils::nsConvertObjectToJSVal((nsISupports *)prop, cx, obj, vp);\n";
static const char kXPIDLObjectGetCaseStr[] =
" // get the js object; n.b., this will do a release on 'prop'\n"
" nsJSUtils::nsConvertXPCObjectToJSVal(prop, NS_GET_IID(%s), cx, obj, vp);\n";
static const char kStringGetCaseStr[] =
" nsJSUtils::nsConvertStringToJSVal(prop, cx, vp);\n";
static const char kIntGetCaseStr[] =
" *vp = INT_TO_JSVAL(prop);\n";
static const char kLongLongGetCaseStr[] =
"#ifdef HAVE_LONG_LONG\n"
" *vp = INT_TO_JSVAL(prop);\n"
"#else\n"
" int i;\n"
" LL_L2UI(i, prop);\n"
" *vp = INT_TO_JSVAL(i);\n"
"#endif\n";
static const char kFloatGetCaseStr[] =
" *vp = DOUBLE_TO_JSVAL(JS_NewDouble(cx, prop));\n";
static const char kBoolGetCaseStr[] =
" *vp = BOOLEAN_TO_JSVAL(prop);\n";
static const char kJSValGetCaseStr[] =
" *vp = prop;\n";
void
JSStubGen::GeneratePropGetter(ofstream *file,
IdlInterface &aInterface,
IdlVariable &aAttribute,
PRInt32 aType)
{
char buf[2048];
char buf2[1024];
char attr_type[128];
char attr_name[128];
const char *case_str;
GetVariableTypeForLocal(attr_type, aAttribute);
if ((JSSTUBGEN_PRIMARY == aType) || (JSSTUBGEN_NONPRIMARY == aType)) {
GetCapitalizedName(attr_name, aAttribute);
}
switch (aAttribute.GetType()) {
case TYPE_BOOLEAN:
case_str = kBoolGetCaseStr;
break;
case TYPE_LONG:
case TYPE_SHORT:
case TYPE_ULONG:
case TYPE_USHORT:
case TYPE_CHAR:
case TYPE_INT:
case TYPE_UINT:
case_str = kIntGetCaseStr;
break;
case TYPE_LONG_LONG:
case TYPE_ULONG_LONG:
case_str = kLongLongGetCaseStr;
break;
case TYPE_FLOAT:
case_str = kFloatGetCaseStr;
break;
case TYPE_JSVAL:
case_str = kJSValGetCaseStr;
break;
case TYPE_STRING:
case_str = kStringGetCaseStr;
break;
case TYPE_OBJECT:
case_str = kObjectGetCaseStr;
break;
case TYPE_XPIDL_OBJECT:
case_str = buf2;
sprintf(buf2, kXPIDLObjectGetCaseStr, aAttribute.GetTypeName(), aAttribute.GetTypeName());
break;
default:
// XXX Fail for other cases
break;
}
if (JSSTUBGEN_PRIMARY == aType) {
sprintf(buf, kGetCaseStr, attr_type, attr_name,
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str);
}
else if (JSSTUBGEN_NONPRIMARY == aType) {
sprintf(buf, kGetCaseNonPrimaryStr, attr_type,
aInterface.GetName(), aInterface.GetName(),
attr_name,
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str);
}
else if (JSSTUBGEN_DEFAULT == aType) {
sprintf(buf, kPropFuncDefaultItemStr, attr_type,
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str);
}
else if (JSSTUBGEN_DEFAULT_NONPRIMARY == aType) {
sprintf(buf, kPropFuncDefaultItemNonPrimaryStr, attr_type,
aInterface.GetName(), aInterface.GetName(),
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str);
}
else if (JSSTUBGEN_DEFAULT_ELLIPSIS == aType) {
sprintf(buf, kPropFuncDefaultItemEllipsisStr);
}
else if (JSSTUBGEN_DEFAULT_NONPRIMARY_ELLIPSIS == aType) {
sprintf(buf, kPropFuncDefaultItemEllipsisNonPrimaryStr,
aInterface.GetName(), aInterface.GetName());
}
else if (JSSTUBGEN_NAMED_ITEM == aType) {
sprintf(buf, kPropFuncNamedItemStr, attr_type,
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str, "Get");
}
else if (JSSTUBGEN_NAMED_ITEM_NONPRIMARY == aType) {
sprintf(buf, kPropFuncNamedItemNonPrimaryStr, attr_type,
aInterface.GetName(), aInterface.GetName(),
aAttribute.GetType() == TYPE_STRING ? "" : "&",
case_str, "Get");
}
else if (JSSTUBGEN_NAMED_ITEM_ELLIPSIS == aType) {
sprintf(buf, kPropFuncNamedItemEllipsisStr);
}
else if (JSSTUBGEN_NAMED_ITEM_NONPRIMARY_ELLIPSIS == aType) {
sprintf(buf, kPropFuncNamedItemEllipsisNonPrimaryStr,
aInterface.GetName(), aInterface.GetName());
}
*file << buf;
}
static const char kSetCaseStr[] =
" %s prop;\n"
"%s \n"
" rv = a->Set%s(prop);\n"
" %s\n";
static const char kSetCaseNonPrimaryStr[] =
" %s prop;\n"
"%s \n"
" nsIDOM%s *b;\n"
" if (NS_OK == a->QueryInterface(kI%sIID, (void **)&b)) {\n"
" b->Set%s(prop);\n"
" NS_RELEASE(b);\n"
" }\n"
" else {\n"
" %s\n"
" rv = NS_ERROR_DOM_WRONG_TYPE_ERR;\n"
" }\n"
" %s\n";
static const char kObjectSetCaseStr[] =
" if (PR_FALSE == nsJSUtils::nsConvertJSValToObject((nsISupports **)&prop,\n"
" kI%sIID, NS_ConvertASCIItoUCS2(\"%s\"),\n"
" cx, *vp)) {\n"
" rv = NS_ERROR_DOM_NOT_OBJECT_ERR;\n"
" break;\n"
" }\n";
static const char kXPIDLObjectSetCaseStr[] =
" if (PR_FALSE == nsJSUtils::nsConvertJSValToXPCObject((nsISupports **) &prop,\n"
" kI%sIID, cx, *vp)) {\n"
" rv = NS_ERROR_DOM_NOT_XPC_OBJECT_ERR;\n"
" break;\n"
" }\n";
static const char kObjectSetCaseEndStr[] = "NS_IF_RELEASE(prop);";
static const char* kXPIDLObjectSetCaseEndStr = kObjectSetCaseEndStr;
static const char kStringSetCaseStr[] =
" nsJSUtils::nsConvertJSValToString(prop, cx, *vp);\n";
static const char kIntSetCaseStr[] =
" int32 temp;\n"
" if (JSVAL_IS_NUMBER(*vp) && JS_ValueToInt32(cx, *vp, &temp)) {\n"
" prop = (%s)temp;\n"
" }\n"
" else {\n"
" rv = NS_ERROR_DOM_NOT_NUMBER_ERR;\n"
" break;\n"
" }\n";
static const char kBoolSetCaseStr[] =
" if (PR_FALSE == nsJSUtils::nsConvertJSValToBool(&prop, cx, *vp)) {\n"
" rv = NS_ERROR_DOM_NOT_BOOLEAN_ERR;\n"
" break;\n"
" }\n";
static const char kJSValSetCaseStr[] =
" prop = *vp;\n";
void
JSStubGen::GeneratePropSetter(ofstream *file,
IdlInterface &aInterface,
IdlAttribute &aAttribute,
PRBool aIsPrimary)
{
char buf[1024];
char attr_type[128];
char attr_name[128];
char case_buf[1024];
const char *end_str;
GetVariableTypeForLocal(attr_type, aAttribute);
GetCapitalizedName(attr_name, aAttribute);
switch (aAttribute.GetType()) {
case TYPE_BOOLEAN:
sprintf(case_buf, kBoolSetCaseStr);
break;
case TYPE_LONG:
case TYPE_LONG_LONG:
case TYPE_SHORT:
case TYPE_ULONG:
case TYPE_ULONG_LONG:
case TYPE_USHORT:
case TYPE_CHAR:
case TYPE_INT:
case TYPE_UINT:
sprintf(case_buf, kIntSetCaseStr, attr_type);
break;
case TYPE_JSVAL:
strcpy(case_buf, kJSValSetCaseStr);
break;
case TYPE_STRING:
strcpy(case_buf, kStringSetCaseStr);
break;
case TYPE_OBJECT:
sprintf(case_buf, kObjectSetCaseStr, aAttribute.GetTypeName(), aAttribute.GetTypeName());
break;
case TYPE_XPIDL_OBJECT:
// Yeah, this sucks. That's life.
{
char* p = aAttribute.GetTypeName();
if (p[0] == 'n' && p[1] == 's' && p[2] == 'I')
p += 3;
sprintf(case_buf, kXPIDLObjectSetCaseStr, p);
}
break;
default:
// XXX Fail for other cases
break;
}
switch (aAttribute.GetType()) {
case TYPE_OBJECT: end_str = kObjectSetCaseEndStr; break;
case TYPE_XPIDL_OBJECT: end_str = kXPIDLObjectSetCaseEndStr; break;
default: end_str = ""; break;
}
if (aIsPrimary) {
sprintf(buf, kSetCaseStr, attr_type, case_buf, attr_name, end_str);
}
else {
sprintf(buf, kSetCaseNonPrimaryStr, attr_type, case_buf,
aInterface.GetName(), aInterface.GetName(), attr_name,
end_str, end_str);
}
*file << buf;
}
static const char kCustomPropFuncBeginStr[] = "\n"
"/***********************************************************************/\n"
"//\n"
"// %s Property %ster\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"%s%s%ster(JSContext *cx, JSObject *obj, jsval id, jsval *vp)\n"
"{\n"
" nsIDOM%s *a = (nsIDOM%s*)nsJSUtils::nsGetNativeThis(cx, obj);\n"
"\n"
" // If there's no private data, this must be the prototype, so ignore\n"
" if (nsnull == a) {\n"
" return JS_TRUE;\n"
" }\n"
"\n"
" nsresult rv;\n"
" nsIScriptSecurityManager *secMan = nsJSUtils::nsGetSecurityManager(cx, obj);\n"
" if (!secMan)\n"
" return PR_FALSE;\n"
" rv = secMan->CheckScriptAccess(cx, obj, NS_DOM_PROP_%s_%s, %s);\n"
" if (NS_FAILED(rv)) {\n"
" return nsJSUtils::nsReportError(cx, obj, rv);\n"
" }\n"
"\n";
static const char kCustomPropGetterFuncEndStr[] = "\n"
" return PR_TRUE;\n"
"}\n";
static const char kCustomPropSetterFuncEndStr[] = "\n"
" JS_DefineProperty(cx, obj, \"%s\", *vp, nsnull, nsnull, JSPROP_ENUMERATE);\n"
" return PR_TRUE;\n"
"}\n";
void
JSStubGen::GenerateCustomPropertyFuncs(IdlSpecification &aSpec)
{
char buf[1024];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
PRBool any = PR_FALSE;
static char* get_str = "Get";
static char* set_str = "Set";
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
char iface_name_upper[128];
strcpy(iface_name, iface->GetName());
strcpy(iface_name_upper, iface->GetName());
StrUpr(iface_name_upper);
int a, acount = iface->AttributeCount();
for (a = 0; a < acount; a++) {
IdlAttribute *attr = iface->GetAttributeAt(a);
if (attr->GetReplaceable()) {
char attr_name[128];
char attr_name_upper[128];
strcpy(attr_name, attr->GetName());
strcpy(attr_name_upper, attr->GetName());
StrUpr(attr_name_upper);
sprintf(buf, kCustomPropFuncBeginStr, attr_name,
get_str, iface_name, attr_name, get_str,
iface_name, iface_name, iface_name_upper, attr_name_upper,
"PR_FALSE");
*file << buf;
GeneratePropGetter(file, *iface, *attr,
iface == primary_iface ? JSSTUBGEN_PRIMARY : JSSTUBGEN_NONPRIMARY);
*file << kCustomPropGetterFuncEndStr;
sprintf(buf, kCustomPropFuncBeginStr, attr_name,
set_str, iface_name, attr_name, set_str,
iface_name, iface_name, iface_name_upper, attr_name_upper,
"PR_TRUE");
*file << buf;
sprintf(buf, kCustomPropSetterFuncEndStr, attr_name);
*file << buf;
}
}
}
}
static const char kFinalizeStr[] =
"\n\n//\n"
"// %s finalizer\n"
"//\n"
"PR_STATIC_CALLBACK(void)\n"
"Finalize%s(JSContext *cx, JSObject *obj)\n"
"{\n"
" nsJSUtils::nsGenericFinalize(cx, obj);\n"
"}\n";
void
JSStubGen::GenerateFinalize(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
IdlInterface *iface = aSpec.GetInterfaceAt(0);
sprintf(buf, kFinalizeStr, iface->GetName(), iface->GetName());
*file << buf;
}
static const char kEnumerateStr[] =
"\n\n//\n"
"// %s enumerate\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"Enumerate%s(JSContext *cx, JSObject *obj)\n"
"{\n"
" return nsJSUtils::nsGenericEnumerate(cx, obj);\n"
"}\n";
#define JSGEN_GENERATE_ENUMERATE(buf, className) \
sprintf(buf, kEnumerateStr, className, className);
void
JSStubGen::GenerateEnumerate(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
IdlInterface *iface = aSpec.GetInterfaceAt(0);
char *name = iface->GetName();
JSGEN_GENERATE_ENUMERATE(buf, name);
*file << buf;
}
static const char kResolveStr[] =
"\n\n//\n"
"// %s resolve\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"Resolve%s(JSContext *cx, JSObject *obj, jsval id)\n"
"{\n"
"%s"
"}\n";
static const char* kGenericResolveStr =
" return nsJSUtils::nsGenericResolve(cx, obj, id);\n";
static const char* kGlobalResolveStr =
" return nsJSUtils::nsGlobalResolve(cx, obj, id);\n";
#define JSGEN_GENERATE_RESOLVE(buf, className, str) \
sprintf(buf, kResolveStr, className, className, str);
void
JSStubGen::GenerateResolve(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
IdlInterface *iface = aSpec.GetInterfaceAt(0);
char *name = iface->GetName();
if (mIsGlobal) {
JSGEN_GENERATE_RESOLVE(buf, name, kGlobalResolveStr);
}
else {
JSGEN_GENERATE_RESOLVE(buf, name, kGenericResolveStr);
}
*file << buf;
}
static const char kMethodBeginStr[] = "\n\n"
"//\n"
"// Native method %s\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"%s%s(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)\n"
"{\n"
" nsIDOM%s *nativeThis = (nsIDOM%s*)nsJSUtils::nsGetNativeThis(cx, obj);\n"
" nsresult result = NS_OK;\n";
static const char kMethodBeginNonPrimaryStr[] = "\n\n"
"//\n"
"// Native method %s\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"%s%s(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)\n"
"{\n"
" nsIDOM%s *privateThis = (nsIDOM%s*)nsJSUtils::nsGetNativeThis(cx, obj);\n"
#ifdef USE_COMPTR
" nsCOMPtr<nsIDOM%s> nativeThis;\n"
#else
" nsIDOM%sPtr nativeThis = nsnull;\n"
#endif
" nsresult result = NS_OK;\n"
#ifdef USE_COMPTR
" if (NS_OK != privateThis->QueryInterface(kI%sIID, getter_AddRefs(nativeThis))) {\n"
#else
" if (NS_OK != privateThis->QueryInterface(kI%sIID, (void **)&nativeThis)) {\n"
#endif
" return nsJSUtils::nsReportError(cx, obj, NS_ERROR_DOM_WRONG_TYPE_ERR);\n"
" }\n"
"\n";
static const char kMethodReturnStr[] =
" %s nativeRet;\n";
static const char kMethodParamStr[] = " %s b%d;\n";
static const char kMethodBodyBeginStr[] =
" *rval = JSVAL_NULL;\n"
" nsIScriptSecurityManager *secMan = nsJSUtils::nsGetSecurityManager(cx, obj);\n"
" if (!secMan)\n"
" return PR_FALSE;\n"
" result = secMan->CheckScriptAccess(cx, obj, NS_DOM_PROP_%s_%s, PR_FALSE);\n"
" if (NS_FAILED(result)) {\n"
" return nsJSUtils::nsReportError(cx, obj, result);\n"
" }\n";
static const char kMethodCheckNullStr[] =
" // If there's no private data, this must be the prototype, so ignore\n"
" if (nsnull == nativeThis) {\n"
" return JS_TRUE;\n"
" }\n"
"\n"
" {\n";
static const char kMethodCheckNullNonPrimaryStr[] =
" // If there's no private data, this must be the prototype, so ignore\n"
#ifdef USE_COMPTR
" if (!nativeThis) {\n"
#else
" if (nativeThis.IsNull()) {\n"
#endif
" return JS_TRUE;\n"
" }\n"
"\n"
" {\n";
static const char kMethodObjectParamStr[] =
#ifdef USE_COMPTR
" if (JS_FALSE == nsJSUtils::nsConvertJSValToObject((nsISupports **)(void**)getter_AddRefs(b%d),\n"
#else
" if (JS_FALSE == nsJSUtils::nsConvertJSValToObject((nsISupports **)&b%d,\n"
#endif
" kI%sIID,\n"
" NS_ConvertASCIItoUCS2(\"%s\"),\n"
" cx,\n"
" argv[%d])) {\n"
" return nsJSUtils::nsReportError(cx, obj, NS_ERROR_DOM_NOT_OBJECT_ERR);\n"
" }\n";
#define JSGEN_GENERATE_OBJECTPARAM(buffer, paramNum, paramType) \
sprintf(buffer, kMethodObjectParamStr, paramNum, paramType, \
paramType, paramNum)
static const char kMethodXPIDLObjectParamStr[] =
#ifdef USE_COMPTR
" if (JS_FALSE == nsJSUtils::nsConvertJSValToXPCObject(getter_AddRefs(b%d),\n"
#else
" if (JS_FALSE == nsJSUtils::nsConvertJSValToXPCObject((nsISupports**) &b%d,\n"
#endif
" kI%sIID, cx, argv[%d])) {\n"
" return nsJSUtils::nsReportError(cx, obj, NS_ERROR_DOM_NOT_XPC_OBJECT_ERR);\n"
" }\n";
#define JSGEN_GENERATE_XPIDL_OBJECTPARAM(buffer, paramNum, paramType) \
sprintf(buffer, kMethodXPIDLObjectParamStr, paramNum, \
(((paramType)[0] == 'n' && (paramType)[1] == 's' && (paramType)[2] == 'I') \
? ((paramType) + 3) : (paramType)), \
paramNum)
static const char kMethodStringParamStr[] =
" nsJSUtils::nsConvertJSValToString(b%d, cx, argv[%d]);\n";
#define JSGEN_GENERATE_STRINGPARAM(buffer, paramNum) \
sprintf(buffer, kMethodStringParamStr, paramNum, paramNum)
static const char kMethodBoolParamStr[] =
" if (!nsJSUtils::nsConvertJSValToBool(&b%d, cx, argv[%d])) {\n"
" return nsJSUtils::nsReportError(cx, obj, NS_ERROR_DOM_NOT_BOOLEAN_ERR);\n"
" }\n";
#define JSGEN_GENERATE_BOOLPARAM(buffer, paramNum) \
sprintf(buffer, kMethodBoolParamStr, paramNum, paramNum)
static const char kMethodIntParamStr[] =
" if (!JS_ValueToInt32(cx, argv[%d], (int32 *)&b%d)) {\n"
" return nsJSUtils::nsReportError(cx, obj, NS_ERROR_DOM_NOT_NUMBER_ERR);\n"
" }\n";
#define JSGEN_GENERATE_INTPARAM(buffer, paramNum) \
sprintf(buffer, kMethodIntParamStr, paramNum, paramNum)
static const char kMethodFloatParamStr[] =
" if (!JS_ValueToNumber(cx, argv[%d], &b%d)) {\n"
" return nsJSUtils::nsReportError(cx, obj, NS_ERROR_DOM_NOT_NUMBER_ERR);\n"
" }\n";
#define JSGEN_GENERATE_FLOATPARAM(buffer, paramNum) \
sprintf(buffer, kMethodFloatParamStr, paramNum, paramNum)
static const char kMethodFuncParamStr[] =
#ifdef USE_COMPTR
" if (!nsJSUtils::nsConvertJSValToFunc(getter_AddRefs(b%d),\n"
#else
" if (!nsJSUtils::nsConvertJSValToFunc((nsIDOMEventListener**)(nsISupports**) &b%d,\n"
#endif
" cx,\n"
" obj,\n"
" argv[%d])) {\n"
" return nsJSUtils::nsReportError(cx, obj, NS_ERROR_DOM_NOT_FUNCTION_ERR);\n"
" }\n";
#define JSGEN_GENERATE_FUNCPARAM(buffer, paramNum, paramType) \
sprintf(buffer, kMethodFuncParamStr, paramNum, paramNum)
static const char kMethodJSValParamStr[] =
" b%d = argv[%d];\n";
#define JSGEN_GENERATE_JSVALPARAM(buffer, paramNum) \
sprintf(buffer, kMethodJSValParamStr, paramNum, paramNum)
static const char kMethodParamListStr[] = "b%d";
static const char kMethodParamListDelimiterStr[] = ", ";
static const char kMethodParamEllipsisStr[] = "cx, argv+%d, argc-%d";
static const char kMethodBodyMiddleStr[] =
"\n"
" result = nativeThis->%s(%s%snativeRet);\n"
" if (NS_FAILED(result)) {\n"
" return nsJSUtils::nsReportError(cx, obj, result);\n"
" }\n"
"\n";
static const char kMethodBodyMiddleNoReturnStr[] =
"\n"
" result = nativeThis->%s(%s);\n"
" if (NS_FAILED(result)) {\n"
" return nsJSUtils::nsReportError(cx, obj, result);\n"
" }\n"
"\n";
static const char kMethodObjectRetStr[] =
" nsJSUtils::nsConvertObjectToJSVal(nativeRet, cx, obj, rval);\n";
static const char kMethodXPIDLObjectRetStr[] =
" // n.b., this will release nativeRet\n"
" nsJSUtils::nsConvertXPCObjectToJSVal(nativeRet, NS_GET_IID(%s), cx, obj, rval);\n";
static const char kMethodStringRetStr[] =
" nsJSUtils::nsConvertStringToJSVal(nativeRet, cx, rval);\n";
static const char kMethodIntRetStr[] =
" *rval = INT_TO_JSVAL(nativeRet);\n";
static const char kMethodFloatRetStr[] =
" *rval = DOUBLE_TO_JSVAL(JS_NewDouble(cx, nativeRet));\n";
static const char kMethodBoolRetStr[] =
" *rval = BOOLEAN_TO_JSVAL(nativeRet);\n";
static const char kMethodVoidRetStr[] =
" *rval = JSVAL_VOID;\n";
static const char kMethodJSValRetStr[] =
" *rval = nativeRet;\n";
static const char kMethodBadParamStr[] =
" if (argc < %d) {\n"
" return nsJSUtils::nsReportError(cx, obj, NS_ERROR_DOM_TOO_FEW_PARAMETERS_ERR);\n"
" }\n"
"\n";
static const char kMethodEndStr[] =
" }\n"
"\n"
" return JS_TRUE;\n"
"}\n";
void
JSStubGen::GenerateMethods(IdlSpecification &aSpec)
{
char buf[1024];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
GetCapitalizedName(iface_name, *iface);
int m, mcount = iface->FunctionCount();
for (m = 0; m < mcount; m++) {
IdlFunction *func = iface->GetFunctionAt(m);
IdlVariable *rval = func->GetReturnValue();
char method_name[128];
char return_type[128];
int p, pcount = func->ParameterCount();
if (func->GetIsNoScript()) {
continue;
}
GetCapitalizedName(method_name, *func);
// If this is a constructor don't have a method for it
if (strcmp(method_name, iface_name) == 0) {
continue;
}
GetVariableTypeForLocal(return_type, *rval);
if (i == 0) {
sprintf(buf, kMethodBeginStr, method_name, iface->GetName(),
method_name, iface->GetName(), iface->GetName());
}
else {
sprintf(buf, kMethodBeginNonPrimaryStr, method_name, iface->GetName(),
method_name, primary_iface->GetName(), primary_iface->GetName(),
iface->GetName(), iface->GetName(), iface->GetName());
}
*file << buf;
if (rval->GetType() != TYPE_VOID) {
sprintf(buf, kMethodReturnStr, return_type);
*file << buf;
}
for (p = 0; p < pcount; p++) {
IdlParameter *param = func->GetParameterAt(p);
GetVariableTypeForMethodLocal(return_type, *param);
sprintf(buf, kMethodParamStr, return_type, p);
*file << buf;
}
char upr_method_name[128];
strcpy(upr_method_name, method_name);
StrUpr(upr_method_name);
char upr_iface_name[128];
strcpy(upr_iface_name, iface->GetName());
StrUpr(upr_iface_name);
sprintf(buf, kMethodBodyBeginStr, upr_iface_name,
upr_method_name);
if (i == 0) {
*file << kMethodCheckNullStr;
}
else {
*file << kMethodCheckNullNonPrimaryStr;
}
*file << buf;
if (pcount > 0) {
sprintf(buf, kMethodBadParamStr, pcount);
*file << buf;
}
for (p = 0; p < pcount; p++) {
IdlParameter *param = func->GetParameterAt(p);
switch(param->GetType()) {
case TYPE_BOOLEAN:
JSGEN_GENERATE_BOOLPARAM(buf, p);
break;
case TYPE_LONG:
case TYPE_LONG_LONG:
case TYPE_SHORT:
case TYPE_ULONG:
case TYPE_ULONG_LONG:
case TYPE_USHORT:
case TYPE_CHAR:
case TYPE_INT:
case TYPE_UINT:
JSGEN_GENERATE_INTPARAM(buf, p);
break;
case TYPE_FLOAT:
JSGEN_GENERATE_FLOATPARAM(buf, p);
break;
case TYPE_JSVAL:
JSGEN_GENERATE_JSVALPARAM(buf, p);
break;
case TYPE_STRING:
JSGEN_GENERATE_STRINGPARAM(buf, p);
break;
case TYPE_OBJECT:
JSGEN_GENERATE_OBJECTPARAM(buf, p, param->GetTypeName());
break;
case TYPE_XPIDL_OBJECT:
JSGEN_GENERATE_XPIDL_OBJECTPARAM(buf, p, param->GetTypeName());
break;
case TYPE_FUNC:
JSGEN_GENERATE_FUNCPARAM(buf, p, param->GetTypeName());
break;
default:
// XXX Fail for other cases
break;
}
*file << buf;
}
char param_buf[512];
char *param_ptr = param_buf;
param_buf[0] = '\0';
for (p = 0; p < pcount; p++) {
if (p > 0) {
strcpy(param_ptr, kMethodParamListDelimiterStr);
param_ptr += strlen(param_ptr);
}
sprintf(param_ptr, kMethodParamListStr, p);
param_ptr += strlen(param_ptr);
}
if (func->GetHasEllipsis()) {
if (pcount > 0) {
strcpy(param_ptr, kMethodParamListDelimiterStr);
param_ptr += strlen(param_ptr);
}
sprintf(param_ptr, kMethodParamEllipsisStr, pcount, pcount);
param_ptr += strlen(param_ptr);
}
if (rval->GetType() != TYPE_VOID) {
if ((pcount > 0) || func->GetHasEllipsis()) {
strcpy(param_ptr, kMethodParamListDelimiterStr);
}
sprintf(buf, kMethodBodyMiddleStr, method_name, param_buf,
rval->GetType() == TYPE_STRING ? "" : "&");
}
else {
sprintf(buf, kMethodBodyMiddleNoReturnStr, method_name, param_buf);
}
*file << buf;
switch(rval->GetType()) {
case TYPE_BOOLEAN:
*file << kMethodBoolRetStr;
break;
case TYPE_LONG:
case TYPE_LONG_LONG:
case TYPE_SHORT:
case TYPE_ULONG:
case TYPE_ULONG_LONG:
case TYPE_USHORT:
case TYPE_CHAR:
case TYPE_INT:
case TYPE_UINT:
*file << kMethodIntRetStr;
break;
case TYPE_FLOAT:
*file << kMethodFloatRetStr;
break;
case TYPE_JSVAL:
*file << kMethodJSValRetStr;
break;
case TYPE_STRING:
*file << kMethodStringRetStr;
break;
case TYPE_OBJECT:
*file << kMethodObjectRetStr;
break;
case TYPE_XPIDL_OBJECT:
{
char buf[1024];
sprintf(buf, kMethodXPIDLObjectRetStr, rval->GetTypeName());
*file << buf;
}
break;
case TYPE_VOID:
*file << kMethodVoidRetStr;
break;
default:
// XXX Fail for other cases
break;
}
*file << kMethodEndStr;
}
}
}
static const char kJSClassStr[] =
"\n\n/***********************************************************************/\n"
"//\n"
"// class for %s\n"
"//\n"
"JSClass %sClass = {\n"
" \"%s\", \n"
" JSCLASS_HAS_PRIVATE | JSCLASS_PRIVATE_IS_NSISUPPORTS,\n"
" JS_PropertyStub,\n"
" JS_PropertyStub,\n"
" Get%sProperty,\n"
" Set%sProperty,\n"
" Enumerate%s,\n"
" Resolve%s,\n"
" JS_ConvertStub,\n"
" Finalize%s,\n"
" nsnull,\n"
" nsJSUtils::nsCheckAccess\n"
"};\n";
#define JSGEN_GENERATE_JSCLASS(buf, className) \
sprintf(buf, kJSClassStr, className, className, className, className, \
className, className, className, className);
void
JSStubGen::GenerateJSClass(IdlSpecification &aSpec)
{
char buf[1024];
ofstream *file = GetFile();
IdlInterface *iface = aSpec.GetInterfaceAt(0);
char *name = iface->GetName();
JSGEN_GENERATE_JSCLASS(buf, name);
*file << buf;
}
static const char kPropSpecBeginStr[] =
"\n\n//\n"
"// %s class properties\n"
"//\n"
"static JSPropertySpec %sProperties[] =\n"
"{\n";
static const char kPropSpecEntryStr[] =
" {\"%s\", %s_%s, JSPROP_ENUMERATE%s},\n";
static const char kPropSpecReadOnlyStr[] = " | JSPROP_READONLY";
static const char kPropSpecEntryReplaceableStr[] =
" {\"%s\", %s_%s, JSPROP_ENUMERATE, %s%sGetter, %s%sSetter},\n";
static const char kPropSpecEndStr[] =
" {0}\n"
"};\n";
void
JSStubGen::GenerateClassProperties(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
sprintf(buf, kPropSpecBeginStr, primary_iface->GetName(),
primary_iface->GetName());
*file << buf;
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
strcpy(iface_name, iface->GetName());
StrUpr(iface_name);
int a, acount = iface->AttributeCount();
for (a = 0; a < acount; a++) {
IdlAttribute *attr = iface->GetAttributeAt(a);
char attr_name[128];
if (attr->GetIsNoScript()) {
continue;
}
strcpy(attr_name, attr->GetName());
StrUpr(attr_name);
if (attr->GetReplaceable()) {
sprintf(buf, kPropSpecEntryReplaceableStr, attr->GetName(),
iface_name, attr_name, iface->GetName(), attr->GetName(),
iface->GetName(), attr->GetName());
}
else {
sprintf(buf, kPropSpecEntryStr, attr->GetName(),
iface_name, attr_name,
attr->GetReadOnly() ? kPropSpecReadOnlyStr : "");
}
*file << buf;
}
}
*file << kPropSpecEndStr;
}
static const char kFuncSpecBeginStr[] =
"\n\n//\n"
"// %s class methods\n"
"//\n"
"static JSFunctionSpec %sMethods[] = \n"
"{\n";
static const char kFuncSpecEntryStr[] =
" {\"%s\", %s%s, %d},\n";
static const char kFuncSpecEndStr[] =
" {0}\n"
"};\n";
void
JSStubGen::GenerateClassFunctions(IdlSpecification &aSpec)
{
char buf[512];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
sprintf(buf, kFuncSpecBeginStr, primary_iface->GetName(),
primary_iface->GetName());
*file << buf;
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
GetCapitalizedName(iface_name, *iface);
int m, mcount = iface->FunctionCount();
for (m = 0; m < mcount; m++) {
char method_name[128];
IdlFunction *func = iface->GetFunctionAt(m);
if (func->GetIsNoScript()) {
continue;
}
GetCapitalizedName(method_name, *func);
// If this is a constructor don't have a method for it.
if (strcmp(method_name, iface_name) == 0) {
continue;
}
sprintf(buf, kFuncSpecEntryStr, func->GetName(),
iface->GetName(), method_name,
func->ParameterCount());
*file << buf;
}
}
*file << kFuncSpecEndStr;
}
static const char kEmptyConstructorStr[] =
"\n\n//\n"
"// %s constructor\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"%s(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)\n"
"{\n"
" return JS_FALSE;\n"
"}\n";
#define JSGEN_GENERATE_EMPTYCONSTRUCTOR(buf, className) \
sprintf(buf, kEmptyConstructorStr, className, className);
static const char kConstructorBeginStr[] =
"\n\n//\n"
"// %s constructor\n"
"//\n"
"PR_STATIC_CALLBACK(JSBool)\n"
"%s(JSContext *cx, JSObject *obj, uintN argc, jsval *argv, jsval *rval)\n"
"{\n"
" nsresult result;\n"
" nsIID interfaceID, classID;\n"
" PRBool isConstructor;\n"
" nsIDOM%s *nativeThis;\n"
" nsIScriptObjectOwner *owner = nsnull;\n"
" nsIJSNativeInitializer* initializer = nsnull;\n"
"\n"
" static NS_DEFINE_IID(kIDOM%sIID, NS_IDOM%s_IID);\n"
" static NS_DEFINE_IID(kIJSNativeInitializerIID, NS_IJSNATIVEINITIALIZER_IID);\n"
"\n"
" nsCOMPtr<nsIScriptContext> scriptCX;\n"
" nsJSUtils::nsGetStaticScriptContext(cx, obj, getter_AddRefs(scriptCX));\n"
" if (!scriptCX) {\n"
" return JS_FALSE;\n"
" }\n"
"\n"
" nsCOMPtr<nsIScriptNameSpaceManager> manager;\n"
" scriptCX->GetNameSpaceManager(getter_AddRefs(manager));\n"
" if (!manager) {\n"
" return JS_FALSE;\n"
" }\n"
"\n"
" result = manager->LookupName(NS_ConvertASCIItoUCS2(\"%s\"), isConstructor, interfaceID, classID);\n"
" if (NS_OK != result) {\n"
" return JS_FALSE;\n"
" }\n"
"\n"
" result = nsComponentManager::CreateInstance(classID,\n"
" nsnull,\n"
" kIDOM%sIID,\n"
" (void **)&nativeThis);\n"
" if (NS_OK != result) {\n"
" return JS_FALSE;\n"
" }\n"
"\n"
" result = nativeThis->QueryInterface(kIJSNativeInitializerIID, (void **)&initializer);\n"
" if (NS_OK == result) {\n"
" result = initializer->Initialize(cx, obj, argc, argv);\n"
" NS_RELEASE(initializer);\n"
"\n"
" if (NS_OK != result) {\n"
" NS_RELEASE(nativeThis);\n"
" return JS_FALSE;\n"
" }\n"
" }\n"
"\n"
" result = nativeThis->QueryInterface(kIScriptObjectOwnerIID, (void **)&owner);\n"
" if (NS_OK != result) {\n"
" NS_RELEASE(nativeThis);\n"
" return JS_FALSE;\n"
" }\n"
"\n"
" owner->SetScriptObject((void *)obj);\n"
" JS_SetPrivate(cx, obj, nativeThis);\n"
"\n"
" NS_RELEASE(owner);\n"
" return JS_TRUE;\n"
"}";
#define JSGEN_GENERATE_CONSTRUCTOR(buf, className, caps) \
sprintf(buf, kConstructorBeginStr, className, className, className, \
className, caps, className, className)
void
JSStubGen::GenerateConstructor(IdlSpecification &aSpec)
{
char buf[4096];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
IdlFunction *constructor;
char *name = primary_iface->GetName();
char caps_name[128];
strcpy(caps_name, name);
StrUpr(caps_name);
if (HasConstructor(*primary_iface, &constructor)) {
JSGEN_GENERATE_CONSTRUCTOR(buf, name, caps_name);
*file << buf;
}
else if (!mIsGlobal) {
JSGEN_GENERATE_EMPTYCONSTRUCTOR(buf, name);
*file << buf;
}
}
static const char kGlobalInitClassStr[] =
"\n\n//\n"
"// %s class initialization\n"
"//\n"
"nsresult NS_Init%sClass(nsIScriptContext *aContext, \n"
" nsIScriptGlobalObject *aGlobal)\n"
"{\n"
" JSContext *jscontext = (JSContext *)aContext->GetNativeContext();\n"
" JSObject *global = JS_GetGlobalObject(jscontext);\n"
"\n"
" JS_DefineProperties(jscontext, global, %sProperties);\n"
" JS_DefineFunctions(jscontext, global, %sMethods);\n"
"\n"
" return NS_OK;\n"
"}\n";
#define JSGEN_GENERATE_GLOBALINITCLASS(buffer, className) \
sprintf(buffer, kGlobalInitClassStr, className, className, className, \
className)
static const char kInitClassBeginStr[] =
"\n\n//\n"
"// %s class initialization\n"
"//\n"
"extern \"C\" NS_DOM nsresult NS_Init%sClass(nsIScriptContext *aContext, void **aPrototype)\n"
"{\n"
" JSContext *jscontext = (JSContext *)aContext->GetNativeContext();\n"
" JSObject *proto = nsnull;\n"
" JSObject *constructor = nsnull;\n"
" JSObject *parent_proto = nsnull;\n"
" JSObject *global = JS_GetGlobalObject(jscontext);\n"
" jsval vp;\n"
"\n"
" if ((PR_TRUE != JS_LookupProperty(jscontext, global, \"%s\", &vp)) ||\n"
" !JSVAL_IS_OBJECT(vp) ||\n"
" ((constructor = JSVAL_TO_OBJECT(vp)) == nsnull) ||\n"
" (PR_TRUE != JS_LookupProperty(jscontext, JSVAL_TO_OBJECT(vp), \"prototype\", &vp)) || \n"
" !JSVAL_IS_OBJECT(vp)) {\n\n";
#define JSGEN_GENERATE_INITCLASSBEGIN(buffer, className) \
sprintf(buffer, kInitClassBeginStr, className, className, className)
static const char kGetParentProtoStr[] =
" if (NS_OK != NS_Init%sClass(aContext, (void **)&parent_proto)) {\n"
" return NS_ERROR_FAILURE;\n"
" }\n";
static const char kInitClassBodyStr[] =
" proto = JS_InitClass(jscontext, // context\n"
" global, // global object\n"
" parent_proto, // parent proto \n"
" &%sClass, // JSClass\n"
" %s, // JSNative ctor\n"
" 0, // ctor args\n"
" %sProperties, // proto props\n"
" %sMethods, // proto funcs\n"
" nsnull, // ctor props (static)\n"
" nsnull); // ctor funcs (static)\n"
" if (nsnull == proto) {\n"
" return NS_ERROR_FAILURE;\n"
" }\n"
"\n";
#define JSGEN_GENERATE_INITCLASSBODY(buffer, className) \
sprintf(buffer, kInitClassBodyStr, className, className, \
className, className)
static const char kAliasConstructorStr[] =
" JS_AliasProperty(jscontext, global, \"%s\", \"%s\");\n";
static const char kInitStaticBeginStr[] =
" if ((PR_TRUE == JS_LookupProperty(jscontext, global, \"%s\", &vp)) &&\n"
" JSVAL_IS_OBJECT(vp) &&\n"
" ((constructor = JSVAL_TO_OBJECT(vp)) != nsnull)) {\n";
static const char kInitStaticEntryStr[] =
" vp = INT_TO_JSVAL(nsIDOM%s::%s);\n"
" JS_SetProperty(jscontext, constructor, \"%s\", &vp);\n"
"\n";
static const char kInitStaticEndStr[] =
" }\n"
"\n";
static const char kInitClassEndStr[] =
" }\n"
" else if ((nsnull != constructor) && JSVAL_IS_OBJECT(vp)) {\n"
" proto = JSVAL_TO_OBJECT(vp);\n"
" }\n"
" else {\n"
" return NS_ERROR_FAILURE;\n"
" }\n"
"\n"
" if (aPrototype) {\n"
" *aPrototype = proto;\n"
" }\n"
" return NS_OK;\n"
"}\n";
void
JSStubGen::GenerateInitClass(IdlSpecification &aSpec)
{
char buf[2048];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
char *primary_class = primary_iface->GetName();
if (mIsGlobal) {
JSGEN_GENERATE_GLOBALINITCLASS(buf, primary_class);
*file << buf;
return;
}
JSGEN_GENERATE_INITCLASSBEGIN(buf, primary_class);
*file << buf;
if (primary_iface->BaseClassCount() > 0) {
char *base_class = primary_iface->GetBaseClassAt(0);
sprintf(buf, kGetParentProtoStr, base_class);
*file << buf;
}
JSGEN_GENERATE_INITCLASSBODY(buf, primary_class);
*file << buf;
int i, icount = aSpec.InterfaceCount();
for (i = 0; i < icount; i++) {
IdlInterface *iface = aSpec.GetInterfaceAt(i);
char iface_name[128];
GetCapitalizedName(iface_name, *iface);
int m, mcount = iface->FunctionCount();
for (m = 0; m < mcount; m++) {
char method_name[128];
IdlFunction *func = iface->GetFunctionAt(m);
GetCapitalizedName(method_name, *func);
// If this is a constructor defined in a non-primary interface
// don't have a method for it...we'll alias it to the constructor
// for the primary interface.
if ((strcmp(method_name, iface_name) == 0) &&
(iface != primary_iface)) {
sprintf(buf, kAliasConstructorStr, primary_class, method_name);
*file << buf;
}
}
}
int c, ccount = primary_iface->ConstCount();
if (ccount > 0) {
sprintf(buf, kInitStaticBeginStr, primary_iface->GetName());
*file << buf;
for (c = 0; c < ccount; c++) {
IdlVariable *var = primary_iface->GetConstAt(c);
if (NULL != var) {
sprintf(buf, kInitStaticEntryStr, primary_iface->GetName(),
var->GetName(), var->GetName());
*file << buf;
}
}
*file << kInitStaticEndStr;
}
*file << kInitClassEndStr;
}
static const char kNewGlobalJSObjectStr[] =
"\n\n//\n"
"// Method for creating a new %s JavaScript object\n"
"//\n"
"extern \"C\" NS_DOM nsresult NS_NewScript%s(nsIScriptContext *aContext, nsISupports *aSupports, nsISupports *aParent, void **aReturn)\n"
"{\n"
" NS_PRECONDITION(nsnull != aContext && nsnull != aSupports && nsnull != aReturn, \"null arg\");\n"
" JSContext *jscontext = (JSContext *)aContext->GetNativeContext();\n"
"\n"
" JSObject *global = ::JS_NewObject(jscontext, &%sClass, NULL, NULL);\n"
" if (global) {\n"
" // The global object has a to be defined in two step:\n"
" // 1- create a generic object, with no prototype and no parent which\n"
" // will be passed to JS_InitStandardClasses. JS_InitStandardClasses \n"
" // will make it the global object\n"
" // 2- define the global object to be what you really want it to be.\n"
" //\n"
" // The js runtime is not fully initialized before JS_InitStandardClasses\n"
" // is called, so part of the global object initialization has to be moved \n"
" // after JS_InitStandardClasses\n"
"\n"
" // assign \"this\" to the js object\n"
" ::JS_SetPrivate(jscontext, global, aSupports);\n"
" NS_ADDREF(aSupports);\n"
"\n"
" JS_DefineProperties(jscontext, global, %sProperties);\n"
" JS_DefineFunctions(jscontext, global, %sMethods);\n"
"\n"
" *aReturn = (void*)global;\n"
" return NS_OK;\n"
" }\n"
"\n"
" return NS_ERROR_FAILURE;\n"
"}\n";
#define JSGEN_GENERATE_NEWGLOBALJSOBJECT(buffer, className) \
sprintf(buffer, kNewGlobalJSObjectStr, className, \
className, className, className, className)
static const char kNewJSObjectStr[] =
"\n\n//\n"
"// Method for creating a new %s JavaScript object\n"
"//\n"
"extern \"C\" NS_DOM nsresult NS_NewScript%s(nsIScriptContext *aContext, nsISupports *aSupports, nsISupports *aParent, void **aReturn)\n"
"{\n"
" NS_PRECONDITION(nsnull != aContext && nsnull != aSupports && nsnull != aReturn, \"null argument to NS_NewScript%s\");\n"
" JSObject *proto;\n"
" JSObject *parent;\n"
" nsIScriptObjectOwner *owner;\n"
" JSContext *jscontext = (JSContext *)aContext->GetNativeContext();\n"
" nsresult result = NS_OK;\n"
" nsIDOM%s *a%s;\n"
"\n"
" if (nsnull == aParent) {\n"
" parent = nsnull;\n"
" }\n"
" else if (NS_OK == aParent->QueryInterface(kIScriptObjectOwnerIID, (void**)&owner)) {\n"
" if (NS_OK != owner->GetScriptObject(aContext, (void **)&parent)) {\n"
" NS_RELEASE(owner);\n"
" return NS_ERROR_FAILURE;\n"
" }\n"
" NS_RELEASE(owner);\n"
" }\n"
" else {\n"
" return NS_ERROR_FAILURE;\n"
" }\n"
"\n"
" if (NS_OK != NS_Init%sClass(aContext, (void **)&proto)) {\n"
" return NS_ERROR_FAILURE;\n"
" }\n"
"\n"
" result = aSupports->QueryInterface(kI%sIID, (void **)&a%s);\n"
" if (NS_OK != result) {\n"
" return result;\n"
" }\n"
"\n"
" // create a js object for this class\n"
" *aReturn = JS_NewObject(jscontext, &%sClass, proto, parent);\n"
" if (nsnull != *aReturn) {\n"
" // connect the native object to the js object\n"
" JS_SetPrivate(jscontext, (JSObject *)*aReturn, a%s);\n"
" }\n"
" else {\n"
" NS_RELEASE(a%s);\n"
" return NS_ERROR_FAILURE; \n"
" }\n"
"\n"
" return NS_OK;\n"
"}\n";
#define JSGEN_GENERATE_NEWJSOBJECT(buffer, className) \
sprintf(buffer, kNewJSObjectStr, className, className, \
className, className, className, className, \
className, className, className, className, className)
void
JSStubGen::GenerateNew(IdlSpecification &aSpec)
{
char buf[2048];
ofstream *file = GetFile();
IdlInterface *primary_iface = aSpec.GetInterfaceAt(0);
char *primary_class = primary_iface->GetName();
if (mIsGlobal) {
JSGEN_GENERATE_NEWGLOBALJSOBJECT(buf, primary_class);
}
else {
JSGEN_GENERATE_NEWJSOBJECT(buf, primary_class);
}
*file << buf;
}