adjusted investigation filters (intelowlproject#2440)

* adjusted investigation filters

* fixed 'playbook to execute' column

* fix

Author: carellamartina

api_app/investigations_manager/filters.py

@@ -4,6 +4,45 @@
 
 
 class InvestigationFilter(filters.FilterSet):
+    name = filters.CharFilter(lookup_expr="icontains")
+
+    owner = filters.CharFilter(method="filter_for_owner")
+    id = filters.CharFilter(method="filter_for_id")
+    tlp = filters.CharFilter(method="filter_for_tlp")
+    tags = filters.CharFilter(method="filter_for_tags")
+
+    @staticmethod
+    def filter_for_owner(queryset, value, owner, *args, **kwargs):
+        return queryset.filter(owner__username__icontains=owner)
+
+    @staticmethod
+    def filter_for_id(queryset, value, _id, *args, **kwargs):
+        try:
+            int_id = int(_id)
+        except ValueError:
+            # this is to manage bad data as input
+            return queryset
+        else:
+            return queryset.filter(id=int_id)
+
+    @staticmethod
+    def filter_for_tlp(queryset, value, tlp, *args, **kwargs):
+        id_list = [
+            investigation.id
+            for investigation in Investigation.objects.all()
+            if investigation.tlp == tlp
+        ]
+        return queryset.filter(id__in=id_list)
+
+    @staticmethod
+    def filter_for_tags(queryset, value, tags, *args, **kwargs):
+        id_list = [
+            investigation.id
+            for investigation in Investigation.objects.all()
+            if tags in investigation.tags
+        ]
+        return queryset.filter(id__in=id_list)
+
     class Meta:
         model = Investigation
         fields = {

frontend/src/components/investigations/table/investigationTableColumns.jsx

@@ -99,8 +99,8 @@ export const investigationTableColumns = [
         (tag) =>
           tag !== null && (
             <JobTag
-              key={`jobtable-tags-${tag?.label}`}
-              tag={tag}
+              key={`jobtable-tags-${tag}`}
+              tag={{ label: tag, color: "#1655D3" }}
               className="ms-2"
             />
           ),

frontend/src/components/plugins/types/pluginTableColumns.jsx

@@ -303,9 +303,9 @@ export const pivotTableColumns = [
   {
     Header: "Playbook to execute",
     id: "playbook",
-    accessor: "playbook_to_execute",
+    accessor: "playbooks_choice",
     Cell: ({ value }) => (
-      <TableCell isCopyToClipboard isTruncate value={value} />
+      <TableCellList value={value} ulKey={value} size={20} />
     ),
     Filter: SelectColumnFilter,
     maxWidth: 145,