feat: add util getSeverityFromPriorityGroup

Author: UlisesGascon

__tests__/utils.test.js

@@ -1,4 +1,4 @@
-const { validateGithubUrl, ensureGithubToken } = require('../src/utils/index')
+const { validateGithubUrl, ensureGithubToken, getSeverityFromPriorityGroup } = require('../src/utils/index')
 
 describe('ensureGithubToken', () => {
   let originalGithubToken
@@ -43,3 +43,21 @@ describe('validateGithubUrl', () => {
     expect(validateGithubUrl(url)).toBe(false)
   })
 })
+
+describe('getSeverityFromPriorityGroup', () => {
+  it('should return the correct severity based on the priority group', () => {
+    expect(getSeverityFromPriorityGroup('P0')).toBe('critical')
+    expect(getSeverityFromPriorityGroup('P1')).toBe('critical')
+    expect(getSeverityFromPriorityGroup('P2')).toBe('critical')
+    expect(getSeverityFromPriorityGroup('P3')).toBe('high')
+    expect(getSeverityFromPriorityGroup('P4')).toBe('high')
+    expect(getSeverityFromPriorityGroup('P5')).toBe('medium')
+    expect(getSeverityFromPriorityGroup('P6')).toBe('medium')
+    expect(getSeverityFromPriorityGroup('P7')).toBe('medium')
+    expect(getSeverityFromPriorityGroup('P8')).toBe('low')
+    expect(getSeverityFromPriorityGroup('P20')).toBe('low')
+    // Recommendations always have 'info' severity
+    expect(getSeverityFromPriorityGroup('R1')).toBe('info')
+    expect(getSeverityFromPriorityGroup('R11')).toBe('info')
+  })
+})

src/utils/index.js

@@ -22,8 +22,28 @@ const logger = {
   log: defineLog('log')
 }
 
+const getSeverityFromPriorityGroup = (priorityGroup) => {
+  const priorityType = priorityGroup[0]
+  const priorityValue = parseInt(priorityGroup.slice(1), 10)
+  // Recommendations (Rxx)
+  if (priorityType === 'R') {
+    return 'info'
+  }
+  // Priorities (Pxx)
+  if (priorityValue >= 0 && priorityValue <= 2) {
+    return 'critical'
+  } else if (priorityValue >= 3 && priorityValue <= 4) {
+    return 'high'
+  } else if (priorityValue >= 5 && priorityValue <= 7) {
+    return 'medium'
+  } else if (priorityValue >= 8) {
+    return 'low'
+  }
+}
+
 module.exports = {
   validateGithubUrl,
   ensureGithubToken,
+  getSeverityFromPriorityGroup,
   logger
 }