import json
import os
from unittest.mock import Mock, patch

from esphome import vscode


def _run_repl_test(input_data):
    """Reusable test function for different input scenarios."""
    input_data.append(_exit())
    with (
        patch("builtins.input", side_effect=input_data),
        patch("sys.stdout") as mock_stdout,
    ):
        args = Mock([])
        args.ace = False
        args.substitution = None
        vscode.read_config(args)

        # Capture printed output
        full_output = "".join(
            call[0][0] for call in mock_stdout.write.call_args_list
        ).strip()
        splitted_output = full_output.split("\n")
        remove_version = splitted_output[1:]  # remove first entry with version info
        return remove_version


def _validate(file_path: str):
    return json.dumps({"type": "validate", "file": file_path})


def _file_response(data: str):
    return json.dumps({"type": "file_response", "content": data})


def _read_file(file_path: str):
    return json.dumps({"type": "read_file", "path": file_path})


def _exit():
    return json.dumps({"type": "exit"})


RESULT_NO_ERROR = '{"type": "result", "yaml_errors": [], "validation_errors": []}'


def test_multi_file():
    source_path = os.path.join("dir_path", "x.yaml")
    output_lines = _run_repl_test(
        [
            _validate(source_path),
            # read_file x.yaml
            _file_response("""esphome:
  name: test1
esp8266:
  board: !secret my_secret_board
"""),
            # read_file secrets.yaml
            _file_response("""my_secret_board: esp1f"""),
        ]
    )

    expected_lines = [
        _read_file(source_path),
        _read_file(os.path.join("dir_path", "secrets.yaml")),
        RESULT_NO_ERROR,
    ]

    assert output_lines == expected_lines


def test_shows_correct_range_error():
    source_path = os.path.join("dir_path", "x.yaml")
    output_lines = _run_repl_test(
        [
            _validate(source_path),
            # read_file x.yaml
            _file_response("""esphome:
  name: test1
esp8266:
  broad: !secret my_secret_board        # typo here
"""),
            # read_file secrets.yaml
            _file_response("""my_secret_board: esp1f"""),
        ]
    )

    assert len(output_lines) == 3
    error = json.loads(output_lines[2])
    validation_error = error["validation_errors"][0]
    assert validation_error["message"].startswith("[broad] is an invalid option for")
    range = validation_error["range"]
    assert range["document"] == source_path
    assert range["start_line"] == 3
    assert range["start_col"] == 2
    assert range["end_line"] == 3
    assert range["end_col"] == 7


def test_shows_correct_loaded_file_error():
    source_path = os.path.join("dir_path", "x.yaml")
    output_lines = _run_repl_test(
        [
            _validate(source_path),
            # read_file x.yaml
            _file_response("""esphome:
  name: test1

packages:
  board: !include .pkg.esp8266.yaml
"""),
            # read_file .pkg.esp8266.yaml
            _file_response("""esp8266:
  broad: esp1f # typo here
"""),
        ]
    )

    assert len(output_lines) == 3
    error = json.loads(output_lines[2])
    validation_error = error["validation_errors"][0]
    assert validation_error["message"].startswith("[broad] is an invalid option for")
    range = validation_error["range"]
    assert range["document"] == os.path.join("dir_path", ".pkg.esp8266.yaml")
    assert range["start_line"] == 1
    assert range["start_col"] == 2
    assert range["end_line"] == 1
    assert range["end_col"] == 7