#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
FLUX风格塑形MCP服务器启动脚本
--------------------------
启动带有FLUX风格塑形功能的MCP服务器
"""

import logging
import argparse
import sys
import asyncio

# 设置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)

# 导入MCP服务器
from mcp_server import MCPServer
from flux_style_resource import register_flux_resources

async def main_async():
    """异步主函数"""
    # 解析命令行参数
    parser = argparse.ArgumentParser(description='启动带有FLUX风格塑形功能的MCP服务器')
    parser.add_argument('--host', type=str, default='127.0.0.1', help='服务器监听地址')
    parser.add_argument('--port', type=int, default=8189, help='服务器监听端口')
    parser.add_argument('--name', type=str, default='FLUX-MCP', help='服务器名称')
    args = parser.parse_args()
    
    try:
        # 创建MCP服务器实例
        logger.info(f"正在初始化MCP服务器: {args.name}...")
        server = MCPServer(server_name=args.name)
        
        # 注册FLUX风格塑形资源和工具
        logger.info("正在注册FLUX风格塑形资源...")
        register_flux_resources(server)
        
        # 启动服务器
        logger.info(f"正在启动MCP服务器，监听地址: {args.host}:{args.port}")
        await server.start(host=args.host, port=args.port)
    except KeyboardInterrupt:
        logger.info("收到中断信号，服务器关闭")
    except Exception as e:
        logger.error(f"服务器启动失败: {e}")
        return 1
    
    return 0

def main():
    """主函数"""
    try:
        return asyncio.run(main_async())
    except KeyboardInterrupt:
        logger.info("收到中断信号，服务器关闭")
        return 0
    except Exception as e:
        logger.error(f"服务器运行出错: {e}")
        return 1

if __name__ == "__main__":
    sys.exit(main())