From ec028e9d178fad17d9d0649f8cd18add434310f9 Mon Sep 17 00:00:00 2001
From: Bryan Frimin <bryan@getprobo.com>
Date: Wed, 23 Apr 2025 09:43:15 -0700
Subject: [PATCH] Fix UI not refresh after create vendor risk

Signed-off-by: Bryan Frimin <bryan@getprobo.com>
---
 .../organizations/vendors/VendorView.tsx      | 66 +------------------
 ...iewCreateRiskAssessmentMutation.graphql.ts |  6 +-
 2 files changed, 6 insertions(+), 66 deletions(-)

diff --git a/apps/console/src/pages/organizations/vendors/VendorView.tsx b/apps/console/src/pages/organizations/vendors/VendorView.tsx
index 1d25648ab..dfc9a97b9 100644
--- a/apps/console/src/pages/organizations/vendors/VendorView.tsx
+++ b/apps/console/src/pages/organizations/vendors/VendorView.tsx
@@ -176,7 +176,7 @@ const createRiskAssessmentMutation = graphql`
     $connections: [ID!]!
   ) {
     createVendorRiskAssessment(input: $input) {
-      vendorRiskAssessmentEdge @appendEdge(connections: $connections) {
+      vendorRiskAssessmentEdge @prependEdge(connections: $connections) {
         node {
           id
           assessedAt
@@ -195,40 +195,6 @@ const createRiskAssessmentMutation = graphql`
   }
 `;
 
-function EditableField({
-  label,
-  value,
-  onChange,
-  type = "text",
-  helpText,
-  disabled = false,
-}: {
-  label: string;
-  value: string;
-  onChange: (value: string) => void;
-  type?: string;
-  helpText?: string;
-  disabled?: boolean;
-}) {
-  return (
-    <div className="space-y-2">
-      <div className="flex items-center gap-2">
-        <HelpCircle className="h-4 w-4 text-tertiary" />
-        <Label className="text-sm">{label}</Label>
-      </div>
-      <div className="space-y-2">
-        <Input
-          type={type}
-          value={value}
-          onChange={(e) => onChange(e.target.value)}
-          disabled={disabled}
-        />
-        {helpText && <p className="text-sm text-secondary">{helpText}</p>}
-      </div>
-    </div>
-  );
-}
-
 // Format date for input field (YYYY-MM-DDTHH:mm)
 function formatDateForInput(date: string | null | undefined): string {
   if (!date) return "";
@@ -689,20 +655,6 @@ function RiskAssessmentsTable({
     return new Date(expiresAt) < new Date();
   };
 
-  // Get severity label based on data sensitivity and business impact
-  const getSeverityLabel = (dataSensitivity: string, businessImpact: string) => {
-    // Simple logic to determine severity - can be adjusted based on requirements
-    if (dataSensitivity === 'CRITICAL' || businessImpact === 'CRITICAL') {
-      return { label: 'Critical', color: 'bg-red-100 text-red-800' };
-    } else if (dataSensitivity === 'HIGH' || businessImpact === 'HIGH') {
-      return { label: 'High', color: 'bg-orange-100 text-orange-800' };
-    } else if (dataSensitivity === 'MEDIUM' || businessImpact === 'MEDIUM') {
-      return { label: 'Medium', color: 'bg-yellow-100 text-yellow-800' };
-    } else {
-      return { label: 'Low', color: 'bg-green-100 text-green-800' };
-    }
-  };
-
   return (
     <div className="space-y-4">
       <div className="rounded-xl border border-[#ECEFEC] bg-white overflow-hidden">
@@ -729,11 +681,6 @@ function RiskAssessmentsTable({
                   <span className="text-xs font-semibold text-[#818780]">Business Impact</span>
                 </div>
               </th>
-              <th className="px-4 py-3 text-left">
-                <div className="flex items-center gap-2">
-                  <span className="text-xs font-semibold text-[#818780]">Severity</span>
-                </div>
-              </th>
               <th className="px-4 py-3 text-left">
                 <div className="flex items-center gap-2">
                   <span className="text-xs font-semibold text-[#818780]">Assessed By</span>
@@ -744,7 +691,6 @@ function RiskAssessmentsTable({
           </thead>
           <tbody>
             {assessments.map((assessment) => {
-              const severity = getSeverityLabel(assessment.dataSensitivity, assessment.businessImpact);
               const expired = isExpired(assessment.expiresAt);
               return (
                 <tr 
@@ -793,14 +739,6 @@ function RiskAssessmentsTable({
                       {assessment.businessImpact.charAt(0) + assessment.businessImpact.slice(1).toLowerCase()}
                     </span>
                   </td>
-                  <td className="px-4 py-3">
-                    <span className={cn(
-                      "text-sm font-normal px-2 py-1 rounded-full",
-                      expired ? "bg-[#F0F0F0] text-[#818780]" : severity.color
-                    )}>
-                      {severity.label}
-                    </span>
-                  </td>
                   <td className="px-4 py-3">
                     <span className={cn(
                       "text-sm font-normal",
@@ -1342,6 +1280,8 @@ function VendorViewContent({
         loadQuery({
           vendorId: data.node.id!,
           organizationId: organizationId!,
+        }, {
+          fetchPolicy: "network-only",
         });
       },
       onError: (error) => {
diff --git a/apps/console/src/pages/organizations/vendors/__generated__/VendorViewCreateRiskAssessmentMutation.graphql.ts b/apps/console/src/pages/organizations/vendors/__generated__/VendorViewCreateRiskAssessmentMutation.graphql.ts
index 66d14c5b1..cf084295d 100644
--- a/apps/console/src/pages/organizations/vendors/__generated__/VendorViewCreateRiskAssessmentMutation.graphql.ts
+++ b/apps/console/src/pages/organizations/vendors/__generated__/VendorViewCreateRiskAssessmentMutation.graphql.ts
@@ -1,5 +1,5 @@
 /**
- * @generated SignedSource<<08de2d8758474132b472d63f413b59a5>>
+ * @generated SignedSource<<863e9f8552e5bb472cc4b02a6dbad1bf>>
  * @lightSyntaxTransform
  * @nogrep
  */
@@ -204,7 +204,7 @@ return {
             "alias": null,
             "args": null,
             "filters": null,
-            "handle": "appendEdge",
+            "handle": "prependEdge",
             "key": "",
             "kind": "LinkedHandle",
             "name": "vendorRiskAssessmentEdge",
@@ -232,6 +232,6 @@ return {
 };
 })();
 
-(node as any).hash = "7def769f97c5a728a7331756a6e3c5b2";
+(node as any).hash = "d28e057eba5f7287f3d0f5e38d444ca5";
 
 export default node;