From 1e827a903bc40e5836591690710fdd8d57ff4c8b Mon Sep 17 00:00:00 2001
From: Br1an67 <932039080@qq.com>
Date: Mon, 2 Mar 2026 00:16:28 +0800
Subject: [PATCH] feat: add support for select, textarea, and file upload in
 web forms

Extend form handling to support:
- <select> dropdowns: detect options, select by visible text or value
- <textarea> fields: fill with text content
- <input type="file">: upload files by absolute path
- All element types are also discovered inside shadow DOM

Changes:
- find_inputs.js: discover <select> (with options) and <textarea>
- browser.py: import Select, handle new element types in
  get_form_inputs() and fill_form_inputs()
---
 sources/browser.py                 | 37 +++++++++++++++++++++++++++---
 sources/web_scripts/find_inputs.js | 31 +++++++++++++++++++++++++
 2 files changed, 65 insertions(+), 3 deletions(-)

diff --git a/sources/browser.py b/sources/browser.py
index 92d7244..cd654de 100644
--- a/sources/browser.py
+++ b/sources/browser.py
@@ -2,7 +2,7 @@ from selenium import webdriver
 from selenium.webdriver.chrome.service import Service
 from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.common.by import By
-from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support.ui import WebDriverWait, Select
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import TimeoutException, WebDriverException
 from selenium.webdriver.common.action_chains import ActionChains
@@ -498,7 +498,16 @@ class Browser:
                 if input_type in ["hidden", "submit", "button", "image"] or not element["displayed"]:
                     continue
                 input_name = element.get("text") or element.get("id") or input_type
-                if input_type == "checkbox" or input_type == "radio":
+                if input_type == "select":
+                    options = element.get("options", [])
+                    options_str = ", ".join(opt["text"] for opt in options if opt["text"])
+                    selected = next((opt["text"] for opt in options if opt.get("selected")), "")
+                    form_strings.append(f"[{input_name}](select: {selected}) options: [{options_str}]")
+                elif input_type == "textarea":
+                    form_strings.append(f"[{input_name}]("")")
+                elif input_type == "file":
+                    form_strings.append(f"[{input_name}](file: )")
+                elif input_type == "checkbox" or input_type == "radio":
                     try:
                         checked_status = "checked" if element.is_selected() else "unchecked"
                     except Exception as e:
@@ -662,7 +671,29 @@ class Browser:
                     self.logger.warning(f"Element '{name}' is not interactable (not displayed or disabled)")
                     continue
                 input_type = (element.get_attribute("type") or "text").lower()
-                if input_type in ["checkbox", "radio"]:
+                tag_name = (element.tag_name or "").lower()
+                if tag_name == "select":
+                    try:
+                        select = Select(element)
+                        select.select_by_visible_text(value)
+                        self.logger.info(f"Selected '{value}' for {name}")
+                    except Exception:
+                        try:
+                            select.select_by_value(value)
+                            self.logger.info(f"Selected value '{value}' for {name}")
+                        except Exception as sel_e:
+                            self.logger.warning(f"Could not select '{value}' for {name}: {sel_e}")
+                elif tag_name == "textarea":
+                    element.clear()
+                    element.send_keys(value)
+                    self.logger.info(f"Filled textarea {name}")
+                elif input_type == "file":
+                    if os.path.isabs(value) and os.path.exists(value):
+                        element.send_keys(value)
+                        self.logger.info(f"Uploaded file '{value}' for {name}")
+                    else:
+                        self.logger.warning(f"File not found: {value}")
+                elif input_type in ["checkbox", "radio"]:
                     is_checked = element.is_selected()
                     should_be_checked = value.lower() == "checked"
 
diff --git a/sources/web_scripts/find_inputs.js b/sources/web_scripts/find_inputs.js
index 80c4b14..acd87fc 100644
--- a/sources/web_scripts/find_inputs.js
+++ b/sources/web_scripts/find_inputs.js
@@ -11,6 +11,37 @@ function findInputs(element, result = []) {
             displayed: isElementDisplayed(input)
         });
     });
+    // Find all <select> elements (dropdowns / multi-choice)
+    const selects = element.querySelectorAll('select');
+    selects.forEach(select => {
+        const options = Array.from(select.options).map(opt => ({
+            value: opt.value,
+            text: opt.textContent.trim(),
+            selected: opt.selected
+        }));
+        result.push({
+            tagName: select.tagName,
+            text: select.name || '',
+            type: 'select',
+            class: select.className || '',
+            xpath: getXPath(select),
+            displayed: isElementDisplayed(select),
+            multiple: select.multiple,
+            options: options
+        });
+    });
+    // Find all <textarea> elements
+    const textareas = element.querySelectorAll('textarea');
+    textareas.forEach(textarea => {
+        result.push({
+            tagName: textarea.tagName,
+            text: textarea.name || '',
+            type: 'textarea',
+            class: textarea.className || '',
+            xpath: getXPath(textarea),
+            displayed: isElementDisplayed(textarea)
+        });
+    });
     const allElements = element.querySelectorAll('*');
     allElements.forEach(el => {
         if (el.shadowRoot) {