# Scenario 1: Simple query to Docs MCP
# This script demonstrates how to connect to the Microsoft Learn Docs MCP server,
# send a query, and print the result.

import asyncio
import logging
import sys
from mcp.client.streamable_http import streamablehttp_client
from mcp import ClientSession

import json

# Microsoft Docs MCP Server Endpoint
MCP_SERVER_URL = "https://learn.microsoft.com/api/mcp"

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    datefmt='%Y-%m-%d %H:%M:%S'
)
logger = logging.getLogger('mcp_client')

def prompt_user():
    """Prompt user for a search query."""
    print("Type your Microsoft Docs search query (or 'exit' to quit):")
    try:
        return input("> ").strip()
    except (KeyboardInterrupt, EOFError):
        print("\nDetected exit signal.")
        return "exit"

async def main():
    logger.info("Connecting to Microsoft Docs MCP Server at: %s", MCP_SERVER_URL)
    try:
        async with streamablehttp_client(MCP_SERVER_URL) as (read_stream, write_stream, _):
            logger.info("Connection established. Initializing MCP session.")
            async with ClientSession(read_stream, write_stream) as session:
                await session.initialize()
                logger.info("Session initialized successfully.")
                print("\n=== Microsoft Docs MCP Client ===")
                print("Tool: microsoft_docs_search")
                print("Enter documentation queries to search Microsoft Learn docs.")
                print("Type 'exit' or 'quit' to end the session.\n")
                
                while True:
                    user_query = prompt_user()
                    if not user_query:
                        print("Query cannot be empty. Please try again.")
                        continue
                    if user_query.lower() in ("exit", "quit"):
                        print("Exiting client. Goodbye!")
                        break
                    
                    try:
                        logger.info("Executing query: %s", user_query)
                        result = await session.call_tool("microsoft_docs_search", {"question": user_query})
                        # print("RESULT:", result)
                        print("\n--- Search Result ---")
                        if hasattr(result, 'content'):
                            # print(result.content)
                            for item in result.content:
                                my_list = json.loads(item.text)
                                for doc in my_list:
                                    print(f"[Title]: {doc.get('title', 'No title')}")
                                    print(f"[Content]: {doc.get('content', 'No content')}")
                                    print("---")
                                    
                        else:
                            print("No content returned from the search.")
                        print("---------------------\n")
                    except Exception as e:
                        logger.error("Query failed: %s", e)
                        print(f"Error: {e}. Please try a different query or check your connection.\n")
                        
    except Exception as e:
        logger.error("Connection error: %s", e)
        print(f"Failed to connect to Microsoft Docs MCP Server: {e}")
        print("Please check your internet connection and try again.")
        sys.exit(1)

if __name__ == "__main__":
    print("Starting Microsoft Docs MCP Client...")
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print("\nApplication terminated by user.")
        sys.exit(0)