import multiprocessing
import uvicorn
import uvicorn.workers
import gunicorn.app.base

from searx import settings


class CustomUvicornWorker(uvicorn.workers.UvicornWorker):
    CONFIG_KWARGS = {}


class StandaloneApplication(gunicorn.app.base.BaseApplication):

    def __init__(self, app, options=None):
        self.options = options or {}
        self.application = app
        super().__init__()

    def load_config(self):
        config = {key: value for key, value in self.options.items()
                if key in self.cfg.settings and value is not None}
        for key, value in config.items():
            self.cfg.set(key.lower(), value)

    def load(self):
        return self.application


def number_of_workers():
    return 1 # (multiprocessing.cpu_count() * 2) + 1


def run_production(app):
    config_kwargs = {
        "loop": "uvloop",
        "http": "httptools",
    }
    base_url = settings['server']['base_url'] or None
    if base_url:
        # ? config_kwargs['proxy_headers'] = True
        config_kwargs['root_path'] = settings['server']['base_url']

    CustomUvicornWorker.CONFIG_KWARGS.update(config_kwargs)

    options = {
        'proc_name': 'searxng',
        'bind': '%s:%s' % (settings['server']['bind_address'], settings['server']['port']),
        'workers': number_of_workers(),
        'worker_class': 'searx.run.CustomUvicornWorker',
        'loglevel': 'debug',
        'capture_output': True,
    }
    StandaloneApplication(app, options).run()


def run_debug():
    kwargs = {
        'reload': True,
        'loop': 'auto',
        'http': 'auto',
        'ws': 'none',
        'host': settings['server']['bind_address'],
        'port': settings['server']['port'],
    }
    base_url = settings['server']['base_url']
    if base_url:
        kwargs['proxy_headers'] = True
        kwargs['root_path'] = settings['server']['base_url']

    uvicorn.run('searx.__main__:app', **kwargs)