Merge pull request #5 from hadriansecurity/olivier/docker

Add Docker

Dockerfile

@@ -0,0 +1,17 @@
+# usage:
+# change example_input.txt to your input file on line 14
+# docker build -t subwiz .
+# docker run subwiz (optionally add any subwiz arguments)
+
+FROM python:3.12-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY subwiz subwiz
+COPY example_input.txt input_domains.txt
+
+ENTRYPOINT ["python", "-m", "subwiz.cli", "-i", "/app/input_domains.txt"]
+CMD []

pyproject.toml

@@ -4,10 +4,10 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "subwiz"
-version = "0.1.4"
+version = "0.1.5"
 description = "A recon tool that uses AI to predict subdomains. Then returns those that resolve."
 readme = "README.md"
-requires-python = ">=3.9"
+requires-python = ">=3.9, <3.13"
 authors = [{name = "Klaas Meinke", email = "[email protected]"}]
 keywords = ["machine learning", "recon", "subdomains", "transformers"]
 license = {text = "MIT License"}

subwiz.py

@@ -0,0 +1,6 @@
+import subwiz
+
+known_subdomains = ['test1.example.com', 'test2.example.com']
+new_subdomains = subwiz.run(input_domains=known_subdomains, no_resolve=True)
+
+print(new_subdomains)

subwiz/cli.py

@@ -5,7 +5,7 @@
     run_inference,
     run_resolution,
 )
-from subwiz.types import (
+from subwiz.type import (
     device_type,
     input_domains_file_type,
     output_file_type,

subwiz/main.py

@@ -11,7 +11,7 @@
 
 from subwiz.model import GPT
 from subwiz.resolve import is_registered_bulk
-from subwiz.types import (
+from subwiz.type import (
     Domain,
     input_domains_type,
     device_type,

subwiz/model.py

@@ -1,5 +1,7 @@
 """
 This code is largely copied from the nanogpt repository by Andrej Karpathy.
+The generate method has been modified to output many sequences.
+
 https://github.com/karpathy/nanoGPT
 
 Full definition of a GPT Language Model, all of it in this single file.

tests/test_domain_inputs.py

@@ -1,7 +1,7 @@
 from argparse import ArgumentTypeError
 import os
 
-from subwiz.types import input_domains_file_type
+from subwiz.type import input_domains_file_type
 
 
 def test_file_doesnt_exist():