include "cefpython.pyx"

cdef int g_pythonCallbackMaxId = 0

cdef dict g_pythonCallbacks = {}

cdef struct PythonCallback:

int callbackId

char uniqueCefBinaryValueSize[16]

cdef CefRefPtr[CefBinaryValue] PutPythonCallback(

object browserId,

object frameId,

object function

) except *:

global g_pythonCallbacks

global g_pythonCallbackMaxId

if not browserId:

raise Exception("PutPythonCallback() FAILED: browserId is empty")

if not frameId:

raise Exception("PutPythonCallback() FAILED: frameId is empty")

cdef PythonCallback pyCallback

g_pythonCallbackMaxId += 1

pyCallback.callbackId = g_pythonCallbackMaxId

cdef CefRefPtr[CefBinaryValue] binaryValue = CefBinaryValue_Create(

&pyCallback, sizeof(pyCallback))

# [0] browserId, [1] frameId, [2] function.

g_pythonCallbacks[g_pythonCallbackMaxId] = (browserId, frameId, function)

return binaryValue

cdef public void RemovePythonCallbacksForFrame(

int frameId

) except * with gil:

# Cannot remove elements from g_pythonCallbacks (dict) while iterating.

cdef list toRemove = []

try:

global g_pythonCallbacks

for callbackId, value in g_pythonCallbacks.iteritems():

if value[1] == frameId:

toRemove.append(callbackId)

for callbackId in toRemove:

del g_pythonCallbacks[callbackId]

Debug("RemovePythonCallbacksForFrame(): " \

"removed python callback, callbackId = %s" \

% callbackId)

except:

(exc_type, exc_value, exc_trace) = sys.exc_info()

sys.excepthook(exc_type, exc_value, exc_trace)

cdef void RemovePythonCallbacksForBrowser(

int browserId) except *:

cdef list toRemove = []

global g_pythonCallbacks

for callbackId, value in g_pythonCallbacks.iteritems():

if value[0] == browserId:

toRemove.append(callbackId)

for callbackId in toRemove:

del g_pythonCallbacks[callbackId]

Debug("RemovePythonCallbacksForBrowser(): " \

"removed python callback, callbackId = %s" \

% callbackId)

cdef public cpp_bool ExecutePythonCallback(

CefRefPtr[CefBrowser] cefBrowser,

int callbackId,

CefRefPtr[CefListValue] cefFunctionArguments,

) except * with gil:

cdef object function

cdef list functionArguments

cdef object returnValue

try:

global g_pythonCallbacks

if callbackId in g_pythonCallbacks:

# [0] browserId, [1] frameId, [2] function.

function = g_pythonCallbacks[callbackId][2]

functionArguments = CefListValueToPyList(

cefBrowser, cefFunctionArguments)

returnValue = function(*functionArguments)

if returnValue is not None:

Debug("ExecutePythonCallback() WARNING: function returned" \

"value, but returning values to javascript is not " \

"supported, function name = %s" % function.__name__)

return True

else:

Debug("ExecutePythonCallback() FAILED: callback not found, " \

"callbackId = %s" % callbackId)

return False

except:

(exc_type, exc_value, exc_trace) = sys.exc_info()

sys.excepthook(exc_type, exc_value, exc_trace)