Добавление прокси-сервера для отладки и перехвата запросов UI-API

Dockerfile

@@ -22,6 +22,9 @@ RUN apt-get clean && apt-get update && apt-get install -y --no-install-recommend
     ca-certificates \
     && apt-get clean && rm -rf /var/lib/apt/lists/* && rm -rf /tmp/*
 
+# Установка Python-пакетов
+RUN pip3 install requests
+
 # Установка Go 1.21
 RUN wget https://golang.org/dl/go1.21.0.linux-amd64.tar.gz && \
     tar -C /usr/local -xzf go1.21.0.linux-amd64.tar.gz && \
@@ -189,7 +192,7 @@ RUN chmod +x /app/app.py && \
 USER tenuser
 
 # Открываем порты
-EXPOSE 7860 8080 3000
+EXPOSE 7860 8080 3000 9090
 
 # Запускаем API сервер и Playground в dev-режиме
 ENTRYPOINT ["python3", "/app/app.py"]

app.py

@@ -12,6 +12,11 @@ import logging
 import urllib.request
 import urllib.error
 import tempfile
+import http.server
+import socketserver
+import requests
+from threading import Thread
+from http import HTTPStatus
 
 # Настройка логирования
 logging.basicConfig(level=logging.INFO, 
@@ -30,6 +35,133 @@ API_BINARY = Path("/app/server/bin/api")
 PLAYGROUND_DIR = Path("/app/playground")
 BACKUP_DIR = Path("/app/backup")
 
+class ProxyHTTPRequestHandler(http.server.BaseHTTPRequestHandler):
+    """HTTP-прокси для отладки и исправления запросов между UI и API"""
+    
+    def __init__(self, *args, **kwargs):
+        # Целевой API сервер
+        self.api_host = "localhost"
+        self.api_port = 8080
+        super().__init__(*args, **kwargs)
+    
+    def do_GET(self):
+        """Обработка GET запросов"""
+        logger.info(f"PROXY: GET запрос: {self.path}")
+        
+        # Специальная обработка для запроса графов
+        if self.path == "/graphs" or self.path == "/api/agents/graphs":
+            self._handle_graphs_request()
+            return
+        
+        # Проксирование запроса к API
+        target_url = f"http://{self.api_host}:{self.api_port}{self.path}"
+        try:
+            response = requests.get(target_url, headers=self._get_headers())
+            self._send_response(response)
+        except Exception as e:
+            logger.error(f"PROXY: Ошибка при проксировании GET-запроса: {e}")
+            self.send_error(HTTPStatus.INTERNAL_SERVER_ERROR, str(e))
+    
+    def do_POST(self):
+        """Обработка POST запросов"""
+        content_length = int(self.headers.get('Content-Length', 0))
+        post_data = self.rfile.read(content_length) if content_length > 0 else b''
+        
+        logger.info(f"PROXY: POST запрос: {self.path}")
+        
+        # Специальные запросы для дизайнера
+        if "/api/designer/v1/packages/reload" in self.path or "/api/dev/v1/packages/reload" in self.path:
+            self._handle_designer_reload()
+            return
+        
+        # Проксирование запроса к API
+        target_url = f"http://{self.api_host}:{self.api_port}{self.path}"
+        try:
+            response = requests.post(target_url, data=post_data, headers=self._get_headers())
+            self._send_response(response)
+        except Exception as e:
+            logger.error(f"PROXY: Ошибка при проксировании POST-запроса: {e}")
+            self.send_error(HTTPStatus.INTERNAL_SERVER_ERROR, str(e))
+    
+    def _handle_graphs_request(self):
+        """Специальная обработка для запроса графов"""
+        try:
+            # Читаем графы из property.json
+            with open(PROPERTY_JSON, 'r') as f:
+                property_data = json.load(f)
+            
+            graphs = property_data.get('graphs', [])
+            logger.info(f"PROXY: Возвращаем графы напрямую: {json.dumps(graphs)}")
+            
+            self.send_response(HTTPStatus.OK)
+            self.send_header('Content-Type', 'application/json')
+            self.end_headers()
+            self.wfile.write(json.dumps(graphs).encode('utf-8'))
+        except Exception as e:
+            logger.error(f"PROXY: Ошибка при обработке запроса графов: {e}")
+            self.send_error(HTTPStatus.INTERNAL_SERVER_ERROR, str(e))
+    
+    def _handle_designer_reload(self):
+        """Обработка запросов дизайнера"""
+        response_data = {
+            "success": True,
+            "packages": [
+                {
+                    "name": "default",
+                    "description": "Default package",
+                    "graphs": []
+                }
+            ]
+        }
+        
+        # Читаем графы из property.json
+        try:
+            with open(PROPERTY_JSON, 'r') as f:
+                property_data = json.load(f)
+            
+            graphs = property_data.get('graphs', [])
+            response_data["packages"][0]["graphs"] = graphs
+            logger.info(f"PROXY: Возвращаем данные для дизайнера с графами: {json.dumps(graphs)}")
+        except Exception as e:
+            logger.error(f"PROXY: Ошибка при чтении графов для дизайнера: {e}")
+        
+        self.send_response(HTTPStatus.OK)
+        self.send_header('Content-Type', 'application/json')
+        self.end_headers()
+        self.wfile.write(json.dumps(response_data).encode('utf-8'))
+    
+    def _get_headers(self):
+        """Получение заголовков для проксирования"""
+        headers = {}
+        for header in self.headers:
+            headers[header] = self.headers[header]
+        return headers
+    
+    def _send_response(self, response):
+        """Отправка ответа клиенту"""
+        self.send_response(response.status_code)
+        
+        # Копируем заголовки
+        for header, value in response.headers.items():
+            self.send_header(header, value)
+        self.end_headers()
+        
+        # Отправляем тело ответа
+        self.wfile.write(response.content)
+    
+    def log_message(self, format, *args):
+        """Переопределение логирования"""
+        logger.debug(f"PROXY: {self.address_string()} - {format % args}")
+
+def run_proxy_server(port=9090):
+    """Запуск прокси-сервера"""
+    try:
+        with socketserver.TCPServer(("", port), ProxyHTTPRequestHandler) as httpd:
+            logger.info(f"Запуск прокси-сервера на порту {port}")
+            httpd.serve_forever()
+    except Exception as e:
+        logger.error(f"Ошибка при запуске прокси-сервера: {e}")
+
 def ensure_directory_permissions(directory_path):
     """Обеспечиваем правильные разрешения для директории"""
     directory = Path(directory_path)
@@ -264,17 +396,23 @@ def main():
         test_thread.daemon = True
         test_thread.start()
         
+        # Запускаем прокси-сервер для перехвата и модификации запросов
+        proxy_port = 9090
+        proxy_thread = Thread(target=run_proxy_server, args=(proxy_port,))
+        proxy_thread.daemon = True
+        proxy_thread.start()
+        
         # Запускаем Playground UI в режиме dev на порту 7860 (порт Hugging Face)
         logger.info("Запуск Playground UI в режиме разработки на порту 7860...")
         os.environ["PORT"] = "7860"
-        os.environ["AGENT_SERVER_URL"] = "http://localhost:8080"
+        os.environ["AGENT_SERVER_URL"] = f"http://localhost:{proxy_port}"  # Указываем прокси вместо прямого API
         os.environ["NEXT_PUBLIC_EDIT_GRAPH_MODE"] = "true"  # Включаем расширенный режим редактирования
         os.environ["NEXT_PUBLIC_DISABLE_CAMERA"] = "true"  # Отключаем запрос на использование камеры
         
         # Важные переменные для отключения запросов к дизайнеру
         os.environ["NEXT_PUBLIC_DEV_MODE"] = "false"
-        os.environ["NEXT_PUBLIC_API_BASE_URL"] = "/api/agents"
-        os.environ["NEXT_PUBLIC_DESIGNER_API_URL"] = "http://localhost:8080"
+        os.environ["NEXT_PUBLIC_API_BASE_URL"] = f"/api/agents"
+        os.environ["NEXT_PUBLIC_DESIGNER_API_URL"] = f"http://localhost:{proxy_port}"
         
         # Запускаем Playground UI
         playground_process = subprocess.Popen(