"""Unit tests for playbook variable discovery + run-time extra-vars file."""
import json

from steward.ansible.executor import build_extra_vars_file
from steward.ansible.sources import discover_playbook_variables


def _names(variables):
    return [v["name"] for v in variables]


def test_vars_block_defaults_surface_as_fields():
    content = """
- hosts: all
  vars:
    steward_url: ""
    agent_interval: 30
    enabled: true
  tasks: []
"""
    variables = discover_playbook_variables(content)
    assert _names(variables) == ["steward_url", "agent_interval", "enabled"]
    interval = next(v for v in variables if v["name"] == "agent_interval")
    assert interval["default"] == 30
    assert interval["required"] is False
    assert interval["secret"] is False


def test_secretish_names_flagged():
    content = """
- hosts: all
  vars:
    db_password: ""
    api_token: ""
    vault_secret: ""
    plain_value: "x"
"""
    by = {v["name"]: v for v in discover_playbook_variables(content)}
    assert by["db_password"]["secret"] is True
    assert by["api_token"]["secret"] is True
    assert by["vault_secret"]["secret"] is True
    assert by["plain_value"]["secret"] is False


def test_vars_prompt_required_and_private():
    content = """
- hosts: all
  vars_prompt:
    - name: release_tag
      prompt: "Which release?"
    - name: admin_pw
      prompt: "Admin password"
      private: true
      default: ""
  tasks: []
"""
    by = {v["name"]: v for v in discover_playbook_variables(content)}
    # No default → required; vars_prompt defaults to private=yes → secret.
    assert by["release_tag"]["required"] is True
    assert by["release_tag"]["secret"] is True
    assert by["release_tag"]["prompt"] == "Which release?"
    # Explicit private + has default → secret but not required.
    assert by["admin_pw"]["secret"] is True
    assert by["admin_pw"]["required"] is False


def test_vars_prompt_wins_on_name_collision():
    content = """
- hosts: all
  vars_prompt:
    - name: dup
      prompt: "from prompt"
      private: false
  vars:
    dup: "from vars"
"""
    variables = discover_playbook_variables(content)
    assert _names(variables) == ["dup"]
    assert variables[0]["prompt"] == "from prompt"


def test_non_scalar_vars_skipped():
    content = """
- hosts: all
  vars:
    scalar: 1
    a_list: [1, 2]
    a_map: {k: v}
"""
    assert _names(discover_playbook_variables(content)) == ["scalar"]


def test_malformed_yaml_returns_empty():
    assert discover_playbook_variables("this: : : not valid") == []
    assert discover_playbook_variables("just a string") == []


def test_extra_vars_file_roundtrip_and_space_safe():
    merged = {"agent_interval": "30", "msg": "hello world", "q": 'a "quote"'}
    args, files = build_extra_vars_file(merged, "/td")
    assert args == ["-e", "@/td/extravars.json"]
    path, content = files[0]
    assert path == "/td/extravars.json"
    # JSON keeps spaces/quotes intact — no shlex hazard like -e key=value.
    assert json.loads(content) == merged


def test_extra_vars_file_empty_is_noop():
    assert build_extra_vars_file({}, "/td") == ([], [])
    assert build_extra_vars_file(None, "/td") == ([], [])