@ampsec/platform-client 66.2.1 → 67.0.0

package/build/src/services/AmpSdk.d.ts

@@ -1,4 +1,4 @@
-import { PlatformActionExecutionDto, PlatformActionExecutionUpsertDto, PlatformAgentDto, PlatformAgentUpsertDto, PlatformCustomActionDto, PlatformCustomActionUpsertDto, PlatformCustomScoreCohortDto, PlatformCustomScoreCohortUpsertDto, PlatformCustomScoreValueDto, PlatformCustomScoreValueUpsertDto, PlatformFindingDto, PlatformFindingSpecDto, PlatformFindingSpecUpsertDto, PlatformFindingUpsertDto, PlatformFlowSpecDto, PlatformFlowSpecUpsertDto, PlatformFlowStateDto, PlatformFlowStateUpsertDto, PlatformJobExecutionStateDto, PlatformJobExecutionStateUpsertDto, PlatformJobSpecDto, PlatformJobSpecUpsertDto, PlatformNotificationDto, PlatformNotificationUpsertDto, PlatformProviderUpsertDto, PlatformReportResultDto, PlatformReportResultUpsertDto, PlatformRiskContributorDto, PlatformRiskContributorUpsertDto, PlatformSaasComponentDto, PlatformSaasComponentUpsertDto, PlatformStagedSaasAssetDto, PlatformStagedSaasAssetUpsertDto, PlatformStagedSaasUserDto, PlatformStagedSaasUserUpsertDto, PlatformTokenDto, PlatformTokenUpsertDto, ProviderDto, TenantDto, TenantUpsertDto } from '../dto';
+import { PlatformActionExecutionDto, PlatformActionExecutionUpsertDto, PlatformCustomActionDto, PlatformCustomActionUpsertDto, PlatformCustomScoreCohortDto, PlatformCustomScoreCohortUpsertDto, PlatformCustomScoreValueDto, PlatformCustomScoreValueUpsertDto, PlatformFindingDto, PlatformFindingSpecDto, PlatformFindingSpecUpsertDto, PlatformFindingUpsertDto, PlatformFlowSpecDto, PlatformFlowSpecUpsertDto, PlatformFlowStateDto, PlatformFlowStateUpsertDto, PlatformJobExecutionStateDto, PlatformJobExecutionStateUpsertDto, PlatformJobSpecDto, PlatformJobSpecUpsertDto, PlatformNotificationDto, PlatformNotificationUpsertDto, PlatformProviderUpsertDto, PlatformReportResultDto, PlatformReportResultUpsertDto, PlatformRiskContributorDto, PlatformRiskContributorUpsertDto, PlatformSaasComponentDto, PlatformSaasComponentUpsertDto, PlatformStagedSaasAssetDto, PlatformStagedSaasAssetUpsertDto, PlatformStagedSaasUserDto, PlatformStagedSaasUserUpsertDto, PlatformTokenDto, PlatformTokenUpsertDto, ProviderDto, TenantDto, TenantUpsertDto } from '../dto';
 import { AmpEntityService, AmpSdkAssetService, AmpSdkTenantService, AmpSdkUserService, TruncatableAmpEntityService } from './entity.service';
 import { AmpRestClientOptions, RestClient } from './rest';
 import { EnumService } from './rest/EnumService';
