include "cefpython.pyx"

cdef class JavascriptBindings:

# By default binding only to top frame.

cdef public py_bool bindToFrames

cdef public py_bool bindToPopups

cdef public dict functions

cdef public dict properties

cdef public dict objects

def __init__(self, bindToFrames=False, bindToPopups=False):

self.functions = {}

self.properties = {}

self.objects = {}

self.bindToFrames = bool(bindToFrames)

self.bindToPopups = bool(bindToPopups)

cpdef py_bool GetBindToFrames(self):

return bool(self.bindToFrames)

cpdef py_bool GetBindToPopups(self):

return bool(self.bindToPopups)

cpdef py_void SetFunction(self, py_string name, object func):

self.SetProperty(name, func)

cpdef py_void SetObject(self, py_string name, object obj):

if not hasattr(obj, "__class__"):

raise Exception("JavascriptBindings.SetObject() failed: name=%s, "

"__class__ attribute missing, this is not an object" % name)

cdef dict methods = {}

cdef py_string key

cdef object method

cdef object predicate = inspect.ismethod

if isinstance(obj, (PyBrowser, PyFrame)):

predicate = inspect.isbuiltin

for value in inspect.getmembers(obj, predicate=predicate):

key = value[0]

method = value[1]

methods[key] = method

self.objects[name] = methods

cpdef object GetFunction(self, py_string name):

if name in self.functions:

return self.functions[name]

cpdef dict GetFunctions(self):

return self.functions

cpdef dict GetObjects(self):

return self.objects

cpdef object GetObjectMethod(self, py_string objectName, py_string methodName):

if objectName in self.objects:

if methodName in self.objects[objectName]:

return self.objects[objectName][methodName]

cpdef object GetFunctionOrMethod(self, py_string name):

# Name can be "someFunc" or "object.someMethod".

cdef list words

if "." in name:

words = name.split(".")

return self.GetObjectMethod(words[0], words[1])

else:

return self.GetFunction(name)

cpdef py_void SetProperty(self, py_string name, object value):

cdef object allowed = self.IsValueAllowedRecursively(value) # returns True or string.

if allowed is not True:

raise Exception("JavascriptBindings.SetProperty() failed: name=%s, "

"not allowed type: %s (this may be a type of a nested value)"

% (name, allowed))

cdef object valueType = type(value)

if IsFunctionOrMethod(valueType):

self.functions[name] = value

else:

self.properties[name] = value

cpdef py_void Rebind(self):

# Rebind() is called for both first-time binding and rebinding.

cdef PyBrowser pyBrowser

cdef dict functions

cdef dict properties

cdef dict objects

cdef dict methods

for browserId, pyBrowser in g_pyBrowsers.iteritems():

if pyBrowser.GetJavascriptBindings() != self:

continue

# Send to the Renderer process: functions, properties,

# objects and its methods, bindToFrames.

functions = {}

for funcName in self.functions:

functions[funcName] = None

properties = self.properties

objects = {}

for objectName in self.objects:

methods = {}

for methodName in self.objects[objectName]:

methods[methodName] = None

objects[objectName] = methods

pyBrowser.SendProcessMessage(cef_types.PID_RENDERER,

0, "DoJavascriptBindings", [{

"functions": functions,

"properties": properties,

"objects": objects,

"bindToFrames": self.bindToFrames

}])

cpdef dict GetProperties(self):

return self.properties

@staticmethod

def IsValueAllowed(object value):

return JavascriptBindings.IsValueAllowedRecursively(value) is True

@staticmethod

def IsValueAllowedRecursively(object value, py_bool recursion=False):

# When making changes here modify also Frame.SetProperty() as it

# checks for FunctionType, MethodType.

cdef object valueType = type(value)

cdef object valueType2

cdef object key

if valueType == list:

for val in value:

valueType2 = JavascriptBindings.IsValueAllowedRecursively(val, True)

if valueType2 is not True:

return valueType2.__name__

return True

elif valueType == bool:

return True

elif valueType == float:

return True

elif valueType == int:

return True

elif valueType == type(None):

return True

elif IsFunctionOrMethod(valueType):

if recursion:

return valueType.__name__

else:

return True

elif valueType == dict:

for key in value:

valueType2 = JavascriptBindings.IsValueAllowedRecursively(value[key], True)

if valueType2 is not True:

return valueType2.__name__

return True

elif valueType == str or valueType == bytes:

return True

elif PY_MAJOR_VERSION < 3 and valueType == unicode:

# The unicode type is not defined in Python 3.

return True

elif valueType == tuple:

return True

else:

return valueType.__name__