{% extends 'base/base.html' %}
{% load static %}

{% block title %}
Add targets
{% endblock title %}


{% block custom_js_css_link %}
{% endblock custom_js_css_link %}

{% block breadcrumb_title %}
<li class="breadcrumb-item"><a href="{% url 'list_target' current_project.slug %}">Targets</a></li>
<li class="breadcrumb-item active">Add Target</li>
{% endblock breadcrumb_title %}

{% block page_title %}
Add or Import Targets
{% endblock page_title %}

{% block main_content %}
<div class="row">
  <div class="col-xl-12">
    <div class="card">
      <div class="card-body">
        <ul class="nav nav-tabs nav-bordered">
          <!-- <li class="nav-item">
            <a href="#pills-add" data-bs-toggle="tab" aria-expanded="false" class="nav-link active">
              Add Target
            </a>
          </li> -->
          <li class="nav-item">
            <a href="#add-multiple-tab" data-bs-toggle="tab" aria-expanded="false" class="nav-link active">
              Add Targets
            </a>
          </li>
          <li class="nav-item">
            <a href="#ip-address-tab" data-bs-toggle="tab" aria-expanded="true" class="nav-link">
              Resolve and add IP Address
            </a>
          </li>
          <li class="nav-item dropdown">
            <a class="nav-link dropdown-toggle arrow-none" href="#" id="import-dropdown" role="button"
            data-bs-toggle="dropdown" aria-haspopup="true" aria-expanded="false">
            Import
            <div class="arrow-down"></div>
          </a>
          <div class="dropdown-menu" aria-labelledby="import-dropdown">
            <a class="dropdown-item" href="#import-text-tab" data-bs-toggle="tab" aria-expanded="false" class="nav-link">Import from Text File</a>
            <a class="dropdown-item" href="#import-csv-tab" data-bs-toggle="tab" aria-expanded="false" class="nav-link">Import from CSV</a>
          </div>
        </li>
      </ul>
      <div class="tab-content" id="pills-tabContent">
        <div class="tab-pane fade" id="ip-address-tab" role="tabpanel" aria-labelledby="ip-address-tab">
          <form method="post">
            {% csrf_token %}
            <div class="row">
              <div class="col-12">
                <div class="alert alert-primary border-0 mb-4" role="alert">
                  Please enter your Target IP address, reNgine will look for the <b>A</b> records and resolve them to the domains.
                </div>
                <label for="ip_address" class="text-dark">IP Address or CIDR</br><span class="text-muted">Example: 192.168.1.1 or 192.168.1.1/24, both are supported</span></label>
                <input type="text" name="ip_address" class="form-control form-control-lg" id="ip_address" placeholder="X.X.X.X">
                <button class="btn btn-primary mt-2 float-end" disabled type="button" name="resolve_ip_button" id="resolve_ip_button"><svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="feather feather-repeat"><polyline points="17 1 21 5 17 9"></polyline><path d="M3 11V9a4 4 0 0 1 4-4h14"></path><polyline points="7 23 3 19 7 15"></polyline><path d="M21 13v2a4 4 0 0 1-4 4H3"></path></svg> Resolve IP Address</button>
              </div>
              <div class="col-12">
                <div id='all_domains_checkbox' class='mt-2 n-chk'>
                  <div class="form-check">
                    <input type="checkbox" class="form-check-input" id="select_all_ip_domains">
                    <label class="form-check-label" for="select_all_ip_domains">Select All Domains</label>
                  </div>
                </div>
              </div>
              <div class="col-12 mt-3">
                <div id="resolved_domains_div">
                </div>
              </div>
              <div class="col-12 mt-3">
                <label for="targetDescription" class="form-label">Target Description (Optional)</label>
                <input type="text" class="form-control form-control-lg" id="targetDescription" placeholder="Interesting Target" name="targetDescription">
              </div>
              {% if user_preferences.bug_bounty_mode %}
              <div class="col-12 mt-3">
                <label for="domainDescription" class="form-label">HackerOne Target Team Handle
                  <br>
                  This is used to send vulnerability reports to the HackerOne Program automatically. Team handle can be found from the program URL, <a href="https://hackerone.com/team_handle">https://hackerone.com/team_handle</a>
                </label>
                <input type="text" class="form-control form-control-lg" id="targetH1TeamHandle" placeholder="team_handle" name="targetH1TeamHandle">
                <button class="btn btn-primary submit-fn mt-2 float-end" disabled type="submit" name="add-ip-target" id="add-ip-target" value="submit"><svg xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="feather feather-plus"><line x1="12" y1="5" x2="12" y2="19"></line><line x1="5" y1="12" x2="19" y2="12"></line></svg> <span id="add_ip_target_btn">Add Target</span></button>
              </div>
              {% endif %}
            </form>
          </div>
        </div>
        <div class="tab-pane fade show active" id="add-multiple-tab" role="tabpanel" aria-labelledby="add-multiple-tab">
          <form method="post">
            {% csrf_token %}
            <div class="row">
              <div class="col-12">
                <label for="importSubdomainFormControlTextarea">You can add one or more targets. If you are adding multiple targets, separate them using new line. If the subdomain already exists, it will be skipped.</label>
                <textarea class="form-control" id="addTargets" rows="8" spellcheck="false" name="addTargets" placeholder="example.com"></textarea>
              </div>
              <div class="col-12 mt-3">
                <label for="targetDescription">Target Description (Optional)</label>
                <input type="text" class="form-control form-control-lg" id="targetDescription" placeholder="Interesting Target" name="targetDescription">
              </div>
              {% if user_preferences.bug_bounty_mode %}
              <div class="col-12 mt-3">
                <label for="domainDescription" class="form-label">HackerOne Target Team Handle
                  <br>
                  This is used to send vulnerability reports to the HackerOne Program automatically. Team handle can be found from the program URL, <a href="https://hackerone.com/team_handle">https://hackerone.com/team_handle</a>
                </label>
                <input type="text" class="form-control form-control-lg" id="targetH1TeamHandle" placeholder="team_handle" name="targetH1TeamHandle">
              </div>
              {% endif %}
              <div class="col-12 mt-3">
                <label for="targetDescription">Target Organization (Optional)</label>
                <input type="text" class="form-control form-control-lg" id="targetOrganization" placeholder="Example Org" name="targetOrganization">
              </div>
            </div>
            <button class="btn btn-primary submit-fn mt-2 float-end" type="submit" id="add-multiple-targets" name="add-multiple-targets" value="submit">Add 0 Target</button>
          </form>
        </div>
        <div class="tab-pane fade" id="import-text-tab" role="tabpanel" aria-labelledby="import-text-tab">
          <div class="row">
            <div class="col-12">
              <div class="alert alert-primary border-0 mb-4" role="alert">
                Your txt file must have list of domains separated by a new line.
                <br><br>
                By default all domains imported from txt will have no description and no organization. If you choose to import multiple domains with description and/or organization, csv import is recommended.
              </div>
              <form method="post" enctype="multipart/form-data">
                <div class="mb-3">
                  {% csrf_token %}
                  <input type="file" class="form-control" id="txtFile" required accept=".txt" name="txtFile" onchange="showname(this, '#selectedTextFileName')">
                  <h6 id="selectedTextFileName" class="text-primary"></h6>
                </div>
                <input type="submit" class="btn btn-primary float-right mt-5 mb-2" value="Upload" name="import-txt-target">
              </form>
            </div>
          </div>
        </div>
        <div class="tab-pane fade" id="import-csv-tab" role="tabpanel" aria-labelledby="import-csv-tab">
          <div class="row">
            <div class="col-12">
              <div class="alert alert-warning border-0 mb-4" role="alert">
                Your csv file must be in the format of <strong>domain, description, organization</strong> separated by a new line.
              </div>
              <form method="post" enctype="multipart/form-data">
                <div class="mb-3">
                  {% csrf_token %}
                  <input type="file" class="form-control" id="csvFile" required accept=".csv" name="csvFile" onchange="showname(this, '#selectedCsvFileName')">
                  <h6 id="selectedCsvFileName" class="text-primary"></h6>
                </div>
                <input type="submit" class="btn btn-primary float-right mt-5 mb-2" value="Upload" name="import-csv-target">
              </form>
            </div>
          </div>
        </div>
      </div>
    </div>
  </div>
