include "../cefpython.pyx"

include "../browser.pyx"

cimport cef_types

TS_ABNORMAL_TERMINATION = cef_types.TS_ABNORMAL_TERMINATION

TS_PROCESS_WAS_KILLED = cef_types.TS_PROCESS_WAS_KILLED

TS_PROCESS_CRASHED = cef_types.TS_PROCESS_CRASHED

cdef PyAuthCallback CreatePyAuthCallback(

CefRefPtr[CefAuthCallback] cefCallback):

cdef PyAuthCallback pyCallback = PyAuthCallback()

pyCallback.cefCallback = cefCallback

return pyCallback

cdef class PyAuthCallback:

PyToCefStringValue(password))

cpdef py_void Cancel(self):

self.cefCallback.get().Cancel()

cdef PyRequestCallback CreatePyRequestCallback(

CefRefPtr[CefRequestCallback] cefCallback):

cdef PyRequestCallback pyCallback = PyRequestCallback()

pyCallback.cefCallback = cefCallback

return pyCallback

cdef class PyRequestCallback:

self.cefCallback.get().Continue(bool(allow))

cpdef py_void Cancel(self):

self.cefCallback.get().Cancel()

cdef public cpp_bool RequestHandler_OnBeforeBrowse(

CefRefPtr[CefBrowser] cefBrowser,

CefRefPtr[CefFrame] cefFrame,

CefRefPtr[CefRequest] cefRequest,

cpp_bool cefIsRedirect

) except * with gil:

cdef PyBrowser pyBrowser

cdef PyFrame pyFrame

cdef PyRequest pyRequest

cdef py_bool pyIsRedirect

cdef object clientCallback

cdef py_bool returnValue

try:

pyBrowser = GetPyBrowser(cefBrowser, "OnBeforeBrowse")

pyFrame = GetPyFrame(cefFrame)

pyRequest = CreatePyRequest(cefRequest)

pyIsRedirect = bool(cefIsRedirect)

clientCallback = pyBrowser.GetClientCallback("OnBeforeBrowse")

if clientCallback:

returnValue = clientCallback(

browser=pyBrowser,

frame=pyFrame,

request=pyRequest,

is_redirect=pyIsRedirect)

return bool(returnValue)

else:

return False

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)

cdef public cpp_bool RequestHandler_OnBeforeResourceLoad(

CefRefPtr[CefBrowser] cefBrowser,

CefRefPtr[CefFrame] cefFrame,

CefRefPtr[CefRequest] cefRequest

) except * with gil:

cdef PyBrowser pyBrowser

cdef PyFrame pyFrame

cdef PyRequest pyRequest

cdef object clientCallback

cdef py_bool returnValue

try:

pyBrowser = GetPyBrowser(cefBrowser, "OnBeforeResourceLoad")

pyFrame = GetPyFrame(cefFrame)

pyRequest = CreatePyRequest(cefRequest)

clientCallback = pyBrowser.GetClientCallback("OnBeforeResourceLoad")

if clientCallback:

returnValue = clientCallback(

browser=pyBrowser,

frame=pyFrame,

request=pyRequest)

return bool(returnValue)

else:

return False

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)

cdef public CefRefPtr[CefResourceHandler] RequestHandler_GetResourceHandler(

CefRefPtr[CefBrowser] cefBrowser,

CefRefPtr[CefFrame] cefFrame,

CefRefPtr[CefRequest] cefRequest

) except * with gil:

cdef PyBrowser pyBrowser

cdef PyFrame pyFrame

cdef PyRequest pyRequest

cdef object clientCallback

cdef object returnValue

try:

pyBrowser = GetPyBrowser(cefBrowser, "GetResourceHandler")

pyFrame = GetPyFrame(cefFrame)

pyRequest = CreatePyRequest(cefRequest)

clientCallback = pyBrowser.GetClientCallback("GetResourceHandler")

if clientCallback:

returnValue = clientCallback(

browser=pyBrowser,

frame=pyFrame,

request=pyRequest)

if returnValue:

return CreateResourceHandler(returnValue)

else:

return <CefRefPtr[CefResourceHandler]>NULL

else:

return <CefRefPtr[CefResourceHandler]>NULL

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)

cdef public void RequestHandler_OnResourceRedirect(

CefRefPtr[CefBrowser] cefBrowser,

CefRefPtr[CefFrame] cefFrame,

const CefString& cefOldUrl,

CefString& cefNewUrl,

CefRefPtr[CefRequest] cefRequest,

CefRefPtr[CefResponse] cefResponse

) except * with gil:

cdef PyBrowser pyBrowser

cdef PyFrame pyFrame

cdef str pyOldUrl

cdef list pyNewUrlOut

cdef PyRequest pyRequest

cdef PyResponse pyResponse

cdef object clientCallback

try:

pyBrowser = GetPyBrowser(cefBrowser, "OnResourceRedirect")

pyFrame = GetPyFrame(cefFrame)

pyOldUrl = CefToPyString(cefOldUrl)

