From b2cda37dd03f5c5c553370c9166319769dc4d133 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@puigdefabregas.eu>
Date: Mon, 28 Oct 2024 17:52:34 +0100
Subject: [PATCH 1/2] add erase server definition

---
 device/models.py                              | 16 +++++
 device/templates/details.html                 | 27 +++++---
 evidence/forms.py                             | 53 ++++++++++++++++
 .../migrations/0002_alter_annotation_type.py  | 25 ++++++++
 evidence/models.py                            |  1 +
 evidence/templates/ev_details.html            | 32 ++++++++--
 evidence/urls.py                              |  1 +
 evidence/views.py                             | 61 ++++++++++++++++---
 lot/migrations/0002_alter_lot_closed.py       | 18 ++++++
 9 files changed, 210 insertions(+), 24 deletions(-)
 create mode 100644 evidence/migrations/0002_alter_annotation_type.py
 create mode 100644 lot/migrations/0002_alter_lot_closed.py

diff --git a/device/models.py b/device/models.py
index da53de8..a351a11 100644
--- a/device/models.py
+++ b/device/models.py
@@ -108,6 +108,22 @@ class Device:
             return
         annotation = annotations.first()
         self.last_evidence = Evidence(annotation.uuid)
+        
+    def is_eraseserver(self):
+        if not self.uuids:
+            self.get_uuids()
+        if not self.uuids:
+            return False
+
+        annotation = Annotation.objects.filter(
+            uuid__in=self.uuids,
+            owner=self.owner,
+            type=Annotation.Type.ERASE_SERVER
+        ).first()
+        
+        if annotation:
+            return True
+        return False
 
     def last_uuid(self):
         return self.uuids[0]
diff --git a/device/templates/details.html b/device/templates/details.html
index 51e97f1..d0e9c02 100644
--- a/device/templates/details.html
+++ b/device/templates/details.html
@@ -12,22 +12,22 @@
   <div class="col">
     <ul class="nav nav-tabs nav-tabs-bordered">
       <li class="nav-items">
-        <a href="#details" class="nav-link active" data-bs-toggle="tab" data-bs-target="#details">General details</a>
+        <a href="#details" class="nav-link active" data-bs-toggle="tab" data-bs-target="#details">{% trans "General details" %}</a>
       </li>
       <li class="nav-items">
-        <a href="#annotations" class="nav-link" data-bs-toggle="tab" data-bs-target="#annotations">User annotations</a>
+        <a href="#annotations" class="nav-link" data-bs-toggle="tab" data-bs-target="#annotations">{% trans "User annotations" %}</a>
       </li>
       <li class="nav-items">
-        <a href="#documents" class="nav-link" data-bs-toggle="tab" data-bs-target="#documents">Documents</a>
+        <a href="#documents" class="nav-link" data-bs-toggle="tab" data-bs-target="#documents">{% trans "Documents" %}</a>
       </li>
       <li class="nav-items">
-        <a href="#lots" class="nav-link" data-bs-toggle="tab" data-bs-target="#lots">Lots</a>
+        <a href="#lots" class="nav-link" data-bs-toggle="tab" data-bs-target="#lots">{% trans "Lots" %}</a>
       </li>
       <li class="nav-items">
-        <a href="#components" class="nav-link" data-bs-toggle="tab" data-bs-target="#components">Components</a>
+        <a href="#components" class="nav-link" data-bs-toggle="tab" data-bs-target="#components">{% trans "Components" %}</a>
       </li>
       <li class="nav-items">
-        <a href="#evidences" class="nav-link" data-bs-toggle="tab" data-bs-target="#evidences">Evidences</a>
+        <a href="#evidences" class="nav-link" data-bs-toggle="tab" data-bs-target="#evidences">{% trans "Evidences" %}</a>
       </li>
       <li class="nav-items">
         <a href="#web" class="nav-link" href="">Web</a>
@@ -38,14 +38,21 @@
 <div class="tab-content pt-2">
 
   <div class="tab-pane fade show active" id="details">
-    <h5 class="card-title">Details</h5>
+    <h5 class="card-title">{% trans "Details" %}</h5>
     <div class="row mb-3">
       <div class="col-lg-3 col-md-4 label ">Phid</div>
       <div class="col-lg-9 col-md-8">{{ object.id }}</div>
     </div>
 
+    {% if object.is_eraseserver %}
+    <div class="row mb-3">
+      <div class="col-lg-3 col-md-4 label">{% trans "Is a erase server" %}</div>
+      <div class="col-lg-9 col-md-8"></div>
+    </div>
+    {% endif %}
+
     <div class="row">
-      <div class="col-lg-3 col-md-4 label ">Type</div>
+      <div class="col-lg-3 col-md-4 label ">{% trans "Type" %}</div>
       <div class="col-lg-9 col-md-8">{{ object.type }}</div>
     </div>
 
@@ -86,12 +93,12 @@
     <div class="btn-group dropdown ml-1 mt-1" uib-dropdown="">
       <a href="{% url 'device:add_annotation' object.pk %}" class="btn btn-primary">
         <i class="bi bi-plus"></i>