@@ -7,6 +7,7 @@ import { AmpSaaSEntityService, AmpSdkSaasAssetService, AmpSdkSaasComponentServic
 import { ContentfulService } from './contentful.service';
 import { FindingsInsightsService } from './findingsInsights.service';
 import { PlatformConnectorService } from './connector.platform.service';
+import { PlatformAgentService } from './AgentsService';
 export type AmpSdkOptions = AmpRestClientOptions;
 /**
  * AMP API
@@ -19,7 +20,7 @@ export type AmpSdkOptions = AmpRestClientOptions;
  */
 export declare class AmpSdkServices {
     readonly actionExecutions: AmpEntityService<PlatformActionExecutionUpsertDto, PlatformActionExecutionDto>;
-    readonly agents: AmpEntityService<PlatformAgentUpsertDto, PlatformAgentDto>;
+    readonly agents: PlatformAgentService;
     readonly asset: AmpSdkAssetService;
     readonly connectors: PlatformConnectorService;
     readonly contentful: ContentfulService;

package/build/src/services/AmpSdk.js

@@ -10,6 +10,7 @@ const saasEntity_service_1 = require("./saasEntity.service");
 const contentful_service_1 = require("./contentful.service");
 const findingsInsights_service_1 = require("./findingsInsights.service");
 const connector_platform_service_1 = require("./connector.platform.service");
+const AgentsService_1 = require("./AgentsService");
 /**
  * AMP API
  * This client is a wrapper around the AMP REST API meant to be used by
@@ -22,7 +23,7 @@ const connector_platform_service_1 = require("./connector.platform.service");
 class AmpSdkServices {
     constructor(rest) {
         this.actionExecutions = new entity_service_1.AmpEntityServiceImpl(rest, constants_1.KIND.ACTION_EXECUTION, constants_1.TARGET_API_PLATFORM);
-        this.agents = new entity_service_1.AmpEntityServiceImpl(rest, constants_1.KIND.AGENTS, constants_1.TARGET_API_PLATFORM);
+        this.agents = new AgentsService_1.PlatformAgentService(rest);
         this.asset = new entity_service_1.AmpSdkAssetService(rest, constants_1.TARGET_API_PLATFORM);
         this.connectors = new connector_platform_service_1.PlatformConnectorService(rest);
         this.contentful = new contentful_service_1.ContentfulService();

package/build/src/services/AmpSdk.js.map

@@ -1 +1 @@
-{"version":3,"file":"AmpSdk.js","sourceRoot":"","sources":["../../../src/services/AmpSdk.ts"],"names":[],"mappings":";;;AA4CA,qDAQ0B;AAC1B,iCAA0E;AAC1E,2CAAsD;AACtD,oDAAmE;AACnE,yDAAyD;AACzD,6DAA+J;AAC/J,6DAAuD;AACvD,yEAAmE;AACnE,6EAAsE;AAItE;;;;;;;;GAQG;AACH,MAAa,cAAc;IAgCzB,YAAY,IAAgB;QAC1B,IAAI,CAAC,gBAAgB,GAAG,IAAI,qCAAoB,CAA+D,IAAI,EAAE,gBAAI,CAAC,gBAAgB,EAAE,+BAAmB,CAAC,CAAC;QACjK,IAAI,CAAC,MAAM,GAAG,IAAI,qCAAoB,CAA2C,IAAI,EAAE,gBAAI,CAAC,MAAM,EAAE,+BAAmB,CAAC,CAAC;QACzH,IAAI,CAAC,KAAK,GAAG,IAAI,mCAAkB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAC/D,IAAI,CAAC,UAAU,GAAG,IAAI,qDAAwB,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,CAAC,UAAU,GAAG,IAAI,sCAAiB,EAAE,CAAC;QAC1C,IAAI,CAAC,aAAa,GAAG,IAAI,qCAAoB,CAAyD,IAAI,EAAE,gBAAI,CAAC,cAAc,EAAE,+BAAmB,CAAC,CAAC;QACtJ,IAAI,CAAC,kBAAkB,GAAG,IAAI,qCAAoB,CAAmE,IAAI,EAAE,gBAAI,CAAC,oBAAoB,EAAE,+BAAmB,CAAC,CAAC;QAC3K,IAAI,CAAC,iBAAiB,GAAG,IAAI,qCAAoB,CAAiE,IAAI,EAAE,gBAAI,CAAC,mBAAmB,EAAE,+BAAmB,CAAC,CAAC;QACvK,IAAI,CAAC,KAAK,GAAG,IAAI,gCAAkB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,GAAG,IAAI,6CAAwB,CAA+C,IAAI,EAAE,gBAAI,CAAC,QAAQ,EAAE,+BAAmB,CAAC,CAAC;QACrI,IAAI,CAAC,YAAY,GAAG,IAAI,6CAAwB,CAAuD,IAAI,EAAE,gBAAI,CAAC,aAAa,EAAE,+BAAmB,CAAC,CAAC;QACtJ,IAAI,CAAC,gBAAgB,GAAG,IAAI,kDAAuB,CAAC,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;QACvG,IAAI,CAAC,SAAS,GAAG,IAAI,6CAAwB,CAAiD,IAAI,EAAE,gBAAI,CAAC,UAAU,EAAE,+BAAmB,CAAC,CAAC;QAC1I,IAAI,CAAC,UAAU,GAAG,IAAI,6CAAwB,CAAmD,IAAI,EAAE,gBAAI,CAAC,WAAW,EAAE,+BAAmB,CAAC,CAAC;QAC9I,IAAI,CAAC,aAAa,GAAG,IAAI,qCAAoB,CAAmE,IAAI,EAAE,gBAAI,CAAC,cAAc,EAAE,+BAAmB,CAAC,CAAC;QAChK,IAAI,CAAC,IAAI,GAAG,IAAI,qCAAoB,CAA+C,IAAI,EAAE,gBAAI,CAAC,QAAQ,EAAE,+BAAmB,CAAC,CAAC;QAC7H,IAAI,CAAC,aAAa,GAAG,IAAI,qCAAoB,CAAyD,IAAI,EAAE,gBAAI,CAAC,aAAa,EAAE,+BAAmB,CAAC,CAAC;QACrJ,IAAI,CAAC,SAAS,GAAG,IAAI,qCAAoB,CAAyC,IAAI,EAAE,gBAAI,CAAC,SAAS,EAAE,+BAAmB,CAAC,CAAC;QAC7H,IAAI,CAAC,aAAa,GAAG,IAAI,6CAAwB,CAAyD,IAAI,EAAE,gBAAI,CAAC,cAAc,EAAE,+BAAmB,CAAC,CAAC;QAC1J,IAAI,CAAC,gBAAgB,GAAG,IAAI,6CAAwB,CAA+D,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;QACtK,IAAI,CAAC,UAAU,GAAG,IAAI,2CAAsB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QACxE,IAAI,CAAC,cAAc,GAAG,IAAI,+CAA0B,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,GAAG,IAAI,0CAAqB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QACtE,IAAI,CAAC,gBAAgB,GAAG,IAAI,gDAA+B,CAA+D,IAAI,EAAE,gBAAI,CAAC,kBAAkB,EAAE,+BAAmB,CAAC,CAAC;QAC9K,IAAI,CAAC,oBAAoB,GAAG,IAAI,gDAA+B,CAC7D,IAAI,EACJ,gBAAI,CAAC,sBAAsB,EAC3B,+BAAmB,CACpB,CAAC;QACF,IAAI,CAAC,eAAe,GAAG,IAAI,gDAA+B,CAA6D,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;QAE1K,IAAI,CAAC,QAAQ,GAAG,IAAI,wCAAqB,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,GAAG,IAAI,qCAAoB,CAA6B,IAAI,EAAE,gBAAI,CAAC,OAAO,EAAE,+BAAmB,CAAC,CAAC;QAC7G,IAAI,CAAC,MAAM,GAAG,IAAI,qCAAoB,CAA2C,IAAI,EAAE,gBAAI,CAAC,MAAM,EAAE,+BAAmB,CAAC,CAAC;QACzH,IAAI,CAAC,KAAK,GAAG,IAAI,kCAAiB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;IAChE,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,OAAsB;QACpC,MAAM,IAAI,GAAG,IAAA,uBAAgB,EAAC,OAAO,CAAC,CAAC;QACvC,OAAO,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;IAClC,CAAC;CACF;AA1ED,wCA0EC"}
+{"version":3,"file":"AmpSdk.js","sourceRoot":"","sources":["../../../src/services/AmpSdk.ts"],"names":[],"mappings":";;;AA0CA,qDAQ0B;AAC1B,iCAA0E;AAC1E,2CAAsD;AACtD,oDAAmE;AACnE,yDAAyD;AACzD,6DAA+J;AAC/J,6DAAuD;AACvD,yEAAmE;AACnE,6EAAsE;AACtE,mDAAqD;AAIrD;;;;;;;;GAQG;AACH,MAAa,cAAc;IAgCzB,YAAY,IAAgB;QAC1B,IAAI,CAAC,gBAAgB,GAAG,IAAI,qCAAoB,CAA+D,IAAI,EAAE,gBAAI,CAAC,gBAAgB,EAAE,+BAAmB,CAAC,CAAC;QACjK,IAAI,CAAC,MAAM,GAAG,IAAI,oCAAoB,CAAC,IAAI,CAAC,CAAC;QAC7C,IAAI,CAAC,KAAK,GAAG,IAAI,mCAAkB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAC/D,IAAI,CAAC,UAAU,GAAG,IAAI,qDAAwB,CAAC,IAAI,CAAC,CAAC;QACrD,IAAI,CAAC,UAAU,GAAG,IAAI,sCAAiB,EAAE,CAAC;QAC1C,IAAI,CAAC,aAAa,GAAG,IAAI,qCAAoB,CAAyD,IAAI,EAAE,gBAAI,CAAC,cAAc,EAAE,+BAAmB,CAAC,CAAC;QACtJ,IAAI,CAAC,kBAAkB,GAAG,IAAI,qCAAoB,CAAmE,IAAI,EAAE,gBAAI,CAAC,oBAAoB,EAAE,+BAAmB,CAAC,CAAC;QAC3K,IAAI,CAAC,iBAAiB,GAAG,IAAI,qCAAoB,CAAiE,IAAI,EAAE,gBAAI,CAAC,mBAAmB,EAAE,+BAAmB,CAAC,CAAC;QACvK,IAAI,CAAC,KAAK,GAAG,IAAI,gCAAkB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,GAAG,IAAI,6CAAwB,CAA+C,IAAI,EAAE,gBAAI,CAAC,QAAQ,EAAE,+BAAmB,CAAC,CAAC;QACrI,IAAI,CAAC,YAAY,GAAG,IAAI,6CAAwB,CAAuD,IAAI,EAAE,gBAAI,CAAC,aAAa,EAAE,+BAAmB,CAAC,CAAC;QACtJ,IAAI,CAAC,gBAAgB,GAAG,IAAI,kDAAuB,CAAC,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;QACvG,IAAI,CAAC,SAAS,GAAG,IAAI,6CAAwB,CAAiD,IAAI,EAAE,gBAAI,CAAC,UAAU,EAAE,+BAAmB,CAAC,CAAC;QAC1I,IAAI,CAAC,UAAU,GAAG,IAAI,6CAAwB,CAAmD,IAAI,EAAE,gBAAI,CAAC,WAAW,EAAE,+BAAmB,CAAC,CAAC;QAC9I,IAAI,CAAC,aAAa,GAAG,IAAI,qCAAoB,CAAmE,IAAI,EAAE,gBAAI,CAAC,cAAc,EAAE,+BAAmB,CAAC,CAAC;QAChK,IAAI,CAAC,IAAI,GAAG,IAAI,qCAAoB,CAA+C,IAAI,EAAE,gBAAI,CAAC,QAAQ,EAAE,+BAAmB,CAAC,CAAC;QAC7H,IAAI,CAAC,aAAa,GAAG,IAAI,qCAAoB,CAAyD,IAAI,EAAE,gBAAI,CAAC,aAAa,EAAE,+BAAmB,CAAC,CAAC;QACrJ,IAAI,CAAC,SAAS,GAAG,IAAI,qCAAoB,CAAyC,IAAI,EAAE,gBAAI,CAAC,SAAS,EAAE,+BAAmB,CAAC,CAAC;QAC7H,IAAI,CAAC,aAAa,GAAG,IAAI,6CAAwB,CAAyD,IAAI,EAAE,gBAAI,CAAC,cAAc,EAAE,+BAAmB,CAAC,CAAC;QAC1J,IAAI,CAAC,gBAAgB,GAAG,IAAI,6CAAwB,CAA+D,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;QACtK,IAAI,CAAC,UAAU,GAAG,IAAI,2CAAsB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QACxE,IAAI,CAAC,cAAc,GAAG,IAAI,+CAA0B,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,GAAG,IAAI,0CAAqB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QACtE,IAAI,CAAC,gBAAgB,GAAG,IAAI,gDAA+B,CAA+D,IAAI,EAAE,gBAAI,CAAC,kBAAkB,EAAE,+BAAmB,CAAC,CAAC;QAC9K,IAAI,CAAC,oBAAoB,GAAG,IAAI,gDAA+B,CAC7D,IAAI,EACJ,gBAAI,CAAC,sBAAsB,EAC3B,+BAAmB,CACpB,CAAC;QACF,IAAI,CAAC,eAAe,GAAG,IAAI,gDAA+B,CAA6D,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;QAE1K,IAAI,CAAC,QAAQ,GAAG,IAAI,wCAAqB,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,GAAG,IAAI,qCAAoB,CAA6B,IAAI,EAAE,gBAAI,CAAC,OAAO,EAAE,+BAAmB,CAAC,CAAC;QAC7G,IAAI,CAAC,MAAM,GAAG,IAAI,qCAAoB,CAA2C,IAAI,EAAE,gBAAI,CAAC,MAAM,EAAE,+BAAmB,CAAC,CAAC;QACzH,IAAI,CAAC,KAAK,GAAG,IAAI,kCAAiB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;IAChE,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,OAAsB;QACpC,MAAM,IAAI,GAAG,IAAA,uBAAgB,EAAC,OAAO,CAAC,CAAC;QACvC,OAAO,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;IAClC,CAAC;CACF;AA1ED,wCA0EC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ampsec/platform-client",
-  "version": "66.2.1",
+  "version": "67.0.0",
   "description": "",
   "main": "build/src/index.js",
   "runkitExampleFilename": "example/main.js",

package/src/dto/agents.dto.ts

@@ -1,39 +1,27 @@
 import {AgentStatus} from './enums/agent.status';
-import {BaseDto, BaseUpsertDto, _BaseDto, _BaseUpsertDto} from './base.dto';
-import {AuthRole} from './enums';
+import {UPSERT_DTO_MASK, _BaseDto, _BaseUpsertDto} from './base.dto';
 import {z} from 'zod';
-// import {AgentRole} from './enums';
+import {AgentAuthRoles} from './enums';
 
-export type AgentUpsertDto = BaseUpsertDto & {
-  /** Agent first name */
-  firstName: string | null;
-  /** Agent last name */
-  lastName: string | null;
-  /** Status of the Agent, e.g. ACTIVE */
-  status: AgentStatus;
-  /** External Id */
-  email: string;
-  /** Agent profile picture */
-  pictureUrl?: string;
-  /** Agent role */
-  // TODO: role: AgentRole; // for now all agents are `OWNER`. Need to map role to tenant
-};
+export const _AgentDto = _BaseDto.extend({
+  firstName: z.string().nullable(),
+  lastName: z.string().nullable(),
+  status: z.nativeEnum(AgentStatus),
+  role: z.nativeEnum(AgentAuthRoles),
+  email: z.string(),
+  pictureUrl: z.string().nullable(),
+  lastLoggedIn: z.string().nullable(),
+});
+export type AgentDto = z.infer<typeof _AgentDto>;
 
-export type AgentDto = BaseDto & AgentUpsertDto;
+export const _AgentUpsertDto = _AgentDto.partial(UPSERT_DTO_MASK);
+export type AgentUpsertDto = z.infer<typeof _AgentUpsertDto>;
 
-export const _AgentSummaryDto = z
-  .object({
-    firstName: z.string().nullable(),
-    lastName: z.string().nullable(),
-    email: z.string(),
-    pictureUrl: z.string().optional(),
-    status: z.nativeEnum(AgentStatus),
-    role: z.nativeEnum(AuthRole),
-    lastLoggedIn: z.date().nullable(),
-  })
-  .merge(_BaseDto);
-
-export type AgentSummaryDto = z.infer<typeof _AgentSummaryDto>;
-
-export const _AgentSummaryUpsertDto = _AgentSummaryDto.partial();
-export type AgentSummaryUpsertDto = z.infer<typeof _AgentSummaryUpsertDto>;
+export const _AgentInviteDto = z.object({
+  email: z.string(),
+  role: z.nativeEnum(AgentAuthRoles),
+  firstName: z.string().optional(),
+  lastName: z.string().optional(),
+  pictureUrl: z.string().optional(),
+});
+export type AgentInviteDto = z.infer<typeof _AgentInviteDto>;

package/src/dto/enums/auth.role.ts

@@ -46,8 +46,36 @@ const authRoleLookupMap: {[key: string]: AuthRole} = {
   SUPPORT: AuthRole.SUPPORT,
   OPERATOR: AuthRole.OPERATOR,
   SERVICE_ACCOUNT: AuthRole.SERVICE_ACCOUNT,
-};
+} as const;
 
 export function lookupAuthRole(role: string): AuthRole {
   return authRoleLookupMap[role] || AuthRole.UNKNOWN;
 }