pyNewUrlOut = [CefToPyString(cefNewUrl)]

pyRequest = CreatePyRequest(cefRequest)

pyResponse = CreatePyResponse(cefResponse)

clientCallback = pyBrowser.GetClientCallback("OnResourceRedirect")

if clientCallback:

clientCallback(

browser=pyBrowser,

frame=pyFrame,

old_url=pyOldUrl,

new_url_out=pyNewUrlOut,

request=pyRequest,

response=pyResponse)

if pyNewUrlOut[0]:

PyToCefString(pyNewUrlOut[0], cefNewUrl)

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)

cdef public cpp_bool RequestHandler_GetAuthCredentials(

CefRefPtr[CefBrowser] cefBrowser,

CefRefPtr[CefFrame] cefFrame,

cpp_bool cefIsProxy,

const CefString& cefHost,

int cefPort,

const CefString& cefRealm,

const CefString& cefScheme,

CefRefPtr[CefAuthCallback] cefAuthCallback

) except * with gil:

cdef PyBrowser pyBrowser

cdef PyFrame pyFrame

cdef py_bool pyIsProxy

cdef str pyHost

cdef int pyPort

cdef str pyRealm

cdef str pyScheme

cdef PyAuthCallback pyAuthCallback

cdef py_bool returnValue

cdef list pyUsernameOut

cdef list pyPasswordOut

cdef object clientCallback

try:

pyBrowser = GetPyBrowser(cefBrowser, "GetAuthCredentials")

pyFrame = GetPyFrame(cefFrame)

pyIsProxy = bool(cefIsProxy)

pyHost = CefToPyString(cefHost)

pyPort = int(cefPort)

pyRealm = CefToPyString(cefRealm)

pyScheme = CefToPyString(cefScheme)

pyAuthCallback = CreatePyAuthCallback(cefAuthCallback)

pyUsernameOut = [""]

pyPasswordOut = [""]

clientCallback = pyBrowser.GetClientCallback("GetAuthCredentials")

if clientCallback:

returnValue = clientCallback(

browser=pyBrowser,

frame=pyFrame,

is_proxy=pyIsProxy,

host=pyHost,

port=pyPort,

realm=pyRealm,

scheme=pyScheme,

callback=pyAuthCallback)

return bool(returnValue)

else:

# TODO: port it from CEF 1, copy the cef1/http_authentication/.

# --

# Default implementation for Windows.

# IF UNAME_SYSNAME == "Windows":

# returnValue = HttpAuthenticationDialog(

# pyBrowser,

# pyIsProxy, pyHost, pyPort, pyRealm, pyScheme,

# pyUsernameOut, pyPasswordOut)

# if returnValue:

# pyAuthCallback.Continue(pyUsernameOut[0], pyPasswordOut[0])

# return True

# return False

# ELSE:

# return False

return False

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)

cdef public cpp_bool RequestHandler_OnQuotaRequest(

CefRefPtr[CefBrowser] cefBrowser,

const CefString& cefOriginUrl,

int64 newSize,

CefRefPtr[CefRequestCallback] cefRequestCallback

) except * with gil:

cdef PyBrowser pyBrowser

cdef py_string pyOriginUrl

cdef py_bool returnValue

cdef object clientCallback

try:

pyBrowser = GetPyBrowser(cefBrowser, "OnQuotaRequest")

pyOriginUrl = CefToPyString(cefOriginUrl)

clientCallback = pyBrowser.GetClientCallback("OnQuotaRequest")

if clientCallback:

returnValue = clientCallback(

browser=pyBrowser,

origin_url=pyOriginUrl,

new_size=newSize,

callback=CreatePyRequestCallback(cefRequestCallback))

return bool(returnValue)

else:

return False

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)

cdef public CefRefPtr[CefCookieManager] RequestHandler_GetCookieManager(

CefRefPtr[CefBrowser] cefBrowser,

const CefString& cefMainUrl

) except * with gil:

# In CEF the GetCookieManager callback belongs to

# CefRequestContextHandler.

# In an exceptional case the browser parameter may be None

# due to limitation in CEF API. No workaround as of now.

cdef PyBrowser pyBrowser

cdef str pyMainUrl

cdef object clientCallback

cdef PyCookieManager returnValue

try:

pyBrowser = GetPyBrowser(cefBrowser, "GetCookieManager")

pyMainUrl = CefToPyString(cefMainUrl)

if pyBrowser:

# Browser may be empty.

clientCallback = pyBrowser.GetClientCallback("GetCookieManager")

if clientCallback:

returnValue = clientCallback(

browser=pyBrowser,

main_url=pyMainUrl)

if returnValue:

if isinstance(returnValue, PyCookieManager):

return returnValue.cefCookieManager

else:

raise Exception("Expected a CookieManager object")

return <CefRefPtr[CefCookieManager]>NULL

else:

return <CefRefPtr[CefCookieManager]>NULL

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)

cdef public void RequestHandler_OnProtocolExecution(

CefRefPtr[CefBrowser] cefBrowser,

const CefString& cefUrl,

cpp_bool& cefAllowOSExecution

) except * with gil:

