Merge pull request #1413 from CybercentreCanada/assemblyline/issues/248

Add interface for updating the POST data field (dev)
CybercentreCanada · Feb 10, 2025 · 4d6c95d · 4d6c95d
2 parents 39b1b7d + 2a65241
commit 4d6c95d
Show file tree

Hide file tree

Showing 4 changed files with 41 additions and 1 deletion.
diff --git a/src/components/models/base/service.ts b/src/components/models/base/service.ts
@@ -271,6 +271,9 @@ export type UpdateSource = {
   /** Pattern used to find files of interest from source */
   pattern?: string;
 
+  /** Data that's sent in POST requests */
+  post_data?: { [key: string]: any };
+
   /** Private key used to authenticate with source */
   private_key?: string;
 
@@ -455,6 +458,7 @@ export const DEFAULT_SOURCE: UpdateSource = {
   override_classification: false,
   password: '',
   pattern: '',
+  post_data: {},
   private_key: '',
   proxy: '',
   ssl_ignore_errors: false,

diff --git a/src/components/routes/manage/signature_sources_details.tsx b/src/components/routes/manage/signature_sources_details.tsx
@@ -69,6 +69,7 @@ const WrappedSourceDetail = ({
   const [showPrivateKey, setShowPrivateKey] = useState<boolean>(false);
 
   const gitFetch = useMemo<boolean>(() => source.fetch_method === 'GIT', [source.fetch_method]);
+  const postFetch = useMemo<boolean>(() => source.fetch_method === 'POST', [source.fetch_method]);
 
   const handleFieldChange = useCallback(
     event => {
@@ -119,6 +120,14 @@ const WrappedSourceDetail = ({
     [setModified, setSource, source]
   );
 
+  const handlePostDataChange = useCallback(
+    event => {
+      setSource({ ...source, post_data: event.updated_src });
+      setModified(true);
+    },
+    [setModified, setSource, source]
+  );
+
   const handleUpdateIntervalChange = useCallback(
     event => {
       if (!event.target.value) {
@@ -470,7 +479,32 @@ const WrappedSourceDetail = ({
                         }}
                       />
                     </Grid>
-
+                    {postFetch && (
+                      <Grid item xs={12}>
+                        <div className={classes.label}>{t('post_data')}</div>
+                        <JSONEditor
+                          name={false}
+                          src={source.post_data}
+                          enableClipboard={false}
+                          groupArraysAfterLength={10}
+                          displayDataTypes={false}
+                          displayObjectSize={false}
+                          onAdd={handlePostDataChange}
+                          onDelete={handlePostDataChange}
+                          onEdit={handlePostDataChange}
+                          collapsed={true}
+                          style={{
+                            border: `1px solid ${theme.palette.divider}`,
+                            borderRadius: '4px',
+                            fontSize: '1rem',
+                            minHeight: theme.spacing(5),
+                            padding: '4px',
+                            overflowX: 'auto',
+                            width: '100%'
+                          }}
+                        />
+                      </Grid>
+                    )}
                     <Grid item xs={12}>
                       <div className={classes.label}>
                         {t('proxy')}

diff --git a/src/locales/en/manage/signature_sources.json b/src/locales/en/manage/signature_sources.json
@@ -31,6 +31,7 @@
   "override_classification": "Override signature-defined classification with source classification",
   "password": "Password",
   "pattern": "Regex pattern",
+  "post_data": "POST Data",
   "private_key": "Private key",
   "private_key_used": "A private key will be used to connect to the source",
   "proxy": "Proxy server",

diff --git a/src/locales/fr/manage/signature_sources.json b/src/locales/fr/manage/signature_sources.json
@@ -31,6 +31,7 @@
   "override_classification": "Override signature-defined classification with source classificationRemplacer la classification définie par la signature par la classification de la source",
   "password": "Mot de passe",
   "pattern": "Modèle Regex",
+  "post_data": "Données POST",
   "private_key": "Clé privée",
   "private_key_used": "Une clé privée sera utilisée pour se connecter à la source",
   "proxy": "Serveur Proxy",