217 lines
7.7 KiB
Python
217 lines
7.7 KiB
Python
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
import logging
|
|
import os
|
|
import signal
|
|
import sys
|
|
|
|
from mopidy.internal.gi import Gst # noqa: F401
|
|
|
|
try:
|
|
# Make GObject's mainloop the event loop for python-dbus
|
|
import dbus.mainloop.glib
|
|
dbus.mainloop.glib.threads_init()
|
|
dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
|
|
except ImportError:
|
|
pass
|
|
|
|
import pykka.debug
|
|
|
|
from mopidy import commands, config as config_lib, ext
|
|
from mopidy.internal import encoding, log, path, process, versioning
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def main():
|
|
log.bootstrap_delayed_logging()
|
|
logger.info('Starting Mopidy %s', versioning.get_version())
|
|
|
|
signal.signal(signal.SIGTERM, process.sigterm_handler)
|
|
# Windows does not have signal.SIGUSR1
|
|
if hasattr(signal, 'SIGUSR1'):
|
|
signal.signal(signal.SIGUSR1, pykka.debug.log_thread_tracebacks)
|
|
|
|
try:
|
|
registry = ext.Registry()
|
|
|
|
root_cmd = commands.RootCommand()
|
|
config_cmd = commands.ConfigCommand()
|
|
deps_cmd = commands.DepsCommand()
|
|
|
|
root_cmd.set(extension=None, registry=registry)
|
|
root_cmd.add_child('config', config_cmd)
|
|
root_cmd.add_child('deps', deps_cmd)
|
|
|
|
extensions_data = ext.load_extensions()
|
|
|
|
for data in extensions_data:
|
|
if data.command: # TODO: check isinstance?
|
|
data.command.set(extension=data.extension)
|
|
root_cmd.add_child(data.extension.ext_name, data.command)
|
|
|
|
args = root_cmd.parse(sys.argv[1:])
|
|
|
|
config, config_errors = config_lib.load(
|
|
args.config_files,
|
|
[d.config_schema for d in extensions_data],
|
|
[d.config_defaults for d in extensions_data],
|
|
args.config_overrides)
|
|
|
|
create_core_dirs(config)
|
|
create_initial_config_file(args, extensions_data)
|
|
|
|
verbosity_level = args.base_verbosity_level
|
|
if args.verbosity_level:
|
|
verbosity_level += args.verbosity_level
|
|
|
|
log.setup_logging(config, verbosity_level, args.save_debug_log)
|
|
|
|
extensions = {
|
|
'validate': [], 'config': [], 'disabled': [], 'enabled': []}
|
|
for data in extensions_data:
|
|
extension = data.extension
|
|
|
|
# TODO: factor out all of this to a helper that can be tested
|
|
if not ext.validate_extension_data(data):
|
|
config[extension.ext_name] = {'enabled': False}
|
|
config_errors[extension.ext_name] = {
|
|
'enabled': 'extension disabled by self check.'}
|
|
extensions['validate'].append(extension)
|
|
elif not config[extension.ext_name]['enabled']:
|
|
config[extension.ext_name] = {'enabled': False}
|
|
config_errors[extension.ext_name] = {
|
|
'enabled': 'extension disabled by user config.'}
|
|
extensions['disabled'].append(extension)
|
|
elif config_errors.get(extension.ext_name):
|
|
config[extension.ext_name]['enabled'] = False
|
|
config_errors[extension.ext_name]['enabled'] = (
|
|
'extension disabled due to config errors.')
|
|
extensions['config'].append(extension)
|
|
else:
|
|
extensions['enabled'].append(extension)
|
|
|
|
log_extension_info([d.extension for d in extensions_data],
|
|
extensions['enabled'])
|
|
|
|
# Config and deps commands are simply special cased for now.
|
|
if args.command == config_cmd:
|
|
schemas = [d.config_schema for d in extensions_data]
|
|
return args.command.run(config, config_errors, schemas)
|
|
elif args.command == deps_cmd:
|
|
return args.command.run()
|
|
|
|
check_config_errors(config, config_errors, extensions)
|
|
|
|
if not extensions['enabled']:
|
|
logger.error('No extension enabled, exiting...')
|
|
sys.exit(1)
|
|
|
|
# Read-only config from here on, please.
|
|
proxied_config = config_lib.Proxy(config)
|
|
|
|
if args.extension and args.extension not in extensions['enabled']:
|
|
logger.error(
|
|
'Unable to run command provided by disabled extension %s',
|
|
args.extension.ext_name)
|
|
return 1
|
|
|
|
for extension in extensions['enabled']:
|
|
try:
|
|
extension.setup(registry)
|
|
except Exception:
|
|
# TODO: would be nice a transactional registry. But sadly this
|
|
# is a bit tricky since our current API is giving out a mutable
|
|
# list. We might however be able to replace this with a
|
|
# collections.Sequence to provide a RO view.
|
|
logger.exception('Extension %s failed during setup, this might'
|
|
' have left the registry in a bad state.',
|
|
extension.ext_name)
|
|
|
|
# Anything that wants to exit after this point must use
|
|
# mopidy.internal.process.exit_process as actors can have been started.
|
|
try:
|
|
return args.command.run(args, proxied_config)
|
|
except NotImplementedError:
|
|
print(root_cmd.format_help())
|
|
return 1
|
|
|
|
except KeyboardInterrupt:
|
|
pass
|
|
except Exception as ex:
|
|
logger.exception(ex)
|
|
raise
|
|
|
|
|
|
def create_core_dirs(config):
|
|
path.get_or_create_dir(config['core']['cache_dir'])
|
|
path.get_or_create_dir(config['core']['config_dir'])
|
|
path.get_or_create_dir(config['core']['data_dir'])
|
|
|
|
|
|
def create_initial_config_file(args, extensions_data):
|
|
"""Initialize whatever the last config file is with defaults"""
|
|
|
|
config_file = args.config_files[-1]
|
|
|
|
if os.path.exists(path.expand_path(config_file)):
|
|
return
|
|
|
|
try:
|
|
default = config_lib.format_initial(extensions_data)
|
|
path.get_or_create_file(config_file, mkdir=False, content=default)
|
|
logger.info('Initialized %s with default config', config_file)
|
|
except IOError as error:
|
|
logger.warning(
|
|
'Unable to initialize %s with default config: %s',
|
|
config_file, encoding.locale_decode(error))
|
|
|
|
|
|
def log_extension_info(all_extensions, enabled_extensions):
|
|
# TODO: distinguish disabled vs blocked by env?
|
|
enabled_names = set(e.ext_name for e in enabled_extensions)
|
|
disabled_names = set(e.ext_name for e in all_extensions) - enabled_names
|
|
logger.info(
|
|
'Enabled extensions: %s', ', '.join(enabled_names) or 'none')
|
|
logger.info(
|
|
'Disabled extensions: %s', ', '.join(disabled_names) or 'none')
|
|
|
|
|
|
def check_config_errors(config, errors, extensions):
|
|
fatal_errors = []
|
|
extension_names = {}
|
|
all_extension_names = set()
|
|
|
|
for state in extensions:
|
|
extension_names[state] = set(e.ext_name for e in extensions[state])
|
|
all_extension_names.update(extension_names[state])
|
|
|
|
for section in sorted(errors):
|
|
if not errors[section]:
|
|
continue
|
|
|
|
if section not in all_extension_names:
|
|
logger.warning('Found fatal %s configuration errors:', section)
|
|
fatal_errors.append(section)
|
|
elif section in extension_names['config']:
|
|
del errors[section]['enabled']
|
|
logger.warning('Found %s configuration errors, the extension '
|
|
'has been automatically disabled:', section)
|
|
else:
|
|
continue
|
|
|
|
for field, msg in errors[section].items():
|
|
logger.warning(' %s/%s %s', section, field, msg)
|
|
|
|
if extensions['config']:
|
|
logger.warning('Please fix the extension configuration errors or '
|
|
'disable the extensions to silence these messages.')
|
|
|
|
if fatal_errors:
|
|
logger.error('Please fix fatal configuration errors, exiting...')
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|