All files / src/handlers/rulegroups CreateRuleGroupHandler.ts

100% Statements 26/26
90% Branches 9/10
100% Functions 2/2
100% Lines 24/24

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 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 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124                                      2x 2x 2x 2x 2x 2x                                                                                 2x           1x 1x     1x 1x               5x 5x   5x               5x 5x 1x 1x 1x         1x                   4x   3x   1x 1x        
/* 
  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
  
  Licensed under the Apache License, Version 2.0 (the "License").
  You may not use this file except in compliance with the License.
  You may obtain a copy of the License at
  
      http://www.apache.org/licenses/LICENSE-2.0
  
  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.
*/
 
import { APIGatewayProxyEvent, Context } from 'aws-lambda';
import { Logger, LoggerFactory } from 'shared_types';
import { AsyncRequestHandler } from 'src/common/AsyncRequestHandler';
import { AppConfiguration } from 'src/common/configuration/AppConfiguration';
import RuleConfigError from 'src/common/RuleConfigError';
import { ServerlessResponse } from 'src/common/ServerlessResponse';
import { RuleGroupDataSourceService } from 'src/service/RuleGroupDataSourceService';
import { CreateRuleGroupInputValidator } from 'src/validators/CreateRuleGroupInputValidator';
import { inject, injectable } from 'tsyringe';
 
 
export type ConfigurationEvaluationTriggerEvent = {
    ruleGroupId: string,
}
/**
 * @api {post} /rulegroups Create new rule group
 * @apiGroup RuleGroup
 * @apiDescription Create new rule group referencing a cloud resource or fixed resource
 * @apiVersion 1.0.0
 *  @apiExample {curl} CURL Example:
 curl --location --request POST 'https://<rest_api_id>.execute-api.ap-southeast-2.amazonaws.com/prod/rulegroups/' --data-raw '{
    "id":"demo-group-demo",
  "description": "demo rule group",
  "ownerGroup": [
    "arn:aws:iam::<account-number>:role/ObjectExtensionSecOpsAdminRole"
  ],
  "ruleGroupArn": "arn:aws:network-firewall:ap-southeast-2:<account-number>:stateful-rulegroup/anfwconfig-demo-rulegroup"
}'
* @apiSuccess (Success 201) Rule Group created
* @apiParam {string} description Description of this rule group
* @apiParam {String[1..100]="[0-9a-zA-Z_-]+"}  id The object's id. id Id of this rule group
* @apiParam {list[]}  ownerGroup The owner group, this is SecOpsAdminRole provided by the solution
* @apiParam {string}  ruleGroupArn The underlying AWS network firewall rule group arn
* @apiParam {String[1..1000]="[ 0-9a-zA-Z_-]+"}  description of the rule group
* @apiError (Error 400) Unsupported Port Type
* @apiError (Error 400) Unsupported Object Type
* @apiError (Error 400) Invalid Object value
* @apiError (Error 502) Time out
* @apiError (Error 503) Internal error
* 
* @apiSuccessExample Success-Response: 
*      HTTP/1.1 201 OK
*  {
    "ruleGroupId": "demo-group-demo-1"
}
* 
* @apiSampleRequest off
*/
@injectable()
export class CreateRuleGroupHandler
    implements AsyncRequestHandler<APIGatewayProxyEvent, ServerlessResponse> {
    private readonly logger: Logger;
    private readonly defaultAggregatorName: string;
 
    constructor(@inject('LoggerFactory') loggerFactory: LoggerFactory,
        @inject('RuleGroupDataSourceService') private ruleGroupDataSourceService: RuleGroupDataSourceService,
        @inject('CreateRuleGroupInputValidator') private validator: CreateRuleGroupInputValidator,
        @inject('AppConfiguration') appConfiguration: AppConfiguration,
    ) {
        this.logger = loggerFactory.getLogger('CreateRuleConfigHandler');
        this.defaultAggregatorName = appConfiguration.defaultAggregatorName;
    }
 
    async handle(
        event: APIGatewayProxyEvent,
        // eslint-disable-next-line @typescript-eslint/no-unused-vars
        _context: Context
    ): Promise<ServerlessResponse> {
        this.logger.debug('Processing create rule group request');
        const input = await this.validator.parseAndValidate(event);
      
        const ruleGroupToBeCreated = {
            ...(input.id && { id: input.id }),
            aggregatorName: input.aggregatorName??this.defaultAggregatorName,
            description: input.description,
            ownerGroup: input.ownerGroup,
            ruleGroupArn: input.ruleGroupArn
        };
 
        try {
            if(input.id) {
                const existingRuleGroup = await this.ruleGroupDataSourceService.getRuleGroupBy(input.id);
                Eif(existingRuleGroup) {
                    this.logger.error(
                        `Error while creating rule group, ${input.id} already exists`,
                        existingRuleGroup
                    );
 
                    return ServerlessResponse.ofError(
                        new RuleConfigError(
                            `Error while creating rule group, ${existingRuleGroup.id} already exists`,
                            409,
                            true
                        )
                    );
                }
            }
 
            const ruleGroupId = await this.ruleGroupDataSourceService.createRuleGroup(ruleGroupToBeCreated);
 
            return ServerlessResponse.ofObject(201, { ruleGroupId: ruleGroupId });
        } catch (e) {
            this.logger.error('Error while updating rule group', e, input);
            return ServerlessResponse.ofObject(503, { message: `Error while creating rule group , ${e.message}` });
        }
    }
}