+
+// Create the type of the subset
+export type UserAuthRoles = Extract<AuthRole, AuthRole.USER>;
+// And the const that goes with it
+export const UserAuthRoles = {
+  [AuthRole.USER]: AuthRole.USER,
+} as const;
+
+// Create the type of the subset
+export type AgentAuthRoles = Extract<AuthRole, AuthRole.ADMIN | AuthRole.AGENT | AuthRole.OWNER | AuthRole.REPORT_VIEWER | AuthRole.VIEWER>;
+// And the const that goes with it
+export const AgentAuthRoles = {
+  [AuthRole.ADMIN]: AuthRole.ADMIN,
+  [AuthRole.AGENT]: AuthRole.AGENT,
+  [AuthRole.OWNER]: AuthRole.OWNER,
+  [AuthRole.REPORT_VIEWER]: AuthRole.REPORT_VIEWER,
+  [AuthRole.VIEWER]: AuthRole.VIEWER,
+} as const;
+
+// Create the type of the subset
+export type PlatformAuthRoles = Extract<AuthRole, AuthRole.CUSTOMER_SUCCESS | AuthRole.OPERATOR | AuthRole.SERVICE_ACCOUNT | AuthRole.SUPPORT>;
+// And the const that goes with it
+export const PlatformAuthRoles = {
+  [AuthRole.CUSTOMER_SUCCESS]: AuthRole.CUSTOMER_SUCCESS,
+  [AuthRole.OPERATOR]: AuthRole.OPERATOR,
+  [AuthRole.SERVICE_ACCOUNT]: AuthRole.SERVICE_ACCOUNT,
+  [AuthRole.SUPPORT]: AuthRole.SUPPORT,
+} as const;

