import asyncio
from magika_enhanced import MagikaAsync

async def main():
    m = MagikaAsync()
    
    # Async single file
    result = await m.identify_path("./document.pdf")
    print(f"{result.path}: {result.output.label} ({result.score:.2%})")
    
    # Async batch with concurrency control
    results = await m.identify_paths(["./file1", "./file2", "./file3"], max_concurrent=10)
    
    # Streaming for large files
    async with aiofiles.open("./large.bin", "rb") as f:
        result = await m.identify_stream(f)

asyncio.run(main())

# rules/african_mobile_money.yaml
rules:
  - name: "mpesa_transaction"
    label: "MPESA_TRANSACTION"
    group: "finance"
    mime_type: "application/x-mpesa"
    extensions: [".mpesa", ".txn"]
    magic_bytes:
      - offset: 0
        pattern: "MPESA\x00"
    description: "M-Pesa mobile money transaction record"
    is_text: false
    priority: 100
  
  - name: "airtel_money_log"
    label: "AIRTEL_MONEY_LOG"
    group: "finance"
    mime_type: "text/x-airtel-money"
    extensions: [".aml"]
    regex: "^AML\\|\\d{10}\\|"
    description: "Airtel Money transaction log"
    is_text: true
    priority: 90

from magika_enhanced import Magika, CustomRules

# Load custom rules
custom = CustomRules.from_yaml("rules/african_mobile_money.yaml")
m = Magika(rules=custom)

result = m.identify_path("./transaction.mpesa")
print(result.output.label)  # MPESA_TRANSACTION

# Start MCP server (stdio transport for Claude Code)
magika-mcp --transport stdio

# Or HTTP transport for web agents
magika-mcp --transport http --host 0.0.0.0 --port 8080

# From PyPI (when published)
pip install magika-enhanced

# With MCP support
pip install magika-enhanced[mcp]

# With benchmarks
pip install magika-enhanced[benchmarks]

# Development install
git clone https://github.com/nakisisageorge/magika-enhanced
cd magika-enhanced
pip install -e ".[dev,mcp,benchmarks]"

# CLI usage (compatible with upstream magika)
magika-enhanced -r ./my_project --json
magika-enhanced ./file.pdf --mime-type
magika-enhanced -r ./data --format jsonl --output-score

# Async Python API
python -c "
import asyncio
from magika_enhanced import MagikaAsync
async def main():
    m = MagikaAsync()
    res = await m.identify_path('./test.py')
    print(res.output.label, res.score)
asyncio.run(main())
"

┌─────────────┐     ┌──────────────┐     ┌─────────────┐
│   CLI /     │────▶│  magika_enhanced │────▶│   magika    │
│   Python API│     │  (async, rules,  │     │  (PyO3 →    │
│   MCP Server│     │   cache, MCP)   │     │   Rust core)│
└─────────────┘     └──────────────┘     └──────┬──────┘
                                                │
                    ┌──────────────┐            │
                    │ Custom Rules │────────────┤
                    │   Engine     │            │
                    └──────────────┘            ▼
                                         ┌─────────────┐
                                         │ ONNX Model  │
                                         │ (~3 MB)     │
                                         └─────────────┘

# Run all tests
pytest

# With coverage
pytest --cov=magika_enhanced --cov-report=html

# Benchmarks
pytest tests/benchmarks/ -v --benchmark-only

# Property-based tests
pytest tests/unit/test_rules.py::test_rules_property -v