All files / modules/75-cf/utils SegmentUtils.ts

76.92% Statements 10/13
100% Branches 0/0
50% Functions 2/4
83.33% Lines 10/12

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              1x                 1x                       1x 4x 2x                   2x             1x     1x     1x     1x  
/*
 * Copyright 2021 Harness Inc. All rights reserved.
 * Use of this source code is governed by the PolyForm Shield 1.0.0 license
 * that can be found in the licenses directory at the root of this repository, also available at
 * https://polyformproject.org/wp-content/uploads/2020/06/PolyForm-Shield-1.0.0.txt.
 */
 
import { PatchSegmentQueryParams, usePatchSegment } from 'services/cf'
 
export interface TargetSegmentParams {
  accountIdentifier: string
  orgIdentifier: string
  environmentIdentifier: string
  projectIdentifier: string
}
 
const makeInstructions = (kind: string, targetIdentifiers: string[]) => ({
  instructions: [
    {
      kind,
      parameters: {
        targets: targetIdentifiers
      }
    }
  ]
})
 
const makeTargetsToSegmentHook =
  (patchKind: string) =>
  ({ accountIdentifier, orgIdentifier, environmentIdentifier, projectIdentifier }: TargetSegmentParams) => {
    const { mutate } = usePatchSegment({
      identifier: '',
      queryParams: {
        accountIdentifier,
        orgIdentifier,
        projectIdentifier,
        environmentIdentifier
      } as PatchSegmentQueryParams
    })
 
    return (segmentIdentifier: string, targetIdentifiers: string[]) => {
      const body = makeInstructions(patchKind, targetIdentifiers)
      return mutate(body, { pathParams: { identifier: segmentIdentifier } })
    }
  }
 
/** Add targets to an include segment list */
export const useAddTargetsToIncludeList = makeTargetsToSegmentHook('addToIncludeList')
 
/** Remove targets to an include segment list */
export const useRemoveTargetsFromIncludeList = makeTargetsToSegmentHook('removeFromIncludeList')
 
/** Add targets to an exclude segment list */
export const useAddTargetsToExcludeList = makeTargetsToSegmentHook('addToExcludeList')
 
/** Remove targets to an exclude segment list */
export const useRemoveTargetsFromExcludeList = makeTargetsToSegmentHook('removeFromExcludeList')