(feat) add delay option for /submit* endpoints

sensepost · Sep 22, 2024 · 1db81b3 · 1db81b3
1 parent a9e254d
commit 1db81b3
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 6 deletions.
diff --git a/web/api/submit.go b/web/api/submit.go
@@ -21,6 +21,7 @@ type submitRequestOptions struct {
 	Y         int    `json:"window_y"`
 	UserAgent string `json:"user_agent"`
 	Timeout   int    `json:"timeout"`
+	Delay     int    `json:"delay"`
 	Format    string `json:"format"`
 }
 
@@ -64,6 +65,9 @@ func (h *ApiHandler) SubmitHandler(w http.ResponseWriter, r *http.Request) {
 		if request.Options.Timeout != 0 {
 			options.Scan.Timeout = request.Options.Timeout
 		}
+		if request.Options.Delay != 0 {
+			options.Scan.Delay = request.Options.Delay
+		}
 		if request.Options.Format != "" {
 			options.Scan.ScreenshotFormat = request.Options.Format
 		}

diff --git a/web/docs/docs.go b/web/docs/docs.go
@@ -328,9 +328,9 @@ const docTemplate = `{
                 ],
                 "responses": {
                     "200": {
-                        "description": "Probing started",
+                        "description": "The URL Result object",
                         "schema": {
-                            "type": "string"
+                            "$ref": "#/definitions/models.Result"
                         }
                     }
                 }
@@ -566,6 +566,9 @@ const docTemplate = `{
         "api.submitRequestOptions": {
             "type": "object",
             "properties": {
+                "delay": {
+                    "type": "integer"
+                },
                 "format": {
                     "type": "string"
                 },

diff --git a/web/docs/swagger.json b/web/docs/swagger.json
@@ -317,9 +317,9 @@
                 ],
                 "responses": {
                     "200": {
-                        "description": "Probing started",
+                        "description": "The URL Result object",
                         "schema": {
-                            "type": "string"
+                            "$ref": "#/definitions/models.Result"
                         }
                     }
                 }
@@ -555,6 +555,9 @@
         "api.submitRequestOptions": {
             "type": "object",
             "properties": {
+                "delay": {
+                    "type": "integer"
+                },
                 "format": {
                     "type": "string"
                 },

diff --git a/web/docs/swagger.yaml b/web/docs/swagger.yaml
@@ -131,6 +131,8 @@ definitions:
     type: object
   api.submitRequestOptions:
     properties:
+      delay:
+        type: integer
       format:
         type: string
       timeout:
@@ -560,9 +562,9 @@ paths:
       - application/json
       responses:
         "200":
-          description: Probing started
+          description: The URL Result object
           schema:
-            type: string
+            $ref: '#/definitions/models.Result'
       summary: Submit a single URL for probing
       tags:
       - Results