Fix CWE leading zero mapping bug and add tests

Co-authored-by: felickz <1760475+felickz@users.noreply.github.com>

Author: Copilot

__tests__/main.test.ts

@@ -4,4 +4,36 @@ describe('main', () => {
     // TODO: Add proper unit tests for the main module
     expect(true).toBe(true)
   })
+
+  describe('CWE ID normalization', () => {
+    it('should handle CWE IDs with leading zeros', () => {
+      // Test that cwe-099 maps to 99
+      const cweIdWithLeadingZero = 'cwe-099'
+      const cweIdPrefix = 'cwe-'
+      const extractedId = cweIdWithLeadingZero.replace(cweIdPrefix, '')
+      const normalizedId = String(parseInt(extractedId, 10))
+      
+      expect(normalizedId).toBe('99')
+    })
+
+    it('should handle CWE IDs without leading zeros', () => {
+      // Test that cwe-89 maps to 89
+      const cweIdNoLeadingZero = 'cwe-89'
+      const cweIdPrefix = 'cwe-'
+      const extractedId = cweIdNoLeadingZero.replace(cweIdPrefix, '')
+      const normalizedId = String(parseInt(extractedId, 10))
+      
+      expect(normalizedId).toBe('89')
+    })
+
+    it('should handle CWE IDs with multiple leading zeros', () => {
+      // Test that cwe-020 maps to 20
+      const cweIdWithLeadingZeros = 'cwe-020'
+      const cweIdPrefix = 'cwe-'
+      const extractedId = cweIdWithLeadingZeros.replace(cweIdPrefix, '')
+      const normalizedId = String(parseInt(extractedId, 10))
+      
+      expect(normalizedId).toBe('20')
+    })
+  })
 })

dist/index.js

@@ -103,9 +103,11 @@ JSONPath({
     for (const tag of tags) {
       if (tag.startsWith(codeQlCweTagPrefix)) {
         const cweId = tag.replace(codeQlCweTagPrefix, '')
-        if (cweIds.includes(cweId)) {
+        // Normalize CWE ID by converting to integer to remove leading zeros
+        const normalizedCweId = String(parseInt(cweId, 10))
+        if (cweIds.includes(normalizedCweId)) {
           tags.push(securityStandardTag)
-          tags.push(...cweCategories[cweId])
+          tags.push(...cweCategories[normalizedCweId])
           return
         }
       }