import json
import os
import pytest
import requests
import signal
import time

from hashlib import sha512
from subprocess import call

from leap.soledad.client import Soledad
from leap.soledad.common.couch import CouchDatabase

# we have to manually setup the events server in order to be able to signal
# events. This is usually done by the enclosing application using soledad
# client (i.e. bitmask client).
from leap.common.events import server
server.ensure_server()


DEFAULT_UUID = '0'
DEFAULT_PASSPHRASE = '123'

DEFAULT_URL = 'http://127.0.0.1:2424'
DEFAULT_PRIVKEY = 'soledad_privkey.pem'
DEFAULT_CERTKEY = 'soledad_certkey.pem'
DEFAULT_TOKEN = 'an-auth-token'


#
# soledad_dbs fixture: provides all databases needed by soledad server in a per
# module scope (same databases for all tests in this module).
#

def _token_dbname():
    dbname = 'tokens_' + \
        str(int(time.time() / (30 * 24 * 3600)))
    return dbname


class SoledadDatabases(object):

    def __init__(self):
        url = 'http://127.0.0.1:5984/'
        self._token_db_url = url + _token_dbname()
        self._shared_db_url = url + 'shared'

    def setup(self):
        self._create_dbs()
        self._add_token()

    def _create_dbs(self):
        requests.put(self._token_db_url)
        requests.put(self._shared_db_url)

    def _add_token(self):
        token = sha512(DEFAULT_TOKEN).hexdigest()
        content = {'type': 'Token', 'user_id': DEFAULT_UUID}
        requests.put(
            self._token_db_url + '/' + token, data=json.dumps(content))

    def teardown(self):
        requests.delete(self._token_db_url)
        requests.delete(self._shared_db_url)


@pytest.fixture(scope='module')
def soledad_dbs(request):
    db = SoledadDatabases()
    db.setup()
    request.addfinalizer(db.teardown)
    return db


#
# user_db fixture: provides an empty database for a given user in a per
# function scope.
#

class UserDatabase(object):

    def __init__(self):
        url = 'http://127.0.0.1:5984/'
        self._user_db_url = url + 'user-%s' % DEFAULT_UUID

    def setup(self):
        CouchDatabase.open_database(
            url=self._user_db_url, create=True, replica_uid=None)

    def teardown(self):
        requests.delete(self._user_db_url)


@pytest.fixture(scope='function')
def user_db(request):
    db = UserDatabase()
    db.setup()
    request.addfinalizer(db.teardown)
    return db


def get_pid(pidfile):
    if not os.path.isfile(pidfile):
        return 0
    try:
        with open(pidfile) as f:
            return int(f.read())
    except IOError:
        return 0


#
# soledad_server fixture: provides a running soledad server in a per module
# context (same soledad server for all tests in this module).
#

class SoledadServer(object):

    def __init__(self, tmpdir_factory):
        tmpdir = tmpdir_factory.mktemp('soledad-server')
        self._pidfile = os.path.join(tmpdir.strpath, 'soledad-server.pid')
        self._logfile = os.path.join(tmpdir.strpath, 'soledad-server.log')

    def start(self):
        call([
            'twistd',
            '--logfile=%s' % self._logfile,
            '--pidfile=%s' % self._pidfile,
            'web',
            '--wsgi=leap.soledad.server.application',
            '--port=2424'
        ])

    def stop(self):
        pid = get_pid(self._pidfile)
        os.kill(pid, signal.SIGKILL)


@pytest.fixture(scope='module')
def soledad_server(tmpdir_factory, request):
    server = SoledadServer(tmpdir_factory)
    server.start()
    request.addfinalizer(server.stop)
    return server


#
# soledad_client fixture: provides a clean soledad client for a test function.
#

@pytest.fixture()
def soledad_client(tmpdir, soledad_server, user_db, soledad_dbs):
    uuid = DEFAULT_UUID
    passphrase = DEFAULT_PASSPHRASE
    secrets_path = os.path.join(tmpdir.strpath, '%s.secret' % uuid)
    local_db_path = os.path.join(tmpdir.strpath, '%s.db' % uuid)
    server_url = DEFAULT_URL
    token = DEFAULT_TOKEN

    # get a soledad instance
    return Soledad(
        uuid,
        unicode(passphrase),
        secrets_path=secrets_path,
        local_db_path=local_db_path,
        server_url=server_url,
        cert_file=None,
        auth_token=token,
        defer_encryption=True)