Iris/mopidy_iris/__init__.py

from __future__ import unicode_literals

import logging, os, json
import tornado.web
import tornado.websocket
import handlers

from mopidy import config, ext
from frontend import IrisFrontend
from handlers import WebsocketHandler, HttpHandler
from core import IrisCore

logger = logging.getLogger(__name__)
__version__ = '3.41.1'

##
# Core extension class
#
# Loads config and gets the party started. Initiates any additional frontends, etc.
##
class Extension( ext.Extension ):

    dist_name = 'Mopidy-Iris'
    ext_name = 'iris'
    version = __version__

    def get_default_config(self):
        conf_file = os.path.join(os.path.dirname(__file__), 'ext.conf')
        return config.read(conf_file)

    def get_config_schema(self):
        schema = config.ConfigSchema(self.ext_name)
        schema['enabled'] = config.Boolean()
        schema['country'] = config.String()
        schema['locale'] = config.String()
        schema['spotify_authorization_url'] = config.String()
        schema['lastfm_authorization_url'] = config.String()
        schema['genius_authorization_url'] = config.String()
        schema['snapcast_enabled'] = config.Boolean()
        schema['snapcast_host'] = config.String()
        schema['snapcast_port'] = config.Integer()
        schema['snapcast_buffer_size'] = config.Integer()
        return schema

    def setup(self, registry):
        
        # Add web extension
        registry.add('http:app', {
            'name': self.ext_name,
            'factory': iris_factory
        })

        # create our core instance
        mem.iris = IrisCore()
        mem.iris.version = self.version

        # Add our frontend
        registry.add('frontend', IrisFrontend)

##
# Customised handler for react router URLS
#
# This routes all URLs to the same path, so that React can handle the path etc
##
class ReactRouterHandler(tornado.web.StaticFileHandler):
    def initialize(self, path):
        self.path = path
        self.dirname, self.filename = os.path.split(path)
        super(ReactRouterHandler, self).initialize(self.dirname)

    def get(self, path=None, include_body=True):
        super(ReactRouterHandler, self).get(self.path, include_body)

##
# Frontend factory
##
def iris_factory(config, core):

    path = os.path.join( os.path.dirname(__file__), 'static')
    
    return [
        (
            r"/images/(.*)", 
            tornado.web.StaticFileHandler,
            {
                'path': config['local-images']['image_dir']
            }
        ),
        (
            r'/http/([^/]*)',
            handlers.HttpHandler, 
            {
                'core': core,
                'config': config
            }
        ),
        (
            r'/ws/?',
            handlers.WebsocketHandler,
            { 
                'core': core,
                'config': config
            }
        ),
        (
            r'/assets/(.*)',
            tornado.web.StaticFileHandler,
            {
                'path': path+'/assets'
            }
        ),
        (
            r'/((.*)(?:css|js|json|map)$)',
            tornado.web.StaticFileHandler,
            {
                'path': path
            }
        ),
        (
            r'/(.*)',
            ReactRouterHandler, {
                'path': path+'/index.html'
            }
        ),
    ]