# This script will generate the AppleEvents interface for Python.

# It uses the "bgen" package to generate C code.

# It execs the file aegen.py which contain the function definitions

# (aegen.py was generated by aescan.py, scanning the <AppleEvents.h> header file).

from macsupport import *

AEArrayType = Type("AEArrayType", "c")

AESendMode = Type("AESendMode", "l")

AESendPriority = Type("AESendPriority", "h")

AEInteractAllowed = Type("AEInteractAllowed", "b")

AEReturnID = Type("AEReturnID", "h")

AETransactionID = Type("AETransactionID", "l")

AEEventClass = OSTypeType('AEEventClass')

AEEventID = OSTypeType('AEEventID')

AEKeyword = OSTypeType('AEKeyword')

DescType = OSTypeType('DescType')

AEDesc = OpaqueType('AEDesc')

AEDesc_ptr = OpaqueType('AEDesc')

AEAddressDesc = OpaqueType('AEAddressDesc', 'AEDesc')

AEAddressDesc_ptr = OpaqueType('AEAddressDesc', 'AEDesc')

AEDescList = OpaqueType('AEDescList', 'AEDesc')

AEDescList_ptr = OpaqueType('AEDescList', 'AEDesc')

AERecord = OpaqueType('AERecord', 'AEDesc')

AERecord_ptr = OpaqueType('AERecord', 'AEDesc')

AppleEvent = OpaqueType('AppleEvent', 'AEDesc')

AppleEvent_ptr = OpaqueType('AppleEvent', 'AEDesc')

class EHType(Type):

    def __init__(self, name = 'EventHandler', format = ''):

        Type.__init__(self, name, format)

    def declare(self, name):

        Output("AEEventHandlerUPP %s__proc__ = upp_GenericEventHandler;", name)

        Output("PyObject *%s;", name)

    def getargsFormat(self):

        return "O"

    def getargsArgs(self, name):

        return "&%s" % name

    def passInput(self, name):

        return "%s__proc__, (long)%s" % (name, name)

    def passOutput(self, name):

        return "&%s__proc__, (long *)&%s" % (name, name)

    def mkvalueFormat(self):

        return "O"

    def mkvalueArgs(self, name):

        return name

    def cleanup(self, name):

        Output("Py_INCREF(%s); /* XXX leak, but needed */", name)

class EHNoRefConType(EHType):

    def passInput(self, name):

        return "upp_GenericEventHandler"

EventHandler = EHType()

EventHandlerNoRefCon = EHNoRefConType()

IdleProcPtr = FakeType("upp_AEIdleProc")

AEIdleUPP = IdleProcPtr

EventFilterProcPtr = FakeType("(AEFilterUPP)0")

AEFilterUPP = EventFilterProcPtr

NMRecPtr = FakeType("(NMRecPtr)0")

EventHandlerProcPtr = FakeType("upp_GenericEventHandler")

AEEventHandlerUPP = EventHandlerProcPtr

AlwaysFalse = FakeType("0")

AEFunction = OSErrWeakLinkFunctionGenerator

AEMethod = OSErrWeakLinkMethodGenerator

includestuff = includestuff + """

#include <Carbon/Carbon.h>

#ifdef USE_TOOLBOX_OBJECT_GLUE

extern PyObject *_AEDesc_New(AEDesc *);

extern int _AEDesc_Convert(PyObject *, AEDesc *);

#define AEDesc_New _AEDesc_New

#define AEDesc_NewBorrowed _AEDesc_NewBorrowed

#define AEDesc_Convert _AEDesc_Convert

#endif

typedef long refcontype;

static pascal OSErr GenericEventHandler(const AppleEvent *request, AppleEvent *reply, refcontype refcon); /* Forward */

AEEventHandlerUPP upp_GenericEventHandler;

static pascal Boolean AEIdleProc(EventRecord *theEvent, long *sleepTime, RgnHandle *mouseRgn)

{

        if ( PyOS_InterruptOccurred() )

                return 1;

        return 0;

}

AEIdleUPP upp_AEIdleProc;

"""

