From f44dadc45343571ecfbc1b9ab061b44733fef369 Mon Sep 17 00:00:00 2001
From: R3D347HR4Y <redeathray@gmail.com>
Date: Sun, 7 Jun 2026 22:05:28 +0200
Subject: [PATCH] feat(admin): add VirusTotal scan settings and mail UI

File policies API key field, conditional scan badge, safe policy merge.
---
 .../sections/file-policies-section.tsx        | 31 +++++++++
 .../gmail/email-view/message-attachments.tsx  | 66 +++++++++----------
 lib/admin-settings/map-api-org-settings.ts    | 23 ++++++-
 lib/admin-settings/org-settings-store.ts      |  1 +
 lib/admin-settings/org-settings-types.ts      |  1 +
 lib/api/admin-org-types.ts                    |  1 +
 lib/api/map-message-attachments.ts            |  2 +
 lib/email-data.ts                             |  2 +
 8 files changed, 91 insertions(+), 36 deletions(-)

diff --git a/components/admin/settings/sections/file-policies-section.tsx b/components/admin/settings/sections/file-policies-section.tsx
index 2507949..3660159 100644
--- a/components/admin/settings/sections/file-policies-section.tsx
+++ b/components/admin/settings/sections/file-policies-section.tsx
@@ -17,6 +17,13 @@ import {
 export function FilePoliciesSection() {
   const filePolicies = useOrgSettingsStore((s) => s.filePolicies)
   const setFilePolicies = useOrgSettingsStore((s) => s.setFilePolicies)
+  const vtKeyConfigured = useOrgSettingsStore(
+    (s) => s.meta?.secrets?.virustotal_api_key?.configured ?? false
+  )
+  const vtKeyMissing =
+    filePolicies.virus_scan_enabled &&
+    !vtKeyConfigured &&
+    !(filePolicies.virustotal_api_key ?? "").trim()
 
   return (
     <OrgSettingsSection
@@ -105,12 +112,36 @@ export function FilePoliciesSection() {
         <label className="flex items-center justify-between gap-4 rounded-lg border p-3 sm:col-span-2">
           <div>
             <p className="text-sm font-medium">Analyse antivirus à l&apos;upload</p>
+            <p className="text-xs text-muted-foreground">
+              VirusTotal — scan synchrone à l&apos;upload Drive et pièces jointes mail
+            </p>
           </div>
           <Switch
             checked={filePolicies.virus_scan_enabled}
             onCheckedChange={(virus_scan_enabled) => setFilePolicies({ virus_scan_enabled })}
           />
         </label>
+        {filePolicies.virus_scan_enabled ? (
+          <div className="sm:col-span-2">
+            <Label>Clé API VirusTotal</Label>
+            <Input
+              className="mt-1 h-9"
+              type="password"
+              autoComplete="off"
+              value={filePolicies.virustotal_api_key ?? ""}
+              onChange={(e) => setFilePolicies({ virustotal_api_key: e.target.value })}
+              placeholder={vtKeyConfigured ? "•••••••• (laisser vide pour conserver)" : "Coller la clé API"}
+            />
+            {vtKeyConfigured && !(filePolicies.virustotal_api_key ?? "").trim() ? (
+              <p className="mt-1 text-xs text-muted-foreground">Clé configurée</p>
+            ) : null}
+            {vtKeyMissing ? (
+              <p className="mt-1 text-xs text-amber-600 dark:text-amber-500">
+                Analyse activée sans clé API — les uploads ne seront pas scannés.
+              </p>
+            ) : null}
+          </div>
+        ) : null}
       </div>
     </OrgSettingsSection>
   )
diff --git a/components/gmail/email-view/message-attachments.tsx b/components/gmail/email-view/message-attachments.tsx
index 06fc497..4e41419 100644
--- a/components/gmail/email-view/message-attachments.tsx
+++ b/components/gmail/email-view/message-attachments.tsx
@@ -85,6 +85,34 @@ function DriveLocationBadge({ folderPath }: { folderPath: string }) {
   )
 }
 
+function attachmentsVirusTotalScanned(attachments: EmailAttachment[]): boolean {
+  return attachments.some((a) => a.virusScanStatus === "clean")
+}
+
+function VirusTotalScanBadge() {
+  return (
+    <>
+      <span aria-hidden> · </span>
+      <span>Analysé par VirusTotal</span>
+      <Tooltip delayDuration={400}>
+        <TooltipTrigger asChild>
+          <button
+            type="button"
+            className="flex h-7 w-7 shrink-0 items-center justify-center rounded-full text-muted-foreground hover:bg-accent"
+            aria-label="Informations sur l'analyse VirusTotal des pièces jointes"
+          >
+            <Info className="size-4" strokeWidth={1.75} />
+          </button>
+        </TooltipTrigger>
+        <TooltipContent side="bottom" className={cn(MAIL_TOOLTIP_CONTENT_CLASS, "max-w-xs text-xs")}>
+          VirusTotal analyse les pièces jointes et les compare à une base de signatures pour
+          repérer les virus et logiciels malveillants.
+        </TooltipContent>
+      </Tooltip>
+    </>
+  )
+}
+
 export function MessageAttachmentsSection({
   messageId,
   attachments,
@@ -117,6 +145,7 @@ export function MessageAttachmentsSection({
 
   const summary = n === 1 ? "Une pièce jointe" : `${n} pièces jointes`
   const asPills = shouldUseAttachmentPillsInPreview(attachments)
+  const showVirusTotal = attachmentsVirusTotalScanned(attachments)
 
   const openPreview = (index: number) => {
     if (!attachments.some((a) => a.id)) {
@@ -167,24 +196,8 @@ export function MessageAttachmentsSection({
           <div className="flex min-w-0 max-w-[min(100%,28rem)] items-center gap-1 text-sm text-muted-foreground">
             <span className="min-w-0 truncate">
               {summary}
-              <span aria-hidden> · </span>
-              <span>Analysé par VirusTotal</span>
+              {showVirusTotal ? <VirusTotalScanBadge /> : null}
             </span>
-            <Tooltip delayDuration={400}>
-              <TooltipTrigger asChild>
-                <button
-                  type="button"
-                  className="flex h-7 w-7 shrink-0 items-center justify-center rounded-full text-muted-foreground hover:bg-accent"
-                  aria-label="Informations sur l'analyse VirusTotal des pièces jointes"
-                >
-                  <Info className="size-4" strokeWidth={1.75} />
-                </button>
-              </TooltipTrigger>
-              <TooltipContent side="bottom" className={cn(MAIL_TOOLTIP_CONTENT_CLASS, "max-w-xs text-xs")}>
-                VirusTotal analyse les pièces jointes et les compare à une base de signatures pour
-                repérer les virus et logiciels malveillants.
-              </TooltipContent>
-            </Tooltip>
           </div>
           {allSaved && uniqueSaveFolders.length === 1 ? (
             <DriveLocationBadge folderPath={uniqueSaveFolders[0]!} />
@@ -316,6 +329,7 @@ export function ConversationAttachmentsSection({
       ? "Une pièce jointe dans cette conversation"
       : `${n} pièces jointes dans cette conversation`
   const asPills = shouldUseAttachmentPillsInPreview(flat.map((item) => item.attachment))
+  const showVirusTotal = attachmentsVirusTotalScanned(flat.map((item) => item.attachment))
 
   const openPreview = (messageId: string, attachments: EmailAttachment[], index: number) => {
     if (!attachments.some((a) => a.id)) {
@@ -336,24 +350,8 @@ export function ConversationAttachmentsSection({
         <div className="flex min-w-0 max-w-[min(100%,28rem)] items-center gap-1 text-sm text-muted-foreground">
           <span className="min-w-0 truncate">
             {summary}
-            <span aria-hidden> · </span>
-            <span>Analysé par VirusTotal</span>
+            {showVirusTotal ? <VirusTotalScanBadge /> : null}
           </span>
-          <Tooltip delayDuration={400}>
-            <TooltipTrigger asChild>
-              <button
-                type="button"
-                className="flex h-7 w-7 shrink-0 items-center justify-center rounded-full text-muted-foreground hover:bg-accent"
-                aria-label="Informations sur l'analyse VirusTotal des pièces jointes"
-              >
-                <Info className="size-4" strokeWidth={1.75} />
-              </button>
-            </TooltipTrigger>
-            <TooltipContent side="bottom" className={cn(MAIL_TOOLTIP_CONTENT_CLASS, "max-w-xs text-xs")}>
-              VirusTotal analyse les pièces jointes et les compare à une base de signatures pour
-              repérer les virus et logiciels malveillants.
-            </TooltipContent>
-          </Tooltip>
         </div>
       </div>
 
diff --git a/lib/admin-settings/map-api-org-settings.ts b/lib/admin-settings/map-api-org-settings.ts
index 3216ab5..c16b05b 100644
--- a/lib/admin-settings/map-api-org-settings.ts
+++ b/lib/admin-settings/map-api-org-settings.ts
@@ -1,6 +1,6 @@
 import type { ApiOrgPolicy, ApiOrgSettingsResponse } from "@/lib/api/admin-org-types"
 import type { OrgPolicySectionKey } from "@/lib/api/admin-org-types"
-import type { IntegrationEntry, OrgSettingsState } from "@/lib/admin-settings/org-settings-types"
+import type { IntegrationEntry, OrgSettingsState, FilePolicySettings } from "@/lib/admin-settings/org-settings-types"
 
 const INTEGRATION_HREFS: Record<string, string> = {
   authentik: "/admin/settings/authentication",
@@ -19,6 +19,25 @@ function mergeIntegrations(
   }))
 }
 
+const DEFAULT_FILE_POLICIES: FilePolicySettings = {
+  max_upload_mib: 512,
+  allowed_extensions: "",
+  block_executable: true,
+  external_sharing: "authenticated",
+  default_link_expiry_days: 30,
+  virus_scan_enabled: false,
+  virustotal_api_key: "",
+  retention_trash_days: 30,
+}
+
+function mergeFilePolicies(fromApi: Partial<FilePolicySettings> | undefined): FilePolicySettings {
+  return {
+    ...DEFAULT_FILE_POLICIES,
+    ...fromApi,
+    virustotal_api_key: fromApi?.virustotal_api_key ?? "",
+  }
+}
+
 export function apiOrgPolicyToStore(policy: ApiOrgPolicy): Partial<OrgSettingsState> {
   return {
     authentik: {
@@ -39,7 +58,7 @@ export function apiOrgPolicyToStore(policy: ApiOrgPolicy): Partial<OrgSettingsSt
     },
     storageQuotas: { ...policy.storage_quotas },
     usageQuotas: { ...policy.usage_quotas },
-    filePolicies: { ...policy.file_policies },
+    filePolicies: mergeFilePolicies(policy.file_policies),
     llm: {
       ...policy.llm,
       providers: policy.llm.providers ?? [],
diff --git a/lib/admin-settings/org-settings-store.ts b/lib/admin-settings/org-settings-store.ts
index a4d7ad8..07e739c 100644
--- a/lib/admin-settings/org-settings-store.ts
+++ b/lib/admin-settings/org-settings-store.ts
@@ -58,6 +58,7 @@ const DEFAULT_FILE_POLICIES: FilePolicySettings = {
   external_sharing: "authenticated",
   default_link_expiry_days: 30,
   virus_scan_enabled: false,
+  virustotal_api_key: "",
   retention_trash_days: 30,
 }
 
diff --git a/lib/admin-settings/org-settings-types.ts b/lib/admin-settings/org-settings-types.ts
index 3ad76fe..d945de1 100644
--- a/lib/admin-settings/org-settings-types.ts
+++ b/lib/admin-settings/org-settings-types.ts
@@ -40,6 +40,7 @@ export type FilePolicySettings = {
   external_sharing: "disabled" | "authenticated" | "public_link"
   default_link_expiry_days: number
   virus_scan_enabled: boolean
+  virustotal_api_key: string
   retention_trash_days: number
 }
 
diff --git a/lib/api/admin-org-types.ts b/lib/api/admin-org-types.ts
index 2ceb89b..d11bc16 100644
--- a/lib/api/admin-org-types.ts
+++ b/lib/api/admin-org-types.ts
@@ -40,6 +40,7 @@ export type ApiOrgFilePolicies = {
   external_sharing: "disabled" | "authenticated" | "public_link"
   default_link_expiry_days: number
   virus_scan_enabled: boolean
+  virustotal_api_key: string
   retention_trash_days: number
 }
 
diff --git a/lib/api/map-message-attachments.ts b/lib/api/map-message-attachments.ts
index c86024a..c976ac6 100644
--- a/lib/api/map-message-attachments.ts
+++ b/lib/api/map-message-attachments.ts
@@ -9,6 +9,7 @@ export interface ApiMessageAttachment {
   is_inline?: boolean
   content_id?: string
   drive_path?: string
+  virus_scan_status?: "clean" | "skipped" | "malicious"
 }
 
 export function mapApiAttachmentsToEmail(
@@ -24,6 +25,7 @@ export function mapApiAttachmentsToEmail(
       contentType: a.content_type || undefined,
       drivePath: a.drive_path || undefined,
       sizeBytes: a.size > 0 ? a.size : undefined,
+      virusScanStatus: a.virus_scan_status,
     }))
 }
 
diff --git a/lib/email-data.ts b/lib/email-data.ts
index 298f6c0..0ab607a 100644
--- a/lib/email-data.ts
+++ b/lib/email-data.ts
@@ -11,6 +11,8 @@ export interface EmailAttachment {
   drivePath?: string
   /** Taille en octets (optionnelle) — tooltips / pills vs cartes dans l’aperçu */
   sizeBytes?: number
+  /** Statut scan VirusTotal côté serveur */
+  virusScanStatus?: "clean" | "skipped" | "malicious"
   /** Contenu texte inline (ex. ICS) — fixtures / import */
   inlineText?: string
 }