-        Add new annotation
+        {% trans "Add new annotation" %}
         <span class="caret"></span>
       </a>
     </div>
 
-    <h5 class="card-title mt-2">Annotations</h5>
+    <h5 class="card-title mt-2">{% trans "Annotations" %}</h5>
     <table class="table table-striped">
       <thead>
         <tr>
diff --git a/evidence/forms.py b/evidence/forms.py
index cf90a71..dab7a60 100644
--- a/evidence/forms.py
+++ b/evidence/forms.py
@@ -162,3 +162,56 @@ class ImportForm(forms.Form):
                 return table
 
         return
+
+
+class EraseServerForm(forms.Form):
+    erase_server = forms.BooleanField(label=_("Is a Erase Server"), required=False)
+
+    def __init__(self, *args, **kwargs):
+        self.pk = None
+        self.uuid = kwargs.pop('uuid', None)
+        self.user = kwargs.pop('user')
+        instance = Annotation.objects.filter(
+            uuid=self.uuid,
+            type=Annotation.Type.ERASE_SERVER,
+            key='ERASE_SERVER',
+            owner=self.user.institution
+        ).first()
+
+        if instance:
+            kwargs["initial"]["erase_server"] = instance.value
+            self.pk = instance.pk
+
+        super().__init__(*args, **kwargs)
+
+    def clean(self):
+        self.erase_server = self.cleaned_data.get('erase_server', False)
+        self.instance = Annotation.objects.filter(
+            uuid=self.uuid,
+            type=Annotation.Type.ERASE_SERVER,
+            key='ERASE_SERVER',
+            owner=self.user.institution
+        ).first()
+
+        return True
+
+    def save(self, user, commit=True):
+        if not commit:
+            return
+
+        if not self.erase_server:
+            if self.instance:
+                self.instance.delete()
+            return
+
+        if self.instance:
+            return
+        
+        Annotation.objects.create(
+            uuid=self.uuid,
+            type=Annotation.Type.ERASE_SERVER,
+            key='ERASE_SERVER',
+            value=self.erase_server,
+            owner=self.user.institution,
+            user=self.user
+        )
diff --git a/evidence/migrations/0002_alter_annotation_type.py b/evidence/migrations/0002_alter_annotation_type.py
new file mode 100644
index 0000000..24957bc
--- /dev/null
+++ b/evidence/migrations/0002_alter_annotation_type.py
@@ -0,0 +1,25 @@
+# Generated by Django 5.0.6 on 2024-10-28 12:56
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("evidence", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="annotation",
+            name="type",
+            field=models.SmallIntegerField(
+                choices=[
+                    (0, "System"),
+                    (1, "User"),
+                    (2, "Document"),
+                    (3, "EraseServer"),
+                ]
+            ),
+        ),
+    ]
diff --git a/evidence/models.py b/evidence/models.py
index 8c5d6f9..fef6a92 100644
--- a/evidence/models.py
+++ b/evidence/models.py
@@ -14,6 +14,7 @@ class Annotation(models.Model):
         SYSTEM= 0, "System"
         USER = 1, "User"
         DOCUMENT = 2, "Document"
+        ERASE_SERVER = 3, "EraseServer"
 
     created = models.DateTimeField(auto_now_add=True)
     uuid = models.UUIDField()
diff --git a/evidence/templates/ev_details.html b/evidence/templates/ev_details.html
index 94c3058..6a40962 100644
--- a/evidence/templates/ev_details.html
+++ b/evidence/templates/ev_details.html
@@ -12,13 +12,16 @@
   <div class="col">
     <ul class="nav nav-tabs nav-tabs-bordered">
       <li class="nav-items">
-        <button class="nav-link active" data-bs-toggle="tab" data-bs-target="#device">Devices</button>
+        <button class="nav-link active" data-bs-toggle="tab" data-bs-target="#device">{% trans "Devices" %}</button>
       </li>
       <li class="nav-items">
-        <button class="nav-link" data-bs-toggle="tab" data-bs-target="#tag">Tag</button>
-      <li class="nav-items">
-        <a href="{% url 'evidence:download' object.uuid %}" class="nav-link">Download File</a>
+	<a href="#tag" class="nav-link" data-bs-toggle="tab" data-bs-target="#tag">{% trans "Tag" %}</a>
       </li>
+      <li class="nav-items">
+        <a href="{% url 'evidence:erase_server' object.uuid %}" class="nav-link">{% trans "Erase Server" %}</a>
+      </li>
+      <li class="nav-items">
+        <a href="{% url 'evidence:download' object.uuid %}" class="nav-link">{% trans "Download File" %}</a>
       </li>
     </ul>
   </div>
@@ -83,3 +86,24 @@
   </div>
 </div>
 {% endblock %}
