From 7d866d3ffc76869bee4e37bb76f90a8b98bc9eaa Mon Sep 17 00:00:00 2001
From: Yogesh Ojha <yogesh.ojha11@gmail.com>
Date: Thu, 29 Jun 2023 22:17:49 +0530
Subject: [PATCH] fix project for add target

---
 web/targetApp/templates/target/list.html | 2 +-
 web/targetApp/urls.py                    | 2 +-
 web/targetApp/views.py                   | 8 ++++++--
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/web/targetApp/templates/target/list.html b/web/targetApp/templates/target/list.html
index db9a5f81..07e79fe2 100644
--- a/web/targetApp/templates/target/list.html
+++ b/web/targetApp/templates/target/list.html
@@ -51,7 +51,7 @@ Targets
 						<a class="btn btn-soft-info float-end disabled ms-1 mt-1" href="#" onclick="scanMultipleTargets()" id="scan_multiple_button">Scan Multiple Targets</a>
 						{% endif %}
 						{% if user|can:'modify_targets' %}
-						<a class="btn btn-soft-primary float-end ms-1 mt-1" href="{% url 'add_target' %}" data-toggle="tooltip" data-placement="top" title="Add New Targets">Add Targets</a>
+						<a class="btn btn-soft-primary float-end ms-1 mt-1" href="{% url 'add_target' request.resolver_match.kwargs.slug %}" data-toggle="tooltip" data-placement="top" title="Add New Targets">Add Targets</a>
 						{% endif %}
 					</div>
 				</div>
diff --git a/web/targetApp/urls.py b/web/targetApp/urls.py
index 80819707..8c0e9fd1 100644
--- a/web/targetApp/urls.py
+++ b/web/targetApp/urls.py
@@ -8,7 +8,7 @@ urlpatterns = [
         views.index,
         name='targetIndex'),
     path(
-        'add/target',
+        '<slug:slug>/add/target',
         views.add_target,
         name='add_target'),
     path(
diff --git a/web/targetApp/views.py b/web/targetApp/views.py
index 4b1328ae..34eab0c4 100644
--- a/web/targetApp/views.py
+++ b/web/targetApp/views.py
@@ -32,12 +32,13 @@ def index(request):
 
 
 @has_permission_decorator(PERM_MODIFY_TARGETS, redirect_url=FOUR_OH_FOUR_URL)
-def add_target(request):
+def add_target(request, slug):
     """Add a new target. Targets can be URLs, IPs, CIDR ranges, or Domains.
 
     Args:
         request: Django request.
     """
+    project = Project.objects.get(slug=slug)
     form = AddTargetForm(request.POST or None)
     if request.method == "POST":
         logger.info(request.POST)
@@ -111,6 +112,7 @@ def add_target(request):
                             name=domain_name,
                             description=description,
                             h1_team_handle=h1_team_handle,
+                            project=project,
                             ip_address_cidr=domain_name if is_ip else None)
                         domain.insert_date = timezone.now()
                         domain.save()
@@ -171,6 +173,7 @@ def add_target(request):
                                 continue
                             Domain.objects.create(
                                 name=target_domain,
+                                project=project,
                                 insert_date=timezone.now())
                             added_target_count += 1
 
@@ -195,6 +198,7 @@ def add_target(request):
                                 continue
                             Domain.objects.create(
                                 name=domain,
+                                project=project,
                                 description=description,
                                 insert_date=timezone.now())
                             added_target_count += 1
@@ -220,7 +224,7 @@ def add_target(request):
         msg = f'{added_target_count} targets added successfully'
         logger.info(msg)
         messages.add_message(request, messages.SUCCESS, msg)
-        return http.HttpResponseRedirect(reverse('list_target'))
+        return http.HttpResponseRedirect(reverse('list_target', kwargs={'slug': slug}))
 
     # GET request
     context = {