import os
import sys
from pathlib import Path

from flask import Config
from pytest import fixture

from newsroom.web.factory import get_app


root = (Path(__file__).parent / '..').resolve()
sys.path.insert(0, str(root))


def update_config(conf):
    conf['CONTENTAPI_URL'] = 'http://localhost:5400'
    conf['MONGO_DBNAME'] = conf['CONTENTAPI_MONGO_DBNAME'] = 'newsroom_test'
    conf['ELASTICSEARCH_INDEX'] = conf['CONTENTAPI_ELASTICSEARCH_INDEX'] = 'newsroom_test'
    conf['MONGO_URI'] = get_mongo_uri('MONGO_URI', conf['MONGO_DBNAME'])
    conf['CONTENTAPI_MONGO_URI'] = get_mongo_uri('CONTENTAPI_MONGO_URI', conf['MONGO_DBNAME'])
    conf['SERVER_NAME'] = 'localhost:5050'
    conf['WTF_CSRF_ENABLED'] = False
    conf['DEBUG'] = True
    conf['TESTING'] = True
    conf['WEBPACK_ASSETS_URL'] = None
    conf['BABEL_DEFAULT_TIMEZONE'] = 'Europe/Prague'
    conf['DEFAULT_TIMEZONE'] = 'Europe/Prague'
    conf['NEWS_API_ENABLED'] = True
    return conf


def get_mongo_uri(key, dbname):
    """Read mongo uri from env variable and replace dbname.

    :param key: env variable name
    :param dbname: mongo db name to use
    """
    env_uri = os.environ.get(key, 'mongodb://localhost/test')
    env_host = env_uri.rsplit('/', 1)[0]
    return '/'.join([env_host, dbname])


def clean_databases(app):
    app.data.mongo.pymongo().cx.drop_database(app.config['CONTENTAPI_MONGO_DBNAME'])
    indices = '%s*' % app.config['CONTENTAPI_ELASTICSEARCH_INDEX']
    es = app.data.elastic.es
    es.indices.delete(indices, ignore=[404])
    with app.app_context():
        app.data.init_elastic(app)


@fixture
def app():
    cfg = Config(root)
    # cfg.from_object('newsroom.web.default_settings')
    update_config(cfg)
    app = get_app(config=cfg, testing=True)

    return app


@fixture
def client(app):
    return app.test_client()


@fixture(autouse=True)
def setup(app):
    with app.app_context():
        app.data.init_elastic(app)
        clean_databases(app)
        yield