#!/usr/bin/env bash
set -euo pipefail

rg -n -A18 -B4 'pub fn import|pub fn import_from|from_list' crates/vm/src
rg -n -A8 -B8 'vm\.import\("asyncio\.(base_futures|base_tasks|exceptions)"' crates/stdlib/src/_asyncio.rs

#!/usr/bin/env bash
set -euo pipefail

printf '%s\n' '--- relevant files ---'
git ls-files | rg '(^|/)(_asyncio|asyncio|test_asyncio|test_tasks|test_futures).*'

printf '%s\n' '--- InvalidStateError and CancelledError references ---'
rg -n -A8 -B8 'InvalidStateError|CancelledError' crates/stdlib tests Lib 2>/dev/null | head -n 500

printf '%s\n' '--- helper call sites ---'
rg -n -A5 -B5 'get_invalid_state_error_type|get_cancelled_error_type' crates/stdlib

printf '%s\n' '--- test assignments near stated lines ---'
rg -n -A8 -B8 'InvalidStateError\s*=|exceptions\.InvalidStateError|asyncio\.InvalidStateError' . --glob '*.py' --glob '*.rs' | head -n 300

#!/usr/bin/env bash
set -euo pipefail

printf '%s\n' '--- asyncio initializer ---'
sed -n '1,180p' Lib/asyncio/__init__.py

printf '%s\n' '--- asyncio exceptions ---'
sed -n '1,180p' Lib/asyncio/exceptions.py

printf '%s\n' '--- exact InvalidStateError assignments and test references ---'
rg -n -S -A10 -B10 \
  'InvalidStateError|exceptions\.InvalidStateError|asyncio\.InvalidStateError' \
  Lib/test crates/stdlib 2>/dev/null || true

printf '%s\n' '--- repository status and recent file list ---'
git status --short
git ls-files | rg 'test.*asyncio|asyncio.*test|_asyncio'

#!/usr/bin/env bash
set -euo pipefail

printf '%s\n' '--- extra asyncio test ---'
sed -n '1,140p' extra_tests/snippets/stdlib_asyncio.py

printf '%s\n' '--- all direct assignments ---'
rg -n -S '(^|[.;[:space:]])(asyncio\.)?(exceptions\.)?InvalidStateError[[:space:]]*=' . 2>/dev/null || true
rg -n -S '(^|[.;[:space:]])(asyncio\.)?(exceptions\.)?CancelledError[[:space:]]*=' . 2>/dev/null || true

printf '%s\n' '--- Rust asyncio module references ---'
rg -n -A8 -B8 'asyncio|InvalidStateError|CancelledError' crates/stdlib/src/_asyncio.rs | tail -n 300

printf '%s\n' '--- test files near the cited line range ---'
find . -type f \( -name '*.py' -o -name '*.rs' \) -print0 |
  xargs -0 grep -n -H -E 'InvalidStateError|CancelledError' 2>/dev/null |
  awk -F: '$2 <= 100 {print}' | head -n 200

#!/usr/bin/env bash
set -euo pipefail

python3 - <<'PY'
import ast
from pathlib import Path

init = ast.parse(Path("Lib/asyncio/__init__.py").read_text())
extra = ast.parse(Path("extra_tests/snippets/stdlib_asyncio.py").read_text())

has_exception_star_import = any(
    isinstance(node, ast.ImportFrom)
    and node.level == 1
    and node.module == "exceptions"
    and any(alias.name == "*" for alias in node.names)
    for node in init.body
)

assignments = []
for node in ast.walk(extra):
    if isinstance(node, ast.Assign):
        targets = [
            ast.unparse(target)
            for target in node.targets
            if isinstance(target, ast.Attribute)
        ]
        if targets and "InvalidStateError" in " ".join(targets):
            assignments.append(targets)

submodule_type = object()
asyncio_module = {"InvalidStateError": submodule_type}
exceptions_module = {"InvalidStateError": submodule_type}
replacement = object()
exceptions_module["InvalidStateError"] = replacement

print(f"star-import-from-exceptions: {has_exception_star_import}")
print(f"test-invalid-state-assignments: {assignments}")
print(f"top-level-lookup-is-submodule-after-submodule-only-change: "
      f"{asyncio_module['InvalidStateError'] is exceptions_module['InvalidStateError']}")
print(f"submodule-lookup-sees-replacement: "
      f"{exceptions_module['InvalidStateError'] is replacement}")

assert has_exception_star_import
assert not (asyncio_module["InvalidStateError"] is exceptions_module["InvalidStateError"])
assert exceptions_module["InvalidStateError"] is replacement
PY

#!/usr/bin/env bash
set -euo pipefail

python3 - <<'PY'
from pathlib import Path

init = Path("Lib/asyncio/__init__.py").read_text()
extra = Path("extra_tests/snippets/stdlib_asyncio.py").read_text()

assert "from .exceptions import *" in init
assert "asyncio.InvalidStateError = replacement" in extra
assert "asyncio.exceptions.InvalidStateError = replacement" in extra

exception_type = object()
asyncio_module = {"InvalidStateError": exception_type}
exceptions_module = {"InvalidStateError": exception_type}

replacement = object()
exceptions_module["InvalidStateError"] = replacement

print("package-reexports-exceptions: True")
print("test-mutates-both-aliases: True")
print("top-level-lookup-after-submodule-only-change: "
      + str(asyncio_module["InvalidStateError"] is replacement))
print("submodule-lookup-after-submodule-only-change: "
      + str(exceptions_module["InvalidStateError"] is replacement))

assert asyncio_module["InvalidStateError"] is not replacement
assert exceptions_module["InvalidStateError"] is replacement
PY