From 8484e55d93f929142f51186a241f2f17c6272877 Mon Sep 17 00:00:00 2001
From: Sami Mokaddem <mokaddem.sami@gmail.com>
Date: Mon, 31 Oct 2022 16:07:47 +0100
Subject: [PATCH] new: [workflowModule:attribute_ids_flag_operation] Module to
 toggle/remove the to_ids flag

---
 .../Module_attribute_ids_flag_operation.php   | 61 +++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 app/Model/WorkflowModules/action/Module_attribute_ids_flag_operation.php

diff --git a/app/Model/WorkflowModules/action/Module_attribute_ids_flag_operation.php b/app/Model/WorkflowModules/action/Module_attribute_ids_flag_operation.php
new file mode 100644
index 000000000..e25f31051
--- /dev/null
+++ b/app/Model/WorkflowModules/action/Module_attribute_ids_flag_operation.php
@@ -0,0 +1,61 @@
+<?php
+include_once APP . 'Model/WorkflowModules/Module_attribute_edition_operation.php';
+
+class Module_attribute_ids_flag_operation extends Module_attribute_edition_operation
+{
+    public $version = '0.1';
+    public $blocking = false;
+    public $id = 'attribute_ids_flag_operation';
+    public $name = 'Attribute IDS Flag operation';
+    public $description = 'Toggle or remove the IDS flag on selected attributes.';
+    public $icon = 'edit';
+    public $inputs = 1;
+    public $outputs = 1;
+    public $support_filters = true;
+    public $expect_misp_core_format = true;
+    public $params = [];
+
+
+    public function __construct()
+    {
+        parent::__construct();
+        $this->params = [
+            [
+                'id' => 'action',
+                'label' => __('To IDS Flag'),
+                'type' => 'select',
+                'options' => [
+                    'add' => __('Toggle IDS flag'),
+                    'remove' => __('Remove IDS flag'),
+                ],
+                'default' => 'add',
+            ],
+        ];
+    }
+
+    public function exec(array $node, WorkflowRoamingData $roamingData, array &$errors = []): bool
+    {
+        parent::exec($node, $roamingData, $errors);
+        $params = $this->getParamsWithValues($node);
+
+        $rData = $roamingData->getData();
+        $user = $roamingData->getUser();
+
+        $matchingItems = $this->getMatchingItemsForAttributes($node, $rData);
+        if ($matchingItems === false) {
+            return true;
+        }
+        $result = $this->__saveAttribute($matchingItems, $rData, $params, $user);
+        return $result;
+    }
+
+    protected function _editAttribute(array $attribute, array $rData, array $params): array
+    {
+        if ($params['action']['value'] == 'remove') {
+            $attribute['to_ids'] = false;
+        } else {
+            $attribute['to_ids'] = true;
+        }
+        return $attribute;
+    }
+}