mirror of
https://github.com/j93es/browser-use-oauth.git
synced 2026-06-04 08:11:53 +09:00
[Refactor] 리팩터링
This commit is contained in:
parent
26b40e0e65
commit
bbd2d6d636
20 changed files with 397 additions and 257 deletions
3
lib/agents/get_sso_list/__init__.py
Normal file
3
lib/agents/get_sso_list/__init__.py
Normal file
|
|
@ -0,0 +1,3 @@
|
|||
from lib.agents.get_sso_list.get_sso_list import get_sso_list
|
||||
|
||||
__all__ = ["get_sso_list"]
|
||||
22
lib/agents/get_sso_list/get_sso_list.py
Normal file
22
lib/agents/get_sso_list/get_sso_list.py
Normal file
|
|
@ -0,0 +1,22 @@
|
|||
from lib.agents.get_sso_list.prompt import get_sso_list_task, FindLoginPageResponse
|
||||
from lib.browser_use_utils.run_task import run_task
|
||||
|
||||
|
||||
NOT_FOUND_LOGIN_PAGE = 0
|
||||
FOUND_LOGIN_PAGE = 1
|
||||
|
||||
async def get_sso_list(target_url) -> tuple[bool, str | FindLoginPageResponse | None]:
|
||||
|
||||
task = get_sso_list_task
|
||||
ReturnModel = FindLoginPageResponse
|
||||
success, response = await run_task(target_url, ReturnModel, task)
|
||||
if not success:
|
||||
return False, response
|
||||
if isinstance(response, str):
|
||||
return False, response
|
||||
|
||||
return True, response
|
||||
|
||||
|
||||
|
||||
|
||||
68
lib/agents/get_sso_list/prompt.py
Normal file
68
lib/agents/get_sso_list/prompt.py
Normal file
|
|
@ -0,0 +1,68 @@
|
|||
from pydantic import BaseModel
|
||||
|
||||
class FindLoginPageResponse(BaseModel):
|
||||
msg: str | None = None
|
||||
url: str | None = None
|
||||
sso_list: list[str] = [] # List of SSO providers found on the login page
|
||||
|
||||
get_sso_list_task = """
|
||||
You are an expert in finding login pages.
|
||||
|
||||
Your task is to navigate to the login page of the given URL. Follow the steps below strictly and return results only in the specified format.
|
||||
|
||||
※ You are NOT allowed to navigate to URLs that are not directly discoverable within the initial domain. Do NOT use search engines or guess external login URLs.
|
||||
|
||||
0. INITIAL BLOCK CHECK
|
||||
- If the browser is blocked when trying to access the page — due to firewall, CAPTCHA, regional restrictions, or other access denials — immediately terminate the process and return the following JSON:
|
||||
```json
|
||||
{
|
||||
"msg": "Blocked",
|
||||
"url": "",
|
||||
"sso_list": []
|
||||
}
|
||||
```
|
||||
- Do NOT proceed to further steps in this case.
|
||||
|
||||
1. LOGIN PAGE NAVIGATION
|
||||
- Navigate only to a **client-side (non-enterprise)** login page within the provided domain.
|
||||
- Do NOT rely on external tools, search engines, or links not directly found on the site.
|
||||
- If a consent popup (e.g. for privacy/cookies) appears, you MUST dismiss or close it before proceeding.
|
||||
- Since step 0 confirmed access, assume the page now loads properly.
|
||||
|
||||
2. SSO BUTTON IDENTIFICATION
|
||||
- On the login page, look for the following social login (SSO) buttons:
|
||||
- Google, GitHub, Facebook, LinkedIn, Microsoft, Naver, Slack, Etc.
|
||||
- ✅ Proceed only if it is clearly an **actual SSO button**.
|
||||
- ❌ Exclude the following:
|
||||
- Passkey-related buttons
|
||||
- Username/password fields
|
||||
- Email-based login
|
||||
- Non-OAuth methods such as certificate or phone verification
|
||||
|
||||
3. RETURN FORMAT
|
||||
- If the login page is successfully found, return:
|
||||
```json
|
||||
{
|
||||
"msg": "Login page found",
|
||||
"url": "https://example.com/login",
|
||||
"sso_list": ["Google", "GitHub"]
|
||||
}
|
||||
```
|
||||
- If the login page cannot be found, return:
|
||||
```json
|
||||
{
|
||||
"msg": "Login page not found",
|
||||
"url": "",
|
||||
"sso_list": []
|
||||
}
|
||||
```
|
||||
- If blocked (as in step 0), return:
|
||||
```json
|
||||
{
|
||||
"msg": "Blocked",
|
||||
"url": "",
|
||||
"sso_list": []
|
||||
}
|
||||
```
|
||||
- Return ONLY the JSON object. Do NOT include any explanation, logging, or extra output.
|
||||
"""
|
||||
Loading…
Add table
Add a link
Reference in a new issue