</div>
{% endblock main_content %}


{% block page_level_script %}
<script src="{% static 'custom/custom.js' %}"></script>
<script type="text/javascript">
  $(document).ready(function(){
    var add_multiple_targets_btn = document.getElementById("add-multiple-targets");
    add_multiple_targets_btn.disabled = true;
    $('#addTargets').on('change keyup',function(){
      var text = $("#addTargets").val();
      var lines = text.split(/\r|\r\n|\n/);
      var count = lines.length;
      add_multiple_targets_btn.innerHTML = `Add ${count} Targets`;
      if (!count) {
        add_multiple_targets_btn.disabled = true;
      }
      else{
        add_multiple_targets_btn.disabled = false;
      }
    });

    $('#ip_address').on('change keyup',function(){
      var resolve_ip_button = document.getElementById("resolve_ip_button");
      var ip_addr = $("#ip_address").val();
      if (/^([0-9]{1,3}\.){3}[0-9]{1,3}(\/([0-9]|[1-2][0-9]|3[0-2]))?$/.test(ip_addr)) {
        resolve_ip_button.disabled = false;
      }
      else {
        resolve_ip_button.disabled = true;
      }
    });

    // hide all checkbox by default
    $("#all_domains_checkbox").hide();
    $("#resolve_ip_button").click(function(){
      // get ip address
      var ip_addr = $("#ip_address").val();
      $('#resolved_domains_div').empty();
      swal.fire({
        title: `Fetching domains`,
        text: `Please wait while we fetch domains associated with IP address or CIDR ${ip_address.value}`,
        padding: '2em',
        onOpen: function() {
          swal.showLoading()
          return fetch(`/api/tools/ip_to_domain/?format=json&ip_address=${ip_address.value}`, {
            method: 'GET',
            credentials: "same-origin",
            headers: {
              "X-CSRFToken": getCookie("csrftoken")
            }
          })
          .then(function(response) {
            return response.json();
          }).then(function(json_data) {
            if (json_data['status']) {
              // #resolved_domains_div
              $("#all_domains_checkbox").show();
              if(Array.isArray(json_data['ip_address'])) {
                $('#resolved_domains_div').append(`<b class='text-info'>${json_data['ip_address'].length} domains associated with IP Address ${json_data['orig']}</b></br>`);
                $('#resolved_domains_div').append(`<b>Please select the domains to import.</b>`);
                $('#resolved_domains_div').append(`<div id='domains_checkbox' class='mt-2 row'></div>`);
                json_data['ip_address'].forEach((ip_info, index, array) => {
                  for (var domain in ip_info['domains']) {
                    $('#domains_checkbox').append(`
                    <div class="col-xl-2 col-md-4 col-sm-6 col-12">
                      <div class="form-check">
                        <input type="checkbox" class="form-check-input resolved_ip_domains" name="resolved_ip_domains" id="${ip_info['domains'][domain]}" value="${ip_info['domains'][domain]}">
                        <label class="form-check-label" for="${ip_info['domains'][domain]}">${ip_info['domains'][domain]}</label>
                      </div>
                    </div>`
                    );
                  }
                })
              }
              swal.close();
              // resolved_ip_domains if any is checked, then only enable add button
              $(".resolved_ip_domains").change(function(){
                var add_multiple_targets_btn = document.getElementById("add-ip-target");
                var checked_count = $('.resolved_ip_domains:checked').length;
                if (checked_count > 0) {
                  add_multiple_targets_btn.disabled = false;
                  $('#add_ip_target_btn').html(`Add ${checked_count} Targets`);
                }
                else{
                  add_multiple_targets_btn.disabled = true;
                }
              });
            }
            else{
              $("#all_domains_checkbox").hide();
              swal.close();
              swal({
                title: `Sorry, no domains associated with IP Address ${ip_address.value}!`,
                type: 'error',
                padding: '2em'
              })
            }
          })
          .catch(function() {
            swal.insertQueueStep({
              type: 'error',
              title: 'Oops! Unable to fetch domains for this IP!'
            })
          })
        }
      });
    });

    // if select all checkbox is checked, select all domains
    $("#select_all_ip_domains").click(function () {
      $(".resolved_ip_domains").prop('checked', $(this).prop('checked'));
      document.getElementById("add-ip-target").disabled = false;
      var checked_count = $('.resolved_ip_domains:checked').length;
      $('#add_ip_target_btn').html(`Add ${checked_count} Targets`);
    });

    $('#domainName').on('change keyup',function(){
      if ($('#domainName').val().length) {
        $("#add-single-target").removeAttr("disabled");
        $("#btn_view_whois").removeClass("btn disabled");
      }
      else{
        $("#add-single-target").prop("disabled", true);
        $("#btn_view_whois").addClass("btn disabled");
      }
    });
  });

  function showname (file, id) {
    $(id).html('Selected file: ' + htmlEncode(file.files.item(0).name));
  };

  function show_whois(){
    // check if target exists or not
    var domain = document.getElementById("domainName").value;
    if (domain) {
      get_domain_whois(domain);
    }
  }

</script>
{% endblock page_level_script %}