添加@app.route('/users/profile', methods=['POST'])

chatgpt-next-web/service.py

@@ -126,7 +126,8 @@ def login():
         'code': 0,
         'message': 'Login successful',
         'data': {
-            'token': token
+            'token': token,
+            'email': eval(user_data)['email']
         }
     })
 
@@ -323,6 +324,35 @@ def handle_payment_notification():
     return jsonify({'code': 0, 'message': 'Notification received'})
 
 
+@app.route('/users/profile', methods=['POST'])
+def get_user_profile():
+    token = parse_token(request)
+
+    # 验证令牌
+    if not validate_token(token):
+        return jsonify({'code': 401, 'message': 'Invalid token'})
+
+    # 获取用户ID
+    user_id = get_user_id_from_token(token)
+    if not user_id:
+        return jsonify({'code': 400, 'message': 'User not found'})
+
+    free_count = get_free_count(user_id)
+    package = get_user_package(user_id)
+    basic_chat_limit = int(package.get(b'basic_chat_limit', 0))
+    advanced_chat_limit = int(package.get(b'advanced_chat_limit', 0))
+
+    return jsonify({
+        'code': 0,
+        'message': 'Success',
+        'data': {
+            'free_count': free_count,
+            'basic_chat_limit': basic_chat_limit,
+            'advanced_chat_limit': advanced_chat_limit
+        }
+    })
+
+
 def get_user_orders(user_id):
     # 从Redis中获取该用户的所有订单数据
     user_orders = []