Merge pull request #31 from jtdub/whois

Whois

Author: jtdub

development/Dockerfile

@@ -14,7 +14,7 @@ COPY poetry.lock .
 
 # hadolint ignore=DL3008
 RUN apt-get update && \
-    apt-get install -y --no-install-recommends nmap iputils-ping traceroute curl mariadb-client libmariadb-dev gcc && \
+    apt-get install -y --no-install-recommends nmap iputils-ping traceroute curl mariadb-client libmariadb-dev gcc whois && \
     apt-get autoremove -y && \
     apt-get clean all && \
     rm -rf /var/lib/apt/lists/*

netopsio/core/admin.py

@@ -1,8 +1,7 @@
 """Register core models in admin panel."""
 
-from django.contrib import admin
-
 from core.models import RequestLog
+from django.contrib import admin
 
 
 class RequestLogAdmin(admin.ModelAdmin):

netopsio/core/api/serializers.py

@@ -1,9 +1,8 @@
 """Core App and Worker Serializers."""
+from core import models
 from django_celery_results.models import TaskResult
 from rest_framework import serializers
 
-from core import models
-
 
 class TaskResultSerializer(serializers.HyperlinkedModelSerializer):
     """TaskResult Serializer."""

netopsio/core/api/urls.py

@@ -1,12 +1,12 @@
 """Netops.io core api urls."""
 
-from rest_framework.routers import DefaultRouter
-
 from core.api import views
+from rest_framework.routers import DefaultRouter
 
 router = DefaultRouter()
 router.register(r"tasks", views.TaskResultViewSet)
 router.register(r"request-logs", views.RequestLogViewSet, basename="requestlog")
 router.register(r"ping", views.PingViewSet, basename="ping")
 router.register(r"traceroute", views.TraceRouteViewSet, basename="traceroute")
 router.register(r"nmap", views.NmapViewSet, basename="nmap")
+router.register(r"whois", views.WhoisViewSet, basename="whois")

netopsio/core/api/views.py

@@ -1,13 +1,13 @@
 """Netops.io core api views."""
 
+from core import models, tasks
+from core.api import serializers
 from django_celery_results.models import TaskResult
 from drf_yasg import openapi
 from drf_yasg.utils import swagger_auto_schema
 from rest_framework import viewsets
 from rest_framework.response import Response
 
-from core import models, tasks
-from core.api import serializers
 from netopsio.utilities import get_ip_address
 
 
@@ -90,3 +90,24 @@ def create(self, request, host=None):
         data = models.TaskModel(task_id=task.task_id)
         serializer = serializers.TaskSerializer(data, context={"request": request})
         return Response(serializer.data)
+
+
+class WhoisViewSet(viewsets.ViewSet):
+    """Whois Viewset."""
+
+    @swagger_auto_schema(
+        request_body=openapi.Schema(
+            type=openapi.TYPE_OBJECT,
+            properties={
+                "host": openapi.Schema(
+                    type=openapi.TYPE_STRING, description="Whois a domain or host."
+                ),
+            },
+        )
+    )  # pylint: disable=no-self-use
+    def create(self, request, host):
+        """whois a host."""
+        task = tasks.whois.delay(host=host)
+        data = models.TaskModel(task_id=task.task_id)
+        serializer = serializers.TaskSerializer(data, context={"request": request})
+        return Response(serializer.data)

netopsio/core/forms.py

@@ -0,0 +1,9 @@
+"""Netops.io core app forms."""
+
+from django import forms
+
+
+class WhoisForm(forms.Form):
+    """Whois form."""
+
+    host = forms.CharField(label="domain or ip address", max_length=256)

netopsio/core/tasks.py

@@ -2,7 +2,6 @@
 import subprocess
 
 from celery import shared_task
-
 from core.models import RequestLog
 
 
@@ -33,3 +32,12 @@ def nmap(host: str) -> str:
     data = RequestLog(ip=host, result=task.stdout.decode("utf-8"), app="nmap")
     data.save()
     return task.stdout.decode("utf-8")
+
+
+@shared_task
+def whois(host: str) -> str:
+    """Whois host worker."""
+    task = subprocess.run(["whois", host], capture_output=True, check=True)
+    data = RequestLog(ip=host, result=task.stdout.decode("utf-8"), app="whois")
+    data.save()
+    return task.stdout.decode("utf-8")

netopsio/core/templates/core/task.html

@@ -9,7 +9,7 @@
         <tbody>
             <tr>
                 <td>{{ status }}</td>
-                <td><a href="/tasks/{{ id }}/">{{ host }}</a></td>
+                <td><a href="{% url 'tasks' %}{{ id }}/">{{ host }}</a></td>
             </tr>
         </tbody>
     </table>

netopsio/core/templates/core/tasks.html

@@ -11,7 +11,7 @@
         <tbody>
         {% for task in results %}
             <tr>
-                <th scope="row"><a href="/tasks/{{ task.task_id }}">{{ task.task_id }}</a></th>
+                <th scope="row"><a href="{% url 'tasks' %}{{ task.task_id }}">{{ task.task_id }}</a></th>
                 <td>{{ task.task_name }}</td>
                 <td>{{ task.status }}</td>
                 <td>{{ task.task_kwargs }}</td>

netopsio/core/templates/core/whois.html

@@ -0,0 +1,9 @@
+{% extends "base.html" %}
+{% block content %}
+{% load crispy_forms_tags %}
+<form action="{% url 'whois' %}" method="post">
+    {% csrf_token %}
+    {{ form|crispy }}
+    <input type="submit" value="Submit" class="btn btn-primary">
+</form>
+{% endblock %}