@ampsec/platform-client 58.1.0 → 58.3.0

package/src/dto/saasUsers.dto.ts

@@ -1,16 +1,21 @@
-import {ChangeAwareDto, ChangeAwareUpsertDto} from './base.dto';
+import {z} from 'zod';
+import {_ChangeAwareDto, _ChangeAwareUpsertDto} from './base.dto';
 
-export type SaasUserUpsertDto = ChangeAwareUpsertDto & {
+export const _SaasUserUpsertDto = _ChangeAwareUpsertDto.extend({
   /** User Id */
-  uid: string;
+  uid: z.string(),
   /** Connector Id */
-  cid: string;
+  cid: z.string(),
   /** External email */
-  email: string;
+  email: z.string(),
   /** External Id */
-  extId: string;
+  extId: z.string(),
   /** Additional information specific to this Saas User */
-  meta: unknown;
-};
+  meta: z.unknown(),
+});
 
-export type SaasUserDto = ChangeAwareDto & SaasUserUpsertDto;
+export type SaasUserUpsertDto = z.infer<typeof _SaasUserUpsertDto>;
+
+export const _SaasUserDto = _ChangeAwareDto.and(_SaasUserUpsertDto);
+
+export type SaasUserDto = z.infer<typeof _SaasUserDto>;

package/src/dto/users.dto.ts

@@ -1,19 +1,19 @@
+import {z} from 'zod';
 import {ChangeAwareDto, ChangeAwareUpsertDto} from './base.dto';
 import {GlobalUserType} from './enums/globalUser.type';
 import {RiskContributorUpsertDto} from './riskContributors.dto';
 
-export type SimpleUserDto = {
-  /** Id of the user */
-  id?: string;
-  /** First name of the user */
-  firstName?: string;
-  /** Last name of the user */
-  lastName?: string;
-  /** Email of the user */
-  email?: string;
-  /** Department of the user */
-  department?: string;
-};
+export const _SimpleUserDto = z.object({
+  id: z.string().optional(),
+  score: z.number().optional(),
+  firstName: z.string().optional(),
+  lastName: z.string().optional(),
+  email: z.string().optional(),
+  organization: z.string().optional(),
+  department: z.string().optional(),
+});
+
+export type SimpleUserDto = z.infer<typeof _SimpleUserDto>;
 
 export type UserUpsertDto = ChangeAwareUpsertDto & {
   /** Tenant ID */

package/src/services/entity.service.ts

@@ -31,6 +31,10 @@ export interface AmpGlobalEntityService<WriteT extends BaseUpsertDto, ReadT exte
   getLookupIds(_tid: string, _options?: EntityCallOptions): Promise<ExtKeyMap>;
 }
 
+export interface AmpPatchEntityService<WriteT extends BaseUpsertDto, ReadT extends BaseDto> extends AmpDataService<ReadT> {
+  patch(_model: WriteT | WriteT[], _options: EntityCallOptions): Promise<Page<ReadT>>;
+}
+
 export class AmpEntityServiceImpl<WriteT extends BaseUpsertDto, ReadT extends BaseDto> extends AmpDataServiceImpl<ReadT> implements AmpEntityService<WriteT, ReadT> {
   constructor(rest: RestClient, kind: string, targetApi: TargetApi = TARGET_API_AGENT) {
     super(rest, kind, targetApi);
@@ -113,7 +117,26 @@ const errorHandler = (error: unknown) => {
   throw error;
 };
 
-export class AmpSdkUserService extends AmpGlobalEntityServiceImpl<PlatformUserUpsertDto, PlatformUserDto> {
+export class AmpPatchUserServiceImpl<WriteT extends BaseUpsertDto, ReadT extends BaseDto>
+  extends AmpGlobalEntityServiceImpl<WriteT, ReadT>
+  implements AmpPatchEntityService<WriteT, ReadT>
+{
+  constructor(rest: RestClient, kind: string, targetApi: TargetApi = 'api') {
+    super(rest, kind, targetApi);
+  }
+
+  patch = (model: WriteT | WriteT[], options: EntityCallOptions): Promise<Page<ReadT>> => {
+    const url = _.isArray(model) ? `/${this.targetApi}/v1/${this.kind}` : `/${this.targetApi}/v1/${this.kind}/${model.id}`;
+    return this.getPage({
+      url,
+      method: 'PATCH',
+      params: options.params as QueryMap,
+      data: model,
+    });
+  };
+}
+
+export class AmpSdkUserService extends AmpPatchUserServiceImpl<PlatformUserUpsertDto, PlatformUserDto> {
   constructor(rest: RestClient, targetApi: TargetApi = TARGET_API_AGENT) {
     super(rest, KIND.USERS, targetApi);
   }

package/src/settings.ts

@@ -87,6 +87,10 @@ const SLACK_SETTINGS = {
 const TENABLE_SETTINGS = {
   TENABLE_ASSET_ARCHIVED_RULE: AmpSettingsMap.asKey<unknown>('tenable.asset.archived.rule', {MATCH_NOTHING: true}),
   TENABLE_ASSET_ARCHIVED_THRESHOLD: AmpSettingsMap.asKey<string>('tenable.asset.archived.threshold', '1 year'),
+  TENABLE_VULN_CRIT_DEADLINE_THRESHOLD: AmpSettingsMap.asKey<string>('tenable.vuln.crit.deadline', '1 day'),
+  TENABLE_VULN_HIGH_DEADLINE_THRESHOLD: AmpSettingsMap.asKey<string>('tenable.vuln.high.deadline', '7 day'),
+  TENABLE_VULN_MEDIUM_DEADLINE_THRESHOLD: AmpSettingsMap.asKey<string>('tenable.vuln.medium.deadline', '30 day'),
+  TENABLE_VULN_LOW_DEADLINE_THRESHOLD: AmpSettingsMap.asKey<string>('tenable.vuln.low.deadline', '90 day'),
 };
 
 /**