"""One-off: rewrite imports for advisor-agent merge (run from repo root)."""
from __future__ import annotations

import re
from pathlib import Path

ROOT = Path(__file__).resolve().parents[2]

REPLACEMENTS = [
    ("from app.model.schemas import", "from app.model.advisor_schemas import"),
    ("from app.model.entities import ScriptTemplate", "from app.model.entities_advisor import ScriptTemplate"),
    ("from app.model.entities import ComplianceRule", "from app.model.entities_advisor import ComplianceRule"),
    ("from app.model.entities import ComplianceCheckLog", "from app.model.entities_advisor import ComplianceCheckLog"),
    ("from app.model.entities import CopyTrackLog", "from app.model.entities_advisor import CopyTrackLog"),
    ("from app.model.entities import KycSession", "from app.model.entities_advisor import KycSession"),
    ("from app.model.entities import MarketAlert", "from app.model.entities_advisor import MarketAlert"),
    ("from app.model.entities import TemplateUseLog", "from app.model.entities_advisor import TemplateUseLog"),
    ("from app.repository.template_repository import", "from app.repository.script_template_repository import"),
    ("from app.service.template_vector_service import", "from app.service.script_template_vector_service import"),
    ("from app.service.template_service import", "from app.service.script_template_service import"),
    ("TemplateRepository", "ScriptTemplateRepository"),
    ("TemplateVectorService", "ScriptTemplateVectorService"),
    ("class TemplateService:", "class ScriptTemplateService:"),
    ("TemplateService(", "ScriptTemplateService("),
    ("from app.api.deps import get_auth_context, require_permission", "from app.api.advisor_auth_adapter import get_advisor_auth, require_advisor_permission"),
    ("from app.api.deps import require_permission", "from app.api.advisor_auth_adapter import require_advisor_permission"),
    ("from app.api.deps import get_auth_context", "from app.api.advisor_auth_adapter import get_advisor_auth"),
    ("require_permission(", "require_advisor_permission("),
    ("get_auth_context", "get_advisor_auth"),
    ("AuthContext", "AdvisorAuthContext"),
    ("from app.model.advisor_schemas import AdvisorAuthContext", "from app.api.advisor_auth_adapter import AdvisorAuthContext"),
    ("success_response", "_advisor_ok"),
    ("from app.utils.response import _advisor_ok", "from app.api.advisor_http import advisor_ok as _advisor_ok"),
]

GLOBS = [
    "app/service/compliance_*.py",
    "app/service/copy_*.py",
    "app/service/kyc_*.py",
    "app/service/market_*.py",
    "app/service/script_template*.py",
    "app/service/input_guard_service.py",
    "app/service/ownership_service.py",
    "app/service/llm_client.py",
    "app/repository/compliance_*.py",
    "app/repository/copy_*.py",
    "app/repository/kyc_*.py",
    "app/repository/market_*.py",
    "app/repository/script_template*.py",
    "app/api/allocation.py",
    "app/api/copy.py",
    "app/api/dashboard.py",
    "app/api/guard.py",
    "app/api/kyc.py",
    "app/api/market.py",
    "app/api/templates.py",
    "scripts/seed/*.py",
    "scripts/sync/sync_template_vectors.py",
]

def main() -> None:
    for pattern in GLOBS:
        for path in ROOT.glob(pattern):
            text = path.read_text(encoding="utf-8")
            orig = text
            for old, new in REPLACEMENTS:
                text = text.replace(old, new)
            if text != orig:
                path.write_text(text, encoding="utf-8")
                print("fixed", path.relative_to(ROOT))

if __name__ == "__main__":
    main()