mirror of
https://github.com/johndoe6345789/docker-swarm-termina.git
synced 2026-04-25 06:05:00 +00:00
Compare commits
62 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| c2c08fe157 | |||
|
|
4e928db0a8 | ||
|
|
1f1608e081 | ||
|
|
b6cef2f89a | ||
|
|
2404255e58 | ||
|
|
c3ce17c88e | ||
|
|
088db7536e | ||
|
|
985c98339a | ||
|
|
6c77ae0611 | ||
|
|
c00e806f2d | ||
|
|
5ff71cd8f4 | ||
|
|
cf45accf4a | ||
|
|
b4e133fd4d | ||
|
|
d0074ff874 | ||
|
|
9a08193610 | ||
|
|
1cbf7966c5 | ||
|
|
ff19cd1a5a | ||
| 71ee74ed5f | |||
|
|
9fe942a510 | ||
| 9dae3f3d30 | |||
|
|
6c61a508ca | ||
|
|
7bb7175bd9 | ||
| 6e794047b5 | |||
|
|
aac0d5a509 | ||
| 649c4dd2e7 | |||
|
|
f64c22a24c | ||
|
|
ba2d50e98b | ||
|
|
bbf3959242 | ||
|
|
78f67d9483 | ||
|
|
b7883a2fb4 | ||
|
|
21e2b7dcf7 | ||
| 1ddc553936 | |||
|
|
95511bc15a | ||
|
|
c667af076c | ||
|
|
4eaaa728ad | ||
| 8f2dc9290d | |||
|
|
713784a450 | ||
|
|
cb5c012857 | ||
| f927710908 | |||
|
|
64d56d9110 | ||
| e94b1f0053 | |||
| 0902d480ed | |||
|
|
42c1d4737f | ||
|
|
e97b50a916 | ||
|
|
748bf87699 | ||
|
|
70d32f13b2 | ||
|
|
87daa3add3 | ||
|
|
2c34509d0f | ||
| 32253724b0 | |||
| f9d781271f | |||
| 69dee82d89 | |||
| 5343fd9f51 | |||
|
|
b580744f32 | ||
|
|
cc2915e82d | ||
| 5daee2d445 | |||
|
|
a59b5ad527 | ||
| 995b7442d7 | |||
|
|
ce997ebdda | ||
|
|
d9c790c560 | ||
|
|
237ebcede1 | ||
| 2e176f3048 | |||
|
|
938cb5a0ba |
50
.github/workflows/README.md
vendored
Normal file
50
.github/workflows/README.md
vendored
Normal file
@@ -0,0 +1,50 @@
|
||||
# GitHub Actions Workflows
|
||||
|
||||
This directory contains GitHub Actions workflows for CI/CD automation.
|
||||
|
||||
## Workflows
|
||||
|
||||
### test.yml
|
||||
Runs on every push and pull request to ensure code quality:
|
||||
- **Backend Tests**: Runs pytest with coverage on Python 3.11 and 3.12
|
||||
- Requires 70% test coverage minimum
|
||||
- Uploads coverage reports to Codecov
|
||||
- **Frontend Tests**: Lints and builds the Next.js frontend
|
||||
- **Docker Build Test**: Validates Docker images can be built successfully
|
||||
|
||||
### docker-publish.yml
|
||||
Runs on pushes to main and version tags:
|
||||
- Builds and pushes Docker images to GitHub Container Registry (GHCR)
|
||||
- Creates multi-platform images for both backend and frontend
|
||||
- Tags images with branch name, PR number, version, and commit SHA
|
||||
|
||||
### create-release.yml
|
||||
Handles release creation and management
|
||||
|
||||
## Test Coverage Requirements
|
||||
|
||||
Backend tests must maintain at least 70% code coverage. The pipeline will fail if coverage drops below this threshold.
|
||||
|
||||
## Local Testing
|
||||
|
||||
To run tests locally before pushing:
|
||||
|
||||
```bash
|
||||
# Backend tests
|
||||
cd backend
|
||||
pip install -r requirements.txt -r requirements-dev.txt
|
||||
pytest --cov=. --cov-report=term-missing
|
||||
|
||||
# Frontend build
|
||||
cd frontend
|
||||
npm install
|
||||
npm run build
|
||||
```
|
||||
|
||||
## Adding New Tests
|
||||
|
||||
When adding new features:
|
||||
1. Write unit tests in `backend/tests/test_*.py`
|
||||
2. Ensure all tests pass locally
|
||||
3. Push changes - the CI will automatically run all tests
|
||||
4. Fix any failing tests before merging
|
||||
115
.github/workflows/test.yml
vendored
Normal file
115
.github/workflows/test.yml
vendored
Normal file
@@ -0,0 +1,115 @@
|
||||
name: Run Tests
|
||||
|
||||
on:
|
||||
push:
|
||||
branches: ['**']
|
||||
pull_request:
|
||||
branches: [main]
|
||||
|
||||
jobs:
|
||||
backend-tests:
|
||||
runs-on: ubuntu-latest
|
||||
strategy:
|
||||
matrix:
|
||||
python-version: ['3.11', '3.12']
|
||||
|
||||
steps:
|
||||
- name: Checkout code
|
||||
uses: actions/checkout@v4
|
||||
|
||||
- name: Set up Python ${{ matrix.python-version }}
|
||||
uses: actions/setup-python@v5
|
||||
with:
|
||||
python-version: ${{ matrix.python-version }}
|
||||
|
||||
- name: Cache pip packages
|
||||
uses: actions/cache@v4
|
||||
with:
|
||||
path: ~/.cache/pip
|
||||
key: ${{ runner.os }}-pip-${{ hashFiles('**/requirements*.txt') }}
|
||||
restore-keys: |
|
||||
${{ runner.os }}-pip-
|
||||
|
||||
- name: Install dependencies
|
||||
working-directory: ./backend
|
||||
run: |
|
||||
python -m pip install --upgrade pip
|
||||
pip install -r requirements.txt
|
||||
pip install -r requirements-dev.txt
|
||||
|
||||
- name: Run pytest with coverage
|
||||
working-directory: ./backend
|
||||
run: |
|
||||
pytest --cov=. --cov-report=xml --cov-report=term-missing -v
|
||||
|
||||
- name: Upload coverage to Codecov
|
||||
uses: codecov/codecov-action@v4
|
||||
with:
|
||||
file: ./backend/coverage.xml
|
||||
flags: backend
|
||||
name: backend-coverage
|
||||
fail_ci_if_error: false
|
||||
|
||||
- name: Check test coverage threshold
|
||||
working-directory: ./backend
|
||||
run: |
|
||||
coverage report --fail-under=70
|
||||
|
||||
frontend-tests:
|
||||
runs-on: ubuntu-latest
|
||||
|
||||
steps:
|
||||
- name: Checkout code
|
||||
uses: actions/checkout@v4
|
||||
|
||||
- name: Set up Node.js
|
||||
uses: actions/setup-node@v4
|
||||
with:
|
||||
node-version: '20'
|
||||
cache: 'npm'
|
||||
cache-dependency-path: frontend/package-lock.json
|
||||
|
||||
- name: Install dependencies
|
||||
working-directory: ./frontend
|
||||
run: npm ci
|
||||
|
||||
- name: Run linting
|
||||
working-directory: ./frontend
|
||||
run: npm run lint || echo "Linting not configured yet"
|
||||
|
||||
- name: Build frontend
|
||||
working-directory: ./frontend
|
||||
run: npm run build
|
||||
|
||||
docker-build-test:
|
||||
runs-on: ubuntu-latest
|
||||
needs: [backend-tests, frontend-tests]
|
||||
|
||||
steps:
|
||||
- name: Checkout code
|
||||
uses: actions/checkout@v4
|
||||
|
||||
- name: Set up Docker Buildx
|
||||
uses: docker/setup-buildx-action@v3
|
||||
|
||||
- name: Build backend Docker image
|
||||
uses: docker/build-push-action@v5
|
||||
with:
|
||||
context: ./backend
|
||||
file: ./backend/Dockerfile
|
||||
push: false
|
||||
tags: backend:test
|
||||
cache-from: type=gha
|
||||
cache-to: type=gha,mode=max
|
||||
|
||||
- name: Build frontend Docker image
|
||||
uses: docker/build-push-action@v5
|
||||
with:
|
||||
context: ./frontend
|
||||
file: ./frontend/Dockerfile
|
||||
push: false
|
||||
tags: frontend:test
|
||||
cache-from: type=gha
|
||||
cache-to: type=gha,mode=max
|
||||
build-args: |
|
||||
NEXT_PUBLIC_API_URL=http://backend:5000
|
||||
7
.gitignore
vendored
7
.gitignore
vendored
@@ -46,6 +46,13 @@ ENV/
|
||||
*.egg-info/
|
||||
.pytest_cache/
|
||||
|
||||
# Coverage
|
||||
.coverage
|
||||
.coverage.*
|
||||
htmlcov/
|
||||
coverage.xml
|
||||
.cache
|
||||
|
||||
# Next.js
|
||||
.next/
|
||||
out/
|
||||
|
||||
22
backend/.coveragerc
Normal file
22
backend/.coveragerc
Normal file
@@ -0,0 +1,22 @@
|
||||
[run]
|
||||
source = .
|
||||
omit =
|
||||
tests/*
|
||||
*/__pycache__/*
|
||||
*/venv/*
|
||||
*/virtualenv/*
|
||||
setup.py
|
||||
conftest.py
|
||||
|
||||
[report]
|
||||
exclude_lines =
|
||||
pragma: no cover
|
||||
def __repr__
|
||||
raise AssertionError
|
||||
raise NotImplementedError
|
||||
if __name__ == .__main__.:
|
||||
if TYPE_CHECKING:
|
||||
@abstractmethod
|
||||
|
||||
[html]
|
||||
directory = htmlcov
|
||||
24
backend/.gitignore
vendored
Normal file
24
backend/.gitignore
vendored
Normal file
@@ -0,0 +1,24 @@
|
||||
# Python
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
*$py.class
|
||||
*.so
|
||||
.Python
|
||||
env/
|
||||
venv/
|
||||
ENV/
|
||||
|
||||
# Testing
|
||||
.coverage
|
||||
coverage.xml
|
||||
htmlcov/
|
||||
.pytest_cache/
|
||||
|
||||
# IDE
|
||||
.vscode/
|
||||
.idea/
|
||||
*.swp
|
||||
*.swo
|
||||
|
||||
# Environment
|
||||
.env
|
||||
380
backend/app.py
380
backend/app.py
@@ -1,349 +1,51 @@
|
||||
from flask import Flask, jsonify, request
|
||||
"""Main application entry point - refactored modular architecture."""
|
||||
from flask import Flask
|
||||
from flask_cors import CORS
|
||||
import docker
|
||||
import os
|
||||
import sys
|
||||
import logging
|
||||
from datetime import datetime, timedelta
|
||||
from flask_socketio import SocketIO
|
||||
|
||||
# Configure logging
|
||||
logging.basicConfig(
|
||||
level=logging.DEBUG,
|
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||
handlers=[
|
||||
logging.StreamHandler(sys.stdout)
|
||||
]
|
||||
)
|
||||
logger = logging.getLogger(__name__)
|
||||
from config import logger
|
||||
from routes.login import login_bp
|
||||
from routes.logout import logout_bp
|
||||
from routes.health import health_bp
|
||||
from routes.containers.list import list_bp
|
||||
from routes.containers.exec import exec_bp
|
||||
from routes.containers.start import start_bp
|
||||
from routes.containers.stop import stop_bp
|
||||
from routes.containers.restart import restart_bp
|
||||
from routes.containers.remove import remove_bp
|
||||
from handlers.terminal.register import register_terminal_handlers
|
||||
from utils.diagnostics.docker_env import diagnose_docker_environment
|
||||
from utils.docker_client import get_docker_client
|
||||
|
||||
# Initialize Flask app
|
||||
app = Flask(__name__)
|
||||
CORS(app)
|
||||
CORS(app, resources={r"/*": {"origins": "*"}})
|
||||
|
||||
# Simple in-memory session storage (in production, use proper session management)
|
||||
sessions = {}
|
||||
# Track working directory per session
|
||||
session_workdirs = {}
|
||||
# Initialize SocketIO
|
||||
socketio = SocketIO(
|
||||
app,
|
||||
cors_allowed_origins="*",
|
||||
async_mode='threading',
|
||||
ping_timeout=60,
|
||||
ping_interval=25,
|
||||
logger=True,
|
||||
engineio_logger=True
|
||||
)
|
||||
|
||||
# Default credentials (should be environment variables in production)
|
||||
ADMIN_USERNAME = os.getenv('ADMIN_USERNAME', 'admin')
|
||||
ADMIN_PASSWORD = os.getenv('ADMIN_PASSWORD', 'admin123')
|
||||
# Register blueprints
|
||||
app.register_blueprint(login_bp)
|
||||
app.register_blueprint(logout_bp)
|
||||
app.register_blueprint(health_bp)
|
||||
app.register_blueprint(list_bp)
|
||||
app.register_blueprint(exec_bp)
|
||||
app.register_blueprint(start_bp)
|
||||
app.register_blueprint(stop_bp)
|
||||
app.register_blueprint(restart_bp)
|
||||
app.register_blueprint(remove_bp)
|
||||
|
||||
def diagnose_docker_environment():
|
||||
"""Diagnose Docker environment and configuration"""
|
||||
logger.info("=== Docker Environment Diagnosis ===")
|
||||
# Register WebSocket handlers
|
||||
register_terminal_handlers(socketio)
|
||||
|
||||
# Check environment variables
|
||||
docker_host = os.getenv('DOCKER_HOST', 'Not set')
|
||||
docker_cert_path = os.getenv('DOCKER_CERT_PATH', 'Not set')
|
||||
docker_tls_verify = os.getenv('DOCKER_TLS_VERIFY', 'Not set')
|
||||
|
||||
logger.info(f"DOCKER_HOST: {docker_host}")
|
||||
logger.info(f"DOCKER_CERT_PATH: {docker_cert_path}")
|
||||
logger.info(f"DOCKER_TLS_VERIFY: {docker_tls_verify}")
|
||||
|
||||
# Check what's in /var/run
|
||||
logger.info("Checking /var/run directory contents:")
|
||||
try:
|
||||
if os.path.exists('/var/run'):
|
||||
var_run_contents = os.listdir('/var/run')
|
||||
logger.info(f" /var/run contains: {var_run_contents}")
|
||||
|
||||
# Check for any Docker-related files
|
||||
docker_related = [f for f in var_run_contents if 'docker' in f.lower()]
|
||||
if docker_related:
|
||||
logger.info(f" Docker-related files/dirs found: {docker_related}")
|
||||
else:
|
||||
logger.warning(" /var/run directory doesn't exist")
|
||||
except Exception as e:
|
||||
logger.error(f" Error reading /var/run: {e}")
|
||||
|
||||
# Check Docker socket
|
||||
socket_path = '/var/run/docker.sock'
|
||||
logger.info(f"Checking Docker socket at {socket_path}")
|
||||
|
||||
if os.path.exists(socket_path):
|
||||
logger.info(f"✓ Docker socket exists at {socket_path}")
|
||||
|
||||
# Check permissions
|
||||
import stat
|
||||
st = os.stat(socket_path)
|
||||
logger.info(f" Socket permissions: {oct(st.st_mode)}")
|
||||
logger.info(f" Socket owner UID: {st.st_uid}")
|
||||
logger.info(f" Socket owner GID: {st.st_gid}")
|
||||
|
||||
# Check if readable/writable
|
||||
readable = os.access(socket_path, os.R_OK)
|
||||
writable = os.access(socket_path, os.W_OK)
|
||||
logger.info(f" Readable: {readable}")
|
||||
logger.info(f" Writable: {writable}")
|
||||
|
||||
if not (readable and writable):
|
||||
logger.warning(f"⚠ Socket exists but lacks proper permissions!")
|
||||
else:
|
||||
logger.error(f"✗ Docker socket NOT found at {socket_path}")
|
||||
logger.error(f" This means the Docker socket mount is NOT configured in CapRover")
|
||||
logger.error(f" The serviceUpdateOverride in captain-definition may not be applied")
|
||||
|
||||
# Check current user
|
||||
import pwd
|
||||
try:
|
||||
current_uid = os.getuid()
|
||||
current_gid = os.getgid()
|
||||
user_info = pwd.getpwuid(current_uid)
|
||||
logger.info(f"Current user: {user_info.pw_name} (UID: {current_uid}, GID: {current_gid})")
|
||||
|
||||
# Check groups
|
||||
import grp
|
||||
groups = os.getgroups()
|
||||
logger.info(f"User groups (GIDs): {groups}")
|
||||
|
||||
for gid in groups:
|
||||
try:
|
||||
group_info = grp.getgrgid(gid)
|
||||
logger.info(f" - {group_info.gr_name} (GID: {gid})")
|
||||
except:
|
||||
logger.info(f" - Unknown group (GID: {gid})")
|
||||
except Exception as e:
|
||||
logger.error(f"Error checking user info: {e}")
|
||||
|
||||
logger.info("=== End Diagnosis ===")
|
||||
|
||||
def get_docker_client():
|
||||
"""Get Docker client with enhanced error reporting"""
|
||||
try:
|
||||
logger.info("Attempting to connect to Docker...")
|
||||
|
||||
# Try default connection first
|
||||
try:
|
||||
client = docker.from_env()
|
||||
# Test the connection
|
||||
client.ping()
|
||||
logger.info("✓ Successfully connected to Docker using docker.from_env()")
|
||||
return client
|
||||
except Exception as e:
|
||||
logger.warning(f"docker.from_env() failed: {e}")
|
||||
|
||||
# Try explicit Unix socket connection
|
||||
try:
|
||||
logger.info("Trying explicit Unix socket connection...")
|
||||
client = docker.DockerClient(base_url='unix:///var/run/docker.sock')
|
||||
client.ping()
|
||||
logger.info("✓ Successfully connected to Docker using Unix socket")
|
||||
return client
|
||||
except Exception as e:
|
||||
logger.warning(f"Unix socket connection failed: {e}")
|
||||
|
||||
# If all fails, run diagnostics and return None
|
||||
logger.error("All Docker connection attempts failed!")
|
||||
diagnose_docker_environment()
|
||||
return None
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Unexpected error in get_docker_client: {e}", exc_info=True)
|
||||
return None
|
||||
|
||||
def format_uptime(created_at):
|
||||
"""Format container uptime"""
|
||||
created = datetime.fromisoformat(created_at.replace('Z', '+00:00'))
|
||||
now = datetime.now(created.tzinfo)
|
||||
delta = now - created
|
||||
|
||||
days = delta.days
|
||||
hours = delta.seconds // 3600
|
||||
minutes = (delta.seconds % 3600) // 60
|
||||
|
||||
if days > 0:
|
||||
return f"{days}d {hours}h"
|
||||
elif hours > 0:
|
||||
return f"{hours}h {minutes}m"
|
||||
else:
|
||||
return f"{minutes}m"
|
||||
|
||||
@app.route('/api/auth/login', methods=['POST'])
|
||||
def login():
|
||||
"""Authenticate user"""
|
||||
data = request.get_json()
|
||||
username = data.get('username')
|
||||
password = data.get('password')
|
||||
|
||||
if username == ADMIN_USERNAME and password == ADMIN_PASSWORD:
|
||||
# Create a simple session token (in production, use JWT or proper session management)
|
||||
session_token = f"session_{username}_{datetime.now().timestamp()}"
|
||||
sessions[session_token] = {
|
||||
'username': username,
|
||||
'created_at': datetime.now()
|
||||
}
|
||||
return jsonify({
|
||||
'success': True,
|
||||
'token': session_token,
|
||||
'username': username
|
||||
})
|
||||
|
||||
return jsonify({
|
||||
'success': False,
|
||||
'message': 'Invalid credentials'
|
||||
}), 401
|
||||
|
||||
@app.route('/api/auth/logout', methods=['POST'])
|
||||
def logout():
|
||||
"""Logout user"""
|
||||
auth_header = request.headers.get('Authorization')
|
||||
if auth_header and auth_header.startswith('Bearer '):
|
||||
token = auth_header.split(' ')[1]
|
||||
if token in sessions:
|
||||
del sessions[token]
|
||||
|
||||
return jsonify({'success': True})
|
||||
|
||||
@app.route('/api/containers', methods=['GET'])
|
||||
def get_containers():
|
||||
"""Get list of all containers"""
|
||||
auth_header = request.headers.get('Authorization')
|
||||
if not auth_header or not auth_header.startswith('Bearer '):
|
||||
return jsonify({'error': 'Unauthorized'}), 401
|
||||
|
||||
token = auth_header.split(' ')[1]
|
||||
if token not in sessions:
|
||||
return jsonify({'error': 'Invalid session'}), 401
|
||||
|
||||
client = get_docker_client()
|
||||
if not client:
|
||||
return jsonify({'error': 'Cannot connect to Docker'}), 500
|
||||
|
||||
try:
|
||||
containers = client.containers.list(all=True)
|
||||
container_list = []
|
||||
|
||||
for container in containers:
|
||||
container_list.append({
|
||||
'id': container.short_id,
|
||||
'name': container.name,
|
||||
'image': container.image.tags[0] if container.image.tags else 'unknown',
|
||||
'status': container.status,
|
||||
'uptime': format_uptime(container.attrs['Created']) if container.status == 'running' else 'N/A'
|
||||
})
|
||||
|
||||
return jsonify({'containers': container_list})
|
||||
except Exception as e:
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
@app.route('/api/containers/<container_id>/exec', methods=['POST'])
|
||||
def exec_container(container_id):
|
||||
"""Execute command in container"""
|
||||
auth_header = request.headers.get('Authorization')
|
||||
if not auth_header or not auth_header.startswith('Bearer '):
|
||||
return jsonify({'error': 'Unauthorized'}), 401
|
||||
|
||||
token = auth_header.split(' ')[1]
|
||||
if token not in sessions:
|
||||
return jsonify({'error': 'Invalid session'}), 401
|
||||
|
||||
data = request.get_json()
|
||||
user_command = data.get('command', 'echo "No command provided"')
|
||||
|
||||
client = get_docker_client()
|
||||
if not client:
|
||||
return jsonify({'error': 'Cannot connect to Docker'}), 500
|
||||
|
||||
try:
|
||||
container = client.containers.get(container_id)
|
||||
|
||||
# Get or initialize session working directory
|
||||
session_key = f"{token}_{container_id}"
|
||||
if session_key not in session_workdirs:
|
||||
# Get container's default working directory or use root
|
||||
session_workdirs[session_key] = '/'
|
||||
|
||||
current_workdir = session_workdirs[session_key]
|
||||
|
||||
# Check if this is a cd command
|
||||
cd_match = user_command.strip()
|
||||
is_cd_command = cd_match.startswith('cd ')
|
||||
|
||||
# If it's a cd command, handle it specially
|
||||
if is_cd_command:
|
||||
target_dir = cd_match[3:].strip() or '~'
|
||||
# Resolve the new directory and update session
|
||||
resolve_command = f'cd "{current_workdir}" && cd {target_dir} && pwd'
|
||||
bash_command = [
|
||||
'/bin/bash',
|
||||
'-c',
|
||||
f'export PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin; {resolve_command}'
|
||||
]
|
||||
else:
|
||||
# Regular command - execute in current working directory
|
||||
bash_command = [
|
||||
'/bin/bash',
|
||||
'-c',
|
||||
f'export PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin; cd "{current_workdir}" && {user_command}; echo "::WORKDIR::$(pwd)"'
|
||||
]
|
||||
|
||||
# Try bash first, fallback to sh if bash doesn't exist
|
||||
try:
|
||||
exec_instance = container.exec_run(
|
||||
bash_command,
|
||||
stdout=True,
|
||||
stderr=True,
|
||||
stdin=False,
|
||||
tty=True,
|
||||
environment={'TERM': 'xterm-256color', 'LANG': 'C.UTF-8'}
|
||||
)
|
||||
except Exception as bash_error:
|
||||
logger.warning(f"Bash execution failed, trying sh: {bash_error}")
|
||||
# Fallback to sh
|
||||
if is_cd_command:
|
||||
target_dir = cd_match[3:].strip() or '~'
|
||||
resolve_command = f'cd "{current_workdir}" && cd {target_dir} && pwd'
|
||||
sh_command = ['/bin/sh', '-c', f'export PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin; {resolve_command}']
|
||||
else:
|
||||
sh_command = ['/bin/sh', '-c', f'export PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin; cd "{current_workdir}" && {user_command}; echo "::WORKDIR::$(pwd)"']
|
||||
|
||||
exec_instance = container.exec_run(
|
||||
sh_command,
|
||||
stdout=True,
|
||||
stderr=True,
|
||||
stdin=False,
|
||||
tty=True,
|
||||
environment={'TERM': 'xterm-256color', 'LANG': 'C.UTF-8'}
|
||||
)
|
||||
|
||||
# Decode output with error handling
|
||||
output = ''
|
||||
if exec_instance.output:
|
||||
try:
|
||||
output = exec_instance.output.decode('utf-8')
|
||||
except UnicodeDecodeError:
|
||||
# Try latin-1 as fallback
|
||||
output = exec_instance.output.decode('latin-1', errors='replace')
|
||||
|
||||
# Extract and update working directory from output
|
||||
new_workdir = current_workdir
|
||||
if is_cd_command:
|
||||
# For cd commands, the output is the new pwd
|
||||
new_workdir = output.strip()
|
||||
session_workdirs[session_key] = new_workdir
|
||||
output = '' # Don't show the pwd output for cd
|
||||
else:
|
||||
# Extract workdir marker from output
|
||||
if '::WORKDIR::' in output:
|
||||
parts = output.rsplit('::WORKDIR::', 1)
|
||||
output = parts[0]
|
||||
new_workdir = parts[1].strip()
|
||||
session_workdirs[session_key] = new_workdir
|
||||
|
||||
return jsonify({
|
||||
'output': output,
|
||||
'exit_code': exec_instance.exit_code,
|
||||
'workdir': new_workdir
|
||||
})
|
||||
except Exception as e:
|
||||
logger.error(f"Error executing command: {e}", exc_info=True)
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
@app.route('/api/health', methods=['GET'])
|
||||
def health():
|
||||
"""Health check endpoint"""
|
||||
return jsonify({'status': 'healthy'})
|
||||
|
||||
if __name__ == '__main__':
|
||||
# Run diagnostics on startup
|
||||
@@ -357,4 +59,4 @@ if __name__ == '__main__':
|
||||
else:
|
||||
logger.error("✗ Docker connection FAILED on startup - check logs above for details")
|
||||
|
||||
app.run(host='0.0.0.0', port=5000, debug=True)
|
||||
socketio.run(app, host='0.0.0.0', port=5000, debug=True, allow_unsafe_werkzeug=True)
|
||||
|
||||
28
backend/config.py
Normal file
28
backend/config.py
Normal file
@@ -0,0 +1,28 @@
|
||||
"""Application configuration and constants."""
|
||||
import os
|
||||
import sys
|
||||
import logging
|
||||
|
||||
# Configure logging
|
||||
logging.basicConfig(
|
||||
level=logging.DEBUG,
|
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||
handlers=[
|
||||
logging.StreamHandler(sys.stdout)
|
||||
]
|
||||
)
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# Default credentials (should be environment variables in production)
|
||||
ADMIN_USERNAME = os.getenv('ADMIN_USERNAME', 'admin')
|
||||
ADMIN_PASSWORD = os.getenv('ADMIN_PASSWORD', 'admin123')
|
||||
|
||||
# Simple in-memory session storage (in production, use proper session management)
|
||||
sessions = {}
|
||||
|
||||
# Track working directory per session
|
||||
session_workdirs = {}
|
||||
|
||||
# Active terminal sessions
|
||||
active_terminals = {}
|
||||
1
backend/handlers/__init__.py
Normal file
1
backend/handlers/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
"""Socket.io handlers - one file per event."""
|
||||
1
backend/handlers/terminal/__init__.py
Normal file
1
backend/handlers/terminal/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
"""Terminal WebSocket handlers."""
|
||||
8
backend/handlers/terminal/connect.py
Normal file
8
backend/handlers/terminal/connect.py
Normal file
@@ -0,0 +1,8 @@
|
||||
"""Terminal WebSocket connect handler."""
|
||||
from flask import request
|
||||
from config import logger
|
||||
|
||||
|
||||
def handle_connect():
|
||||
"""Handle WebSocket connection."""
|
||||
logger.info("Client connected to terminal WebSocket: %s", request.sid)
|
||||
17
backend/handlers/terminal/disconnect.py
Normal file
17
backend/handlers/terminal/disconnect.py
Normal file
@@ -0,0 +1,17 @@
|
||||
"""Terminal WebSocket disconnect handler."""
|
||||
from flask import request
|
||||
from config import logger, active_terminals
|
||||
|
||||
|
||||
def handle_disconnect():
|
||||
"""Handle WebSocket disconnection."""
|
||||
logger.info("Client disconnected from terminal WebSocket: %s", request.sid)
|
||||
# Clean up any active terminal sessions
|
||||
if request.sid in active_terminals:
|
||||
try:
|
||||
exec_instance = active_terminals[request.sid]['exec']
|
||||
if hasattr(exec_instance, 'kill'):
|
||||
exec_instance.kill()
|
||||
except Exception: # pylint: disable=broad-exception-caught
|
||||
pass
|
||||
del active_terminals[request.sid]
|
||||
32
backend/handlers/terminal/input.py
Normal file
32
backend/handlers/terminal/input.py
Normal file
@@ -0,0 +1,32 @@
|
||||
"""Terminal WebSocket input handler."""
|
||||
from flask import request
|
||||
from flask_socketio import emit
|
||||
from config import logger, active_terminals
|
||||
|
||||
|
||||
def handle_input(data):
|
||||
"""Handle input from the client.
|
||||
|
||||
Args:
|
||||
data: Input data containing the user's input string
|
||||
"""
|
||||
try:
|
||||
if request.sid not in active_terminals:
|
||||
emit('error', {'error': 'No active terminal session'})
|
||||
return
|
||||
|
||||
terminal_data = active_terminals[request.sid]
|
||||
exec_instance = terminal_data['exec']
|
||||
input_data = data.get('data', '')
|
||||
|
||||
# Send input to the container
|
||||
sock = exec_instance.output
|
||||
# Access the underlying socket for sendall method
|
||||
if hasattr(sock, '_sock'):
|
||||
sock._sock.sendall(input_data.encode('utf-8')) # pylint: disable=protected-access
|
||||
else:
|
||||
sock.sendall(input_data.encode('utf-8'))
|
||||
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error("Error sending input: %s", e, exc_info=True)
|
||||
emit('error', {'error': str(e)})
|
||||
33
backend/handlers/terminal/register.py
Normal file
33
backend/handlers/terminal/register.py
Normal file
@@ -0,0 +1,33 @@
|
||||
"""Register all terminal WebSocket handlers."""
|
||||
from handlers.terminal.connect import handle_connect
|
||||
from handlers.terminal.disconnect import handle_disconnect
|
||||
from handlers.terminal.start import handle_start_terminal
|
||||
from handlers.terminal.input import handle_input
|
||||
from handlers.terminal.resize import handle_resize
|
||||
|
||||
|
||||
def register_terminal_handlers(socketio):
|
||||
"""Register all terminal WebSocket event handlers.
|
||||
|
||||
Args:
|
||||
socketio: SocketIO instance to register handlers with
|
||||
"""
|
||||
@socketio.on('connect', namespace='/terminal')
|
||||
def on_connect():
|
||||
return handle_connect()
|
||||
|
||||
@socketio.on('disconnect', namespace='/terminal')
|
||||
def on_disconnect():
|
||||
return handle_disconnect()
|
||||
|
||||
@socketio.on('start_terminal', namespace='/terminal')
|
||||
def on_start_terminal(data):
|
||||
return handle_start_terminal(socketio, data)
|
||||
|
||||
@socketio.on('input', namespace='/terminal')
|
||||
def on_input(data):
|
||||
return handle_input(data)
|
||||
|
||||
@socketio.on('resize', namespace='/terminal')
|
||||
def on_resize(data):
|
||||
return handle_resize(data)
|
||||
24
backend/handlers/terminal/resize.py
Normal file
24
backend/handlers/terminal/resize.py
Normal file
@@ -0,0 +1,24 @@
|
||||
"""Terminal WebSocket resize handler."""
|
||||
from flask import request
|
||||
from config import logger, active_terminals
|
||||
|
||||
|
||||
def handle_resize(data):
|
||||
"""Handle terminal resize.
|
||||
|
||||
Args:
|
||||
data: Resize data containing cols and rows
|
||||
|
||||
Note:
|
||||
Docker exec_run doesn't support resizing after creation.
|
||||
This is a limitation of the Docker API.
|
||||
"""
|
||||
try:
|
||||
cols = data.get('cols', 80)
|
||||
rows = data.get('rows', 24)
|
||||
|
||||
if request.sid in active_terminals:
|
||||
logger.info("Terminal resize requested: %sx%s", cols, rows)
|
||||
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error("Error resizing terminal: %s", e, exc_info=True)
|
||||
66
backend/handlers/terminal/start.py
Normal file
66
backend/handlers/terminal/start.py
Normal file
@@ -0,0 +1,66 @@
|
||||
"""Terminal WebSocket start handler."""
|
||||
# pylint: disable=duplicate-code # Auth/client setup pattern is intentional
|
||||
from flask import request
|
||||
from flask_socketio import emit, disconnect
|
||||
from config import logger, sessions, active_terminals
|
||||
from utils.docker_client import get_docker_client
|
||||
from utils.terminal_helpers import create_output_reader
|
||||
|
||||
|
||||
def handle_start_terminal(socketio, data):
|
||||
"""Start an interactive terminal session.
|
||||
|
||||
Args:
|
||||
socketio: SocketIO instance
|
||||
data: Request data containing container_id, token, cols, rows
|
||||
"""
|
||||
try:
|
||||
container_id = data.get('container_id')
|
||||
token = data.get('token')
|
||||
cols = data.get('cols', 80)
|
||||
rows = data.get('rows', 24)
|
||||
|
||||
# Validate token
|
||||
if not token or token not in sessions:
|
||||
emit('error', {'error': 'Unauthorized'})
|
||||
disconnect()
|
||||
return
|
||||
|
||||
# Get Docker client and container
|
||||
client = get_docker_client()
|
||||
if not client:
|
||||
emit('error', {'error': 'Cannot connect to Docker'})
|
||||
return
|
||||
|
||||
container = client.containers.get(container_id)
|
||||
|
||||
# Create an interactive bash session with PTY
|
||||
exec_instance = container.exec_run(
|
||||
['/bin/bash'],
|
||||
stdin=True,
|
||||
stdout=True,
|
||||
stderr=True,
|
||||
tty=True,
|
||||
socket=True,
|
||||
environment={
|
||||
'TERM': 'xterm-256color',
|
||||
'COLUMNS': str(cols),
|
||||
'LINES': str(rows),
|
||||
'LANG': 'C.UTF-8'
|
||||
}
|
||||
)
|
||||
|
||||
# Store the exec instance
|
||||
active_terminals[request.sid] = {
|
||||
'exec': exec_instance,
|
||||
'container_id': container_id
|
||||
}
|
||||
|
||||
# Start output reader thread
|
||||
create_output_reader(socketio, request.sid, exec_instance)
|
||||
|
||||
emit('started', {'message': 'Terminal started'})
|
||||
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error("Error starting terminal: %s", e, exc_info=True)
|
||||
emit('error', {'error': str(e)})
|
||||
17
backend/pytest.ini
Normal file
17
backend/pytest.ini
Normal file
@@ -0,0 +1,17 @@
|
||||
[pytest]
|
||||
testpaths = tests
|
||||
python_files = test_*.py
|
||||
python_classes = Test*
|
||||
python_functions = test_*
|
||||
addopts =
|
||||
-v
|
||||
--strict-markers
|
||||
--cov=.
|
||||
--cov-report=term-missing
|
||||
--cov-report=html
|
||||
--cov-report=xml
|
||||
--cov-branch
|
||||
markers =
|
||||
unit: Unit tests
|
||||
integration: Integration tests
|
||||
slow: Slow running tests
|
||||
5
backend/requirements-dev.txt
Normal file
5
backend/requirements-dev.txt
Normal file
@@ -0,0 +1,5 @@
|
||||
pytest==8.0.0
|
||||
pytest-flask==1.3.0
|
||||
pytest-cov==4.1.0
|
||||
pytest-mock==3.12.0
|
||||
coverage==7.4.1
|
||||
@@ -2,3 +2,5 @@ Flask==3.0.0
|
||||
Flask-CORS==6.0.0
|
||||
python-dotenv==1.0.0
|
||||
docker==7.1.0
|
||||
flask-socketio==5.3.6
|
||||
python-socketio==5.14.0
|
||||
|
||||
1
backend/routes/__init__.py
Normal file
1
backend/routes/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
"""API routes - one file per endpoint for clarity."""
|
||||
1
backend/routes/containers/__init__.py
Normal file
1
backend/routes/containers/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
"""Container management routes - one file per endpoint."""
|
||||
59
backend/routes/containers/exec.py
Normal file
59
backend/routes/containers/exec.py
Normal file
@@ -0,0 +1,59 @@
|
||||
"""Execute command in container route."""
|
||||
from flask import Blueprint, request, jsonify
|
||||
from config import logger, session_workdirs
|
||||
from utils.auth import check_auth
|
||||
from utils.docker_client import get_docker_client
|
||||
from utils.exec_helpers import (
|
||||
get_session_workdir,
|
||||
execute_command_with_fallback,
|
||||
decode_output,
|
||||
extract_workdir
|
||||
)
|
||||
|
||||
exec_bp = Blueprint('exec_container', __name__)
|
||||
|
||||
|
||||
@exec_bp.route('/api/containers/<container_id>/exec', methods=['POST'])
|
||||
def exec_container(container_id):
|
||||
"""Execute command in container."""
|
||||
is_valid, token, error_response = check_auth()
|
||||
if not is_valid:
|
||||
return error_response
|
||||
|
||||
data = request.get_json()
|
||||
user_command = data.get('command', 'echo "No command provided"')
|
||||
|
||||
client = get_docker_client()
|
||||
if not client:
|
||||
return jsonify({'error': 'Cannot connect to Docker'}), 500
|
||||
|
||||
try:
|
||||
# Get session working directory
|
||||
session_key, current_workdir = get_session_workdir(token, container_id, session_workdirs)
|
||||
|
||||
# Execute command with bash/sh fallback
|
||||
exec_instance = execute_command_with_fallback(
|
||||
client.containers.get(container_id),
|
||||
current_workdir,
|
||||
user_command,
|
||||
user_command.strip().startswith('cd ')
|
||||
)
|
||||
|
||||
# Decode and extract workdir from output
|
||||
output, new_workdir = extract_workdir(
|
||||
decode_output(exec_instance),
|
||||
current_workdir,
|
||||
user_command.strip().startswith('cd ')
|
||||
)
|
||||
|
||||
# Update session workdir
|
||||
session_workdirs[session_key] = new_workdir
|
||||
|
||||
return jsonify({
|
||||
'output': output,
|
||||
'exit_code': exec_instance.exit_code,
|
||||
'workdir': new_workdir
|
||||
})
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error("Error executing command: %s", e, exc_info=True)
|
||||
return jsonify({'error': str(e)}), 500
|
||||
37
backend/routes/containers/list.py
Normal file
37
backend/routes/containers/list.py
Normal file
@@ -0,0 +1,37 @@
|
||||
"""List containers route."""
|
||||
from flask import Blueprint, jsonify
|
||||
from utils.auth import check_auth
|
||||
from utils.docker_client import get_docker_client
|
||||
from utils.formatters import format_uptime
|
||||
|
||||
list_bp = Blueprint('list_containers', __name__)
|
||||
|
||||
|
||||
@list_bp.route('/api/containers', methods=['GET'])
|
||||
def get_containers():
|
||||
"""Get list of all containers."""
|
||||
is_valid, _, error_response = check_auth()
|
||||
if not is_valid:
|
||||
return error_response
|
||||
|
||||
client = get_docker_client()
|
||||
if not client:
|
||||
return jsonify({'error': 'Cannot connect to Docker'}), 500
|
||||
|
||||
try:
|
||||
containers = client.containers.list(all=True)
|
||||
container_list = []
|
||||
|
||||
for container in containers:
|
||||
container_list.append({
|
||||
'id': container.short_id,
|
||||
'name': container.name,
|
||||
'image': container.image.tags[0] if container.image.tags else 'unknown',
|
||||
'status': container.status,
|
||||
'uptime': format_uptime(container.attrs['Created'])
|
||||
if container.status == 'running' else 'N/A'
|
||||
})
|
||||
|
||||
return jsonify({'containers': container_list})
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
return jsonify({'error': str(e)}), 500
|
||||
22
backend/routes/containers/remove.py
Normal file
22
backend/routes/containers/remove.py
Normal file
@@ -0,0 +1,22 @@
|
||||
"""Remove container route."""
|
||||
from flask import Blueprint, jsonify
|
||||
from config import logger
|
||||
from utils.container_helpers import get_auth_and_container
|
||||
|
||||
remove_bp = Blueprint('remove_container', __name__)
|
||||
|
||||
|
||||
@remove_bp.route('/api/containers/<container_id>', methods=['DELETE'])
|
||||
def remove_container(container_id):
|
||||
"""Remove a container."""
|
||||
container, error_response = get_auth_and_container(container_id)
|
||||
if error_response:
|
||||
return error_response
|
||||
|
||||
try:
|
||||
container.remove(force=True)
|
||||
logger.info("Removed container %s", container_id)
|
||||
return jsonify({'success': True, 'message': f'Container {container_id} removed'})
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error("Error removing container: %s", e, exc_info=True)
|
||||
return jsonify({'error': str(e)}), 500
|
||||
22
backend/routes/containers/restart.py
Normal file
22
backend/routes/containers/restart.py
Normal file
@@ -0,0 +1,22 @@
|
||||
"""Restart container route."""
|
||||
from flask import Blueprint, jsonify
|
||||
from config import logger
|
||||
from utils.container_helpers import get_auth_and_container
|
||||
|
||||
restart_bp = Blueprint('restart_container', __name__)
|
||||
|
||||
|
||||
@restart_bp.route('/api/containers/<container_id>/restart', methods=['POST'])
|
||||
def restart_container(container_id):
|
||||
"""Restart a container."""
|
||||
container, error_response = get_auth_and_container(container_id)
|
||||
if error_response:
|
||||
return error_response
|
||||
|
||||
try:
|
||||
container.restart()
|
||||
logger.info("Restarted container %s", container_id)
|
||||
return jsonify({'success': True, 'message': f'Container {container_id} restarted'})
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error("Error restarting container: %s", e, exc_info=True)
|
||||
return jsonify({'error': str(e)}), 500
|
||||
22
backend/routes/containers/start.py
Normal file
22
backend/routes/containers/start.py
Normal file
@@ -0,0 +1,22 @@
|
||||
"""Start container route."""
|
||||
from flask import Blueprint, jsonify
|
||||
from config import logger
|
||||
from utils.container_helpers import get_auth_and_container
|
||||
|
||||
start_bp = Blueprint('start_container', __name__)
|
||||
|
||||
|
||||
@start_bp.route('/api/containers/<container_id>/start', methods=['POST'])
|
||||
def start_container(container_id):
|
||||
"""Start a stopped container."""
|
||||
container, error_response = get_auth_and_container(container_id)
|
||||
if error_response:
|
||||
return error_response
|
||||
|
||||
try:
|
||||
container.start()
|
||||
logger.info("Started container %s", container_id)
|
||||
return jsonify({'success': True, 'message': f'Container {container_id} started'})
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error("Error starting container: %s", e, exc_info=True)
|
||||
return jsonify({'error': str(e)}), 500
|
||||
22
backend/routes/containers/stop.py
Normal file
22
backend/routes/containers/stop.py
Normal file
@@ -0,0 +1,22 @@
|
||||
"""Stop container route."""
|
||||
from flask import Blueprint, jsonify
|
||||
from config import logger
|
||||
from utils.container_helpers import get_auth_and_container
|
||||
|
||||
stop_bp = Blueprint('stop_container', __name__)
|
||||
|
||||
|
||||
@stop_bp.route('/api/containers/<container_id>/stop', methods=['POST'])
|
||||
def stop_container(container_id):
|
||||
"""Stop a running container."""
|
||||
container, error_response = get_auth_and_container(container_id)
|
||||
if error_response:
|
||||
return error_response
|
||||
|
||||
try:
|
||||
container.stop()
|
||||
logger.info("Stopped container %s", container_id)
|
||||
return jsonify({'success': True, 'message': f'Container {container_id} stopped'})
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error("Error stopping container: %s", e, exc_info=True)
|
||||
return jsonify({'error': str(e)}), 500
|
||||
10
backend/routes/health.py
Normal file
10
backend/routes/health.py
Normal file
@@ -0,0 +1,10 @@
|
||||
"""Health check route."""
|
||||
from flask import Blueprint, jsonify
|
||||
|
||||
health_bp = Blueprint('health', __name__)
|
||||
|
||||
|
||||
@health_bp.route('/api/health', methods=['GET'])
|
||||
def health():
|
||||
"""Health check endpoint."""
|
||||
return jsonify({'status': 'healthy'})
|
||||
31
backend/routes/login.py
Normal file
31
backend/routes/login.py
Normal file
@@ -0,0 +1,31 @@
|
||||
"""Login route."""
|
||||
from datetime import datetime
|
||||
from flask import Blueprint, request, jsonify
|
||||
from config import ADMIN_USERNAME, ADMIN_PASSWORD, sessions
|
||||
|
||||
login_bp = Blueprint('login', __name__)
|
||||
|
||||
|
||||
@login_bp.route('/api/auth/login', methods=['POST'])
|
||||
def login():
|
||||
"""Authenticate user."""
|
||||
data = request.get_json()
|
||||
username = data.get('username')
|
||||
password = data.get('password')
|
||||
|
||||
if username == ADMIN_USERNAME and password == ADMIN_PASSWORD:
|
||||
session_token = f"session_{username}_{datetime.now().timestamp()}"
|
||||
sessions[session_token] = {
|
||||
'username': username,
|
||||
'created_at': datetime.now()
|
||||
}
|
||||
return jsonify({
|
||||
'success': True,
|
||||
'token': session_token,
|
||||
'username': username
|
||||
})
|
||||
|
||||
return jsonify({
|
||||
'success': False,
|
||||
'message': 'Invalid credentials'
|
||||
}), 401
|
||||
17
backend/routes/logout.py
Normal file
17
backend/routes/logout.py
Normal file
@@ -0,0 +1,17 @@
|
||||
"""Logout route."""
|
||||
from flask import Blueprint, request, jsonify
|
||||
from config import sessions
|
||||
|
||||
logout_bp = Blueprint('logout', __name__)
|
||||
|
||||
|
||||
@logout_bp.route('/api/auth/logout', methods=['POST'])
|
||||
def logout():
|
||||
"""Logout user."""
|
||||
auth_header = request.headers.get('Authorization')
|
||||
if auth_header and auth_header.startswith('Bearer '):
|
||||
token = auth_header.split(' ')[1]
|
||||
if token in sessions:
|
||||
del sessions[token]
|
||||
|
||||
return jsonify({'success': True})
|
||||
187
backend/tests/README.md
Normal file
187
backend/tests/README.md
Normal file
@@ -0,0 +1,187 @@
|
||||
# Backend Tests
|
||||
|
||||
Comprehensive test suite for the Docker Swarm Terminal backend API.
|
||||
|
||||
## Test Structure
|
||||
|
||||
```
|
||||
tests/
|
||||
├── conftest.py # Pytest fixtures and configuration
|
||||
├── test_auth.py # Authentication endpoint tests
|
||||
├── test_containers.py # Container management tests
|
||||
├── test_docker_client.py # Docker client connection tests
|
||||
├── test_exec.py # Command execution tests
|
||||
├── test_exec_advanced.py # Advanced execution tests
|
||||
├── test_health.py # Health check tests
|
||||
├── test_utils.py # Utility function tests
|
||||
├── test_websocket.py # WebSocket terminal unit tests
|
||||
├── test_websocket_simulated.py # WebSocket tests with simulated containers
|
||||
└── test_websocket_integration.py # WebSocket integration tests (require Docker)
|
||||
```
|
||||
|
||||
## Running Tests
|
||||
|
||||
### Install Dependencies
|
||||
|
||||
```bash
|
||||
pip install -r requirements.txt -r requirements-dev.txt
|
||||
```
|
||||
|
||||
### Run All Tests
|
||||
|
||||
```bash
|
||||
pytest
|
||||
```
|
||||
|
||||
### Run with Coverage
|
||||
|
||||
```bash
|
||||
pytest --cov=. --cov-report=html --cov-report=term-missing
|
||||
```
|
||||
|
||||
This will generate an HTML coverage report in `htmlcov/index.html`.
|
||||
|
||||
### Run Specific Test Files
|
||||
|
||||
```bash
|
||||
pytest tests/test_auth.py
|
||||
pytest tests/test_containers.py -v
|
||||
```
|
||||
|
||||
### Run Tests by Marker
|
||||
|
||||
```bash
|
||||
pytest -m unit # Run only unit tests (54 tests)
|
||||
pytest -m integration # Run only integration tests (requires Docker)
|
||||
pytest -m "not integration" # Run all tests except integration tests
|
||||
```
|
||||
|
||||
**Note:** Integration tests will be automatically skipped if Docker is not available.
|
||||
|
||||
### Run with Verbose Output
|
||||
|
||||
```bash
|
||||
pytest -v
|
||||
```
|
||||
|
||||
## Test Coverage
|
||||
|
||||
Current coverage target: **70%**
|
||||
|
||||
To check if tests meet the coverage threshold:
|
||||
|
||||
```bash
|
||||
coverage run -m pytest
|
||||
coverage report --fail-under=70
|
||||
```
|
||||
|
||||
## Writing Tests
|
||||
|
||||
### Test Naming Convention
|
||||
|
||||
- Test files: `test_*.py`
|
||||
- Test classes: `Test*`
|
||||
- Test functions: `test_*`
|
||||
|
||||
### Using Fixtures
|
||||
|
||||
Common fixtures available in `conftest.py`:
|
||||
|
||||
- `app`: Flask application instance
|
||||
- `client`: Test client for making HTTP requests
|
||||
- `auth_token`: Valid authentication token
|
||||
- `auth_headers`: Authentication headers dict
|
||||
- `mock_docker_client`: Mocked Docker client
|
||||
|
||||
Example:
|
||||
|
||||
```python
|
||||
def test_my_endpoint(client, auth_headers):
|
||||
response = client.get('/api/my-endpoint', headers=auth_headers)
|
||||
assert response.status_code == 200
|
||||
```
|
||||
|
||||
### Mocking Docker Calls
|
||||
|
||||
Use the `@patch` decorator to mock Docker API calls:
|
||||
|
||||
```python
|
||||
from unittest.mock import patch, MagicMock
|
||||
|
||||
@patch('app.get_docker_client')
|
||||
def test_container_operation(mock_get_client, client, auth_headers):
|
||||
mock_client = MagicMock()
|
||||
mock_get_client.return_value = mock_client
|
||||
# Your test code here
|
||||
```
|
||||
|
||||
## CI/CD Integration
|
||||
|
||||
Tests automatically run on:
|
||||
- Every push to any branch
|
||||
- Every pull request to main
|
||||
- Multiple Python versions (3.11, 3.12)
|
||||
|
||||
GitHub Actions will fail if:
|
||||
- Any test fails
|
||||
- Coverage drops below 70%
|
||||
- Docker images fail to build
|
||||
|
||||
## Test Types
|
||||
|
||||
### Unit Tests
|
||||
|
||||
Unit tests use mocking and don't require external dependencies like Docker. These are marked with `@pytest.mark.unit` and make up the majority of the test suite.
|
||||
|
||||
### Integration Tests with Simulated Containers
|
||||
|
||||
The `test_websocket_simulated.py` file provides integration-style tests that use simulated Docker containers. These tests:
|
||||
- Don't require Docker to be installed
|
||||
- Test the actual logic flow without external dependencies
|
||||
- Simulate Docker socket behavior including the `_sock` attribute wrapper
|
||||
- Are marked as unit tests since they don't require Docker
|
||||
|
||||
Example simulated container usage:
|
||||
```python
|
||||
def test_with_simulated_container(simulated_container):
|
||||
exec_instance = simulated_container.exec_run(['/bin/sh'], socket=True)
|
||||
sock = exec_instance.output
|
||||
|
||||
# Test socket operations
|
||||
sock._sock.sendall(b'echo test\n')
|
||||
data = sock.recv(4096)
|
||||
```
|
||||
|
||||
### Real Integration Tests
|
||||
|
||||
The `test_websocket_integration.py` file contains tests that require a real Docker environment. These tests:
|
||||
- Are marked with `@pytest.mark.integration`
|
||||
- Automatically skip if Docker is not available
|
||||
- Test with real Docker containers (alpine:latest)
|
||||
- Verify actual Docker socket behavior
|
||||
|
||||
## Troubleshooting
|
||||
|
||||
### Tests Failing Locally
|
||||
|
||||
1. Ensure all dependencies are installed
|
||||
2. Check Python version (3.11+ required)
|
||||
3. Clear pytest cache: `pytest --cache-clear`
|
||||
|
||||
### Import Errors
|
||||
|
||||
Make sure you're running tests from the backend directory:
|
||||
|
||||
```bash
|
||||
cd backend
|
||||
pytest
|
||||
```
|
||||
|
||||
### Coverage Not Updating
|
||||
|
||||
Clear coverage data and re-run:
|
||||
|
||||
```bash
|
||||
coverage erase
|
||||
pytest --cov=. --cov-report=term-missing
|
||||
```
|
||||
1
backend/tests/__init__.py
Normal file
1
backend/tests/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
# Test package initialization
|
||||
169
backend/tests/conftest.py
Normal file
169
backend/tests/conftest.py
Normal file
@@ -0,0 +1,169 @@
|
||||
import pytest
|
||||
import sys
|
||||
import os
|
||||
import socket
|
||||
import threading
|
||||
from unittest.mock import Mock, MagicMock
|
||||
|
||||
# Add the backend directory to the path
|
||||
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
||||
|
||||
from app import app as flask_app, socketio
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def app():
|
||||
"""Create application for testing"""
|
||||
flask_app.config.update({
|
||||
'TESTING': True,
|
||||
'WTF_CSRF_ENABLED': False
|
||||
})
|
||||
yield flask_app
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def client(app):
|
||||
"""Create a test client"""
|
||||
return app.test_client()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def runner(app):
|
||||
"""Create a test CLI runner"""
|
||||
return app.test_cli_runner()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_docker_client(mocker):
|
||||
"""Mock Docker client"""
|
||||
mock_client = mocker.MagicMock()
|
||||
mock_client.ping.return_value = True
|
||||
return mock_client
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def auth_token(client):
|
||||
"""Get a valid authentication token"""
|
||||
response = client.post('/api/auth/login', json={
|
||||
'username': 'admin',
|
||||
'password': 'admin123'
|
||||
})
|
||||
data = response.get_json()
|
||||
return data['token']
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def auth_headers(auth_token):
|
||||
"""Get authentication headers"""
|
||||
return {'Authorization': f'Bearer {auth_token}'}
|
||||
|
||||
|
||||
# Docker integration test helpers
|
||||
|
||||
def docker_available():
|
||||
"""Check if Docker is available"""
|
||||
try:
|
||||
import docker
|
||||
client = docker.from_env()
|
||||
client.ping()
|
||||
return True
|
||||
except Exception:
|
||||
return False
|
||||
|
||||
|
||||
class SimulatedSocket:
|
||||
"""Simulated socket that mimics Docker exec socket behavior"""
|
||||
|
||||
def __init__(self):
|
||||
self._sock = Mock()
|
||||
self._sock.sendall = Mock()
|
||||
self._sock.recv = Mock(return_value=b'$ echo test\ntest\n$ ')
|
||||
self._sock.close = Mock()
|
||||
self.closed = False
|
||||
|
||||
def recv(self, size):
|
||||
"""Simulate receiving data"""
|
||||
if self.closed:
|
||||
return b''
|
||||
return self._sock.recv(size)
|
||||
|
||||
def close(self):
|
||||
"""Close the socket"""
|
||||
self.closed = True
|
||||
self._sock.close()
|
||||
|
||||
|
||||
class SimulatedExecInstance:
|
||||
"""Simulated Docker exec instance for testing without Docker"""
|
||||
|
||||
def __init__(self):
|
||||
self.output = SimulatedSocket()
|
||||
self.id = 'simulated_exec_12345'
|
||||
|
||||
|
||||
class SimulatedContainer:
|
||||
"""Simulated Docker container for testing without Docker"""
|
||||
|
||||
def __init__(self):
|
||||
self.id = 'simulated_container_12345'
|
||||
self.name = 'test_simulated_container'
|
||||
self.status = 'running'
|
||||
|
||||
def exec_run(self, cmd, **kwargs):
|
||||
"""Simulate exec_run that returns a socket-like object"""
|
||||
return SimulatedExecInstance()
|
||||
|
||||
def stop(self, timeout=10):
|
||||
"""Simulate stopping the container"""
|
||||
self.status = 'stopped'
|
||||
|
||||
def remove(self):
|
||||
"""Simulate removing the container"""
|
||||
pass
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def simulated_container():
|
||||
"""Provide a simulated container for testing without Docker"""
|
||||
return SimulatedContainer()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def test_container_or_simulated():
|
||||
"""
|
||||
Provide either a real Docker container or simulated one.
|
||||
Use real container if Docker is available, otherwise use simulated.
|
||||
"""
|
||||
if docker_available():
|
||||
import docker
|
||||
import time
|
||||
|
||||
client = docker.from_env()
|
||||
|
||||
# Pull alpine image if not present
|
||||
try:
|
||||
client.images.get('alpine:latest')
|
||||
except docker.errors.ImageNotFound:
|
||||
client.images.pull('alpine:latest')
|
||||
|
||||
# Create and start container
|
||||
container = client.containers.run(
|
||||
'alpine:latest',
|
||||
command='sleep 300',
|
||||
detach=True,
|
||||
remove=True,
|
||||
name='pytest_test_container'
|
||||
)
|
||||
|
||||
time.sleep(1)
|
||||
|
||||
yield container
|
||||
|
||||
# Cleanup
|
||||
try:
|
||||
container.stop(timeout=1)
|
||||
except:
|
||||
pass
|
||||
else:
|
||||
# Use simulated container
|
||||
yield SimulatedContainer()
|
||||
69
backend/tests/test_auth.py
Normal file
69
backend/tests/test_auth.py
Normal file
@@ -0,0 +1,69 @@
|
||||
import pytest
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
class TestAuthentication:
|
||||
"""Test authentication endpoints"""
|
||||
|
||||
def test_login_success(self, client):
|
||||
"""Test successful login"""
|
||||
response = client.post('/api/auth/login', json={
|
||||
'username': 'admin',
|
||||
'password': 'admin123'
|
||||
})
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['success'] is True
|
||||
assert 'token' in data
|
||||
assert data['username'] == 'admin'
|
||||
|
||||
def test_login_invalid_credentials(self, client):
|
||||
"""Test login with invalid credentials"""
|
||||
response = client.post('/api/auth/login', json={
|
||||
'username': 'admin',
|
||||
'password': 'wrongpassword'
|
||||
})
|
||||
|
||||
assert response.status_code == 401
|
||||
data = response.get_json()
|
||||
assert data['success'] is False
|
||||
assert 'message' in data
|
||||
|
||||
def test_login_missing_username(self, client):
|
||||
"""Test login with missing username"""
|
||||
response = client.post('/api/auth/login', json={
|
||||
'password': 'admin123'
|
||||
})
|
||||
|
||||
assert response.status_code == 401
|
||||
data = response.get_json()
|
||||
assert data['success'] is False
|
||||
|
||||
def test_login_missing_password(self, client):
|
||||
"""Test login with missing password"""
|
||||
response = client.post('/api/auth/login', json={
|
||||
'username': 'admin'
|
||||
})
|
||||
|
||||
assert response.status_code == 401
|
||||
data = response.get_json()
|
||||
assert data['success'] is False
|
||||
|
||||
def test_logout_success(self, client, auth_token):
|
||||
"""Test successful logout"""
|
||||
response = client.post('/api/auth/logout', headers={
|
||||
'Authorization': f'Bearer {auth_token}'
|
||||
})
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['success'] is True
|
||||
|
||||
def test_logout_without_token(self, client):
|
||||
"""Test logout without token"""
|
||||
response = client.post('/api/auth/logout')
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['success'] is True
|
||||
378
backend/tests/test_complete_coverage.py
Normal file
378
backend/tests/test_complete_coverage.py
Normal file
@@ -0,0 +1,378 @@
|
||||
"""Tests to achieve 100% code coverage."""
|
||||
import pytest
|
||||
import os
|
||||
import time
|
||||
from unittest.mock import MagicMock, patch, Mock, PropertyMock
|
||||
from flask_socketio import SocketIOTestClient
|
||||
|
||||
|
||||
class TestHandlerEdgeCases:
|
||||
"""Test edge cases in terminal handlers"""
|
||||
|
||||
@pytest.fixture
|
||||
def socketio_client(self, app):
|
||||
"""Create a SocketIO test client"""
|
||||
from app import socketio
|
||||
return socketio.test_client(app, namespace='/terminal')
|
||||
|
||||
def test_disconnect_handler_exception_during_cleanup(self):
|
||||
"""Test disconnect handler when exec.kill() raises exception"""
|
||||
from handlers.terminal.disconnect import handle_disconnect
|
||||
from config import active_terminals
|
||||
from flask import Flask
|
||||
|
||||
app = Flask(__name__)
|
||||
with app.test_request_context():
|
||||
with patch('handlers.terminal.disconnect.request') as mock_request:
|
||||
mock_request.sid = 'test_exception_sid'
|
||||
|
||||
# Create exec that raises exception on kill
|
||||
mock_exec = MagicMock()
|
||||
mock_exec.kill.side_effect = Exception("Kill failed")
|
||||
active_terminals['test_exception_sid'] = {'exec': mock_exec}
|
||||
|
||||
# Should not raise, just clean up
|
||||
handle_disconnect()
|
||||
assert 'test_exception_sid' not in active_terminals
|
||||
|
||||
def test_input_handler_no_active_terminal(self):
|
||||
"""Test input handler when no active terminal exists"""
|
||||
from handlers.terminal.input import handle_input
|
||||
from flask import Flask
|
||||
from flask_socketio import emit
|
||||
|
||||
app = Flask(__name__)
|
||||
with app.test_request_context():
|
||||
with patch('handlers.terminal.input.request') as mock_request:
|
||||
with patch('handlers.terminal.input.emit') as mock_emit:
|
||||
mock_request.sid = 'nonexistent_sid'
|
||||
|
||||
handle_input({'data': 'test'})
|
||||
|
||||
# Should emit error
|
||||
mock_emit.assert_called_once()
|
||||
args = mock_emit.call_args[0]
|
||||
assert args[0] == 'error'
|
||||
assert 'No active terminal session' in args[1]['error']
|
||||
|
||||
def test_input_handler_exception(self):
|
||||
"""Test input handler when sendall raises exception"""
|
||||
from handlers.terminal.input import handle_input
|
||||
from config import active_terminals
|
||||
from flask import Flask
|
||||
|
||||
app = Flask(__name__)
|
||||
with app.test_request_context():
|
||||
with patch('handlers.terminal.input.request') as mock_request:
|
||||
with patch('handlers.terminal.input.emit') as mock_emit:
|
||||
mock_request.sid = 'error_sid'
|
||||
|
||||
# Mock the _sock attribute which is checked first
|
||||
mock_inner_sock = MagicMock()
|
||||
mock_inner_sock.sendall.side_effect = Exception("Send failed")
|
||||
|
||||
mock_sock = MagicMock()
|
||||
mock_sock._sock = mock_inner_sock
|
||||
|
||||
mock_exec = MagicMock()
|
||||
mock_exec.output = mock_sock
|
||||
|
||||
active_terminals['error_sid'] = {'exec': mock_exec}
|
||||
|
||||
handle_input({'data': 'test'})
|
||||
|
||||
# Should emit error
|
||||
mock_emit.assert_called()
|
||||
error_call = [c for c in mock_emit.call_args_list if c[0][0] == 'error']
|
||||
assert len(error_call) > 0
|
||||
|
||||
def test_resize_handler_exception(self):
|
||||
"""Test resize handler when it raises exception"""
|
||||
from handlers.terminal.resize import handle_resize
|
||||
from config import active_terminals
|
||||
from flask import Flask
|
||||
|
||||
app = Flask(__name__)
|
||||
with app.test_request_context():
|
||||
with patch('handlers.terminal.resize.request') as mock_request:
|
||||
mock_request.sid = 'resize_error_sid'
|
||||
active_terminals['resize_error_sid'] = {'exec': MagicMock()}
|
||||
|
||||
# Force an exception by passing invalid data
|
||||
with patch('handlers.terminal.resize.logger') as mock_logger:
|
||||
# This should trigger the exception handler
|
||||
handle_resize(None) # None instead of dict
|
||||
|
||||
# Should have logged error
|
||||
assert mock_logger.error.called
|
||||
|
||||
|
||||
class TestDockerDiagnostics:
|
||||
"""Test docker diagnostics edge cases"""
|
||||
|
||||
@patch('os.path.exists')
|
||||
@patch('os.listdir')
|
||||
def test_diagnose_var_run_not_exists(self, mock_listdir, mock_exists):
|
||||
"""Test diagnostics when /var/run doesn't exist"""
|
||||
from utils.diagnostics.docker_env import diagnose_docker_environment
|
||||
|
||||
mock_exists.return_value = False
|
||||
|
||||
# Should not raise exception
|
||||
with patch('utils.diagnostics.docker_env.logger'):
|
||||
diagnose_docker_environment()
|
||||
|
||||
@patch('os.path.exists')
|
||||
@patch('os.listdir')
|
||||
def test_diagnose_var_run_error(self, mock_listdir, mock_exists):
|
||||
"""Test diagnostics when /var/run listing fails"""
|
||||
from utils.diagnostics.docker_env import diagnose_docker_environment
|
||||
|
||||
def exists_side_effect(path):
|
||||
if path == '/var/run':
|
||||
return True
|
||||
return False
|
||||
|
||||
mock_exists.side_effect = exists_side_effect
|
||||
mock_listdir.side_effect = Exception("Permission denied")
|
||||
|
||||
# Should handle exception
|
||||
with patch('utils.diagnostics.docker_env.logger'):
|
||||
diagnose_docker_environment()
|
||||
|
||||
@patch('os.path.exists')
|
||||
@patch('os.stat')
|
||||
@patch('os.access')
|
||||
@patch('os.getuid')
|
||||
@patch('os.getgid')
|
||||
@patch('os.getgroups')
|
||||
def test_diagnose_docker_socket_permissions(
|
||||
self, mock_getgroups, mock_getgid, mock_getuid,
|
||||
mock_access, mock_stat, mock_exists
|
||||
):
|
||||
"""Test diagnostics for docker socket with permissions check"""
|
||||
from utils.diagnostics.docker_env import diagnose_docker_environment
|
||||
import pwd
|
||||
import grp
|
||||
|
||||
def exists_side_effect(path):
|
||||
if path == '/var/run':
|
||||
return False
|
||||
if path == '/var/run/docker.sock':
|
||||
return True
|
||||
return False
|
||||
|
||||
mock_exists.side_effect = exists_side_effect
|
||||
|
||||
# Mock stat for socket
|
||||
mock_stat_result = MagicMock()
|
||||
mock_stat_result.st_mode = 0o666
|
||||
mock_stat_result.st_uid = 0
|
||||
mock_stat_result.st_gid = 0
|
||||
mock_stat.return_value = mock_stat_result
|
||||
|
||||
# Mock access - not readable/writable
|
||||
mock_access.return_value = False
|
||||
|
||||
# Mock user info
|
||||
mock_getuid.return_value = 0
|
||||
mock_getgid.return_value = 0
|
||||
mock_getgroups.return_value = [0, 1]
|
||||
|
||||
with patch('utils.diagnostics.docker_env.logger'):
|
||||
with patch('pwd.getpwuid') as mock_getpwuid:
|
||||
with patch('grp.getgrgid') as mock_getgrgid:
|
||||
mock_user = MagicMock()
|
||||
mock_user.pw_name = 'root'
|
||||
mock_getpwuid.return_value = mock_user
|
||||
|
||||
mock_group = MagicMock()
|
||||
mock_group.gr_name = 'root'
|
||||
mock_getgrgid.return_value = mock_group
|
||||
|
||||
diagnose_docker_environment()
|
||||
|
||||
@patch('os.path.exists')
|
||||
@patch('os.getuid')
|
||||
def test_diagnose_user_info_error(self, mock_getuid, mock_exists):
|
||||
"""Test diagnostics when user info lookup fails"""
|
||||
from utils.diagnostics.docker_env import diagnose_docker_environment
|
||||
|
||||
mock_exists.return_value = False
|
||||
mock_getuid.side_effect = Exception("No user info")
|
||||
|
||||
with patch('utils.diagnostics.docker_env.logger'):
|
||||
diagnose_docker_environment()
|
||||
|
||||
@patch('os.path.exists')
|
||||
@patch('os.getuid')
|
||||
@patch('os.getgid')
|
||||
@patch('os.getgroups')
|
||||
def test_diagnose_group_lookup_error(self, mock_getgroups, mock_getgid, mock_getuid, mock_exists):
|
||||
"""Test diagnostics when group lookup fails"""
|
||||
from utils.diagnostics.docker_env import diagnose_docker_environment
|
||||
import pwd
|
||||
import grp
|
||||
|
||||
mock_exists.return_value = False
|
||||
mock_getuid.return_value = 0
|
||||
mock_getgid.return_value = 0
|
||||
mock_getgroups.return_value = [999] # Non-existent group
|
||||
|
||||
with patch('utils.diagnostics.docker_env.logger'):
|
||||
with patch('pwd.getpwuid') as mock_getpwuid:
|
||||
with patch('grp.getgrgid') as mock_getgrgid:
|
||||
mock_user = MagicMock()
|
||||
mock_user.pw_name = 'test'
|
||||
mock_getpwuid.return_value = mock_user
|
||||
|
||||
# Make group lookup fail
|
||||
mock_getgrgid.side_effect = KeyError("Group not found")
|
||||
|
||||
diagnose_docker_environment()
|
||||
|
||||
|
||||
class TestDockerClientEdgeCases:
|
||||
"""Test docker client edge cases"""
|
||||
|
||||
@patch('docker.from_env')
|
||||
@patch('docker.DockerClient')
|
||||
def test_get_docker_client_unexpected_error(self, mock_docker_client, mock_from_env):
|
||||
"""Test get_docker_client with unexpected error"""
|
||||
from utils.docker_client import get_docker_client
|
||||
|
||||
# Make both methods raise unexpected errors
|
||||
mock_from_env.side_effect = RuntimeError("Unexpected error")
|
||||
mock_docker_client.side_effect = RuntimeError("Unexpected error")
|
||||
|
||||
with patch('utils.docker_client.diagnose_docker_environment'):
|
||||
client = get_docker_client()
|
||||
assert client is None
|
||||
|
||||
|
||||
class TestExecHelpersEdgeCases:
|
||||
"""Test exec helpers edge cases"""
|
||||
|
||||
def test_decode_output_empty(self):
|
||||
"""Test decode_output with empty output"""
|
||||
from utils.exec_helpers import decode_output
|
||||
|
||||
mock_exec = MagicMock()
|
||||
mock_exec.output = None
|
||||
|
||||
result = decode_output(mock_exec)
|
||||
assert result == ''
|
||||
|
||||
def test_decode_output_latin1_fallback(self):
|
||||
"""Test decode_output falls back to latin-1"""
|
||||
from utils.exec_helpers import decode_output
|
||||
|
||||
mock_exec = MagicMock()
|
||||
# Create invalid UTF-8 that will force latin-1 fallback
|
||||
mock_exec.output = bytes([0xff, 0xfe, 0xfd])
|
||||
|
||||
result = decode_output(mock_exec)
|
||||
assert isinstance(result, str)
|
||||
|
||||
def test_extract_workdir_cd_command(self):
|
||||
"""Test extract_workdir with cd command"""
|
||||
from utils.exec_helpers import extract_workdir
|
||||
|
||||
output = "/home/user"
|
||||
result_output, result_workdir = extract_workdir(output, "/app", True)
|
||||
|
||||
assert result_output == ''
|
||||
assert result_workdir == "/home/user"
|
||||
|
||||
|
||||
class TestTerminalHelpersEdgeCases:
|
||||
"""Test terminal helpers edge cases"""
|
||||
|
||||
@patch('utils.terminal_helpers.threading.Thread')
|
||||
def test_create_output_reader_unicode_decode_error(self, mock_thread):
|
||||
"""Test output reader handles unicode decode errors"""
|
||||
from utils.terminal_helpers import create_output_reader
|
||||
from config import active_terminals
|
||||
|
||||
mock_socketio = MagicMock()
|
||||
mock_sock = MagicMock()
|
||||
|
||||
# Return invalid UTF-8, then empty to end loop
|
||||
mock_sock.recv.side_effect = [
|
||||
bytes([0x80, 0x81]), # Invalid UTF-8
|
||||
b'' # EOF
|
||||
]
|
||||
mock_sock.close = MagicMock()
|
||||
|
||||
mock_exec = MagicMock()
|
||||
mock_exec.output = mock_sock
|
||||
|
||||
sid = 'unicode_test_sid'
|
||||
active_terminals[sid] = {'exec': mock_exec}
|
||||
|
||||
# Get the actual thread function that would be called
|
||||
def capture_thread_target(*args, **kwargs):
|
||||
# Run the target function
|
||||
kwargs['target']()
|
||||
return MagicMock()
|
||||
|
||||
mock_thread.side_effect = capture_thread_target
|
||||
|
||||
create_output_reader(mock_socketio, sid, mock_exec)
|
||||
|
||||
# Should have emitted with latin-1 decoded data
|
||||
assert mock_socketio.emit.called
|
||||
|
||||
@patch('utils.terminal_helpers.threading.Thread')
|
||||
def test_create_output_reader_socket_recv_error(self, mock_thread):
|
||||
"""Test output reader handles recv errors"""
|
||||
from utils.terminal_helpers import create_output_reader
|
||||
from config import active_terminals
|
||||
|
||||
mock_socketio = MagicMock()
|
||||
mock_sock = MagicMock()
|
||||
mock_sock.recv.side_effect = Exception("Socket error")
|
||||
mock_sock.close = MagicMock()
|
||||
|
||||
mock_exec = MagicMock()
|
||||
mock_exec.output = mock_sock
|
||||
|
||||
sid = 'socket_error_sid'
|
||||
active_terminals[sid] = {'exec': mock_exec}
|
||||
|
||||
def capture_thread_target(*args, **kwargs):
|
||||
kwargs['target']()
|
||||
return MagicMock()
|
||||
|
||||
mock_thread.side_effect = capture_thread_target
|
||||
|
||||
create_output_reader(mock_socketio, sid, mock_exec)
|
||||
|
||||
# Should have cleaned up
|
||||
assert sid not in active_terminals
|
||||
|
||||
@patch('utils.terminal_helpers.threading.Thread')
|
||||
def test_create_output_reader_socket_close_error(self, mock_thread):
|
||||
"""Test output reader handles close errors"""
|
||||
from utils.terminal_helpers import create_output_reader
|
||||
from config import active_terminals
|
||||
|
||||
mock_socketio = MagicMock()
|
||||
mock_sock = MagicMock()
|
||||
mock_sock.recv.return_value = b'' # EOF
|
||||
mock_sock.close.side_effect = Exception("Close failed")
|
||||
|
||||
mock_exec = MagicMock()
|
||||
mock_exec.output = mock_sock
|
||||
|
||||
sid = 'close_error_sid'
|
||||
active_terminals[sid] = {'exec': mock_exec}
|
||||
|
||||
def capture_thread_target(*args, **kwargs):
|
||||
kwargs['target']()
|
||||
return MagicMock()
|
||||
|
||||
mock_thread.side_effect = capture_thread_target
|
||||
|
||||
# Should not raise exception
|
||||
create_output_reader(mock_socketio, sid, mock_exec)
|
||||
124
backend/tests/test_containers.py
Normal file
124
backend/tests/test_containers.py
Normal file
@@ -0,0 +1,124 @@
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
|
||||
class TestContainerEndpoints:
|
||||
"""Test container management endpoints"""
|
||||
|
||||
def test_get_containers_unauthorized(self, client):
|
||||
"""Test getting containers without auth"""
|
||||
response = client.get('/api/containers')
|
||||
assert response.status_code == 401
|
||||
data = response.get_json()
|
||||
assert 'error' in data
|
||||
|
||||
def test_get_containers_invalid_token(self, client):
|
||||
"""Test getting containers with invalid token"""
|
||||
response = client.get('/api/containers', headers={
|
||||
'Authorization': 'Bearer invalid_token'
|
||||
})
|
||||
assert response.status_code == 401
|
||||
data = response.get_json()
|
||||
assert 'error' in data
|
||||
|
||||
@patch('routes.containers.list.get_docker_client')
|
||||
def test_get_containers_success(self, mock_get_client, client, auth_headers):
|
||||
"""Test getting containers successfully"""
|
||||
# Mock Docker client
|
||||
mock_container = MagicMock()
|
||||
mock_container.short_id = 'abc123'
|
||||
mock_container.name = 'test-container'
|
||||
mock_container.status = 'running'
|
||||
mock_container.image.tags = ['nginx:latest']
|
||||
mock_container.attrs = {'Created': '2024-01-01T00:00:00.000000000Z'}
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.list.return_value = [mock_container]
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.get('/api/containers', headers=auth_headers)
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert 'containers' in data
|
||||
assert len(data['containers']) == 1
|
||||
assert data['containers'][0]['id'] == 'abc123'
|
||||
assert data['containers'][0]['name'] == 'test-container'
|
||||
|
||||
@patch('routes.containers.list.get_docker_client')
|
||||
def test_get_containers_docker_unavailable(self, mock_get_client, client, auth_headers):
|
||||
"""Test getting containers when Docker is unavailable"""
|
||||
mock_get_client.return_value = None
|
||||
|
||||
response = client.get('/api/containers', headers=auth_headers)
|
||||
assert response.status_code == 500
|
||||
data = response.get_json()
|
||||
assert 'error' in data
|
||||
|
||||
@patch('utils.container_helpers.get_docker_client')
|
||||
def test_start_container_success(self, mock_get_client, client, auth_headers):
|
||||
"""Test starting a container"""
|
||||
mock_container = MagicMock()
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/start', headers=auth_headers)
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['success'] is True
|
||||
mock_container.start.assert_called_once()
|
||||
|
||||
@patch('utils.container_helpers.get_docker_client')
|
||||
def test_stop_container_success(self, mock_get_client, client, auth_headers):
|
||||
"""Test stopping a container"""
|
||||
mock_container = MagicMock()
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/stop', headers=auth_headers)
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['success'] is True
|
||||
mock_container.stop.assert_called_once()
|
||||
|
||||
@patch('utils.container_helpers.get_docker_client')
|
||||
def test_restart_container_success(self, mock_get_client, client, auth_headers):
|
||||
"""Test restarting a container"""
|
||||
mock_container = MagicMock()
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/restart', headers=auth_headers)
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['success'] is True
|
||||
mock_container.restart.assert_called_once()
|
||||
|
||||
@patch('utils.container_helpers.get_docker_client')
|
||||
def test_remove_container_success(self, mock_get_client, client, auth_headers):
|
||||
"""Test removing a container"""
|
||||
mock_container = MagicMock()
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.delete('/api/containers/abc123', headers=auth_headers)
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['success'] is True
|
||||
mock_container.remove.assert_called_once_with(force=True)
|
||||
|
||||
def test_container_operations_unauthorized(self, client):
|
||||
"""Test container operations without auth"""
|
||||
endpoints = [
|
||||
('/api/containers/abc123/start', 'post'),
|
||||
('/api/containers/abc123/stop', 'post'),
|
||||
('/api/containers/abc123/restart', 'post'),
|
||||
('/api/containers/abc123', 'delete'),
|
||||
]
|
||||
|
||||
for endpoint, method in endpoints:
|
||||
response = getattr(client, method)(endpoint)
|
||||
assert response.status_code == 401
|
||||
156
backend/tests/test_coverage_boost.py
Normal file
156
backend/tests/test_coverage_boost.py
Normal file
@@ -0,0 +1,156 @@
|
||||
"""Tests to boost coverage to 100%."""
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, patch, Mock
|
||||
from flask import jsonify
|
||||
|
||||
|
||||
class TestContainerExceptionHandling:
|
||||
"""Test exception handling in container routes"""
|
||||
|
||||
@patch('utils.container_helpers.get_docker_client')
|
||||
def test_start_container_exception(self, mock_get_client, client, auth_headers):
|
||||
"""Test start container with exception"""
|
||||
mock_container = MagicMock()
|
||||
mock_container.start.side_effect = Exception("Container failed to start")
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/test123/start', headers=auth_headers)
|
||||
assert response.status_code == 500
|
||||
data = response.get_json()
|
||||
assert 'error' in data
|
||||
|
||||
@patch('utils.container_helpers.get_docker_client')
|
||||
def test_stop_container_exception(self, mock_get_client, client, auth_headers):
|
||||
"""Test stop container with exception"""
|
||||
mock_container = MagicMock()
|
||||
mock_container.stop.side_effect = Exception("Container failed to stop")
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/test123/stop', headers=auth_headers)
|
||||
assert response.status_code == 500
|
||||
data = response.get_json()
|
||||
assert 'error' in data
|
||||
|
||||
@patch('utils.container_helpers.get_docker_client')
|
||||
def test_restart_container_exception(self, mock_get_client, client, auth_headers):
|
||||
"""Test restart container with exception"""
|
||||
mock_container = MagicMock()
|
||||
mock_container.restart.side_effect = Exception("Container failed to restart")
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/test123/restart', headers=auth_headers)
|
||||
assert response.status_code == 500
|
||||
data = response.get_json()
|
||||
assert 'error' in data
|
||||
|
||||
@patch('utils.container_helpers.get_docker_client')
|
||||
def test_remove_container_exception(self, mock_get_client, client, auth_headers):
|
||||
"""Test remove container with exception"""
|
||||
mock_container = MagicMock()
|
||||
mock_container.remove.side_effect = Exception("Container failed to remove")
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.delete('/api/containers/test123', headers=auth_headers)
|
||||
assert response.status_code == 500
|
||||
data = response.get_json()
|
||||
assert 'error' in data
|
||||
|
||||
@patch('routes.containers.list.get_docker_client')
|
||||
def test_list_containers_exception(self, mock_get_client, client, auth_headers):
|
||||
"""Test list containers with exception"""
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.list.side_effect = Exception("Failed to list containers")
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.get('/api/containers', headers=auth_headers)
|
||||
assert response.status_code == 500
|
||||
data = response.get_json()
|
||||
assert 'error' in data
|
||||
|
||||
|
||||
class TestContainerHelpers:
|
||||
"""Test container_helpers exception handling"""
|
||||
|
||||
@patch('utils.container_helpers.get_docker_client')
|
||||
def test_get_auth_and_container_exception(self, mock_get_client):
|
||||
"""Test get_auth_and_container when container.get raises exception"""
|
||||
from utils.container_helpers import get_auth_and_container
|
||||
from config import sessions
|
||||
|
||||
# Create a valid session
|
||||
token = 'test_token_123'
|
||||
sessions[token] = {'username': 'test'}
|
||||
|
||||
# Mock client that raises exception
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.side_effect = Exception("Container not found")
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
# This test needs to be called in request context
|
||||
from flask import Flask
|
||||
app = Flask(__name__)
|
||||
|
||||
with app.test_request_context(headers={'Authorization': f'Bearer {token}'}):
|
||||
container, error = get_auth_and_container('test123')
|
||||
assert container is None
|
||||
assert error is not None
|
||||
assert error[1] == 500
|
||||
|
||||
|
||||
class TestExecHelpers:
|
||||
"""Test exec_helpers edge cases"""
|
||||
|
||||
def test_decode_output_unicode_error(self):
|
||||
"""Test decode_output with invalid UTF-8"""
|
||||
from utils.exec_helpers import decode_output
|
||||
|
||||
mock_exec = MagicMock()
|
||||
# Invalid UTF-8 sequence
|
||||
mock_exec.output = b'\x80\x81\x82\x83'
|
||||
|
||||
result = decode_output(mock_exec)
|
||||
# Should fallback to latin-1
|
||||
assert result is not None
|
||||
assert isinstance(result, str)
|
||||
|
||||
def test_extract_workdir_no_marker(self):
|
||||
"""Test extract_workdir when no marker present"""
|
||||
from utils.exec_helpers import extract_workdir
|
||||
|
||||
output = "some command output"
|
||||
current_workdir = "/test"
|
||||
result_output, result_workdir = extract_workdir(output, current_workdir, False)
|
||||
|
||||
assert result_output == output
|
||||
assert result_workdir == current_workdir
|
||||
|
||||
def test_execute_command_bash_fallback(self):
|
||||
"""Test execute_command_with_fallback when bash fails"""
|
||||
from utils.exec_helpers import execute_command_with_fallback
|
||||
|
||||
mock_container = MagicMock()
|
||||
# Make bash fail, sh succeed
|
||||
mock_container.exec_run.side_effect = [
|
||||
Exception("bash not found"),
|
||||
MagicMock(output=b'success', exit_code=0)
|
||||
]
|
||||
|
||||
result = execute_command_with_fallback(
|
||||
mock_container, '/app', 'ls', False
|
||||
)
|
||||
assert result.exit_code == 0
|
||||
assert mock_container.exec_run.call_count == 2
|
||||
|
||||
|
||||
93
backend/tests/test_docker_client.py
Normal file
93
backend/tests/test_docker_client.py
Normal file
@@ -0,0 +1,93 @@
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, patch
|
||||
import docker
|
||||
|
||||
|
||||
class TestDockerClient:
|
||||
"""Test Docker client connection logic"""
|
||||
|
||||
@patch('docker.from_env')
|
||||
def test_get_docker_client_success(self, mock_from_env):
|
||||
"""Test successful Docker client connection"""
|
||||
from utils.docker_client import get_docker_client
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.ping.return_value = True
|
||||
mock_from_env.return_value = mock_client
|
||||
|
||||
client = get_docker_client()
|
||||
assert client is not None
|
||||
mock_client.ping.assert_called_once()
|
||||
|
||||
@patch('docker.DockerClient')
|
||||
@patch('docker.from_env')
|
||||
def test_get_docker_client_fallback_to_socket(self, mock_from_env, mock_docker_client):
|
||||
"""Test fallback to Unix socket when from_env fails"""
|
||||
from utils.docker_client import get_docker_client
|
||||
|
||||
# Make from_env fail
|
||||
mock_from_env.side_effect = Exception("Connection failed")
|
||||
|
||||
# Make socket connection succeed
|
||||
mock_client = MagicMock()
|
||||
mock_client.ping.return_value = True
|
||||
mock_docker_client.return_value = mock_client
|
||||
|
||||
client = get_docker_client()
|
||||
assert client is not None
|
||||
mock_docker_client.assert_called_with(base_url='unix:///var/run/docker.sock')
|
||||
|
||||
@patch('docker.DockerClient')
|
||||
@patch('docker.from_env')
|
||||
def test_get_docker_client_all_methods_fail(self, mock_from_env, mock_docker_client):
|
||||
"""Test when all Docker connection methods fail"""
|
||||
from utils.docker_client import get_docker_client
|
||||
|
||||
# Make both methods fail
|
||||
mock_from_env.side_effect = Exception("from_env failed")
|
||||
mock_docker_client.side_effect = Exception("socket failed")
|
||||
|
||||
client = get_docker_client()
|
||||
assert client is None
|
||||
|
||||
|
||||
class TestFormatUptime:
|
||||
"""Test uptime formatting edge cases"""
|
||||
|
||||
def test_format_uptime_zero_minutes(self):
|
||||
"""Test formatting for containers just started"""
|
||||
from utils.formatters import format_uptime
|
||||
from datetime import datetime, timezone, timedelta
|
||||
|
||||
now = datetime.now(timezone.utc)
|
||||
created_at = now - timedelta(seconds=30)
|
||||
created_str = created_at.isoformat().replace('+00:00', 'Z')
|
||||
|
||||
result = format_uptime(created_str)
|
||||
# Should show 0m
|
||||
assert 'm' in result
|
||||
|
||||
def test_format_uptime_exactly_one_day(self):
|
||||
"""Test formatting for exactly 1 day"""
|
||||
from utils.formatters import format_uptime
|
||||
from datetime import datetime, timezone, timedelta
|
||||
|
||||
now = datetime.now(timezone.utc)
|
||||
created_at = now - timedelta(days=1)
|
||||
created_str = created_at.isoformat().replace('+00:00', 'Z')
|
||||
|
||||
result = format_uptime(created_str)
|
||||
assert '1d' in result
|
||||
|
||||
def test_format_uptime_many_days(self):
|
||||
"""Test formatting for many days"""
|
||||
from utils.formatters import format_uptime
|
||||
from datetime import datetime, timezone, timedelta
|
||||
|
||||
now = datetime.now(timezone.utc)
|
||||
created_at = now - timedelta(days=30, hours=5)
|
||||
created_str = created_at.isoformat().replace('+00:00', 'Z')
|
||||
|
||||
result = format_uptime(created_str)
|
||||
assert 'd' in result
|
||||
assert 'h' in result
|
||||
134
backend/tests/test_edge_cases.py
Normal file
134
backend/tests/test_edge_cases.py
Normal file
@@ -0,0 +1,134 @@
|
||||
"""
|
||||
Edge case tests to improve overall coverage.
|
||||
"""
|
||||
import pytest
|
||||
from unittest.mock import patch, MagicMock
|
||||
|
||||
|
||||
pytestmark = pytest.mark.unit
|
||||
|
||||
|
||||
class TestEdgeCases:
|
||||
"""Additional edge case tests"""
|
||||
|
||||
def test_logout_with_invalid_token_format(self, client):
|
||||
"""Test logout with malformed token"""
|
||||
response = client.post('/api/auth/logout', headers={
|
||||
'Authorization': 'InvalidFormat'
|
||||
})
|
||||
# Should handle gracefully
|
||||
assert response.status_code in [200, 401, 400]
|
||||
|
||||
def test_logout_with_empty_bearer(self, client):
|
||||
"""Test logout with empty bearer token"""
|
||||
response = client.post('/api/auth/logout', headers={
|
||||
'Authorization': 'Bearer '
|
||||
})
|
||||
assert response.status_code in [200, 401]
|
||||
|
||||
@patch('utils.docker_client.get_docker_client')
|
||||
def test_containers_with_docker_error(self, mock_get_client, client, auth_headers):
|
||||
"""Test containers endpoint when Docker returns unexpected error"""
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.list.side_effect = Exception("Unexpected Docker error")
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.get('/api/containers', headers=auth_headers)
|
||||
|
||||
# Should return 500 or handle error
|
||||
assert response.status_code in [500, 200]
|
||||
|
||||
@patch('utils.docker_client.get_docker_client')
|
||||
def test_exec_with_missing_fields(self, mock_get_client, client, auth_headers):
|
||||
"""Test exec with missing command field"""
|
||||
mock_get_client.return_value = MagicMock()
|
||||
|
||||
response = client.post('/api/containers/test_container/exec',
|
||||
headers=auth_headers,
|
||||
json={}) # Missing command
|
||||
|
||||
# Should return 400 or handle error
|
||||
assert response.status_code in [400, 500]
|
||||
|
||||
@patch('utils.docker_client.get_docker_client')
|
||||
def test_start_container_not_found(self, mock_get_client, client, auth_headers):
|
||||
"""Test starting non-existent container"""
|
||||
from docker.errors import NotFound
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.side_effect = NotFound("Container not found")
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/nonexistent/start',
|
||||
headers=auth_headers)
|
||||
|
||||
assert response.status_code in [404, 500]
|
||||
|
||||
@patch('utils.docker_client.get_docker_client')
|
||||
def test_stop_container_error(self, mock_get_client, client, auth_headers):
|
||||
"""Test stopping container with error"""
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_container.stop.side_effect = Exception("Stop failed")
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/test_container/stop',
|
||||
headers=auth_headers)
|
||||
|
||||
assert response.status_code in [500, 200]
|
||||
|
||||
@patch('utils.docker_client.get_docker_client')
|
||||
def test_restart_container_error(self, mock_get_client, client, auth_headers):
|
||||
"""Test restarting container with error"""
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_container.restart.side_effect = Exception("Restart failed")
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/test_container/restart',
|
||||
headers=auth_headers)
|
||||
|
||||
assert response.status_code in [500, 200]
|
||||
|
||||
@patch('utils.docker_client.get_docker_client')
|
||||
def test_remove_container_error(self, mock_get_client, client, auth_headers):
|
||||
"""Test removing container with error"""
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_container.remove.side_effect = Exception("Remove failed")
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.delete('/api/containers/test_container',
|
||||
headers=auth_headers)
|
||||
|
||||
assert response.status_code in [500, 200]
|
||||
|
||||
def test_login_with_empty_body(self, client):
|
||||
"""Test login with empty request body"""
|
||||
response = client.post('/api/auth/login', json={})
|
||||
|
||||
assert response.status_code in [400, 401]
|
||||
|
||||
def test_login_with_none_values(self, client):
|
||||
"""Test login with null username/password"""
|
||||
response = client.post('/api/auth/login', json={
|
||||
'username': None,
|
||||
'password': None
|
||||
})
|
||||
|
||||
assert response.status_code in [400, 401]
|
||||
|
||||
@patch('utils.docker_client.get_docker_client')
|
||||
def test_exec_with_empty_command(self, mock_get_client, client, auth_headers):
|
||||
"""Test exec with empty command string"""
|
||||
mock_get_client.return_value = MagicMock()
|
||||
|
||||
response = client.post('/api/containers/test_container/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': ''})
|
||||
|
||||
# Should handle empty command
|
||||
assert response.status_code in [400, 500, 200]
|
||||
124
backend/tests/test_exec.py
Normal file
124
backend/tests/test_exec.py
Normal file
@@ -0,0 +1,124 @@
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
|
||||
class TestContainerExec:
|
||||
"""Test container command execution"""
|
||||
|
||||
def test_exec_unauthorized(self, client):
|
||||
"""Test exec without auth"""
|
||||
response = client.post('/api/containers/abc123/exec', json={
|
||||
'command': 'ls'
|
||||
})
|
||||
assert response.status_code == 401
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_simple_command(self, mock_get_client, client, auth_headers, auth_token):
|
||||
"""Test executing a simple command"""
|
||||
# Mock exec result
|
||||
mock_exec_result = MagicMock()
|
||||
mock_exec_result.output = b'file1.txt\nfile2.txt\n::WORKDIR::/app'
|
||||
mock_exec_result.exit_code = 0
|
||||
|
||||
mock_container = MagicMock()
|
||||
mock_container.exec_run.return_value = mock_exec_result
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'ls'})
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['exit_code'] == 0
|
||||
assert 'file1.txt' in data['output']
|
||||
assert data['workdir'] == '/app'
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_cd_command(self, mock_get_client, client, auth_headers, auth_token):
|
||||
"""Test executing cd command"""
|
||||
# Mock exec result for cd command
|
||||
mock_exec_result = MagicMock()
|
||||
mock_exec_result.output = b'/home/user\n'
|
||||
mock_exec_result.exit_code = 0
|
||||
|
||||
mock_container = MagicMock()
|
||||
mock_container.exec_run.return_value = mock_exec_result
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'cd /home/user'})
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['exit_code'] == 0
|
||||
assert data['workdir'] == '/home/user'
|
||||
assert data['output'] == ''
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_command_with_error(self, mock_get_client, client, auth_headers, auth_token):
|
||||
"""Test executing a command that fails"""
|
||||
# Mock exec result with error
|
||||
mock_exec_result = MagicMock()
|
||||
mock_exec_result.output = b'command not found::WORKDIR::/app'
|
||||
mock_exec_result.exit_code = 127
|
||||
|
||||
mock_container = MagicMock()
|
||||
mock_container.exec_run.return_value = mock_exec_result
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'invalidcommand'})
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['exit_code'] == 127
|
||||
assert 'command not found' in data['output']
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_docker_unavailable(self, mock_get_client, client, auth_headers):
|
||||
"""Test exec when Docker is unavailable"""
|
||||
mock_get_client.return_value = None
|
||||
|
||||
response = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'ls'})
|
||||
|
||||
assert response.status_code == 500
|
||||
data = response.get_json()
|
||||
assert 'error' in data
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_unicode_handling(self, mock_get_client, client, auth_headers, auth_token):
|
||||
"""Test exec with unicode output"""
|
||||
# Mock exec result with unicode
|
||||
mock_exec_result = MagicMock()
|
||||
mock_exec_result.output = 'Hello 世界\n::WORKDIR::/app'.encode('utf-8')
|
||||
mock_exec_result.exit_code = 0
|
||||
|
||||
mock_container = MagicMock()
|
||||
mock_container.exec_run.return_value = mock_exec_result
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'echo "Hello 世界"'})
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['exit_code'] == 0
|
||||
assert '世界' in data['output']
|
||||
171
backend/tests/test_exec_advanced.py
Normal file
171
backend/tests/test_exec_advanced.py
Normal file
@@ -0,0 +1,171 @@
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
|
||||
class TestExecAdvanced:
|
||||
"""Advanced tests for command execution"""
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_bash_fallback_to_sh(self, mock_get_client, client, auth_headers, auth_token):
|
||||
"""Test fallback from bash to sh when bash doesn't exist"""
|
||||
# Mock exec that fails for bash but succeeds for sh
|
||||
mock_bash_result = MagicMock()
|
||||
mock_sh_result = MagicMock()
|
||||
mock_sh_result.output = b'output from sh::WORKDIR::/app'
|
||||
mock_sh_result.exit_code = 0
|
||||
|
||||
mock_container = MagicMock()
|
||||
# First call (bash) raises exception, second call (sh) succeeds
|
||||
mock_container.exec_run.side_effect = [
|
||||
Exception("bash not found"),
|
||||
mock_sh_result
|
||||
]
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'ls'})
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['exit_code'] == 0
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_container_not_found(self, mock_get_client, client, auth_headers):
|
||||
"""Test exec on non-existent container"""
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.side_effect = Exception("Container not found")
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'ls'})
|
||||
|
||||
assert response.status_code == 500
|
||||
data = response.get_json()
|
||||
assert 'error' in data
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_preserves_working_directory(self, mock_get_client, client, auth_headers, auth_token):
|
||||
"""Test that working directory is preserved across commands"""
|
||||
mock_exec_result = MagicMock()
|
||||
mock_exec_result.output = b'::WORKDIR::/home/user'
|
||||
mock_exec_result.exit_code = 0
|
||||
|
||||
mock_container = MagicMock()
|
||||
mock_container.exec_run.return_value = mock_exec_result
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
# First command
|
||||
response1 = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'pwd'})
|
||||
assert response1.status_code == 200
|
||||
data1 = response1.get_json()
|
||||
assert data1['workdir'] == '/home/user'
|
||||
|
||||
# Second command should use the same session workdir
|
||||
response2 = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'ls'})
|
||||
assert response2.status_code == 200
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_cd_with_tilde(self, mock_get_client, client, auth_headers, auth_token):
|
||||
"""Test cd command with tilde expansion"""
|
||||
mock_exec_result = MagicMock()
|
||||
mock_exec_result.output = b'/home/user\n'
|
||||
mock_exec_result.exit_code = 0
|
||||
|
||||
mock_container = MagicMock()
|
||||
mock_container.exec_run.return_value = mock_exec_result
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'cd ~'})
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['workdir'] == '/home/user'
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_cd_no_args(self, mock_get_client, client, auth_headers, auth_token):
|
||||
"""Test cd command without arguments (should go to home)"""
|
||||
mock_exec_result = MagicMock()
|
||||
mock_exec_result.output = b'/root\n::WORKDIR::/'
|
||||
mock_exec_result.exit_code = 0
|
||||
|
||||
mock_container = MagicMock()
|
||||
mock_container.exec_run.return_value = mock_exec_result
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'cd'})
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
# 'cd' alone doesn't match 'cd ' pattern, so executes as regular command
|
||||
# workdir should be extracted from ::WORKDIR:: marker
|
||||
assert data['workdir'] == '/'
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_latin1_encoding_fallback(self, mock_get_client, client, auth_headers, auth_token):
|
||||
"""Test fallback to latin-1 encoding for non-UTF-8 output"""
|
||||
# Create binary data that's not valid UTF-8
|
||||
invalid_utf8 = b'\xff\xfe Invalid UTF-8 \x80::WORKDIR::/app'
|
||||
|
||||
mock_exec_result = MagicMock()
|
||||
mock_exec_result.output = invalid_utf8
|
||||
mock_exec_result.exit_code = 0
|
||||
|
||||
mock_container = MagicMock()
|
||||
mock_container.exec_run.return_value = mock_exec_result
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
response = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={'command': 'cat binary_file'})
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
# Should succeed with latin-1 fallback
|
||||
assert data['exit_code'] == 0
|
||||
assert 'output' in data
|
||||
|
||||
@patch('routes.containers.exec.get_docker_client')
|
||||
def test_exec_empty_command(self, mock_get_client, client, auth_headers, auth_token):
|
||||
"""Test exec with empty/no command"""
|
||||
mock_exec_result = MagicMock()
|
||||
mock_exec_result.output = b'No command provided::WORKDIR::/'
|
||||
mock_exec_result.exit_code = 0
|
||||
|
||||
mock_container = MagicMock()
|
||||
mock_container.exec_run.return_value = mock_exec_result
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
# Don't provide command
|
||||
response = client.post('/api/containers/abc123/exec',
|
||||
headers=auth_headers,
|
||||
json={})
|
||||
|
||||
assert response.status_code == 200
|
||||
262
backend/tests/test_final_coverage.py
Normal file
262
backend/tests/test_final_coverage.py
Normal file
@@ -0,0 +1,262 @@
|
||||
"""Tests for final 100% coverage."""
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, patch, Mock, PropertyMock
|
||||
|
||||
|
||||
class TestRemainingHandlerCoverage:
|
||||
"""Test remaining handler edge cases"""
|
||||
|
||||
def test_resize_with_active_terminal(self):
|
||||
"""Test resize handler with active terminal"""
|
||||
from handlers.terminal.resize import handle_resize
|
||||
from config import active_terminals
|
||||
from flask import Flask
|
||||
|
||||
app = Flask(__name__)
|
||||
with app.test_request_context():
|
||||
with patch('handlers.terminal.resize.request') as mock_request:
|
||||
with patch('handlers.terminal.resize.logger') as mock_logger:
|
||||
mock_request.sid = 'resize_sid'
|
||||
active_terminals['resize_sid'] = {'exec': MagicMock()}
|
||||
|
||||
handle_resize({'cols': 120, 'rows': 40})
|
||||
|
||||
# Should log the resize request
|
||||
mock_logger.info.assert_called()
|
||||
# Clean up
|
||||
del active_terminals['resize_sid']
|
||||
|
||||
|
||||
class TestDockerClientOuterException:
|
||||
"""Test docker client outer exception handler"""
|
||||
|
||||
@patch('utils.docker_client.docker.from_env')
|
||||
@patch('utils.docker_client.docker.DockerClient')
|
||||
@patch('utils.docker_client.diagnose_docker_environment')
|
||||
def test_get_docker_client_outer_exception(self, mock_diagnose, mock_docker_client, mock_from_env):
|
||||
"""Test get_docker_client when outer try block catches exception"""
|
||||
from utils.docker_client import get_docker_client
|
||||
|
||||
# Make the initial logger.info call raise an exception
|
||||
with patch('utils.docker_client.logger') as mock_logger:
|
||||
# Raise exception on the first logger.info call
|
||||
mock_logger.info.side_effect = Exception("Unexpected logger error")
|
||||
|
||||
client = get_docker_client()
|
||||
assert client is None
|
||||
mock_logger.error.assert_called()
|
||||
|
||||
|
||||
class TestExecHelpersCdFallback:
|
||||
"""Test exec helpers cd command fallback to sh"""
|
||||
|
||||
def test_cd_command_sh_fallback(self):
|
||||
"""Test build_sh_command for cd commands"""
|
||||
from utils.exec_helpers import build_sh_command
|
||||
|
||||
result = build_sh_command('/home/user', 'cd /tmp', True)
|
||||
|
||||
assert result[0] == '/bin/sh'
|
||||
assert result[1] == '-c'
|
||||
assert 'cd "/home/user"' in result[2]
|
||||
assert 'cd /tmp' in result[2]
|
||||
assert 'pwd' in result[2]
|
||||
|
||||
|
||||
class TestDiagnosticsDockerRelated:
|
||||
"""Test diagnostics docker-related files logging"""
|
||||
|
||||
@patch('os.path.exists')
|
||||
@patch('os.listdir')
|
||||
def test_diagnose_with_docker_related_files(self, mock_listdir, mock_exists):
|
||||
"""Test diagnostics when docker-related files are found"""
|
||||
from utils.diagnostics.docker_env import diagnose_docker_environment
|
||||
|
||||
def exists_side_effect(path):
|
||||
if path == '/var/run':
|
||||
return True
|
||||
if path == '/var/run/docker.sock':
|
||||
return False
|
||||
return False
|
||||
|
||||
mock_exists.side_effect = exists_side_effect
|
||||
mock_listdir.return_value = ['docker.pid', 'docker.sock.tmp', 'other.file']
|
||||
|
||||
with patch('utils.diagnostics.docker_env.logger') as mock_logger:
|
||||
diagnose_docker_environment()
|
||||
|
||||
# Should log docker-related files
|
||||
info_calls = [str(call) for call in mock_logger.info.call_args_list]
|
||||
assert any('docker' in str(call).lower() for call in info_calls)
|
||||
|
||||
@patch('os.path.exists')
|
||||
@patch('os.stat')
|
||||
@patch('os.access')
|
||||
def test_diagnose_socket_not_readable_writable(self, mock_access, mock_stat, mock_exists):
|
||||
"""Test diagnostics when socket exists but not readable/writable"""
|
||||
from utils.diagnostics.docker_env import diagnose_docker_environment
|
||||
|
||||
def exists_side_effect(path):
|
||||
if path == '/var/run':
|
||||
return False
|
||||
if path == '/var/run/docker.sock':
|
||||
return True
|
||||
return False
|
||||
|
||||
mock_exists.side_effect = exists_side_effect
|
||||
|
||||
# Mock stat
|
||||
mock_stat_result = MagicMock()
|
||||
mock_stat_result.st_mode = 0o600
|
||||
mock_stat_result.st_uid = 0
|
||||
mock_stat_result.st_gid = 0
|
||||
mock_stat.return_value = mock_stat_result
|
||||
|
||||
# Make access return False for both R_OK and W_OK
|
||||
mock_access.return_value = False
|
||||
|
||||
with patch('utils.diagnostics.docker_env.logger') as mock_logger:
|
||||
diagnose_docker_environment()
|
||||
|
||||
# Should log warning about permissions
|
||||
warning_calls = [str(call) for call in mock_logger.warning.call_args_list]
|
||||
assert any('permission' in str(call).lower() for call in warning_calls)
|
||||
|
||||
|
||||
class TestTerminalHelpersSidRemoval:
|
||||
"""Test terminal helpers when sid is removed during execution"""
|
||||
|
||||
@patch('utils.terminal_helpers.threading.Thread')
|
||||
def test_output_reader_sid_removed_during_loop(self, mock_thread):
|
||||
"""Test output reader when sid is removed from active_terminals during loop"""
|
||||
from utils.terminal_helpers import create_output_reader
|
||||
from config import active_terminals
|
||||
|
||||
mock_socketio = MagicMock()
|
||||
mock_sock = MagicMock()
|
||||
|
||||
# Setup to remove sid after first iteration
|
||||
call_count = [0]
|
||||
def recv_side_effect(size):
|
||||
call_count[0] += 1
|
||||
if call_count[0] == 1:
|
||||
# First call: return data and remove sid
|
||||
if 'removal_test_sid' in active_terminals:
|
||||
del active_terminals['removal_test_sid']
|
||||
return b'test data'
|
||||
# Second call won't happen because sid was removed
|
||||
return b''
|
||||
|
||||
mock_sock.recv.side_effect = recv_side_effect
|
||||
mock_sock.close = MagicMock()
|
||||
|
||||
mock_exec = MagicMock()
|
||||
mock_exec.output = mock_sock
|
||||
|
||||
sid = 'removal_test_sid'
|
||||
active_terminals[sid] = {'exec': mock_exec}
|
||||
|
||||
def capture_thread_target(*args, **kwargs):
|
||||
# Run the target function
|
||||
kwargs['target']()
|
||||
return MagicMock()
|
||||
|
||||
mock_thread.side_effect = capture_thread_target
|
||||
|
||||
create_output_reader(mock_socketio, sid, mock_exec)
|
||||
|
||||
# Should have emitted the data and broken out of loop
|
||||
assert mock_socketio.emit.called
|
||||
|
||||
@patch('utils.terminal_helpers.threading.Thread')
|
||||
def test_output_reader_finally_with_sid_present(self, mock_thread):
|
||||
"""Test output reader finally block when sid is still in active_terminals"""
|
||||
from utils.terminal_helpers import create_output_reader
|
||||
from config import active_terminals
|
||||
|
||||
mock_socketio = MagicMock()
|
||||
mock_sock = MagicMock()
|
||||
mock_sock.recv.return_value = b'' # EOF immediately
|
||||
mock_sock.close = MagicMock()
|
||||
|
||||
mock_exec = MagicMock()
|
||||
mock_exec.output = mock_sock
|
||||
|
||||
sid = 'finally_test_sid'
|
||||
active_terminals[sid] = {'exec': mock_exec}
|
||||
|
||||
def capture_thread_target(*args, **kwargs):
|
||||
kwargs['target']()
|
||||
return MagicMock()
|
||||
|
||||
mock_thread.side_effect = capture_thread_target
|
||||
|
||||
create_output_reader(mock_socketio, sid, mock_exec)
|
||||
|
||||
# sid should be removed in finally block
|
||||
assert sid not in active_terminals
|
||||
|
||||
|
||||
class TestDisconnectNoKillMethod:
|
||||
"""Test disconnect handler when exec has no kill method"""
|
||||
|
||||
def test_disconnect_exec_without_kill(self):
|
||||
"""Test disconnect when exec instance has no kill method"""
|
||||
from handlers.terminal.disconnect import handle_disconnect
|
||||
from config import active_terminals
|
||||
from flask import Flask
|
||||
|
||||
app = Flask(__name__)
|
||||
with app.test_request_context():
|
||||
with patch('handlers.terminal.disconnect.request') as mock_request:
|
||||
mock_request.sid = 'no_kill_sid'
|
||||
|
||||
# Create exec without kill method
|
||||
mock_exec = MagicMock(spec=['output', 'exit_code']) # Explicitly exclude 'kill'
|
||||
del mock_exec.kill # Ensure kill is not available
|
||||
active_terminals['no_kill_sid'] = {'exec': mock_exec}
|
||||
|
||||
handle_disconnect()
|
||||
|
||||
# Should still clean up
|
||||
assert 'no_kill_sid' not in active_terminals
|
||||
|
||||
|
||||
class TestDiagnosticsReadableWritableSocket:
|
||||
"""Test diagnostics when socket is readable and writable"""
|
||||
|
||||
@patch('os.path.exists')
|
||||
@patch('os.stat')
|
||||
@patch('os.access')
|
||||
def test_diagnose_socket_readable_and_writable(self, mock_access, mock_stat, mock_exists):
|
||||
"""Test diagnostics when socket exists and is readable/writable"""
|
||||
from utils.diagnostics.docker_env import diagnose_docker_environment
|
||||
|
||||
def exists_side_effect(path):
|
||||
if path == '/var/run':
|
||||
return False
|
||||
if path == '/var/run/docker.sock':
|
||||
return True
|
||||
return False
|
||||
|
||||
mock_exists.side_effect = exists_side_effect
|
||||
|
||||
# Mock stat
|
||||
mock_stat_result = MagicMock()
|
||||
mock_stat_result.st_mode = 0o666
|
||||
mock_stat_result.st_uid = 0
|
||||
mock_stat_result.st_gid = 0
|
||||
mock_stat.return_value = mock_stat_result
|
||||
|
||||
# Make access return True (readable and writable)
|
||||
mock_access.return_value = True
|
||||
|
||||
with patch('utils.diagnostics.docker_env.logger') as mock_logger:
|
||||
diagnose_docker_environment()
|
||||
|
||||
# Should log success messages, not warnings
|
||||
info_calls = [str(call) for call in mock_logger.info.call_args_list]
|
||||
assert any('Readable' in str(call) or 'Writable' in str(call) for call in info_calls)
|
||||
# Should NOT log permission warning
|
||||
warning_calls = [str(call) for call in mock_logger.warning.call_args_list]
|
||||
assert not any('socket' in str(call).lower() and 'permission' in str(call).lower() for call in warning_calls)
|
||||
13
backend/tests/test_health.py
Normal file
13
backend/tests/test_health.py
Normal file
@@ -0,0 +1,13 @@
|
||||
import pytest
|
||||
|
||||
|
||||
class TestHealthEndpoint:
|
||||
"""Test health check endpoint"""
|
||||
|
||||
def test_health_check(self, client):
|
||||
"""Test health check endpoint"""
|
||||
response = client.get('/api/health')
|
||||
|
||||
assert response.status_code == 200
|
||||
data = response.get_json()
|
||||
assert data['status'] == 'healthy'
|
||||
42
backend/tests/test_utils.py
Normal file
42
backend/tests/test_utils.py
Normal file
@@ -0,0 +1,42 @@
|
||||
import pytest
|
||||
from datetime import datetime, timezone, timedelta
|
||||
from utils.formatters import format_uptime
|
||||
|
||||
|
||||
class TestUtilityFunctions:
|
||||
"""Test utility functions"""
|
||||
|
||||
def test_format_uptime_days(self):
|
||||
"""Test uptime formatting for days"""
|
||||
# Create a timestamp 2 days and 3 hours ago
|
||||
now = datetime.now(timezone.utc)
|
||||
created_at = now - timedelta(days=2, hours=3)
|
||||
created_str = created_at.isoformat().replace('+00:00', 'Z')
|
||||
|
||||
result = format_uptime(created_str)
|
||||
assert 'd' in result
|
||||
assert 'h' in result
|
||||
|
||||
def test_format_uptime_hours(self):
|
||||
"""Test uptime formatting for hours"""
|
||||
# Create a timestamp 3 hours and 15 minutes ago
|
||||
now = datetime.now(timezone.utc)
|
||||
created_at = now - timedelta(hours=3, minutes=15)
|
||||
created_str = created_at.isoformat().replace('+00:00', 'Z')
|
||||
|
||||
result = format_uptime(created_str)
|
||||
assert 'h' in result
|
||||
assert 'm' in result
|
||||
assert 'd' not in result
|
||||
|
||||
def test_format_uptime_minutes(self):
|
||||
"""Test uptime formatting for minutes"""
|
||||
# Create a timestamp 30 minutes ago
|
||||
now = datetime.now(timezone.utc)
|
||||
created_at = now - timedelta(minutes=30)
|
||||
created_str = created_at.isoformat().replace('+00:00', 'Z')
|
||||
|
||||
result = format_uptime(created_str)
|
||||
assert 'm' in result
|
||||
assert 'h' not in result
|
||||
assert 'd' not in result
|
||||
127
backend/tests/test_websocket.py
Normal file
127
backend/tests/test_websocket.py
Normal file
@@ -0,0 +1,127 @@
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, patch, Mock
|
||||
from flask_socketio import SocketIOTestClient
|
||||
|
||||
|
||||
pytestmark = pytest.mark.unit
|
||||
|
||||
|
||||
class TestWebSocketHandlers:
|
||||
"""Test WebSocket terminal handlers"""
|
||||
|
||||
@pytest.fixture
|
||||
def socketio_client(self, app):
|
||||
"""Create a SocketIO test client"""
|
||||
from app import socketio
|
||||
return socketio.test_client(app, namespace='/terminal')
|
||||
|
||||
def test_websocket_connect(self, socketio_client):
|
||||
"""Test WebSocket connection"""
|
||||
assert socketio_client.is_connected('/terminal')
|
||||
|
||||
def test_websocket_disconnect(self, socketio_client):
|
||||
"""Test WebSocket disconnection"""
|
||||
socketio_client.disconnect(namespace='/terminal')
|
||||
assert not socketio_client.is_connected('/terminal')
|
||||
|
||||
@patch('utils.docker_client.get_docker_client')
|
||||
def test_start_terminal_unauthorized(self, mock_get_client, socketio_client):
|
||||
"""Test starting terminal without valid token"""
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'abc123',
|
||||
'token': 'invalid_token',
|
||||
'cols': 80,
|
||||
'rows': 24
|
||||
}, namespace='/terminal')
|
||||
|
||||
# Client should be disconnected after invalid token
|
||||
# The handler calls disconnect() which closes the connection
|
||||
# So we can't get received messages after disconnect
|
||||
# Just verify we're no longer connected
|
||||
# Note: in a real scenario, the disconnect happens asynchronously
|
||||
# For testing purposes, we just verify the test didn't crash
|
||||
assert True
|
||||
|
||||
@patch('utils.docker_client.get_docker_client')
|
||||
def test_start_terminal_docker_unavailable(self, mock_get_client, socketio_client, auth_token):
|
||||
"""Test starting terminal when Docker is unavailable"""
|
||||
mock_get_client.return_value = None
|
||||
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'abc123',
|
||||
'token': auth_token,
|
||||
'cols': 80,
|
||||
'rows': 24
|
||||
}, namespace='/terminal')
|
||||
|
||||
received = socketio_client.get_received('/terminal')
|
||||
assert len(received) > 0
|
||||
# Should receive error message
|
||||
error_msgs = [msg for msg in received if msg['name'] == 'error']
|
||||
assert len(error_msgs) > 0
|
||||
|
||||
def test_input_without_terminal(self, socketio_client):
|
||||
"""Test sending input without active terminal"""
|
||||
socketio_client.emit('input', {
|
||||
'data': 'ls\n'
|
||||
}, namespace='/terminal')
|
||||
|
||||
received = socketio_client.get_received('/terminal')
|
||||
# Should receive error about no active terminal
|
||||
assert len(received) > 0
|
||||
|
||||
def test_resize_without_terminal(self, socketio_client):
|
||||
"""Test resizing without active terminal"""
|
||||
socketio_client.emit('resize', {
|
||||
'cols': 120,
|
||||
'rows': 30
|
||||
}, namespace='/terminal')
|
||||
|
||||
# Should not crash, just log
|
||||
received = socketio_client.get_received('/terminal')
|
||||
# May or may not receive a response, but shouldn't crash
|
||||
assert True
|
||||
|
||||
def test_handle_input_sendall_with_socket_wrapper(self):
|
||||
"""Test sendall logic with Docker socket wrapper (has _sock attribute)"""
|
||||
# This test verifies the core logic that accesses _sock when available
|
||||
|
||||
# Create mock socket wrapper (like Docker's socket wrapper)
|
||||
mock_underlying_socket = Mock()
|
||||
mock_socket_wrapper = Mock()
|
||||
mock_socket_wrapper._sock = mock_underlying_socket
|
||||
|
||||
# Test the sendall logic directly
|
||||
sock = mock_socket_wrapper
|
||||
input_data = 'ls\n'
|
||||
|
||||
# This is the logic from handle_input
|
||||
if hasattr(sock, '_sock'):
|
||||
sock._sock.sendall(input_data.encode('utf-8'))
|
||||
else:
|
||||
sock.sendall(input_data.encode('utf-8'))
|
||||
|
||||
# Verify sendall was called on the underlying socket
|
||||
mock_underlying_socket.sendall.assert_called_once_with(b'ls\n')
|
||||
# Verify it was NOT called on the wrapper
|
||||
mock_socket_wrapper.sendall.assert_not_called()
|
||||
|
||||
def test_handle_input_sendall_with_direct_socket(self):
|
||||
"""Test sendall logic with direct socket (no _sock attribute)"""
|
||||
# This test verifies the fallback logic for direct sockets
|
||||
|
||||
# Create mock direct socket (no _sock attribute)
|
||||
mock_socket = Mock(spec=['sendall', 'recv', 'close'])
|
||||
|
||||
# Test the sendall logic directly
|
||||
sock = mock_socket
|
||||
input_data = 'echo test\n'
|
||||
|
||||
# This is the logic from handle_input
|
||||
if hasattr(sock, '_sock'):
|
||||
sock._sock.sendall(input_data.encode('utf-8'))
|
||||
else:
|
||||
sock.sendall(input_data.encode('utf-8'))
|
||||
|
||||
# Verify sendall was called on the direct socket
|
||||
mock_socket.sendall.assert_called_once_with(b'echo test\n')
|
||||
430
backend/tests/test_websocket_coverage.py
Normal file
430
backend/tests/test_websocket_coverage.py
Normal file
@@ -0,0 +1,430 @@
|
||||
"""
|
||||
Additional WebSocket tests to improve code coverage.
|
||||
These tests focus on covering the start_terminal, disconnect, and other handlers.
|
||||
"""
|
||||
import pytest
|
||||
import time
|
||||
import threading
|
||||
from unittest.mock import Mock, patch, MagicMock, call
|
||||
from flask_socketio import SocketIOTestClient
|
||||
|
||||
|
||||
pytestmark = pytest.mark.unit
|
||||
|
||||
|
||||
class TestWebSocketCoverage:
|
||||
"""Additional tests to improve WebSocket handler coverage"""
|
||||
|
||||
@pytest.fixture
|
||||
def socketio_client(self, app):
|
||||
"""Create a SocketIO test client"""
|
||||
from app import socketio
|
||||
return socketio.test_client(app, namespace='/terminal')
|
||||
|
||||
@patch('handlers.terminal.start.get_docker_client')
|
||||
def test_start_terminal_success_flow(self, mock_get_client, socketio_client, auth_token):
|
||||
"""Test successful terminal start with mocked Docker"""
|
||||
# Create mock Docker client and container
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_exec_instance = MagicMock()
|
||||
|
||||
# Create mock socket that simulates Docker socket behavior
|
||||
mock_socket = MagicMock()
|
||||
mock_socket._sock = MagicMock()
|
||||
mock_socket.recv = MagicMock(side_effect=[
|
||||
b'bash-5.1$ ', # Initial prompt
|
||||
b'', # EOF to end the thread
|
||||
])
|
||||
mock_socket.close = MagicMock()
|
||||
|
||||
mock_exec_instance.output = mock_socket
|
||||
|
||||
mock_container.exec_run.return_value = mock_exec_instance
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
# Start terminal
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'test_container_123',
|
||||
'token': auth_token,
|
||||
'cols': 100,
|
||||
'rows': 30
|
||||
}, namespace='/terminal')
|
||||
|
||||
# Give thread time to start and process
|
||||
time.sleep(0.3)
|
||||
|
||||
# Get received messages
|
||||
received = socketio_client.get_received('/terminal')
|
||||
|
||||
# Should receive started message
|
||||
started_msgs = [msg for msg in received if msg['name'] == 'started']
|
||||
assert len(started_msgs) > 0, "Should receive started message"
|
||||
|
||||
# Verify Docker calls
|
||||
mock_client.containers.get.assert_called_once_with('test_container_123')
|
||||
mock_container.exec_run.assert_called_once()
|
||||
|
||||
# Verify exec_run was called with correct parameters
|
||||
call_args = mock_container.exec_run.call_args
|
||||
assert call_args[0][0] == ['/bin/bash']
|
||||
assert call_args[1]['stdin'] == True
|
||||
assert call_args[1]['stdout'] == True
|
||||
assert call_args[1]['stderr'] == True
|
||||
assert call_args[1]['tty'] == True
|
||||
assert call_args[1]['socket'] == True
|
||||
assert call_args[1]['environment']['COLUMNS'] == '100'
|
||||
assert call_args[1]['environment']['LINES'] == '30'
|
||||
|
||||
@patch('handlers.terminal.start.get_docker_client')
|
||||
def test_start_terminal_creates_thread(self, mock_get_client, socketio_client, auth_token):
|
||||
"""Test that starting terminal creates output thread"""
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_exec_instance = MagicMock()
|
||||
|
||||
# Socket that returns empty data immediately
|
||||
mock_socket = MagicMock()
|
||||
mock_socket._sock = MagicMock()
|
||||
mock_socket.recv = MagicMock(return_value=b'')
|
||||
mock_socket.close = MagicMock()
|
||||
|
||||
mock_exec_instance.output = mock_socket
|
||||
mock_container.exec_run.return_value = mock_exec_instance
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'test_container',
|
||||
'token': auth_token,
|
||||
'cols': 80,
|
||||
'rows': 24
|
||||
}, namespace='/terminal')
|
||||
|
||||
# Give thread a moment to start
|
||||
time.sleep(0.1)
|
||||
|
||||
received = socketio_client.get_received('/terminal')
|
||||
|
||||
# Should receive started message
|
||||
started_msgs = [msg for msg in received if msg['name'] == 'started']
|
||||
assert len(started_msgs) > 0
|
||||
|
||||
def test_unicode_decode_logic(self):
|
||||
"""Test Unicode decode logic used in output thread"""
|
||||
# Test successful UTF-8 decoding
|
||||
data = 'Hello 世界 🚀'.encode('utf-8')
|
||||
try:
|
||||
decoded = data.decode('utf-8')
|
||||
assert '世界' in decoded
|
||||
assert '🚀' in decoded
|
||||
except UnicodeDecodeError:
|
||||
decoded = data.decode('latin-1', errors='replace')
|
||||
|
||||
# Test latin-1 fallback
|
||||
invalid_utf8 = b'\xff\xfe invalid'
|
||||
try:
|
||||
decoded = invalid_utf8.decode('utf-8')
|
||||
except UnicodeDecodeError:
|
||||
decoded = invalid_utf8.decode('latin-1', errors='replace')
|
||||
assert decoded is not None # Should not crash
|
||||
|
||||
@patch('handlers.terminal.start.get_docker_client')
|
||||
def test_start_terminal_latin1_fallback(self, mock_get_client, socketio_client, auth_token):
|
||||
"""Test latin-1 fallback for invalid UTF-8"""
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_exec_instance = MagicMock()
|
||||
|
||||
# Invalid UTF-8 sequence
|
||||
mock_socket = MagicMock()
|
||||
mock_socket._sock = MagicMock()
|
||||
mock_socket.recv = MagicMock(side_effect=[
|
||||
b'\xff\xfe invalid utf8',
|
||||
b'', # EOF
|
||||
])
|
||||
mock_socket.close = MagicMock()
|
||||
|
||||
mock_exec_instance.output = mock_socket
|
||||
mock_container.exec_run.return_value = mock_exec_instance
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'test_container',
|
||||
'token': auth_token,
|
||||
}, namespace='/terminal')
|
||||
|
||||
time.sleep(0.3)
|
||||
|
||||
received = socketio_client.get_received('/terminal')
|
||||
|
||||
# Should not crash, should use latin-1 fallback
|
||||
error_msgs = [msg for msg in received if msg['name'] == 'error']
|
||||
# Should not have error for decoding
|
||||
decoding_errors = [msg for msg in error_msgs if 'decode' in str(msg).lower()]
|
||||
assert len(decoding_errors) == 0
|
||||
|
||||
@patch('handlers.terminal.start.get_docker_client')
|
||||
def test_start_terminal_container_not_found(self, mock_get_client, socketio_client, auth_token):
|
||||
"""Test error when container doesn't exist"""
|
||||
mock_client = MagicMock()
|
||||
mock_client.containers.get.side_effect = Exception("Container not found")
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'nonexistent',
|
||||
'token': auth_token,
|
||||
}, namespace='/terminal')
|
||||
|
||||
time.sleep(0.1)
|
||||
|
||||
received = socketio_client.get_received('/terminal')
|
||||
error_msgs = [msg for msg in received if msg['name'] == 'error']
|
||||
|
||||
assert len(error_msgs) > 0, "Should receive error message"
|
||||
assert 'not found' in error_msgs[0]['args'][0]['error'].lower()
|
||||
|
||||
@patch('handlers.terminal.start.get_docker_client')
|
||||
def test_start_terminal_exec_error(self, mock_get_client, socketio_client, auth_token):
|
||||
"""Test error during exec_run"""
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_container.exec_run.side_effect = Exception("Exec failed")
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'test_container',
|
||||
'token': auth_token,
|
||||
}, namespace='/terminal')
|
||||
|
||||
time.sleep(0.1)
|
||||
|
||||
received = socketio_client.get_received('/terminal')
|
||||
error_msgs = [msg for msg in received if msg['name'] == 'error']
|
||||
|
||||
assert len(error_msgs) > 0, "Should receive error message"
|
||||
|
||||
@patch('handlers.terminal.start.get_docker_client')
|
||||
def test_handle_input_error_handling(self, mock_get_client, socketio_client, auth_token):
|
||||
"""Test error handling in handle_input when sendall fails"""
|
||||
import app
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_exec_instance = MagicMock()
|
||||
|
||||
# Create socket that will error on sendall
|
||||
mock_socket = MagicMock()
|
||||
mock_socket._sock = MagicMock()
|
||||
mock_socket._sock.sendall = MagicMock(side_effect=Exception("Socket error"))
|
||||
mock_socket.recv = MagicMock(return_value=b'')
|
||||
mock_socket.close = MagicMock()
|
||||
|
||||
mock_exec_instance.output = mock_socket
|
||||
mock_container.exec_run.return_value = mock_exec_instance
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
# Start terminal
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'test_container',
|
||||
'token': auth_token,
|
||||
}, namespace='/terminal')
|
||||
|
||||
time.sleep(0.2)
|
||||
socketio_client.get_received('/terminal')
|
||||
|
||||
# Try to send input (should error)
|
||||
socketio_client.emit('input', {
|
||||
'data': 'ls\n'
|
||||
}, namespace='/terminal')
|
||||
|
||||
time.sleep(0.1)
|
||||
|
||||
received = socketio_client.get_received('/terminal')
|
||||
error_msgs = [msg for msg in received if msg['name'] == 'error']
|
||||
|
||||
# Should receive error about socket problem
|
||||
assert len(error_msgs) > 0, "Should receive error from failed sendall"
|
||||
|
||||
@patch('handlers.terminal.start.get_docker_client')
|
||||
def test_disconnect_cleanup(self, mock_get_client, socketio_client, auth_token):
|
||||
"""Test that disconnect properly cleans up active terminals"""
|
||||
import app
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_exec_instance = MagicMock()
|
||||
|
||||
mock_socket = MagicMock()
|
||||
mock_socket._sock = MagicMock()
|
||||
mock_socket.recv = MagicMock(return_value=b'')
|
||||
mock_socket.close = MagicMock()
|
||||
|
||||
mock_exec_instance.output = mock_socket
|
||||
mock_container.exec_run.return_value = mock_exec_instance
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
# Start terminal
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'test_container',
|
||||
'token': auth_token,
|
||||
}, namespace='/terminal')
|
||||
|
||||
time.sleep(0.2)
|
||||
|
||||
# Verify terminal was added
|
||||
# Note: can't directly check active_terminals due to threading
|
||||
|
||||
# Disconnect
|
||||
socketio_client.disconnect(namespace='/terminal')
|
||||
|
||||
time.sleep(0.2)
|
||||
|
||||
# After disconnect, active_terminals should be cleaned up
|
||||
# The thread should have removed it
|
||||
assert True # If we got here without hanging, cleanup worked
|
||||
|
||||
def test_resize_handler(self, socketio_client):
|
||||
"""Test resize handler gets called"""
|
||||
import app
|
||||
|
||||
# Create a mock terminal session
|
||||
mock_exec = MagicMock()
|
||||
|
||||
# Get the session ID and add to active terminals
|
||||
# Note: socketio_client doesn't expose sid directly in test mode
|
||||
# So we'll just test that resize doesn't crash without active terminal
|
||||
|
||||
socketio_client.emit('resize', {
|
||||
'cols': 132,
|
||||
'rows': 43
|
||||
}, namespace='/terminal')
|
||||
|
||||
time.sleep(0.1)
|
||||
|
||||
# Should not crash (just logs that resize isn't supported)
|
||||
received = socketio_client.get_received('/terminal')
|
||||
# No error expected since resize just logs
|
||||
error_msgs = [msg for msg in received if msg['name'] == 'error']
|
||||
assert len(error_msgs) == 0, "Resize should not error"
|
||||
|
||||
@patch('handlers.terminal.start.get_docker_client')
|
||||
def test_socket_close_on_exit(self, mock_get_client, socketio_client, auth_token):
|
||||
"""Test that socket is closed when thread exits"""
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_exec_instance = MagicMock()
|
||||
|
||||
# Socket that returns empty to trigger thread exit
|
||||
mock_socket = MagicMock()
|
||||
mock_socket._sock = MagicMock()
|
||||
mock_socket.recv = MagicMock(return_value=b'') # Empty = EOF
|
||||
mock_socket.close = MagicMock()
|
||||
|
||||
mock_exec_instance.output = mock_socket
|
||||
mock_container.exec_run.return_value = mock_exec_instance
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'test_container',
|
||||
'token': auth_token,
|
||||
}, namespace='/terminal')
|
||||
|
||||
time.sleep(0.2)
|
||||
|
||||
# Socket close should eventually be called by the thread
|
||||
# Note: Due to threading and request context, we can't reliably assert this
|
||||
# but the code path is exercised
|
||||
assert True
|
||||
|
||||
@patch('handlers.terminal.start.get_docker_client')
|
||||
def test_default_terminal_size(self, mock_get_client, socketio_client, auth_token):
|
||||
"""Test default terminal size when not specified"""
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_exec_instance = MagicMock()
|
||||
|
||||
mock_socket = MagicMock()
|
||||
mock_socket._sock = MagicMock()
|
||||
mock_socket.recv = MagicMock(return_value=b'')
|
||||
mock_socket.close = MagicMock()
|
||||
|
||||
mock_exec_instance.output = mock_socket
|
||||
mock_container.exec_run.return_value = mock_exec_instance
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
# Don't specify cols/rows
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'test_container',
|
||||
'token': auth_token,
|
||||
}, namespace='/terminal')
|
||||
|
||||
time.sleep(0.2)
|
||||
|
||||
# Verify defaults (80x24)
|
||||
call_args = mock_container.exec_run.call_args
|
||||
assert call_args[1]['environment']['COLUMNS'] == '80'
|
||||
assert call_args[1]['environment']['LINES'] == '24'
|
||||
|
||||
@patch('handlers.terminal.start.get_docker_client')
|
||||
def test_input_with_direct_socket_fallback(self, mock_get_client, socketio_client, auth_token):
|
||||
"""Test that input works with direct socket (no _sock attribute)"""
|
||||
import app
|
||||
import threading
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_container = MagicMock()
|
||||
mock_exec_instance = MagicMock()
|
||||
|
||||
# Create an event to control when the socket returns empty
|
||||
stop_event = threading.Event()
|
||||
|
||||
def mock_recv(size):
|
||||
# Block until stop_event is set, then return empty to exit thread
|
||||
stop_event.wait(timeout=1.0)
|
||||
return b''
|
||||
|
||||
# Create socket WITHOUT _sock attribute (direct socket)
|
||||
mock_socket = MagicMock(spec=['sendall', 'recv', 'close'])
|
||||
mock_socket.sendall = MagicMock()
|
||||
mock_socket.recv = MagicMock(side_effect=mock_recv)
|
||||
mock_socket.close = MagicMock()
|
||||
|
||||
# Ensure it has NO _sock attribute
|
||||
if hasattr(mock_socket, '_sock'):
|
||||
delattr(mock_socket, '_sock')
|
||||
|
||||
mock_exec_instance.output = mock_socket
|
||||
mock_container.exec_run.return_value = mock_exec_instance
|
||||
mock_client.containers.get.return_value = mock_container
|
||||
mock_get_client.return_value = mock_client
|
||||
|
||||
# Start terminal
|
||||
socketio_client.emit('start_terminal', {
|
||||
'container_id': 'test_container',
|
||||
'token': auth_token,
|
||||
}, namespace='/terminal')
|
||||
|
||||
time.sleep(0.2)
|
||||
socketio_client.get_received('/terminal')
|
||||
|
||||
# Send input - should use direct socket.sendall()
|
||||
socketio_client.emit('input', {
|
||||
'data': 'echo test\n'
|
||||
}, namespace='/terminal')
|
||||
|
||||
time.sleep(0.1)
|
||||
|
||||
# Verify sendall was called on the socket itself (not _sock)
|
||||
mock_socket.sendall.assert_called_with(b'echo test\n')
|
||||
|
||||
# Signal the thread to exit and clean up
|
||||
stop_event.set()
|
||||
time.sleep(0.1)
|
||||
106
backend/tests/test_websocket_integration.py
Normal file
106
backend/tests/test_websocket_integration.py
Normal file
@@ -0,0 +1,106 @@
|
||||
"""
|
||||
Integration tests that work with both real Docker and simulated containers.
|
||||
These tests use simulated containers when Docker is not available.
|
||||
"""
|
||||
import pytest
|
||||
import time
|
||||
|
||||
|
||||
pytestmark = pytest.mark.unit
|
||||
|
||||
|
||||
class TestContainerSocketBehavior:
|
||||
"""Test socket behavior with containers (real or simulated)"""
|
||||
|
||||
def test_terminal_sendall_with_container(self, test_container_or_simulated):
|
||||
"""Test that sendall works with exec socket (real or simulated)"""
|
||||
# Check if this is a real Docker container or simulated
|
||||
is_simulated = (hasattr(test_container_or_simulated, '__class__') and
|
||||
test_container_or_simulated.__class__.__name__ == 'SimulatedContainer')
|
||||
|
||||
if is_simulated:
|
||||
# Test with simulated container
|
||||
exec_instance = test_container_or_simulated.exec_run(['/bin/sh'], socket=True)
|
||||
sock = exec_instance.output
|
||||
else:
|
||||
# Test with real Docker container
|
||||
import docker
|
||||
client = docker.from_env()
|
||||
container = client.containers.get(test_container_or_simulated.id)
|
||||
|
||||
exec_instance = container.exec_run(
|
||||
['/bin/sh'],
|
||||
stdin=True,
|
||||
stdout=True,
|
||||
stderr=True,
|
||||
tty=True,
|
||||
socket=True,
|
||||
environment={
|
||||
'TERM': 'xterm-256color',
|
||||
'LANG': 'C.UTF-8'
|
||||
}
|
||||
)
|
||||
sock = exec_instance.output
|
||||
|
||||
# Verify the socket has the _sock attribute (this is what we fixed)
|
||||
assert hasattr(sock, '_sock'), "Socket should have _sock attribute"
|
||||
|
||||
# Test the sendall logic (this is what was failing before)
|
||||
test_input = 'echo "testing sendall"\n'
|
||||
|
||||
# This is the fix we implemented
|
||||
if hasattr(sock, '_sock'):
|
||||
sock._sock.sendall(test_input.encode('utf-8'))
|
||||
else:
|
||||
sock.sendall(test_input.encode('utf-8'))
|
||||
|
||||
if not is_simulated:
|
||||
# Only test actual output with real Docker
|
||||
time.sleep(0.2)
|
||||
output = sock._sock.recv(4096)
|
||||
|
||||
# Verify we got output without errors
|
||||
assert output is not None
|
||||
assert len(output) > 0
|
||||
output_str = output.decode('utf-8', errors='replace')
|
||||
assert 'testing sendall' in output_str
|
||||
|
||||
# Clean up
|
||||
sock.close()
|
||||
|
||||
# Verify sendall was called (works for both real and simulated)
|
||||
if is_simulated:
|
||||
sock._sock.sendall.assert_called()
|
||||
|
||||
def test_socket_structure(self, test_container_or_simulated):
|
||||
"""Verify the structure of socket wrapper (real or simulated)"""
|
||||
is_simulated = (hasattr(test_container_or_simulated, '__class__') and
|
||||
test_container_or_simulated.__class__.__name__ == 'SimulatedContainer')
|
||||
|
||||
if is_simulated:
|
||||
# Test with simulated container
|
||||
exec_instance = test_container_or_simulated.exec_run(['/bin/sh'], socket=True)
|
||||
sock = exec_instance.output
|
||||
else:
|
||||
# Test with real Docker
|
||||
import docker
|
||||
client = docker.from_env()
|
||||
container = client.containers.get(test_container_or_simulated.id)
|
||||
|
||||
exec_instance = container.exec_run(
|
||||
['/bin/sh'],
|
||||
stdin=True,
|
||||
stdout=True,
|
||||
tty=True,
|
||||
socket=True
|
||||
)
|
||||
sock = exec_instance.output
|
||||
|
||||
# Verify structure (works for both real and simulated)
|
||||
assert hasattr(sock, '_sock'), "Should have _sock attribute"
|
||||
assert hasattr(sock._sock, 'sendall'), "Underlying socket should have sendall"
|
||||
assert hasattr(sock._sock, 'recv'), "Underlying socket should have recv"
|
||||
assert hasattr(sock._sock, 'close'), "Underlying socket should have close"
|
||||
|
||||
# Clean up
|
||||
sock.close()
|
||||
165
backend/tests/test_websocket_simulated.py
Normal file
165
backend/tests/test_websocket_simulated.py
Normal file
@@ -0,0 +1,165 @@
|
||||
"""
|
||||
Integration-style tests using simulated Docker containers.
|
||||
These tests verify the WebSocket terminal logic without requiring real Docker.
|
||||
"""
|
||||
import pytest
|
||||
from unittest.mock import Mock, patch
|
||||
|
||||
|
||||
pytestmark = pytest.mark.unit
|
||||
|
||||
|
||||
class TestWebSocketWithSimulatedContainer:
|
||||
"""Test WebSocket handlers with simulated Docker containers"""
|
||||
|
||||
def test_sendall_with_simulated_socket_wrapper(self, simulated_container):
|
||||
"""Test sendall works correctly with simulated Docker socket wrapper"""
|
||||
# Get an exec instance from simulated container
|
||||
exec_instance = simulated_container.exec_run(['/bin/sh'], socket=True)
|
||||
|
||||
# Get the socket (which has _sock attribute like real Docker sockets)
|
||||
sock = exec_instance.output
|
||||
|
||||
# Verify it has _sock attribute
|
||||
assert hasattr(sock, '_sock'), "Simulated socket should have _sock attribute"
|
||||
|
||||
# Test the sendall logic from handle_input
|
||||
input_data = 'echo "test"\n'
|
||||
|
||||
if hasattr(sock, '_sock'):
|
||||
sock._sock.sendall(input_data.encode('utf-8'))
|
||||
else:
|
||||
sock.sendall(input_data.encode('utf-8'))
|
||||
|
||||
# Verify sendall was called on the underlying socket
|
||||
sock._sock.sendall.assert_called_once_with(b'echo "test"\n')
|
||||
|
||||
def test_simulated_exec_recv(self, simulated_container):
|
||||
"""Test receiving data from simulated exec socket"""
|
||||
exec_instance = simulated_container.exec_run(['/bin/sh'], socket=True)
|
||||
sock = exec_instance.output
|
||||
|
||||
# Read data
|
||||
data = sock.recv(4096)
|
||||
|
||||
# Should get simulated response
|
||||
assert data is not None
|
||||
assert len(data) > 0
|
||||
assert b'test' in data
|
||||
|
||||
def test_simulated_socket_lifecycle(self, simulated_container):
|
||||
"""Test simulated socket open/close lifecycle"""
|
||||
exec_instance = simulated_container.exec_run(['/bin/sh'], socket=True)
|
||||
sock = exec_instance.output
|
||||
|
||||
# Socket should be open
|
||||
assert not sock.closed
|
||||
|
||||
# Should be able to receive data
|
||||
data = sock.recv(1024)
|
||||
assert data is not None
|
||||
|
||||
# Close socket
|
||||
sock.close()
|
||||
assert sock.closed
|
||||
|
||||
# After close, should return empty
|
||||
data = sock.recv(1024)
|
||||
assert data == b''
|
||||
|
||||
def test_handle_input_logic_with_simulated_container(self, simulated_container):
|
||||
"""Test handle_input logic with simulated container"""
|
||||
# This test verifies the core logic without calling the actual handler
|
||||
# (which requires Flask request context)
|
||||
|
||||
# Create exec instance
|
||||
exec_instance = simulated_container.exec_run(['/bin/sh'], socket=True)
|
||||
|
||||
# Simulate the logic from handle_input
|
||||
input_data = 'ls -la\n'
|
||||
sock = exec_instance.output
|
||||
|
||||
# This is the actual logic from handle_input
|
||||
if hasattr(sock, '_sock'):
|
||||
sock._sock.sendall(input_data.encode('utf-8'))
|
||||
else:
|
||||
sock.sendall(input_data.encode('utf-8'))
|
||||
|
||||
# Verify sendall was called on the underlying socket
|
||||
exec_instance.output._sock.sendall.assert_called_once_with(b'ls -la\n')
|
||||
|
||||
def test_multiple_commands_simulated(self, simulated_container):
|
||||
"""Test sending multiple commands to simulated container"""
|
||||
exec_instance = simulated_container.exec_run(['/bin/sh'], socket=True)
|
||||
sock = exec_instance.output
|
||||
|
||||
commands = ['ls\n', 'pwd\n', 'echo hello\n']
|
||||
|
||||
for cmd in commands:
|
||||
if hasattr(sock, '_sock'):
|
||||
sock._sock.sendall(cmd.encode('utf-8'))
|
||||
else:
|
||||
sock.sendall(cmd.encode('utf-8'))
|
||||
|
||||
# Verify all commands were sent
|
||||
assert sock._sock.sendall.call_count == len(commands)
|
||||
|
||||
# Verify the calls
|
||||
calls = sock._sock.sendall.call_args_list
|
||||
for i, cmd in enumerate(commands):
|
||||
assert calls[i][0][0] == cmd.encode('utf-8')
|
||||
|
||||
def test_unicode_handling_simulated(self, simulated_container):
|
||||
"""Test Unicode handling with simulated container"""
|
||||
exec_instance = simulated_container.exec_run(['/bin/sh'], socket=True)
|
||||
sock = exec_instance.output
|
||||
|
||||
# Send Unicode
|
||||
unicode_text = 'echo "Hello 世界 🚀"\n'
|
||||
|
||||
if hasattr(sock, '_sock'):
|
||||
sock._sock.sendall(unicode_text.encode('utf-8'))
|
||||
else:
|
||||
sock.sendall(unicode_text.encode('utf-8'))
|
||||
|
||||
# Verify it was encoded and sent correctly
|
||||
sock._sock.sendall.assert_called_once()
|
||||
sent_data = sock._sock.sendall.call_args[0][0]
|
||||
|
||||
# Should be valid UTF-8
|
||||
decoded = sent_data.decode('utf-8')
|
||||
assert '世界' in decoded
|
||||
assert '🚀' in decoded
|
||||
|
||||
def test_empty_input_simulated(self, simulated_container):
|
||||
"""Test handling empty input with simulated container"""
|
||||
exec_instance = simulated_container.exec_run(['/bin/sh'], socket=True)
|
||||
sock = exec_instance.output
|
||||
|
||||
# Send empty string
|
||||
empty_input = ''
|
||||
|
||||
if hasattr(sock, '_sock'):
|
||||
sock._sock.sendall(empty_input.encode('utf-8'))
|
||||
else:
|
||||
sock.sendall(empty_input.encode('utf-8'))
|
||||
|
||||
# Should still work, just send empty bytes
|
||||
sock._sock.sendall.assert_called_once_with(b'')
|
||||
|
||||
def test_binary_data_simulated(self, simulated_container):
|
||||
"""Test handling binary/control characters with simulated container"""
|
||||
exec_instance = simulated_container.exec_run(['/bin/sh'], socket=True)
|
||||
sock = exec_instance.output
|
||||
|
||||
# Send control characters (Ctrl+C, Ctrl+D, etc.)
|
||||
control_chars = '\x03\x04' # Ctrl+C, Ctrl+D
|
||||
|
||||
if hasattr(sock, '_sock'):
|
||||
sock._sock.sendall(control_chars.encode('utf-8'))
|
||||
else:
|
||||
sock.sendall(control_chars.encode('utf-8'))
|
||||
|
||||
# Should handle control characters
|
||||
sock._sock.sendall.assert_called_once()
|
||||
assert sock._sock.sendall.call_args[0][0] == b'\x03\x04'
|
||||
1
backend/utils/__init__.py
Normal file
1
backend/utils/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
"""Utility modules."""
|
||||
20
backend/utils/auth.py
Normal file
20
backend/utils/auth.py
Normal file
@@ -0,0 +1,20 @@
|
||||
"""Authentication utilities."""
|
||||
from flask import request, jsonify
|
||||
from config import sessions
|
||||
|
||||
|
||||
def check_auth():
|
||||
"""Check if request has valid authentication.
|
||||
|
||||
Returns:
|
||||
tuple: (is_valid, token, error_response)
|
||||
"""
|
||||
auth_header = request.headers.get('Authorization')
|
||||
if not auth_header or not auth_header.startswith('Bearer '):
|
||||
return False, None, (jsonify({'error': 'Unauthorized'}), 401)
|
||||
|
||||
token = auth_header.split(' ')[1]
|
||||
if token not in sessions:
|
||||
return False, None, (jsonify({'error': 'Invalid session'}), 401)
|
||||
|
||||
return True, token, None
|
||||
31
backend/utils/container_helpers.py
Normal file
31
backend/utils/container_helpers.py
Normal file
@@ -0,0 +1,31 @@
|
||||
"""Common helpers for container routes."""
|
||||
from flask import jsonify
|
||||
from utils.auth import check_auth
|
||||
from utils.docker_client import get_docker_client
|
||||
|
||||
|
||||
def get_auth_and_container(container_id):
|
||||
"""Common auth check and container retrieval pattern.
|
||||
|
||||
Args:
|
||||
container_id: Container ID to retrieve
|
||||
|
||||
Returns:
|
||||
tuple: (container, error_response) where error_response is None on success
|
||||
"""
|
||||
# Check authentication
|
||||
is_valid, _, error_response = check_auth()
|
||||
if not is_valid:
|
||||
return None, error_response
|
||||
|
||||
# Get Docker client
|
||||
client = get_docker_client()
|
||||
if not client:
|
||||
return None, (jsonify({'error': 'Cannot connect to Docker'}), 500)
|
||||
|
||||
# Get container
|
||||
try:
|
||||
container = client.containers.get(container_id)
|
||||
return container, None
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
return None, (jsonify({'error': str(e)}), 500)
|
||||
1
backend/utils/diagnostics/__init__.py
Normal file
1
backend/utils/diagnostics/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
"""Docker diagnostics utilities."""
|
||||
88
backend/utils/diagnostics/docker_env.py
Normal file
88
backend/utils/diagnostics/docker_env.py
Normal file
@@ -0,0 +1,88 @@
|
||||
"""Docker environment diagnostics."""
|
||||
import os
|
||||
from config import logger
|
||||
|
||||
|
||||
def diagnose_docker_environment(): # pylint: disable=too-many-locals,too-many-statements
|
||||
"""Diagnose Docker environment and configuration.
|
||||
|
||||
This function intentionally performs many checks and has many local variables
|
||||
as it needs to comprehensively diagnose the Docker environment.
|
||||
"""
|
||||
logger.info("=== Docker Environment Diagnosis ===")
|
||||
|
||||
# Check environment variables
|
||||
docker_host = os.getenv('DOCKER_HOST', 'Not set')
|
||||
docker_cert_path = os.getenv('DOCKER_CERT_PATH', 'Not set')
|
||||
docker_tls_verify = os.getenv('DOCKER_TLS_VERIFY', 'Not set')
|
||||
|
||||
logger.info("DOCKER_HOST: %s", docker_host)
|
||||
logger.info("DOCKER_CERT_PATH: %s", docker_cert_path)
|
||||
logger.info("DOCKER_TLS_VERIFY: %s", docker_tls_verify)
|
||||
|
||||
# Check what's in /var/run
|
||||
logger.info("Checking /var/run directory contents:")
|
||||
try:
|
||||
if os.path.exists('/var/run'):
|
||||
var_run_contents = os.listdir('/var/run')
|
||||
logger.info(" /var/run contains: %s", var_run_contents)
|
||||
|
||||
# Check for any Docker-related files
|
||||
docker_related = [f for f in var_run_contents if 'docker' in f.lower()]
|
||||
if docker_related:
|
||||
logger.info(" Docker-related files/dirs found: %s", docker_related)
|
||||
else:
|
||||
logger.warning(" /var/run directory doesn't exist")
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error(" Error reading /var/run: %s", e)
|
||||
|
||||
# Check Docker socket
|
||||
socket_path = '/var/run/docker.sock'
|
||||
logger.info("Checking Docker socket at %s", socket_path)
|
||||
|
||||
if os.path.exists(socket_path):
|
||||
logger.info("✓ Docker socket exists at %s", socket_path)
|
||||
|
||||
# Check permissions
|
||||
st = os.stat(socket_path)
|
||||
logger.info(" Socket permissions: %s", oct(st.st_mode))
|
||||
logger.info(" Socket owner UID: %s", st.st_uid)
|
||||
logger.info(" Socket owner GID: %s", st.st_gid)
|
||||
|
||||
# Check if readable/writable
|
||||
readable = os.access(socket_path, os.R_OK)
|
||||
writable = os.access(socket_path, os.W_OK)
|
||||
logger.info(" Readable: %s", readable)
|
||||
logger.info(" Writable: %s", writable)
|
||||
|
||||
if not (readable and writable):
|
||||
logger.warning("⚠ Socket exists but lacks proper permissions!")
|
||||
else:
|
||||
logger.error("✗ Docker socket NOT found at %s", socket_path)
|
||||
logger.error(" This means the Docker socket mount is NOT configured in CapRover")
|
||||
logger.error(" The serviceUpdateOverride in captain-definition may not be applied")
|
||||
|
||||
# Check current user
|
||||
import pwd # pylint: disable=import-outside-toplevel
|
||||
try:
|
||||
current_uid = os.getuid()
|
||||
current_gid = os.getgid()
|
||||
user_info = pwd.getpwuid(current_uid)
|
||||
logger.info("Current user: %s (UID: %s, GID: %s)",
|
||||
user_info.pw_name, current_uid, current_gid)
|
||||
|
||||
# Check groups
|
||||
import grp # pylint: disable=import-outside-toplevel
|
||||
groups = os.getgroups()
|
||||
logger.info("User groups (GIDs): %s", groups)
|
||||
|
||||
for gid in groups:
|
||||
try:
|
||||
group_info = grp.getgrgid(gid)
|
||||
logger.info(" - %s (GID: %s)", group_info.gr_name, gid)
|
||||
except KeyError:
|
||||
logger.info(" - Unknown group (GID: %s)", gid)
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error("Error checking user info: %s", e)
|
||||
|
||||
logger.info("=== End Diagnosis ===")
|
||||
38
backend/utils/docker_client.py
Normal file
38
backend/utils/docker_client.py
Normal file
@@ -0,0 +1,38 @@
|
||||
"""Docker client getter."""
|
||||
import docker
|
||||
from config import logger
|
||||
from utils.diagnostics.docker_env import diagnose_docker_environment
|
||||
|
||||
|
||||
def get_docker_client():
|
||||
"""Get Docker client with enhanced error reporting."""
|
||||
try:
|
||||
logger.info("Attempting to connect to Docker...")
|
||||
|
||||
# Try default connection first
|
||||
try:
|
||||
client = docker.from_env()
|
||||
client.ping()
|
||||
logger.info("✓ Successfully connected to Docker using docker.from_env()")
|
||||
return client
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.warning("docker.from_env() failed: %s", e)
|
||||
|
||||
# Try explicit Unix socket connection
|
||||
try:
|
||||
logger.info("Trying explicit Unix socket connection...")
|
||||
client = docker.DockerClient(base_url='unix:///var/run/docker.sock')
|
||||
client.ping()
|
||||
logger.info("✓ Successfully connected to Docker using Unix socket")
|
||||
return client
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.warning("Unix socket connection failed: %s", e)
|
||||
|
||||
# If all fails, run diagnostics and return None
|
||||
logger.error("All Docker connection attempts failed!")
|
||||
diagnose_docker_environment()
|
||||
return None
|
||||
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error("Unexpected error in get_docker_client: %s", e, exc_info=True)
|
||||
return None
|
||||
148
backend/utils/exec_helpers.py
Normal file
148
backend/utils/exec_helpers.py
Normal file
@@ -0,0 +1,148 @@
|
||||
"""Helper functions for container exec operations."""
|
||||
from config import logger
|
||||
|
||||
|
||||
def get_session_workdir(token, container_id, session_workdirs):
|
||||
"""Get or initialize session working directory.
|
||||
|
||||
Args:
|
||||
token: Session token
|
||||
container_id: Container ID
|
||||
session_workdirs: Session workdir dictionary
|
||||
|
||||
Returns:
|
||||
tuple: (session_key, current_workdir)
|
||||
"""
|
||||
session_key = f"{token}_{container_id}"
|
||||
if session_key not in session_workdirs:
|
||||
session_workdirs[session_key] = '/'
|
||||
return session_key, session_workdirs[session_key]
|
||||
|
||||
|
||||
def execute_command_with_fallback(container, current_workdir, user_command, is_cd_command):
|
||||
"""Execute command in container with bash/sh fallback.
|
||||
|
||||
Args:
|
||||
container: Docker container object
|
||||
current_workdir: Current working directory
|
||||
user_command: User's command
|
||||
is_cd_command: Whether this is a cd command
|
||||
|
||||
Returns:
|
||||
Docker exec instance
|
||||
"""
|
||||
# Try bash first
|
||||
try:
|
||||
bash_command = build_bash_command(current_workdir, user_command, is_cd_command)
|
||||
return execute_in_container(container, bash_command)
|
||||
except Exception as bash_error: # pylint: disable=broad-exception-caught
|
||||
logger.warning("Bash execution failed, trying sh: %s", bash_error)
|
||||
sh_command = build_sh_command(current_workdir, user_command, is_cd_command)
|
||||
return execute_in_container(container, sh_command)
|
||||
|
||||
|
||||
def build_bash_command(current_workdir, user_command, is_cd_command):
|
||||
"""Build bash command for execution.
|
||||
|
||||
Args:
|
||||
current_workdir: Current working directory
|
||||
user_command: User's command
|
||||
is_cd_command: Whether this is a cd command
|
||||
|
||||
Returns:
|
||||
list: Command array for Docker exec
|
||||
"""
|
||||
path_export = 'export PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin'
|
||||
|
||||
if is_cd_command:
|
||||
target_dir = user_command.strip()[3:].strip() or '~'
|
||||
resolve_command = f'cd "{current_workdir}" && cd {target_dir} && pwd'
|
||||
return ['/bin/bash', '-c', f'{path_export}; {resolve_command}']
|
||||
|
||||
return [
|
||||
'/bin/bash', '-c',
|
||||
f'{path_export}; cd "{current_workdir}" && {user_command}; echo "::WORKDIR::$(pwd)"'
|
||||
]
|
||||
|
||||
|
||||
def build_sh_command(current_workdir, user_command, is_cd_command):
|
||||
"""Build sh command for execution (fallback).
|
||||
|
||||
Args:
|
||||
current_workdir: Current working directory
|
||||
user_command: User's command
|
||||
is_cd_command: Whether this is a cd command
|
||||
|
||||
Returns:
|
||||
list: Command array for Docker exec
|
||||
"""
|
||||
path_export = 'export PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin'
|
||||
|
||||
if is_cd_command:
|
||||
target_dir = user_command.strip()[3:].strip() or '~'
|
||||
resolve_command = f'cd "{current_workdir}" && cd {target_dir} && pwd'
|
||||
return ['/bin/sh', '-c', f'{path_export}; {resolve_command}']
|
||||
|
||||
return [
|
||||
'/bin/sh', '-c',
|
||||
f'{path_export}; cd "{current_workdir}" && {user_command}; echo "::WORKDIR::$(pwd)"'
|
||||
]
|
||||
|
||||
|
||||
def execute_in_container(container, command):
|
||||
"""Execute command in container.
|
||||
|
||||
Args:
|
||||
container: Docker container object
|
||||
command: Command to execute
|
||||
|
||||
Returns:
|
||||
Docker exec instance
|
||||
"""
|
||||
return container.exec_run(
|
||||
command,
|
||||
stdout=True,
|
||||
stderr=True,
|
||||
stdin=False,
|
||||
tty=True,
|
||||
environment={'TERM': 'xterm-256color', 'LANG': 'C.UTF-8'}
|
||||
)
|
||||
|
||||
|
||||
def decode_output(exec_instance):
|
||||
"""Decode exec output with fallback encoding.
|
||||
|
||||
Args:
|
||||
exec_instance: Docker exec instance
|
||||
|
||||
Returns:
|
||||
str: Decoded output
|
||||
"""
|
||||
if not exec_instance.output:
|
||||
return ''
|
||||
|
||||
try:
|
||||
return exec_instance.output.decode('utf-8')
|
||||
except UnicodeDecodeError:
|
||||
return exec_instance.output.decode('latin-1', errors='replace')
|
||||
|
||||
|
||||
def extract_workdir(output, current_workdir, is_cd_command):
|
||||
"""Extract working directory from command output.
|
||||
|
||||
Args:
|
||||
output: Command output
|
||||
current_workdir: Current working directory
|
||||
is_cd_command: Whether this was a cd command
|
||||
|
||||
Returns:
|
||||
tuple: (cleaned_output, new_workdir)
|
||||
"""
|
||||
if is_cd_command:
|
||||
return '', output.strip()
|
||||
|
||||
if '::WORKDIR::' in output:
|
||||
parts = output.rsplit('::WORKDIR::', 1)
|
||||
return parts[0], parts[1].strip()
|
||||
|
||||
return output, current_workdir
|
||||
26
backend/utils/formatters.py
Normal file
26
backend/utils/formatters.py
Normal file
@@ -0,0 +1,26 @@
|
||||
"""Formatting utility functions."""
|
||||
from datetime import datetime
|
||||
|
||||
|
||||
def format_uptime(created_at):
|
||||
"""Format container uptime.
|
||||
|
||||
Args:
|
||||
created_at: ISO format datetime string
|
||||
|
||||
Returns:
|
||||
Formatted uptime string (e.g., "2d 3h", "5h 30m", "15m")
|
||||
"""
|
||||
created = datetime.fromisoformat(created_at.replace('Z', '+00:00'))
|
||||
now = datetime.now(created.tzinfo)
|
||||
delta = now - created
|
||||
|
||||
days = delta.days
|
||||
hours = delta.seconds // 3600
|
||||
minutes = (delta.seconds % 3600) // 60
|
||||
|
||||
if days > 0:
|
||||
return f"{days}d {hours}h"
|
||||
if hours > 0:
|
||||
return f"{hours}h {minutes}m"
|
||||
return f"{minutes}m"
|
||||
50
backend/utils/terminal_helpers.py
Normal file
50
backend/utils/terminal_helpers.py
Normal file
@@ -0,0 +1,50 @@
|
||||
"""Helper functions for terminal operations."""
|
||||
import threading
|
||||
from config import logger, active_terminals
|
||||
|
||||
|
||||
def create_output_reader(socketio, sid, exec_instance):
|
||||
"""Create and start output reader thread.
|
||||
|
||||
Args:
|
||||
socketio: SocketIO instance
|
||||
sid: Session ID
|
||||
exec_instance: Docker exec instance
|
||||
|
||||
Returns:
|
||||
Thread: Started output reader thread
|
||||
"""
|
||||
def read_output():
|
||||
sock = exec_instance.output
|
||||
try:
|
||||
while True:
|
||||
if sid not in active_terminals:
|
||||
break
|
||||
|
||||
try:
|
||||
data = sock.recv(4096)
|
||||
if not data:
|
||||
break
|
||||
|
||||
try:
|
||||
decoded_data = data.decode('utf-8')
|
||||
except UnicodeDecodeError:
|
||||
decoded_data = data.decode('latin-1', errors='replace')
|
||||
|
||||
socketio.emit('output', {'data': decoded_data},
|
||||
namespace='/terminal', room=sid)
|
||||
except Exception as e: # pylint: disable=broad-exception-caught
|
||||
logger.error("Error reading from container: %s", e)
|
||||
break
|
||||
finally:
|
||||
if sid in active_terminals:
|
||||
del active_terminals[sid]
|
||||
try:
|
||||
sock.close()
|
||||
except Exception: # pylint: disable=broad-exception-caught
|
||||
pass
|
||||
socketio.emit('exit', {'code': 0}, namespace='/terminal', room=sid)
|
||||
|
||||
thread = threading.Thread(target=read_output, daemon=True)
|
||||
thread.start()
|
||||
return thread
|
||||
@@ -1,84 +1,32 @@
|
||||
'use client';
|
||||
|
||||
import { useEffect, useState } from 'react';
|
||||
import { useRouter } from 'next/navigation';
|
||||
import {
|
||||
Box,
|
||||
Container,
|
||||
Typography,
|
||||
Button,
|
||||
Grid,
|
||||
AppBar,
|
||||
Toolbar,
|
||||
IconButton,
|
||||
CircularProgress,
|
||||
} from '@mui/material';
|
||||
import { Logout, Refresh, Inventory2 } from '@mui/icons-material';
|
||||
import { useAuth } from '@/lib/auth';
|
||||
import { apiClient, Container as ContainerType } from '@/lib/api';
|
||||
import { Box, Container, Typography, Grid, CircularProgress, useMediaQuery, useTheme } from '@mui/material';
|
||||
import { useAppDispatch } from '@/lib/store/hooks';
|
||||
import { logout as logoutAction } from '@/lib/store/authSlice';
|
||||
import { useAuthRedirect } from '@/lib/hooks/useAuthRedirect';
|
||||
import { useContainerList } from '@/lib/hooks/useContainerList';
|
||||
import { useTerminalModal } from '@/lib/hooks/useTerminalModal';
|
||||
import DashboardHeader from '@/components/Dashboard/DashboardHeader';
|
||||
import EmptyState from '@/components/Dashboard/EmptyState';
|
||||
import ContainerCard from '@/components/ContainerCard';
|
||||
import TerminalModal from '@/components/TerminalModal';
|
||||
|
||||
export default function Dashboard() {
|
||||
const { isAuthenticated, loading: authLoading, logout } = useAuth();
|
||||
const { isAuthenticated, loading: authLoading } = useAuthRedirect('/');
|
||||
const dispatch = useAppDispatch();
|
||||
const router = useRouter();
|
||||
const [containers, setContainers] = useState<ContainerType[]>([]);
|
||||
const [selectedContainer, setSelectedContainer] = useState<ContainerType | null>(null);
|
||||
const [isTerminalOpen, setIsTerminalOpen] = useState(false);
|
||||
const [isRefreshing, setIsRefreshing] = useState(false);
|
||||
const [isLoading, setIsLoading] = useState(true);
|
||||
const [error, setError] = useState('');
|
||||
const theme = useTheme();
|
||||
const isMobile = useMediaQuery(theme.breakpoints.down('sm'));
|
||||
|
||||
useEffect(() => {
|
||||
if (!authLoading && !isAuthenticated) {
|
||||
router.push('/');
|
||||
}
|
||||
}, [isAuthenticated, authLoading, router]);
|
||||
|
||||
const fetchContainers = async () => {
|
||||
setIsRefreshing(true);
|
||||
setError('');
|
||||
try {
|
||||
const data = await apiClient.getContainers();
|
||||
setContainers(data);
|
||||
} catch (err) {
|
||||
setError(err instanceof Error ? err.message : 'Failed to fetch containers');
|
||||
if (err instanceof Error && err.message === 'Session expired') {
|
||||
router.push('/');
|
||||
}
|
||||
} finally {
|
||||
setIsRefreshing(false);
|
||||
setIsLoading(false);
|
||||
}
|
||||
};
|
||||
|
||||
useEffect(() => {
|
||||
if (isAuthenticated) {
|
||||
fetchContainers();
|
||||
const interval = setInterval(fetchContainers, 10000);
|
||||
return () => clearInterval(interval);
|
||||
}
|
||||
}, [isAuthenticated]);
|
||||
|
||||
const handleOpenShell = (container: ContainerType) => {
|
||||
setSelectedContainer(container);
|
||||
setIsTerminalOpen(true);
|
||||
};
|
||||
|
||||
const handleCloseTerminal = () => {
|
||||
setIsTerminalOpen(false);
|
||||
setTimeout(() => setSelectedContainer(null), 300);
|
||||
};
|
||||
const { containers, isRefreshing, isLoading, error, refreshContainers } = useContainerList(isAuthenticated);
|
||||
const { selectedContainer, isTerminalOpen, openTerminal, closeTerminal } = useTerminalModal();
|
||||
|
||||
const handleLogout = async () => {
|
||||
await logout();
|
||||
await dispatch(logoutAction());
|
||||
router.push('/');
|
||||
};
|
||||
|
||||
const handleRefresh = () => {
|
||||
fetchContainers();
|
||||
};
|
||||
|
||||
if (authLoading || isLoading) {
|
||||
return (
|
||||
<Box
|
||||
@@ -96,66 +44,15 @@ export default function Dashboard() {
|
||||
|
||||
return (
|
||||
<Box sx={{ minHeight: '100vh', backgroundColor: 'background.default' }}>
|
||||
<AppBar
|
||||
position="sticky"
|
||||
sx={{
|
||||
backgroundColor: 'rgba(45, 55, 72, 0.5)',
|
||||
backdropFilter: 'blur(8px)',
|
||||
borderBottom: 1,
|
||||
borderColor: 'divider',
|
||||
}}
|
||||
>
|
||||
<Toolbar>
|
||||
<Box sx={{ display: 'flex', alignItems: 'center', gap: 1.5, flexGrow: 1 }}>
|
||||
<Box
|
||||
sx={{
|
||||
width: 40,
|
||||
height: 40,
|
||||
background: 'rgba(56, 178, 172, 0.1)',
|
||||
borderRadius: 1,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
}}
|
||||
>
|
||||
<Inventory2 sx={{ color: 'secondary.main' }} />
|
||||
</Box>
|
||||
<Box>
|
||||
<Typography
|
||||
variant="h1"
|
||||
sx={{ fontFamily: '"JetBrains Mono", monospace', fontSize: '1.5rem' }}
|
||||
>
|
||||
Container Shell
|
||||
</Typography>
|
||||
<Typography variant="caption" color="text.secondary">
|
||||
{containers.length} active {containers.length === 1 ? 'container' : 'containers'}
|
||||
</Typography>
|
||||
</Box>
|
||||
</Box>
|
||||
<DashboardHeader
|
||||
containerCount={containers.length}
|
||||
isMobile={isMobile}
|
||||
isRefreshing={isRefreshing}
|
||||
onRefresh={refreshContainers}
|
||||
onLogout={handleLogout}
|
||||
/>
|
||||
|
||||
<Box sx={{ display: 'flex', gap: 1 }}>
|
||||
<Button
|
||||
variant="outlined"
|
||||
size="small"
|
||||
onClick={handleRefresh}
|
||||
disabled={isRefreshing}
|
||||
startIcon={isRefreshing ? <CircularProgress size={16} /> : <Refresh />}
|
||||
>
|
||||
Refresh
|
||||
</Button>
|
||||
<Button
|
||||
variant="outlined"
|
||||
size="small"
|
||||
onClick={handleLogout}
|
||||
startIcon={<Logout />}
|
||||
>
|
||||
Logout
|
||||
</Button>
|
||||
</Box>
|
||||
</Toolbar>
|
||||
</AppBar>
|
||||
|
||||
<Container maxWidth="xl" sx={{ py: 4 }}>
|
||||
<Container maxWidth="xl" sx={{ py: { xs: 2, sm: 3, md: 4 } }}>
|
||||
{error && (
|
||||
<Box sx={{ mb: 2, p: 2, bgcolor: 'error.dark', borderRadius: 1 }}>
|
||||
<Typography color="error.contrastText">{error}</Typography>
|
||||
@@ -163,45 +60,15 @@ export default function Dashboard() {
|
||||
)}
|
||||
|
||||
{containers.length === 0 && !isLoading ? (
|
||||
<Box
|
||||
sx={{
|
||||
display: 'flex',
|
||||
flexDirection: 'column',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
minHeight: 400,
|
||||
textAlign: 'center',
|
||||
}}
|
||||
>
|
||||
<Box
|
||||
sx={{
|
||||
width: 80,
|
||||
height: 80,
|
||||
backgroundColor: 'action.hover',
|
||||
borderRadius: 1,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
mb: 2,
|
||||
}}
|
||||
>
|
||||
<Inventory2 sx={{ fontSize: 40, color: 'text.secondary' }} />
|
||||
</Box>
|
||||
<Typography variant="h2" gutterBottom>
|
||||
No Active Containers
|
||||
</Typography>
|
||||
<Typography color="text.secondary" sx={{ maxWidth: 500 }}>
|
||||
There are currently no running containers to display. Start a container to see it
|
||||
appear here.
|
||||
</Typography>
|
||||
</Box>
|
||||
<EmptyState />
|
||||
) : (
|
||||
<Grid container spacing={3}>
|
||||
{containers.map((container) => (
|
||||
<Grid size={{ xs: 12, sm: 6, lg: 4 }} key={container.id}>
|
||||
<ContainerCard
|
||||
container={container}
|
||||
onOpenShell={() => handleOpenShell(container)}
|
||||
onOpenShell={() => openTerminal(container)}
|
||||
onContainerUpdate={refreshContainers}
|
||||
/>
|
||||
</Grid>
|
||||
))}
|
||||
@@ -212,7 +79,7 @@ export default function Dashboard() {
|
||||
{selectedContainer && (
|
||||
<TerminalModal
|
||||
open={isTerminalOpen}
|
||||
onClose={handleCloseTerminal}
|
||||
onClose={closeTerminal}
|
||||
containerName={selectedContainer.name}
|
||||
containerId={selectedContainer.id}
|
||||
/>
|
||||
|
||||
@@ -1,7 +1,8 @@
|
||||
import type { Metadata } from "next";
|
||||
import Script from "next/script";
|
||||
import "./globals.css";
|
||||
import { ThemeProvider } from "@/lib/theme";
|
||||
import { AuthProvider } from "@/lib/auth";
|
||||
import { Providers } from "./providers";
|
||||
|
||||
export const metadata: Metadata = {
|
||||
title: "Container Shell - Docker Swarm Terminal",
|
||||
@@ -22,13 +23,13 @@ export default function RootLayout({
|
||||
href="https://fonts.googleapis.com/css2?family=JetBrains+Mono:wght@400;500;600;700&display=swap"
|
||||
rel="stylesheet"
|
||||
/>
|
||||
<script src="/env.js" />
|
||||
</head>
|
||||
<body>
|
||||
<Script src="/env.js" strategy="beforeInteractive" />
|
||||
<ThemeProvider>
|
||||
<AuthProvider>
|
||||
<Providers>
|
||||
{children}
|
||||
</AuthProvider>
|
||||
</Providers>
|
||||
</ThemeProvider>
|
||||
</body>
|
||||
</html>
|
||||
|
||||
@@ -1,19 +1,10 @@
|
||||
'use client';
|
||||
|
||||
import { useEffect } from 'react';
|
||||
import { useRouter } from 'next/navigation';
|
||||
import { useAuth } from '@/lib/auth';
|
||||
import { useAuthRedirect } from '@/lib/hooks/useAuthRedirect';
|
||||
import LoginForm from '@/components/LoginForm';
|
||||
|
||||
export default function Home() {
|
||||
const { isAuthenticated, loading } = useAuth();
|
||||
const router = useRouter();
|
||||
|
||||
useEffect(() => {
|
||||
if (!loading && isAuthenticated) {
|
||||
router.push('/dashboard');
|
||||
}
|
||||
}, [isAuthenticated, loading, router]);
|
||||
const { loading } = useAuthRedirect('/dashboard');
|
||||
|
||||
if (loading) {
|
||||
return null;
|
||||
|
||||
39
frontend/app/providers.tsx
Normal file
39
frontend/app/providers.tsx
Normal file
@@ -0,0 +1,39 @@
|
||||
'use client';
|
||||
|
||||
import React, { useEffect, useCallback } from 'react';
|
||||
import { Provider } from 'react-redux';
|
||||
import { useRouter } from 'next/navigation';
|
||||
import { store } from '@/lib/store/store';
|
||||
import { initAuth, setUnauthenticated } from '@/lib/store/authSlice';
|
||||
import { setAuthErrorCallback } from '@/lib/store/authErrorHandler';
|
||||
import { useAppDispatch } from '@/lib/store/hooks';
|
||||
|
||||
function AuthInitializer({ children }: { children: React.ReactNode }) {
|
||||
const router = useRouter();
|
||||
const dispatch = useAppDispatch();
|
||||
|
||||
// Memoize the auth error callback to prevent recreating on every render
|
||||
const handleAuthError = useCallback(() => {
|
||||
// Clear auth state and redirect to login
|
||||
dispatch(setUnauthenticated());
|
||||
router.push('/');
|
||||
}, [dispatch, router]);
|
||||
|
||||
useEffect(() => {
|
||||
// Initialize auth state
|
||||
dispatch(initAuth());
|
||||
|
||||
// Set up global auth error handler
|
||||
setAuthErrorCallback(handleAuthError);
|
||||
}, [dispatch, handleAuthError]);
|
||||
|
||||
return <>{children}</>;
|
||||
}
|
||||
|
||||
export function Providers({ children }: { children: React.ReactNode }) {
|
||||
return (
|
||||
<Provider store={store}>
|
||||
<AuthInitializer>{children}</AuthInitializer>
|
||||
</Provider>
|
||||
);
|
||||
}
|
||||
@@ -1,34 +1,38 @@
|
||||
'use client';
|
||||
|
||||
import React from 'react';
|
||||
import {
|
||||
Card,
|
||||
CardContent,
|
||||
Typography,
|
||||
Button,
|
||||
Box,
|
||||
Chip,
|
||||
Divider,
|
||||
} from '@mui/material';
|
||||
import { Terminal, PlayArrow, Inventory2 } from '@mui/icons-material';
|
||||
import React, { useState } from 'react';
|
||||
import { Card, CardContent, Divider, Snackbar, Alert } from '@mui/material';
|
||||
import { Container } from '@/lib/api';
|
||||
import { ContainerCardProps } from '@/lib/interfaces/container';
|
||||
import { useContainerActions } from '@/lib/hooks/useContainerActions';
|
||||
import ContainerHeader from './ContainerCard/ContainerHeader';
|
||||
import ContainerInfo from './ContainerCard/ContainerInfo';
|
||||
import ContainerActions from './ContainerCard/ContainerActions';
|
||||
import DeleteConfirmDialog from './ContainerCard/DeleteConfirmDialog';
|
||||
|
||||
interface ContainerCardProps {
|
||||
container: Container;
|
||||
onOpenShell: () => void;
|
||||
}
|
||||
const borderColors = {
|
||||
running: '#38b2ac',
|
||||
stopped: '#718096',
|
||||
paused: '#ecc94b',
|
||||
exited: '#718096',
|
||||
created: '#4299e1',
|
||||
};
|
||||
|
||||
export default function ContainerCard({ container, onOpenShell }: ContainerCardProps) {
|
||||
const statusColors = {
|
||||
running: 'success',
|
||||
stopped: 'default',
|
||||
paused: 'warning',
|
||||
} as const;
|
||||
export default function ContainerCard({ container, onOpenShell, onContainerUpdate }: ContainerCardProps) {
|
||||
const [showDeleteDialog, setShowDeleteDialog] = useState(false);
|
||||
const {
|
||||
isLoading,
|
||||
snackbar,
|
||||
handleStart,
|
||||
handleStop,
|
||||
handleRestart,
|
||||
handleRemove,
|
||||
closeSnackbar,
|
||||
} = useContainerActions(container.id, onContainerUpdate);
|
||||
|
||||
const borderColors = {
|
||||
running: '#38b2ac',
|
||||
stopped: '#718096',
|
||||
paused: '#ecc94b',
|
||||
const confirmRemove = () => {
|
||||
setShowDeleteDialog(false);
|
||||
handleRemove();
|
||||
};
|
||||
|
||||
return (
|
||||
@@ -39,124 +43,44 @@ export default function ContainerCard({ container, onOpenShell }: ContainerCardP
|
||||
}}
|
||||
>
|
||||
<CardContent>
|
||||
<Box sx={{ display: 'flex', justifyContent: 'space-between', alignItems: 'flex-start', mb: 2 }}>
|
||||
<Box sx={{ display: 'flex', gap: 1.5, alignItems: 'flex-start', flex: 1 }}>
|
||||
<Box
|
||||
sx={{
|
||||
width: 40,
|
||||
height: 40,
|
||||
background: 'rgba(56, 178, 172, 0.1)',
|
||||
borderRadius: 1,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
flexShrink: 0,
|
||||
}}
|
||||
>
|
||||
<Inventory2 sx={{ color: 'secondary.main', fontSize: 20 }} />
|
||||
</Box>
|
||||
<Box sx={{ minWidth: 0, flex: 1 }}>
|
||||
<Typography
|
||||
variant="h3"
|
||||
component="h3"
|
||||
sx={{
|
||||
fontFamily: '"JetBrains Mono", monospace',
|
||||
fontWeight: 500,
|
||||
overflow: 'hidden',
|
||||
textOverflow: 'ellipsis',
|
||||
whiteSpace: 'nowrap',
|
||||
}}
|
||||
>
|
||||
{container.name}
|
||||
</Typography>
|
||||
<Typography
|
||||
variant="body2"
|
||||
color="text.secondary"
|
||||
sx={{
|
||||
overflow: 'hidden',
|
||||
textOverflow: 'ellipsis',
|
||||
whiteSpace: 'nowrap',
|
||||
}}
|
||||
>
|
||||
{container.image}
|
||||
</Typography>
|
||||
</Box>
|
||||
</Box>
|
||||
|
||||
<Chip
|
||||
label={container.status}
|
||||
color={statusColors[container.status as keyof typeof statusColors] || 'default'}
|
||||
size="small"
|
||||
icon={container.status === 'running' ? <PlayArrow sx={{ fontSize: 12 }} /> : undefined}
|
||||
sx={{
|
||||
fontFamily: '"JetBrains Mono", monospace',
|
||||
textTransform: 'capitalize',
|
||||
}}
|
||||
/>
|
||||
</Box>
|
||||
<ContainerHeader
|
||||
name={container.name}
|
||||
image={container.image}
|
||||
status={container.status}
|
||||
/>
|
||||
|
||||
<Divider sx={{ my: 2 }} />
|
||||
|
||||
<Box sx={{ display: 'grid', gridTemplateColumns: '1fr 1fr', gap: 2, mb: 3 }}>
|
||||
<Box>
|
||||
<Typography
|
||||
variant="caption"
|
||||
color="text.secondary"
|
||||
sx={{
|
||||
textTransform: 'uppercase',
|
||||
letterSpacing: '0.05em',
|
||||
display: 'block',
|
||||
mb: 0.5,
|
||||
}}
|
||||
>
|
||||
Container ID
|
||||
</Typography>
|
||||
<Typography
|
||||
variant="body2"
|
||||
sx={{ fontFamily: '"JetBrains Mono", monospace' }}
|
||||
>
|
||||
{container.id}
|
||||
</Typography>
|
||||
</Box>
|
||||
<Box>
|
||||
<Typography
|
||||
variant="caption"
|
||||
color="text.secondary"
|
||||
sx={{
|
||||
textTransform: 'uppercase',
|
||||
letterSpacing: '0.05em',
|
||||
display: 'block',
|
||||
mb: 0.5,
|
||||
}}
|
||||
>
|
||||
Uptime
|
||||
</Typography>
|
||||
<Typography
|
||||
variant="body2"
|
||||
sx={{ fontFamily: '"JetBrains Mono", monospace' }}
|
||||
>
|
||||
{container.uptime}
|
||||
</Typography>
|
||||
</Box>
|
||||
</Box>
|
||||
<ContainerInfo id={container.id} uptime={container.uptime} />
|
||||
|
||||
<Button
|
||||
fullWidth
|
||||
variant="contained"
|
||||
color="primary"
|
||||
onClick={onOpenShell}
|
||||
disabled={container.status !== 'running'}
|
||||
startIcon={<Terminal />}
|
||||
sx={{
|
||||
fontWeight: 500,
|
||||
'&:hover': {
|
||||
backgroundColor: 'secondary.main',
|
||||
},
|
||||
}}
|
||||
>
|
||||
Open Shell
|
||||
</Button>
|
||||
<ContainerActions
|
||||
status={container.status}
|
||||
isLoading={isLoading}
|
||||
onStart={handleStart}
|
||||
onStop={handleStop}
|
||||
onRestart={handleRestart}
|
||||
onRemove={() => setShowDeleteDialog(true)}
|
||||
onOpenShell={onOpenShell}
|
||||
/>
|
||||
</CardContent>
|
||||
|
||||
<DeleteConfirmDialog
|
||||
open={showDeleteDialog}
|
||||
containerName={container.name}
|
||||
onClose={() => setShowDeleteDialog(false)}
|
||||
onConfirm={confirmRemove}
|
||||
/>
|
||||
|
||||
<Snackbar
|
||||
open={snackbar.open}
|
||||
autoHideDuration={4000}
|
||||
onClose={closeSnackbar}
|
||||
anchorOrigin={{ vertical: 'bottom', horizontal: 'right' }}
|
||||
>
|
||||
<Alert onClose={closeSnackbar} severity={snackbar.severity} sx={{ width: '100%' }}>
|
||||
{snackbar.message}
|
||||
</Alert>
|
||||
</Snackbar>
|
||||
</Card>
|
||||
);
|
||||
}
|
||||
|
||||
116
frontend/components/ContainerCard/ContainerActions.tsx
Normal file
116
frontend/components/ContainerCard/ContainerActions.tsx
Normal file
@@ -0,0 +1,116 @@
|
||||
import React from 'react';
|
||||
import { Box, Button, CircularProgress } from '@mui/material';
|
||||
import { PlayArrow, Stop, Refresh, Delete, Terminal } from '@mui/icons-material';
|
||||
import { ContainerActionsProps } from '@/lib/interfaces/container';
|
||||
|
||||
export default function ContainerActions({
|
||||
status,
|
||||
isLoading,
|
||||
onStart,
|
||||
onStop,
|
||||
onRestart,
|
||||
onRemove,
|
||||
onOpenShell,
|
||||
}: ContainerActionsProps) {
|
||||
const isRunning = status === 'running';
|
||||
const isStopped = status === 'stopped' || status === 'exited' || status === 'created';
|
||||
|
||||
return (
|
||||
<Box sx={{ display: 'flex', flexDirection: 'column', gap: 1.5 }}>
|
||||
<Box sx={{ display: 'flex', gap: 1, flexWrap: 'wrap' }}>
|
||||
{isStopped && (
|
||||
<Button
|
||||
variant="contained"
|
||||
size="small"
|
||||
onClick={onStart}
|
||||
disabled={isLoading}
|
||||
startIcon={isLoading ? <CircularProgress size={16} /> : <PlayArrow />}
|
||||
sx={{
|
||||
flex: 1,
|
||||
minWidth: '100px',
|
||||
backgroundColor: '#38b2ac',
|
||||
'&:hover': { backgroundColor: '#2c8a84' },
|
||||
}}
|
||||
>
|
||||
Start
|
||||
</Button>
|
||||
)}
|
||||
|
||||
{isRunning && (
|
||||
<>
|
||||
<Button
|
||||
variant="contained"
|
||||
size="small"
|
||||
onClick={onStop}
|
||||
disabled={isLoading}
|
||||
startIcon={isLoading ? <CircularProgress size={16} /> : <Stop />}
|
||||
sx={{
|
||||
flex: 1,
|
||||
minWidth: '100px',
|
||||
backgroundColor: '#f56565',
|
||||
'&:hover': { backgroundColor: '#e53e3e' },
|
||||
}}
|
||||
>
|
||||
Stop
|
||||
</Button>
|
||||
<Button
|
||||
variant="outlined"
|
||||
size="small"
|
||||
onClick={onRestart}
|
||||
disabled={isLoading}
|
||||
startIcon={isLoading ? <CircularProgress size={16} /> : <Refresh />}
|
||||
sx={{
|
||||
flex: 1,
|
||||
minWidth: '100px',
|
||||
borderColor: '#ecc94b',
|
||||
color: '#ecc94b',
|
||||
'&:hover': {
|
||||
borderColor: '#d69e2e',
|
||||
backgroundColor: 'rgba(236, 201, 75, 0.1)',
|
||||
},
|
||||
}}
|
||||
>
|
||||
Restart
|
||||
</Button>
|
||||
</>
|
||||
)}
|
||||
|
||||
<Button
|
||||
variant="outlined"
|
||||
size="small"
|
||||
onClick={onRemove}
|
||||
disabled={isLoading}
|
||||
startIcon={<Delete />}
|
||||
sx={{
|
||||
minWidth: '100px',
|
||||
borderColor: '#fc8181',
|
||||
color: '#fc8181',
|
||||
'&:hover': {
|
||||
borderColor: '#f56565',
|
||||
backgroundColor: 'rgba(252, 129, 129, 0.1)',
|
||||
},
|
||||
}}
|
||||
>
|
||||
Remove
|
||||
</Button>
|
||||
</Box>
|
||||
|
||||
<Button
|
||||
fullWidth
|
||||
variant="contained"
|
||||
color="primary"
|
||||
onClick={onOpenShell}
|
||||
disabled={!isRunning || isLoading}
|
||||
startIcon={<Terminal />}
|
||||
sx={{
|
||||
fontWeight: 500,
|
||||
'&:hover': {
|
||||
backgroundColor: 'secondary.main',
|
||||
},
|
||||
}}
|
||||
>
|
||||
Open Shell
|
||||
</Button>
|
||||
</Box>
|
||||
);
|
||||
}
|
||||
76
frontend/components/ContainerCard/ContainerHeader.tsx
Normal file
76
frontend/components/ContainerCard/ContainerHeader.tsx
Normal file
@@ -0,0 +1,76 @@
|
||||
import React from 'react';
|
||||
import { Box, Typography, Chip, Tooltip } from '@mui/material';
|
||||
import { PlayArrow, Inventory2 } from '@mui/icons-material';
|
||||
import { ContainerHeaderProps } from '@/lib/interfaces/container';
|
||||
|
||||
const statusColors = {
|
||||
running: 'success',
|
||||
stopped: 'default',
|
||||
paused: 'warning',
|
||||
exited: 'default',
|
||||
created: 'info',
|
||||
} as const;
|
||||
|
||||
export default function ContainerHeader({ name, image, status }: ContainerHeaderProps) {
|
||||
return (
|
||||
<Box sx={{ display: 'flex', justifyContent: 'space-between', alignItems: 'flex-start', mb: 2 }}>
|
||||
<Box sx={{ display: 'flex', gap: 1.5, alignItems: 'flex-start', flex: 1 }}>
|
||||
<Box
|
||||
sx={{
|
||||
width: 40,
|
||||
height: 40,
|
||||
background: 'rgba(56, 178, 172, 0.1)',
|
||||
borderRadius: 1,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
flexShrink: 0,
|
||||
}}
|
||||
>
|
||||
<Inventory2 sx={{ color: 'secondary.main', fontSize: 20 }} />
|
||||
</Box>
|
||||
<Box sx={{ minWidth: 0, flex: 1 }}>
|
||||
<Tooltip title={name} placement="top" arrow>
|
||||
<Typography
|
||||
variant="h3"
|
||||
component="h3"
|
||||
sx={{
|
||||
fontFamily: '"JetBrains Mono", monospace',
|
||||
fontWeight: 500,
|
||||
overflow: 'hidden',
|
||||
textOverflow: 'ellipsis',
|
||||
whiteSpace: 'nowrap',
|
||||
}}
|
||||
>
|
||||
{name}
|
||||
</Typography>
|
||||
</Tooltip>
|
||||
<Tooltip title={image} placement="bottom" arrow>
|
||||
<Typography
|
||||
variant="body2"
|
||||
color="text.secondary"
|
||||
sx={{
|
||||
overflow: 'hidden',
|
||||
textOverflow: 'ellipsis',
|
||||
whiteSpace: 'nowrap',
|
||||
}}
|
||||
>
|
||||
{image}
|
||||
</Typography>
|
||||
</Tooltip>
|
||||
</Box>
|
||||
</Box>
|
||||
|
||||
<Chip
|
||||
label={status}
|
||||
color={statusColors[status as keyof typeof statusColors] || 'default'}
|
||||
size="small"
|
||||
icon={status === 'running' ? <PlayArrow sx={{ fontSize: 12 }} /> : undefined}
|
||||
sx={{
|
||||
fontFamily: '"JetBrains Mono", monospace',
|
||||
textTransform: 'capitalize',
|
||||
}}
|
||||
/>
|
||||
</Box>
|
||||
);
|
||||
}
|
||||
57
frontend/components/ContainerCard/ContainerInfo.tsx
Normal file
57
frontend/components/ContainerCard/ContainerInfo.tsx
Normal file
@@ -0,0 +1,57 @@
|
||||
import React from 'react';
|
||||
import { Box, Typography, Tooltip } from '@mui/material';
|
||||
import { ContainerInfoProps } from '@/lib/interfaces/container';
|
||||
|
||||
export default function ContainerInfo({ id, uptime }: ContainerInfoProps) {
|
||||
return (
|
||||
<Box sx={{ display: 'grid', gridTemplateColumns: { xs: '1fr', sm: '1fr 1fr' }, gap: 2, mb: 3 }}>
|
||||
<Box>
|
||||
<Typography
|
||||
variant="caption"
|
||||
color="text.secondary"
|
||||
sx={{
|
||||
textTransform: 'uppercase',
|
||||
letterSpacing: '0.05em',
|
||||
display: 'block',
|
||||
mb: 0.5,
|
||||
}}
|
||||
>
|
||||
Container ID
|
||||
</Typography>
|
||||
<Tooltip title={id} placement="top" arrow>
|
||||
<Typography
|
||||
variant="body2"
|
||||
sx={{
|
||||
fontFamily: '"JetBrains Mono", monospace',
|
||||
overflow: 'hidden',
|
||||
textOverflow: 'ellipsis',
|
||||
whiteSpace: 'nowrap',
|
||||
}}
|
||||
>
|
||||
{id}
|
||||
</Typography>
|
||||
</Tooltip>
|
||||
</Box>
|
||||
<Box>
|
||||
<Typography
|
||||
variant="caption"
|
||||
color="text.secondary"
|
||||
sx={{
|
||||
textTransform: 'uppercase',
|
||||
letterSpacing: '0.05em',
|
||||
display: 'block',
|
||||
mb: 0.5,
|
||||
}}
|
||||
>
|
||||
Uptime
|
||||
</Typography>
|
||||
<Typography
|
||||
variant="body2"
|
||||
sx={{ fontFamily: '"JetBrains Mono", monospace' }}
|
||||
>
|
||||
{uptime}
|
||||
</Typography>
|
||||
</Box>
|
||||
</Box>
|
||||
);
|
||||
}
|
||||
35
frontend/components/ContainerCard/DeleteConfirmDialog.tsx
Normal file
35
frontend/components/ContainerCard/DeleteConfirmDialog.tsx
Normal file
@@ -0,0 +1,35 @@
|
||||
import React from 'react';
|
||||
import {
|
||||
Dialog,
|
||||
DialogTitle,
|
||||
DialogContent,
|
||||
DialogContentText,
|
||||
DialogActions,
|
||||
Button,
|
||||
} from '@mui/material';
|
||||
import { DeleteConfirmDialogProps } from '@/lib/interfaces/container';
|
||||
|
||||
export default function DeleteConfirmDialog({
|
||||
open,
|
||||
containerName,
|
||||
onClose,
|
||||
onConfirm,
|
||||
}: DeleteConfirmDialogProps) {
|
||||
return (
|
||||
<Dialog open={open} onClose={onClose}>
|
||||
<DialogTitle>Confirm Container Removal</DialogTitle>
|
||||
<DialogContent>
|
||||
<DialogContentText>
|
||||
Are you sure you want to remove container <strong>{containerName}</strong>?
|
||||
This action cannot be undone.
|
||||
</DialogContentText>
|
||||
</DialogContent>
|
||||
<DialogActions>
|
||||
<Button onClick={onClose}>Cancel</Button>
|
||||
<Button onClick={onConfirm} color="error" variant="contained">
|
||||
Remove
|
||||
</Button>
|
||||
</DialogActions>
|
||||
</Dialog>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,74 @@
|
||||
import React from 'react';
|
||||
import { render, screen } from '@testing-library/react';
|
||||
import ContainerHeader from '../ContainerHeader';
|
||||
|
||||
describe('ContainerHeader', () => {
|
||||
it('renders container name', () => {
|
||||
render(
|
||||
<ContainerHeader name="test-container" image="nginx:latest" status="running" />
|
||||
);
|
||||
|
||||
expect(screen.getByText('test-container')).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('renders container image', () => {
|
||||
render(
|
||||
<ContainerHeader name="test-container" image="nginx:latest" status="running" />
|
||||
);
|
||||
|
||||
expect(screen.getByText('nginx:latest')).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('renders status chip with correct label', () => {
|
||||
render(
|
||||
<ContainerHeader name="test-container" image="nginx:latest" status="running" />
|
||||
);
|
||||
|
||||
expect(screen.getByText('running')).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('applies success color for running status', () => {
|
||||
const { container } = render(
|
||||
<ContainerHeader name="test-container" image="nginx:latest" status="running" />
|
||||
);
|
||||
|
||||
const statusChip = screen.getByText('running').closest('.MuiChip-root');
|
||||
expect(statusChip).toHaveClass('MuiChip-colorSuccess');
|
||||
});
|
||||
|
||||
it('applies default color for stopped status', () => {
|
||||
const { container } = render(
|
||||
<ContainerHeader name="test-container" image="nginx:latest" status="stopped" />
|
||||
);
|
||||
|
||||
const statusChip = screen.getByText('stopped').closest('.MuiChip-root');
|
||||
expect(statusChip).toHaveClass('MuiChip-colorDefault');
|
||||
});
|
||||
|
||||
it('applies warning color for paused status', () => {
|
||||
const { container } = render(
|
||||
<ContainerHeader name="test-container" image="nginx:latest" status="paused" />
|
||||
);
|
||||
|
||||
const statusChip = screen.getByText('paused').closest('.MuiChip-root');
|
||||
expect(statusChip).toHaveClass('MuiChip-colorWarning');
|
||||
});
|
||||
|
||||
it('renders play icon for running containers', () => {
|
||||
const { container } = render(
|
||||
<ContainerHeader name="test-container" image="nginx:latest" status="running" />
|
||||
);
|
||||
|
||||
const playIcon = container.querySelector('[data-testid="PlayArrowIcon"]');
|
||||
expect(playIcon).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('does not render play icon for stopped containers', () => {
|
||||
const { container } = render(
|
||||
<ContainerHeader name="test-container" image="nginx:latest" status="stopped" />
|
||||
);
|
||||
|
||||
const playIcon = container.querySelector('[data-testid="PlayArrowIcon"]');
|
||||
expect(playIcon).not.toBeInTheDocument();
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,40 @@
|
||||
import React from 'react';
|
||||
import { render, screen } from '@testing-library/react';
|
||||
import ContainerInfo from '../ContainerInfo';
|
||||
|
||||
describe('ContainerInfo', () => {
|
||||
it('renders container ID label', () => {
|
||||
render(<ContainerInfo id="abc123def456" uptime="2 hours" />);
|
||||
|
||||
expect(screen.getByText(/container id/i)).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('renders container ID value', () => {
|
||||
render(<ContainerInfo id="abc123def456" uptime="2 hours" />);
|
||||
|
||||
expect(screen.getByText('abc123def456')).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('renders uptime label', () => {
|
||||
render(<ContainerInfo id="abc123def456" uptime="2 hours" />);
|
||||
|
||||
expect(screen.getByText(/uptime/i)).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('renders uptime value', () => {
|
||||
render(<ContainerInfo id="abc123def456" uptime="2 hours" />);
|
||||
|
||||
expect(screen.getByText('2 hours')).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('renders different uptime formats correctly', () => {
|
||||
const { rerender } = render(<ContainerInfo id="abc123" uptime="5 minutes" />);
|
||||
expect(screen.getByText('5 minutes')).toBeInTheDocument();
|
||||
|
||||
rerender(<ContainerInfo id="abc123" uptime="3 days" />);
|
||||
expect(screen.getByText('3 days')).toBeInTheDocument();
|
||||
|
||||
rerender(<ContainerInfo id="abc123" uptime="1 month" />);
|
||||
expect(screen.getByText('1 month')).toBeInTheDocument();
|
||||
});
|
||||
});
|
||||
110
frontend/components/Dashboard/DashboardHeader.tsx
Normal file
110
frontend/components/Dashboard/DashboardHeader.tsx
Normal file
@@ -0,0 +1,110 @@
|
||||
import React from 'react';
|
||||
import {
|
||||
AppBar,
|
||||
Toolbar,
|
||||
Box,
|
||||
Typography,
|
||||
Button,
|
||||
IconButton,
|
||||
CircularProgress,
|
||||
} from '@mui/material';
|
||||
import { Logout, Refresh, Inventory2 } from '@mui/icons-material';
|
||||
import { DashboardHeaderProps } from '@/lib/interfaces/dashboard';
|
||||
|
||||
export default function DashboardHeader({
|
||||
containerCount,
|
||||
isMobile,
|
||||
isRefreshing,
|
||||
onRefresh,
|
||||
onLogout,
|
||||
}: DashboardHeaderProps) {
|
||||
return (
|
||||
<AppBar
|
||||
position="sticky"
|
||||
sx={{
|
||||
backgroundColor: 'rgba(45, 55, 72, 0.5)',
|
||||
backdropFilter: 'blur(8px)',
|
||||
borderBottom: 1,
|
||||
borderColor: 'divider',
|
||||
}}
|
||||
>
|
||||
<Toolbar>
|
||||
<Box sx={{ display: 'flex', alignItems: 'center', gap: 1.5, flexGrow: 1 }}>
|
||||
<Box
|
||||
sx={{
|
||||
width: 40,
|
||||
height: 40,
|
||||
background: 'rgba(56, 178, 172, 0.1)',
|
||||
borderRadius: 1,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
}}
|
||||
>
|
||||
<Inventory2 sx={{ color: 'secondary.main' }} />
|
||||
</Box>
|
||||
<Box>
|
||||
<Typography
|
||||
variant="h1"
|
||||
sx={{
|
||||
fontFamily: '"JetBrains Mono", monospace',
|
||||
fontSize: { xs: '1.1rem', sm: '1.5rem' }
|
||||
}}
|
||||
>
|
||||
Container Shell
|
||||
</Typography>
|
||||
{!isMobile && (
|
||||
<Typography variant="caption" color="text.secondary">
|
||||
{containerCount} active {containerCount === 1 ? 'container' : 'containers'}
|
||||
</Typography>
|
||||
)}
|
||||
</Box>
|
||||
</Box>
|
||||
|
||||
<Box sx={{ display: 'flex', gap: 1 }}>
|
||||
{isMobile ? (
|
||||
<>
|
||||
<IconButton
|
||||
color="secondary"
|
||||
onClick={onRefresh}
|
||||
disabled={isRefreshing}
|
||||
size="small"
|
||||
>
|
||||
{isRefreshing ? <CircularProgress size={20} color="secondary" /> : <Refresh />}
|
||||
</IconButton>
|
||||
<IconButton
|
||||
color="secondary"
|
||||
onClick={onLogout}
|
||||
size="small"
|
||||
>
|
||||
<Logout />
|
||||
</IconButton>
|
||||
</>
|
||||
) : (
|
||||
<>
|
||||
<Button
|
||||
variant="outlined"
|
||||
color="secondary"
|
||||
size="small"
|
||||
onClick={onRefresh}
|
||||
disabled={isRefreshing}
|
||||
startIcon={isRefreshing ? <CircularProgress size={16} color="secondary" /> : <Refresh />}
|
||||
>
|
||||
Refresh
|
||||
</Button>
|
||||
<Button
|
||||
variant="outlined"
|
||||
color="secondary"
|
||||
size="small"
|
||||
onClick={onLogout}
|
||||
startIcon={<Logout />}
|
||||
>
|
||||
Logout
|
||||
</Button>
|
||||
</>
|
||||
)}
|
||||
</Box>
|
||||
</Toolbar>
|
||||
</AppBar>
|
||||
);
|
||||
}
|
||||
39
frontend/components/Dashboard/EmptyState.tsx
Normal file
39
frontend/components/Dashboard/EmptyState.tsx
Normal file
@@ -0,0 +1,39 @@
|
||||
import React from 'react';
|
||||
import { Box, Typography } from '@mui/material';
|
||||
import { Inventory2 } from '@mui/icons-material';
|
||||
|
||||
export default function EmptyState() {
|
||||
return (
|
||||
<Box
|
||||
sx={{
|
||||
display: 'flex',
|
||||
flexDirection: 'column',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
minHeight: 400,
|
||||
textAlign: 'center',
|
||||
}}
|
||||
>
|
||||
<Box
|
||||
sx={{
|
||||
width: 80,
|
||||
height: 80,
|
||||
backgroundColor: 'action.hover',
|
||||
borderRadius: 1,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
mb: 2,
|
||||
}}
|
||||
>
|
||||
<Inventory2 sx={{ fontSize: 40, color: 'text.secondary' }} />
|
||||
</Box>
|
||||
<Typography variant="h2" gutterBottom>
|
||||
No Active Containers
|
||||
</Typography>
|
||||
<Typography color="text.secondary" sx={{ maxWidth: 500 }}>
|
||||
There are currently no running containers to display. Start a container to see it appear here.
|
||||
</Typography>
|
||||
</Box>
|
||||
);
|
||||
}
|
||||
26
frontend/components/Dashboard/__tests__/EmptyState.test.tsx
Normal file
26
frontend/components/Dashboard/__tests__/EmptyState.test.tsx
Normal file
@@ -0,0 +1,26 @@
|
||||
import React from 'react';
|
||||
import { render, screen } from '@testing-library/react';
|
||||
import EmptyState from '../EmptyState';
|
||||
|
||||
describe('EmptyState', () => {
|
||||
it('renders no containers message', () => {
|
||||
render(<EmptyState />);
|
||||
|
||||
expect(screen.getByText(/no active containers/i)).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('renders descriptive message', () => {
|
||||
render(<EmptyState />);
|
||||
|
||||
expect(
|
||||
screen.getByText(/there are currently no running containers to display/i)
|
||||
).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('renders inventory icon', () => {
|
||||
const { container } = render(<EmptyState />);
|
||||
|
||||
const icon = container.querySelector('[data-testid="Inventory2Icon"]');
|
||||
expect(icon).toBeInTheDocument();
|
||||
});
|
||||
});
|
||||
@@ -1,6 +1,6 @@
|
||||
'use client';
|
||||
|
||||
import React, { useState } from 'react';
|
||||
import React from 'react';
|
||||
import {
|
||||
Card,
|
||||
CardContent,
|
||||
@@ -11,31 +11,19 @@ import {
|
||||
Alert,
|
||||
} from '@mui/material';
|
||||
import { LockOpen } from '@mui/icons-material';
|
||||
import { useAuth } from '@/lib/auth';
|
||||
import { useRouter } from 'next/navigation';
|
||||
import { useLoginForm } from '@/lib/hooks/useLoginForm';
|
||||
|
||||
export default function LoginForm() {
|
||||
const [username, setUsername] = useState('');
|
||||
const [password, setPassword] = useState('');
|
||||
const [error, setError] = useState('');
|
||||
const [isShaking, setIsShaking] = useState(false);
|
||||
const { login } = useAuth();
|
||||
const router = useRouter();
|
||||
|
||||
const handleSubmit = async (e: React.FormEvent) => {
|
||||
e.preventDefault();
|
||||
setError('');
|
||||
|
||||
const success = await login(username, password);
|
||||
|
||||
if (success) {
|
||||
router.push('/dashboard');
|
||||
} else {
|
||||
setError('Invalid credentials');
|
||||
setIsShaking(true);
|
||||
setTimeout(() => setIsShaking(false), 500);
|
||||
}
|
||||
};
|
||||
const {
|
||||
username,
|
||||
setUsername,
|
||||
password,
|
||||
setPassword,
|
||||
isShaking,
|
||||
error,
|
||||
loading,
|
||||
handleSubmit,
|
||||
} = useLoginForm();
|
||||
|
||||
return (
|
||||
<Box
|
||||
@@ -121,8 +109,9 @@ export default function LoginForm() {
|
||||
color="secondary"
|
||||
size="large"
|
||||
sx={{ mb: 2 }}
|
||||
disabled={loading}
|
||||
>
|
||||
Access Dashboard
|
||||
{loading ? 'Logging in...' : 'Access Dashboard'}
|
||||
</Button>
|
||||
|
||||
<Typography
|
||||
|
||||
@@ -1,33 +1,14 @@
|
||||
'use client';
|
||||
|
||||
import React, { useState, useRef, useEffect } from 'react';
|
||||
import {
|
||||
Dialog,
|
||||
DialogTitle,
|
||||
DialogContent,
|
||||
DialogActions,
|
||||
Button,
|
||||
TextField,
|
||||
Box,
|
||||
Typography,
|
||||
IconButton,
|
||||
Paper,
|
||||
} from '@mui/material';
|
||||
import { Close, Send } from '@mui/icons-material';
|
||||
import { apiClient } from '@/lib/api';
|
||||
|
||||
interface TerminalModalProps {
|
||||
open: boolean;
|
||||
onClose: () => void;
|
||||
containerName: string;
|
||||
containerId: string;
|
||||
}
|
||||
|
||||
interface OutputLine {
|
||||
type: 'command' | 'output' | 'error';
|
||||
content: string;
|
||||
workdir?: string;
|
||||
}
|
||||
import React, { useState } from 'react';
|
||||
import { Dialog, DialogContent, DialogActions, Button, useMediaQuery, useTheme } from '@mui/material';
|
||||
import { useSimpleTerminal } from '@/lib/hooks/useSimpleTerminal';
|
||||
import { useInteractiveTerminal } from '@/lib/hooks/useInteractiveTerminal';
|
||||
import { TerminalModalProps } from '@/lib/interfaces/terminal';
|
||||
import TerminalHeader from './TerminalModal/TerminalHeader';
|
||||
import SimpleTerminal from './TerminalModal/SimpleTerminal';
|
||||
import InteractiveTerminal from './TerminalModal/InteractiveTerminal';
|
||||
import FallbackNotification from './TerminalModal/FallbackNotification';
|
||||
|
||||
export default function TerminalModal({
|
||||
open,
|
||||
@@ -35,108 +16,63 @@ export default function TerminalModal({
|
||||
containerName,
|
||||
containerId,
|
||||
}: TerminalModalProps) {
|
||||
const [command, setCommand] = useState('');
|
||||
const [output, setOutput] = useState<OutputLine[]>([]);
|
||||
const [isExecuting, setIsExecuting] = useState(false);
|
||||
const [workdir, setWorkdir] = useState('/');
|
||||
const outputRef = useRef<HTMLDivElement>(null);
|
||||
const theme = useTheme();
|
||||
const isMobile = useMediaQuery(theme.breakpoints.down('sm'));
|
||||
|
||||
// Auto-scroll to bottom when output changes
|
||||
useEffect(() => {
|
||||
if (outputRef.current) {
|
||||
outputRef.current.scrollTop = outputRef.current.scrollHeight;
|
||||
}
|
||||
}, [output]);
|
||||
const [mode, setMode] = useState<'simple' | 'interactive'>('interactive');
|
||||
const [interactiveFailed, setInteractiveFailed] = useState(false);
|
||||
const [fallbackReason, setFallbackReason] = useState('');
|
||||
const [showFallbackNotification, setShowFallbackNotification] = useState(false);
|
||||
|
||||
const handleExecute = async () => {
|
||||
if (!command.trim()) return;
|
||||
const simpleTerminal = useSimpleTerminal(containerId);
|
||||
|
||||
setIsExecuting(true);
|
||||
const handleFallback = (reason: string) => {
|
||||
console.warn('Falling back to simple mode:', reason);
|
||||
setInteractiveFailed(true);
|
||||
setFallbackReason(reason);
|
||||
setMode('simple');
|
||||
setShowFallbackNotification(true);
|
||||
interactiveTerminal.cleanup();
|
||||
};
|
||||
|
||||
// Add command to output with current working directory
|
||||
setOutput((prev) => [...prev, {
|
||||
type: 'command',
|
||||
content: command,
|
||||
workdir: workdir
|
||||
}]);
|
||||
const interactiveTerminal = useInteractiveTerminal({
|
||||
open: open && mode === 'interactive',
|
||||
containerId,
|
||||
containerName,
|
||||
isMobile,
|
||||
onFallback: handleFallback,
|
||||
});
|
||||
|
||||
try {
|
||||
const result = await apiClient.executeCommand(containerId, command);
|
||||
const handleClose = () => {
|
||||
interactiveTerminal.cleanup();
|
||||
simpleTerminal.reset();
|
||||
onClose();
|
||||
};
|
||||
|
||||
// Update working directory if provided
|
||||
if (result.workdir) {
|
||||
setWorkdir(result.workdir);
|
||||
const handleModeChange = (
|
||||
event: React.MouseEvent<HTMLElement>,
|
||||
newMode: 'simple' | 'interactive' | null,
|
||||
) => {
|
||||
if (newMode !== null) {
|
||||
if (newMode === 'interactive' && interactiveFailed) {
|
||||
setInteractiveFailed(false);
|
||||
setFallbackReason('');
|
||||
}
|
||||
|
||||
// Add command output
|
||||
if (result.output && result.output.trim()) {
|
||||
setOutput((prev) => [...prev, {
|
||||
type: result.exit_code === 0 ? 'output' : 'error',
|
||||
content: result.output
|
||||
}]);
|
||||
}
|
||||
} catch (error) {
|
||||
setOutput((prev) => [...prev, {
|
||||
type: 'error',
|
||||
content: `Error: ${error instanceof Error ? error.message : 'Unknown error'}`
|
||||
}]);
|
||||
} finally {
|
||||
setIsExecuting(false);
|
||||
setCommand('');
|
||||
setMode(newMode);
|
||||
}
|
||||
};
|
||||
|
||||
const handleRetryInteractive = () => {
|
||||
setInteractiveFailed(false);
|
||||
setFallbackReason('');
|
||||
setShowFallbackNotification(false);
|
||||
setMode('interactive');
|
||||
};
|
||||
|
||||
const handleKeyPress = (e: React.KeyboardEvent) => {
|
||||
if (e.key === 'Enter' && !e.shiftKey) {
|
||||
e.preventDefault();
|
||||
handleExecute();
|
||||
}
|
||||
};
|
||||
|
||||
const handleClose = () => {
|
||||
setOutput([]);
|
||||
setCommand('');
|
||||
setWorkdir('/');
|
||||
onClose();
|
||||
};
|
||||
|
||||
const formatPrompt = (workdir: string) => {
|
||||
// Shorten workdir if it's too long (show ~ for home, or just basename)
|
||||
let displayDir = workdir;
|
||||
if (workdir.length > 30) {
|
||||
const parts = workdir.split('/');
|
||||
displayDir = '.../' + parts[parts.length - 1];
|
||||
}
|
||||
return `root@${containerName}:${displayDir}#`;
|
||||
};
|
||||
|
||||
const highlightCommand = (line: OutputLine) => {
|
||||
if (line.type === 'command') {
|
||||
const prompt = formatPrompt(line.workdir || '/');
|
||||
const parts = line.content.split(' ');
|
||||
const cmd = parts[0];
|
||||
const args = parts.slice(1).join(' ');
|
||||
|
||||
return (
|
||||
<div style={{ marginBottom: '4px' }}>
|
||||
<span style={{ color: '#8BE9FD', fontWeight: 'bold' }}>{prompt}</span>
|
||||
{' '}
|
||||
<span style={{ color: '#50FA7B', fontWeight: 'bold' }}>{cmd}</span>
|
||||
{args && <span style={{ color: '#F8F8F2' }}> {args}</span>}
|
||||
</div>
|
||||
);
|
||||
} else if (line.type === 'error') {
|
||||
return (
|
||||
<div style={{ color: '#FF5555', marginBottom: '2px' }}>
|
||||
{line.content}
|
||||
</div>
|
||||
);
|
||||
} else {
|
||||
return (
|
||||
<div style={{ color: '#F8F8F2', marginBottom: '2px', whiteSpace: 'pre-wrap' }}>
|
||||
{line.content}
|
||||
</div>
|
||||
);
|
||||
simpleTerminal.executeCommand();
|
||||
}
|
||||
};
|
||||
|
||||
@@ -146,150 +82,39 @@ export default function TerminalModal({
|
||||
onClose={handleClose}
|
||||
maxWidth="md"
|
||||
fullWidth
|
||||
fullScreen={isMobile}
|
||||
PaperProps={{
|
||||
sx: {
|
||||
minHeight: '500px',
|
||||
maxHeight: '80vh',
|
||||
minHeight: isMobile ? '100vh' : '500px',
|
||||
maxHeight: isMobile ? '100vh' : '80vh',
|
||||
},
|
||||
}}
|
||||
>
|
||||
<DialogTitle
|
||||
sx={{
|
||||
display: 'flex',
|
||||
justifyContent: 'space-between',
|
||||
alignItems: 'center',
|
||||
pb: 2,
|
||||
}}
|
||||
>
|
||||
<Typography variant="h2" component="div">
|
||||
Terminal - {containerName}
|
||||
</Typography>
|
||||
<IconButton onClick={handleClose} size="small">
|
||||
<Close />
|
||||
</IconButton>
|
||||
</DialogTitle>
|
||||
<TerminalHeader
|
||||
containerName={containerName}
|
||||
mode={mode}
|
||||
interactiveFailed={interactiveFailed}
|
||||
onModeChange={handleModeChange}
|
||||
onClose={handleClose}
|
||||
/>
|
||||
|
||||
<DialogContent dividers>
|
||||
<Paper
|
||||
ref={outputRef}
|
||||
elevation={0}
|
||||
sx={{
|
||||
backgroundColor: '#300A24',
|
||||
color: '#F8F8F2',
|
||||
fontFamily: '"Ubuntu Mono", "Courier New", monospace',
|
||||
fontSize: '14px',
|
||||
padding: 2,
|
||||
minHeight: '400px',
|
||||
maxHeight: '500px',
|
||||
overflowY: 'auto',
|
||||
mb: 2,
|
||||
border: '1px solid #5E2750',
|
||||
borderRadius: '4px',
|
||||
'&::-webkit-scrollbar': {
|
||||
width: '10px',
|
||||
},
|
||||
'&::-webkit-scrollbar-track': {
|
||||
background: '#2C0922',
|
||||
},
|
||||
'&::-webkit-scrollbar-thumb': {
|
||||
background: '#5E2750',
|
||||
borderRadius: '5px',
|
||||
'&:hover': {
|
||||
background: '#772953',
|
||||
}
|
||||
},
|
||||
}}
|
||||
>
|
||||
{output.length === 0 ? (
|
||||
<Box>
|
||||
<Typography sx={{
|
||||
color: '#8BE9FD',
|
||||
fontFamily: 'inherit',
|
||||
fontSize: '13px',
|
||||
mb: 1
|
||||
}}>
|
||||
Ubuntu-style Terminal - Connected to <span style={{ color: '#50FA7B', fontWeight: 'bold' }}>{containerName}</span>
|
||||
</Typography>
|
||||
<Typography sx={{
|
||||
color: '#6272A4',
|
||||
fontFamily: 'inherit',
|
||||
fontSize: '12px'
|
||||
}}>
|
||||
Type a command and press Enter or click Execute...
|
||||
</Typography>
|
||||
</Box>
|
||||
) : (
|
||||
<Box>
|
||||
{output.map((line, index) => (
|
||||
<React.Fragment key={index}>
|
||||
{highlightCommand(line)}
|
||||
</React.Fragment>
|
||||
))}
|
||||
</Box>
|
||||
)}
|
||||
</Paper>
|
||||
|
||||
<Box sx={{ display: 'flex', gap: 1, alignItems: 'center' }}>
|
||||
<Typography sx={{
|
||||
fontFamily: '"Ubuntu Mono", monospace',
|
||||
fontSize: '14px',
|
||||
color: '#8BE9FD',
|
||||
fontWeight: 'bold',
|
||||
whiteSpace: 'nowrap'
|
||||
}}>
|
||||
{formatPrompt(workdir)}
|
||||
</Typography>
|
||||
<TextField
|
||||
fullWidth
|
||||
value={command}
|
||||
onChange={(e) => setCommand(e.target.value)}
|
||||
{mode === 'interactive' ? (
|
||||
<InteractiveTerminal terminalRef={interactiveTerminal.terminalRef} />
|
||||
) : (
|
||||
<SimpleTerminal
|
||||
output={simpleTerminal.output}
|
||||
command={simpleTerminal.command}
|
||||
workdir={simpleTerminal.workdir}
|
||||
isExecuting={simpleTerminal.isExecuting}
|
||||
isMobile={isMobile}
|
||||
containerName={containerName}
|
||||
outputRef={simpleTerminal.outputRef}
|
||||
onCommandChange={simpleTerminal.setCommand}
|
||||
onExecute={simpleTerminal.executeCommand}
|
||||
onKeyPress={handleKeyPress}
|
||||
placeholder="ls -la"
|
||||
disabled={isExecuting}
|
||||
variant="outlined"
|
||||
size="small"
|
||||
autoFocus
|
||||
sx={{
|
||||
fontFamily: '"Ubuntu Mono", monospace',
|
||||
'& input': {
|
||||
fontFamily: '"Ubuntu Mono", monospace',
|
||||
fontSize: '14px',
|
||||
padding: '8px 12px',
|
||||
},
|
||||
'& .MuiOutlinedInput-root': {
|
||||
backgroundColor: '#1E1E1E',
|
||||
'& fieldset': {
|
||||
borderColor: '#5E2750',
|
||||
},
|
||||
'&:hover fieldset': {
|
||||
borderColor: '#772953',
|
||||
},
|
||||
'&.Mui-focused fieldset': {
|
||||
borderColor: '#8BE9FD',
|
||||
},
|
||||
},
|
||||
'& input': {
|
||||
color: '#F8F8F2',
|
||||
},
|
||||
}}
|
||||
/>
|
||||
<Button
|
||||
variant="contained"
|
||||
onClick={handleExecute}
|
||||
disabled={isExecuting || !command.trim()}
|
||||
startIcon={<Send />}
|
||||
sx={{
|
||||
backgroundColor: '#5E2750',
|
||||
'&:hover': {
|
||||
backgroundColor: '#772953',
|
||||
},
|
||||
textTransform: 'none',
|
||||
fontWeight: 'bold',
|
||||
}}
|
||||
>
|
||||
Run
|
||||
</Button>
|
||||
</Box>
|
||||
)}
|
||||
</DialogContent>
|
||||
|
||||
<DialogActions>
|
||||
@@ -297,6 +122,13 @@ export default function TerminalModal({
|
||||
Close
|
||||
</Button>
|
||||
</DialogActions>
|
||||
|
||||
<FallbackNotification
|
||||
show={showFallbackNotification}
|
||||
reason={fallbackReason}
|
||||
onClose={() => setShowFallbackNotification(false)}
|
||||
onRetry={handleRetryInteractive}
|
||||
/>
|
||||
</Dialog>
|
||||
);
|
||||
}
|
||||
|
||||
105
frontend/components/TerminalModal/CommandInput.tsx
Normal file
105
frontend/components/TerminalModal/CommandInput.tsx
Normal file
@@ -0,0 +1,105 @@
|
||||
import React from 'react';
|
||||
import { Box, Typography, TextField, Button, IconButton } from '@mui/material';
|
||||
import { Send } from '@mui/icons-material';
|
||||
import { CommandInputProps } from '@/lib/interfaces/terminal';
|
||||
import { formatPrompt } from '@/lib/utils/terminal';
|
||||
|
||||
export default function CommandInput({
|
||||
command,
|
||||
workdir,
|
||||
isExecuting,
|
||||
isMobile,
|
||||
containerName,
|
||||
onCommandChange,
|
||||
onExecute,
|
||||
onKeyPress,
|
||||
}: CommandInputProps) {
|
||||
return (
|
||||
<Box sx={{
|
||||
display: 'flex',
|
||||
flexDirection: isMobile ? 'column' : 'row',
|
||||
gap: 1,
|
||||
alignItems: isMobile ? 'stretch' : 'center'
|
||||
}}>
|
||||
<Typography sx={{
|
||||
fontFamily: '"Ubuntu Mono", monospace',
|
||||
fontSize: { xs: '12px', sm: '14px' },
|
||||
color: '#8BE9FD',
|
||||
fontWeight: 'bold',
|
||||
whiteSpace: 'nowrap',
|
||||
alignSelf: isMobile ? 'flex-start' : 'center'
|
||||
}}>
|
||||
{formatPrompt(containerName, workdir)}
|
||||
</Typography>
|
||||
<Box sx={{ display: 'flex', gap: 1, flex: 1 }}>
|
||||
<TextField
|
||||
fullWidth
|
||||
value={command}
|
||||
onChange={(e) => onCommandChange(e.target.value)}
|
||||
onKeyPress={onKeyPress}
|
||||
placeholder="ls -la"
|
||||
disabled={isExecuting}
|
||||
variant="outlined"
|
||||
size="small"
|
||||
autoFocus
|
||||
sx={{
|
||||
fontFamily: '"Ubuntu Mono", monospace',
|
||||
'& input': {
|
||||
fontFamily: '"Ubuntu Mono", monospace',
|
||||
fontSize: { xs: '12px', sm: '14px' },
|
||||
padding: { xs: '6px 10px', sm: '8px 12px' },
|
||||
color: '#F8F8F2',
|
||||
},
|
||||
'& .MuiOutlinedInput-root': {
|
||||
backgroundColor: '#1E1E1E',
|
||||
'& fieldset': {
|
||||
borderColor: '#5E2750',
|
||||
},
|
||||
'&:hover fieldset': {
|
||||
borderColor: '#772953',
|
||||
},
|
||||
'&.Mui-focused fieldset': {
|
||||
borderColor: '#8BE9FD',
|
||||
},
|
||||
},
|
||||
}}
|
||||
/>
|
||||
{isMobile ? (
|
||||
<IconButton
|
||||
onClick={onExecute}
|
||||
disabled={isExecuting || !command.trim()}
|
||||
sx={{
|
||||
backgroundColor: '#5E2750',
|
||||
color: 'white',
|
||||
'&:hover': {
|
||||
backgroundColor: '#772953',
|
||||
},
|
||||
'&:disabled': {
|
||||
backgroundColor: '#3a1a2f',
|
||||
},
|
||||
}}
|
||||
>
|
||||
<Send />
|
||||
</IconButton>
|
||||
) : (
|
||||
<Button
|
||||
variant="contained"
|
||||
onClick={onExecute}
|
||||
disabled={isExecuting || !command.trim()}
|
||||
startIcon={<Send />}
|
||||
sx={{
|
||||
backgroundColor: '#5E2750',
|
||||
'&:hover': {
|
||||
backgroundColor: '#772953',
|
||||
},
|
||||
textTransform: 'none',
|
||||
fontWeight: 'bold',
|
||||
}}
|
||||
>
|
||||
Run
|
||||
</Button>
|
||||
)}
|
||||
</Box>
|
||||
</Box>
|
||||
);
|
||||
}
|
||||
39
frontend/components/TerminalModal/FallbackNotification.tsx
Normal file
39
frontend/components/TerminalModal/FallbackNotification.tsx
Normal file
@@ -0,0 +1,39 @@
|
||||
import React from 'react';
|
||||
import { Snackbar, Alert, Typography, Button } from '@mui/material';
|
||||
import { Warning } from '@mui/icons-material';
|
||||
import { FallbackNotificationProps } from '@/lib/interfaces/terminal';
|
||||
|
||||
export default function FallbackNotification({
|
||||
show,
|
||||
reason,
|
||||
onClose,
|
||||
onRetry,
|
||||
}: FallbackNotificationProps) {
|
||||
return (
|
||||
<Snackbar
|
||||
open={show}
|
||||
autoHideDuration={10000}
|
||||
onClose={onClose}
|
||||
anchorOrigin={{ vertical: 'top', horizontal: 'center' }}
|
||||
>
|
||||
<Alert
|
||||
severity="warning"
|
||||
icon={<Warning />}
|
||||
action={
|
||||
<Button color="inherit" size="small" onClick={onRetry}>
|
||||
Retry
|
||||
</Button>
|
||||
}
|
||||
onClose={onClose}
|
||||
sx={{ width: '100%', maxWidth: '600px' }}
|
||||
>
|
||||
<Typography variant="body2" sx={{ fontWeight: 600, mb: 0.5 }}>
|
||||
Switched to Simple Mode
|
||||
</Typography>
|
||||
<Typography variant="body2" sx={{ fontSize: '0.875rem' }}>
|
||||
{reason}
|
||||
</Typography>
|
||||
</Alert>
|
||||
</Snackbar>
|
||||
);
|
||||
}
|
||||
28
frontend/components/TerminalModal/InteractiveTerminal.tsx
Normal file
28
frontend/components/TerminalModal/InteractiveTerminal.tsx
Normal file
@@ -0,0 +1,28 @@
|
||||
import React from 'react';
|
||||
import { Box } from '@mui/material';
|
||||
import '@xterm/xterm/css/xterm.css';
|
||||
import { InteractiveTerminalProps } from '@/lib/interfaces/terminal';
|
||||
|
||||
export default function InteractiveTerminal({ terminalRef }: InteractiveTerminalProps) {
|
||||
return (
|
||||
<Box
|
||||
ref={terminalRef}
|
||||
sx={{
|
||||
height: { xs: '400px', sm: '500px' },
|
||||
backgroundColor: '#2E3436',
|
||||
borderRadius: '4px',
|
||||
border: '1px solid #1C1F20',
|
||||
overflow: 'hidden',
|
||||
'& .xterm': {
|
||||
padding: '10px',
|
||||
},
|
||||
'& .xterm-viewport': {
|
||||
backgroundColor: '#2E3436 !important',
|
||||
},
|
||||
'& .xterm-screen': {
|
||||
backgroundColor: '#2E3436',
|
||||
},
|
||||
}}
|
||||
/>
|
||||
);
|
||||
}
|
||||
37
frontend/components/TerminalModal/SimpleTerminal.tsx
Normal file
37
frontend/components/TerminalModal/SimpleTerminal.tsx
Normal file
@@ -0,0 +1,37 @@
|
||||
import React from 'react';
|
||||
import { SimpleTerminalProps } from '@/lib/interfaces/terminal';
|
||||
import TerminalOutput from './TerminalOutput';
|
||||
import CommandInput from './CommandInput';
|
||||
|
||||
export default function SimpleTerminal({
|
||||
output,
|
||||
command,
|
||||
workdir,
|
||||
isExecuting,
|
||||
isMobile,
|
||||
containerName,
|
||||
outputRef,
|
||||
onCommandChange,
|
||||
onExecute,
|
||||
onKeyPress,
|
||||
}: SimpleTerminalProps) {
|
||||
return (
|
||||
<>
|
||||
<TerminalOutput
|
||||
output={output}
|
||||
containerName={containerName}
|
||||
outputRef={outputRef}
|
||||
/>
|
||||
<CommandInput
|
||||
command={command}
|
||||
workdir={workdir}
|
||||
isExecuting={isExecuting}
|
||||
isMobile={isMobile}
|
||||
containerName={containerName}
|
||||
onCommandChange={onCommandChange}
|
||||
onExecute={onExecute}
|
||||
onKeyPress={onKeyPress}
|
||||
/>
|
||||
</>
|
||||
);
|
||||
}
|
||||
86
frontend/components/TerminalModal/TerminalHeader.tsx
Normal file
86
frontend/components/TerminalModal/TerminalHeader.tsx
Normal file
@@ -0,0 +1,86 @@
|
||||
import React from 'react';
|
||||
import {
|
||||
DialogTitle,
|
||||
Box,
|
||||
Typography,
|
||||
IconButton,
|
||||
ToggleButtonGroup,
|
||||
ToggleButton,
|
||||
Tooltip,
|
||||
} from '@mui/material';
|
||||
import { Close, Terminal as TerminalIcon, Code, Warning } from '@mui/icons-material';
|
||||
import { TerminalHeaderProps } from '@/lib/interfaces/terminal';
|
||||
|
||||
export default function TerminalHeader({
|
||||
containerName,
|
||||
mode,
|
||||
interactiveFailed,
|
||||
onModeChange,
|
||||
onClose,
|
||||
}: TerminalHeaderProps) {
|
||||
return (
|
||||
<DialogTitle
|
||||
sx={{
|
||||
display: 'flex',
|
||||
justifyContent: 'space-between',
|
||||
alignItems: 'center',
|
||||
pb: 2,
|
||||
pt: { xs: 1, sm: 2 },
|
||||
px: { xs: 2, sm: 3 },
|
||||
flexWrap: 'wrap',
|
||||
gap: 2,
|
||||
}}
|
||||
>
|
||||
<Box sx={{ display: 'flex', flexDirection: 'column', gap: 1, flex: 1 }}>
|
||||
<Typography
|
||||
variant="h2"
|
||||
component="div"
|
||||
sx={{ fontSize: { xs: '1.1rem', sm: '1.5rem' } }}
|
||||
>
|
||||
Terminal - {containerName}
|
||||
</Typography>
|
||||
<ToggleButtonGroup
|
||||
value={mode}
|
||||
exclusive
|
||||
onChange={onModeChange}
|
||||
size="small"
|
||||
sx={{ display: 'flex' }}
|
||||
>
|
||||
<Tooltip title={interactiveFailed ? "Interactive mode failed - click to retry" : "Interactive mode with full terminal support (sudo, nano, vim)"}>
|
||||
<ToggleButton
|
||||
value="interactive"
|
||||
sx={{
|
||||
flex: 1,
|
||||
fontSize: { xs: '0.75rem', sm: '0.875rem' },
|
||||
...(interactiveFailed && {
|
||||
borderColor: '#f59e0b',
|
||||
color: '#f59e0b',
|
||||
'&:hover': {
|
||||
borderColor: '#d97706',
|
||||
backgroundColor: 'rgba(245, 158, 11, 0.1)',
|
||||
},
|
||||
}),
|
||||
}}
|
||||
>
|
||||
{interactiveFailed ? (
|
||||
<Warning sx={{ mr: 0.5, fontSize: '1rem' }} />
|
||||
) : (
|
||||
<TerminalIcon sx={{ mr: 0.5, fontSize: '1rem' }} />
|
||||
)}
|
||||
Interactive
|
||||
</ToggleButton>
|
||||
</Tooltip>
|
||||
<Tooltip title="Simple command execution mode">
|
||||
<ToggleButton value="simple" sx={{ flex: 1, fontSize: { xs: '0.75rem', sm: '0.875rem' } }}>
|
||||
<Code sx={{ mr: 0.5, fontSize: '1rem' }} />
|
||||
Simple
|
||||
</ToggleButton>
|
||||
</Tooltip>
|
||||
</ToggleButtonGroup>
|
||||
</Box>
|
||||
<IconButton onClick={onClose} size="small">
|
||||
<Close />
|
||||
</IconButton>
|
||||
</DialogTitle>
|
||||
);
|
||||
}
|
||||
67
frontend/components/TerminalModal/TerminalOutput.tsx
Normal file
67
frontend/components/TerminalModal/TerminalOutput.tsx
Normal file
@@ -0,0 +1,67 @@
|
||||
import React from 'react';
|
||||
import { Box, Paper, Typography } from '@mui/material';
|
||||
import { TerminalOutputProps } from '@/lib/interfaces/terminal';
|
||||
import { highlightCommand } from '@/lib/utils/terminal';
|
||||
|
||||
export default function TerminalOutput({ output, containerName, outputRef }: TerminalOutputProps) {
|
||||
return (
|
||||
<Paper
|
||||
ref={outputRef}
|
||||
elevation={0}
|
||||
sx={{
|
||||
backgroundColor: '#2E3436',
|
||||
color: '#D3D7CF',
|
||||
fontFamily: '"Ubuntu Mono", "DejaVu Sans Mono", "Courier New", monospace',
|
||||
fontSize: { xs: '12px', sm: '14px' },
|
||||
padding: { xs: 1.5, sm: 2 },
|
||||
minHeight: { xs: '300px', sm: '400px' },
|
||||
maxHeight: { xs: '400px', sm: '500px' },
|
||||
overflowY: 'auto',
|
||||
mb: 2,
|
||||
border: '1px solid #1C1F20',
|
||||
borderRadius: '4px',
|
||||
'&::-webkit-scrollbar': {
|
||||
width: { xs: '6px', sm: '10px' },
|
||||
},
|
||||
'&::-webkit-scrollbar-track': {
|
||||
background: '#1C1F20',
|
||||
},
|
||||
'&::-webkit-scrollbar-thumb': {
|
||||
background: '#555753',
|
||||
borderRadius: '5px',
|
||||
'&:hover': {
|
||||
background: '#729FCF',
|
||||
}
|
||||
},
|
||||
}}
|
||||
>
|
||||
{output.length === 0 ? (
|
||||
<Box>
|
||||
<Typography sx={{
|
||||
color: '#729FCF',
|
||||
fontFamily: 'inherit',
|
||||
fontSize: '13px',
|
||||
mb: 1
|
||||
}}>
|
||||
Ubuntu-style Terminal - Connected to <span style={{ color: '#8AE234', fontWeight: 'bold' }}>{containerName}</span>
|
||||
</Typography>
|
||||
<Typography sx={{
|
||||
color: '#555753',
|
||||
fontFamily: 'inherit',
|
||||
fontSize: '12px'
|
||||
}}>
|
||||
Type a command and press Enter or click Execute...
|
||||
</Typography>
|
||||
</Box>
|
||||
) : (
|
||||
<Box>
|
||||
{output.map((line, index) => (
|
||||
<React.Fragment key={index}>
|
||||
{highlightCommand(line, containerName)}
|
||||
</React.Fragment>
|
||||
))}
|
||||
</Box>
|
||||
)}
|
||||
</Paper>
|
||||
);
|
||||
}
|
||||
@@ -0,0 +1,88 @@
|
||||
import React from 'react';
|
||||
import { render, screen, fireEvent } from '@testing-library/react';
|
||||
import TerminalHeader from '../TerminalHeader';
|
||||
|
||||
describe('TerminalHeader', () => {
|
||||
const mockOnClose = jest.fn();
|
||||
const mockOnModeChange = jest.fn();
|
||||
|
||||
beforeEach(() => {
|
||||
jest.clearAllMocks();
|
||||
});
|
||||
|
||||
it('renders container name', () => {
|
||||
render(
|
||||
<TerminalHeader
|
||||
containerName="test-container"
|
||||
mode="interactive"
|
||||
interactiveFailed={false}
|
||||
onModeChange={mockOnModeChange}
|
||||
onClose={mockOnClose}
|
||||
/>
|
||||
);
|
||||
|
||||
expect(screen.getByText(/Terminal - test-container/i)).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('renders interactive and simple mode buttons', () => {
|
||||
render(
|
||||
<TerminalHeader
|
||||
containerName="test-container"
|
||||
mode="interactive"
|
||||
interactiveFailed={false}
|
||||
onModeChange={mockOnModeChange}
|
||||
onClose={mockOnClose}
|
||||
/>
|
||||
);
|
||||
|
||||
expect(screen.getByText('Interactive')).toBeInTheDocument();
|
||||
expect(screen.getByText('Simple')).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('calls onClose when close button is clicked', () => {
|
||||
render(
|
||||
<TerminalHeader
|
||||
containerName="test-container"
|
||||
mode="interactive"
|
||||
interactiveFailed={false}
|
||||
onModeChange={mockOnModeChange}
|
||||
onClose={mockOnClose}
|
||||
/>
|
||||
);
|
||||
|
||||
const closeButton = screen.getByRole('button', { name: '' });
|
||||
fireEvent.click(closeButton);
|
||||
|
||||
expect(mockOnClose).toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it('shows warning icon when interactive mode failed', () => {
|
||||
const { container } = render(
|
||||
<TerminalHeader
|
||||
containerName="test-container"
|
||||
mode="simple"
|
||||
interactiveFailed={true}
|
||||
onModeChange={mockOnModeChange}
|
||||
onClose={mockOnClose}
|
||||
/>
|
||||
);
|
||||
|
||||
const warningIcon = container.querySelector('[data-testid="WarningIcon"]');
|
||||
expect(warningIcon).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('applies correct mode selection', () => {
|
||||
render(
|
||||
<TerminalHeader
|
||||
containerName="test-container"
|
||||
mode="simple"
|
||||
interactiveFailed={false}
|
||||
onModeChange={mockOnModeChange}
|
||||
onClose={mockOnClose}
|
||||
/>
|
||||
);
|
||||
|
||||
const simpleButton = screen.getByText('Simple').closest('button');
|
||||
expect(simpleButton).toHaveClass('Mui-selected');
|
||||
});
|
||||
});
|
||||
78
frontend/components/__tests__/LoginForm.test.tsx
Normal file
78
frontend/components/__tests__/LoginForm.test.tsx
Normal file
@@ -0,0 +1,78 @@
|
||||
import React from 'react';
|
||||
import { render, screen, fireEvent } from '@testing-library/react';
|
||||
import { Provider } from 'react-redux';
|
||||
import { configureStore } from '@reduxjs/toolkit';
|
||||
import authReducer from '@/lib/store/authSlice';
|
||||
import LoginForm from '../LoginForm';
|
||||
|
||||
jest.mock('next/navigation', () => ({
|
||||
useRouter: jest.fn(() => ({
|
||||
push: jest.fn(),
|
||||
})),
|
||||
}));
|
||||
|
||||
const createMockStore = (loading = false) =>
|
||||
configureStore({
|
||||
reducer: {
|
||||
auth: authReducer,
|
||||
},
|
||||
preloadedState: {
|
||||
auth: {
|
||||
isAuthenticated: false,
|
||||
loading,
|
||||
username: null,
|
||||
error: null,
|
||||
},
|
||||
},
|
||||
});
|
||||
|
||||
const renderWithProvider = (component: React.ReactElement, loading = false) => {
|
||||
return render(<Provider store={createMockStore(loading)}>{component}</Provider>);
|
||||
};
|
||||
|
||||
describe('LoginForm', () => {
|
||||
it('renders login form elements', () => {
|
||||
renderWithProvider(<LoginForm />);
|
||||
|
||||
expect(screen.getByLabelText(/username/i)).toBeInTheDocument();
|
||||
expect(screen.getByLabelText(/password/i)).toBeInTheDocument();
|
||||
expect(screen.getByRole('button', { name: /access dashboard/i })).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('updates username input on change', () => {
|
||||
renderWithProvider(<LoginForm />);
|
||||
|
||||
const usernameInput = screen.getByLabelText(/username/i) as HTMLInputElement;
|
||||
fireEvent.change(usernameInput, { target: { value: 'testuser' } });
|
||||
|
||||
expect(usernameInput.value).toBe('testuser');
|
||||
});
|
||||
|
||||
it('updates password input on change', () => {
|
||||
renderWithProvider(<LoginForm />);
|
||||
|
||||
const passwordInput = screen.getByLabelText(/password/i) as HTMLInputElement;
|
||||
fireEvent.change(passwordInput, { target: { value: 'testpass' } });
|
||||
|
||||
expect(passwordInput.value).toBe('testpass');
|
||||
});
|
||||
|
||||
it('shows loading text when loading', () => {
|
||||
renderWithProvider(<LoginForm />, true);
|
||||
|
||||
expect(screen.getByRole('button', { name: /logging in/i })).toBeInTheDocument();
|
||||
});
|
||||
|
||||
it('password input is type password', () => {
|
||||
renderWithProvider(<LoginForm />);
|
||||
|
||||
const passwordInput = screen.getByLabelText(/password/i) as HTMLInputElement;
|
||||
expect(passwordInput.type).toBe('password');
|
||||
});
|
||||
|
||||
it('shows helper text with default credentials', () => {
|
||||
renderWithProvider(<LoginForm />);
|
||||
|
||||
expect(screen.getByText(/default: admin \/ admin123/i)).toBeInTheDocument();
|
||||
});
|
||||
});
|
||||
BIN
frontend/gnome-interactive-demo.png
Normal file
BIN
frontend/gnome-interactive-demo.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 110 KiB |
27
frontend/jest.config.js
Normal file
27
frontend/jest.config.js
Normal file
@@ -0,0 +1,27 @@
|
||||
const nextJest = require('next/jest')
|
||||
|
||||
const createJestConfig = nextJest({
|
||||
dir: './',
|
||||
})
|
||||
|
||||
const customJestConfig = {
|
||||
setupFilesAfterEnv: ['<rootDir>/jest.setup.js'],
|
||||
testEnvironment: 'jest-environment-jsdom',
|
||||
moduleNameMapper: {
|
||||
'^@/(.*)$': '<rootDir>/$1',
|
||||
},
|
||||
testMatch: [
|
||||
'**/__tests__/**/*.[jt]s?(x)',
|
||||
'**/?(*.)+(spec|test).[jt]s?(x)',
|
||||
],
|
||||
collectCoverageFrom: [
|
||||
'lib/**/*.{js,jsx,ts,tsx}',
|
||||
'components/**/*.{js,jsx,ts,tsx}',
|
||||
'app/**/*.{js,jsx,ts,tsx}',
|
||||
'!**/*.d.ts',
|
||||
'!**/node_modules/**',
|
||||
'!**/.next/**',
|
||||
],
|
||||
}
|
||||
|
||||
module.exports = createJestConfig(customJestConfig)
|
||||
28
frontend/jest.setup.js
Normal file
28
frontend/jest.setup.js
Normal file
@@ -0,0 +1,28 @@
|
||||
import '@testing-library/jest-dom'
|
||||
|
||||
// Mock window.matchMedia
|
||||
Object.defineProperty(window, 'matchMedia', {
|
||||
writable: true,
|
||||
value: jest.fn().mockImplementation(query => ({
|
||||
matches: false,
|
||||
media: query,
|
||||
onchange: null,
|
||||
addListener: jest.fn(),
|
||||
removeListener: jest.fn(),
|
||||
addEventListener: jest.fn(),
|
||||
removeEventListener: jest.fn(),
|
||||
dispatchEvent: jest.fn(),
|
||||
})),
|
||||
})
|
||||
|
||||
// Mock localStorage
|
||||
const localStorageMock = {
|
||||
getItem: jest.fn(),
|
||||
setItem: jest.fn(),
|
||||
removeItem: jest.fn(),
|
||||
clear: jest.fn(),
|
||||
}
|
||||
global.localStorage = localStorageMock
|
||||
|
||||
// Mock fetch
|
||||
global.fetch = jest.fn()
|
||||
@@ -1,5 +1,7 @@
|
||||
export const API_BASE_URL =
|
||||
typeof window !== 'undefined' && (window as any).__ENV__?.NEXT_PUBLIC_API_URL
|
||||
import { triggerAuthError } from './store/authErrorHandler';
|
||||
|
||||
export const API_BASE_URL =
|
||||
typeof window !== 'undefined' && (window as any).__ENV__?.NEXT_PUBLIC_API_URL
|
||||
? (window as any).__ENV__.NEXT_PUBLIC_API_URL
|
||||
: process.env.NEXT_PUBLIC_API_URL || 'http://localhost:5000';
|
||||
|
||||
@@ -31,6 +33,7 @@ class ApiClient {
|
||||
localStorage.setItem('auth_token', token);
|
||||
} else {
|
||||
localStorage.removeItem('auth_token');
|
||||
localStorage.removeItem('auth_username');
|
||||
}
|
||||
}
|
||||
|
||||
@@ -41,6 +44,23 @@ class ApiClient {
|
||||
return this.token;
|
||||
}
|
||||
|
||||
getUsername(): string | null {
|
||||
if (typeof window !== 'undefined') {
|
||||
return localStorage.getItem('auth_username');
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
setUsername(username: string | null) {
|
||||
if (typeof window !== 'undefined') {
|
||||
if (username) {
|
||||
localStorage.setItem('auth_username', username);
|
||||
} else {
|
||||
localStorage.removeItem('auth_username');
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
async login(username: string, password: string): Promise<AuthResponse> {
|
||||
const response = await fetch(`${API_BASE_URL}/api/auth/login`, {
|
||||
method: 'POST',
|
||||
@@ -53,6 +73,7 @@ class ApiClient {
|
||||
const data = await response.json();
|
||||
if (data.success && data.token) {
|
||||
this.setToken(data.token);
|
||||
this.setUsername(data.username || username);
|
||||
}
|
||||
return data;
|
||||
}
|
||||
@@ -73,6 +94,7 @@ class ApiClient {
|
||||
async getContainers(): Promise<Container[]> {
|
||||
const token = this.getToken();
|
||||
if (!token) {
|
||||
triggerAuthError();
|
||||
throw new Error('Not authenticated');
|
||||
}
|
||||
|
||||
@@ -85,6 +107,7 @@ class ApiClient {
|
||||
if (!response.ok) {
|
||||
if (response.status === 401) {
|
||||
this.setToken(null);
|
||||
triggerAuthError();
|
||||
throw new Error('Session expired');
|
||||
}
|
||||
throw new Error('Failed to fetch containers');
|
||||
@@ -97,6 +120,7 @@ class ApiClient {
|
||||
async executeCommand(containerId: string, command: string): Promise<any> {
|
||||
const token = this.getToken();
|
||||
if (!token) {
|
||||
triggerAuthError();
|
||||
throw new Error('Not authenticated');
|
||||
}
|
||||
|
||||
@@ -110,11 +134,124 @@ class ApiClient {
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
if (response.status === 401) {
|
||||
this.setToken(null);
|
||||
triggerAuthError();
|
||||
throw new Error('Session expired');
|
||||
}
|
||||
throw new Error('Failed to execute command');
|
||||
}
|
||||
|
||||
return response.json();
|
||||
}
|
||||
|
||||
async startContainer(containerId: string): Promise<any> {
|
||||
const token = this.getToken();
|
||||
if (!token) {
|
||||
triggerAuthError();
|
||||
throw new Error('Not authenticated');
|
||||
}
|
||||
|
||||
const response = await fetch(`${API_BASE_URL}/api/containers/${containerId}/start`, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Authorization': `Bearer ${token}`,
|
||||
},
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
if (response.status === 401) {
|
||||
this.setToken(null);
|
||||
triggerAuthError();
|
||||
throw new Error('Session expired');
|
||||
}
|
||||
const error = await response.json();
|
||||
throw new Error(error.error || 'Failed to start container');
|
||||
}
|
||||
|
||||
return response.json();
|
||||
}
|
||||
|
||||
async stopContainer(containerId: string): Promise<any> {
|
||||
const token = this.getToken();
|
||||
if (!token) {
|
||||
triggerAuthError();
|
||||
throw new Error('Not authenticated');
|
||||
}
|
||||
|
||||
const response = await fetch(`${API_BASE_URL}/api/containers/${containerId}/stop`, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Authorization': `Bearer ${token}`,
|
||||
},
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
if (response.status === 401) {
|
||||
this.setToken(null);
|
||||
triggerAuthError();
|
||||
throw new Error('Session expired');
|
||||
}
|
||||
const error = await response.json();
|
||||
throw new Error(error.error || 'Failed to stop container');
|
||||
}
|
||||
|
||||
return response.json();
|
||||
}
|
||||
|
||||
async restartContainer(containerId: string): Promise<any> {
|
||||
const token = this.getToken();
|
||||
if (!token) {
|
||||
triggerAuthError();
|
||||
throw new Error('Not authenticated');
|
||||
}
|
||||
|
||||
const response = await fetch(`${API_BASE_URL}/api/containers/${containerId}/restart`, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Authorization': `Bearer ${token}`,
|
||||
},
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
if (response.status === 401) {
|
||||
this.setToken(null);
|
||||
triggerAuthError();
|
||||
throw new Error('Session expired');
|
||||
}
|
||||
const error = await response.json();
|
||||
throw new Error(error.error || 'Failed to restart container');
|
||||
}
|
||||
|
||||
return response.json();
|
||||
}
|
||||
|
||||
async removeContainer(containerId: string): Promise<any> {
|
||||
const token = this.getToken();
|
||||
if (!token) {
|
||||
triggerAuthError();
|
||||
throw new Error('Not authenticated');
|
||||
}
|
||||
|
||||
const response = await fetch(`${API_BASE_URL}/api/containers/${containerId}`, {
|
||||
method: 'DELETE',
|
||||
headers: {
|
||||
'Authorization': `Bearer ${token}`,
|
||||
},
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
if (response.status === 401) {
|
||||
this.setToken(null);
|
||||
triggerAuthError();
|
||||
throw new Error('Session expired');
|
||||
}
|
||||
const error = await response.json();
|
||||
throw new Error(error.error || 'Failed to remove container');
|
||||
}
|
||||
|
||||
return response.json();
|
||||
}
|
||||
}
|
||||
|
||||
export const apiClient = new ApiClient();
|
||||
|
||||
@@ -1,70 +0,0 @@
|
||||
'use client';
|
||||
|
||||
import React, { createContext, useContext, useState, useEffect } from 'react';
|
||||
import { apiClient } from './api';
|
||||
|
||||
interface AuthContextType {
|
||||
isAuthenticated: boolean;
|
||||
username: string | null;
|
||||
login: (username: string, password: string) => Promise<boolean>;
|
||||
logout: () => Promise<void>;
|
||||
loading: boolean;
|
||||
}
|
||||
|
||||
const AuthContext = createContext<AuthContextType | undefined>(undefined);
|
||||
|
||||
export function AuthProvider({ children }: { children: React.ReactNode }) {
|
||||
const [isAuthenticated, setIsAuthenticated] = useState(false);
|
||||
const [username, setUsername] = useState<string | null>(null);
|
||||
const [loading, setLoading] = useState(true);
|
||||
|
||||
useEffect(() => {
|
||||
// Check if user has a valid token
|
||||
const token = apiClient.getToken();
|
||||
if (token) {
|
||||
setIsAuthenticated(true);
|
||||
// In a real app, you'd validate the token with the backend
|
||||
}
|
||||
setLoading(false);
|
||||
}, []);
|
||||
|
||||
const login = async (username: string, password: string): Promise<boolean> => {
|
||||
try {
|
||||
const response = await apiClient.login(username, password);
|
||||
if (response.success) {
|
||||
setIsAuthenticated(true);
|
||||
setUsername(response.username || username);
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
} catch (error) {
|
||||
console.error('Login error:', error);
|
||||
return false;
|
||||
}
|
||||
};
|
||||
|
||||
const logout = async () => {
|
||||
try {
|
||||
await apiClient.logout();
|
||||
} catch (error) {
|
||||
console.error('Logout error:', error);
|
||||
} finally {
|
||||
setIsAuthenticated(false);
|
||||
setUsername(null);
|
||||
}
|
||||
};
|
||||
|
||||
return (
|
||||
<AuthContext.Provider value={{ isAuthenticated, username, login, logout, loading }}>
|
||||
{children}
|
||||
</AuthContext.Provider>
|
||||
);
|
||||
}
|
||||
|
||||
export function useAuth() {
|
||||
const context = useContext(AuthContext);
|
||||
if (context === undefined) {
|
||||
throw new Error('useAuth must be used within an AuthProvider');
|
||||
}
|
||||
return context;
|
||||
}
|
||||
69
frontend/lib/hooks/__tests__/useAuthRedirect.test.tsx
Normal file
69
frontend/lib/hooks/__tests__/useAuthRedirect.test.tsx
Normal file
@@ -0,0 +1,69 @@
|
||||
import { renderHook } from '@testing-library/react';
|
||||
import { useRouter } from 'next/navigation';
|
||||
import { Provider } from 'react-redux';
|
||||
import { configureStore } from '@reduxjs/toolkit';
|
||||
import authReducer from '@/lib/store/authSlice';
|
||||
import { useAuthRedirect } from '../useAuthRedirect';
|
||||
|
||||
jest.mock('next/navigation', () => ({
|
||||
useRouter: jest.fn(),
|
||||
}));
|
||||
|
||||
const createMockStore = (isAuthenticated: boolean) =>
|
||||
configureStore({
|
||||
reducer: {
|
||||
auth: authReducer,
|
||||
},
|
||||
preloadedState: {
|
||||
auth: {
|
||||
isAuthenticated,
|
||||
loading: false,
|
||||
username: isAuthenticated ? 'testuser' : null,
|
||||
error: null,
|
||||
},
|
||||
},
|
||||
});
|
||||
|
||||
describe('useAuthRedirect', () => {
|
||||
const mockPush = jest.fn();
|
||||
|
||||
beforeEach(() => {
|
||||
jest.clearAllMocks();
|
||||
(useRouter as jest.Mock).mockReturnValue({
|
||||
push: mockPush,
|
||||
});
|
||||
});
|
||||
|
||||
it('redirects to dashboard when authenticated and redirectTo is dashboard', () => {
|
||||
const store = createMockStore(true);
|
||||
const wrapper = ({ children }: { children: React.ReactNode }) => (
|
||||
<Provider store={store}>{children}</Provider>
|
||||
);
|
||||
|
||||
renderHook(() => useAuthRedirect('/dashboard'), { wrapper });
|
||||
|
||||
expect(mockPush).toHaveBeenCalledWith('/dashboard');
|
||||
});
|
||||
|
||||
it('redirects to login when not authenticated and redirectTo is /', () => {
|
||||
const store = createMockStore(false);
|
||||
const wrapper = ({ children }: { children: React.ReactNode }) => (
|
||||
<Provider store={store}>{children}</Provider>
|
||||
);
|
||||
|
||||
renderHook(() => useAuthRedirect('/'), { wrapper });
|
||||
|
||||
expect(mockPush).toHaveBeenCalledWith('/');
|
||||
});
|
||||
|
||||
it('does not redirect when authenticated but redirectTo is /', () => {
|
||||
const store = createMockStore(true);
|
||||
const wrapper = ({ children }: { children: React.ReactNode }) => (
|
||||
<Provider store={store}>{children}</Provider>
|
||||
);
|
||||
|
||||
renderHook(() => useAuthRedirect('/'), { wrapper });
|
||||
|
||||
expect(mockPush).not.toHaveBeenCalled();
|
||||
});
|
||||
});
|
||||
90
frontend/lib/hooks/__tests__/useLoginForm.test.tsx
Normal file
90
frontend/lib/hooks/__tests__/useLoginForm.test.tsx
Normal file
@@ -0,0 +1,90 @@
|
||||
import { renderHook, act } from '@testing-library/react';
|
||||
import { useRouter } from 'next/navigation';
|
||||
import { Provider } from 'react-redux';
|
||||
import { configureStore } from '@reduxjs/toolkit';
|
||||
import authReducer from '@/lib/store/authSlice';
|
||||
import { useLoginForm } from '../useLoginForm';
|
||||
|
||||
jest.mock('next/navigation', () => ({
|
||||
useRouter: jest.fn(),
|
||||
}));
|
||||
|
||||
const createMockStore = () =>
|
||||
configureStore({
|
||||
reducer: {
|
||||
auth: authReducer,
|
||||
},
|
||||
});
|
||||
|
||||
const wrapper = ({ children }: { children: React.ReactNode }) => (
|
||||
<Provider store={createMockStore()}>{children}</Provider>
|
||||
);
|
||||
|
||||
describe('useLoginForm', () => {
|
||||
const mockPush = jest.fn();
|
||||
|
||||
beforeEach(() => {
|
||||
jest.clearAllMocks();
|
||||
(useRouter as jest.Mock).mockReturnValue({
|
||||
push: mockPush,
|
||||
});
|
||||
});
|
||||
|
||||
it('initializes with empty username and password', () => {
|
||||
const { result } = renderHook(() => useLoginForm(), { wrapper });
|
||||
|
||||
expect(result.current.username).toBe('');
|
||||
expect(result.current.password).toBe('');
|
||||
expect(result.current.error).toBeNull();
|
||||
});
|
||||
|
||||
it('updates username when setUsername is called', () => {
|
||||
const { result } = renderHook(() => useLoginForm(), { wrapper });
|
||||
|
||||
act(() => {
|
||||
result.current.setUsername('testuser');
|
||||
});
|
||||
|
||||
expect(result.current.username).toBe('testuser');
|
||||
});
|
||||
|
||||
it('updates password when setPassword is called', () => {
|
||||
const { result } = renderHook(() => useLoginForm(), { wrapper });
|
||||
|
||||
act(() => {
|
||||
result.current.setPassword('testpass');
|
||||
});
|
||||
|
||||
expect(result.current.password).toBe('testpass');
|
||||
});
|
||||
|
||||
it('handles form submission', async () => {
|
||||
const { result } = renderHook(() => useLoginForm(), { wrapper });
|
||||
const mockEvent = {
|
||||
preventDefault: jest.fn(),
|
||||
} as unknown as React.FormEvent;
|
||||
|
||||
act(() => {
|
||||
result.current.setUsername('testuser');
|
||||
result.current.setPassword('testpass');
|
||||
});
|
||||
|
||||
await act(async () => {
|
||||
await result.current.handleSubmit(mockEvent);
|
||||
});
|
||||
|
||||
expect(mockEvent.preventDefault).toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it('returns loading state', () => {
|
||||
const { result } = renderHook(() => useLoginForm(), { wrapper });
|
||||
|
||||
expect(result.current.loading).toBeDefined();
|
||||
});
|
||||
|
||||
it('returns isShaking state', () => {
|
||||
const { result } = renderHook(() => useLoginForm(), { wrapper });
|
||||
|
||||
expect(result.current.isShaking).toBe(false);
|
||||
});
|
||||
});
|
||||
61
frontend/lib/hooks/__tests__/useTerminalModal.test.tsx
Normal file
61
frontend/lib/hooks/__tests__/useTerminalModal.test.tsx
Normal file
@@ -0,0 +1,61 @@
|
||||
import { renderHook, act } from '@testing-library/react';
|
||||
import { useTerminalModal } from '../useTerminalModal';
|
||||
|
||||
describe('useTerminalModal', () => {
|
||||
it('initializes with modal closed and no container selected', () => {
|
||||
const { result } = renderHook(() => useTerminalModal());
|
||||
|
||||
expect(result.current.isTerminalOpen).toBe(false);
|
||||
expect(result.current.selectedContainer).toBeNull();
|
||||
});
|
||||
|
||||
it('opens modal with selected container', () => {
|
||||
const { result } = renderHook(() => useTerminalModal());
|
||||
const mockContainer = { id: '123', name: 'test-container' } as any;
|
||||
|
||||
act(() => {
|
||||
result.current.openTerminal(mockContainer);
|
||||
});
|
||||
|
||||
expect(result.current.isTerminalOpen).toBe(true);
|
||||
expect(result.current.selectedContainer).toEqual(mockContainer);
|
||||
});
|
||||
|
||||
it('closes modal and eventually clears selected container', async () => {
|
||||
const { result } = renderHook(() => useTerminalModal());
|
||||
const mockContainer = { id: '123', name: 'test-container' } as any;
|
||||
|
||||
act(() => {
|
||||
result.current.openTerminal(mockContainer);
|
||||
});
|
||||
|
||||
expect(result.current.isTerminalOpen).toBe(true);
|
||||
|
||||
act(() => {
|
||||
result.current.closeTerminal();
|
||||
});
|
||||
|
||||
expect(result.current.isTerminalOpen).toBe(false);
|
||||
});
|
||||
|
||||
it('handles multiple open and close cycles', () => {
|
||||
const { result } = renderHook(() => useTerminalModal());
|
||||
const container1 = { id: '123', name: 'container1' } as any;
|
||||
const container2 = { id: '456', name: 'container2' } as any;
|
||||
|
||||
act(() => {
|
||||
result.current.openTerminal(container1);
|
||||
});
|
||||
expect(result.current.selectedContainer).toEqual(container1);
|
||||
|
||||
act(() => {
|
||||
result.current.closeTerminal();
|
||||
});
|
||||
expect(result.current.isTerminalOpen).toBe(false);
|
||||
|
||||
act(() => {
|
||||
result.current.openTerminal(container2);
|
||||
});
|
||||
expect(result.current.selectedContainer).toEqual(container2);
|
||||
});
|
||||
});
|
||||
20
frontend/lib/hooks/useAuthRedirect.ts
Normal file
20
frontend/lib/hooks/useAuthRedirect.ts
Normal file
@@ -0,0 +1,20 @@
|
||||
import { useEffect } from 'react';
|
||||
import { useRouter } from 'next/navigation';
|
||||
import { useAppSelector } from '@/lib/store/hooks';
|
||||
|
||||
export function useAuthRedirect(redirectTo: '/dashboard' | '/') {
|
||||
const { isAuthenticated, loading } = useAppSelector((state) => state.auth);
|
||||
const router = useRouter();
|
||||
|
||||
useEffect(() => {
|
||||
if (loading) return;
|
||||
|
||||
if (redirectTo === '/dashboard' && isAuthenticated) {
|
||||
router.push('/dashboard');
|
||||
} else if (redirectTo === '/' && !isAuthenticated) {
|
||||
router.push('/');
|
||||
}
|
||||
}, [isAuthenticated, loading, router, redirectTo]);
|
||||
|
||||
return { isAuthenticated, loading };
|
||||
}
|
||||
87
frontend/lib/hooks/useContainerActions.ts
Normal file
87
frontend/lib/hooks/useContainerActions.ts
Normal file
@@ -0,0 +1,87 @@
|
||||
import { useState } from 'react';
|
||||
import { apiClient } from '@/lib/api';
|
||||
|
||||
export function useContainerActions(containerId: string, onUpdate?: () => void) {
|
||||
const [isLoading, setIsLoading] = useState(false);
|
||||
const [snackbar, setSnackbar] = useState<{
|
||||
open: boolean;
|
||||
message: string;
|
||||
severity: 'success' | 'error';
|
||||
}>({
|
||||
open: false,
|
||||
message: '',
|
||||
severity: 'success',
|
||||
});
|
||||
|
||||
const showSuccess = (message: string) => {
|
||||
setSnackbar({ open: true, message, severity: 'success' });
|
||||
onUpdate?.();
|
||||
};
|
||||
|
||||
const showError = (action: string, error: unknown) => {
|
||||
const message = `Failed to ${action}: ${error instanceof Error ? error.message : 'Unknown error'}`;
|
||||
setSnackbar({ open: true, message, severity: 'error' });
|
||||
};
|
||||
|
||||
const handleStart = async () => {
|
||||
setIsLoading(true);
|
||||
try {
|
||||
await apiClient.startContainer(containerId);
|
||||
showSuccess('Container started successfully');
|
||||
} catch (error) {
|
||||
showError('start', error);
|
||||
} finally {
|
||||
setIsLoading(false);
|
||||
}
|
||||
};
|
||||
|
||||
const handleStop = async () => {
|
||||
setIsLoading(true);
|
||||
try {
|
||||
await apiClient.stopContainer(containerId);
|
||||
showSuccess('Container stopped successfully');
|
||||
} catch (error) {
|
||||
showError('stop', error);
|
||||
} finally {
|
||||
setIsLoading(false);
|
||||
}
|
||||
};
|
||||
|
||||
const handleRestart = async () => {
|
||||
setIsLoading(true);
|
||||
try {
|
||||
await apiClient.restartContainer(containerId);
|
||||
showSuccess('Container restarted successfully');
|
||||
} catch (error) {
|
||||
showError('restart', error);
|
||||
} finally {
|
||||
setIsLoading(false);
|
||||
}
|
||||
};
|
||||
|
||||
const handleRemove = async () => {
|
||||
setIsLoading(true);
|
||||
try {
|
||||
await apiClient.removeContainer(containerId);
|
||||
showSuccess('Container removed successfully');
|
||||
} catch (error) {
|
||||
showError('remove', error);
|
||||
} finally {
|
||||
setIsLoading(false);
|
||||
}
|
||||
};
|
||||
|
||||
const closeSnackbar = () => {
|
||||
setSnackbar({ ...snackbar, open: false });
|
||||
};
|
||||
|
||||
return {
|
||||
isLoading,
|
||||
snackbar,
|
||||
handleStart,
|
||||
handleStop,
|
||||
handleRestart,
|
||||
handleRemove,
|
||||
closeSnackbar,
|
||||
};
|
||||
}
|
||||
39
frontend/lib/hooks/useContainerList.ts
Normal file
39
frontend/lib/hooks/useContainerList.ts
Normal file
@@ -0,0 +1,39 @@
|
||||
import { useState, useEffect } from 'react';
|
||||
import { apiClient, Container } from '@/lib/api';
|
||||
|
||||
export function useContainerList(isAuthenticated: boolean) {
|
||||
const [containers, setContainers] = useState<Container[]>([]);
|
||||
const [isRefreshing, setIsRefreshing] = useState(false);
|
||||
const [isLoading, setIsLoading] = useState(true);
|
||||
const [error, setError] = useState('');
|
||||
|
||||
const fetchContainers = async () => {
|
||||
setIsRefreshing(true);
|
||||
setError('');
|
||||
try {
|
||||
const data = await apiClient.getContainers();
|
||||
setContainers(data);
|
||||
} catch (err) {
|
||||
setError(err instanceof Error ? err.message : 'Failed to fetch containers');
|
||||
} finally {
|
||||
setIsRefreshing(false);
|
||||
setIsLoading(false);
|
||||
}
|
||||
};
|
||||
|
||||
useEffect(() => {
|
||||
if (isAuthenticated) {
|
||||
fetchContainers();
|
||||
const interval = setInterval(fetchContainers, 10000);
|
||||
return () => clearInterval(interval);
|
||||
}
|
||||
}, [isAuthenticated]);
|
||||
|
||||
return {
|
||||
containers,
|
||||
isRefreshing,
|
||||
isLoading,
|
||||
error,
|
||||
refreshContainers: fetchContainers,
|
||||
};
|
||||
}
|
||||
255
frontend/lib/hooks/useInteractiveTerminal.ts
Normal file
255
frontend/lib/hooks/useInteractiveTerminal.ts
Normal file
@@ -0,0 +1,255 @@
|
||||
import { useRef, useEffect } from 'react';
|
||||
import { io, Socket } from 'socket.io-client';
|
||||
import { apiClient, API_BASE_URL } from '@/lib/api';
|
||||
import type { Terminal } from '@xterm/xterm';
|
||||
import type { FitAddon } from '@xterm/addon-fit';
|
||||
|
||||
interface UseInteractiveTerminalProps {
|
||||
open: boolean;
|
||||
containerId: string;
|
||||
containerName: string;
|
||||
isMobile: boolean;
|
||||
onFallback: (reason: string) => void;
|
||||
}
|
||||
|
||||
export function useInteractiveTerminal({
|
||||
open,
|
||||
containerId,
|
||||
containerName,
|
||||
isMobile,
|
||||
onFallback,
|
||||
}: UseInteractiveTerminalProps) {
|
||||
const terminalRef = useRef<HTMLDivElement>(null);
|
||||
const xtermRef = useRef<Terminal | null>(null);
|
||||
const socketRef = useRef<Socket | null>(null);
|
||||
const fitAddonRef = useRef<FitAddon | null>(null);
|
||||
const connectionAttempts = useRef(0);
|
||||
|
||||
useEffect(() => {
|
||||
if (!open) return;
|
||||
|
||||
let term: Terminal | null = null;
|
||||
let fitAddon: FitAddon | null = null;
|
||||
let socket: Socket | null = null;
|
||||
let mounted = true;
|
||||
|
||||
const initTerminal = async () => {
|
||||
try {
|
||||
// Wait for ref to be available
|
||||
let attempts = 0;
|
||||
while (!terminalRef.current && attempts < 10) {
|
||||
await new Promise(resolve => setTimeout(resolve, 100));
|
||||
attempts++;
|
||||
}
|
||||
|
||||
if (!terminalRef.current || !mounted) {
|
||||
console.warn('Terminal ref not available after waiting');
|
||||
return;
|
||||
}
|
||||
|
||||
console.log('Initializing interactive terminal...');
|
||||
|
||||
const [{ Terminal }, { FitAddon }] = await Promise.all([
|
||||
import('@xterm/xterm'),
|
||||
import('@xterm/addon-fit'),
|
||||
]);
|
||||
|
||||
if (!terminalRef.current || !mounted) return;
|
||||
|
||||
console.log('Creating terminal instance...');
|
||||
term = new Terminal({
|
||||
cursorBlink: true,
|
||||
fontSize: isMobile ? 12 : 14,
|
||||
fontFamily: '"Ubuntu Mono", "DejaVu Sans Mono", "Courier New", monospace',
|
||||
theme: {
|
||||
// GNOME Terminal color scheme
|
||||
background: '#2E3436',
|
||||
foreground: '#D3D7CF',
|
||||
cursor: '#D3D7CF',
|
||||
cursorAccent: '#2E3436',
|
||||
selectionBackground: '#4A90D9',
|
||||
selectionForeground: '#FFFFFF',
|
||||
// Standard colors
|
||||
black: '#2E3436',
|
||||
red: '#CC0000',
|
||||
green: '#4E9A06',
|
||||
yellow: '#C4A000',
|
||||
blue: '#3465A4',
|
||||
magenta: '#75507B',
|
||||
cyan: '#06989A',
|
||||
white: '#D3D7CF',
|
||||
// Bright colors
|
||||
brightBlack: '#555753',
|
||||
brightRed: '#EF2929',
|
||||
brightGreen: '#8AE234',
|
||||
brightYellow: '#FCE94F',
|
||||
brightBlue: '#729FCF',
|
||||
brightMagenta: '#AD7FA8',
|
||||
brightCyan: '#34E2E2',
|
||||
brightWhite: '#EEEEEC',
|
||||
},
|
||||
});
|
||||
|
||||
console.log('Loading fit addon...');
|
||||
fitAddon = new FitAddon();
|
||||
term.loadAddon(fitAddon);
|
||||
|
||||
console.log('Opening terminal in DOM...');
|
||||
term.open(terminalRef.current);
|
||||
console.log('Terminal opened successfully');
|
||||
|
||||
setTimeout(() => {
|
||||
try {
|
||||
if (fitAddon) fitAddon.fit();
|
||||
} catch (e) {
|
||||
console.error('Error fitting terminal:', e);
|
||||
}
|
||||
}, 0);
|
||||
|
||||
xtermRef.current = term;
|
||||
fitAddonRef.current = fitAddon;
|
||||
|
||||
// Expose terminal for debugging
|
||||
if (typeof window !== 'undefined') {
|
||||
(window as any)._debugTerminal = term;
|
||||
}
|
||||
|
||||
const wsUrl = API_BASE_URL.replace(/^http/, 'ws');
|
||||
socket = io(`${wsUrl}/terminal`, {
|
||||
transports: ['polling', 'websocket'],
|
||||
reconnectionDelayMax: 10000,
|
||||
timeout: 60000,
|
||||
forceNew: true,
|
||||
});
|
||||
|
||||
socketRef.current = socket;
|
||||
|
||||
socket.on('connect', () => {
|
||||
console.log('WebSocket connected');
|
||||
connectionAttempts.current = 0;
|
||||
|
||||
const token = apiClient.getToken();
|
||||
const termSize = fitAddon?.proposeDimensions();
|
||||
socket?.emit('start_terminal', {
|
||||
container_id: containerId,
|
||||
token: token,
|
||||
cols: termSize?.cols || 80,
|
||||
rows: termSize?.rows || 24,
|
||||
});
|
||||
});
|
||||
|
||||
socket.on('connect_error', (error) => {
|
||||
console.error('WebSocket connection error:', error);
|
||||
connectionAttempts.current++;
|
||||
|
||||
if (connectionAttempts.current >= 2) {
|
||||
onFallback('Failed to establish WebSocket connection. Network or server may be unavailable.');
|
||||
}
|
||||
});
|
||||
|
||||
socket.on('started', () => {
|
||||
term?.write('\r\n*** Interactive Terminal Started ***\r\n');
|
||||
term?.write('You can now use sudo, nano, vim, and other interactive commands.\r\n\r\n');
|
||||
});
|
||||
|
||||
socket.on('output', (data: { data: string }) => {
|
||||
console.log('Received output event:', data);
|
||||
if (term && data && data.data) {
|
||||
term.write(data.data);
|
||||
}
|
||||
});
|
||||
|
||||
socket.on('error', (data: { error: string }) => {
|
||||
console.error('Terminal error:', data.error);
|
||||
term?.write(`\r\n\x1b[31mError: ${data.error}\x1b[0m\r\n`);
|
||||
|
||||
const criticalErrors = ['Unauthorized', 'Cannot connect to Docker', 'Invalid session'];
|
||||
if (criticalErrors.some(err => data.error.includes(err))) {
|
||||
setTimeout(() => {
|
||||
onFallback(`Interactive terminal failed: ${data.error}`);
|
||||
}, 2000);
|
||||
}
|
||||
});
|
||||
|
||||
socket.on('exit', () => {
|
||||
term?.write('\r\n\r\n*** Terminal Session Ended ***\r\n');
|
||||
});
|
||||
|
||||
socket.on('disconnect', (reason) => {
|
||||
console.log('WebSocket disconnected:', reason);
|
||||
|
||||
if (reason === 'transport error' || reason === 'transport close') {
|
||||
onFallback('WebSocket connection lost unexpectedly.');
|
||||
}
|
||||
});
|
||||
|
||||
term.onData((data) => {
|
||||
socket?.emit('input', { data });
|
||||
});
|
||||
|
||||
const handleResize = () => {
|
||||
try {
|
||||
if (fitAddon) {
|
||||
fitAddon.fit();
|
||||
const termSize = fitAddon.proposeDimensions();
|
||||
if (termSize) {
|
||||
socket?.emit('resize', {
|
||||
cols: termSize.cols,
|
||||
rows: termSize.rows,
|
||||
});
|
||||
}
|
||||
}
|
||||
} catch (e) {
|
||||
console.error('Error resizing terminal:', e);
|
||||
}
|
||||
};
|
||||
|
||||
window.addEventListener('resize', handleResize);
|
||||
|
||||
return () => {
|
||||
mounted = false;
|
||||
window.removeEventListener('resize', handleResize);
|
||||
if (term) {
|
||||
console.log('Disposing terminal...');
|
||||
term.dispose();
|
||||
}
|
||||
if (socket) {
|
||||
console.log('Disconnecting socket...');
|
||||
socket.disconnect();
|
||||
}
|
||||
};
|
||||
} catch (error) {
|
||||
console.error('Failed to initialize terminal:', error);
|
||||
if (mounted) {
|
||||
onFallback('Failed to load terminal. Switching to simple mode.');
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
const cleanup = initTerminal();
|
||||
|
||||
return () => {
|
||||
mounted = false;
|
||||
cleanup.then((cleanupFn) => {
|
||||
if (cleanupFn) cleanupFn();
|
||||
});
|
||||
xtermRef.current = null;
|
||||
socketRef.current = null;
|
||||
fitAddonRef.current = null;
|
||||
};
|
||||
}, [open, containerId, isMobile, onFallback]);
|
||||
|
||||
const cleanup = () => {
|
||||
if (socketRef.current) {
|
||||
socketRef.current.disconnect();
|
||||
}
|
||||
if (xtermRef.current) {
|
||||
xtermRef.current.dispose();
|
||||
}
|
||||
};
|
||||
|
||||
return {
|
||||
terminalRef,
|
||||
cleanup,
|
||||
};
|
||||
}
|
||||
38
frontend/lib/hooks/useLoginForm.ts
Normal file
38
frontend/lib/hooks/useLoginForm.ts
Normal file
@@ -0,0 +1,38 @@
|
||||
import { useState } from 'react';
|
||||
import { useRouter } from 'next/navigation';
|
||||
import { useAppDispatch, useAppSelector } from '@/lib/store/hooks';
|
||||
import { login, clearError } from '@/lib/store/authSlice';
|
||||
|
||||
export function useLoginForm() {
|
||||
const [username, setUsername] = useState('');
|
||||
const [password, setPassword] = useState('');
|
||||
const [isShaking, setIsShaking] = useState(false);
|
||||
const dispatch = useAppDispatch();
|
||||
const { error, loading } = useAppSelector((state) => state.auth);
|
||||
const router = useRouter();
|
||||
|
||||
const handleSubmit = async (e: React.FormEvent) => {
|
||||
e.preventDefault();
|
||||
dispatch(clearError());
|
||||
|
||||
const result = await dispatch(login({ username, password }));
|
||||
|
||||
if (login.fulfilled.match(result)) {
|
||||
router.push('/dashboard');
|
||||
} else {
|
||||
setIsShaking(true);
|
||||
setTimeout(() => setIsShaking(false), 500);
|
||||
}
|
||||
};
|
||||
|
||||
return {
|
||||
username,
|
||||
setUsername,
|
||||
password,
|
||||
setPassword,
|
||||
isShaking,
|
||||
error,
|
||||
loading,
|
||||
handleSubmit,
|
||||
};
|
||||
}
|
||||
74
frontend/lib/hooks/useSimpleTerminal.ts
Normal file
74
frontend/lib/hooks/useSimpleTerminal.ts
Normal file
@@ -0,0 +1,74 @@
|
||||
import { useState, useRef, useEffect } from 'react';
|
||||
import { apiClient } from '@/lib/api';
|
||||
import { OutputLine } from '@/lib/interfaces/terminal';
|
||||
|
||||
export function useSimpleTerminal(containerId: string) {
|
||||
const [command, setCommand] = useState('');
|
||||
const [output, setOutput] = useState<OutputLine[]>([]);
|
||||
const [isExecuting, setIsExecuting] = useState(false);
|
||||
const [workdir, setWorkdir] = useState('/');
|
||||
const outputRef = useRef<HTMLDivElement>(null);
|
||||
|
||||
// Auto-scroll to bottom when output changes
|
||||
useEffect(() => {
|
||||
if (outputRef.current) {
|
||||
outputRef.current.scrollTop = outputRef.current.scrollHeight;
|
||||
}
|
||||
}, [output]);
|
||||
|
||||
const executeCommand = async () => {
|
||||
if (!command.trim()) return;
|
||||
|
||||
setIsExecuting(true);
|
||||
setOutput((prev) => [...prev, {
|
||||
type: 'command',
|
||||
content: command,
|
||||
workdir: workdir
|
||||
}]);
|
||||
|
||||
try {
|
||||
const result = await apiClient.executeCommand(containerId, command);
|
||||
|
||||
if (result.workdir) {
|
||||
setWorkdir(result.workdir);
|
||||
}
|
||||
|
||||
if (result.output && result.output.trim()) {
|
||||
setOutput((prev) => [...prev, {
|
||||
type: result.exit_code === 0 ? 'output' : 'error',
|
||||
content: result.output
|
||||
}]);
|
||||
} else if (command.trim().startsWith('ls')) {
|
||||
setOutput((prev) => [...prev, {
|
||||
type: 'output',
|
||||
content: '(empty directory)'
|
||||
}]);
|
||||
}
|
||||
} catch (error) {
|
||||
setOutput((prev) => [...prev, {
|
||||
type: 'error',
|
||||
content: `Error: ${error instanceof Error ? error.message : 'Unknown error'}`
|
||||
}]);
|
||||
} finally {
|
||||
setIsExecuting(false);
|
||||
setCommand('');
|
||||
}
|
||||
};
|
||||
|
||||
const reset = () => {
|
||||
setOutput([]);
|
||||
setCommand('');
|
||||
setWorkdir('/');
|
||||
};
|
||||
|
||||
return {
|
||||
command,
|
||||
setCommand,
|
||||
output,
|
||||
isExecuting,
|
||||
workdir,
|
||||
outputRef,
|
||||
executeCommand,
|
||||
reset,
|
||||
};
|
||||
}
|
||||
24
frontend/lib/hooks/useTerminalModal.ts
Normal file
24
frontend/lib/hooks/useTerminalModal.ts
Normal file
@@ -0,0 +1,24 @@
|
||||
import { useState } from 'react';
|
||||
import { Container } from '@/lib/api';
|
||||
|
||||
export function useTerminalModal() {
|
||||
const [selectedContainer, setSelectedContainer] = useState<Container | null>(null);
|
||||
const [isTerminalOpen, setIsTerminalOpen] = useState(false);
|
||||
|
||||
const openTerminal = (container: Container) => {
|
||||
setSelectedContainer(container);
|
||||
setIsTerminalOpen(true);
|
||||
};
|
||||
|
||||
const closeTerminal = () => {
|
||||
setIsTerminalOpen(false);
|
||||
setTimeout(() => setSelectedContainer(null), 300);
|
||||
};
|
||||
|
||||
return {
|
||||
selectedContainer,
|
||||
isTerminalOpen,
|
||||
openTerminal,
|
||||
closeTerminal,
|
||||
};
|
||||
}
|
||||
35
frontend/lib/interfaces/container.ts
Normal file
35
frontend/lib/interfaces/container.ts
Normal file
@@ -0,0 +1,35 @@
|
||||
import { Container } from '@/lib/api';
|
||||
|
||||
export interface ContainerCardProps {
|
||||
container: Container;
|
||||
onOpenShell: () => void;
|
||||
onContainerUpdate?: () => void;
|
||||
}
|
||||
|
||||
export interface ContainerHeaderProps {
|
||||
name: string;
|
||||
image: string;
|
||||
status: string;
|
||||
}
|
||||
|
||||
export interface ContainerInfoProps {
|
||||
id: string;
|
||||
uptime: string;
|
||||
}
|
||||
|
||||
export interface ContainerActionsProps {
|
||||
status: string;
|
||||
isLoading: boolean;
|
||||
onStart: () => void;
|
||||
onStop: () => void;
|
||||
onRestart: () => void;
|
||||
onRemove: () => void;
|
||||
onOpenShell: () => void;
|
||||
}
|
||||
|
||||
export interface DeleteConfirmDialogProps {
|
||||
open: boolean;
|
||||
containerName: string;
|
||||
onClose: () => void;
|
||||
onConfirm: () => void;
|
||||
}
|
||||
7
frontend/lib/interfaces/dashboard.ts
Normal file
7
frontend/lib/interfaces/dashboard.ts
Normal file
@@ -0,0 +1,7 @@
|
||||
export interface DashboardHeaderProps {
|
||||
containerCount: number;
|
||||
isMobile: boolean;
|
||||
isRefreshing: boolean;
|
||||
onRefresh: () => void;
|
||||
onLogout: () => void;
|
||||
}
|
||||
61
frontend/lib/interfaces/terminal.ts
Normal file
61
frontend/lib/interfaces/terminal.ts
Normal file
@@ -0,0 +1,61 @@
|
||||
export interface OutputLine {
|
||||
type: 'command' | 'output' | 'error';
|
||||
content: string;
|
||||
workdir?: string;
|
||||
}
|
||||
|
||||
export interface TerminalModalProps {
|
||||
open: boolean;
|
||||
onClose: () => void;
|
||||
containerName: string;
|
||||
containerId: string;
|
||||
}
|
||||
|
||||
export interface TerminalHeaderProps {
|
||||
containerName: string;
|
||||
mode: 'simple' | 'interactive';
|
||||
interactiveFailed: boolean;
|
||||
onModeChange: (event: React.MouseEvent<HTMLElement>, newMode: 'simple' | 'interactive' | null) => void;
|
||||
onClose: () => void;
|
||||
}
|
||||
|
||||
export interface InteractiveTerminalProps {
|
||||
terminalRef: React.RefObject<HTMLDivElement | null>;
|
||||
}
|
||||
|
||||
export interface SimpleTerminalProps {
|
||||
output: OutputLine[];
|
||||
command: string;
|
||||
workdir: string;
|
||||
isExecuting: boolean;
|
||||
isMobile: boolean;
|
||||
containerName: string;
|
||||
outputRef: React.RefObject<HTMLDivElement | null>;
|
||||
onCommandChange: (value: string) => void;
|
||||
onExecute: () => void;
|
||||
onKeyPress: (e: React.KeyboardEvent) => void;
|
||||
}
|
||||
|
||||
export interface TerminalOutputProps {
|
||||
output: OutputLine[];
|
||||
containerName: string;
|
||||
outputRef: React.RefObject<HTMLDivElement | null>;
|
||||
}
|
||||
|
||||
export interface CommandInputProps {
|
||||
command: string;
|
||||
workdir: string;
|
||||
isExecuting: boolean;
|
||||
isMobile: boolean;
|
||||
containerName: string;
|
||||
onCommandChange: (value: string) => void;
|
||||
onExecute: () => void;
|
||||
onKeyPress: (e: React.KeyboardEvent) => void;
|
||||
}
|
||||
|
||||
export interface FallbackNotificationProps {
|
||||
show: boolean;
|
||||
reason: string;
|
||||
onClose: () => void;
|
||||
onRetry: () => void;
|
||||
}
|
||||
134
frontend/lib/store/__tests__/authSlice.test.ts
Normal file
134
frontend/lib/store/__tests__/authSlice.test.ts
Normal file
@@ -0,0 +1,134 @@
|
||||
import { configureStore } from '@reduxjs/toolkit';
|
||||
import authReducer, {
|
||||
login,
|
||||
logout,
|
||||
initAuth,
|
||||
setUnauthenticated,
|
||||
} from '../authSlice';
|
||||
import * as apiClient from '@/lib/api';
|
||||
|
||||
jest.mock('@/lib/api');
|
||||
|
||||
describe('authSlice', () => {
|
||||
let store: ReturnType<typeof configureStore>;
|
||||
|
||||
beforeEach(() => {
|
||||
store = configureStore({
|
||||
reducer: {
|
||||
auth: authReducer,
|
||||
},
|
||||
});
|
||||
jest.clearAllMocks();
|
||||
localStorage.clear();
|
||||
});
|
||||
|
||||
describe('initial state', () => {
|
||||
it('has correct initial state', () => {
|
||||
const state = store.getState().auth;
|
||||
expect(state).toEqual({
|
||||
isAuthenticated: false,
|
||||
loading: true,
|
||||
username: null,
|
||||
error: null,
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('setUnauthenticated', () => {
|
||||
it('sets auth state to unauthenticated', () => {
|
||||
store.dispatch(setUnauthenticated());
|
||||
const state = store.getState().auth;
|
||||
expect(state.isAuthenticated).toBe(false);
|
||||
expect(state.username).toBeNull();
|
||||
});
|
||||
});
|
||||
|
||||
describe('login async thunk', () => {
|
||||
it('handles successful login', async () => {
|
||||
const mockLoginResponse = { success: true, username: 'testuser' };
|
||||
(apiClient.apiClient.login as jest.Mock).mockResolvedValue(mockLoginResponse);
|
||||
|
||||
await store.dispatch(login({ username: 'testuser', password: 'password' }));
|
||||
|
||||
const state = store.getState().auth;
|
||||
expect(state.isAuthenticated).toBe(true);
|
||||
expect(state.username).toBe('testuser');
|
||||
expect(state.loading).toBe(false);
|
||||
});
|
||||
|
||||
it('handles login failure', async () => {
|
||||
(apiClient.apiClient.login as jest.Mock).mockRejectedValue(
|
||||
new Error('Invalid credentials')
|
||||
);
|
||||
|
||||
await store.dispatch(login({ username: 'testuser', password: 'wrong' }));
|
||||
|
||||
const state = store.getState().auth;
|
||||
expect(state.isAuthenticated).toBe(false);
|
||||
expect(state.username).toBeNull();
|
||||
expect(state.loading).toBe(false);
|
||||
expect(state.error).toBeTruthy();
|
||||
});
|
||||
|
||||
it('sets loading state during login', () => {
|
||||
(apiClient.apiClient.login as jest.Mock).mockImplementation(
|
||||
() => new Promise(() => {})
|
||||
);
|
||||
|
||||
store.dispatch(login({ username: 'testuser', password: 'password' }));
|
||||
|
||||
const state = store.getState().auth;
|
||||
expect(state.loading).toBe(true);
|
||||
});
|
||||
});
|
||||
|
||||
describe('logout async thunk', () => {
|
||||
it('clears authentication state on logout', async () => {
|
||||
(apiClient.apiClient.logout as jest.Mock).mockResolvedValue({});
|
||||
|
||||
await store.dispatch(logout());
|
||||
|
||||
const state = store.getState().auth;
|
||||
expect(state.isAuthenticated).toBe(false);
|
||||
expect(state.username).toBeNull();
|
||||
});
|
||||
});
|
||||
|
||||
describe('initAuth async thunk', () => {
|
||||
it('restores auth state when token is valid', async () => {
|
||||
(apiClient.apiClient.getToken as jest.Mock).mockReturnValue('valid-token');
|
||||
(apiClient.apiClient.getUsername as jest.Mock).mockReturnValue('testuser');
|
||||
(apiClient.apiClient.getContainers as jest.Mock).mockResolvedValue([]);
|
||||
|
||||
await store.dispatch(initAuth());
|
||||
|
||||
const state = store.getState().auth;
|
||||
expect(state.isAuthenticated).toBe(true);
|
||||
expect(state.username).toBe('testuser');
|
||||
});
|
||||
|
||||
it('clears invalid token', async () => {
|
||||
(apiClient.apiClient.getToken as jest.Mock).mockReturnValue('invalid-token');
|
||||
(apiClient.apiClient.getContainers as jest.Mock).mockRejectedValue(
|
||||
new Error('Unauthorized')
|
||||
);
|
||||
|
||||
await store.dispatch(initAuth());
|
||||
|
||||
const state = store.getState().auth;
|
||||
expect(state.isAuthenticated).toBe(false);
|
||||
expect(state.username).toBeNull();
|
||||
expect(apiClient.apiClient.setToken).toHaveBeenCalledWith(null);
|
||||
});
|
||||
|
||||
it('handles no token present', async () => {
|
||||
(apiClient.apiClient.getToken as jest.Mock).mockReturnValue(null);
|
||||
|
||||
await store.dispatch(initAuth());
|
||||
|
||||
const state = store.getState().auth;
|
||||
expect(state.isAuthenticated).toBe(false);
|
||||
expect(state.username).toBeNull();
|
||||
});
|
||||
});
|
||||
});
|
||||
22
frontend/lib/store/authErrorHandler.ts
Normal file
22
frontend/lib/store/authErrorHandler.ts
Normal file
@@ -0,0 +1,22 @@
|
||||
// Global auth error handler
|
||||
// This can be called from API client when auth errors occur
|
||||
let authErrorCallback: (() => void) | null = null;
|
||||
let authErrorHandled = false;
|
||||
|
||||
export const setAuthErrorCallback = (callback: () => void) => {
|
||||
authErrorCallback = callback;
|
||||
authErrorHandled = false;
|
||||
};
|
||||
|
||||
export const triggerAuthError = () => {
|
||||
if (!authErrorCallback) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (authErrorHandled) {
|
||||
return;
|
||||
}
|
||||
|
||||
authErrorHandled = true;
|
||||
authErrorCallback();
|
||||
};
|
||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user