From 38c0d38f9206f2e078ed799d506caaefb4fb429a Mon Sep 17 00:00:00 2001
From: Bryan Frimin <bryan@getprobo.com>
Date: Thu, 18 Dec 2025 10:32:26 +0100
Subject: [PATCH] Update selector with new frameworks

Signed-off-by: Bryan Frimin <bryan@getprobo.com>
---
 packages/ui/src/Atoms/Frameworks/CCPA.tsx     |  5 +++-
 packages/ui/src/Atoms/Frameworks/DORA.tsx     |  5 +++-
 packages/ui/src/Atoms/Frameworks/GDPR.tsx     |  5 +++-
 packages/ui/src/Atoms/Frameworks/NIS2.tsx     |  5 +++-
 .../FrameworkSelector/FrameworkSelector.tsx   | 28 +++++++++++++++++++
 5 files changed, 44 insertions(+), 4 deletions(-)

diff --git a/packages/ui/src/Atoms/Frameworks/CCPA.tsx b/packages/ui/src/Atoms/Frameworks/CCPA.tsx
index 327c45261..ea20c7557 100644
--- a/packages/ui/src/Atoms/Frameworks/CCPA.tsx
+++ b/packages/ui/src/Atoms/Frameworks/CCPA.tsx
@@ -1,6 +1,9 @@
-export function CCPA() {
+export function CCPA(props: { className?: string }) {
+    const { className } = props;
+
     return (
         <svg
+            className={className}
             version="1.1"
             id="Layer_1"
             xmlns="http://www.w3.org/2000/svg"
diff --git a/packages/ui/src/Atoms/Frameworks/DORA.tsx b/packages/ui/src/Atoms/Frameworks/DORA.tsx
index 40e8b9686..9f12713fd 100644
--- a/packages/ui/src/Atoms/Frameworks/DORA.tsx
+++ b/packages/ui/src/Atoms/Frameworks/DORA.tsx
@@ -1,6 +1,9 @@
-export function DORA() {
+export function DORA(props: { className?: string }) {
+    const { className } = props;
+
     return (
         <svg
+            className={className}
             version="1.1"
             id="Layer_1"
             xmlns="http://www.w3.org/2000/svg"
diff --git a/packages/ui/src/Atoms/Frameworks/GDPR.tsx b/packages/ui/src/Atoms/Frameworks/GDPR.tsx
index 23817e028..dcc18ab1e 100644
--- a/packages/ui/src/Atoms/Frameworks/GDPR.tsx
+++ b/packages/ui/src/Atoms/Frameworks/GDPR.tsx
@@ -1,6 +1,9 @@
-export function GDPR() {
+export function GDPR(props: { className?: string }) {
+    const { className } = props;
+
     return (
         <svg
+            className={className}
             version="1.1"
             id="Layer_1"
             xmlns="http://www.w3.org/2000/svg"
diff --git a/packages/ui/src/Atoms/Frameworks/NIS2.tsx b/packages/ui/src/Atoms/Frameworks/NIS2.tsx
index 7c47aef9d..2a684d11c 100644
--- a/packages/ui/src/Atoms/Frameworks/NIS2.tsx
+++ b/packages/ui/src/Atoms/Frameworks/NIS2.tsx
@@ -1,6 +1,9 @@
-export function NIS2() {
+export function NIS2(props: { className?: string }) {
+    const { className } = props;
+
     return (
         <svg
+            className={className}
             version="1.1"
             id="Layer_1"
             xmlns="http://www.w3.org/2000/svg"
diff --git a/packages/ui/src/Molecules/FrameworkSelector/FrameworkSelector.tsx b/packages/ui/src/Molecules/FrameworkSelector/FrameworkSelector.tsx
index 3c260ef2a..9834f9b6a 100644
--- a/packages/ui/src/Molecules/FrameworkSelector/FrameworkSelector.tsx
+++ b/packages/ui/src/Molecules/FrameworkSelector/FrameworkSelector.tsx
@@ -5,6 +5,10 @@ import { useTranslate } from "@probo/i18n";
 import { ISO27001 } from "../../Atoms/Frameworks/ISO27001";
 import { SOC2 } from "../../Atoms/Frameworks/SOC2";
 import { HIPAA } from "../../Atoms/Frameworks/HIPAA";
+import { CCPA } from "../../Atoms/Frameworks/CCPA";
+import { GDPR } from "../../Atoms/Frameworks/GDPR";
+import { NIS2 } from "../../Atoms/Frameworks/NIS2";
+import { DORA } from "../../Atoms/Frameworks/DORA";
 
 const availableFrameworks = [
     {
@@ -25,6 +29,30 @@ const availableFrameworks = [
         logo: <HIPAA className="size-8" />,
         description: "Health Insurance Portability and Accountability Act",
     },
+    {
+        id: "CCPA",
+        name: "CCPA",
+        logo: <CCPA className="size-8" />,
+        description: "California Consumer Privacy Act",
+    },
+    {
+        id: "NIS2",
+        name: "NIS 2",
+        logo: <NIS2 className="size-8" />,
+        description: "Network and Information Systems Directive 2",
+    },
+    {
+        id: "GDPR",
+        name: "GDPR",
+        logo: <GDPR className="size-8" />,
+        description: "General Data Protection Regulation",
+    },
+    {
+        id: "DORA",
+        name: "DORA",
+        logo: <DORA className="size-8" />,
+        description: "Digital Operational Readiness Assessment",
+    },
 ];
 
 type Framework = (typeof availableFrameworks)[number];