Skip to content

remediation

remediation

API router for remediation operations.

generate_remediation(request) async

Generate remediation and rollback configurations.

Source code in hier_config_api/routers/remediation.py 
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
@router.post("/generate", response_model=GenerateRemediationResponse)
async def generate_remediation(request: GenerateRemediationRequest) -> GenerateRemediationResponse:
    """Generate remediation and rollback configurations."""
    try:
        result = RemediationService.generate_remediation(
            platform=request.platform,
            running_config=request.running_config,
            intended_config=request.intended_config,
            tag_rules=request.tag_rules,
            include_tags=request.include_tags,
            exclude_tags=request.exclude_tags,
        )

        # Store remediation
        remediation_id = storage.store_remediation(result)
        result["remediation_id"] = remediation_id

        return GenerateRemediationResponse(
            remediation_id=remediation_id,
            platform=result["platform"],
            remediation_config=result["remediation_config"],
            rollback_config=result["rollback_config"],
            summary=result["summary"],
            tags=result["tags"],
        )
    except Exception as e:
        raise HTTPException(
            status_code=400, detail=f"Failed to generate remediation: {str(e)}"
        ) from e

apply_tags(remediation_id, request) async

Apply tags to an existing remediation.

Source code in hier_config_api/routers/remediation.py 
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
@router.post("/{remediation_id}/tags", response_model=ApplyTagsResponse)
async def apply_tags(remediation_id: str, request: ApplyTagsRequest) -> ApplyTagsResponse:
    """Apply tags to an existing remediation."""
    remediation_data = storage.get_remediation(remediation_id)
    if not remediation_data:
        raise HTTPException(status_code=404, detail="Remediation not found")

    try:
        remediation_config = remediation_data["remediation_config"]
        tagged_config, tags = RemediationService.apply_tags(remediation_config, request.tag_rules)

        # Update stored remediation
        storage.update_remediation(remediation_id, {"tags": tags})

        return ApplyTagsResponse(
            remediation_id=remediation_id, remediation_config=tagged_config, tags=tags
        )
    except Exception as e:
        raise HTTPException(status_code=400, detail=f"Failed to apply tags: {str(e)}") from e

filter_remediation(remediation_id, include_tags=Query(None), exclude_tags=Query(None)) async

Filter remediation by tags.

Source code in hier_config_api/routers/remediation.py 
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
@router.get("/{remediation_id}/filter", response_model=FilterRemediationResponse)
async def filter_remediation(
    remediation_id: str,
    include_tags: list[str] = Query(None),
    exclude_tags: list[str] = Query(None),
) -> FilterRemediationResponse:
    """Filter remediation by tags."""
    remediation_data = storage.get_remediation(remediation_id)
    if not remediation_data:
        raise HTTPException(status_code=404, detail="Remediation not found")

    try:
        remediation_config = remediation_data["remediation_config"]
        tags = remediation_data.get("tags", {})

        filtered_config, summary = RemediationService.filter_remediation(
            remediation_config, tags, include_tags, exclude_tags
        )

        return FilterRemediationResponse(
            remediation_id=remediation_id, filtered_config=filtered_config, summary=summary
        )
    except Exception as e:
        raise HTTPException(
            status_code=400, detail=f"Failed to filter remediation: {str(e)}"
        ) from e