"""
MCP Lab - Main Entry Point
A unified entry point for all MCP servers in the crash course.
"""
import sys
import argparse
from mcp_crash_course.src.servers.daily_news import mcp as daily_news_mcp
from mcp_crash_course.src.servers.stock_news import mcp as stock_news_mcp
from mcp_crash_course.src.servers.wikipedia_summary import mcp as wikipedia_mcp


def main():
    parser = argparse.ArgumentParser(
        description="MCP Lab - Run MCP servers from the crash course"
    )
    parser.add_argument(
        "server",
        choices=["daily-news", "stock-news", "wikipedia"],
        help="Which MCP server to run"
    )

    args = parser.parse_args()

    if args.server == "daily-news":
        print("Starting daily-news MCP server...")
        daily_news_mcp.run()
    elif args.server == "stock-news":
        print("Starting stock-news MCP server...")
        stock_news_mcp.run()
    elif args.server == "wikipedia":
        print("Starting wikipedia-summary MCP server...")
        wikipedia_mcp.run()


if __name__ == "__main__":
    main()