53755a31d6
- Add C2Error exception to adapter ABC - Add promote_to_in_progress() helper to simulation_workflow (pending→in_progress) - Flesh out MythicAdapter: list_callbacks() (GraphQL query) + create_task() (mutation) - Expand FakeAdapter to 3 deterministic callbacks; switch task store to per-instance - Add GET /api/engagements/<id>/c2/callbacks — lists active callbacks via adapter - Add POST /api/simulations/<id>/c2/execute — issues tasks, stores C2Task rows, auto-transitions pending→in_progress, blocks on done (409) - Both endpoints: SOC=403, 503 no-key, 502 adapter error, sanitized error messages - Add requests-mock==1.12.1 to requirements.txt - 42 new tests (342 total, 300 M1 baseline preserved green) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
"""Flask application factory."""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from flask import Flask, jsonify, send_from_directory
|
|
|
|
from backend.app.api import (
|
|
auth_bp,
|
|
c2_bp,
|
|
engagements_bp,
|
|
sims_c2_bp,
|
|
simulations_bp,
|
|
templates_bp,
|
|
users_bp,
|
|
)
|
|
from backend.app.cli import register_cli
|
|
from backend.app.config import Config, TestConfig
|
|
from backend.app.errors import register_error_handlers
|
|
from backend.app.extensions import db, migrate
|
|
|
|
|
|
def create_app(config_object: object | None = None) -> Flask:
|
|
"""Application factory.
|
|
|
|
`config_object` is an *instance* (not a class) of Config or a subclass.
|
|
If None, picks TestConfig when MIMIC_TESTING=1, otherwise Config.
|
|
"""
|
|
static_folder = str(Path(__file__).parent / "static")
|
|
app = Flask(__name__, static_folder=static_folder, static_url_path="/static")
|
|
|
|
if config_object is None:
|
|
config_object = TestConfig() if os.environ.get("MIMIC_TESTING") == "1" else Config()
|
|
app.config.from_object(config_object)
|
|
|
|
db.init_app(app)
|
|
migrations_dir = str(Path(__file__).parent.parent / "migrations")
|
|
migrate.init_app(app, db, directory=migrations_dir)
|
|
|
|
# Ensure models are imported so Alembic/metadata see them.
|
|
from backend.app import models # noqa: F401
|
|
|
|
app.register_blueprint(auth_bp)
|
|
app.register_blueprint(users_bp)
|
|
app.register_blueprint(engagements_bp)
|
|
app.register_blueprint(simulations_bp)
|
|
app.register_blueprint(templates_bp)
|
|
app.register_blueprint(c2_bp)
|
|
app.register_blueprint(sims_c2_bp)
|
|
|
|
from backend.app.services import mitre as mitre_svc
|
|
mitre_svc.load_bundle()
|
|
|
|
register_error_handlers(app)
|
|
register_cli(app)
|
|
|
|
static_root = Path(static_folder)
|
|
|
|
@app.get("/api/health")
|
|
def health():
|
|
return {"status": "ok"}, 200
|
|
|
|
# Serve the built frontend (Vite output copied into app/static at image build time).
|
|
@app.get("/")
|
|
def index():
|
|
index_path = static_root / "index.html"
|
|
if index_path.exists():
|
|
return send_from_directory(static_folder, "index.html")
|
|
return {"status": "ok", "message": "Mimic API running. Frontend not built."}, 200
|
|
|
|
@app.get("/<path:path>")
|
|
def spa_fallback(path: str):
|
|
# Unknown /api/* paths must stay JSON 404 — never shadowed by index.html.
|
|
if path.startswith("api/"):
|
|
return jsonify({"error": "Not found"}), 404
|
|
# Serve static assets if present; otherwise hand back index.html for client routing.
|
|
candidate = static_root / path
|
|
if candidate.is_file():
|
|
return send_from_directory(static_folder, path)
|
|
index_path = static_root / "index.html"
|
|
if index_path.exists():
|
|
return send_from_directory(static_folder, "index.html")
|
|
return jsonify({"error": "Not found"}), 404
|
|
|
|
return app
|