import pytest

from autogpt.config import Config
from autogpt.plugins import (
    blacklist_whitelist_check,
    inspect_zip_for_module,
    scan_plugins,
)

PLUGINS_TEST_DIR = "tests/unit/data/test_plugins"
PLUGIN_TEST_ZIP_FILE = "Auto-GPT-Plugin-Test-master.zip"
PLUGIN_TEST_INIT_PY = "Auto-GPT-Plugin-Test-master/src/auto_gpt_vicuna/__init__.py"
PLUGIN_TEST_OPENAI = "https://weathergpt.vercel.app/"


def test_inspect_zip_for_module():
    result = inspect_zip_for_module(str(f"{PLUGINS_TEST_DIR}/{PLUGIN_TEST_ZIP_FILE}"))
    assert result == PLUGIN_TEST_INIT_PY


@pytest.fixture
def mock_config_blacklist_whitelist_check():
    class MockConfig:
        plugins_blacklist = ["BadPlugin"]
        plugins_whitelist = ["GoodPlugin"]

    return MockConfig()


def test_blacklist_whitelist_check_blacklist(
    mock_config_blacklist_whitelist_check, monkeypatch
):
    monkeypatch.setattr("builtins.input", lambda _: "y")
    assert not blacklist_whitelist_check(
        "BadPlugin", mock_config_blacklist_whitelist_check
    )


def test_blacklist_whitelist_check_whitelist(
    mock_config_blacklist_whitelist_check, monkeypatch
):
    monkeypatch.setattr("builtins.input", lambda _: "y")
    assert blacklist_whitelist_check(
        "GoodPlugin", mock_config_blacklist_whitelist_check
    )


def test_blacklist_whitelist_check_user_input_yes(
    mock_config_blacklist_whitelist_check, monkeypatch
):
    monkeypatch.setattr("builtins.input", lambda _: "y")
    assert blacklist_whitelist_check(
        "UnknownPlugin", mock_config_blacklist_whitelist_check
    )


def test_blacklist_whitelist_check_user_input_no(
    mock_config_blacklist_whitelist_check, monkeypatch
):
    monkeypatch.setattr("builtins.input", lambda _: "n")
    assert not blacklist_whitelist_check(
        "UnknownPlugin", mock_config_blacklist_whitelist_check
    )


def test_blacklist_whitelist_check_user_input_invalid(
    mock_config_blacklist_whitelist_check, monkeypatch
):
    monkeypatch.setattr("builtins.input", lambda _: "invalid")
    assert not blacklist_whitelist_check(
        "UnknownPlugin", mock_config_blacklist_whitelist_check
    )


@pytest.fixture
def config_with_plugins():
    cfg = Config()
    cfg.plugins_dir = PLUGINS_TEST_DIR
    cfg.plugins_openai = ["https://weathergpt.vercel.app/"]
    return cfg


@pytest.fixture
def mock_config_openai_plugin():
    class MockConfig:
        plugins_dir = PLUGINS_TEST_DIR
        plugins_openai = [PLUGIN_TEST_OPENAI]
        plugins_blacklist = ["AutoGPTPVicuna"]
        plugins_whitelist = [PLUGIN_TEST_OPENAI]

    return MockConfig()


def test_scan_plugins_openai(mock_config_openai_plugin):
    result = scan_plugins(mock_config_openai_plugin, debug=True)
    assert len(result) == 1


@pytest.fixture
def mock_config_generic_plugin():
    class MockConfig:
        plugins_dir = PLUGINS_TEST_DIR
        plugins_openai = []
        plugins_blacklist = []
        plugins_whitelist = ["AutoGPTPVicuna"]

    return MockConfig()


def test_scan_plugins_generic(mock_config_generic_plugin):
    result = scan_plugins(mock_config_generic_plugin, debug=True)
    assert len(result) == 1