114 lines
3.6 KiB
Python
Executable File
114 lines
3.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""Generate a cryptographically secure API key for AegisGitea MCP."""
|
|
|
|
import sys
|
|
from datetime import datetime, timedelta, timezone
|
|
from pathlib import Path
|
|
|
|
# Add src to path for imports
|
|
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
|
|
|
|
from aegis_gitea_mcp.auth import generate_api_key
|
|
|
|
|
|
def main() -> None:
|
|
"""Generate and display a new API key."""
|
|
print("=" * 70)
|
|
print("AegisGitea MCP - API Key Generator")
|
|
print("=" * 70)
|
|
print()
|
|
|
|
# Get optional description
|
|
description = input("Enter description for this key (e.g., 'ChatGPT Business'): ").strip()
|
|
if not description:
|
|
description = "Generated key"
|
|
|
|
# Generate key
|
|
api_key = generate_api_key(length=64)
|
|
|
|
# Calculate expiration date (90 days from now)
|
|
created_at = datetime.now(timezone.utc)
|
|
expires_at = created_at + timedelta(days=90)
|
|
|
|
print()
|
|
print("✓ API Key Generated Successfully!")
|
|
print()
|
|
print("-" * 70)
|
|
print(f"Description: {description}")
|
|
print(f"Created: {created_at.strftime('%Y-%m-%d %H:%M:%S UTC')}")
|
|
print(f"Expires: {expires_at.strftime('%Y-%m-%d %H:%M:%S UTC')} (90 days)")
|
|
print("-" * 70)
|
|
print()
|
|
print("API KEY:")
|
|
print(f" {api_key}")
|
|
print()
|
|
print("-" * 70)
|
|
print()
|
|
print("📋 Next Steps:")
|
|
print()
|
|
print("1. Add this key to your .env file:")
|
|
print()
|
|
print(f" MCP_API_KEYS={api_key}")
|
|
print()
|
|
print(" (If you have multiple keys, separate them with commas)")
|
|
print()
|
|
print("2. Restart the MCP server:")
|
|
print()
|
|
print(" docker-compose restart aegis-mcp")
|
|
print()
|
|
print("3. Configure ChatGPT Business:")
|
|
print()
|
|
print(" - Go to ChatGPT Settings > MCP Servers")
|
|
print(" - Add custom header:")
|
|
print(f" Authorization: Bearer {api_key}")
|
|
print()
|
|
print("4. Test the connection:")
|
|
print()
|
|
print(" Ask ChatGPT: 'List my Gitea repositories'")
|
|
print()
|
|
print("-" * 70)
|
|
print()
|
|
print("⚠️ IMPORTANT:")
|
|
print()
|
|
print(" • Store this key securely - it won't be shown again")
|
|
print(" • This key should be rotated in 90 days")
|
|
print(" • Set a reminder to rotate before expiration")
|
|
print(" • Never commit this key to version control")
|
|
print()
|
|
print("=" * 70)
|
|
|
|
# Offer to save metadata to file
|
|
save = input("\nSave key metadata to keys/ directory? [y/N]: ").strip().lower()
|
|
if save == "y":
|
|
keys_dir = Path(__file__).parent.parent / "keys"
|
|
keys_dir.mkdir(exist_ok=True)
|
|
|
|
# Create metadata file
|
|
key_id = api_key[:12]
|
|
metadata_file = keys_dir / f"key-{key_id}-{created_at.strftime('%Y%m%d')}.txt"
|
|
|
|
with open(metadata_file, "w", encoding="utf-8") as f:
|
|
f.write(f"API Key Metadata\n")
|
|
f.write(f"================\n\n")
|
|
f.write(f"Key ID: {key_id}...\n")
|
|
f.write(f"Description: {description}\n")
|
|
f.write(f"Created: {created_at.isoformat()}\n")
|
|
f.write(f"Expires: {expires_at.isoformat()}\n")
|
|
f.write(f"\n")
|
|
f.write(f"NOTE: The actual API key is NOT stored in this file for security.\n")
|
|
f.write(f" Only metadata is saved for reference.\n")
|
|
|
|
print(f"\n✓ Metadata saved to: {metadata_file}")
|
|
print(f"\n (The actual key is NOT saved - only you have it)")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
print("\n\nOperation cancelled.")
|
|
sys.exit(1)
|
|
except Exception as e:
|
|
print(f"\n❌ Error: {e}", file=sys.stderr)
|
|
sys.exit(1)
|