77a0bfb8c8
- tasks.py: 빈 sr_ids 목록도 HTTP 400 반환 (이전: 200) - messenger.py: /sr 봇명령 _cmd_create_sr에서 db 파라미터 제거 (BackgroundTask 호환) - main.py: Windows cp949 인코딩 오류 제거 (em dash → ASCII 하이픈) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
295 lines
12 KiB
Python
295 lines
12 KiB
Python
from contextlib import asynccontextmanager
|
||
|
||
from fastapi import FastAPI
|
||
from fastapi.responses import FileResponse
|
||
from fastapi.staticfiles import StaticFiles
|
||
|
||
from database import init_db
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
|
||
from routers import (
|
||
approvals, assign, audit, auth, cmdb, dashboard, kb, nlcmd, rating, tasks, work,
|
||
institutions, shell_scripts, timetable, attachments, notifications,
|
||
messenger, ssh, projects, vibe,
|
||
ssl_manager, pm, incidents, oncall, batch,
|
||
si_projects, si_wbs, si_requirements, si_issues,
|
||
si_risks, si_milestones, si_change_requests, si_tests,
|
||
agents,
|
||
analytics,
|
||
ws as ws_router,
|
||
timeline,
|
||
code_review,
|
||
anomaly,
|
||
chatbot,
|
||
kb_agent,
|
||
orchestrator,
|
||
predictive,
|
||
change,
|
||
problem,
|
||
capacity,
|
||
catalog,
|
||
ldap,
|
||
pam,
|
||
vuln_scan,
|
||
report,
|
||
metrics,
|
||
finops,
|
||
tenant_mgmt,
|
||
gateway,
|
||
license as license_router,
|
||
learning,
|
||
push as push_router,
|
||
)
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
# 디렉토리 생성
|
||
from pathlib import Path
|
||
(Path(__file__).parent / "uploads" / "sr_files").mkdir(parents=True, exist_ok=True)
|
||
(Path(__file__).parent / "uploads" / "workspaces").mkdir(parents=True, exist_ok=True)
|
||
|
||
await init_db()
|
||
from database import SessionLocal
|
||
from core.seed import seed_all
|
||
async with SessionLocal() as db:
|
||
await seed_all(db)
|
||
|
||
# 라이선스 상태 확인 (시작 시) — TRIAL 키는 GUARDIA_LICENSE_KEY 없이도 동작
|
||
from routers.license import get_license_status
|
||
async with SessionLocal() as db:
|
||
lic_status = await get_license_status(db)
|
||
if lic_status.get("valid"):
|
||
edition = lic_status["edition"]
|
||
days = lic_status["days_remaining"]
|
||
cust = lic_status["customer"]
|
||
if lic_status.get("is_trial"):
|
||
print(f"[LICENSE] TRIAL 체험판 활성 (D-{days}) - {cust}")
|
||
else:
|
||
print(f"[LICENSE] {edition} 라이선스 활성 ({days}일 남음) - {cust}")
|
||
if lic_status.get("expiry_warning"):
|
||
print(f"[LICENSE] 만료 {days}일 남음 - 갱신을 준비하세요.")
|
||
elif lic_status.get("expired"):
|
||
print("[LICENSE] 라이선스가 만료되었습니다. 갱신이 필요합니다.")
|
||
else:
|
||
print("[LICENSE] 라이선스 미등록 - /license 에서 무료 체험을 시작하거나 키를 등록하세요.")
|
||
|
||
# A-1: WebSocket ↔ SSE 통합 패치
|
||
from routers.ws import _integrate_with_sse_bus
|
||
_integrate_with_sse_bus()
|
||
|
||
# 백그라운드 스케줄러 시작
|
||
from core.scheduler import start_scheduler, init_batch_jobs_from_db
|
||
start_scheduler()
|
||
await init_batch_jobs_from_db() # DB에서 활성 배치 잡 자동 등록
|
||
|
||
yield
|
||
|
||
# 스케줄러 종료
|
||
from core.scheduler import stop_scheduler
|
||
stop_scheduler()
|
||
|
||
# F-2: Redis 연결 종료
|
||
try:
|
||
from core.cache import close_redis
|
||
await close_redis()
|
||
except Exception:
|
||
pass
|
||
|
||
|
||
app = FastAPI(title="GUARDiA ITSM", version="1.0.0", lifespan=lifespan)
|
||
|
||
# ── F-2: Redis 캐시 종료 훅 ──────────────────────────────────────────────────
|
||
# (lifespan의 yield 이후에 실행 — close_redis는 shutdown시 호출)
|
||
|
||
# ── F-3: Rate Limiting 미들웨어 등록 ─────────────────────────────────────────
|
||
from core.ratelimit import setup_rate_limiting
|
||
setup_rate_limiting(app)
|
||
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=["http://localhost:8001", "http://127.0.0.1:8001"],
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|
||
|
||
app.include_router(auth.router)
|
||
app.include_router(dashboard.router)
|
||
app.include_router(assign.router)
|
||
app.include_router(kb.router)
|
||
app.include_router(nlcmd.router)
|
||
app.include_router(tasks.router)
|
||
app.include_router(approvals.router)
|
||
app.include_router(cmdb.router)
|
||
app.include_router(audit.router)
|
||
app.include_router(work.router)
|
||
app.include_router(rating.router)
|
||
app.include_router(institutions.router)
|
||
app.include_router(shell_scripts.router)
|
||
app.include_router(timetable.router)
|
||
app.include_router(attachments.router)
|
||
app.include_router(notifications.router)
|
||
app.include_router(messenger.router)
|
||
app.include_router(ssh.router)
|
||
app.include_router(projects.router)
|
||
app.include_router(vibe.router)
|
||
app.include_router(ssl_manager.router)
|
||
app.include_router(pm.router)
|
||
app.include_router(incidents.router)
|
||
app.include_router(oncall.router)
|
||
app.include_router(batch.router)
|
||
|
||
# ── SI 프로젝트 관리 (분석→설계→구현→인도) ─────────────────────────────────
|
||
app.include_router(si_projects.router)
|
||
app.include_router(si_wbs.router)
|
||
app.include_router(si_requirements.router)
|
||
app.include_router(si_issues.router)
|
||
app.include_router(si_risks.router)
|
||
app.include_router(si_milestones.router)
|
||
app.include_router(si_change_requests.router)
|
||
app.include_router(si_tests.router)
|
||
|
||
# ── AI 에이전트 (Paperclip × GUARDiA, Ollama 로컬 LLM) ──────────────────────
|
||
app.include_router(agents.router)
|
||
|
||
# ── Analytics (E-2 배포 성공률 트렌드, E-3 엔지니어 워크로드) ─────────────────
|
||
app.include_router(analytics.router)
|
||
|
||
# ── A-1: WebSocket 실시간 대시보드 ───────────────────────────────────────────
|
||
app.include_router(ws_router.router)
|
||
|
||
# ── A-4: 운영 이벤트 타임라인 ────────────────────────────────────────────────
|
||
app.include_router(timeline.router)
|
||
|
||
# ── B-3: 코드 리뷰 에이전트 ──────────────────────────────────────────────────
|
||
app.include_router(code_review.router)
|
||
|
||
# ── B-1: AI 이상 탐지 ────────────────────────────────────────────────────────
|
||
app.include_router(anomaly.router)
|
||
|
||
# ── B-2: 자연어 SR 접수 챗봇 ─────────────────────────────────────────────────
|
||
app.include_router(chatbot.router)
|
||
|
||
# ── B-4: KB 자동 업데이트 에이전트 ───────────────────────────────────────────
|
||
app.include_router(kb_agent.router)
|
||
|
||
# ── B-5: 멀티 에이전트 협업 오케스트레이션 ────────────────────────────────────
|
||
app.include_router(orchestrator.router)
|
||
|
||
# ── B-6: 예측 유지보수 ────────────────────────────────────────────────────────
|
||
app.include_router(predictive.router)
|
||
|
||
# ── C-2: 변경 관리 CAB ───────────────────────────────────────────────────────
|
||
app.include_router(change.router)
|
||
|
||
# ── C-3: Problem Management ─────────────────────────────────────────────────
|
||
app.include_router(problem.router)
|
||
|
||
# ── C-4: 용량 관리 대시보드 ──────────────────────────────────────────────────
|
||
app.include_router(capacity.router)
|
||
|
||
# ── C-5: 서비스 카탈로그 ──────────────────────────────────────────────────────
|
||
app.include_router(catalog.router)
|
||
|
||
# ── D-1: LDAP/AD 연동 ────────────────────────────────────────────────────────
|
||
app.include_router(ldap.router)
|
||
|
||
# ── D-3: 특권 접근 관리 (PAM) ─────────────────────────────────────────────────
|
||
app.include_router(pam.router)
|
||
|
||
# ── D-4: 보안 취약점 자동 스캔 ───────────────────────────────────────────────
|
||
app.include_router(vuln_scan.router)
|
||
|
||
# ── E-1: 월별 리포트 자동 생성 ───────────────────────────────────────────────
|
||
app.include_router(report.router)
|
||
|
||
# ── E-4: Grafana 연동 (Prometheus 메트릭) ─────────────────────────────────
|
||
app.include_router(metrics.router)
|
||
|
||
# ── E-5: FinOps 비용 분석 ────────────────────────────────────────────────
|
||
app.include_router(finops.router)
|
||
|
||
# ── F-1: 멀티테넌트 데이터 격리 ──────────────────────────────────────────
|
||
from middleware.tenant import TenantMiddleware
|
||
app.add_middleware(TenantMiddleware)
|
||
app.include_router(tenant_mgmt.router)
|
||
|
||
# ── F-5: OpenAPI 외부 연동 게이트웨이 ────────────────────────────────────
|
||
app.include_router(gateway.router)
|
||
|
||
# ── Self-Improving Learning Loop ──────────────────────────────────────────
|
||
app.include_router(learning.router)
|
||
|
||
# ── 라이선스 관리 ──────────────────────────────────────────────────────────
|
||
app.include_router(license_router.router)
|
||
|
||
# ── G-10: PWA Push 알림 ──────────────────────────────────────────────────
|
||
app.include_router(push_router.router)
|
||
|
||
app.mount("/static", StaticFiles(directory="static"), name="static")
|
||
|
||
|
||
@app.get("/")
|
||
async def index():
|
||
return FileResponse("static/index.html")
|
||
|
||
|
||
@app.get("/customer")
|
||
async def customer():
|
||
return FileResponse("static/customer.html")
|
||
|
||
|
||
@app.get("/login")
|
||
async def login_page():
|
||
return FileResponse("static/login.html")
|
||
|
||
|
||
@app.get("/change-password")
|
||
async def change_password_page():
|
||
return FileResponse("static/change-password.html")
|
||
|
||
|
||
@app.get("/agents")
|
||
async def agents_page():
|
||
return FileResponse("static/agents.html")
|
||
|
||
|
||
@app.get("/incidents")
|
||
async def incidents_page():
|
||
return FileResponse("static/incidents.html")
|
||
|
||
|
||
@app.get("/ssl")
|
||
async def ssl_page():
|
||
return FileResponse("static/ssl.html")
|
||
|
||
|
||
@app.get("/pm")
|
||
async def pm_page():
|
||
return FileResponse("static/pm.html")
|
||
|
||
|
||
@app.get("/oncall")
|
||
async def oncall_page():
|
||
return FileResponse("static/oncall.html")
|
||
|
||
|
||
@app.get("/batch")
|
||
async def batch_page():
|
||
return FileResponse("static/batch.html")
|
||
|
||
|
||
@app.get("/vibe")
|
||
async def vibe_page():
|
||
return FileResponse("static/vibe.html")
|
||
|
||
|
||
@app.get("/si")
|
||
async def si_page():
|
||
return FileResponse("static/si.html")
|
||
|
||
|
||
@app.get("/license")
|
||
async def license_page():
|
||
return FileResponse("static/license.html")
|