update

.gitignore

@@ -1 +1,2 @@
-**/__pycache__/
+**/__pycache__/
+.env

airs/base.py

@@ -0,0 +1,112 @@
+import os
+from supabase import create_client, Client
+
+# 创建 Base 类，Base类主要实现应用的用户管理，数据库连接等功能
+class Base:
+    def __init__(self):
+        supabase_url = os.getenv("SUPABASE_URL")
+        supabase_key = os.getenv("SUPABASE_KEY")
+
+        if not supabase_url or not supabase_key:
+            raise ValueError("SUPABASE_URL and SUPABASE_KEY environment variables must be set.")
+
+        self.supabase: Client = create_client(supabase_url, supabase_key)
+
+    def sign_up_with_email(self, email, password):
+        """
+        使用邮箱和密码注册新用户
+
+        Args:
+            email: 用户邮箱
+            password: 用户密码
+
+        Returns:
+            成功返回用户数据，失败返回错误信息
+        """
+        try:
+            # 发送注册请求
+            response = self.supabase.auth.sign_up({
+                "email": email,
+                "password": password
+            })
+
+            # 检查是否有错误
+            if response.user is None: # Supabase Python SDK 在错误时 user 为 None
+                return {"success": False, "error": response.session.user.identities[0].identity_data['error']} # 尝试获取更具体的错误信息
+            
+            # 注册成功
+            return {
+                "success": True,
+                "user": response.user.model_dump(), # 将 UserModel 转换为字典
+                "message": "注册成功！请检查邮箱完成验证。"
+            }
+        except Exception as e:
+            return {"success": False, "error": str(e)}
+
+    def sign_in_with_email(self, email, password):
+        """
+        使用邮箱和密码登录
+
+        Args:
+            email: 用户邮箱
+            password: 用户密码
+
+        Returns:
+            成功返回用户数据和会话信息，失败返回错误信息
+        """
+        try:
+            # 发送登录请求
+            response = self.supabase.auth.sign_in_with_password({
+                "email": email,
+                "password": password
+            })
+
+            # 检查是否有错误
+            if response.user is None: # Supabase Python SDK 在错误时 user 为 None
+                return {"success": False, "error": response.session.user.identities[0].identity_data['error']} # 尝试获取更具体的错误信息
+
+            # 登录成功，保存会话信息
+            return {
+                "success": True,
+                "user": response.user.model_dump(), # 将 UserModel 转换为字典
+                "session": response.session.model_dump(), # 将 SessionModel 转换为字典
+                "message": "登录成功！"
+            }
+        except Exception as e:
+            return {"success": False, "error": str(e)}
+
+    def get_current_user(self):
+        """
+        获取当前登录用户
+
+        Returns:
+            用户信息或 None
+        """
+        try:
+            # 获取当前会话
+            response = self.supabase.auth.get_session()
+
+            if response.session:
+                return response.user.model_dump() # 将 UserModel 转换为字典
+            else:
+                return None
+        except Exception as e:
+            print(f"获取用户信息出错: {str(e)}")
+            return None
+
+    def sign_out(self):
+        """
+        退出登录
+
+        Returns:
+            成功返回 True，失败返回 False
+        """
+        try:
+            response = self.supabase.auth.sign_out()
+            # Supabase Python SDK 的 sign_out 不直接返回 error 属性，
+            # 成功时通常返回 None 或一个空对象
+            # 如果有异常会直接抛出
+            return True
+        except Exception as e:
+            print(f"退出登录出错: {str(e)}")
+            return False

app.py

@@ -1,11 +1,17 @@
 from fastapi import FastAPI
+from dotenv import load_dotenv # 导入 load_dotenv
+
+# 加载 .env 文件中的环境变量
+load_dotenv()
 
 from apps.app01 import App01
 from apps.app02 import App02
+from apps.airs_tasks import Tasks
 
 # 初始化应用
 app01_instance = App01()
 app02_instance = App02()
+tasks_instance = Tasks()
 
 # 创建 FastAPI 应用
 app = FastAPI()
@@ -14,6 +20,7 @@ app = FastAPI()
 # 使用 Mount 来指定路径前缀
 app.mount("/app01", app=app01_instance.mcp.sse_app())
 app.mount("/app02", app=app02_instance.mcp.sse_app())
+app.mount("/tasks", app=tasks_instance.mcp.sse_app())
 
 # 根路径，用于测试 FastAPI 是否正常工作
 @app.get("/")

apps/airs_tasks.py

@@ -0,0 +1,40 @@
+import asyncio
+from fastapi import Request
+from mcp.server.fastmcp import FastMCP
+import os # 用于获取环境变量
+from airs.base import Base # 导入 Base 类
+
+# 创建 MCP 应用实例
+class Tasks(Base): # Tasks 类继承自 Base 类
+    def __init__(self):
+        super().__init__() # 初始化 Base 类
+        self.mcp = FastMCP(name="tasks", stateless_http=True)
+        self.register_endpoints()
+        
+        # 登录测试
+        
+        test_user = os.getenv("TEST_USER")
+        test_passsword = os.getenv("TEST_PASSWORD")
+        print("尝试登录...")
+        login_result = self.sign_in_with_email(test_user, test_passsword)
+        if login_result["success"]:
+            # print("登录成功！用户信息:", login_result["user"])
+            print("登录成功！")
+        else:
+            print("登录失败！错误信息:", login_result["error"])
+
+    def register_endpoints(self):
+        @self.mcp.resource("tasks://info")
+        def info() -> str:
+            """Information about Tasks"""
+            return "This is Airs Tasks"
+
+        @self.mcp.tool()
+        def echo(message: str) -> str:
+            """Echo a message in Tasks"""
+            return f"Tasks Echo: {message}"
+
+        @self.mcp.tool()
+        def app01_add_integers(num1: int, num2: int) -> int:
+            """Add two integers in App01"""
+            return num1 + num2

requirements.txt

@@ -2,3 +2,5 @@ fastapi
 uvicorn
 fastmcp
 sse_starlette
+supabase
+python-dotenv