From b1357612a45138222f0fac53dd1a1d28d6dae945 Mon Sep 17 00:00:00 2001
From: Simon Erkelens <github@casa-laguna.net>
Date: Fri, 30 Jun 2023 10:32:50 +1200
Subject: [PATCH] Add connect-src, add searchable for type

---
 src/Models/CSPDomain.php | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/Models/CSPDomain.php b/src/Models/CSPDomain.php
index 2805090..43309ac 100644
--- a/src/Models/CSPDomain.php
+++ b/src/Models/CSPDomain.php
@@ -5,6 +5,8 @@
 use Page;
 use SilverStripe\Forms\FieldList;
 use SilverStripe\ORM\DataObject;
+use SilverStripe\ORM\FieldType\DBEnum;
+use SilverStripe\ORM\FieldType\DBVarchar;
 use SilverStripe\ORM\ManyManyList;
 use SilverStripe\Security\Member;
 use SilverStripe\Security\Permission;
@@ -25,8 +27,8 @@ class CSPDomain extends DataObject implements PermissionProvider
     private static $table_name = 'CSPDomain';
 
     private static $db = [
-        'Domain' => 'Varchar(255)',
-        'Source' => 'Enum("default,script,style,img,media,font,form,frame,ancestor,worker")'
+        'Domain' => DBVarchar::class,
+        'Source' => DBEnum::class . '("default,script,style,img,media,font,form,frame,ancestor,worker,connect")'
     ];
 
     private static $belongs_many_many = [
@@ -55,11 +57,13 @@ class CSPDomain extends DataObject implements PermissionProvider
         'font'    => 'Fonts',
         'form'    => 'Forms',
         'frame'   => 'Iframes',
-        'worker'  => 'Worker'
+        'worker'  => 'Worker',
+        'connect' => 'Connect'
     ];
 
     private static $searchable_fields = [
         'Domain',
+        'Source'
     ];
 
     /**