+
+{% block extrascript %}
+<script>
+  document.addEventListener("DOMContentLoaded", function() {
+    // Obtener el hash de la URL (ejemplo: #components)
+    const hash = window.location.hash;
+
+    // Verificar si hay un hash en la URL
+    if (hash) {
+      // Buscar el botón o enlace que corresponde al hash y activarlo
+      const tabTrigger = document.querySelector(`[data-bs-target="${hash}"]`);
+      
+      if (tabTrigger) {
+        // Crear una instancia de tab de Bootstrap para activar el tab
+        const tab = new bootstrap.Tab(tabTrigger);
+        tab.show();
+      }
+    }
+  });
+</script>
+{% endblock %}
diff --git a/evidence/urls.py b/evidence/urls.py
index 87ccdd5..9a4e2a2 100644
--- a/evidence/urls.py
+++ b/evidence/urls.py
@@ -18,6 +18,7 @@ urlpatterns = [
     path("upload", views.UploadView.as_view(), name="upload"),
     path("import", views.ImportView.as_view(), name="import"),
     path("<uuid:pk>", views.EvidenceView.as_view(), name="details"),
+    path("<uuid:pk>/eraseserver", views.EraseServerView.as_view(), name="erase_server"),
     path("<uuid:pk>/download", views.DownloadEvidenceView.as_view(), name="download"),
     path('annotation/<int:pk>/del', views.AnnotationDeleteView.as_view(), name='delete_annotation'),
 ]
diff --git a/evidence/views.py b/evidence/views.py
index 81ff767..201a621 100644
--- a/evidence/views.py
+++ b/evidence/views.py
@@ -11,18 +11,14 @@ from django.views.generic.edit import (
     FormView,
 )
 
-
 from dashboard.mixins import  DashboardView, Http403
 from evidence.models import Evidence, Annotation
-from evidence.forms import UploadForm, UserTagForm, ImportForm
-# from django.shortcuts import render
-# from rest_framework import viewsets
-# from snapshot.serializers import SnapshotSerializer
-
-
-# class SnapshotViewSet(viewsets.ModelViewSet):
-#     queryset = Snapshot.objects.all()
-#     serializer_class = SnapshotSerializer
+from evidence.forms import (
+    UploadForm,
+    UserTagForm,
+    ImportForm,
+    EraseServerForm
+)
 
 
 class ListEvidencesView(DashboardView, TemplateView):
@@ -167,3 +163,48 @@ class AnnotationDeleteView(DashboardView, DeleteView):
 
 
         return redirect(url_name, **kwargs_view)
+
+
+class EraseServerView(DashboardView, FormView):
+    template_name = "ev_eraseserver.html"
+    section = "evidences"
+    title = _("Evidences")
+    breadcrumb = "Evidences / Details"
+    success_url = reverse_lazy('evidence:list')
+    form_class = EraseServerForm
+
+    def get(self, request, *args, **kwargs):
+        self.pk = kwargs['pk']
+        self.object = Evidence(self.pk)
+        if self.object.owner != self.request.user.institution:
+            raise Http403
+
+        self.object.get_annotations()
+        return super().get(request, *args, **kwargs)
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context.update({
+            'object': self.object,
+        })
+        return context
+
+    def get_form_kwargs(self):
+        self.pk = self.kwargs.get('pk')
+        kwargs = super().get_form_kwargs()
+        kwargs['uuid'] = self.pk
+        kwargs['user'] = self.request.user
+        return kwargs
+
+    def form_valid(self, form):
+        form.save(self.request.user)
+        response = super().form_valid(form)
+        return response
+
+    def form_invalid(self, form):
+        response = super().form_invalid(form)
+        return response
+
+    def get_success_url(self):
+        success_url = reverse_lazy('evidence:details', args=[self.pk])
+        return success_url
diff --git a/lot/migrations/0002_alter_lot_closed.py b/lot/migrations/0002_alter_lot_closed.py
new file mode 100644
index 0000000..a694df6
--- /dev/null
+++ b/lot/migrations/0002_alter_lot_closed.py
@@ -0,0 +1,18 @@
+# Generated by Django 5.0.6 on 2024-10-28 12:56
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("lot", "0001_initial"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="lot",
+            name="closed",
+            field=models.BooleanField(default=False),
+        ),
+    ]

From 356dd590d22aabe1ffe67f69a56e2eb4a8218df2 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@puigdefabregas.eu>
Date: Mon, 28 Oct 2024 18:09:45 +0100
Subject: [PATCH 2/2] fix duplicates in search view

---
 dashboard/views.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/dashboard/views.py b/dashboard/views.py
index 36369c5..8d91732 100644
--- a/dashboard/views.py
+++ b/dashboard/views.py
@@ -70,10 +70,17 @@ class SearchView(InventaryMixin):
             return self.search_hids(query, offset, limit)
 
         devices = []
+        dev_id = []
+
         for x in matches:
-            devices.append(self.get_annotations(x))
+            # devices.append(self.get_annotations(x))
+            dev = self.get_annotations(x)
+            if dev.id not in dev_id:
+                devices.append(dev)
+                dev_id.append(dev.id)
 
         count = matches.size()
+        # TODO fix of pagination, the count is not correct
         return devices, count
 
     def get_annotations(self, xp):