# SPDX-FileCopyrightText: 2023-2026 Espressif Systems (Shanghai) CO LTD
# SPDX-License-Identifier: Apache-2.0
import logging
import os
import re
import shutil
import subprocess
from pathlib import Path

from test_build_system_helpers import EnvDict
from test_build_system_helpers import run_idf_py


def run_git_cmd(*args: str, workdir: Path, env: EnvDict | None = None) -> subprocess.CompletedProcess:
    cmd = ['git'] + list(args)
    env_dict = dict(**os.environ)
    if env:
        env_dict.update(env)
    logging.debug('running {} in {}'.format(' '.join(cmd), workdir))

    return subprocess.run(cmd, cwd=workdir, env=env_dict, capture_output=True)


# In this test, the action needs to be performed in ESP-IDF that is valid git directory
# Copying ESP-IDF is not possible
def test_git_custom_tag() -> None:
    try:
        env_dict = dict(**os.environ)
        idf_build_test_app_path = Path(env_dict['IDF_PATH'], 'tools', 'test_build_system', 'build_test_app')

        logging.info('Project is in ESP-IDF which has a custom tag')
        env_dict['GIT_COMMITTER_NAME'] = 'No One'
        env_dict['GIT_COMMITTER_EMAIL'] = 'noone@espressif.com'
        run_git_cmd('tag', 'mytag', '-a', '-m', 'mytag', workdir=idf_build_test_app_path, env=env_dict)
        idf_ret = run_idf_py('reconfigure', workdir=idf_build_test_app_path)
        assert 'App "build_test_app" version: mytag' in idf_ret.stdout, 'Project version should be the custom tag'
        version = run_idf_py('--version', workdir=idf_build_test_app_path)
        assert 'mytag' not in version.stdout, 'IDF version should not contain mytag'

    finally:
        run_git_cmd('tag', '-d', 'mytag', workdir=idf_build_test_app_path)
        shutil.rmtree(idf_build_test_app_path / 'build')
        os.unlink(idf_build_test_app_path / 'sdkconfig')


def test_support_git_worktree(minimal_git_app: Path) -> None:
    logging.info('Supports git worktree')
    run_git_cmd('branch', 'test_build_system', workdir=minimal_git_app)
    run_git_cmd('worktree', 'add', '../esp-idf-worktree-app', 'test_build_system', workdir=minimal_git_app)
    try:
        idf_ret_template_app = run_idf_py('reconfigure', workdir=minimal_git_app)
        idf_ret_worktree_app = run_idf_py(
            'reconfigure', workdir=os.path.join(os.path.dirname(minimal_git_app), 'esp-idf-worktree-app')
        )
        assert (
            re.search(r'-- App \"app-template\".*', idf_ret_template_app.stdout).group()  # type: ignore
            == re.search(r'-- App \"app-template\".*', idf_ret_worktree_app.stdout).group()  # type: ignore
        )
    finally:
        run_git_cmd('worktree', 'remove', '../esp-idf-worktree-app', workdir=minimal_git_app)
        run_git_cmd('branch', '-d', 'test_build_system', workdir=minimal_git_app)