feat: Add proper type for ProductStatus_t

csaf-poc · Aug 31, 2023 · 827aa00 · 827aa00
1 parent eb3115d
commit 827aa00
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 4 deletions.
diff --git a/src/lib/singleview/productvulnerabilities/productvulnerabilities.ts b/src/lib/singleview/productvulnerabilities/productvulnerabilities.ts
@@ -10,7 +10,9 @@ import {
   ProductStatusSymbol,
   type Vulnerability,
   type Product,
-  type Relationship
+  type Relationship,
+  type ProductStatus_t,
+  type ProductStatus_t_Key
 } from "./productvulnerabilitiestypes";
 
 const generateProductVulnerabilities = (jsonDocument: any) => {
@@ -31,7 +33,7 @@ const generateCrossTableFrom = (products: Product[], vulnerabilities: Vulnerabil
   const getCVE = vulnerabilities.map((vulnerability: Vulnerability) => vulnerability.cve);
   header = header.concat(getCVE);
   result.push(header);
-  let productLines = products.map((product: any) => {
+  let productLines = products.map((product: Product) => {
     let line = [product.name];
     line = line.concat(generateLineWith(product, vulnerabilities));
     return line;
@@ -110,8 +112,8 @@ const isProduct = (branch: any) => {
   return branch.product && branch.product.product_id && branch.product.name;
 };
 
-const generateDictFrom = (productStatus: any, section: string) => {
-  return productStatus[section].reduce((o: any, n: string) => {
+const generateDictFrom = (productStatus: ProductStatus_t, section: ProductStatus_t_Key) => {
+  return productStatus[section]!.reduce((o: any, n: string) => {
     o[n] = n;
     return o;
   }, {});

diff --git a/src/lib/singleview/productvulnerabilities/productvulnerabilitiestypes.ts b/src/lib/singleview/productvulnerabilities/productvulnerabilitiestypes.ts
@@ -14,6 +14,19 @@ export const ProductStatus = {
   RECOMMENDED: "RECOMMENDED"
 } as const;
 
+export type ProductStatus_t = {
+  first_affected?: string[];
+  first_fixed?: string[];
+  fixed?: string[];
+  known_affected?: string[];
+  known_not_affected?: string[];
+  last_affected?: string[];
+  recommended?: string[];
+  under_investigation?: string[];
+};
+
+export type ProductStatus_t_Key = keyof ProductStatus_t;
+
 export const ProductStatusSymbol = {
   FIXED: "F",
   UNDER_INVESTIGATION: "U",