package/src/dto/findings.dto.ts

@@ -1,6 +1,6 @@
 import {z} from 'zod';
 import {SimpleAssetDto} from './assets.dto';
-import {BaseDto, BaseUpsertDto, _BaseDto} from './base.dto';
+import {BaseDto, BaseUpsertDto, UPSERT_DTO_MASK, _BaseDto} from './base.dto';
 import {Category, FindingKind, FindingSpecKind, SaasComponentKind} from './enums';
 import {FindingOutcome, FindingSeverity, FindingStatus} from './enums';
 import {SimpleProviderDto} from './providers.dto';
@@ -107,12 +107,7 @@ export const _FindingSpecDto = _BaseDto.extend({
 });
 export type FindingSpecDto = z.infer<typeof _FindingSpecDto>;
 
-export const _FindingSpecUpsertDto = _FindingSpecDto.partial({
-  id: true,
-  createdAt: true,
-  updatedAt: true,
-  deletedAt: true,
-});
+export const _FindingSpecUpsertDto = _FindingSpecDto.partial(UPSERT_DTO_MASK);
 export type FindingSpecUpsertDto = z.infer<typeof _FindingSpecUpsertDto>;
 
 export const _FindingSummaryDto = z.object({

package/src/dto/flows.dto.ts

@@ -131,8 +131,8 @@ export const _FlowInterval = z.object({
 export type FlowInterval = z.infer<typeof _FlowInterval>;
 
 const _FlowActionDetails = z.object({
-  actionId: z.string(),
-  findingStatus: z.nativeEnum(FindingStatus).optional(),
+  actionId: z.string().nullable(),
+  findingStatus: z.nativeEnum(FindingStatus).optional().nullable(),
 });
 
 export const _FlowActions = z.object({
@@ -157,6 +157,8 @@ export const _FlowActions = z.object({
    */
   resolveNowActionId: z.string().optional(),
   resolveNowAction: _FlowActionDetails.optional(),
+
+  scheduleResolutionAction: _FlowActionDetails.optional(),
 });
 export type FlowActions = z.infer<typeof _FlowActions>;
 export const _FlowRewardsConfig = z.object({
@@ -188,6 +190,9 @@ export const _FlowSpecDto = _BaseDto.merge(
     rewards: z.array(_FlowRewardsConfig).optional(),
     customMessage: z.object({template: z.string().optional(), append: z.boolean().optional()}).optional(),
     additionalContext: z.object({data: z.string().optional()}).optional(),
+    /**
+     * @deprecated Use scheduleResolutionAction instead
+     */
     scheduleConfig: z.object({required: z.boolean()}).optional(),
     activity: z
       .object({

package/src/dto/platform/platform.agents.dto.ts

@@ -1,6 +1,26 @@
-import {AgentDto, AgentUpsertDto} from '../agents.dto';
-import {MultiTenantBased, UpsertMultiTenantBased} from './multi.tenant.based.dto';
+import {z} from 'zod';
+import {AgentAuthRoles, AgentStatus} from '../enums';
+import {UPSERT_DTO_MASK, _BaseDto} from '../base.dto';
+import {_AgentUpsertDto} from '../agents.dto';
 
-export type PlatformAgentUpsertDto = AgentUpsertDto & UpsertMultiTenantBased;
+export const _PlatformAgentMembership = z.object({
+  tid: z.string(),
+  role: z.nativeEnum(AgentAuthRoles),
+  status: z.nativeEnum(AgentStatus),
+  organization: z.string().nullable(),
+  department: z.string().nullable(),
+  lastLoggedIn: z.string().nullable(),
+});
+export type PlatformAgentMembership = z.infer<typeof _PlatformAgentMembership>;
 
-export type PlatformAgentDto = AgentDto & MultiTenantBased;
+export const _PlatformAgentDto = _BaseDto.extend({
+  firstName: z.string().nullable(),
+  lastName: z.string().nullable(),
+  email: z.string(),
+  memberships: z.array(_PlatformAgentMembership),
+  pictureUrl: z.string().nullable(),
+});
+export type PlatformAgentDto = z.infer<typeof _PlatformAgentDto>;
+
+export const _PlatformAgentUpsertDto = _AgentUpsertDto.partial(UPSERT_DTO_MASK);
+export type PlatformAgentUpsertDto = z.infer<typeof _PlatformAgentUpsertDto>;

package/src/etag.ts

@@ -8,7 +8,7 @@ export function calculateEtag(model: unknown): string {
 }
 
 export function withEtag(model: ChangeAwareDto | ChangeAwareUpsertDto) {
-  model.etag = calculateEtag(model);
+  model.etag = calculateEtag({...model, etag: ''});
 }
 
 export function createWithEtag<T extends {etag: string}>(model: T): T {

package/src/services/AgentsService.ts

@@ -1,25 +1,58 @@
-import {z} from 'zod';
-import {AgentDto, AgentUpsertDto, Page, _AgentSummaryUpsertDto} from '../dto';
-import {KIND} from './constants';
+import {AgentAuthRoles, AgentDto, AgentInviteDto, AgentStatus, AgentUpsertDto, Page, PlatformAgentDto, _AgentUpsertDto} from '../dto';
+import {KIND, TARGET_API_PLATFORM} from './constants';
+import {AmpDataServiceImpl} from './data.service';
 import {AmpEntityServiceImpl} from './entity.service';
 import {RestClient} from './rest';
 
-export const _AgentSummaryPayload = _AgentSummaryUpsertDto.required({email: true, role: true});
-export type AgentSummaryPayload = z.infer<typeof _AgentSummaryPayload>;
-
 export class AgentsService extends AmpEntityServiceImpl<AgentUpsertDto, AgentDto> {
   constructor(rest: RestClient) {
     super(rest, KIND.AGENTS);
   }
-  invite = async (agentSavePayload: AgentSummaryPayload): Promise<Page<AgentDto>> => {
-    const payload = _AgentSummaryPayload.parse(agentSavePayload);
-
+  invite = async (agentInvitePayload: AgentInviteDto): Promise<Page<AgentDto>> => {
     const res = await this.rest.call({
       url: `/${this.targetApi}/v1/${this.kind}/invite`,
       method: 'POST',
-      data: payload,
+      data: agentInvitePayload,
     });
 
     return res.data as Page<AgentDto>;
   };
 }
+
+export class PlatformAgentService extends AmpDataServiceImpl<PlatformAgentDto> {
+  constructor(rest: RestClient) {
+    super(rest, KIND.AGENTS, TARGET_API_PLATFORM);
+  }
+  create = async (email: string) => {
+    return this.rest.call({
+      url: `/${this.targetApi}/v1/${this.kind}`,
+      method: 'POST',
+      data: {email},
+      headers: {
+        'Content-Type': 'application/json',
+      },
+    });
+  };
+
+  delete = async (id: string) => {
+    return this.rest.call({
+      url: `/${this.targetApi}/v1/${this.kind}/${id}`,
+      method: 'DELETE',
+    });
+  };
+
+  upsertMembership = async ({tid, agentId, role, status}: {tid: string; agentId: string; role: AgentAuthRoles; status: AgentStatus}) => {
+    return this.rest.call({
+      url: `/${this.targetApi}/v1/${this.kind}/memberships`,
+      method: 'PUT',
+      data: {tid, agentId, role, status},
+    });
+  };
+
+  deleteMembership = async ({tid, agentId}: {tid: string; agentId: string}) => {
+    return this.rest.call({
+      url: `/${this.targetApi}/v1/${this.kind}/memberships/agentId/${agentId}/tid/${tid}`,
+      method: 'DELETE',
+    });
+  };
+}

package/src/services/AmpSdk.ts

@@ -1,8 +1,6 @@
 import {
   PlatformActionExecutionDto,
   PlatformActionExecutionUpsertDto,
-  PlatformAgentDto,
-  PlatformAgentUpsertDto,
   PlatformCustomActionDto,
   PlatformCustomActionUpsertDto,
   PlatformCustomScoreCohortDto,
@@ -59,6 +57,7 @@ import {AmpSaaSEntityService, AmpSaaSEntityServiceImpl, AmpSdkSaasAssetService,
 import {ContentfulService} from './contentful.service';
 import {FindingsInsightsService} from './findingsInsights.service';
 import {PlatformConnectorService} from './connector.platform.service';
+import {PlatformAgentService} from './AgentsService';
 
 export type AmpSdkOptions = AmpRestClientOptions;
 
@@ -73,7 +72,7 @@ export type AmpSdkOptions = AmpRestClientOptions;
  */
 export class AmpSdkServices {
   readonly actionExecutions: AmpEntityService<PlatformActionExecutionUpsertDto, PlatformActionExecutionDto>;
-  readonly agents: AmpEntityService<PlatformAgentUpsertDto, PlatformAgentDto>;
+  readonly agents: PlatformAgentService;
   readonly asset: AmpSdkAssetService;
   readonly connectors: PlatformConnectorService;
   readonly contentful: ContentfulService;
@@ -105,7 +104,7 @@ export class AmpSdkServices {
 
   constructor(rest: RestClient) {
     this.actionExecutions = new AmpEntityServiceImpl<PlatformActionExecutionUpsertDto, PlatformActionExecutionDto>(rest, KIND.ACTION_EXECUTION, TARGET_API_PLATFORM);
-    this.agents = new AmpEntityServiceImpl<PlatformAgentUpsertDto, PlatformAgentDto>(rest, KIND.AGENTS, TARGET_API_PLATFORM);
+    this.agents = new PlatformAgentService(rest);
     this.asset = new AmpSdkAssetService(rest, TARGET_API_PLATFORM);
     this.connectors = new PlatformConnectorService(rest);
     this.contentful = new ContentfulService();