security-c4po/security-c4po-angular/src/shared/stores/project-state/project-state.ts

import {Action, Selector, State, StateContext} from '@ngxs/store';
import {Injectable} from '@angular/core';
import {Project} from '@shared/models/project.model';
import {
  ChangeCategory,
  ChangePentest,
  ChangeProject,
  InitProjectState, UpdatePentestComments,
  UpdatePentestFindings
} from '@shared/stores/project-state/project-state.actions';
import {Category} from '@shared/models/category.model';
import {Pentest} from '@shared/models/pentest.model';

export const PROJECT_STATE_NAME = 'project';

export interface ProjectStateModel {
  selectedProject: Project;
  // Manages Categories
  disabledCategories: Array<string>;
  selectedCategory: Category;
  // Manages objectives of Category
  disabledPentests: Array<string>;
  selectedPentest: Pentest;
}

@State<ProjectStateModel>({
  name: PROJECT_STATE_NAME,
  defaults: {
    selectedProject: null,
    disabledCategories: [],
    selectedCategory: Category.INFORMATION_GATHERING,
    disabledPentests: [],
    selectedPentest: null
  }
})
@Injectable()
export class ProjectState {
  @Selector()
  static project(state: ProjectStateModel): Project {
    return state.selectedProject;
  }

  @Selector()
  static selectedCategory(state: ProjectStateModel): Category {
    return state.selectedCategory;
  }

  @Selector()
  static pentest(state: ProjectStateModel): Pentest {
    return state.selectedPentest;
  }

  @Action(InitProjectState)
  initProjectState(ctx: StateContext<ProjectStateModel>, action: InitProjectState): void {
    ctx.setState({
      selectedProject: action.project,
      disabledCategories: action.disabledCategories,
      selectedCategory: Category.INFORMATION_GATHERING,
      disabledPentests: action.disabledPentests,
      selectedPentest: null
    });
  }

  @Action(ChangeProject)
  changeProject(ctx: StateContext<ProjectStateModel>, {project}: ChangeProject): void {
    const state = ctx.getState();
    // ToDo: Add logic to change selectedCategory if disabled
    ctx.patchState({
      selectedProject: project
    });
  }

  @Action(ChangeCategory)
  changeCategory(ctx: StateContext<ProjectStateModel>, {category}: ChangeCategory): void {
    const state = ctx.getState();
    // ToDo: Add logic to change selectedCategory if disabled
    ctx.patchState({
      selectedCategory: category
    });
  }

  @Action(ChangePentest)
  changePentest(ctx: StateContext<ProjectStateModel>, {pentest}: ChangePentest): void {
    const state = ctx.getState();
    ctx.patchState({
      selectedPentest: {...pentest, projectId: state.selectedProject.id}
    });
  }

  @Action(UpdatePentestFindings)
  updatePentestFindings(ctx: StateContext<ProjectStateModel>, {findingId}: UpdatePentestFindings): void {
    const state = ctx.getState();
    let stateSelectedPentest: Pentest = state.selectedPentest;
    const stateFindingIds: Array<string> = stateSelectedPentest.findingIds || [];
    let updatedFindingIds: Array<string> = [];
    if (!stateFindingIds.includes(findingId)) {
      updatedFindingIds = [...stateFindingIds, findingId];
    } else {
      // ToDo: Add logic to remove findingId from array
    }
    // overwrites only findingIds
    stateSelectedPentest = {
      ...stateSelectedPentest,
      findingIds: updatedFindingIds
    };
    // path project state
    ctx.patchState({
      selectedPentest: stateSelectedPentest
    });
  }

  @Action(UpdatePentestComments)
  updatePentestComments(ctx: StateContext<ProjectStateModel>, {commentId}: UpdatePentestComments): void {
    const state = ctx.getState();
    let stateSelectedPentest: Pentest = state.selectedPentest;
    const stateCommentIds: Array<string> = stateSelectedPentest.commentIds || [];
    let updatedCommentIds: Array<string> = [];
    if (!stateCommentIds.includes(commentId)) {
      updatedCommentIds = [...stateCommentIds, commentId];
    } else {
      // ToDo: Add logic to remove commentId from array
    }
    // overwrites only findingIds
    stateSelectedPentest = {
      ...stateSelectedPentest,
      commentIds: updatedCommentIds
    };
    // path project state
    ctx.patchState({
      selectedPentest: stateSelectedPentest
    });
  }
}