osint-dev-team · manmolecular · Jul 29, 2020 · Jul 19, 2020 · Jul 20, 2020 · Jul 20, 2020
diff --git a/src/scripts/recon/get_title/__init__.py b/src/scripts/recon/get_title/__init__.py
@@ -0,0 +1,5 @@
+import sys
+from pathlib import Path
+
+__root_dir = Path(__file__).parents[4]
+sys.path.append(str(__root_dir))
diff --git a/src/scripts/recon/get_title/__main__.py b/src/scripts/recon/get_title/__main__.py
@@ -0,0 +1,12 @@
+#!/usr/bin/env python3
+
+from pprint import pprint
+from sys import argv
+
+from src.core.utils.module import run_module
+from .module import Runner
+
+result = run_module(
+    Runner, args=argv, arg_name="url", arg_default="https://www.vk.com/"
+)
+pprint(result)
diff --git a/src/scripts/recon/get_title/module.py b/src/scripts/recon/get_title/module.py
@@ -0,0 +1,34 @@
+#!/usr/bin/env python3
+
+import requests
+
+from re import search
+from src.core.base.recon import ReconRunner
+from src.core.utils.response import ScriptResponse
+
+
+class Runner(ReconRunner):
+    def __init__(self, logger: str = __name__):
+        super(Runner, self).__init__(logger)
+
+    def run(self, *args, **kwargs):
+        """
+        Try to find http title
+        :param args: args from core runner
+        :param kwargs: kwargs from core runner
+        :return: http title if it's exist
+        """
+        url = kwargs.get("url")
+        try:
+            response = requests.get(url, verify=False).text
+        except Exception as get_err:
+            return ScriptResponse.error(result=None, message=str(get_err))
+
+        search_title = search(r"<title>(.*)</title>", response)
+        result = search_title.group(1) if search_title else None
+        return ScriptResponse.success(
+            result=result,
+            message=f"Successfully found the title for {url}"
+            if result
+            else "Can not get the title or bad url",
+        )
diff --git a/src/scripts/recon/get_title/requirements.txt b/src/scripts/recon/get_title/requirements.txt
@@ -0,0 +1,5 @@
+certifi==2020.6.20
+chardet==3.0.4
+idna==2.10
+requests==2.24.0
+urllib3==1.25.9
diff --git a/src/scripts/recon/get_title/test_module.py b/src/scripts/recon/get_title/test_module.py
@@ -0,0 +1,96 @@
+#!/usr/bin/env python3
+
+from http.server import HTTPServer, BaseHTTPRequestHandler
+from threading import Thread
+from unittest import TestCase
+
+from src.core.values.defaults import TestDefaults
+from .module import Runner
+
+DefaultValues = TestDefaults()
+
+
+class TestClassHTTPRequestHandler(BaseHTTPRequestHandler):
+    """
+    Defines mocking server class
+    """
+
+    def do_GET(self) -> None:
+        """
+        Defines basic handlers
+        :return: None
+        """
+        if self.path == "/":
+            self.mock_endpoint()
+
+    def mock_endpoint(self) -> None:
+        """
+        Mock '/' endpoint. Return HTML with custom headers.
+        :return: None
+        """
+        self.send_response(200)
+        self.send_header("Content-type", "text/html; charset=utf-8")
+        self.end_headers()
+        self.wfile.write(
+            b"<html><body><title>Mock page from Mocking Class</title></body></html>"
+        )
+
+
+class GetTitleTest(TestCase):
+
+    server: Thread or None = None
+
+    def setUp(self):
+        """
+        Setup something before each test function
+        :return: None
+        """
+        self.runner = Runner()
+
+    @classmethod
+    def setUpClass(cls) -> None:
+        """
+        Start up server on class initialization in daemon thread
+        :return: None
+        """
+        super(GetTitleTest, cls).setUpClass()
+        cls.server = HTTPServer(
+            server_address=(DefaultValues.HOST, DefaultValues.PORT),
+            RequestHandlerClass=TestClassHTTPRequestHandler,
+        )
+        server_daemon = Thread(target=cls.server.serve_forever, daemon=True)
+        server_daemon.start()
+
+    @classmethod
+    def tearDownClass(cls) -> None:
+        super(GetTitleTest, cls).tearDownClass()
+        cls.server.shutdown()
+
+    def test_create_runner(self) -> None:
+        """
+        Test creation of the class instance
+        :return: None
+        """
+        self.assertIsNotNone(self.runner)
+        self.assertIsInstance(self.runner, Runner)
+
+    def test_local_server(self) -> None:
+        """
+        Test values from the local server
+        :return: None
+        """
+        response = self.runner.run(
+            url=f"http://{DefaultValues.HOST}:{DefaultValues.PORT}"
+        )
+        self.assertEqual(response.get("status"), "success")
+        self.assertEqual(response.get("result"), "Mock page from Mocking Class")
+
+    def test_remote_server(self) -> None:
+        """
+        Test values from the remote server
+        :return: None
+        """
+        response = self.runner.run(url=f"https://dsakdjsalkjdkasjdlksadjlaks.co")
+        self.assertEqual(response.get("status"), "error")
+        response = self.runner.run(url=f"https://facebook.com")
+        self.assertEqual(response.get("status"), "success")