"""Tools for enabling and registering chart themes"""

from ...utils.theme import ThemeRegistry

VEGA_THEMES = ['ggplot2', 'quartz', 'vox', 'fivethirtyeight', 'dark', 'latimes']


class VegaTheme(object):
    """Implementation of a builtin vega theme."""
    def __init__(self, theme):
        self.theme = theme
        
    def __call__(self):
        return {"usermeta": {"embedOptions": {"theme": self.theme}},
                "config": {"view": {"continuousWidth": 400, "continuousHeight": 300}}}

    def __repr__(self):
        return "VegaTheme({!r})".format(self.theme)


# The entry point group that can be used by other packages to declare other
# renderers that will be auto-detected. Explicit registration is also
# allowed by the PluginRegistery API.
ENTRY_POINT_GROUP = 'altair.vegalite.v4.theme'  # type: str
themes = ThemeRegistry(entry_point_group=ENTRY_POINT_GROUP)

themes.register('default', lambda: {"config": {"view": {"continuousWidth": 400, "continuousHeight": 300}}})
themes.register('opaque', lambda: {"config": {"background": "white",
                                              "view": {"continuousWidth": 400, "continuousHeight": 300}}})
themes.register('none', lambda: {})
  
for theme in VEGA_THEMES:
    themes.register(theme, VegaTheme(theme))

themes.enable('default')

"""Tools for enabling and registering chart themes"""

from ...utils.theme import ThemeRegistry

VEGA_THEMES = ['ggplot2', 'quartz', 'vox', 'fivethirtyeight', 'dark', 'latimes', 'urbaninstitute']


class VegaTheme(object):
    """Implementation of a builtin vega theme."""
    def __init__(self, theme):
        self.theme = theme
        
    def __call__(self):
        return {"usermeta": {"embedOptions": {"theme": self.theme}},
                "config": {"view": {"continuousWidth": 400, "continuousHeight": 300}}}

    def __repr__(self):
        return "VegaTheme({!r})".format(self.theme)


# The entry point group that can be used by other packages to declare other
# renderers that will be auto-detected. Explicit registration is also
# allowed by the PluginRegistery API.
ENTRY_POINT_GROUP = 'altair.vegalite.v4.theme'  # type: str
themes = ThemeRegistry(entry_point_group=ENTRY_POINT_GROUP)

themes.register('default', lambda: {"config": {"view": {"continuousWidth": 400, "continuousHeight": 300}}})
themes.register('opaque', lambda: {"config": {"background": "white",
                                              "view": {"continuousWidth": 400, "continuousHeight": 300}}})
themes.register('none', lambda: {})
  
for theme in VEGA_THEMES:
    themes.register(theme, VegaTheme(theme))

themes.enable('default')

add Urban Institute theme