# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========

from dataclasses import dataclass

from mirage.core.github._client import github_get
from mirage.core.github.config import GitHubConfig
from mirage.types import PathSpec


@dataclass
class SearchResult:
    path: str
    sha: str


async def search_code(
    config: GitHubConfig,
    owner: str,
    repo: str,
    query: str,
    path_filter: str | None = None,
) -> list[SearchResult]:
    q = f"{query} repo:{owner}/{repo}"
    if path_filter:
        q += f" path:{path_filter}"
    data = await github_get(config.token, "/search/code", params={"q": q})
    return [
        SearchResult(path=item["path"], sha=item["sha"])
        for item in data.get("items", [])
    ]


async def narrow_paths(
    config: GitHubConfig,
    owner: str,
    repo: str,
    pattern: str,
    paths: list[PathSpec],
) -> list[PathSpec]:
    """Use GitHub code search to narrow paths for grep/rg.

    Returns narrowed PathSpecs (one per matching file). Empty list means
    search returned no results.
    """
    narrowed: list[str] = []
    for p in paths:
        p_prefix = p.prefix if isinstance(p, PathSpec) else ""
        path_filter = p.original if isinstance(p, PathSpec) else p
        if p_prefix and path_filter.startswith(p_prefix):
            path_filter = path_filter[len(p_prefix):] or "/"
        try:
            results = await search_code(
                config,
                owner,
                repo,
                query=pattern,
                path_filter=path_filter or None,
            )
            narrowed.extend(r.path for r in results)
        except Exception:
            pass
    return [PathSpec(original=n, directory="") for n in narrowed]
