# Copyright (c) 2012-2014 The CEF Python authors. All rights reserved.
# License: New BSD License.
# Website: http://code.google.com/p/cefpython/

include "cefpython.pyx"

import weakref

# -----------------------------------------------------------------------------
# PyStringVisitor
# -----------------------------------------------------------------------------

cdef object g_userStringVisitors = weakref.WeakValueDictionary()
cdef int g_userStringVisitorMaxId = 0

cdef int StoreUserStringVisitor(object userStringVisitor) except *:
    global g_userStringVisitorMaxId
    global g_userStringVisitors
    g_userStringVisitorMaxId += 1
    g_userStringVisitors[g_userStringVisitorMaxId] = userStringVisitor
    return g_userStringVisitorMaxId

cdef PyStringVisitor GetUserStringVisitor(int stringVisitorId):
    global g_userStringVisitors
    cdef object userStringVisitor
    cdef PyStringVisitor pyStringVisitor
    if stringVisitorId in g_userStringVisitors:
        userStringVisitor = g_userStringVisitors[stringVisitorId]
        pyStringVisitor = PyStringVisitor(userStringVisitor)
        return pyStringVisitor

cdef class PyStringVisitor:
    cdef object userStringVisitor

    def __init__(self, object userStringVisitor):
        self.userStringVisitor = userStringVisitor

    cdef object GetCallback(self, str funcName):
        if self.userStringVisitor and (
                hasattr(self.userStringVisitor, funcName) and (
                callable(getattr(self.userStringVisitor, funcName)))):
            return getattr(self.userStringVisitor, funcName)

# -----------------------------------------------------------------------------
# C++ StringVisitor
# -----------------------------------------------------------------------------

cdef CefRefPtr[CefStringVisitor] CreateStringVisitor(
        object userStringVisitor) except *:
    if not userStringVisitor:
        raise Exception("userStringVisitor object missing")
    if not hasattr(userStringVisitor, "Visit"):
        raise Exception("userStringVisitor object is missing Visit() method")
    cdef int stringVisitorId = StoreUserStringVisitor(userStringVisitor)
    cdef CefRefPtr[CefStringVisitor] cefStringVisitor = (
            new StringVisitor(stringVisitorId))
    return cefStringVisitor

cdef public void StringVisitor_Visit(
        int stringVisitorId,
        const CefString& string
        ) except * with gil:
    cdef str pyString
    cdef PyStringVisitor userStringVisitor
    cdef object callback
    try:
        pyString = CefToPyString(string)
        userStringVisitor = GetUserStringVisitor(stringVisitorId)
        if userStringVisitor:
            callback = userStringVisitor.GetCallback("Visit")
            if callback:
                callback(pyString)
    except:
        (exc_type, exc_value, exc_trace) = sys.exc_info()
        sys.excepthook(exc_type, exc_value, exc_trace)