finalstuff = finalstuff + """

static pascal OSErr

GenericEventHandler(const AppleEvent *request, AppleEvent *reply, refcontype refcon)

{

        PyObject *handler = (PyObject *)refcon;

        AEDescObject *requestObject, *replyObject;

        PyObject *args, *res;

        if ((requestObject = (AEDescObject *)AEDesc_New((AppleEvent *)request)) == NULL) {

                return -1;

        }

        if ((replyObject = (AEDescObject *)AEDesc_New(reply)) == NULL) {

                Py_DECREF(requestObject);

                return -1;

        }

        if ((args = Py_BuildValue("OO", requestObject, replyObject)) == NULL) {

                Py_DECREF(requestObject);

                Py_DECREF(replyObject);

                return -1;

        }

        res = PyEval_CallObject(handler, args);

        requestObject->ob_itself.descriptorType = 'null';

        requestObject->ob_itself.dataHandle = NULL;

        replyObject->ob_itself.descriptorType = 'null';

        replyObject->ob_itself.dataHandle = NULL;

        Py_DECREF(args);

        if (res == NULL) {

                PySys_WriteStderr("Exception in AE event handler function\\n");

                PyErr_Print();

                return -1;

        }

        Py_DECREF(res);

        return noErr;

}

PyObject *AEDesc_NewBorrowed(AEDesc *itself)

{

        PyObject *it;

        it = AEDesc_New(itself);

        if (it)

                ((AEDescObject *)it)->ob_owned = 0;

        return (PyObject *)it;

}

"""

initstuff = initstuff + """

        upp_AEIdleProc = NewAEIdleUPP(AEIdleProc);

        upp_GenericEventHandler = NewAEEventHandlerUPP(GenericEventHandler);

        PyMac_INIT_TOOLBOX_OBJECT_NEW(AEDesc *, AEDesc_New);

        PyMac_INIT_TOOLBOX_OBJECT_NEW(AEDesc *, AEDesc_NewBorrowed);

        PyMac_INIT_TOOLBOX_OBJECT_CONVERT(AEDesc, AEDesc_Convert);

"""

module = MacModule('_AE', 'AE', includestuff, finalstuff, initstuff)

class AEDescDefinition(PEP253Mixin, GlobalObjectDefinition):

    getsetlist = [(

            'type',

            'return PyMac_BuildOSType(self->ob_itself.descriptorType);',

            None,

            'Type of this AEDesc'

            ), (

            'data',

            """

            PyObject *res;

            Size size;

            char *ptr;

            OSErr err;

            size = AEGetDescDataSize(&self->ob_itself);

            if ( (res = PyString_FromStringAndSize(NULL, size)) == NULL )

                    return NULL;

            if ( (ptr = PyString_AsString(res)) == NULL )

                    return NULL;

            if ( (err=AEGetDescData(&self->ob_itself, ptr, size)) < 0 )

                    return PyMac_Error(err);

            return res;

            """,

            None,

            'The raw data in this AEDesc'

            )]

    def __init__(self, name, prefix = None, itselftype = None):

        GlobalObjectDefinition.__init__(self, name, prefix or name, itselftype or name)

        self.argref = "*"

    def outputStructMembers(self):

        GlobalObjectDefinition.outputStructMembers(self)

        Output("int ob_owned;")

    def outputInitStructMembers(self):

        GlobalObjectDefinition.outputInitStructMembers(self)

        Output("it->ob_owned = 1;")

    def outputCleanupStructMembers(self):

        Output("if (self->ob_owned) AEDisposeDesc(&self->ob_itself);")

aedescobject = AEDescDefinition('AEDesc')

module.addobject(aedescobject)

functions = []

aedescmethods = []

execfile('aegen.py')

##execfile('aedatamodelgen.py')

# Manual generator

AutoDispose_body = """

int onoff, old;

if (!PyArg_ParseTuple(_args, "i", &onoff))

        return NULL;

old = _self->ob_owned;

_self->ob_owned = onoff;

_res = Py_BuildValue("i", old);

return _res;

"""

f = ManualGenerator("AutoDispose", AutoDispose_body)

f.docstring = lambda: "(int)->int. Automatically AEDisposeDesc the object on Python object cleanup"

aedescmethods.append(f)

for f in functions: module.add(f)

for f in aedescmethods: aedescobject.add(f)

SetOutputFileName('_AEmodule.c')

module.generate()