feat(scan): add template to scan create

src/Commands/RetestScan.ts

@@ -10,12 +10,24 @@ export class RetestScan implements CommandModule {
 
   public builder(argv: Argv): Argv {
     return argv
+      .option('name', {
+        alias: 'n',
+        describe: 'Name of the scan.',
+        string: false,
+        demandOption: false
+      })
       .option('token', {
         alias: 't',
         describe: 'Bright API-key',
         requiresArg: true,
         demandOption: true
       })
+      .option('template', {
+        alias: 'tp',
+        requiresArg: false,
+        string: true,
+        describe: 'ID of the template'
+      })
       .positional('scan', {
         describe: 'ID of an existing scan which you want to re-run.',
         type: 'string',
@@ -35,8 +47,20 @@ export class RetestScan implements CommandModule {
 
   public async handler(args: Arguments): Promise<void> {
     try {
+      if (!args.name && args.template) {
+        throw Error('please make sure that name is specified.');
+      }
+
+      const body = args.name && {
+        templateId: args.template as string,
+        name: args.name as string
+      };
+
       const scanManager: Scans = container.resolve(Scans);
-      const scanId: string = await scanManager.retest(args.scan as string);
+      const scanId: string = await scanManager.retest(
+        args.scan as string,
+        body
+      );
 
       // eslint-disable-next-line no-console
       console.log(scanId);

src/Commands/RunScan.ts

@@ -94,6 +94,12 @@ export class RunScan implements CommandModule {
         string: true,
         describe: 'ID of the project'
       })
+      .option('template', {
+        alias: 'tp',
+        requiresArg: false,
+        string: true,
+        describe: 'ID of the template'
+      })
       .option('module', {
         default: Module.DAST,
         requiresArg: true,
@@ -178,6 +184,7 @@ export class RunScan implements CommandModule {
         module: args.module,
         authObjectId: args.auth,
         projectId: args.project,
+        templateId: args.template,
         buckets: args.bucket,
         hostsFilter: args.hostFilter,
         headers: Helpers.parseHeaders(args.header as string[]),

src/Scan/RestScans.spec.ts

@@ -65,6 +65,7 @@ describe('RestScans', () => {
         const scanConfig: ScanConfig = {
           name: 'scan',
           module: Module.DAST,
+          buckets: [],
           tests: [],
           fileId: 'id'
         };
@@ -100,6 +101,7 @@ describe('RestScans', () => {
       const scanConfig: ScanConfig = {
         name: 'scan',
         module: Module.DAST,
+        buckets: [],
         tests: [],
         fileId: 'id'
       };

src/Scan/RestScans.ts

@@ -2,6 +2,7 @@ import {
   Discovery,
   Header,
   ScanConfig,
+  ScanRetestConfig,
   Scans,
   ScanState,
   SourceType,
@@ -52,8 +53,12 @@ export class RestScans implements Scans {
     return id;
   }
 
-  public async retest(scanId: string): Promise<string> {
+  public async retest(
+    scanId: string,
+    body?: ScanRetestConfig
+  ): Promise<string> {
     const { id }: { id: string } = await this.client.post({
+      ...(body?.name && { body: { config: body } }),
       uri: `/api/v1/scans/${scanId}/retest`
     });
 

src/Scan/Scans.ts

@@ -135,6 +135,7 @@ export interface ScanConfig {
   module: Module;
   authObjectId?: string;
   projectId?: string;
+  templateId?: string;
   discoveryTypes?: Discovery[];
   tests: TestType[];
   buckets: string[];
@@ -150,6 +151,11 @@ export interface ScanConfig {
   repeaters?: string[];
 }
 
+export interface ScanRetestConfig {
+  name?: string;
+  templateId?: string;
+}
+
 export enum IssueCategory {
   MEDIUM = 'Medium',
   HIGH = 'High',
@@ -192,7 +198,7 @@ export interface StorageFile {
 export interface Scans {
   create(body: ScanConfig): Promise<string>;
 
-  retest(scanId: string): Promise<string>;
+  retest(scanId: string, body?: ScanRetestConfig): Promise<string>;
 
   stop(scanId: string): Promise<void>;