#!/usr/bin/env python

"""
Test collection management
"""

from http import HTTPStatus
import json
import os
import pwd
import time

from pytest_cs import wait_for_log, wait_for_http

import pytest
import yaml

pytestmark = pytest.mark.docker


def test_install_two_collections(crowdsec, flavor):
    """Test installing collections at startup"""
    it1 = 'crowdsecurity/apache2'
    it2 = 'crowdsecurity/asterisk'
    env = {
        'COLLECTIONS': f'{it1} {it2}'
    }
    with crowdsec(flavor=flavor, environment=env) as cont:
        wait_for_http(cont, 8080, '/health', want_status=HTTPStatus.OK)
        res = cont.exec_run('cscli collections list -o json')
        assert res.exit_code == 0
        j = json.loads(res.output)
        items = {c['name']: c for c in j['collections']}
        assert items[it1]['status'] == 'enabled'
        assert items[it2]['status'] == 'enabled'
        wait_for_log(cont, [
            # f'*collections install "{it1}"*'
            # f'*collections install "{it2}"*'
            f'*Enabled collections : {it1}*',
            f'*Enabled collections : {it2}*',
        ])


def test_disable_collection(crowdsec, flavor):
    """Test removing a pre-installed collection at startup"""
    it = 'crowdsecurity/linux'
    env = {
        'DISABLE_COLLECTIONS': it
    }
    with crowdsec(flavor=flavor, environment=env) as cont:
        wait_for_log(cont, "*Starting processing data*")
        wait_for_http(cont, 8080, '/health', want_status=HTTPStatus.OK)
        res = cont.exec_run('cscli collections list -o json')
        assert res.exit_code == 0
        j = json.loads(res.output)
        items = {c['name'] for c in j['collections']}
        assert it not in items
        wait_for_log(cont, [
            # f'*collections remove "{it}*",
            f'*Removed symlink [[]{it}[]]*',
        ])


def test_install_and_disable_collection(crowdsec, flavor):
    """Declare a collection to install AND disable: disable wins"""
    it = 'crowdsecurity/apache2'
    env = {
        'COLLECTIONS': it,
        'DISABLE_COLLECTIONS': it,
    }
    with crowdsec(flavor=flavor, environment=env) as cont:
        wait_for_log(cont, "*Starting processing data*")
        wait_for_http(cont, 8080, '/health', want_status=HTTPStatus.OK)
        res = cont.exec_run('cscli collections list -o json')
        assert res.exit_code == 0
        j = json.loads(res.output)
        items = {c['name'] for c in j['collections']}
        assert it not in items
        logs = cont.logs().decode().splitlines()
        # check that there was no attempt to install
        assert not any(f'Enabled collections : {it}' in line for line in logs)


# already done in bats, prividing here as example of a somewhat complex test
def test_taint_bubble_up(crowdsec, tmp_path_factory, flavor):
    coll = 'crowdsecurity/nginx'
    env = {
        'COLLECTIONS': f'{coll}'
    }

    hub = tmp_path_factory.mktemp("hub")
    volumes = {
        hub: {'bind': '/etc/crowdsec/hub', 'mode': 'rw'}
    }

    with crowdsec(flavor=flavor, environment=env, volumes=volumes) as cont:
        wait_for_http(cont, 8080, '/health', want_status=HTTPStatus.OK)
        res = cont.exec_run('cscli collections list -o json')
        assert res.exit_code == 0
        j = json.loads(res.output)
        items = {c['name']: c for c in j['collections']}
        # implicit check for tainted=False
        assert items[coll]['status'] == 'enabled'
        wait_for_log(cont, [
            f'*Enabled collections : {coll}*',
        ])

        # change file permissions to allow edit
        current_uid = pwd.getpwuid(os.getuid()).pw_uid
        res = cont.exec_run(f'chown -R {current_uid} /etc/crowdsec/hub')
        assert res.exit_code == 0

    scenario = 'crowdsecurity/http-crawl-non_statics'
    scenario_file = hub / f'scenarios/{scenario}.yaml'

    with open(scenario_file) as f:
        yml = yaml.safe_load(f)

    yml['description'] += ' (tainted)'
    # won't be able to read it back because description is taken from the index

    with open(scenario_file, 'w') as f:
        yaml.dump(yml, f)

    with crowdsec(flavor=flavor, environment=env, volumes=volumes) as cont:
        wait_for_http(cont, 8080, '/health', want_status=HTTPStatus.OK)
        res = cont.exec_run(f'cscli scenarios inspect {scenario} -o json')
        assert res.exit_code == 0
        j = json.loads(res.output)
        assert j['tainted'] is True

        res = cont.exec_run('cscli collections list -o json')
        assert res.exit_code == 0
        j = json.loads(res.output)
        items = {c['name']: c for c in j['collections']}
        assert items['crowdsecurity/nginx']['status'] == 'enabled,tainted'
        assert items['crowdsecurity/base-http-scenarios']['status'] == 'enabled,tainted'