cdef PyBrowser pyBrowser

cdef str pyUrl

cdef list pyAllowOSExecutionOut

cdef object clientCallback

try:

pyBrowser = GetPyBrowser(cefBrowser, "OnProtocolExecution")

pyUrl = CefToPyString(cefUrl)

pyAllowOSExecutionOut = [bool(cefAllowOSExecution)]

clientCallback = pyBrowser.GetClientCallback("OnProtocolExecution")

if clientCallback:

clientCallback(

browser=pyBrowser,

url=pyUrl,

allow_execution_out=pyAllowOSExecutionOut)

# Since Cython 0.17.4 assigning a value to an argument

# passed by reference will throw an error, the fix is to

# to use "(&arg)[0] =" instead of "arg =", see this topic:

# https://groups.google.com/forum/#!msg/cython-users/j58Sp3QMrD4/y9vJy9YBi_kJ

# For CefRefPtr you should use swap() method instead.

(&cefAllowOSExecution)[0] = 1

#(&cefAllowOSExecution)[0] = <cpp_bool>bool(pyAllowOSExecutionOut[0])

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)

cdef public cpp_bool RequestHandler_OnBeforePluginLoad(

CefRefPtr[CefBrowser] browser,

const CefString& mime_type,

const CefString& plugin_url,

cpp_bool is_main_frame,

const CefString& top_origin_url,

CefRefPtr[CefWebPluginInfo] plugin_info,

cef_types.cef_plugin_policy_t* plugin_policy

) except * with gil:

cdef PyBrowser pyBrowser

cdef PyWebPluginInfo pyInfo

cdef py_bool returnValue

cdef object clientCallback

try:

# OnBeforePluginLoad is called from RequestContexthandler.

# The Browser object might not be available, because it is

# being set synchronously during CreateBrowserSync, after

# Browser is created. From testing it always works, however

# better be safe.

if not browser.get():

Debug("WARNING: RequestHandler_OnBeforePluginLoad() failed,"

" Browser object is not available")

return False

py_browser = GetPyBrowser(browser, "OnBeforePluginLoad")

py_plugin_info = CreatePyWebPluginInfo(plugin_info)

clientCallback = GetGlobalClientCallback("OnBeforePluginLoad")

if clientCallback:

returnValue = clientCallback(

browser=py_browser,

mime_type=CefToPyString(mime_type),

plugin_url=CefToPyString(plugin_url),

is_main_frame=bool(is_main_frame),

top_origin_url=CefToPyString(top_origin_url),

plugin_info=py_plugin_info)

if returnValue:

plugin_policy[0] = cef_types.PLUGIN_POLICY_DISABLE

return bool(returnValue)

else:

return False

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)

cdef public cpp_bool RequestHandler_OnCertificateError(

int certError,

const CefString& cefRequestUrl,

CefRefPtr[CefRequestCallback] cefCertCallback

) except * with gil:

cdef py_bool returnValue

cdef object clientCallback

try:

clientCallback = GetGlobalClientCallback("OnCertificateError")

if clientCallback:

returnValue = clientCallback(

cert_error=certError,

request_url=CefToPyString(cefRequestUrl),

callback=CreatePyRequestCallback(cefCertCallback))

return bool(returnValue)

else:

return False

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)

cdef public void RequestHandler_OnRendererProcessTerminated(

CefRefPtr[CefBrowser] cefBrowser,

cef_types.cef_termination_status_t cefStatus

) except * with gil:

# TODO: proccess may crash during browser creation. Let this callback

# to be set either through cefpython.SetGlobalClientCallback()

# or PyBrowser.SetClientCallback(). Modify the

# PyBrowser.GetClientCallback() implementation to return a global

# callback first if set.

cdef PyBrowser pyBrowser

cdef object clientCallback

try:

pyBrowser = GetPyBrowser(cefBrowser, "OnRendererProcessTerminated")

clientCallback = pyBrowser.GetClientCallback(

"OnRendererProcessTerminated")

if clientCallback:

clientCallback(browser=pyBrowser, status=cefStatus)

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)

cdef public void RequestHandler_OnPluginCrashed(

CefRefPtr[CefBrowser] cefBrowser,

const CefString& cefPluginPath

) except * with gil:

# TODO: plugin may crash during browser creation. Let this callback

# to be set either through cefpython.SetGlobalClientCallback()

# or PyBrowser.SetClientCallback(). Modify the

# PyBrowser.GetClientCallback() implementation to return a global

# callback first if set.

cdef PyBrowser pyBrowser

cdef object clientCallback

try:

pyBrowser = GetPyBrowser(cefBrowser, "OnPluginCrashed")

clientCallback = pyBrowser.GetClientCallback("OnPluginCrashed")

if clientCallback:

clientCallback(

browser=pyBrowser,

plugin_path=CefToPyString(cefPluginPath))

except:

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

sys.excepthook(exc_type, exc_value, exc_trace)