Enhance Dockerfile, README, and server configuration for improved security and functionality

- Added non-root user creation in Dockerfile for enhanced security.
- Configured permissions and environment variables in Dockerfile.
- Updated README.md to include security features, API endpoints, and local development instructions.
- Expanded requirements.txt with additional packages for logging and monitoring.
- Enhanced server.py with rate limiting and message size configuration for better performance.

Dockerfile

@@ -1,6 +1,9 @@
 # Dockerfile
 FROM python:3.11-slim
 
+# Crear usuario no root
+RUN adduser --disabled-password --gecos '' appuser
+
 WORKDIR /app
 
 # Instalar dependencias del sistema si son necesarias
@@ -16,8 +19,18 @@ COPY hypercorn.toml .
 # Instalar dependencias de Python
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Variables de entorno
+# Configurar permisos
+RUN chown -R appuser:appuser /app && \
+    mkdir -p /home/appuser/.cache/huggingface && \
+    chown -R appuser:appuser /home/appuser/.cache
+
+# Variables de entorno de seguridad
 ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONPATH=/app
+
+# Cambiar al usuario no root
+USER appuser
 
 # Exponer el puerto que Hugging Face Spaces espera
 EXPOSE 7860

README.md

@@ -10,15 +10,66 @@ pinned: false
 
 # Kaio API Server
 
-Este servidor proporciona una API REST y WebSocket para ejecutar comandos y gestionar archivos.
+Este servidor proporciona una API REST y WebSocket para ejecutar comandos y gestionar archivos de forma segura.
+
+## Características de Seguridad
+
+- Autenticación mediante tokens de sesión
+- Rate limiting por IP/sesión
+- Ejecución de comandos en entorno aislado
+- Validación de entrada con Pydantic
+- CORS configurado para endpoints específicos
+- Logs estructurados y monitorizables
 
 ## Endpoints
 
-- REST API: `https://<space-name>.hf.space/`
-- WebSocket: `wss://<space-name>.hf.space/session`
+### REST API
+- `GET /status` - Estado del servidor y versión
+- `GET /health` - Health check del servidor
+- `POST /command` - Ejecutar comando (requiere autenticación)
+
+### WebSocket
+- `wss://<space-name>.hf.space/session` - Conexión WebSocket para sesiones
+
+## Configuración
+
+El servidor se puede configurar mediante variables de entorno:
+
+```env
+TIMEOUT=60
+LOG_LEVEL=INFO
+MAX_MESSAGE_SIZE=16777216
+RATE_LIMIT=100
+SESSION_TIMEOUT=3600
+```
+
+## Monitoreo
+
+- Métricas Prometheus disponibles en `/metrics`
+- Logs estructurados en formato JSON
+- Health checks periódicos
+
+## Seguridad
+
+- Ejecuta como usuario no-root
+- Aislamiento de contenedor
+- Validación de entrada estricta
+- Límites de recursos configurables
+
+## Desarrollo Local
+
+```bash
+# Construir imagen
+docker build -t kaio-server .
+
+# Ejecutar servidor
+docker run -p 7860:7860 kaio-server
+```
 
-## Estado
+## Producción
 
-Puedes verificar el estado del servidor accediendo a:
-- `https://<space-name>.hf.space/health`
-- `https://<space-name>.hf.space/status`
+Se recomienda:
+1. Configurar límites de recursos
+2. Habilitar logging estructurado
+3. Configurar monitoreo
+4. Revisar configuración CORS

requirements.txt

@@ -1,4 +1,9 @@
-quart>=0.18.4
-quart-schema>=0.15.0
-hypercorn>=0.14.3
-websockets>=11.0.3 
+quart==0.18.4
+quart-schema==0.15.0
+quart-cors==0.7.0
+hypercorn==0.14.3
+websockets==11.0.3
+python-json-logger==2.0.7
+prometheus-client==0.17.1
+pydantic==2.4.2
+python-dotenv==1.0.0 

server.py

@@ -17,10 +17,18 @@ from broker import SessionBroker, SessionDoesNotExist, ClientRequest, ClientResp
 TIMEOUT: int = 60
 LOG_LEVEL: int = logging.INFO
 TRUSTED_HOSTS: list[str] = ["127.0.0.1"]
+MAX_MESSAGE_SIZE: int = 16 * 1024 * 1024  # 16MB
+RATE_LIMIT: int = 100  # requests per minute
+SESSION_TIMEOUT: int = 3600  # 1 hour
 
-# Crear aplicación con CORS habilitado
+# Crear aplicación con CORS habilitado y configuración de seguridad
 app = Quart(__name__)
-app = cors(app, allow_origin="*")
+app = cors(app, 
+    allow_origin="*",
+    allow_methods=["GET", "POST", "OPTIONS"],
+    allow_headers=["Content-Type"],
+    max_age=3600
+)
 QuartSchema(app)
 app.logger.setLevel(LOG_LEVEL)