added proper console logs to pipeline

Omegapoint · Sep 14, 2023 · 2c329cc · 2c329cc
1 parent 6fb4bdf
commit 2c329cc
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 14 deletions.
diff --git a/src/Pentest/PentestService.ts b/src/Pentest/PentestService.ts
@@ -1,17 +1,15 @@
 import * as core from '@actions/core';
-import * as github from '@actions/github';
+//import * as github from '@actions/github';
 
 export class PentestService {
   public static async getStateOfPentest(pentestDate: { date: string; boardsTag: string }): Promise<void> {
-    console.log('Running Pentest Controls');
-
     if (!pentestDate.date || pentestDate.date === 'date-of-pentest') {
       core.warning('Pentest Date is not set!');
       return;
     }
     core.exportVariable('pentestDate', pentestDate.date);
 
-    console.log('Pentest Date is set', pentestDate.date);
+    //console.log('Pentest Date is set', pentestDate.date);
 
     // const token = core.getInput('repo-token');
     // console.log('Got the token');

diff --git a/src/branchprotection/BranchProtectionService.ts b/src/branchprotection/BranchProtectionService.ts
@@ -2,31 +2,32 @@ import * as core from '@actions/core';
 import * as github from '@actions/github';
 export class BranchProtectionService {
   public static async getStateOfBranchProtection(): Promise<void> {
-    console.log('Custom task is working!');
+    console.log('\n Running branch protection control');
     const token = core.getInput('repo-token');
-    console.log('Got the token');
+    //console.log('Got the token');
 
     const octokit = github.getOctokit(token);
-    console.log('octoKit authenticated');
+    //console.log('octoKit authenticated');
 
     const { owner, repo } = github.context.repo;
-    console.log(`Owner: ${owner}`);
-    console.log(`Repo: ${repo}`);
+    //console.log(`Owner: ${owner}`);
+    //console.log(`Repo: ${repo}`);
 
-    console.log('Going to get branch protection');
+    //console.log('Going to get branch protection');
     await octokit.rest.repos
       .getBranchProtection({
         owner: owner,
         repo: repo,
         branch: 'main',
       })
       .then((response) => {
-        console.log('Got the branch protection');
+        //console.log('Got the branch protection');
         console.log(response.data);
       })
       .catch((error) => {
-        core.warning('Error getting branch protection!');
-        console.log('Error: ', error.message);
+        console.log('Branch protections is not enabled for repository:' + repo)
+        //core.warning('Error getting branch protection!');
+        //console.log('Error: ', error.message);
       });
   }
 }
diff --git a/src/index.ts b/src/index.ts
@@ -3,16 +3,19 @@ import * as github from '@actions/github';
 import { BranchProtectionService } from './branchprotection/BranchProtectionService';
 import { CyDigConfig } from './types/CyDigConfig';
 import { getContentOfFile } from './helpfunctions/JsonService';
-import { PentestService } from './Pentest/PentestService';
+import { PentestService } from './pentest/PentestService';
 import { ThreatModelingService } from './threatmodeling/ThreatModelingService';
 /**
  * The main function for the action.
  * @returns {Promise<void>} Resolves when the action is complete.
  */
 export async function run(): Promise<void> {
   try {
+    console.log('\n Running controls on your repository')
     const cydigConfig: CyDigConfig = getContentOfFile(core.getInput('cydigConfigPath'));
     await BranchProtectionService.getStateOfBranchProtection();
+
+    console.log('\n Running controls on your boards');
     await PentestService.getStateOfPentest(cydigConfig.pentest);
     await ThreatModelingService.getStateOfThreatModeling(cydigConfig.threatModeling);
   } catch (error) {