import os

import pytest

from solidlsp import SolidLanguageServer
from solidlsp.ls_config import Language
from solidlsp.ls_utils import SymbolUtils
from test.conftest import language_tests_enabled
from test.solidlsp.conftest import format_symbol_for_assert, has_malformed_name, request_all_symbols


@pytest.mark.skipif(not language_tests_enabled(Language.ELM), reason="Elm tests are disabled (elm compiler not available)")
@pytest.mark.elm
class TestElmLanguageServer:
    @pytest.mark.parametrize("language_server", [Language.ELM], indirect=True)
    def test_find_symbol(self, language_server: SolidLanguageServer) -> None:
        symbols = language_server.request_full_symbol_tree()
        assert SymbolUtils.symbol_tree_contains_name(symbols, "greet"), "greet function not found in symbol tree"
        assert SymbolUtils.symbol_tree_contains_name(symbols, "calculateSum"), "calculateSum function not found in symbol tree"
        assert SymbolUtils.symbol_tree_contains_name(symbols, "formatMessage"), "formatMessage function not found in symbol tree"
        assert SymbolUtils.symbol_tree_contains_name(symbols, "addNumbers"), "addNumbers function not found in symbol tree"

    @pytest.mark.parametrize("language_server", [Language.ELM], indirect=True)
    def test_find_references_within_file(self, language_server: SolidLanguageServer) -> None:
        file_path = os.path.join("Main.elm")
        symbols = language_server.request_document_symbols(file_path).get_all_symbols_and_roots()
        greet_symbol = None
        for sym in symbols[0]:
            if sym.get("name") == "greet":
                greet_symbol = sym
                break
        assert greet_symbol is not None, "Could not find 'greet' symbol in Main.elm"
        sel_start = greet_symbol["selectionRange"]["start"]
        refs = language_server.request_references(file_path, sel_start["line"], sel_start["character"])
        assert any("Main.elm" in ref.get("relativePath", "") for ref in refs), "Main.elm should reference greet function"

    @pytest.mark.parametrize("language_server", [Language.ELM], indirect=True)
    def test_find_references_across_files(self, language_server: SolidLanguageServer) -> None:
        # Test formatMessage function which is defined in Utils.elm and used in Main.elm
        utils_path = os.path.join("Utils.elm")
        symbols = language_server.request_document_symbols(utils_path).get_all_symbols_and_roots()
        formatMessage_symbol = None
        for sym in symbols[0]:
            if sym.get("name") == "formatMessage":
                formatMessage_symbol = sym
                break
        assert formatMessage_symbol is not None, "Could not find 'formatMessage' symbol in Utils.elm"

        # Get references from the definition in Utils.elm
        sel_start = formatMessage_symbol["selectionRange"]["start"]
        refs = language_server.request_references(utils_path, sel_start["line"], sel_start["character"])

        # Verify that we found references
        assert refs, "Expected to find references for formatMessage"

        # Verify that at least one reference is in Main.elm (where formatMessage is used)
        assert any("Main.elm" in ref.get("relativePath", "") for ref in refs), "Expected to find usage of formatMessage in Main.elm"

    @pytest.mark.parametrize("language_server", [Language.ELM], indirect=True)
    def test_bare_symbol_names(self, language_server) -> None:
        all_symbols = request_all_symbols(language_server)
        malformed_symbols = []
        for s in all_symbols:
            if has_malformed_name(s):
                malformed_symbols.append(s)
        if malformed_symbols:
            pytest.fail(
                f"Found malformed symbols: {[format_symbol_for_assert(sym) for sym in malformed_symbols]}",
                pytrace=False,
            )