w
This commit is contained in:
+22
-14
@@ -12,8 +12,8 @@ import json
|
||||
|
||||
from mcp.server.fastmcp import Context
|
||||
|
||||
from owa_mcp.server import mcp, AppContext
|
||||
from owa_mcp.owa_client import OWAClient, SessionExpiredError
|
||||
from owa_mcp.server import AppContext, mcp
|
||||
from owa_mcp.server_lifecycle import require_client
|
||||
|
||||
|
||||
@@ -87,16 +87,20 @@ async def login(
|
||||
# Parse and validate cookies
|
||||
cookies_str = cookies.strip()
|
||||
if not cookies_str:
|
||||
return json.dumps({
|
||||
"success": False,
|
||||
"error": "Cookies string is empty. Paste your browser cookies as name=value pairs, one per line.",
|
||||
})
|
||||
return json.dumps(
|
||||
{
|
||||
"success": False,
|
||||
"error": "Cookies string is empty. Paste your browser cookies as name=value pairs, one per line.",
|
||||
}
|
||||
)
|
||||
|
||||
# Save to disk
|
||||
try:
|
||||
client.cookie_file.write_text(cookies_str + "\n")
|
||||
except Exception as e:
|
||||
return json.dumps({"success": False, "error": f"Failed to write cookie file: {e}"})
|
||||
return json.dumps(
|
||||
{"success": False, "error": f"Failed to write cookie file: {e}"}
|
||||
)
|
||||
|
||||
# Load into memory
|
||||
try:
|
||||
@@ -106,12 +110,16 @@ async def login(
|
||||
|
||||
# Verify session
|
||||
if _session_is_valid(client):
|
||||
return json.dumps({
|
||||
"success": True,
|
||||
"message": f"Logged in. Session cookies saved to {client.cookie_file}.",
|
||||
})
|
||||
return json.dumps(
|
||||
{
|
||||
"success": True,
|
||||
"message": f"Logged in. Session cookies saved to {client.cookie_file}.",
|
||||
}
|
||||
)
|
||||
else:
|
||||
return json.dumps({
|
||||
"success": False,
|
||||
"error": "Cookies loaded but session verification failed. The cookies may have expired.",
|
||||
})
|
||||
return json.dumps(
|
||||
{
|
||||
"success": False,
|
||||
"error": "Cookies loaded but session verification failed. The cookies may have expired.",
|
||||
}
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user