mirror of
https://github.com/Dvorinka/Containr.git
synced 2026-10-28 15:51:26 +00:00
Compare commits
11
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
549790c611 | ||
|
|
6bcbc1369e | ||
|
|
81a4ca28d8 | ||
|
|
269e031761 | ||
|
|
ce71b5af24 | ||
|
|
cca20f7eae | ||
|
|
202fdd7b5a | ||
|
|
687555fc78 | ||
|
|
302c595380 | ||
|
|
ca8eab4a65 | ||
|
|
a6b8397dd8 |
No files matched your search
+13
-5
@@ -252,9 +252,12 @@ goroutines; this fixes correctness *and* unlocks the ops surface.
|
|||||||
- [x] **Graph deploy pipeline** — creates services in dependency order,
|
- [x] **Graph deploy pipeline** — creates services in dependency order,
|
||||||
resolves vars/secrets, triggers builds. `deployTemplateGraph`,
|
resolves vars/secrets, triggers builds. `deployTemplateGraph`,
|
||||||
`POST /templates/deploy` (inline), `POST /projects/:id/import-compose`.
|
`POST /templates/deploy` (inline), `POST /projects/:id/import-compose`.
|
||||||
- [ ] **Visual template builder** — canvas mode: drop git/image/db
|
- [x] **Visual template builder** — canvas mode in the template editor:
|
||||||
nodes, wire refs, set order, save/publish. Reuses existing canvas
|
web/worker/cron/database nodes, handles wire `depends_on`, inspector
|
||||||
primitives (dflow's `templates/compose` is exactly this).
|
edits image/repo/engine/port/env/volumes, live validation, applies
|
||||||
|
back to the version-2 graph JSON. Positions persist under
|
||||||
|
`config.canvas`; flat legacy configs open as a single node.
|
||||||
|
`builder-model.ts` + `TemplateBuilder.tsx`.
|
||||||
- [x] **Catalog growth** — seeded graph stacks for n8n, Gitea,
|
- [x] **Catalog growth** — seeded graph stacks for n8n, Gitea,
|
||||||
Nextcloud, Vaultwarden, Umami, Immich (Plausible already shipped).
|
Nextcloud, Vaultwarden, Umami, Immich (Plausible already shipped).
|
||||||
`TestSeedTemplatesResolveCleanly` keeps every seed parseable,
|
`TestSeedTemplatesResolveCleanly` keeps every seed parseable,
|
||||||
@@ -534,7 +537,7 @@ and wired refs, one click or one CLI call.
|
|||||||
workspace, CLI `environments` + MCP parity. Per-env vars/domains ride the
|
workspace, CLI `environments` + MCP parity. Per-env vars/domains ride the
|
||||||
service's own scope (one env per service); promote ships via preview
|
service's own scope (one env per service); promote ships via preview
|
||||||
promote.
|
promote.
|
||||||
- [ ] **sqlc port** — ~140 raw `database/sql` sites → `sqlc/queries/`.
|
- [ ] **sqlc port** — ~265 raw `database/sql` sites → `sqlc/queries/`. Done in tranches (dynamic queries stay raw). Tranches merged: metrics storage + service domains (#75), service clone/move ops (#76), preview environments (#77), deployments + rollback (#78). Remaining: runtime, compliance, cron, git, security, auth, sleeper, scanner, gateway, and smaller files.
|
||||||
- [ ] **Tests** — canvas/wizard frontend coverage, deployment/template/
|
- [ ] **Tests** — canvas/wizard frontend coverage, deployment/template/
|
||||||
webhook table tests, Playwright e2e golden path.
|
webhook table tests, Playwright e2e golden path.
|
||||||
- [x] **First-run wizard** — public `GET /setup/status` + authed
|
- [x] **First-run wizard** — public `GET /setup/status` + authed
|
||||||
@@ -543,7 +546,12 @@ and wired refs, one click or one CLI call.
|
|||||||
optional Cloudflare tunnel token → done. Shell redirects signed-in
|
optional Cloudflare tunnel token → done. Shell redirects signed-in
|
||||||
users to `/setup` while `needs_setup`; installs with existing projects
|
users to `/setup` while `needs_setup`; installs with existing projects
|
||||||
auto-complete so nobody gets nagged retroactively.
|
auto-complete so nobody gets nagged retroactively.
|
||||||
- [ ] **Self-upgrade loop verification**, docs freshness, v1.0.0 release.
|
- [x] **Self-upgrade loop verification** — live-verified: admin
|
||||||
|
`POST /agents/:id/upgrade` → `self_upgrade` command → agent downloads
|
||||||
|
`/api/agents/download/linux-amd64` → sha256 verify → `--version`
|
||||||
|
sanity → atomic rename → result posts → `syscall.Exec` re-exec. Agent
|
||||||
|
ran `v0.0.1-old`, came back `v9.9.9-test`, command row `completed`.
|
||||||
|
- [ ] **Docs freshness, v1.0.0 release.**
|
||||||
|
|
||||||
## 11. Deliberately not imported from dflow
|
## 11. Deliberately not imported from dflow
|
||||||
|
|
||||||
|
|||||||
@@ -3,6 +3,8 @@ package api
|
|||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"containr/internal/database"
|
"containr/internal/database"
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
|
"context"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
@@ -63,11 +65,16 @@ func handleLogin(c *gin.Context) {
|
|||||||
// Find user by email
|
// Find user by email
|
||||||
var user User
|
var user User
|
||||||
var hashedPassword string
|
var hashedPassword string
|
||||||
err := db.QueryRow(`
|
urow, err := sqlcdb.New(db.DB).GetUserByEmailForAuth(context.Background(), req.Email)
|
||||||
SELECT id, email, password_hash, name, COALESCE(avatar_url, ''), is_admin, created_at
|
user = User{
|
||||||
FROM users
|
ID: urow.ID.String(),
|
||||||
WHERE email = $1
|
Email: urow.Email,
|
||||||
`, req.Email).Scan(&user.ID, &user.Email, &hashedPassword, &user.Name, &user.AvatarURL, &user.IsAdmin, &user.CreatedAt)
|
Name: urow.Name,
|
||||||
|
AvatarURL: urow.AvatarUrl,
|
||||||
|
IsAdmin: urow.IsAdmin,
|
||||||
|
CreatedAt: urow.CreatedAt.Time.String(),
|
||||||
|
}
|
||||||
|
hashedPassword = urow.PasswordHash
|
||||||
|
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
authUser, ok := verifyBetterAuthCredentials(c, req.Email, req.Password)
|
authUser, ok := verifyBetterAuthCredentials(c, req.Email, req.Password)
|
||||||
@@ -257,25 +264,30 @@ func createLocalUserFromBetterAuth(db *database.DB, authUser betterAuthLoginUser
|
|||||||
}
|
}
|
||||||
|
|
||||||
// A mirrored first account owns the platform, same as handleRegister.
|
// A mirrored first account owns the platform, same as handleRegister.
|
||||||
var total int
|
q := sqlcdb.New(db.DB)
|
||||||
if err := db.QueryRow("SELECT COUNT(*) FROM users").Scan(&total); err != nil {
|
total64, err := q.CountUsers(context.Background())
|
||||||
|
if err != nil {
|
||||||
return User{}, err
|
return User{}, err
|
||||||
}
|
}
|
||||||
|
|
||||||
var user User
|
urow, err := q.UpsertLocalUser(context.Background(), sqlcdb.UpsertLocalUserParams{
|
||||||
err = db.QueryRow(`
|
Email: email,
|
||||||
INSERT INTO users (email, password_hash, name, avatar_url, is_admin)
|
PasswordHash: string(hashedPassword),
|
||||||
VALUES ($1, $2, $3, NULLIF($4, ''), $5)
|
Name: name,
|
||||||
ON CONFLICT (email) DO UPDATE
|
Column4: strings.TrimSpace(authUser.Image),
|
||||||
SET name = EXCLUDED.name,
|
IsAdmin: total64 == 0,
|
||||||
avatar_url = COALESCE(EXCLUDED.avatar_url, users.avatar_url),
|
})
|
||||||
updated_at = NOW()
|
|
||||||
RETURNING id, email, name, COALESCE(avatar_url, ''), is_admin, created_at
|
|
||||||
`, email, string(hashedPassword), name, strings.TrimSpace(authUser.Image), total == 0).
|
|
||||||
Scan(&user.ID, &user.Email, &user.Name, &user.AvatarURL, &user.IsAdmin, &user.CreatedAt)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return User{}, err
|
return User{}, err
|
||||||
}
|
}
|
||||||
|
user := User{
|
||||||
|
ID: urow.ID.String(),
|
||||||
|
Email: urow.Email,
|
||||||
|
Name: urow.Name,
|
||||||
|
AvatarURL: urow.AvatarUrl,
|
||||||
|
IsAdmin: urow.IsAdmin,
|
||||||
|
CreatedAt: urow.CreatedAt.Time.String(),
|
||||||
|
}
|
||||||
|
|
||||||
return user, nil
|
return user, nil
|
||||||
}
|
}
|
||||||
@@ -302,8 +314,8 @@ func handleRegister(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var existing int
|
q := sqlcdb.New(db.DB)
|
||||||
err = db.QueryRow("SELECT COUNT(*) FROM users WHERE email = $1", strings.ToLower(strings.TrimSpace(req.Email))).Scan(&existing)
|
existing, err := q.CountUsersByEmail(context.Background(), strings.ToLower(strings.TrimSpace(req.Email)))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
||||||
return
|
return
|
||||||
@@ -322,12 +334,20 @@ func handleRegister(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// First account owns the platform.
|
// First account owns the platform.
|
||||||
var user User
|
urow, err := q.InsertUserAdmin(context.Background(), sqlcdb.InsertUserAdminParams{
|
||||||
err = db.QueryRow(`
|
Email: req.Email,
|
||||||
INSERT INTO users (email, password_hash, name, is_admin)
|
PasswordHash: string(hashedPassword),
|
||||||
VALUES ($1, $2, $3, $4)
|
Name: req.Name,
|
||||||
RETURNING id, email, name, COALESCE(avatar_url, ''), is_admin, created_at
|
IsAdmin: total == 0,
|
||||||
`, req.Email, string(hashedPassword), req.Name, total == 0).Scan(&user.ID, &user.Email, &user.Name, &user.AvatarURL, &user.IsAdmin, &user.CreatedAt)
|
})
|
||||||
|
user := User{
|
||||||
|
ID: urow.ID.String(),
|
||||||
|
Email: urow.Email,
|
||||||
|
Name: urow.Name,
|
||||||
|
AvatarURL: urow.AvatarUrl,
|
||||||
|
IsAdmin: urow.IsAdmin,
|
||||||
|
CreatedAt: urow.CreatedAt.Time.String(),
|
||||||
|
}
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create user"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create user"})
|
||||||
@@ -365,8 +385,8 @@ func handleCreateUser(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var existing int
|
existing, err := sqlcdb.New(db.DB).CountUsersByEmail(context.Background(), req.Email)
|
||||||
if err := db.QueryRow("SELECT COUNT(*) FROM users WHERE email = $1", req.Email).Scan(&existing); err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -390,12 +410,19 @@ func handleCreateUser(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var user User
|
urow, err := sqlcdb.New(db.DB).InsertUser(context.Background(), sqlcdb.InsertUserParams{
|
||||||
err = db.QueryRow(`
|
Email: req.Email,
|
||||||
INSERT INTO users (email, password_hash, name)
|
PasswordHash: string(hashedPassword),
|
||||||
VALUES ($1, $2, $3)
|
Name: req.Name,
|
||||||
RETURNING id, email, name, COALESCE(avatar_url, ''), is_admin, created_at
|
})
|
||||||
`, req.Email, string(hashedPassword), req.Name).Scan(&user.ID, &user.Email, &user.Name, &user.AvatarURL, &user.IsAdmin, &user.CreatedAt)
|
user := User{
|
||||||
|
ID: urow.ID.String(),
|
||||||
|
Email: urow.Email,
|
||||||
|
Name: urow.Name,
|
||||||
|
AvatarURL: urow.AvatarUrl,
|
||||||
|
IsAdmin: urow.IsAdmin,
|
||||||
|
CreatedAt: urow.CreatedAt.Time.String(),
|
||||||
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusConflict, gin.H{"error": "User already exists"})
|
c.JSON(http.StatusConflict, gin.H{"error": "User already exists"})
|
||||||
return
|
return
|
||||||
@@ -458,10 +485,11 @@ func createBetterAuthUser(c *gin.Context, name string, email string, password st
|
|||||||
}
|
}
|
||||||
|
|
||||||
func countLocalUsers(db *database.DB) (int, error) {
|
func countLocalUsers(db *database.DB) (int, error) {
|
||||||
var count int
|
count64, err := sqlcdb.New(db.DB).CountUsers(context.Background())
|
||||||
if err := db.QueryRow("SELECT COUNT(*) FROM users").Scan(&count); err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
|
count := int(count64)
|
||||||
// Better Auth writes to auth_users; a row there only mirrors into users on
|
// Better Auth writes to auth_users; a row there only mirrors into users on
|
||||||
// the first authenticated call. Count it too or a second public sign-up
|
// the first authenticated call. Count it too or a second public sign-up
|
||||||
// slips through the bootstrap window. UNION by email avoids double-counting
|
// slips through the bootstrap window. UNION by email avoids double-counting
|
||||||
@@ -489,11 +517,15 @@ func handleGetProfile(c *gin.Context) {
|
|||||||
db := c.MustGet("db").(*database.DB)
|
db := c.MustGet("db").(*database.DB)
|
||||||
|
|
||||||
var user User
|
var user User
|
||||||
err := db.QueryRow(`
|
urow, err := sqlcdb.New(db.DB).GetUserByID(context.Background(), uuid.MustParse(userID))
|
||||||
SELECT id, email, name, COALESCE(avatar_url, ''), is_admin, created_at
|
user = User{
|
||||||
FROM users
|
ID: urow.ID.String(),
|
||||||
WHERE id = $1
|
Email: urow.Email,
|
||||||
`, userID).Scan(&user.ID, &user.Email, &user.Name, &user.AvatarURL, &user.IsAdmin, &user.CreatedAt)
|
Name: urow.Name,
|
||||||
|
AvatarURL: urow.AvatarUrl,
|
||||||
|
IsAdmin: urow.IsAdmin,
|
||||||
|
CreatedAt: urow.CreatedAt.Time.String(),
|
||||||
|
}
|
||||||
|
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
|
||||||
@@ -521,11 +553,11 @@ func handleUpdateProfile(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Update user profile
|
// Update user profile
|
||||||
_, err := db.Exec(`
|
err := sqlcdb.New(db.DB).UpdateUserProfile(context.Background(), sqlcdb.UpdateUserProfileParams{
|
||||||
UPDATE users
|
Column1: req.Name,
|
||||||
SET name = COALESCE($1, name), avatar_url = COALESCE($2, avatar_url)
|
Column2: req.AvatarURL,
|
||||||
WHERE id = $3
|
ID: uuid.MustParse(userID),
|
||||||
`, req.Name, req.AvatarURL, userID)
|
})
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update profile"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update profile"})
|
||||||
|
|||||||
+143
-119
@@ -65,13 +65,47 @@ type UpdateCronJobRequest struct {
|
|||||||
Retention int `json:"retention"`
|
Retention int `json:"retention"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// cronJobFrom maps the sqlc row onto the string-ID API model.
|
||||||
|
func cronJobFrom(j sqlcdb.CronJob) CronJob {
|
||||||
|
return CronJob{
|
||||||
|
ID: j.ID.String(),
|
||||||
|
ProjectID: j.ProjectID.String(),
|
||||||
|
ServiceID: j.ServiceID.String(),
|
||||||
|
Name: j.Name,
|
||||||
|
Schedule: j.Schedule,
|
||||||
|
Command: j.Command,
|
||||||
|
Timezone: j.Timezone.String,
|
||||||
|
Enabled: j.Enabled.Bool,
|
||||||
|
LastRunAt: timePtrNT(j.LastRunAt),
|
||||||
|
NextRunAt: timePtrNT(j.NextRunAt),
|
||||||
|
LastStatus: j.LastStatus.String,
|
||||||
|
LastOutput: j.LastOutput.String,
|
||||||
|
Retention: int(j.Retention.Int32),
|
||||||
|
CreatedAt: j.CreatedAt.Time,
|
||||||
|
UpdatedAt: j.UpdatedAt.Time,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// cronExecutionFrom maps the sqlc row onto the string-ID API model.
|
||||||
|
func cronExecutionFrom(e sqlcdb.CronExecution) CronExecution {
|
||||||
|
return CronExecution{
|
||||||
|
ID: e.ID.String(),
|
||||||
|
CronJobID: e.CronJobID.String(),
|
||||||
|
StartedAt: e.StartedAt,
|
||||||
|
FinishedAt: timePtrNT(e.FinishedAt),
|
||||||
|
Status: e.Status.String,
|
||||||
|
Output: e.Output.String,
|
||||||
|
Error: e.Error.String,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// cronJobProjectID resolves the project a cron job belongs to.
|
// cronJobProjectID resolves the project a cron job belongs to.
|
||||||
func cronJobProjectID(db *database.DB, jobID string) (uuid.UUID, bool) {
|
func cronJobProjectID(db *database.DB, jobID string) (uuid.UUID, bool) {
|
||||||
var projectID uuid.UUID
|
id, err := uuid.Parse(jobID)
|
||||||
err := db.QueryRow(
|
if err != nil {
|
||||||
`SELECT cj.project_id FROM cron_jobs cj WHERE cj.id = $1`,
|
return uuid.Nil, false
|
||||||
jobID,
|
}
|
||||||
).Scan(&projectID)
|
projectID, err := sqlcdb.New(db.DB).GetCronJobProjectID(context.Background(), id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return uuid.Nil, false
|
return uuid.Nil, false
|
||||||
}
|
}
|
||||||
@@ -137,15 +171,14 @@ func handleCreateCronJob(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var ownerCheck string
|
svcID, err := uuid.Parse(req.ServiceID)
|
||||||
err := db.QueryRow(
|
if err != nil {
|
||||||
`SELECT p.owner_id FROM projects p
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
JOIN services s ON s.project_id = p.id
|
return
|
||||||
WHERE s.id = $1`,
|
}
|
||||||
req.ServiceID,
|
ownerCheck, err := sqlcdb.New(db.DB).GetCronOwnerViaService(context.Background(), svcID)
|
||||||
).Scan(&ownerCheck)
|
|
||||||
|
|
||||||
if err != nil || (ownerCheck != userID && !contextIsAdmin(c)) {
|
if err != nil || (ownerCheck.String() != userID && !contextIsAdmin(c)) {
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -178,11 +211,20 @@ func handleCreateCronJob(c *gin.Context) {
|
|||||||
UpdatedAt: time.Now(),
|
UpdatedAt: time.Now(),
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = db.Exec(
|
err = sqlcdb.New(db.DB).InsertCronJob(context.Background(), sqlcdb.InsertCronJobParams{
|
||||||
`INSERT INTO cron_jobs (id, project_id, service_id, name, schedule, command, timezone, enabled, next_run_at, retention, created_at, updated_at)
|
ID: uuid.MustParse(job.ID),
|
||||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)`,
|
ProjectID: uuid.MustParse(job.ProjectID),
|
||||||
job.ID, job.ProjectID, job.ServiceID, job.Name, job.Schedule, job.Command, job.Timezone, job.Enabled, job.NextRunAt, job.Retention, job.CreatedAt, job.UpdatedAt,
|
ServiceID: uuid.MustParse(job.ServiceID),
|
||||||
)
|
Name: job.Name,
|
||||||
|
Schedule: job.Schedule,
|
||||||
|
Command: job.Command,
|
||||||
|
Timezone: sql.NullString{String: job.Timezone, Valid: true},
|
||||||
|
Enabled: sql.NullBool{Bool: job.Enabled, Valid: true},
|
||||||
|
NextRunAt: ntPtr(job.NextRunAt),
|
||||||
|
Retention: sql.NullInt32{Int32: int32(job.Retention), Valid: true},
|
||||||
|
CreatedAt: sql.NullTime{Time: job.CreatedAt, Valid: true},
|
||||||
|
UpdatedAt: sql.NullTime{Time: job.UpdatedAt, Valid: true},
|
||||||
|
})
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create cron job"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create cron job"})
|
||||||
@@ -201,22 +243,17 @@ func handleGetCronJob(c *gin.Context) {
|
|||||||
db := c.MustGet("db").(*database.DB)
|
db := c.MustGet("db").(*database.DB)
|
||||||
jobID := c.Param("id")
|
jobID := c.Param("id")
|
||||||
|
|
||||||
var job CronJob
|
jid, err := uuid.Parse(jobID)
|
||||||
err := db.QueryRow(
|
|
||||||
`SELECT cj.id, cj.project_id, cj.service_id, cj.name, cj.schedule, cj.timezone,
|
|
||||||
cj.enabled, cj.last_run_at, cj.next_run_at, cj.last_status, cj.last_output,
|
|
||||||
cj.retention, cj.created_at, cj.updated_at
|
|
||||||
FROM cron_jobs cj
|
|
||||||
WHERE cj.id = $1`,
|
|
||||||
jobID,
|
|
||||||
).Scan(&job.ID, &job.ProjectID, &job.ServiceID, &job.Name, &job.Schedule, &job.Timezone,
|
|
||||||
&job.Enabled, &job.LastRunAt, &job.NextRunAt, &job.LastStatus, &job.LastOutput,
|
|
||||||
&job.Retention, &job.CreatedAt, &job.UpdatedAt)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Cron job not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Cron job not found"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
row, err := sqlcdb.New(db.DB).GetCronJob(context.Background(), jid)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusNotFound, gin.H{"error": "Cron job not found"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
job := cronJobFrom(row)
|
||||||
|
|
||||||
projectID, found := cronJobProjectID(db, jobID)
|
projectID, found := cronJobProjectID(db, jobID)
|
||||||
if !found {
|
if !found {
|
||||||
@@ -242,15 +279,14 @@ func handleUpdateCronJob(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var ownerCheck string
|
jid, err := uuid.Parse(jobID)
|
||||||
err := db.QueryRow(
|
if err != nil {
|
||||||
`SELECT p.owner_id FROM cron_jobs cj
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
JOIN projects p ON cj.project_id = p.id
|
return
|
||||||
WHERE cj.id = $1`,
|
}
|
||||||
jobID,
|
ownerCheck, err := sqlcdb.New(db.DB).GetCronJobOwner(context.Background(), jid)
|
||||||
).Scan(&ownerCheck)
|
|
||||||
|
|
||||||
if err != nil || (ownerCheck != userID && !contextIsAdmin(c)) {
|
if err != nil || (ownerCheck.String() != userID && !contextIsAdmin(c)) {
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -312,21 +348,19 @@ func handleDeleteCronJob(c *gin.Context) {
|
|||||||
db := c.MustGet("db").(*database.DB)
|
db := c.MustGet("db").(*database.DB)
|
||||||
jobID := c.Param("id")
|
jobID := c.Param("id")
|
||||||
|
|
||||||
var ownerCheck string
|
jid, err := uuid.Parse(jobID)
|
||||||
err := db.QueryRow(
|
if err != nil {
|
||||||
`SELECT p.owner_id FROM cron_jobs cj
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
JOIN projects p ON cj.project_id = p.id
|
return
|
||||||
WHERE cj.id = $1`,
|
}
|
||||||
jobID,
|
ownerCheck, err := sqlcdb.New(db.DB).GetCronJobOwner(context.Background(), jid)
|
||||||
).Scan(&ownerCheck)
|
|
||||||
|
|
||||||
if err != nil || (ownerCheck != userID && !contextIsAdmin(c)) {
|
if err != nil || (ownerCheck.String() != userID && !contextIsAdmin(c)) {
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = db.Exec("DELETE FROM cron_jobs WHERE id = $1", jobID)
|
if err := sqlcdb.New(db.DB).DeleteCronJob(context.Background(), jid); err != nil {
|
||||||
if err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to delete cron job"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to delete cron job"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -350,29 +384,20 @@ func handleGetCronExecutions(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
rows, err := db.Query(
|
jid, err := uuid.Parse(jobID)
|
||||||
`SELECT id, cron_job_id, started_at, finished_at, status, output, error
|
if err != nil {
|
||||||
FROM cron_executions
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to fetch executions"})
|
||||||
WHERE cron_job_id = $1
|
return
|
||||||
ORDER BY started_at DESC
|
}
|
||||||
LIMIT 100`,
|
rows, err := sqlcdb.New(db.DB).ListCronExecutions(context.Background(), jid)
|
||||||
jobID,
|
|
||||||
)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to fetch executions"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to fetch executions"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var executions []CronExecution
|
var executions []CronExecution
|
||||||
for rows.Next() {
|
for _, e := range rows {
|
||||||
var exec CronExecution
|
executions = append(executions, cronExecutionFrom(e))
|
||||||
err := rows.Scan(&exec.ID, &exec.CronJobID, &exec.StartedAt, &exec.FinishedAt, &exec.Status, &exec.Output, &exec.Error)
|
|
||||||
if err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
executions = append(executions, exec)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
c.JSON(http.StatusOK, gin.H{"executions": executions})
|
c.JSON(http.StatusOK, gin.H{"executions": executions})
|
||||||
@@ -383,29 +408,33 @@ func handleTriggerCronJob(c *gin.Context) {
|
|||||||
db := c.MustGet("db").(*database.DB)
|
db := c.MustGet("db").(*database.DB)
|
||||||
jobID := c.Param("id")
|
jobID := c.Param("id")
|
||||||
|
|
||||||
var job CronJob
|
jid, err := uuid.Parse(jobID)
|
||||||
var ownerCheck string
|
if err != nil {
|
||||||
err := db.QueryRow(
|
|
||||||
`SELECT cj.service_id, cj.command, cj.schedule, cj.timezone, cj.retention, p.owner_id
|
|
||||||
FROM cron_jobs cj
|
|
||||||
JOIN projects p ON cj.project_id = p.id
|
|
||||||
WHERE cj.id = $1`,
|
|
||||||
jobID,
|
|
||||||
).Scan(&job.ServiceID, &job.Command, &job.Schedule, &job.Timezone, &job.Retention, &ownerCheck)
|
|
||||||
|
|
||||||
if err != nil || (ownerCheck != userID && !contextIsAdmin(c)) {
|
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
trig, err := sqlcdb.New(db.DB).GetCronJobForTrigger(context.Background(), jid)
|
||||||
|
if err != nil || (trig.OwnerID.String() != userID && !contextIsAdmin(c)) {
|
||||||
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
|
return
|
||||||
|
}
|
||||||
|
job := CronJob{
|
||||||
|
ServiceID: trig.ServiceID.String(),
|
||||||
|
Command: trig.Command,
|
||||||
|
Schedule: trig.Schedule,
|
||||||
|
Timezone: trig.Timezone.String,
|
||||||
|
Retention: int(trig.Retention.Int32),
|
||||||
|
}
|
||||||
|
|
||||||
execID := uuid.New().String()
|
execID := uuid.New().String()
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
|
|
||||||
_, err = db.Exec(
|
_ = sqlcdb.New(db.DB).InsertCronExecution(context.Background(), sqlcdb.InsertCronExecutionParams{
|
||||||
`INSERT INTO cron_executions (id, cron_job_id, started_at, status)
|
ID: uuid.MustParse(execID),
|
||||||
VALUES ($1, $2, $3, $4)`,
|
CronJobID: jid,
|
||||||
execID, jobID, now, "running",
|
StartedAt: now,
|
||||||
)
|
Status: sql.NullString{String: "running", Valid: true},
|
||||||
|
})
|
||||||
|
|
||||||
dockerClient, _ := c.Get("docker_client")
|
dockerClient, _ := c.Get("docker_client")
|
||||||
dockerCli, _ := dockerClient.(*docker.Client)
|
dockerCli, _ := dockerClient.(*docker.Client)
|
||||||
@@ -458,39 +487,25 @@ func StartCronScheduler(ctx context.Context, db *database.DB, dockerClient *dock
|
|||||||
}
|
}
|
||||||
|
|
||||||
func runDueCronJobs(db *database.DB, dockerClient *docker.Client) {
|
func runDueCronJobs(db *database.DB, dockerClient *docker.Client) {
|
||||||
rows, err := db.Query(
|
q := sqlcdb.New(db.DB)
|
||||||
`SELECT id, project_id, service_id, name, schedule, command, timezone, enabled, retention
|
jobs, err := q.ListDueCronJobs(context.Background())
|
||||||
FROM cron_jobs
|
|
||||||
WHERE enabled = TRUE AND next_run_at IS NOT NULL AND next_run_at <= NOW()`)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("cron scheduler: failed to list due jobs: %v", err)
|
log.Printf("cron scheduler: failed to list due jobs: %v", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
type dueJob struct {
|
|
||||||
CronJob
|
|
||||||
Command string
|
|
||||||
}
|
|
||||||
var jobs []dueJob
|
|
||||||
for rows.Next() {
|
|
||||||
var j dueJob
|
|
||||||
if err := rows.Scan(&j.ID, &j.ProjectID, &j.ServiceID, &j.Name, &j.Schedule,
|
|
||||||
&j.Command, &j.Timezone, &j.Enabled, &j.Retention); err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
jobs = append(jobs, j)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, j := range jobs {
|
for _, j := range jobs {
|
||||||
execID := uuid.New().String()
|
execID := uuid.New().String()
|
||||||
if _, err := db.Exec(
|
if err := q.InsertCronExecution(context.Background(), sqlcdb.InsertCronExecutionParams{
|
||||||
`INSERT INTO cron_executions (id, cron_job_id, started_at, status) VALUES ($1, $2, $3, $4)`,
|
ID: uuid.MustParse(execID),
|
||||||
execID, j.ID, time.Now(), "running"); err != nil {
|
CronJobID: j.ID,
|
||||||
|
StartedAt: time.Now(),
|
||||||
|
Status: sql.NullString{String: "running", Valid: true},
|
||||||
|
}); err != nil {
|
||||||
log.Printf("cron scheduler: failed to record execution for %s: %v", j.ID, err)
|
log.Printf("cron scheduler: failed to record execution for %s: %v", j.ID, err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
go executeCronJob(db, dockerClient, j.ID, j.ServiceID, execID, j.Command, j.Schedule, j.Timezone, j.Retention)
|
go executeCronJob(db, dockerClient, j.ID.String(), j.ServiceID.String(), execID, j.Command, j.Schedule, j.Timezone.String, int(j.Retention.Int32))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -584,23 +599,32 @@ func executeCronJob(db *database.DB, dockerClient *docker.Client, jobID, service
|
|||||||
}
|
}
|
||||||
|
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
db.Exec(
|
q := sqlcdb.New(db.DB)
|
||||||
`UPDATE cron_executions SET finished_at = $1, status = $2, output = $3, error = $4 WHERE id = $5`,
|
if eid, err := uuid.Parse(execID); err == nil {
|
||||||
now, status, output, errText, execID,
|
_ = q.FinishCronExecution(context.Background(), sqlcdb.FinishCronExecutionParams{
|
||||||
)
|
FinishedAt: sql.NullTime{Time: now, Valid: true},
|
||||||
|
Status: sql.NullString{String: status, Valid: true},
|
||||||
|
Output: sql.NullString{String: output, Valid: true},
|
||||||
|
Error: sql.NullString{String: errText, Valid: true},
|
||||||
|
ID: eid,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
next, _ := calculateNextRun(schedule, timezone)
|
next, _ := calculateNextRun(schedule, timezone)
|
||||||
db.Exec(
|
if jid, err := uuid.Parse(jobID); err == nil {
|
||||||
`UPDATE cron_jobs SET last_run_at = $1, last_status = $2, last_output = $3, next_run_at = $4 WHERE id = $5`,
|
_ = q.UpdateCronJobRun(context.Background(), sqlcdb.UpdateCronJobRunParams{
|
||||||
now, status, output, next, jobID,
|
LastRunAt: sql.NullTime{Time: now, Valid: true},
|
||||||
)
|
LastStatus: sql.NullString{String: status, Valid: true},
|
||||||
|
LastOutput: sql.NullString{String: output, Valid: true},
|
||||||
|
NextRunAt: ntPtr(next),
|
||||||
|
ID: jid,
|
||||||
|
})
|
||||||
|
|
||||||
if retention > 0 {
|
if retention > 0 {
|
||||||
db.Exec(
|
_ = q.TrimCronExecutions(context.Background(), sqlcdb.TrimCronExecutionsParams{
|
||||||
`DELETE FROM cron_executions WHERE cron_job_id = $1 AND id NOT IN (
|
CronJobID: jid,
|
||||||
SELECT id FROM cron_executions WHERE cron_job_id = $1
|
Limit: int32(retention),
|
||||||
ORDER BY started_at DESC LIMIT $2)`,
|
})
|
||||||
jobID, retention,
|
}
|
||||||
)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,193 @@
|
|||||||
|
package api
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"containr/internal/database"
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestCalculateNextRun(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
name string
|
||||||
|
schedule string
|
||||||
|
timezone string
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{"every minute", "* * * * *", "", false},
|
||||||
|
{"daily at 4am", "0 4 * * *", "UTC", false},
|
||||||
|
{"weekday run", "0 9 * * 1-5", "Europe/Prague", false},
|
||||||
|
{"whitespace tolerated", " */15 * * * * ", "", false},
|
||||||
|
{"empty schedule", "", "", true},
|
||||||
|
{"bad field count", "* * *", "", true},
|
||||||
|
{"garbage", "not a cron", "", true},
|
||||||
|
{"bad timezone", "0 4 * * *", "Mars/Olympus", true},
|
||||||
|
}
|
||||||
|
now := time.Now()
|
||||||
|
for _, tc := range cases {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
next, err := calculateNextRun(tc.schedule, tc.timezone)
|
||||||
|
if tc.wantErr {
|
||||||
|
if err == nil {
|
||||||
|
t.Fatalf("expected error, got next=%v", next)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
if next == nil {
|
||||||
|
t.Fatal("expected non-nil next run")
|
||||||
|
}
|
||||||
|
if !next.After(now) {
|
||||||
|
t.Errorf("next run %v is not after %v", next, now)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCalculateNextRunTimezone(t *testing.T) {
|
||||||
|
// CRON_TZ must change the computed instant: 04:00 in a fixed-offset zone
|
||||||
|
// lands at a different UTC hour than 04:00 UTC.
|
||||||
|
utcNext, err := calculateNextRun("0 4 * * *", "UTC")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("utc: %v", err)
|
||||||
|
}
|
||||||
|
tokyoNext, err := calculateNextRun("0 4 * * *", "Asia/Tokyo")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("tokyo: %v", err)
|
||||||
|
}
|
||||||
|
if utcNext.Equal(*tokyoNext) {
|
||||||
|
t.Fatalf("timezone had no effect: both %v", utcNext)
|
||||||
|
}
|
||||||
|
if !strings.EqualFold(utcNext.Location().String(), "Local") {
|
||||||
|
// UTC-scheduled runs still land in server-local time representation.
|
||||||
|
t.Logf("utc next: %v / tokyo next: %v", utcNext, tokyoNext)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestCronLiveInvariants exercises the cron.sql queries on live Postgres:
|
||||||
|
// insert/read/owner join, execution lifecycle, run bookkeeping, retention trim.
|
||||||
|
func TestCronLiveInvariants(t *testing.T) {
|
||||||
|
dsn := testDSNString(t)
|
||||||
|
db, err := database.NewConnection(dsn)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("connect: %v", err)
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
ctx := context.Background()
|
||||||
|
q := sqlcdb.New(db.DB)
|
||||||
|
|
||||||
|
userID := uuid.New()
|
||||||
|
projectID := uuid.New()
|
||||||
|
envID := uuid.New()
|
||||||
|
serviceID := uuid.New()
|
||||||
|
jobID := uuid.New()
|
||||||
|
suffix := strings.ReplaceAll(jobID.String()[:8], "-", "")
|
||||||
|
must := func(err error) {
|
||||||
|
t.Helper()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("setup: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = db.Exec(`INSERT INTO users (id, email, password_hash, name) VALUES ($1, $2, 'x', 'test')`,
|
||||||
|
userID, "cron-"+suffix+"@test.local")
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO projects (id, name, owner_id, is_approved) VALUES ($1, $2, $3, true)`,
|
||||||
|
projectID, "cron-"+suffix, userID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO environments (id, name, project_id) VALUES ($1, 'production', $2)`,
|
||||||
|
envID, projectID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO services (id, name, project_id, environment_id, service_type, source_type)
|
||||||
|
VALUES ($1, $2, $3, $4, 'web', 'image')`,
|
||||||
|
serviceID, "cron-"+suffix, projectID, envID)
|
||||||
|
must(err)
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM users WHERE id = $1`, userID) }()
|
||||||
|
|
||||||
|
next, err := calculateNextRun("0 4 * * *", "UTC")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("schedule: %v", err)
|
||||||
|
}
|
||||||
|
must(q.InsertCronJob(ctx, sqlcdb.InsertCronJobParams{
|
||||||
|
ID: jobID, ProjectID: projectID, ServiceID: serviceID,
|
||||||
|
Name: "job-" + suffix, Schedule: "0 4 * * *", Command: "echo hi",
|
||||||
|
Timezone: sql.NullString{String: "UTC", Valid: true},
|
||||||
|
Enabled: sql.NullBool{Bool: true, Valid: true},
|
||||||
|
NextRunAt: ntPtr(next),
|
||||||
|
}))
|
||||||
|
job, err := q.GetCronJob(ctx, jobID)
|
||||||
|
if err != nil || job.Name != "job-"+suffix || job.Command != "echo hi" {
|
||||||
|
t.Fatalf("job: %v %+v", err, job)
|
||||||
|
}
|
||||||
|
owner, err := q.GetCronJobOwner(ctx, jobID)
|
||||||
|
if err != nil || owner != userID {
|
||||||
|
t.Fatalf("owner: %v %v", err, owner)
|
||||||
|
}
|
||||||
|
owner2, err := q.GetCronOwnerViaService(ctx, serviceID)
|
||||||
|
if err != nil || owner2 != userID {
|
||||||
|
t.Fatalf("owner via service: %v %v", err, owner2)
|
||||||
|
}
|
||||||
|
pid, err := q.GetCronJobProjectID(ctx, jobID)
|
||||||
|
if err != nil || pid != projectID {
|
||||||
|
t.Fatalf("project: %v %v", err, pid)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Trigger-shaped read.
|
||||||
|
trig, err := q.GetCronJobForTrigger(ctx, jobID)
|
||||||
|
if err != nil || trig.ServiceID != serviceID || trig.OwnerID != userID {
|
||||||
|
t.Fatalf("trigger: %v %+v", err, trig)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Execution lifecycle + retention trim.
|
||||||
|
for i := 0; i < 3; i++ {
|
||||||
|
execID := uuid.New()
|
||||||
|
must(q.InsertCronExecution(ctx, sqlcdb.InsertCronExecutionParams{
|
||||||
|
ID: execID, CronJobID: jobID,
|
||||||
|
StartedAt: time.Now().Add(time.Duration(i) * time.Second),
|
||||||
|
Status: sql.NullString{String: "running", Valid: true},
|
||||||
|
}))
|
||||||
|
must(q.FinishCronExecution(ctx, sqlcdb.FinishCronExecutionParams{
|
||||||
|
FinishedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
|
Status: sql.NullString{String: "success", Valid: true},
|
||||||
|
Output: sql.NullString{String: "ok", Valid: true},
|
||||||
|
Error: sql.NullString{String: "", Valid: true},
|
||||||
|
ID: execID,
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
execs, err := q.ListCronExecutions(ctx, jobID)
|
||||||
|
if err != nil || len(execs) != 3 {
|
||||||
|
t.Fatalf("execs: %v %d", err, len(execs))
|
||||||
|
}
|
||||||
|
if execs[0].Status.String != "success" || !execs[0].FinishedAt.Valid {
|
||||||
|
t.Fatalf("exec row: %+v", execs[0])
|
||||||
|
}
|
||||||
|
must(q.TrimCronExecutions(ctx, sqlcdb.TrimCronExecutionsParams{CronJobID: jobID, Limit: 1}))
|
||||||
|
execs, _ = q.ListCronExecutions(ctx, jobID)
|
||||||
|
if len(execs) != 1 {
|
||||||
|
t.Fatalf("trim kept %d, want 1", len(execs))
|
||||||
|
}
|
||||||
|
|
||||||
|
must(q.UpdateCronJobRun(ctx, sqlcdb.UpdateCronJobRunParams{
|
||||||
|
LastRunAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
|
LastStatus: sql.NullString{String: "success", Valid: true},
|
||||||
|
LastOutput: sql.NullString{String: "ok", Valid: true},
|
||||||
|
NextRunAt: ntPtr(next),
|
||||||
|
ID: jobID,
|
||||||
|
}))
|
||||||
|
job, _ = q.GetCronJob(ctx, jobID)
|
||||||
|
if job.LastStatus.String != "success" || !job.LastRunAt.Valid {
|
||||||
|
t.Fatalf("post-run job: %+v", job)
|
||||||
|
}
|
||||||
|
|
||||||
|
must(q.DeleteCronJob(ctx, jobID))
|
||||||
|
if _, err := q.GetCronJob(ctx, jobID); err == nil {
|
||||||
|
t.Fatal("deleted job still readable")
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -2,12 +2,14 @@ package api
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"containr/internal/database"
|
"containr/internal/database"
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
"containr/internal/deployment"
|
"containr/internal/deployment"
|
||||||
"containr/internal/deployqueue"
|
"containr/internal/deployqueue"
|
||||||
"containr/internal/docker"
|
"containr/internal/docker"
|
||||||
"containr/internal/source"
|
"containr/internal/source"
|
||||||
"containr/internal/types"
|
"containr/internal/types"
|
||||||
"context"
|
"context"
|
||||||
|
"database/sql"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"log/slog"
|
"log/slog"
|
||||||
@@ -37,6 +39,72 @@ type DeploymentModel struct {
|
|||||||
UpdatedAt time.Time `json:"updated_at" db:"updated_at"`
|
UpdatedAt time.Time `json:"updated_at" db:"updated_at"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// deploymentRowShape carries the deployment columns shared by the
|
||||||
|
// generated list/detail/rollback row types.
|
||||||
|
type deploymentRowShape struct {
|
||||||
|
ID uuid.UUID
|
||||||
|
ServiceID uuid.UUID
|
||||||
|
CommitHash sql.NullString
|
||||||
|
Status sql.NullString
|
||||||
|
ImageName sql.NullString
|
||||||
|
ImageTag sql.NullString
|
||||||
|
BuildLog sql.NullString
|
||||||
|
RuntimeLog sql.NullString
|
||||||
|
Error sql.NullString
|
||||||
|
StartedAt sql.NullTime
|
||||||
|
CompletedAt sql.NullTime
|
||||||
|
CreatedAt sql.NullTime
|
||||||
|
UpdatedAt sql.NullTime
|
||||||
|
}
|
||||||
|
|
||||||
|
func strPtrNS(v sql.NullString) *string {
|
||||||
|
if !v.Valid {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
s := v.String
|
||||||
|
return &s
|
||||||
|
}
|
||||||
|
|
||||||
|
func ptrStr(s *string) string {
|
||||||
|
if s == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return *s
|
||||||
|
}
|
||||||
|
|
||||||
|
func timePtrNT(v sql.NullTime) *time.Time {
|
||||||
|
if !v.Valid {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
t := v.Time
|
||||||
|
return &t
|
||||||
|
}
|
||||||
|
|
||||||
|
func ntPtr(t *time.Time) sql.NullTime {
|
||||||
|
if t == nil {
|
||||||
|
return sql.NullTime{}
|
||||||
|
}
|
||||||
|
return sql.NullTime{Time: *t, Valid: true}
|
||||||
|
}
|
||||||
|
|
||||||
|
func deploymentModelFrom(r deploymentRowShape) DeploymentModel {
|
||||||
|
return DeploymentModel{
|
||||||
|
ID: r.ID,
|
||||||
|
ServiceID: r.ServiceID,
|
||||||
|
CommitHash: strPtrNS(r.CommitHash),
|
||||||
|
Status: r.Status.String,
|
||||||
|
ImageName: r.ImageName.String,
|
||||||
|
ImageTag: r.ImageTag.String,
|
||||||
|
BuildLog: r.BuildLog.String,
|
||||||
|
RuntimeLog: r.RuntimeLog.String,
|
||||||
|
Error: strPtrNS(r.Error),
|
||||||
|
StartedAt: timePtrNT(r.StartedAt),
|
||||||
|
CompletedAt: timePtrNT(r.CompletedAt),
|
||||||
|
CreatedAt: r.CreatedAt.Time,
|
||||||
|
UpdatedAt: r.UpdatedAt.Time,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
type CreateDeploymentRequest struct {
|
type CreateDeploymentRequest struct {
|
||||||
CommitHash string `json:"commit_hash"`
|
CommitHash string `json:"commit_hash"`
|
||||||
Branch string `json:"branch"`
|
Branch string `json:"branch"`
|
||||||
@@ -84,44 +152,29 @@ func handleGetDeployments(c *gin.Context) {
|
|||||||
|
|
||||||
offset, limit := pageWindow(c, 50, 100)
|
offset, limit := pageWindow(c, 50, 100)
|
||||||
|
|
||||||
var total int
|
q := sqlcdb.New(db.(*database.DB).DB)
|
||||||
_ = db.(*database.DB).QueryRow(
|
total, _ := q.CountDeploymentsForService(c.Request.Context(), serviceID)
|
||||||
`SELECT COUNT(*) FROM deployments WHERE service_id = $1`, serviceID,
|
|
||||||
).Scan(&total)
|
|
||||||
|
|
||||||
rows, err := db.(*database.DB).Query(
|
rows, err := q.ListDeploymentsForService(c.Request.Context(), sqlcdb.ListDeploymentsForServiceParams{
|
||||||
`SELECT id, service_id, commit_hash, status, image_name, image_tag,
|
ServiceID: serviceID, Limit: int32(limit), Offset: int32(offset)})
|
||||||
build_log, runtime_log, error, started_at, completed_at, created_at, updated_at
|
|
||||||
FROM deployments
|
|
||||||
WHERE service_id = $1
|
|
||||||
ORDER BY created_at DESC
|
|
||||||
LIMIT $2 OFFSET $3`,
|
|
||||||
serviceID, limit, offset,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to retrieve deployments"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to retrieve deployments"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var deployments []DeploymentModel
|
deployments := make([]DeploymentModel, 0, len(rows))
|
||||||
for rows.Next() {
|
for _, r := range rows {
|
||||||
var d DeploymentModel
|
deployments = append(deployments, deploymentModelFrom(deploymentRowShape{
|
||||||
err := rows.Scan(
|
ID: r.ID, ServiceID: r.ServiceID, CommitHash: r.CommitHash, Status: r.Status,
|
||||||
&d.ID, &d.ServiceID, &d.CommitHash, &d.Status, &d.ImageName, &d.ImageTag,
|
ImageName: r.ImageName, ImageTag: r.ImageTag, BuildLog: r.BuildLog,
|
||||||
&d.BuildLog, &d.RuntimeLog, &d.Error, &d.StartedAt, &d.CompletedAt,
|
RuntimeLog: r.RuntimeLog, Error: r.Error, StartedAt: r.StartedAt,
|
||||||
&d.CreatedAt, &d.UpdatedAt,
|
CompletedAt: r.CompletedAt, CreatedAt: r.CreatedAt, UpdatedAt: r.UpdatedAt,
|
||||||
)
|
}))
|
||||||
if err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to scan deployment"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
deployments = append(deployments, d)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
c.JSON(http.StatusOK, gin.H{
|
c.JSON(http.StatusOK, gin.H{
|
||||||
"deployments": deployments,
|
"deployments": deployments,
|
||||||
"pagination": paginationMeta(offset, limit, total),
|
"pagination": paginationMeta(offset, limit, int(total)),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -152,51 +205,33 @@ func handleGetRecentDeployments(c *gin.Context) {
|
|||||||
|
|
||||||
offset, limit := pageWindow(c, 10, 50)
|
offset, limit := pageWindow(c, 10, 50)
|
||||||
|
|
||||||
var total int
|
q := sqlcdb.New(db.(*database.DB).DB)
|
||||||
if err := db.(*database.DB).QueryRow(
|
total, err := q.CountAccessibleDeployments(c.Request.Context(), sqlcdb.CountAccessibleDeploymentsParams{
|
||||||
`SELECT COUNT(*)
|
OwnerID: userID, Column2: isAdmin})
|
||||||
FROM deployments d
|
if err != nil {
|
||||||
JOIN services s ON s.id = d.service_id
|
|
||||||
JOIN projects p ON p.id = s.project_id
|
|
||||||
WHERE p.is_approved OR p.owner_id = $1 OR $2::bool
|
|
||||||
OR EXISTS (SELECT 1 FROM project_members pm WHERE pm.project_id = p.id AND pm.user_id = $1)`,
|
|
||||||
userID, isAdmin,
|
|
||||||
).Scan(&total); err != nil {
|
|
||||||
total = 0
|
total = 0
|
||||||
}
|
}
|
||||||
|
|
||||||
rows, err := db.(*database.DB).Query(
|
rows, err := q.ListRecentAccessibleDeployments(c.Request.Context(), sqlcdb.ListRecentAccessibleDeploymentsParams{
|
||||||
`SELECT d.id, d.service_id, s.name, p.name, d.status,
|
OwnerID: userID, Column2: isAdmin, Limit: int32(limit), Offset: int32(offset)})
|
||||||
COALESCE(d.image_name, ''), d.started_at, d.completed_at, d.created_at
|
|
||||||
FROM deployments d
|
|
||||||
JOIN services s ON s.id = d.service_id
|
|
||||||
JOIN projects p ON p.id = s.project_id
|
|
||||||
WHERE p.is_approved OR p.owner_id = $1 OR $2::bool
|
|
||||||
OR EXISTS (SELECT 1 FROM project_members pm WHERE pm.project_id = p.id AND pm.user_id = $1)
|
|
||||||
ORDER BY d.created_at DESC
|
|
||||||
LIMIT $3 OFFSET $4`,
|
|
||||||
userID, isAdmin, limit, offset,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to retrieve deployments"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to retrieve deployments"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
deployments := []RecentDeployment{}
|
deployments := make([]RecentDeployment, 0, len(rows))
|
||||||
for rows.Next() {
|
for _, r := range rows {
|
||||||
var d RecentDeployment
|
deployments = append(deployments, RecentDeployment{
|
||||||
if err := rows.Scan(&d.ID, &d.ServiceID, &d.ServiceName, &d.ProjectName,
|
ID: r.ID, ServiceID: r.ServiceID, ServiceName: r.ServiceName,
|
||||||
&d.Status, &d.ImageName, &d.StartedAt, &d.CompletedAt, &d.CreatedAt); err != nil {
|
ProjectName: r.ProjectName, Status: r.Status.String, ImageName: r.ImageName,
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to scan deployment"})
|
StartedAt: timePtrNT(r.StartedAt), CompletedAt: timePtrNT(r.CompletedAt),
|
||||||
return
|
CreatedAt: r.CreatedAt.Time,
|
||||||
}
|
})
|
||||||
deployments = append(deployments, d)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
c.JSON(http.StatusOK, gin.H{
|
c.JSON(http.StatusOK, gin.H{
|
||||||
"deployments": deployments,
|
"deployments": deployments,
|
||||||
"pagination": paginationMeta(offset, limit, total),
|
"pagination": paginationMeta(offset, limit, int(total)),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -230,38 +265,27 @@ func handleCreateDeployment(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var service Service
|
ctx := c.Request.Context()
|
||||||
var projectOwner string
|
q := sqlcdb.New(db.(*database.DB).DB)
|
||||||
err = db.(*database.DB).QueryRow(
|
sr, err := q.GetServiceForDeployWithOwner(ctx, serviceID)
|
||||||
`SELECT s.id, s.project_id, s.name, s.type, s.status, s.image, s.command,
|
|
||||||
s.environment, s.git_repo, s.git_branch, s.build_path, s.cpu, s.memory,
|
|
||||||
COALESCE(s.replicas, 1), COALESCE(s.port, 0),
|
|
||||||
COALESCE(s.domain, ''), COALESCE(s.healthcheck_path, ''),
|
|
||||||
COALESCE(s.restart_policy, 'unless-stopped'),
|
|
||||||
COALESCE(s.builder, 'auto'), COALESCE(s.cpu_reserve, ''),
|
|
||||||
COALESCE(s.memory_reserve, ''), COALESCE(s.static_build_cmd, ''),
|
|
||||||
COALESCE(s.static_dir, ''),
|
|
||||||
s.created_at, s.updated_at, p.owner_id
|
|
||||||
FROM services s
|
|
||||||
JOIN projects p ON s.project_id = p.id
|
|
||||||
WHERE s.id = $1`,
|
|
||||||
serviceID,
|
|
||||||
).Scan(
|
|
||||||
&service.ID, &service.ProjectID, &service.Name, &service.Type, &service.Status,
|
|
||||||
&service.Image, &service.Command, &service.Environment, &service.GitRepo,
|
|
||||||
&service.GitBranch, &service.BuildPath, &service.CPU, &service.Memory,
|
|
||||||
&service.Replicas, &service.Port, &service.Domain, &service.HealthCheckPath,
|
|
||||||
&service.RestartPolicy, &service.Builder, &service.CPUReserve,
|
|
||||||
&service.MemoryReserve, &service.StaticBuildCmd, &service.StaticDir,
|
|
||||||
&service.CreatedAt, &service.UpdatedAt, &projectOwner,
|
|
||||||
)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
service := serviceForDeployFromRow(sqlcdb.GetServiceForDeployRow{
|
||||||
|
ID: sr.ID, ProjectID: sr.ProjectID, Name: sr.Name, Type: sr.Type,
|
||||||
|
Status: sr.Status, Image: sr.Image, Command: sr.Command,
|
||||||
|
Environment: sr.Environment, GitRepo: sr.GitRepo, GitBranch: sr.GitBranch,
|
||||||
|
BuildPath: sr.BuildPath, Cpu: sr.Cpu, Memory: sr.Memory,
|
||||||
|
Replicas: sr.Replicas, Port: sr.Port, Domain: sr.Domain,
|
||||||
|
HealthcheckPath: sr.HealthcheckPath, RestartPolicy: sr.RestartPolicy,
|
||||||
|
Builder: sr.Builder, CpuReserve: sr.CpuReserve, MemoryReserve: sr.MemoryReserve,
|
||||||
|
StaticBuildCmd: sr.StaticBuildCmd, StaticDir: sr.StaticDir,
|
||||||
|
CreatedAt: sr.CreatedAt, UpdatedAt: sr.UpdatedAt,
|
||||||
|
})
|
||||||
|
projectOwner := sr.OwnerID
|
||||||
|
|
||||||
if projectOwner != userID.(string) {
|
if projectOwner.String() != userID.(string) {
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -287,12 +311,17 @@ func handleCreateDeployment(c *gin.Context) {
|
|||||||
UpdatedAt: now,
|
UpdatedAt: now,
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = db.(*database.DB).Exec(
|
err = q.InsertDeployment(ctx, sqlcdb.InsertDeploymentParams{
|
||||||
`INSERT INTO deployments
|
ID: d.ID,
|
||||||
(id, service_id, version, commit_hash, status, image_name, image_tag, created_at, updated_at)
|
ServiceID: d.ServiceID,
|
||||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)`,
|
Version: fmt.Sprintf("v%d", now.Unix()),
|
||||||
d.ID, d.ServiceID, fmt.Sprintf("v%d", now.Unix()), d.CommitHash, d.Status, d.ImageName, d.ImageTag, d.CreatedAt, d.UpdatedAt,
|
CommitHash: sql.NullString{String: ptrStr(d.CommitHash), Valid: d.CommitHash != nil},
|
||||||
)
|
Status: sql.NullString{String: d.Status, Valid: true},
|
||||||
|
ImageName: sql.NullString{String: d.ImageName, Valid: true},
|
||||||
|
ImageTag: sql.NullString{String: d.ImageTag, Valid: true},
|
||||||
|
CreatedAt: sql.NullTime{Time: d.CreatedAt, Valid: true},
|
||||||
|
UpdatedAt: sql.NullTime{Time: d.UpdatedAt, Valid: true},
|
||||||
|
})
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create deployment"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create deployment"})
|
||||||
@@ -303,20 +332,20 @@ func handleCreateDeployment(c *gin.Context) {
|
|||||||
if !exists || engine == nil {
|
if !exists || engine == nil {
|
||||||
unavailableErr := "Deployment engine unavailable. Docker may not be configured on this server."
|
unavailableErr := "Deployment engine unavailable. Docker may not be configured on this server."
|
||||||
completedAt := time.Now()
|
completedAt := time.Now()
|
||||||
_, _ = db.(*database.DB).Exec(
|
_ = q.FailDeployment(ctx, sqlcdb.FailDeploymentParams{
|
||||||
`UPDATE deployments
|
Error: sql.NullString{String: unavailableErr, Valid: true},
|
||||||
SET status = 'failed', error = $1, completed_at = $2, updated_at = $2
|
CompletedAt: sql.NullTime{Time: completedAt, Valid: true},
|
||||||
WHERE id = $3`,
|
ID: d.ID,
|
||||||
unavailableErr, completedAt, d.ID,
|
})
|
||||||
)
|
|
||||||
d.Status = "failed"
|
d.Status = "failed"
|
||||||
d.Error = &unavailableErr
|
d.Error = &unavailableErr
|
||||||
d.CompletedAt = &completedAt
|
d.CompletedAt = &completedAt
|
||||||
} else {
|
} else {
|
||||||
_, err = db.(*database.DB).Exec(
|
err = q.SetServiceStatus(ctx, sqlcdb.SetServiceStatusParams{
|
||||||
`UPDATE services SET status = 'building', updated_at = $1 WHERE id = $2`,
|
Status: sql.NullString{String: "building", Valid: true},
|
||||||
time.Now(), serviceID,
|
UpdatedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
)
|
ID: serviceID,
|
||||||
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update service status"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update service status"})
|
||||||
return
|
return
|
||||||
@@ -331,14 +360,17 @@ func handleCreateDeployment(c *gin.Context) {
|
|||||||
})
|
})
|
||||||
if pos > 0 {
|
if pos > 0 {
|
||||||
d.Status = "queued"
|
d.Status = "queued"
|
||||||
_, _ = db.(*database.DB).Exec(
|
now := time.Now()
|
||||||
`UPDATE deployments SET status = 'queued', updated_at = $1 WHERE id = $2`,
|
_ = q.SetDeploymentStatus(ctx, sqlcdb.SetDeploymentStatusParams{
|
||||||
time.Now(), d.ID,
|
Status: sql.NullString{String: "queued", Valid: true},
|
||||||
)
|
UpdatedAt: sql.NullTime{Time: now, Valid: true},
|
||||||
_, _ = db.(*database.DB).Exec(
|
ID: d.ID,
|
||||||
`UPDATE services SET status = 'queued', updated_at = $1 WHERE id = $2`,
|
})
|
||||||
time.Now(), serviceID,
|
_ = q.SetServiceStatus(ctx, sqlcdb.SetServiceStatusParams{
|
||||||
)
|
Status: sql.NullString{String: "queued", Valid: true},
|
||||||
|
UpdatedAt: sql.NullTime{Time: now, Valid: true},
|
||||||
|
ID: serviceID,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -380,6 +412,7 @@ func runDeploymentAndSyncWithImage(
|
|||||||
) {
|
) {
|
||||||
ctx, cancel := context.WithTimeout(parentCtx, 30*time.Minute)
|
ctx, cancel := context.WithTimeout(parentCtx, 30*time.Minute)
|
||||||
defer cancel()
|
defer cancel()
|
||||||
|
q := sqlcdb.New(db.DB)
|
||||||
|
|
||||||
// The checkout and any shipped artifacts die with this deployment.
|
// The checkout and any shipped artifacts die with this deployment.
|
||||||
var artifactIDs []string
|
var artifactIDs []string
|
||||||
@@ -392,10 +425,11 @@ func runDeploymentAndSyncWithImage(
|
|||||||
}()
|
}()
|
||||||
failDeploy := func(msg string) {
|
failDeploy := func(msg string) {
|
||||||
failedAt := time.Now()
|
failedAt := time.Now()
|
||||||
_, _ = db.Exec(
|
_ = q.FailDeployment(ctx, sqlcdb.FailDeploymentParams{
|
||||||
`UPDATE deployments SET status = 'failed', error = $1, completed_at = $2, updated_at = $2 WHERE id = $3`,
|
Error: sql.NullString{String: msg, Valid: true},
|
||||||
msg, failedAt, dbDeployment.ID,
|
CompletedAt: sql.NullTime{Time: failedAt, Valid: true},
|
||||||
)
|
ID: dbDeployment.ID,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
sourcePath := strings.TrimSpace(service.BuildPath)
|
sourcePath := strings.TrimSpace(service.BuildPath)
|
||||||
@@ -423,11 +457,10 @@ func runDeploymentAndSyncWithImage(
|
|||||||
replicas = 1
|
replicas = 1
|
||||||
}
|
}
|
||||||
|
|
||||||
var publishedPort int32
|
|
||||||
// Best effort: reuse the host port from the last live deployment so public
|
// Best effort: reuse the host port from the last live deployment so public
|
||||||
// URLs survive redeploys. Column exists post-migration; older DBs get
|
// URLs survive redeploys. Column exists post-migration; older DBs get
|
||||||
// ephemeral ports.
|
// ephemeral ports.
|
||||||
_ = db.QueryRow(`SELECT published_port FROM services WHERE id = $1`, service.ID).Scan(&publishedPort)
|
publishedPort, _ := q.GetServicePublishedPort(ctx, service.ID)
|
||||||
|
|
||||||
publicPort := int32(service.Port)
|
publicPort := int32(service.Port)
|
||||||
if service.Builder == "static" && publicPort == 0 {
|
if service.Builder == "static" && publicPort == 0 {
|
||||||
@@ -451,7 +484,9 @@ func runDeploymentAndSyncWithImage(
|
|||||||
// remote builds package it into a context artifact the node builds.
|
// remote builds package it into a context artifact the node builds.
|
||||||
if service.GitRepo != "" && imageOverride == "" {
|
if service.GitRepo != "" && imageOverride == "" {
|
||||||
var ownerID string
|
var ownerID string
|
||||||
_ = db.QueryRow(`SELECT owner_id FROM projects WHERE id = $1`, service.ProjectID).Scan(&ownerID)
|
if o, err := q.GetProjectOwner(ctx, service.ProjectID); err == nil {
|
||||||
|
ownerID = o.String()
|
||||||
|
}
|
||||||
branch := req.Branch
|
branch := req.Branch
|
||||||
if branch == "" {
|
if branch == "" {
|
||||||
branch = service.GitBranch
|
branch = service.GitBranch
|
||||||
@@ -535,7 +570,9 @@ func runDeploymentAndSyncWithImage(
|
|||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
var ownerID string
|
var ownerID string
|
||||||
_ = db.QueryRow(`SELECT owner_id FROM projects WHERE id = $1`, service.ProjectID).Scan(&ownerID)
|
if o, err := q.GetProjectOwner(ctx, service.ProjectID); err == nil {
|
||||||
|
ownerID = o.String()
|
||||||
|
}
|
||||||
auth := registryAuthFor(db, ownerID, image)
|
auth := registryAuthFor(db, ownerID, image)
|
||||||
deployReq.BuildConfig = &deployment.BuildConfig{
|
deployReq.BuildConfig = &deployment.BuildConfig{
|
||||||
BuildType: "prebuilt",
|
BuildType: "prebuilt",
|
||||||
@@ -605,10 +642,11 @@ func runDeploymentAndSyncWithImage(
|
|||||||
|
|
||||||
if msg := capacityCheck(parentCtx, db, engine.DockerClient(), service, replicas); msg != "" {
|
if msg := capacityCheck(parentCtx, db, engine.DockerClient(), service, replicas); msg != "" {
|
||||||
failedAt := time.Now()
|
failedAt := time.Now()
|
||||||
_, _ = db.Exec(
|
_ = q.FailDeployment(ctx, sqlcdb.FailDeploymentParams{
|
||||||
`UPDATE deployments SET status = 'failed', error = $1, completed_at = $2, updated_at = $2 WHERE id = $3`,
|
Error: sql.NullString{String: msg, Valid: true},
|
||||||
msg, failedAt, dbDeployment.ID,
|
CompletedAt: sql.NullTime{Time: failedAt, Valid: true},
|
||||||
)
|
ID: dbDeployment.ID,
|
||||||
|
})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -616,16 +654,16 @@ func runDeploymentAndSyncWithImage(
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
failedAt := time.Now()
|
failedAt := time.Now()
|
||||||
failure := "Failed to start deployment engine: " + err.Error()
|
failure := "Failed to start deployment engine: " + err.Error()
|
||||||
_, _ = db.Exec(
|
_ = q.FailDeployment(ctx, sqlcdb.FailDeploymentParams{
|
||||||
`UPDATE deployments
|
Error: sql.NullString{String: failure, Valid: true},
|
||||||
SET status = 'failed', error = $1, completed_at = $2, updated_at = $2
|
CompletedAt: sql.NullTime{Time: failedAt, Valid: true},
|
||||||
WHERE id = $3`,
|
ID: dbDeployment.ID,
|
||||||
failure, failedAt, dbDeployment.ID,
|
})
|
||||||
)
|
_ = q.SetServiceStatus(ctx, sqlcdb.SetServiceStatusParams{
|
||||||
_, _ = db.Exec(
|
Status: sql.NullString{String: "failed", Valid: true},
|
||||||
`UPDATE services SET status = 'failed', updated_at = $1 WHERE id = $2`,
|
UpdatedAt: sql.NullTime{Time: failedAt, Valid: true},
|
||||||
failedAt, service.ID,
|
ID: service.ID,
|
||||||
)
|
})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -640,16 +678,17 @@ func runDeploymentAndSyncWithImage(
|
|||||||
if errors.Is(ctx.Err(), context.Canceled) {
|
if errors.Is(ctx.Err(), context.Canceled) {
|
||||||
finalStatus, finalErr = "cancelled", "Deployment cancelled"
|
finalStatus, finalErr = "cancelled", "Deployment cancelled"
|
||||||
}
|
}
|
||||||
_, _ = db.Exec(
|
_ = q.CompleteDeployment(ctx, sqlcdb.CompleteDeploymentParams{
|
||||||
`UPDATE deployments
|
Status: sql.NullString{String: finalStatus, Valid: true},
|
||||||
SET status = $1, error = $2, completed_at = $3, updated_at = $3
|
Error: sql.NullString{String: finalErr, Valid: true},
|
||||||
WHERE id = $4`,
|
CompletedAt: sql.NullTime{Time: failedAt, Valid: true},
|
||||||
finalStatus, finalErr, failedAt, dbDeployment.ID,
|
ID: dbDeployment.ID,
|
||||||
)
|
})
|
||||||
_, _ = db.Exec(
|
_ = q.SetServiceStatus(ctx, sqlcdb.SetServiceStatusParams{
|
||||||
`UPDATE services SET status = $1, updated_at = $2 WHERE id = $3`,
|
Status: sql.NullString{String: finalStatus, Valid: true},
|
||||||
finalStatus, failedAt, service.ID,
|
UpdatedAt: sql.NullTime{Time: failedAt, Valid: true},
|
||||||
)
|
ID: service.ID,
|
||||||
|
})
|
||||||
return
|
return
|
||||||
case <-syncTicker.C:
|
case <-syncTicker.C:
|
||||||
current, getErr := engine.GetDeployment(engineDeployment.ID)
|
current, getErr := engine.GetDeployment(engineDeployment.ID)
|
||||||
@@ -660,49 +699,36 @@ func runDeploymentAndSyncWithImage(
|
|||||||
dbStatus := mapEngineStatusToDBStatus(current.Status)
|
dbStatus := mapEngineStatusToDBStatus(current.Status)
|
||||||
imageName, imageTag := splitImageReference(current.ImageName, dbDeployment.ImageTag)
|
imageName, imageTag := splitImageReference(current.ImageName, dbDeployment.ImageTag)
|
||||||
|
|
||||||
var dbError interface{}
|
now := time.Now()
|
||||||
if current.Error != "" {
|
_ = q.SyncDeploymentProgress(ctx, sqlcdb.SyncDeploymentProgressParams{
|
||||||
dbError = current.Error
|
Status: sql.NullString{String: dbStatus, Valid: true},
|
||||||
}
|
ImageName: sql.NullString{String: imageName, Valid: true},
|
||||||
|
ImageTag: sql.NullString{String: imageTag, Valid: true},
|
||||||
_, _ = db.Exec(
|
BuildLog: sql.NullString{String: current.BuildLog, Valid: true},
|
||||||
`UPDATE deployments
|
RuntimeLog: sql.NullString{String: current.DeployLog, Valid: true},
|
||||||
SET status = $1,
|
Error: sql.NullString{String: current.Error, Valid: current.Error != ""},
|
||||||
image_name = $2,
|
StartedAt: ntPtr(current.StartedAt),
|
||||||
image_tag = $3,
|
CompletedAt: ntPtr(current.CompletedAt),
|
||||||
build_log = $4,
|
UpdatedAt: sql.NullTime{Time: now, Valid: true},
|
||||||
runtime_log = $5,
|
ID: dbDeployment.ID,
|
||||||
error = $6,
|
})
|
||||||
started_at = $7,
|
|
||||||
completed_at = $8,
|
|
||||||
updated_at = $9
|
|
||||||
WHERE id = $10`,
|
|
||||||
dbStatus,
|
|
||||||
imageName,
|
|
||||||
imageTag,
|
|
||||||
current.BuildLog,
|
|
||||||
current.DeployLog,
|
|
||||||
dbError,
|
|
||||||
current.StartedAt,
|
|
||||||
current.CompletedAt,
|
|
||||||
time.Now(),
|
|
||||||
dbDeployment.ID,
|
|
||||||
)
|
|
||||||
|
|
||||||
switch dbStatus {
|
switch dbStatus {
|
||||||
case "deployed":
|
case "deployed":
|
||||||
_, _ = db.Exec(
|
_ = q.SetServiceStatus(ctx, sqlcdb.SetServiceStatusParams{
|
||||||
`UPDATE services SET status = 'running', updated_at = $1 WHERE id = $2`,
|
Status: sql.NullString{String: "running", Valid: true},
|
||||||
time.Now(), service.ID,
|
UpdatedAt: sql.NullTime{Time: now, Valid: true},
|
||||||
)
|
ID: service.ID,
|
||||||
|
})
|
||||||
insertUserNotification(db, userID, "deployment", "Deployment succeeded",
|
insertUserNotification(db, userID, "deployment", "Deployment succeeded",
|
||||||
fmt.Sprintf("Service %s is now running.", service.Name), "service", service.ID.String())
|
fmt.Sprintf("Service %s is now running.", service.Name), "service", service.ID.String())
|
||||||
return
|
return
|
||||||
case "failed":
|
case "failed":
|
||||||
_, _ = db.Exec(
|
_ = q.SetServiceStatus(ctx, sqlcdb.SetServiceStatusParams{
|
||||||
`UPDATE services SET status = 'failed', updated_at = $1 WHERE id = $2`,
|
Status: sql.NullString{String: "failed", Valid: true},
|
||||||
time.Now(), service.ID,
|
UpdatedAt: sql.NullTime{Time: now, Valid: true},
|
||||||
)
|
ID: service.ID,
|
||||||
|
})
|
||||||
body := fmt.Sprintf("Service %s failed to deploy.", service.Name)
|
body := fmt.Sprintf("Service %s failed to deploy.", service.Name)
|
||||||
if current.Error != "" {
|
if current.Error != "" {
|
||||||
body = fmt.Sprintf("Service %s failed to deploy: %s", service.Name, current.Error)
|
body = fmt.Sprintf("Service %s failed to deploy: %s", service.Name, current.Error)
|
||||||
@@ -754,21 +780,16 @@ func handleCancelDeployment(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var status, owner string
|
ctx := c.Request.Context()
|
||||||
var serviceID uuid.UUID
|
q := sqlcdb.New(db.(*database.DB).DB)
|
||||||
err = db.(*database.DB).QueryRow(
|
access, err := q.GetDeploymentAccess(ctx, deploymentID)
|
||||||
`SELECT d.status, d.service_id, p.owner_id
|
|
||||||
FROM deployments d
|
|
||||||
JOIN services s ON d.service_id = s.id
|
|
||||||
JOIN projects p ON s.project_id = p.id
|
|
||||||
WHERE d.id = $1`,
|
|
||||||
deploymentID,
|
|
||||||
).Scan(&status, &serviceID, &owner)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Deployment not found", "code": "NOT_FOUND"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Deployment not found", "code": "NOT_FOUND"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if owner != userID.(string) && !contextIsAdmin(c) {
|
serviceID := access.ServiceID
|
||||||
|
status := access.Status.String
|
||||||
|
if access.OwnerID.String() != userID.(string) && !contextIsAdmin(c) {
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied", "code": "FORBIDDEN"})
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied", "code": "FORBIDDEN"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -782,14 +803,17 @@ func handleCancelDeployment(c *gin.Context) {
|
|||||||
switch getDeployQueue(c).Cancel(serviceID, deploymentID) {
|
switch getDeployQueue(c).Cancel(serviceID, deploymentID) {
|
||||||
case deployqueue.WasQueued:
|
case deployqueue.WasQueued:
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
_, _ = db.(*database.DB).Exec(
|
_ = q.CompleteDeployment(ctx, sqlcdb.CompleteDeploymentParams{
|
||||||
`UPDATE deployments SET status = 'cancelled', error = 'Deployment cancelled', completed_at = $1, updated_at = $1 WHERE id = $2`,
|
Status: sql.NullString{String: "cancelled", Valid: true},
|
||||||
now, deploymentID,
|
Error: sql.NullString{String: "Deployment cancelled", Valid: true},
|
||||||
)
|
CompletedAt: sql.NullTime{Time: now, Valid: true},
|
||||||
_, _ = db.(*database.DB).Exec(
|
ID: deploymentID,
|
||||||
`UPDATE services SET status = 'cancelled', updated_at = $1 WHERE id = $2`,
|
})
|
||||||
now, serviceID,
|
_ = q.SetServiceStatus(ctx, sqlcdb.SetServiceStatusParams{
|
||||||
)
|
Status: sql.NullString{String: "cancelled", Valid: true},
|
||||||
|
UpdatedAt: sql.NullTime{Time: now, Valid: true},
|
||||||
|
ID: serviceID,
|
||||||
|
})
|
||||||
c.JSON(http.StatusOK, gin.H{"status": "cancelled"})
|
c.JSON(http.StatusOK, gin.H{"status": "cancelled"})
|
||||||
case deployqueue.WasActive:
|
case deployqueue.WasActive:
|
||||||
c.JSON(http.StatusAccepted, gin.H{"status": "cancelling"})
|
c.JSON(http.StatusAccepted, gin.H{"status": "cancelling"})
|
||||||
@@ -832,27 +856,18 @@ func handleGetDeployment(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var d DeploymentModel
|
r, err := sqlcdb.New(db.(*database.DB).DB).GetDeploymentWithProject(c.Request.Context(), deploymentID)
|
||||||
var projectID uuid.UUID
|
|
||||||
err = db.(*database.DB).QueryRow(
|
|
||||||
`SELECT d.id, d.service_id, d.commit_hash, d.status, d.image_name, d.image_tag,
|
|
||||||
d.build_log, d.runtime_log, d.error, d.started_at, d.completed_at,
|
|
||||||
d.created_at, d.updated_at, p.id
|
|
||||||
FROM deployments d
|
|
||||||
JOIN services s ON d.service_id = s.id
|
|
||||||
JOIN projects p ON s.project_id = p.id
|
|
||||||
WHERE d.id = $1`,
|
|
||||||
deploymentID,
|
|
||||||
).Scan(
|
|
||||||
&d.ID, &d.ServiceID, &d.CommitHash, &d.Status, &d.ImageName, &d.ImageTag,
|
|
||||||
&d.BuildLog, &d.RuntimeLog, &d.Error, &d.StartedAt, &d.CompletedAt,
|
|
||||||
&d.CreatedAt, &d.UpdatedAt, &projectID,
|
|
||||||
)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Deployment not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Deployment not found"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
projectID := r.ProjectID
|
||||||
|
d := deploymentModelFrom(deploymentRowShape{
|
||||||
|
ID: r.ID, ServiceID: r.ServiceID, CommitHash: r.CommitHash, Status: r.Status,
|
||||||
|
ImageName: r.ImageName, ImageTag: r.ImageTag, BuildLog: r.BuildLog,
|
||||||
|
RuntimeLog: r.RuntimeLog, Error: r.Error, StartedAt: r.StartedAt,
|
||||||
|
CompletedAt: r.CompletedAt, CreatedAt: r.CreatedAt, UpdatedAt: r.UpdatedAt,
|
||||||
|
})
|
||||||
|
|
||||||
if _, allowed := projectReadAccess(c, db.(*database.DB), projectID); !allowed {
|
if _, allowed := projectReadAccess(c, db.(*database.DB), projectID); !allowed {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Deployment not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Deployment not found"})
|
||||||
@@ -882,33 +897,23 @@ func handleRollbackDeployment(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var targetDeployment DeploymentModel
|
ctx := c.Request.Context()
|
||||||
var serviceID uuid.UUID
|
q := sqlcdb.New(db.(*database.DB).DB)
|
||||||
var ownerCheck string
|
r, err := q.GetDeploymentForRollback(ctx, deploymentID)
|
||||||
|
|
||||||
err = db.(*database.DB).QueryRow(
|
|
||||||
`SELECT d.id, d.service_id, d.commit_hash, d.status, d.image_name, d.image_tag,
|
|
||||||
d.build_log, d.runtime_log, d.error, d.started_at, d.completed_at,
|
|
||||||
d.created_at, d.updated_at, p.owner_id
|
|
||||||
FROM deployments d
|
|
||||||
JOIN services s ON d.service_id = s.id
|
|
||||||
JOIN projects p ON s.project_id = p.id
|
|
||||||
WHERE d.id = $1`,
|
|
||||||
deploymentID,
|
|
||||||
).Scan(
|
|
||||||
&targetDeployment.ID, &serviceID, &targetDeployment.CommitHash, &targetDeployment.Status,
|
|
||||||
&targetDeployment.ImageName, &targetDeployment.ImageTag, &targetDeployment.BuildLog,
|
|
||||||
&targetDeployment.RuntimeLog, &targetDeployment.Error, &targetDeployment.StartedAt,
|
|
||||||
&targetDeployment.CompletedAt, &targetDeployment.CreatedAt, &targetDeployment.UpdatedAt,
|
|
||||||
&ownerCheck,
|
|
||||||
)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Deployment not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Deployment not found"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
serviceID := r.ServiceID
|
||||||
|
targetDeployment := deploymentModelFrom(deploymentRowShape{
|
||||||
|
ID: r.ID, ServiceID: r.ServiceID, CommitHash: r.CommitHash, Status: r.Status,
|
||||||
|
ImageName: r.ImageName, ImageTag: r.ImageTag, BuildLog: r.BuildLog,
|
||||||
|
RuntimeLog: r.RuntimeLog, Error: r.Error, StartedAt: r.StartedAt,
|
||||||
|
CompletedAt: r.CompletedAt, CreatedAt: r.CreatedAt, UpdatedAt: r.UpdatedAt,
|
||||||
|
})
|
||||||
|
ownerCheck := r.OwnerID
|
||||||
|
|
||||||
if ownerCheck != userID.(string) && !contextIsAdmin(c) {
|
if ownerCheck.String() != userID.(string) && !contextIsAdmin(c) {
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -931,23 +936,28 @@ func handleRollbackDeployment(c *gin.Context) {
|
|||||||
UpdatedAt: now,
|
UpdatedAt: now,
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = db.(*database.DB).Exec(
|
err = q.InsertDeployment(ctx, sqlcdb.InsertDeploymentParams{
|
||||||
`INSERT INTO deployments
|
ID: rollback.ID,
|
||||||
(id, service_id, version, commit_hash, status, image_name, image_tag, created_at, updated_at)
|
ServiceID: rollback.ServiceID,
|
||||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)`,
|
Version: fmt.Sprintf("rollback-%d", now.Unix()),
|
||||||
rollback.ID, rollback.ServiceID, fmt.Sprintf("rollback-%d", now.Unix()), rollback.CommitHash, rollback.Status,
|
CommitHash: sql.NullString{String: ptrStr(rollback.CommitHash), Valid: rollback.CommitHash != nil},
|
||||||
rollback.ImageName, rollback.ImageTag, rollback.CreatedAt, rollback.UpdatedAt,
|
Status: sql.NullString{String: rollback.Status, Valid: true},
|
||||||
)
|
ImageName: sql.NullString{String: rollback.ImageName, Valid: true},
|
||||||
|
ImageTag: sql.NullString{String: rollback.ImageTag, Valid: true},
|
||||||
|
CreatedAt: sql.NullTime{Time: rollback.CreatedAt, Valid: true},
|
||||||
|
UpdatedAt: sql.NullTime{Time: rollback.UpdatedAt, Valid: true},
|
||||||
|
})
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create rollback deployment"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create rollback deployment"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = db.(*database.DB).Exec(
|
err = q.SetServiceStatus(ctx, sqlcdb.SetServiceStatusParams{
|
||||||
`UPDATE services SET status = 'building', updated_at = $1 WHERE id = $2`,
|
Status: sql.NullString{String: "building", Valid: true},
|
||||||
time.Now(), serviceID,
|
UpdatedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
)
|
ID: serviceID,
|
||||||
|
})
|
||||||
|
|
||||||
// Real rollback: redeploy the target deployment's image.
|
// Real rollback: redeploy the target deployment's image.
|
||||||
engineValue, _ := c.Get("deployment_engine")
|
engineValue, _ := c.Get("deployment_engine")
|
||||||
@@ -963,36 +973,18 @@ func handleRollbackDeployment(c *gin.Context) {
|
|||||||
if engine == nil {
|
if engine == nil {
|
||||||
reason = "Deployment engine unavailable. Docker may not be configured on this server."
|
reason = "Deployment engine unavailable. Docker may not be configured on this server."
|
||||||
}
|
}
|
||||||
_, _ = db.(*database.DB).Exec(
|
_ = q.FailDeployment(ctx, sqlcdb.FailDeploymentParams{
|
||||||
`UPDATE deployments SET status = 'failed', error = $1, completed_at = $2, updated_at = $2 WHERE id = $3`,
|
Error: sql.NullString{String: reason, Valid: true},
|
||||||
reason, completedAt, rollbackID,
|
CompletedAt: sql.NullTime{Time: completedAt, Valid: true},
|
||||||
)
|
ID: rollbackID,
|
||||||
_, _ = db.(*database.DB).Exec(
|
})
|
||||||
`UPDATE services SET status = 'failed', updated_at = $1 WHERE id = $2`,
|
_ = q.SetServiceStatus(ctx, sqlcdb.SetServiceStatusParams{
|
||||||
completedAt, serviceID,
|
Status: sql.NullString{String: "failed", Valid: true},
|
||||||
)
|
UpdatedAt: sql.NullTime{Time: completedAt, Valid: true},
|
||||||
|
ID: serviceID,
|
||||||
|
})
|
||||||
} else {
|
} else {
|
||||||
var service Service
|
service, _ := loadServiceForDeploy(db.(*database.DB), serviceID)
|
||||||
_ = db.(*database.DB).QueryRow(
|
|
||||||
`SELECT s.id, s.project_id, s.name, s.type, s.status, s.image, s.command,
|
|
||||||
s.environment, s.git_repo, s.git_branch, s.build_path, s.cpu, s.memory,
|
|
||||||
COALESCE(s.replicas, 1), COALESCE(s.port, 0),
|
|
||||||
COALESCE(s.domain, ''), COALESCE(s.healthcheck_path, ''),
|
|
||||||
COALESCE(s.restart_policy, 'unless-stopped'),
|
|
||||||
COALESCE(s.builder, 'auto'), COALESCE(s.cpu_reserve, ''),
|
|
||||||
COALESCE(s.memory_reserve, ''), COALESCE(s.static_build_cmd, ''),
|
|
||||||
COALESCE(s.static_dir, ''),
|
|
||||||
s.created_at, s.updated_at
|
|
||||||
FROM services s WHERE s.id = $1`, serviceID,
|
|
||||||
).Scan(
|
|
||||||
&service.ID, &service.ProjectID, &service.Name, &service.Type, &service.Status,
|
|
||||||
&service.Image, &service.Command, &service.Environment, &service.GitRepo,
|
|
||||||
&service.GitBranch, &service.BuildPath, &service.CPU, &service.Memory,
|
|
||||||
&service.Replicas, &service.Port, &service.Domain, &service.HealthCheckPath,
|
|
||||||
&service.RestartPolicy, &service.Builder, &service.CPUReserve,
|
|
||||||
&service.MemoryReserve, &service.StaticBuildCmd, &service.StaticDir,
|
|
||||||
&service.CreatedAt, &service.UpdatedAt,
|
|
||||||
)
|
|
||||||
|
|
||||||
rollbackReq := CreateDeploymentRequest{
|
rollbackReq := CreateDeploymentRequest{
|
||||||
CommitHash: func() string {
|
CommitHash: func() string {
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"regexp"
|
"regexp"
|
||||||
@@ -44,19 +45,26 @@ func validHostname(d string) bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func loadServiceDomains(db *database.DB, serviceID uuid.UUID) []ServiceDomain {
|
func loadServiceDomains(db *database.DB, serviceID uuid.UUID) []ServiceDomain {
|
||||||
rows, err := db.Query(
|
rows, err := sqlcdb.New(db.DB).ListServiceDomains(context.Background(), serviceID)
|
||||||
`SELECT id, service_id, domain, is_default, cert_type, cert_status, last_checked_at, created_at
|
|
||||||
FROM service_domains WHERE service_id = $1 ORDER BY is_default DESC, created_at ASC`, serviceID)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
out := make([]ServiceDomain, 0, len(rows))
|
||||||
out := []ServiceDomain{}
|
for _, r := range rows {
|
||||||
for rows.Next() {
|
d := ServiceDomain{
|
||||||
var d ServiceDomain
|
ID: r.ID,
|
||||||
if err := rows.Scan(&d.ID, &d.ServiceID, &d.Domain, &d.IsDefault, &d.CertType, &d.CertStatus, &d.LastCheckedAt, &d.CreatedAt); err == nil {
|
ServiceID: r.ServiceID,
|
||||||
out = append(out, d)
|
Domain: r.Domain,
|
||||||
|
IsDefault: r.IsDefault,
|
||||||
|
CertType: r.CertType,
|
||||||
|
CertStatus: r.CertStatus,
|
||||||
|
CreatedAt: r.CreatedAt,
|
||||||
}
|
}
|
||||||
|
if r.LastCheckedAt.Valid {
|
||||||
|
t := r.LastCheckedAt.Time
|
||||||
|
d.LastCheckedAt = &t
|
||||||
|
}
|
||||||
|
out = append(out, d)
|
||||||
}
|
}
|
||||||
return out
|
return out
|
||||||
}
|
}
|
||||||
@@ -104,22 +112,20 @@ func dnsLabel(name string) string {
|
|||||||
// base domain; spread services union every eligible node's base domain.
|
// base domain; spread services union every eligible node's base domain.
|
||||||
// Local services get nothing — there is no instance-level base domain yet.
|
// Local services get nothing — there is no instance-level base domain yet.
|
||||||
func serviceAutoDomains(db *database.DB, serviceID uuid.UUID) []string {
|
func serviceAutoDomains(db *database.DB, serviceID uuid.UUID) []string {
|
||||||
var name string
|
ctx := context.Background()
|
||||||
var nodeID sql.NullString
|
q := sqlcdb.New(db.DB)
|
||||||
var spread bool
|
brief, err := q.GetServicePlacementBrief(ctx, serviceID)
|
||||||
if err := db.QueryRow(
|
if err != nil {
|
||||||
`SELECT name, node_id, COALESCE(spread, false) FROM services WHERE id = $1`, serviceID,
|
|
||||||
).Scan(&name, &nodeID, &spread); err != nil {
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
label := dnsLabel(name)
|
label := dnsLabel(brief.Name)
|
||||||
if label == "" {
|
if label == "" {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
var bases []string
|
var bases []string
|
||||||
ctx := context.Background()
|
nodeID := brief.NodeID
|
||||||
q := sqlcdb.New(db.DB)
|
spread := brief.Spread
|
||||||
switch {
|
switch {
|
||||||
case nodeID.Valid && nodeID.String != "":
|
case nodeID.Valid && nodeID.String != "":
|
||||||
if a, err := q.GetAgent(ctx, nodeID.String); err == nil && a.DefaultDomain != "" {
|
if a, err := q.GetAgent(ctx, nodeID.String); err == nil && a.DefaultDomain != "" {
|
||||||
@@ -142,24 +148,25 @@ func serviceAutoDomains(db *database.DB, serviceID uuid.UUID) []string {
|
|||||||
// syncDefaultDomain writes the is_default row back to services.domain so
|
// syncDefaultDomain writes the is_default row back to services.domain so
|
||||||
// legacy reads (CLI tables, public_url) keep working.
|
// legacy reads (CLI tables, public_url) keep working.
|
||||||
func syncDefaultDomain(db *database.DB, serviceID uuid.UUID) {
|
func syncDefaultDomain(db *database.DB, serviceID uuid.UUID) {
|
||||||
var domain string
|
ctx := context.Background()
|
||||||
err := db.QueryRow(
|
q := sqlcdb.New(db.DB)
|
||||||
`SELECT domain FROM service_domains WHERE service_id = $1 AND is_default`, serviceID).Scan(&domain)
|
domain, err := q.GetServiceDomainDefault(ctx, serviceID)
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
err = db.QueryRow(
|
domain, err = q.GetServiceDomainFirst(ctx, serviceID)
|
||||||
`SELECT domain FROM service_domains WHERE service_id = $1 ORDER BY created_at ASC LIMIT 1`, serviceID).Scan(&domain)
|
|
||||||
}
|
}
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
domain = ""
|
domain = ""
|
||||||
}
|
}
|
||||||
_, _ = db.Exec(`UPDATE services SET domain = $1 WHERE id = $2`, domain, serviceID)
|
_ = q.SetServiceLegacyDomain(ctx, sqlcdb.SetServiceLegacyDomainParams{Domain: domain, ID: serviceID})
|
||||||
}
|
}
|
||||||
|
|
||||||
// serviceAccess loads the maintenance/basic-auth columns for the runtime spec.
|
// serviceAccess loads the maintenance/basic-auth columns for the runtime spec.
|
||||||
func serviceAccess(db *database.DB, serviceID uuid.UUID) (maintenance bool, basicAuth string) {
|
func serviceAccess(db *database.DB, serviceID uuid.UUID) (maintenance bool, basicAuth string) {
|
||||||
_ = db.QueryRow(`SELECT maintenance_mode, basic_auth_users FROM services WHERE id = $1`, serviceID).
|
row, err := sqlcdb.New(db.DB).GetServiceAccess(context.Background(), serviceID)
|
||||||
Scan(&maintenance, &basicAuth)
|
if err != nil {
|
||||||
return maintenance, basicAuth
|
return false, ""
|
||||||
|
}
|
||||||
|
return row.MaintenanceMode, row.BasicAuthUsers
|
||||||
}
|
}
|
||||||
|
|
||||||
// maintenanceURL is the redirect target for maintenance mode — the public
|
// maintenanceURL is the redirect target for maintenance mode — the public
|
||||||
@@ -204,11 +211,12 @@ func basicAuthUsernames(encoded string) []string {
|
|||||||
// serviceWriteAccess mirrors the service update handler's access check.
|
// serviceWriteAccess mirrors the service update handler's access check.
|
||||||
func serviceWriteAccess(c *gin.Context, db *database.DB, serviceID uuid.UUID) bool {
|
func serviceWriteAccess(c *gin.Context, db *database.DB, serviceID uuid.UUID) bool {
|
||||||
userID, _ := c.Get("user_id")
|
userID, _ := c.Get("user_id")
|
||||||
var ok bool
|
uid, _ := uuid.Parse(fmt.Sprint(userID))
|
||||||
err := db.QueryRow(
|
ok, err := sqlcdb.New(db.DB).ServiceWriteAccess(context.Background(), sqlcdb.ServiceWriteAccessParams{
|
||||||
`SELECT EXISTS(SELECT 1 FROM services s JOIN projects p ON s.project_id = p.id
|
ID: serviceID,
|
||||||
WHERE s.id = $1 AND (p.owner_id = $2 OR $3::bool))`,
|
OwnerID: uid,
|
||||||
serviceID, userID, contextIsAdmin(c)).Scan(&ok)
|
Column3: contextIsAdmin(c),
|
||||||
|
})
|
||||||
return err == nil && ok
|
return err == nil && ok
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -255,9 +263,11 @@ func handleAddServiceDomain(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ctx := c.Request.Context()
|
||||||
|
q := sqlcdb.New(db.(*database.DB).DB)
|
||||||
|
|
||||||
// First domain on a service becomes the default automatically.
|
// First domain on a service becomes the default automatically.
|
||||||
var count int
|
count, _ := q.CountServiceDomains(ctx, serviceID)
|
||||||
_ = db.(*database.DB).QueryRow(`SELECT COUNT(*) FROM service_domains WHERE service_id = $1`, serviceID).Scan(&count)
|
|
||||||
makeDefault := req.IsDefault || count == 0
|
makeDefault := req.IsDefault || count == 0
|
||||||
|
|
||||||
tx, err := db.(*database.DB).Begin()
|
tx, err := db.(*database.DB).Begin()
|
||||||
@@ -266,19 +276,18 @@ func handleAddServiceDomain(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer tx.Rollback()
|
defer tx.Rollback()
|
||||||
|
qtx := q.WithTx(tx)
|
||||||
if makeDefault {
|
if makeDefault {
|
||||||
if _, err := tx.Exec(`UPDATE service_domains SET is_default = false WHERE service_id = $1`, serviceID); err != nil {
|
if err := qtx.ClearServiceDomainDefault(ctx, serviceID); err != nil {
|
||||||
respondError(c, http.StatusInternalServerError, "INTERNAL", "Failed to add domain")
|
respondError(c, http.StatusInternalServerError, "INTERNAL", "Failed to add domain")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
var inserted ServiceDomain
|
row, err := qtx.CreateServiceDomain(ctx, sqlcdb.CreateServiceDomainParams{
|
||||||
err = tx.QueryRow(
|
ServiceID: serviceID,
|
||||||
`INSERT INTO service_domains (service_id, domain, is_default)
|
Domain: domain,
|
||||||
VALUES ($1, $2, $3)
|
IsDefault: makeDefault,
|
||||||
RETURNING id, service_id, domain, is_default, cert_type, cert_status, created_at`,
|
})
|
||||||
serviceID, domain, makeDefault).
|
|
||||||
Scan(&inserted.ID, &inserted.ServiceID, &inserted.Domain, &inserted.IsDefault, &inserted.CertType, &inserted.CertStatus, &inserted.CreatedAt)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if strings.Contains(err.Error(), "duplicate") || strings.Contains(err.Error(), "unique") {
|
if strings.Contains(err.Error(), "duplicate") || strings.Contains(err.Error(), "unique") {
|
||||||
respondError(c, http.StatusConflict, "CONFLICT", "Domain already attached to this service")
|
respondError(c, http.StatusConflict, "CONFLICT", "Domain already attached to this service")
|
||||||
@@ -292,7 +301,15 @@ func handleAddServiceDomain(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
syncDefaultDomain(db.(*database.DB), serviceID)
|
syncDefaultDomain(db.(*database.DB), serviceID)
|
||||||
c.JSON(http.StatusCreated, gin.H{"domain": inserted})
|
c.JSON(http.StatusCreated, gin.H{"domain": ServiceDomain{
|
||||||
|
ID: row.ID,
|
||||||
|
ServiceID: row.ServiceID,
|
||||||
|
Domain: row.Domain,
|
||||||
|
IsDefault: row.IsDefault,
|
||||||
|
CertType: row.CertType,
|
||||||
|
CertStatus: row.CertStatus,
|
||||||
|
CreatedAt: row.CreatedAt,
|
||||||
|
}})
|
||||||
}
|
}
|
||||||
|
|
||||||
func handleDeleteServiceDomain(c *gin.Context) {
|
func handleDeleteServiceDomain(c *gin.Context) {
|
||||||
@@ -315,9 +332,8 @@ func handleDeleteServiceDomain(c *gin.Context) {
|
|||||||
respondError(c, http.StatusBadRequest, "VALIDATION", "Invalid domain ID")
|
respondError(c, http.StatusBadRequest, "VALIDATION", "Invalid domain ID")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
res, err := db.(*database.DB).Exec(
|
affected, err := sqlcdb.New(db.(*database.DB).DB).DeleteServiceDomain(
|
||||||
`DELETE FROM service_domains WHERE id = $1 AND service_id = $2`, domainID, serviceID)
|
c.Request.Context(), sqlcdb.DeleteServiceDomainParams{ID: domainID, ServiceID: serviceID})
|
||||||
affected, _ := res.RowsAffected()
|
|
||||||
if err != nil || affected == 0 {
|
if err != nil || affected == 0 {
|
||||||
respondError(c, http.StatusNotFound, "NOT_FOUND", "Domain not found")
|
respondError(c, http.StatusNotFound, "NOT_FOUND", "Domain not found")
|
||||||
return
|
return
|
||||||
@@ -346,14 +362,12 @@ func handleSetDefaultDomain(c *gin.Context) {
|
|||||||
respondError(c, http.StatusBadRequest, "VALIDATION", "Invalid domain ID")
|
respondError(c, http.StatusBadRequest, "VALIDATION", "Invalid domain ID")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
res, err := db.(*database.DB).Exec(
|
affected, err := sqlcdb.New(db.(*database.DB).DB).SetServiceDomainDefault(
|
||||||
`UPDATE service_domains SET is_default = (id = $1) WHERE service_id = $2`,
|
c.Request.Context(), sqlcdb.SetServiceDomainDefaultParams{ID: domainID, ServiceID: serviceID})
|
||||||
domainID, serviceID)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
respondError(c, http.StatusInternalServerError, "INTERNAL", "Failed to set default domain")
|
respondError(c, http.StatusInternalServerError, "INTERNAL", "Failed to set default domain")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
affected, _ := res.RowsAffected()
|
|
||||||
if affected == 0 {
|
if affected == 0 {
|
||||||
respondError(c, http.StatusNotFound, "NOT_FOUND", "Domain not found")
|
respondError(c, http.StatusNotFound, "NOT_FOUND", "Domain not found")
|
||||||
return
|
return
|
||||||
@@ -421,7 +435,7 @@ func handleCheckServiceDomains(c *gin.Context) {
|
|||||||
r.Status = "ok"
|
r.Status = "ok"
|
||||||
}
|
}
|
||||||
results = append(results, r)
|
results = append(results, r)
|
||||||
_, _ = dbh.Exec(`UPDATE service_domains SET last_checked_at = NOW() WHERE id = $1`, d.ID)
|
_ = sqlcdb.New(dbh.DB).TouchServiceDomainChecked(c.Request.Context(), d.ID)
|
||||||
}
|
}
|
||||||
c.JSON(http.StatusOK, gin.H{"domains": results})
|
c.JSON(http.StatusOK, gin.H{"domains": results})
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,30 @@
|
|||||||
package api
|
package api
|
||||||
|
|
||||||
import "testing"
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"containr/internal/database"
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
func testDSNString(t *testing.T) string {
|
||||||
|
t.Helper()
|
||||||
|
dsn := os.Getenv("METRICS_TEST_DSN")
|
||||||
|
if dsn == "" {
|
||||||
|
dsn = os.Getenv("DATABASE_URL")
|
||||||
|
}
|
||||||
|
if dsn == "" {
|
||||||
|
t.Skip("METRICS_TEST_DSN not set")
|
||||||
|
}
|
||||||
|
return dsn
|
||||||
|
}
|
||||||
|
|
||||||
func TestDNSLabel(t *testing.T) {
|
func TestDNSLabel(t *testing.T) {
|
||||||
cases := map[string]string{
|
cases := map[string]string{
|
||||||
@@ -24,3 +48,554 @@ func TestDNSLabelLengthCap(t *testing.T) {
|
|||||||
t.Fatalf("label over 63 chars: %d", len(got))
|
t.Fatalf("label over 63 chars: %d", len(got))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Live invariant test against real Postgres — set METRICS_TEST_DSN (or
|
||||||
|
// DATABASE_URL). Verifies the generated domain queries preserve the
|
||||||
|
// default-domain and legacy services.domain invariants.
|
||||||
|
func TestServiceDomainsLiveInvariants(t *testing.T) {
|
||||||
|
dsn := testDSNString(t)
|
||||||
|
db, err := database.NewConnection(dsn)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("connect: %v", err)
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
ctx := context.Background()
|
||||||
|
q := sqlcdb.New(db.DB)
|
||||||
|
|
||||||
|
userID := uuid.New()
|
||||||
|
projectID := uuid.New()
|
||||||
|
envID := uuid.New()
|
||||||
|
serviceID := uuid.New()
|
||||||
|
suffix := strings.ReplaceAll(serviceID.String()[:8], "-", "")
|
||||||
|
|
||||||
|
must := func(err error) {
|
||||||
|
t.Helper()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("setup: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = db.Exec(`INSERT INTO users (id, email, password_hash, name) VALUES ($1, $2, 'x', 'test')`,
|
||||||
|
userID, "domtest-"+suffix+"@test.local")
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO projects (id, name, owner_id, is_approved) VALUES ($1, $2, $3, true)`,
|
||||||
|
projectID, "domtest-"+suffix, userID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO environments (id, name, project_id) VALUES ($1, 'production', $2)`,
|
||||||
|
envID, projectID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO services (id, name, project_id, environment_id, service_type, source_type)
|
||||||
|
VALUES ($1, 'domtest-`+suffix+`', $2, $3, 'web', 'image')`,
|
||||||
|
serviceID, projectID, envID)
|
||||||
|
must(err)
|
||||||
|
defer func() {
|
||||||
|
_, _ = db.Exec(`DELETE FROM users WHERE id = $1`, userID)
|
||||||
|
}()
|
||||||
|
|
||||||
|
d1, err := q.CreateServiceDomain(ctx, sqlcdb.CreateServiceDomainParams{
|
||||||
|
ServiceID: serviceID, Domain: "a-" + suffix + ".example.com", IsDefault: true})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("create d1: %v", err)
|
||||||
|
}
|
||||||
|
d2, err := q.CreateServiceDomain(ctx, sqlcdb.CreateServiceDomainParams{
|
||||||
|
ServiceID: serviceID, Domain: "b-" + suffix + ".example.com", IsDefault: false})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("create d2: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
syncDefaultDomain(db, serviceID)
|
||||||
|
var legacy string
|
||||||
|
if err := db.QueryRow(`SELECT domain FROM services WHERE id = $1`, serviceID).Scan(&legacy); err != nil {
|
||||||
|
t.Fatalf("read legacy: %v", err)
|
||||||
|
}
|
||||||
|
if legacy != d1.Domain {
|
||||||
|
t.Fatalf("legacy domain = %q, want %q", legacy, d1.Domain)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Switch default: exactly one row stays default, legacy follows.
|
||||||
|
if err := q.ClearServiceDomainDefault(ctx, serviceID); err != nil {
|
||||||
|
t.Fatalf("clear: %v", err)
|
||||||
|
}
|
||||||
|
if _, err := q.SetServiceDomainDefault(ctx, sqlcdb.SetServiceDomainDefaultParams{
|
||||||
|
ID: d2.ID, ServiceID: serviceID}); err != nil {
|
||||||
|
t.Fatalf("set default: %v", err)
|
||||||
|
}
|
||||||
|
syncDefaultDomain(db, serviceID)
|
||||||
|
if err := db.QueryRow(`SELECT domain FROM services WHERE id = $1`, serviceID).Scan(&legacy); err != nil {
|
||||||
|
t.Fatalf("read legacy2: %v", err)
|
||||||
|
}
|
||||||
|
if legacy != d2.Domain {
|
||||||
|
t.Fatalf("legacy after switch = %q, want %q", legacy, d2.Domain)
|
||||||
|
}
|
||||||
|
|
||||||
|
doms := loadServiceDomains(db, serviceID)
|
||||||
|
if len(doms) != 2 || !doms[0].IsDefault || doms[0].Domain != d2.Domain {
|
||||||
|
t.Fatalf("list order wrong: %+v", doms)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deleting the default falls back to the remaining domain.
|
||||||
|
if _, err := q.DeleteServiceDomain(ctx, sqlcdb.DeleteServiceDomainParams{
|
||||||
|
ID: d2.ID, ServiceID: serviceID}); err != nil {
|
||||||
|
t.Fatalf("delete d2: %v", err)
|
||||||
|
}
|
||||||
|
syncDefaultDomain(db, serviceID)
|
||||||
|
if err := db.QueryRow(`SELECT domain FROM services WHERE id = $1`, serviceID).Scan(&legacy); err != nil {
|
||||||
|
t.Fatalf("read legacy3: %v", err)
|
||||||
|
}
|
||||||
|
if legacy != d1.Domain {
|
||||||
|
t.Fatalf("legacy after delete = %q, want %q", legacy, d1.Domain)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Live clone test: cloneServiceRow must copy config, domains, and variables
|
||||||
|
// verbatim, start stopped, and land in the target project/environment.
|
||||||
|
func TestCloneServiceRowLive(t *testing.T) {
|
||||||
|
dsn := testDSNString(t)
|
||||||
|
db, err := database.NewConnection(dsn)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("connect: %v", err)
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
ctx := context.Background()
|
||||||
|
q := sqlcdb.New(db.DB)
|
||||||
|
|
||||||
|
userID := uuid.New()
|
||||||
|
projectID := uuid.New()
|
||||||
|
envID := uuid.New()
|
||||||
|
serviceID := uuid.New()
|
||||||
|
suffix := strings.ReplaceAll(serviceID.String()[:8], "-", "")
|
||||||
|
must := func(err error) {
|
||||||
|
t.Helper()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("setup: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = db.Exec(`INSERT INTO users (id, email, password_hash, name) VALUES ($1, $2, 'x', 'test')`,
|
||||||
|
userID, "clone-"+suffix+"@test.local")
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO projects (id, name, owner_id, is_approved) VALUES ($1, $2, $3, true)`,
|
||||||
|
projectID, "clone-"+suffix, userID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO environments (id, name, project_id) VALUES ($1, 'production', $2)`,
|
||||||
|
envID, projectID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO services (id, name, project_id, environment_id, service_type, source_type,
|
||||||
|
image_name, port, volumes, traefik_labels, maintenance_mode)
|
||||||
|
VALUES ($1, $2, $3, $4, 'web', 'image', 'traefik/whoami:latest', 80,
|
||||||
|
'[{"mount":"/data"}]'::jsonb,
|
||||||
|
'{"middlewares.rl.ratelimit.average":"100"}'::jsonb, true)`,
|
||||||
|
serviceID, "clonesrc-"+suffix, projectID, envID)
|
||||||
|
must(err)
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM users WHERE id = $1`, userID) }()
|
||||||
|
|
||||||
|
if _, err := q.CreateServiceDomain(ctx, sqlcdb.CreateServiceDomainParams{
|
||||||
|
ServiceID: serviceID, Domain: "src-" + suffix + ".example.com", IsDefault: true}); err != nil {
|
||||||
|
t.Fatalf("seed domain: %v", err)
|
||||||
|
}
|
||||||
|
if _, err := db.Exec(`INSERT INTO environment_variables (id, service_id, key, value, is_secret)
|
||||||
|
VALUES ($1, $2, 'FOO', 'bar', false)`, uuid.New(), serviceID); err != nil {
|
||||||
|
t.Fatalf("seed var: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cloneID, err := cloneServiceRow(db, serviceID, projectID, "clone-"+suffix, "")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("clone: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var status, imageName, env string
|
||||||
|
var maintenance bool
|
||||||
|
var volumes, labels []byte
|
||||||
|
if err := db.QueryRow(`SELECT status, image_name, COALESCE(environment,''), maintenance_mode,
|
||||||
|
volumes, traefik_labels FROM services WHERE id = $1`, cloneID).
|
||||||
|
Scan(&status, &imageName, &env, &maintenance, &volumes, &labels); err != nil {
|
||||||
|
t.Fatalf("read clone: %v", err)
|
||||||
|
}
|
||||||
|
if status != "stopped" || imageName != "traefik/whoami:latest" || env != "production" {
|
||||||
|
t.Fatalf("clone mismatch: %v %v %v", status, imageName, env)
|
||||||
|
}
|
||||||
|
if !maintenance || string(volumes) == "[]" || string(labels) == "{}" {
|
||||||
|
t.Fatalf("clone dropped config: maint=%v vol=%s labels=%s", maintenance, volumes, labels)
|
||||||
|
}
|
||||||
|
doms := loadServiceDomains(db, cloneID)
|
||||||
|
if len(doms) != 1 || doms[0].Domain != "src-"+suffix+".example.com" || !doms[0].IsDefault {
|
||||||
|
t.Fatalf("domains not cloned: %+v", doms)
|
||||||
|
}
|
||||||
|
var varVal string
|
||||||
|
if err := db.QueryRow(`SELECT value FROM environment_variables WHERE service_id = $1 AND key = 'FOO'`,
|
||||||
|
cloneID).Scan(&varVal); err != nil || varVal != "bar" {
|
||||||
|
t.Fatalf("vars not cloned: %v %q", err, varVal)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Live preview-environment test: generated queries must preserve the
|
||||||
|
// access-check join, the status-sync CASE, and the sweep listing.
|
||||||
|
func TestPreviewEnvironmentsLiveInvariants(t *testing.T) {
|
||||||
|
dsn := testDSNString(t)
|
||||||
|
db, err := database.NewConnection(dsn)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("connect: %v", err)
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
ctx := context.Background()
|
||||||
|
q := sqlcdb.New(db.DB)
|
||||||
|
|
||||||
|
userID := uuid.New()
|
||||||
|
projectID := uuid.New()
|
||||||
|
envID := uuid.New()
|
||||||
|
srcID := uuid.New()
|
||||||
|
cloneID := uuid.New()
|
||||||
|
previewID := uuid.New()
|
||||||
|
suffix := strings.ReplaceAll(previewID.String()[:8], "-", "")
|
||||||
|
must := func(err error) {
|
||||||
|
t.Helper()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("setup: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = db.Exec(`INSERT INTO users (id, email, password_hash, name) VALUES ($1, $2, 'x', 'test')`,
|
||||||
|
userID, "prev-"+suffix+"@test.local")
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO projects (id, name, owner_id, is_approved) VALUES ($1, $2, $3, true)`,
|
||||||
|
projectID, "prev-"+suffix, userID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO environments (id, name, project_id) VALUES ($1, 'production', $2)`,
|
||||||
|
envID, projectID)
|
||||||
|
must(err)
|
||||||
|
for _, p := range [][3]interface{}{{srcID, "src-" + suffix, "running"}, {cloneID, "clone-" + suffix, "running"}} {
|
||||||
|
_, err = db.Exec(`INSERT INTO services (id, name, project_id, environment_id, service_type, source_type, status)
|
||||||
|
VALUES ($1, $2, $3, $4, 'web', 'image', $5)`,
|
||||||
|
p[0], p[1], projectID, envID, p[2])
|
||||||
|
must(err)
|
||||||
|
}
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM users WHERE id = $1`, userID) }()
|
||||||
|
|
||||||
|
must(q.CreatePreviewEnvironment(ctx, sqlcdb.CreatePreviewEnvironmentParams{
|
||||||
|
ID: previewID,
|
||||||
|
ProjectID: projectID,
|
||||||
|
ServiceID: srcID,
|
||||||
|
PreviewServiceID: uuid.NullUUID{UUID: cloneID, Valid: true},
|
||||||
|
BranchName: "feat-" + suffix,
|
||||||
|
Environment: "preview-feat-" + suffix,
|
||||||
|
Status: "building",
|
||||||
|
ExpiresAt: time.Now().Add(-time.Hour), // already past TTL → sweepable
|
||||||
|
}))
|
||||||
|
|
||||||
|
// List + service join.
|
||||||
|
rows, err := q.ListPreviewEnvironmentsForProject(ctx, projectID)
|
||||||
|
if err != nil || len(rows) != 1 {
|
||||||
|
t.Fatalf("list: %v %d", err, len(rows))
|
||||||
|
}
|
||||||
|
if rows[0].ID != previewID || !rows[0].SvcID.Valid || rows[0].ServiceName.String != "src-"+suffix {
|
||||||
|
t.Fatalf("list row: %+v", rows[0])
|
||||||
|
}
|
||||||
|
|
||||||
|
// Access-check join: owner reads; a stranger does not.
|
||||||
|
if _, err := q.GetPreviewEnvironment(ctx, sqlcdb.GetPreviewEnvironmentParams{
|
||||||
|
ID: previewID, OwnerID: userID}); err != nil {
|
||||||
|
t.Fatalf("owner read: %v", err)
|
||||||
|
}
|
||||||
|
if _, err := q.GetPreviewEnvironment(ctx, sqlcdb.GetPreviewEnvironmentParams{
|
||||||
|
ID: previewID, OwnerID: uuid.New()}); err != nil {
|
||||||
|
t.Fatalf("approved project should be public-read: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Status sync folds clone's 'running' into the preview row.
|
||||||
|
if err := q.SyncPreviewStatuses(ctx); err != nil {
|
||||||
|
t.Fatalf("sync: %v", err)
|
||||||
|
}
|
||||||
|
var status string
|
||||||
|
if err := db.QueryRow(`SELECT status FROM preview_environments WHERE id = $1`, previewID).Scan(&status); err != nil {
|
||||||
|
t.Fatalf("read status: %v", err)
|
||||||
|
}
|
||||||
|
if status != "running" {
|
||||||
|
t.Fatalf("status after sync = %q, want running", status)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TTL sweep picks it up (expired at insert), marks expired.
|
||||||
|
sweep, err := q.ListSweepablePreviews(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("sweep list: %v", err)
|
||||||
|
}
|
||||||
|
found := false
|
||||||
|
for _, s := range sweep {
|
||||||
|
if s.ID == previewID && s.PreviewServiceID.UUID == cloneID {
|
||||||
|
found = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !found {
|
||||||
|
t.Fatalf("sweepable list missing preview: %+v", sweep)
|
||||||
|
}
|
||||||
|
must(q.MarkPreviewEnvironmentStatus(ctx, sqlcdb.MarkPreviewEnvironmentStatusParams{Status: "expired", ID: previewID}))
|
||||||
|
sweep, _ = q.ListSweepablePreviews(ctx)
|
||||||
|
for _, s := range sweep {
|
||||||
|
if s.ID == previewID {
|
||||||
|
t.Fatal("expired preview still sweepable")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Owner lookup and delete.
|
||||||
|
owner, err := q.GetPreviewEnvironmentOwner(ctx, previewID)
|
||||||
|
if err != nil || owner != userID {
|
||||||
|
t.Fatalf("owner: %v %v", owner, err)
|
||||||
|
}
|
||||||
|
must(q.DeletePreviewEnvironment(ctx, previewID))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Live deployment lifecycle: generated queries preserve list ordering,
|
||||||
|
// the access join, and the status-transition writes.
|
||||||
|
func TestDeploymentsLiveInvariants(t *testing.T) {
|
||||||
|
dsn := testDSNString(t)
|
||||||
|
db, err := database.NewConnection(dsn)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("connect: %v", err)
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
ctx := context.Background()
|
||||||
|
q := sqlcdb.New(db.DB)
|
||||||
|
|
||||||
|
userID := uuid.New()
|
||||||
|
projectID := uuid.New()
|
||||||
|
envID := uuid.New()
|
||||||
|
serviceID := uuid.New()
|
||||||
|
deployID := uuid.New()
|
||||||
|
suffix := strings.ReplaceAll(deployID.String()[:8], "-", "")
|
||||||
|
must := func(err error) {
|
||||||
|
t.Helper()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("setup: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = db.Exec(`INSERT INTO users (id, email, password_hash, name) VALUES ($1, $2, 'x', 'test')`,
|
||||||
|
userID, "dep-"+suffix+"@test.local")
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO projects (id, name, owner_id, is_approved) VALUES ($1, $2, $3, true)`,
|
||||||
|
projectID, "dep-"+suffix, userID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO environments (id, name, project_id) VALUES ($1, 'production', $2)`,
|
||||||
|
envID, projectID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO services (id, name, project_id, environment_id, service_type, source_type, status)
|
||||||
|
VALUES ($1, $2, $3, $4, 'web', 'image', 'running')`,
|
||||||
|
serviceID, "dep-"+suffix, projectID, envID)
|
||||||
|
must(err)
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM users WHERE id = $1`, userID) }()
|
||||||
|
|
||||||
|
must(q.InsertDeployment(ctx, sqlcdb.InsertDeploymentParams{
|
||||||
|
ID: deployID,
|
||||||
|
ServiceID: serviceID,
|
||||||
|
Version: "v1-" + suffix,
|
||||||
|
Status: sql.NullString{String: "pending", Valid: true},
|
||||||
|
}))
|
||||||
|
total, err := q.CountDeploymentsForService(ctx, serviceID)
|
||||||
|
if err != nil || total != 1 {
|
||||||
|
t.Fatalf("count: %v %d", err, total)
|
||||||
|
}
|
||||||
|
rows, err := q.ListDeploymentsForService(ctx, sqlcdb.ListDeploymentsForServiceParams{
|
||||||
|
ServiceID: serviceID, Limit: 10, Offset: 0})
|
||||||
|
if err != nil || len(rows) != 1 || rows[0].ID != deployID {
|
||||||
|
t.Fatalf("list: %v %+v", err, rows)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Access join returns owner.
|
||||||
|
access, err := q.GetDeploymentAccess(ctx, deployID)
|
||||||
|
if err != nil || access.OwnerID != userID || access.ServiceID != serviceID {
|
||||||
|
t.Fatalf("access: %v %+v", err, access)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Recent feed honors the access predicate.
|
||||||
|
feed, err := q.ListRecentAccessibleDeployments(ctx, sqlcdb.ListRecentAccessibleDeploymentsParams{
|
||||||
|
OwnerID: userID, Limit: 10, Offset: 0})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("feed: %v", err)
|
||||||
|
}
|
||||||
|
found := false
|
||||||
|
for _, d := range feed {
|
||||||
|
if d.ID == deployID && d.ServiceName == "dep-"+suffix {
|
||||||
|
found = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !found {
|
||||||
|
t.Fatalf("feed missing deployment: %+v", feed)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Status transitions.
|
||||||
|
must(q.SetDeploymentStatus(ctx, sqlcdb.SetDeploymentStatusParams{
|
||||||
|
Status: sql.NullString{String: "queued", Valid: true}, UpdatedAt: sql.NullTime{Time: time.Now(), Valid: true}, ID: deployID}))
|
||||||
|
must(q.SyncDeploymentProgress(ctx, sqlcdb.SyncDeploymentProgressParams{
|
||||||
|
Status: sql.NullString{String: "deployed", Valid: true},
|
||||||
|
ImageName: sql.NullString{String: "nginx", Valid: true},
|
||||||
|
ImageTag: sql.NullString{String: "alpine", Valid: true},
|
||||||
|
UpdatedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
|
ID: deployID,
|
||||||
|
}))
|
||||||
|
r, err := q.GetDeploymentWithProject(ctx, deployID)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("get: %v", err)
|
||||||
|
}
|
||||||
|
if r.Status.String != "deployed" || r.ImageName.String != "nginx" || r.ProjectID != projectID {
|
||||||
|
t.Fatalf("post-sync row: %+v", r)
|
||||||
|
}
|
||||||
|
must(q.FailDeployment(ctx, sqlcdb.FailDeploymentParams{
|
||||||
|
Error: sql.NullString{String: "boom", Valid: true},
|
||||||
|
CompletedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
|
ID: deployID,
|
||||||
|
}))
|
||||||
|
r, _ = q.GetDeploymentWithProject(ctx, deployID)
|
||||||
|
if r.Status.String != "failed" || r.Error.String != "boom" {
|
||||||
|
t.Fatalf("post-fail row: %+v", r)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestRuntimeLiveInvariants exercises the runtime.sql queries: the
|
||||||
|
// legacy-column COALESCE chain in GetServiceRuntimeWithOwner, the
|
||||||
|
// deployed-image concat, and the conditional published-port update.
|
||||||
|
func TestRuntimeLiveInvariants(t *testing.T) {
|
||||||
|
dsn := testDSNString(t)
|
||||||
|
db, err := database.NewConnection(dsn)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("connect: %v", err)
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
ctx := context.Background()
|
||||||
|
q := sqlcdb.New(db.DB)
|
||||||
|
|
||||||
|
userID := uuid.New()
|
||||||
|
projectID := uuid.New()
|
||||||
|
envID := uuid.New()
|
||||||
|
serviceID := uuid.New()
|
||||||
|
suffix := strings.ReplaceAll(serviceID.String()[:8], "-", "")
|
||||||
|
must := func(err error) {
|
||||||
|
t.Helper()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("setup: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = db.Exec(`INSERT INTO users (id, email, password_hash, name) VALUES ($1, $2, 'x', 'test')`,
|
||||||
|
userID, "rt-"+suffix+"@test.local")
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO projects (id, name, owner_id, is_approved) VALUES ($1, $2, $3, true)`,
|
||||||
|
projectID, "rt-"+suffix, userID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO environments (id, name, project_id) VALUES ($1, 'production', $2)`,
|
||||||
|
envID, projectID)
|
||||||
|
must(err)
|
||||||
|
// Only legacy columns populated — the COALESCE chain must surface them.
|
||||||
|
_, err = db.Exec(`INSERT INTO services (id, name, project_id, environment_id, service_type, source_type,
|
||||||
|
source_url, image_name, start_command, port) VALUES ($1, $2, $3, $4, 'worker', 'git', $5, $6, 'run.sh', 8080)`,
|
||||||
|
serviceID, "rt-"+suffix, projectID, envID, "https://git.example/x.git", "repo/img:latest")
|
||||||
|
must(err)
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM users WHERE id = $1`, userID) }()
|
||||||
|
|
||||||
|
row, err := q.GetServiceRuntimeWithOwner(ctx, serviceID)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("runtime row: %v", err)
|
||||||
|
}
|
||||||
|
if row.Type != "worker" || row.Image != "repo/img:latest" || row.Command != "run.sh" ||
|
||||||
|
row.GitRepo != "https://git.example/x.git" || row.OwnerID != userID {
|
||||||
|
t.Fatalf("COALESCE chain lost legacy columns: %+v", row)
|
||||||
|
}
|
||||||
|
svc := serviceFromRuntimeRow(row)
|
||||||
|
if svc.Type != "worker" || svc.Image != "repo/img:latest" || svc.Port != 8080 {
|
||||||
|
t.Fatalf("mapped service wrong: %+v", svc)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sibling refs + port lookup.
|
||||||
|
refs, err := q.ListProjectServiceRefs(ctx, projectID)
|
||||||
|
if err != nil || len(refs) != 1 || refs[0].Name != "rt-"+suffix {
|
||||||
|
t.Fatalf("refs: %v %+v", err, refs)
|
||||||
|
}
|
||||||
|
port, err := q.GetServicePort(ctx, serviceID)
|
||||||
|
if err != nil || port != 8080 {
|
||||||
|
t.Fatalf("port: %v %d", err, port)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Variables round-trip.
|
||||||
|
_, err = db.Exec(`INSERT INTO environment_variables (id, service_id, key, value, is_secret)
|
||||||
|
VALUES (gen_random_uuid(), $1, 'API_KEY', 'shh', true), (gen_random_uuid(), $1, 'PLAIN', 'p', false)`,
|
||||||
|
serviceID)
|
||||||
|
must(err)
|
||||||
|
vars, err := q.ListServiceVariablesWithSecret(ctx, serviceID)
|
||||||
|
if err != nil || len(vars) != 2 {
|
||||||
|
t.Fatalf("vars: %v %+v", err, vars)
|
||||||
|
}
|
||||||
|
vals, err := q.ListServiceVariableValues(ctx, serviceID)
|
||||||
|
if err != nil || len(vals) != 2 {
|
||||||
|
t.Fatalf("vals: %v %+v", err, vals)
|
||||||
|
}
|
||||||
|
got, err := q.GetServiceVariableValue(ctx, sqlcdb.GetServiceVariableValueParams{ServiceID: serviceID, Key: "API_KEY"})
|
||||||
|
if err != nil || got != "shh" {
|
||||||
|
t.Fatalf("var value: %v %q", err, got)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Node agent listing through container_instances.
|
||||||
|
_, err = db.Exec(`INSERT INTO node_agents (id, name, hostname, ip_address, port) VALUES ($1, $2, 'h', '127.0.0.1', 9999)`,
|
||||||
|
"agent-"+suffix, "agent-"+suffix)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO container_instances (id, name, image, project_id, service_id, node_agent_id)
|
||||||
|
VALUES ($1, 'c', 'img', $2, $3, $4)`,
|
||||||
|
"ci-"+suffix, projectID.String(), serviceID.String(), "agent-"+suffix)
|
||||||
|
must(err)
|
||||||
|
agents, err := q.ListNodeAgentsForService(ctx, serviceID.String())
|
||||||
|
if err != nil || len(agents) != 1 || agents[0] != "agent-"+suffix {
|
||||||
|
t.Fatalf("agents: %v %+v", err, agents)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Last deployed image: only 'deployed' rows, latest first.
|
||||||
|
d1, d2 := uuid.New(), uuid.New()
|
||||||
|
must(q.InsertDeployment(ctx, sqlcdb.InsertDeploymentParams{
|
||||||
|
ID: d1, ServiceID: serviceID, Version: "v1",
|
||||||
|
Status: sql.NullString{String: "failed", Valid: true},
|
||||||
|
}))
|
||||||
|
must(q.InsertDeployment(ctx, sqlcdb.InsertDeploymentParams{
|
||||||
|
ID: d2, ServiceID: serviceID, Version: "v2",
|
||||||
|
Status: sql.NullString{String: "deployed", Valid: true},
|
||||||
|
}))
|
||||||
|
must(q.SyncDeploymentProgress(ctx, sqlcdb.SyncDeploymentProgressParams{
|
||||||
|
Status: sql.NullString{String: "deployed", Valid: true},
|
||||||
|
ImageName: sql.NullString{String: "app", Valid: true},
|
||||||
|
ImageTag: sql.NullString{String: "v2", Valid: true},
|
||||||
|
UpdatedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
|
ID: d2,
|
||||||
|
}))
|
||||||
|
img, err := q.GetLastDeployedImage(ctx, serviceID)
|
||||||
|
if err != nil || img != "app:v2" {
|
||||||
|
t.Fatalf("last image: %v %q", err, img)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Conditional published-port update: writes once, no-ops on repeat.
|
||||||
|
must(q.SetServicePublishedPort(ctx, sqlcdb.SetServicePublishedPortParams{PublishedPort: 30001, ID: serviceID}))
|
||||||
|
p, err := q.GetServicePublishedPort(ctx, serviceID)
|
||||||
|
if err != nil || p != 30001 {
|
||||||
|
t.Fatalf("published port: %v %d", err, p)
|
||||||
|
}
|
||||||
|
must(q.SetServiceReplicas(ctx, sqlcdb.SetServiceReplicasParams{
|
||||||
|
Replicas: 3, UpdatedAt: sql.NullTime{Time: time.Now(), Valid: true}, ID: serviceID}))
|
||||||
|
row, _ = q.GetServiceRuntimeWithOwner(ctx, serviceID)
|
||||||
|
if row.Replicas != 3 {
|
||||||
|
t.Fatalf("replicas: %+v", row)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sleeper reads: candidate sweep + domain/status text lookups.
|
||||||
|
_, err = db.Exec(`UPDATE services SET sleep_enabled = true, domain = 'rt.example.com', status = 'running' WHERE id = $1`, serviceID)
|
||||||
|
must(err)
|
||||||
|
cands, err := q.ListSleepCandidates(ctx)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("sleep candidates: %v", err)
|
||||||
|
}
|
||||||
|
var mine *sqlcdb.ListSleepCandidatesRow
|
||||||
|
for i := range cands {
|
||||||
|
if cands[i].ID == serviceID {
|
||||||
|
mine = &cands[i]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if mine == nil || mine.SleepIdleMinutes != 15 {
|
||||||
|
t.Fatalf("candidate missing/idle wrong: %+v", mine)
|
||||||
|
}
|
||||||
|
dom, err := q.GetServiceDomainText(ctx, serviceID)
|
||||||
|
if err != nil || dom != "rt.example.com" {
|
||||||
|
t.Fatalf("domain text: %v %q", err, dom)
|
||||||
|
}
|
||||||
|
st, err := q.GetServiceStatusText(ctx, serviceID)
|
||||||
|
if err != nil || st != "running" {
|
||||||
|
t.Fatalf("status text: %v %q", err, st)
|
||||||
|
}
|
||||||
|
}
|
||||||
+197
-221
@@ -2,6 +2,8 @@ package api
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"containr/internal/database"
|
"containr/internal/database"
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
|
"context"
|
||||||
"crypto/rand"
|
"crypto/rand"
|
||||||
"crypto/rsa"
|
"crypto/rsa"
|
||||||
"crypto/x509"
|
"crypto/x509"
|
||||||
@@ -92,41 +94,73 @@ type CreateWebhookRequest struct {
|
|||||||
Branch string `json:"branch"`
|
Branch string `json:"branch"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func gitRepoFrom(r sqlcdb.ListGitRepositoriesRow) GitRepository {
|
||||||
|
return GitRepository{
|
||||||
|
ID: r.ID.String(),
|
||||||
|
ProviderID: r.ProviderID.String(),
|
||||||
|
Name: r.Name,
|
||||||
|
FullName: r.FullName,
|
||||||
|
Description: r.Description,
|
||||||
|
CloneURL: r.CloneUrl,
|
||||||
|
WebhookURL: r.WebhookUrl,
|
||||||
|
DefaultBranch: r.DefaultBranch.String,
|
||||||
|
IsPrivate: r.IsPrivate.Bool,
|
||||||
|
UserID: r.UserID.String(),
|
||||||
|
CreatedAt: r.CreatedAt.Time,
|
||||||
|
UpdatedAt: r.UpdatedAt.Time,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func gitRepoFromFullNameRow(r sqlcdb.GetGitRepositoryByFullNameRow) GitRepository {
|
||||||
|
return GitRepository{
|
||||||
|
ID: r.ID.String(),
|
||||||
|
ProviderID: r.ProviderID.String(),
|
||||||
|
Name: r.Name,
|
||||||
|
FullName: r.FullName,
|
||||||
|
Description: r.Description,
|
||||||
|
CloneURL: r.CloneUrl,
|
||||||
|
WebhookURL: r.WebhookUrl,
|
||||||
|
DefaultBranch: r.DefaultBranch.String,
|
||||||
|
IsPrivate: r.IsPrivate.Bool,
|
||||||
|
UserID: r.UserID.String(),
|
||||||
|
CreatedAt: r.CreatedAt.Time,
|
||||||
|
UpdatedAt: r.UpdatedAt.Time,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func gitProviderFrom(p sqlcdb.ListGitProvidersRow) GitProvider {
|
||||||
|
return GitProvider{
|
||||||
|
ID: p.ID.String(),
|
||||||
|
Name: p.Name,
|
||||||
|
DisplayName: p.DisplayName,
|
||||||
|
APIUrl: p.ApiUrl,
|
||||||
|
WebhookUrl: p.WebhookUrl,
|
||||||
|
UserID: p.UserID.String(),
|
||||||
|
CreatedAt: p.CreatedAt.Time,
|
||||||
|
UpdatedAt: p.UpdatedAt.Time,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func handleGetGitProviders(c *gin.Context) {
|
func handleGetGitProviders(c *gin.Context) {
|
||||||
ctx, ok := requireGitRequestContext(c)
|
ctx, ok := requireGitRequestContext(c)
|
||||||
if !ok {
|
if !ok {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
rows, err := ctx.db.Query(`
|
userID, err := uuid.Parse(ctx.userID)
|
||||||
SELECT id, name, display_name, api_url, webhook_url, user_id, created_at, updated_at
|
if err != nil {
|
||||||
FROM git_providers
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
||||||
WHERE user_id = $1
|
return
|
||||||
ORDER BY created_at DESC
|
}
|
||||||
`, ctx.userID)
|
rows, err := sqlcdb.New(ctx.db.DB).ListGitProviders(context.Background(), userID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
providers := make([]GitProvider, 0)
|
providers := make([]GitProvider, 0, len(rows))
|
||||||
for rows.Next() {
|
for _, p := range rows {
|
||||||
var provider GitProvider
|
providers = append(providers, gitProviderFrom(p))
|
||||||
if err := rows.Scan(
|
|
||||||
&provider.ID,
|
|
||||||
&provider.Name,
|
|
||||||
&provider.DisplayName,
|
|
||||||
&provider.APIUrl,
|
|
||||||
&provider.WebhookUrl,
|
|
||||||
&provider.UserID,
|
|
||||||
&provider.CreatedAt,
|
|
||||||
&provider.UpdatedAt,
|
|
||||||
); err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
providers = append(providers, provider)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
c.JSON(http.StatusOK, gin.H{"providers": providers})
|
c.JSON(http.StatusOK, gin.H{"providers": providers})
|
||||||
@@ -173,21 +207,22 @@ func handleConnectGitHubApp(c *gin.Context) {
|
|||||||
UserID: ctx.userID,
|
UserID: ctx.userID,
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ctx.db.QueryRow(`
|
prow, err := sqlcdb.New(ctx.db.DB).UpsertGitHubAppProvider(context.Background(), sqlcdb.UpsertGitHubAppProviderParams{
|
||||||
INSERT INTO git_providers (id, name, display_name, api_url, webhook_url, access_token, user_id, created_at, updated_at)
|
ID: uuid.MustParse(provider.ID),
|
||||||
VALUES ($1, $2, $3, $4, $5, $6, $7, NOW(), NOW())
|
Name: provider.Name,
|
||||||
ON CONFLICT (name, user_id)
|
DisplayName: provider.DisplayName,
|
||||||
DO UPDATE SET
|
ApiUrl: provider.APIUrl,
|
||||||
display_name = EXCLUDED.display_name,
|
WebhookUrl: provider.WebhookUrl,
|
||||||
api_url = EXCLUDED.api_url,
|
AccessToken: provider.AccessToken,
|
||||||
webhook_url = EXCLUDED.webhook_url,
|
UserID: uuid.MustParse(provider.UserID),
|
||||||
access_token = EXCLUDED.access_token,
|
})
|
||||||
updated_at = NOW()
|
if err != nil {
|
||||||
RETURNING id, created_at, updated_at
|
|
||||||
`, provider.ID, provider.Name, provider.DisplayName, provider.APIUrl, provider.WebhookUrl, provider.AccessToken, provider.UserID).Scan(&provider.ID, &provider.CreatedAt, &provider.UpdatedAt); err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to connect GitHub App installation"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to connect GitHub App installation"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
provider.ID = prow.ID.String()
|
||||||
|
provider.CreatedAt = prow.CreatedAt.Time
|
||||||
|
provider.UpdatedAt = prow.UpdatedAt.Time
|
||||||
|
|
||||||
provider.AccessToken = ""
|
provider.AccessToken = ""
|
||||||
c.JSON(http.StatusOK, gin.H{
|
c.JSON(http.StatusOK, gin.H{
|
||||||
@@ -259,10 +294,16 @@ func handleCreateGitProvider(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if _, err := ctx.db.Exec(`
|
q := sqlcdb.New(ctx.db.DB)
|
||||||
INSERT INTO git_providers (id, name, display_name, api_url, webhook_url, access_token, user_id, created_at, updated_at)
|
if err := q.InsertGitProvider(context.Background(), sqlcdb.InsertGitProviderParams{
|
||||||
VALUES ($1, $2, $3, $4, $5, $6, $7, NOW(), NOW())
|
ID: uuid.MustParse(provider.ID),
|
||||||
`, provider.ID, provider.Name, provider.DisplayName, provider.APIUrl, provider.WebhookUrl, provider.AccessToken, provider.UserID); err != nil {
|
Name: provider.Name,
|
||||||
|
DisplayName: provider.DisplayName,
|
||||||
|
ApiUrl: provider.APIUrl,
|
||||||
|
WebhookUrl: provider.WebhookUrl,
|
||||||
|
AccessToken: provider.AccessToken,
|
||||||
|
UserID: uuid.MustParse(provider.UserID),
|
||||||
|
}); err != nil {
|
||||||
if isUniqueConstraintError(err) {
|
if isUniqueConstraintError(err) {
|
||||||
c.JSON(http.StatusConflict, gin.H{"error": "Provider already connected for this account"})
|
c.JSON(http.StatusConflict, gin.H{"error": "Provider already connected for this account"})
|
||||||
return
|
return
|
||||||
@@ -271,13 +312,12 @@ func handleCreateGitProvider(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ctx.db.QueryRow(`
|
if ts, err := q.GetGitProviderTimestamps(context.Background(), uuid.MustParse(provider.ID)); err != nil {
|
||||||
SELECT created_at, updated_at
|
|
||||||
FROM git_providers
|
|
||||||
WHERE id = $1
|
|
||||||
`, provider.ID).Scan(&provider.CreatedAt, &provider.UpdatedAt); err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to read created provider"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to read created provider"})
|
||||||
return
|
return
|
||||||
|
} else {
|
||||||
|
provider.CreatedAt = ts.CreatedAt.Time
|
||||||
|
provider.UpdatedAt = ts.UpdatedAt.Time
|
||||||
}
|
}
|
||||||
|
|
||||||
provider.AccessToken = ""
|
provider.AccessToken = ""
|
||||||
@@ -296,12 +336,11 @@ func handleGetGitRepositories(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var providerName, providerDisplayName, providerAccessToken, providerAPIURL string
|
q := sqlcdb.New(ctx.db.DB)
|
||||||
err := ctx.db.QueryRow(`
|
provRow, err := q.GetGitProviderForRepos(context.Background(), sqlcdb.GetGitProviderForReposParams{
|
||||||
SELECT name, display_name, access_token, api_url
|
ID: uuid.MustParse(providerID),
|
||||||
FROM git_providers
|
UserID: uuid.MustParse(ctx.userID),
|
||||||
WHERE id = $1 AND user_id = $2
|
})
|
||||||
`, providerID, ctx.userID).Scan(&providerName, &providerDisplayName, &providerAccessToken, &providerAPIURL)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if errors.Is(err, sql.ErrNoRows) {
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Provider not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Provider not found"})
|
||||||
@@ -310,6 +349,10 @@ func handleGetGitRepositories(c *gin.Context) {
|
|||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
providerName := provRow.Name
|
||||||
|
providerDisplayName := provRow.DisplayName
|
||||||
|
providerAccessToken := provRow.AccessToken
|
||||||
|
providerAPIURL := provRow.ApiUrl
|
||||||
|
|
||||||
if c.DefaultQuery("sync", "1") != "0" {
|
if c.DefaultQuery("sync", "1") != "0" {
|
||||||
if syncErr := syncProviderRepositories(ctx.db, providerID, ctx.userID, providerName, providerAPIURL, providerAccessToken); syncErr != nil {
|
if syncErr := syncProviderRepositories(ctx.db, providerID, ctx.userID, providerName, providerAPIURL, providerAccessToken); syncErr != nil {
|
||||||
@@ -324,53 +367,30 @@ func handleGetGitRepositories(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
offset := (page - 1) * limit
|
offset := (page - 1) * limit
|
||||||
search := strings.TrimSpace(c.Query("search"))
|
search := strings.TrimSpace(c.Query("search"))
|
||||||
searchNullable := sql.NullString{String: search, Valid: search != ""}
|
|
||||||
|
|
||||||
rows, err := ctx.db.Query(`
|
repoRows, err := q.ListGitRepositories(context.Background(), sqlcdb.ListGitRepositoriesParams{
|
||||||
SELECT id, provider_id, name, full_name, COALESCE(description, ''), clone_url, COALESCE(webhook_url, ''),
|
ProviderID: uuid.MustParse(providerID),
|
||||||
default_branch, is_private, user_id, created_at, updated_at
|
UserID: uuid.MustParse(ctx.userID),
|
||||||
FROM git_repositories
|
Column3: search,
|
||||||
WHERE provider_id = $1 AND user_id = $2
|
Limit: int32(limit),
|
||||||
AND ($3::text IS NULL OR full_name ILIKE ('%' || $3::text || '%') OR name ILIKE ('%' || $3::text || '%'))
|
Offset: int32(offset),
|
||||||
ORDER BY updated_at DESC
|
})
|
||||||
LIMIT $4 OFFSET $5
|
|
||||||
`, providerID, ctx.userID, searchNullable, limit, offset)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
repositories := make([]GitRepository, 0)
|
repositories := make([]GitRepository, 0, len(repoRows))
|
||||||
for rows.Next() {
|
for _, r := range repoRows {
|
||||||
var repo GitRepository
|
repositories = append(repositories, gitRepoFrom(r))
|
||||||
if err := rows.Scan(
|
|
||||||
&repo.ID,
|
|
||||||
&repo.ProviderID,
|
|
||||||
&repo.Name,
|
|
||||||
&repo.FullName,
|
|
||||||
&repo.Description,
|
|
||||||
&repo.CloneURL,
|
|
||||||
&repo.WebhookURL,
|
|
||||||
&repo.DefaultBranch,
|
|
||||||
&repo.IsPrivate,
|
|
||||||
&repo.UserID,
|
|
||||||
&repo.CreatedAt,
|
|
||||||
&repo.UpdatedAt,
|
|
||||||
); err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
repositories = append(repositories, repo)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var total int
|
total, err := q.CountGitRepositories(context.Background(), sqlcdb.CountGitRepositoriesParams{
|
||||||
if err := ctx.db.QueryRow(`
|
ProviderID: uuid.MustParse(providerID),
|
||||||
SELECT COUNT(*)
|
UserID: uuid.MustParse(ctx.userID),
|
||||||
FROM git_repositories
|
Column3: search,
|
||||||
WHERE provider_id = $1 AND user_id = $2
|
})
|
||||||
AND ($3::text IS NULL OR full_name ILIKE ('%' || $3::text || '%') OR name ILIKE ('%' || $3::text || '%'))
|
if err != nil {
|
||||||
`, providerID, ctx.userID, searchNullable).Scan(&total); err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -408,12 +428,11 @@ func handleConnectGitRepository(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var providerName string
|
q := sqlcdb.New(ctx.db.DB)
|
||||||
err := ctx.db.QueryRow(`
|
providerName, err := q.GetGitProviderName(context.Background(), sqlcdb.GetGitProviderNameParams{
|
||||||
SELECT name
|
ID: uuid.MustParse(providerID),
|
||||||
FROM git_providers
|
UserID: uuid.MustParse(ctx.userID),
|
||||||
WHERE id = $1 AND user_id = $2
|
})
|
||||||
`, providerID, ctx.userID).Scan(&providerName)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if errors.Is(err, sql.ErrNoRows) {
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Provider not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Provider not found"})
|
||||||
@@ -429,31 +448,16 @@ func handleConnectGitRepository(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var existing GitRepository
|
existingRow, err := q.GetGitRepositoryByFullName(context.Background(), sqlcdb.GetGitRepositoryByFullNameParams{
|
||||||
err = ctx.db.QueryRow(`
|
ProviderID: uuid.MustParse(providerID),
|
||||||
SELECT id, provider_id, name, full_name, COALESCE(description, ''), clone_url, COALESCE(webhook_url, ''),
|
FullName: fullName,
|
||||||
default_branch, is_private, user_id, created_at, updated_at
|
UserID: uuid.MustParse(ctx.userID),
|
||||||
FROM git_repositories
|
})
|
||||||
WHERE provider_id = $1 AND full_name = $2 AND user_id = $3
|
|
||||||
`, providerID, fullName, ctx.userID).Scan(
|
|
||||||
&existing.ID,
|
|
||||||
&existing.ProviderID,
|
|
||||||
&existing.Name,
|
|
||||||
&existing.FullName,
|
|
||||||
&existing.Description,
|
|
||||||
&existing.CloneURL,
|
|
||||||
&existing.WebhookURL,
|
|
||||||
&existing.DefaultBranch,
|
|
||||||
&existing.IsPrivate,
|
|
||||||
&existing.UserID,
|
|
||||||
&existing.CreatedAt,
|
|
||||||
&existing.UpdatedAt,
|
|
||||||
)
|
|
||||||
switch {
|
switch {
|
||||||
case err == nil:
|
case err == nil:
|
||||||
c.JSON(http.StatusOK, gin.H{
|
c.JSON(http.StatusOK, gin.H{
|
||||||
"message": "Repository already connected",
|
"message": "Repository already connected",
|
||||||
"repository": existing,
|
"repository": gitRepoFromFullNameRow(existingRow),
|
||||||
})
|
})
|
||||||
return
|
return
|
||||||
case !errors.Is(err, sql.ErrNoRows):
|
case !errors.Is(err, sql.ErrNoRows):
|
||||||
@@ -473,13 +477,18 @@ func handleConnectGitRepository(c *gin.Context) {
|
|||||||
UserID: ctx.userID,
|
UserID: ctx.userID,
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ctx.db.QueryRow(`
|
ins, err := q.InsertGitRepository(context.Background(), sqlcdb.InsertGitRepositoryParams{
|
||||||
INSERT INTO git_repositories (
|
ID: uuid.MustParse(repo.ID),
|
||||||
id, provider_id, name, full_name, description, clone_url, default_branch, is_private, user_id, created_at, updated_at
|
ProviderID: uuid.MustParse(repo.ProviderID),
|
||||||
)
|
Name: repo.Name,
|
||||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, NOW(), NOW())
|
FullName: repo.FullName,
|
||||||
RETURNING created_at, updated_at
|
Description: sql.NullString{String: repo.Description, Valid: true},
|
||||||
`, repo.ID, repo.ProviderID, repo.Name, repo.FullName, repo.Description, repo.CloneURL, repo.DefaultBranch, repo.IsPrivate, repo.UserID).Scan(&repo.CreatedAt, &repo.UpdatedAt); err != nil {
|
CloneUrl: repo.CloneURL,
|
||||||
|
DefaultBranch: sql.NullString{String: repo.DefaultBranch, Valid: true},
|
||||||
|
IsPrivate: sql.NullBool{Bool: repo.IsPrivate, Valid: true},
|
||||||
|
UserID: uuid.MustParse(repo.UserID),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
if isUniqueConstraintError(err) {
|
if isUniqueConstraintError(err) {
|
||||||
c.JSON(http.StatusConflict, gin.H{"error": "Repository is already connected"})
|
c.JSON(http.StatusConflict, gin.H{"error": "Repository is already connected"})
|
||||||
return
|
return
|
||||||
@@ -487,6 +496,8 @@ func handleConnectGitRepository(c *gin.Context) {
|
|||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to connect repository"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to connect repository"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
repo.CreatedAt = ins.CreatedAt.Time
|
||||||
|
repo.UpdatedAt = ins.UpdatedAt.Time
|
||||||
|
|
||||||
c.JSON(http.StatusCreated, gin.H{
|
c.JSON(http.StatusCreated, gin.H{
|
||||||
"message": "Repository connected successfully",
|
"message": "Repository connected successfully",
|
||||||
@@ -516,12 +527,10 @@ func handleCreateWebhook(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var providerID string
|
providerUUID, err := sqlcdb.New(ctx.db.DB).GetGitRepositoryProviderID(context.Background(), sqlcdb.GetGitRepositoryProviderIDParams{
|
||||||
err := ctx.db.QueryRow(`
|
ID: uuid.MustParse(repoID),
|
||||||
SELECT provider_id
|
UserID: uuid.MustParse(ctx.userID),
|
||||||
FROM git_repositories
|
})
|
||||||
WHERE id = $1 AND user_id = $2
|
|
||||||
`, repoID, ctx.userID).Scan(&providerID)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if errors.Is(err, sql.ErrNoRows) {
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Repository not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Repository not found"})
|
||||||
@@ -530,6 +539,7 @@ func handleCreateWebhook(c *gin.Context) {
|
|||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
providerID := providerUUID.String()
|
||||||
|
|
||||||
eventsJSON, err := json.Marshal(req.Events)
|
eventsJSON, err := json.Marshal(req.Events)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -552,23 +562,22 @@ func handleCreateWebhook(c *gin.Context) {
|
|||||||
Active: true,
|
Active: true,
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ctx.db.QueryRow(`
|
wrow, err := sqlcdb.New(ctx.db.DB).UpsertGitWebhook(context.Background(), sqlcdb.UpsertGitWebhookParams{
|
||||||
INSERT INTO git_webhooks (
|
ID: uuid.MustParse(webhook.ID),
|
||||||
id, repo_id, provider_id, events, webhook_secret, active, branch_filter, created_at, updated_at
|
RepoID: uuid.MustParse(webhook.RepoID),
|
||||||
)
|
ProviderID: uuid.MustParse(webhook.ProviderID),
|
||||||
VALUES ($1, $2, $3, $4, $5, TRUE, $6, NOW(), NOW())
|
Events: webhook.Events,
|
||||||
ON CONFLICT (repo_id, provider_id)
|
WebhookSecret: webhook.Secret,
|
||||||
DO UPDATE SET
|
BranchFilter: sql.NullString{String: strings.TrimSpace(req.Branch), Valid: true},
|
||||||
events = EXCLUDED.events,
|
})
|
||||||
webhook_secret = EXCLUDED.webhook_secret,
|
if err != nil {
|
||||||
active = TRUE,
|
|
||||||
branch_filter = EXCLUDED.branch_filter,
|
|
||||||
updated_at = NOW()
|
|
||||||
RETURNING id, active, created_at, updated_at
|
|
||||||
`, webhook.ID, webhook.RepoID, webhook.ProviderID, webhook.Events, webhook.Secret, strings.TrimSpace(req.Branch)).Scan(&webhook.ID, &webhook.Active, &webhook.CreatedAt, &webhook.UpdatedAt); err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to configure webhook"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to configure webhook"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
webhook.ID = wrow.ID.String()
|
||||||
|
webhook.Active = wrow.Active.Bool
|
||||||
|
webhook.CreatedAt = wrow.CreatedAt.Time
|
||||||
|
webhook.UpdatedAt = wrow.UpdatedAt.Time
|
||||||
|
|
||||||
c.JSON(http.StatusCreated, gin.H{
|
c.JSON(http.StatusCreated, gin.H{
|
||||||
"message": "Webhook configured successfully",
|
"message": "Webhook configured successfully",
|
||||||
@@ -598,69 +607,39 @@ func handleGetConnectedRepositories(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
offset := (page - 1) * limit
|
offset := (page - 1) * limit
|
||||||
|
|
||||||
rows, err := ctx.db.Query(`
|
q := sqlcdb.New(ctx.db.DB)
|
||||||
SELECT r.id, r.provider_id, r.name, r.full_name, COALESCE(r.description, ''), r.clone_url,
|
repoRows, err := q.ListConnectedRepositories(context.Background(), sqlcdb.ListConnectedRepositoriesParams{
|
||||||
r.default_branch, r.is_private, r.user_id, r.created_at, r.updated_at,
|
UserID: uuid.MustParse(ctx.userID),
|
||||||
p.name as provider_name, p.display_name
|
Limit: int32(limit),
|
||||||
FROM git_repositories r
|
Offset: int32(offset),
|
||||||
JOIN git_providers p ON r.provider_id = p.id
|
})
|
||||||
WHERE r.user_id = $1
|
|
||||||
ORDER BY r.updated_at DESC
|
|
||||||
LIMIT $2 OFFSET $3
|
|
||||||
`, ctx.userID, limit, offset)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
repositories := make([]map[string]interface{}, 0)
|
|
||||||
for rows.Next() {
|
|
||||||
var repo GitRepository
|
|
||||||
var providerName, providerDisplayName string
|
|
||||||
if err := rows.Scan(
|
|
||||||
&repo.ID,
|
|
||||||
&repo.ProviderID,
|
|
||||||
&repo.Name,
|
|
||||||
&repo.FullName,
|
|
||||||
&repo.Description,
|
|
||||||
&repo.CloneURL,
|
|
||||||
&repo.DefaultBranch,
|
|
||||||
&repo.IsPrivate,
|
|
||||||
&repo.UserID,
|
|
||||||
&repo.CreatedAt,
|
|
||||||
&repo.UpdatedAt,
|
|
||||||
&providerName,
|
|
||||||
&providerDisplayName,
|
|
||||||
); err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
|
repositories := make([]map[string]interface{}, 0, len(repoRows))
|
||||||
|
for _, r := range repoRows {
|
||||||
repositories = append(repositories, map[string]interface{}{
|
repositories = append(repositories, map[string]interface{}{
|
||||||
"id": repo.ID,
|
"id": r.ID.String(),
|
||||||
"provider_id": repo.ProviderID,
|
"provider_id": r.ProviderID.String(),
|
||||||
"name": repo.Name,
|
"name": r.Name,
|
||||||
"full_name": repo.FullName,
|
"full_name": r.FullName,
|
||||||
"description": repo.Description,
|
"description": r.Description,
|
||||||
"clone_url": repo.CloneURL,
|
"clone_url": r.CloneUrl,
|
||||||
"default_branch": repo.DefaultBranch,
|
"default_branch": r.DefaultBranch.String,
|
||||||
"is_private": repo.IsPrivate,
|
"is_private": r.IsPrivate.Bool,
|
||||||
"created_at": repo.CreatedAt,
|
"created_at": r.CreatedAt.Time,
|
||||||
"updated_at": repo.UpdatedAt,
|
"updated_at": r.UpdatedAt.Time,
|
||||||
"provider": map[string]string{
|
"provider": map[string]string{
|
||||||
"name": providerName,
|
"name": r.ProviderName,
|
||||||
"display_name": providerDisplayName,
|
"display_name": r.DisplayName,
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
var total int
|
total, err := q.CountUserRepositories(context.Background(), uuid.MustParse(ctx.userID))
|
||||||
if err := ctx.db.QueryRow(`
|
if err != nil {
|
||||||
SELECT COUNT(*)
|
|
||||||
FROM git_repositories
|
|
||||||
WHERE user_id = $1
|
|
||||||
`, ctx.userID).Scan(&total); err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -705,20 +684,17 @@ func syncProviderRepositories(db *database.DB, providerID, userID, providerName,
|
|||||||
defaultBranch = "main"
|
defaultBranch = "main"
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err := db.Exec(`
|
err := sqlcdb.New(db.DB).UpsertGitRepository(context.Background(), sqlcdb.UpsertGitRepositoryParams{
|
||||||
INSERT INTO git_repositories (
|
ID: uuid.New(),
|
||||||
id, provider_id, name, full_name, description, clone_url, default_branch, is_private, user_id, created_at, updated_at
|
ProviderID: uuid.MustParse(providerID),
|
||||||
)
|
Name: repository.Name,
|
||||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, NOW(), NOW())
|
FullName: repository.FullName,
|
||||||
ON CONFLICT (provider_id, full_name)
|
Description: sql.NullString{String: repository.Description, Valid: true},
|
||||||
DO UPDATE SET
|
CloneUrl: cloneURL,
|
||||||
name = EXCLUDED.name,
|
DefaultBranch: sql.NullString{String: defaultBranch, Valid: true},
|
||||||
description = EXCLUDED.description,
|
IsPrivate: sql.NullBool{Bool: repository.IsPrivate, Valid: true},
|
||||||
clone_url = EXCLUDED.clone_url,
|
UserID: uuid.MustParse(userID),
|
||||||
default_branch = EXCLUDED.default_branch,
|
})
|
||||||
is_private = EXCLUDED.is_private,
|
|
||||||
updated_at = NOW()
|
|
||||||
`, uuid.NewString(), providerID, repository.Name, repository.FullName, repository.Description, cloneURL, defaultBranch, repository.IsPrivate, userID)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -1323,16 +1299,15 @@ func handleDeleteGitProvider(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
result, err := ctx.db.Exec(
|
affected, err := sqlcdb.New(ctx.db.DB).DeleteGitProvider(context.Background(), sqlcdb.DeleteGitProviderParams{
|
||||||
"DELETE FROM git_providers WHERE id = $1 AND user_id = $2",
|
ID: uuid.MustParse(providerID),
|
||||||
providerID, ctx.userID,
|
UserID: uuid.MustParse(ctx.userID),
|
||||||
)
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to delete provider"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to delete provider"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
affected, _ := result.RowsAffected()
|
|
||||||
if affected == 0 {
|
if affected == 0 {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Provider not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Provider not found"})
|
||||||
return
|
return
|
||||||
@@ -1362,12 +1337,10 @@ func handleGetGitRepositoryBranches(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var providerName, providerAPIURL, providerAccessToken string
|
pf, err := sqlcdb.New(ctx.db.DB).GetGitProviderForFetch(context.Background(), sqlcdb.GetGitProviderForFetchParams{
|
||||||
err := ctx.db.QueryRow(`
|
ID: uuid.MustParse(providerID),
|
||||||
SELECT name, api_url, access_token
|
UserID: uuid.MustParse(ctx.userID),
|
||||||
FROM git_providers
|
})
|
||||||
WHERE id = $1 AND user_id = $2
|
|
||||||
`, providerID, ctx.userID).Scan(&providerName, &providerAPIURL, &providerAccessToken)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if errors.Is(err, sql.ErrNoRows) {
|
if errors.Is(err, sql.ErrNoRows) {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Provider not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Provider not found"})
|
||||||
@@ -1376,6 +1349,9 @@ func handleGetGitRepositoryBranches(c *gin.Context) {
|
|||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Database error"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
providerName := pf.Name
|
||||||
|
providerAPIURL := pf.ApiUrl
|
||||||
|
providerAccessToken := pf.AccessToken
|
||||||
|
|
||||||
branches, err := fetchRepositoryBranches(providerName, providerAPIURL, providerAccessToken, owner, repo)
|
branches, err := fetchRepositoryBranches(providerName, providerAPIURL, providerAccessToken, owner, repo)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ import (
|
|||||||
"crypto/hmac"
|
"crypto/hmac"
|
||||||
"crypto/sha256"
|
"crypto/sha256"
|
||||||
"crypto/subtle"
|
"crypto/subtle"
|
||||||
|
"database/sql"
|
||||||
"encoding/hex"
|
"encoding/hex"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
@@ -14,6 +15,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"containr/internal/database"
|
"containr/internal/database"
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
"containr/internal/deployment"
|
"containr/internal/deployment"
|
||||||
"containr/internal/deployqueue"
|
"containr/internal/deployqueue"
|
||||||
|
|
||||||
@@ -38,22 +40,15 @@ func handleGitWebhookPush(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
q := sqlcdb.New(db.DB)
|
||||||
secret string
|
hook, err := q.GetGitWebhookForPush(context.Background(), uuid.MustParse(webhookID))
|
||||||
repoID string
|
|
||||||
providerID string
|
|
||||||
branchFilter string
|
|
||||||
active bool
|
|
||||||
)
|
|
||||||
err := db.QueryRow(`
|
|
||||||
SELECT webhook_secret, repo_id, provider_id, COALESCE(branch_filter, ''), active
|
|
||||||
FROM git_webhooks WHERE id = $1`, webhookID).
|
|
||||||
Scan(&secret, &repoID, &providerID, &branchFilter, &active)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Webhook not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Webhook not found"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if !active {
|
secret := hook.WebhookSecret
|
||||||
|
branchFilter := hook.BranchFilter
|
||||||
|
if !hook.Active.Bool {
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "Webhook is disabled"})
|
c.JSON(http.StatusForbidden, gin.H{"error": "Webhook is disabled"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -64,8 +59,7 @@ func handleGitWebhookPush(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var providerName string
|
providerName, _ := q.GetGitProviderNameByID(context.Background(), hook.ProviderID)
|
||||||
_ = db.QueryRow(`SELECT name FROM git_providers WHERE id = $1`, providerID).Scan(&providerName)
|
|
||||||
|
|
||||||
if !verifyGitWebhookSignature(providerName, secret, c.Request, body) {
|
if !verifyGitWebhookSignature(providerName, secret, c.Request, body) {
|
||||||
c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid signature"})
|
c.JSON(http.StatusUnauthorized, gin.H{"error": "Invalid signature"})
|
||||||
@@ -118,14 +112,12 @@ func handleGitWebhookPush(c *gin.Context) {
|
|||||||
commit = payload.After
|
commit = payload.After
|
||||||
}
|
}
|
||||||
|
|
||||||
var cloneURL, fullName, repoUserID string
|
repo, err := q.GetGitRepoForPush(context.Background(), hook.RepoID)
|
||||||
err = db.QueryRow(
|
|
||||||
`SELECT clone_url, full_name, user_id FROM git_repositories WHERE id = $1`, repoID,
|
|
||||||
).Scan(&cloneURL, &fullName, &repoUserID)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Repository not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Repository not found"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
cloneURL, fullName, repoUserID := repo.CloneUrl, repo.FullName, repo.UserID.String()
|
||||||
|
|
||||||
engineValue, _ := c.Get("deployment_engine")
|
engineValue, _ := c.Get("deployment_engine")
|
||||||
engine, _ := engineValue.(*deployment.DeploymentEngine)
|
engine, _ := engineValue.(*deployment.DeploymentEngine)
|
||||||
@@ -147,34 +139,37 @@ func handleGitWebhookPush(c *gin.Context) {
|
|||||||
// the pushed repo+branch. Shared by the per-repo webhook receiver and the
|
// the pushed repo+branch. Shared by the per-repo webhook receiver and the
|
||||||
// GitHub App webhook endpoint.
|
// GitHub App webhook endpoint.
|
||||||
func dispatchPushToServices(c *gin.Context, db *database.DB, engine *deployment.DeploymentEngine, cloneURL, fullName, branch, commit, repoUserID string) (int, error) {
|
func dispatchPushToServices(c *gin.Context, db *database.DB, engine *deployment.DeploymentEngine, cloneURL, fullName, branch, commit, repoUserID string) (int, error) {
|
||||||
// Sync-style events may carry no branch — match services on repo only.
|
q := sqlcdb.New(db.DB)
|
||||||
branchClause := "s.git_branch = $1"
|
ctx := context.Background()
|
||||||
if branch == "" {
|
// Sync-style events may carry no branch — empty branch matches all.
|
||||||
branchClause = "TRUE"
|
rows, err := q.ListServicesForPush(ctx, sqlcdb.ListServicesForPushParams{
|
||||||
}
|
GitBranch: sql.NullString{String: branch, Valid: true},
|
||||||
rows, err := db.Query(fmt.Sprintf(`
|
GitRepo: sql.NullString{String: cloneURL, Valid: true},
|
||||||
SELECT s.id, s.project_id, s.name, s.type, s.status, s.image, s.command,
|
GitRepo_2: sql.NullString{String: fullName, Valid: true},
|
||||||
s.environment, s.git_repo, s.git_branch, s.build_path, s.cpu, s.memory,
|
})
|
||||||
s.created_at, s.updated_at
|
|
||||||
FROM services s
|
|
||||||
WHERE %s AND (s.git_repo = $2 OR s.git_repo = $3)`, branchClause),
|
|
||||||
branch, cloneURL, fullName)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var services []Service
|
var services []Service
|
||||||
for rows.Next() {
|
for _, row := range rows {
|
||||||
var s Service
|
services = append(services, Service{
|
||||||
if err := rows.Scan(
|
ID: row.ID,
|
||||||
&s.ID, &s.ProjectID, &s.Name, &s.Type, &s.Status, &s.Image, &s.Command,
|
ProjectID: row.ProjectID,
|
||||||
&s.Environment, &s.GitRepo, &s.GitBranch, &s.BuildPath, &s.CPU, &s.Memory,
|
Name: row.Name,
|
||||||
&s.CreatedAt, &s.UpdatedAt,
|
Type: row.Type,
|
||||||
); err != nil {
|
Status: row.Status,
|
||||||
return 0, err
|
Image: row.Image,
|
||||||
}
|
Command: row.Command,
|
||||||
services = append(services, s)
|
Environment: row.Environment,
|
||||||
|
GitRepo: row.GitRepo,
|
||||||
|
GitBranch: row.GitBranch,
|
||||||
|
BuildPath: row.BuildPath,
|
||||||
|
CPU: row.Cpu,
|
||||||
|
Memory: row.Memory,
|
||||||
|
CreatedAt: row.CreatedAt.Time,
|
||||||
|
UpdatedAt: row.UpdatedAt.Time,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
enqueued := 0
|
enqueued := 0
|
||||||
@@ -192,27 +187,34 @@ func dispatchPushToServices(c *gin.Context, db *database.DB, engine *deployment.
|
|||||||
CreatedAt: now,
|
CreatedAt: now,
|
||||||
UpdatedAt: now,
|
UpdatedAt: now,
|
||||||
}
|
}
|
||||||
if _, err := db.Exec(
|
if err := q.InsertDeployment(ctx, sqlcdb.InsertDeploymentParams{
|
||||||
`INSERT INTO deployments
|
ID: d.ID,
|
||||||
(id, service_id, version, commit_hash, status, image_name, image_tag, created_at, updated_at)
|
ServiceID: d.ServiceID,
|
||||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)`,
|
Version: fmt.Sprintf("v%d", now.Unix()),
|
||||||
d.ID, d.ServiceID, fmt.Sprintf("v%d", now.Unix()), d.CommitHash, d.Status, d.ImageName, d.ImageTag, d.CreatedAt, d.UpdatedAt,
|
CommitHash: sql.NullString{String: ptrStr(d.CommitHash), Valid: d.CommitHash != nil},
|
||||||
); err != nil {
|
Status: sql.NullString{String: d.Status, Valid: true},
|
||||||
|
ImageName: sql.NullString{String: d.ImageName, Valid: true},
|
||||||
|
ImageTag: sql.NullString{String: d.ImageTag, Valid: true},
|
||||||
|
CreatedAt: sql.NullTime{Time: d.CreatedAt, Valid: true},
|
||||||
|
UpdatedAt: sql.NullTime{Time: d.UpdatedAt, Valid: true},
|
||||||
|
}); err != nil {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if engine == nil {
|
if engine == nil {
|
||||||
failedAt := time.Now()
|
|
||||||
failure := "Deployment engine unavailable. Docker may not be configured on this server."
|
failure := "Deployment engine unavailable. Docker may not be configured on this server."
|
||||||
_, _ = db.Exec(
|
failedAt := time.Now()
|
||||||
`UPDATE deployments SET status = 'failed', error = $1, completed_at = $2, updated_at = $2 WHERE id = $3`,
|
_ = q.FailDeployment(ctx, sqlcdb.FailDeploymentParams{
|
||||||
failure, failedAt, d.ID,
|
Error: sql.NullString{String: failure, Valid: true},
|
||||||
)
|
CompletedAt: sql.NullTime{Time: failedAt, Valid: true},
|
||||||
|
ID: d.ID,
|
||||||
|
})
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
_, _ = db.Exec(
|
_ = q.SetServiceStatus(ctx, sqlcdb.SetServiceStatusParams{
|
||||||
`UPDATE services SET status = 'building', updated_at = $1 WHERE id = $2`,
|
Status: sql.NullString{String: "building", Valid: true},
|
||||||
time.Now(), service.ID,
|
UpdatedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
)
|
ID: service.ID,
|
||||||
|
})
|
||||||
// Webhook pushes default to a clean build — stale layers are the
|
// Webhook pushes default to a clean build — stale layers are the
|
||||||
// classic silent-rollback failure (dflow convention).
|
// classic silent-rollback failure (dflow convention).
|
||||||
branchForBuild := branch
|
branchForBuild := branch
|
||||||
@@ -226,14 +228,17 @@ func dispatchPushToServices(c *gin.Context, db *database.DB, engine *deployment.
|
|||||||
runDeploymentAndSync(jctx, db, engine, &d, service, webhookReq, repoUserID)
|
runDeploymentAndSync(jctx, db, engine, &d, service, webhookReq, repoUserID)
|
||||||
},
|
},
|
||||||
}); pos > 0 {
|
}); pos > 0 {
|
||||||
_, _ = db.Exec(
|
now := sql.NullTime{Time: time.Now(), Valid: true}
|
||||||
`UPDATE deployments SET status = 'queued', updated_at = $1 WHERE id = $2`,
|
_ = q.SetDeploymentStatus(ctx, sqlcdb.SetDeploymentStatusParams{
|
||||||
time.Now(), d.ID,
|
Status: sql.NullString{String: "queued", Valid: true},
|
||||||
)
|
UpdatedAt: now,
|
||||||
_, _ = db.Exec(
|
ID: d.ID,
|
||||||
`UPDATE services SET status = 'queued', updated_at = $1 WHERE id = $2`,
|
})
|
||||||
time.Now(), service.ID,
|
_ = q.SetServiceStatus(ctx, sqlcdb.SetServiceStatusParams{
|
||||||
)
|
Status: sql.NullString{String: "queued", Valid: true},
|
||||||
|
UpdatedAt: now,
|
||||||
|
ID: service.ID,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
enqueued++
|
enqueued++
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,16 @@
|
|||||||
package api
|
package api
|
||||||
|
|
||||||
import "testing"
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"containr/internal/database"
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
func TestNormalizeRepositoryFullName(t *testing.T) {
|
func TestNormalizeRepositoryFullName(t *testing.T) {
|
||||||
cases := []struct {
|
cases := []struct {
|
||||||
@@ -83,3 +93,195 @@ func TestDeriveCloneURL(t *testing.T) {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Live invariant test against real Postgres — set METRICS_TEST_DSN (or
|
||||||
|
// DATABASE_URL). Covers the generated git provider/repo/webhook queries plus
|
||||||
|
// the push-dispatch service matcher (empty branch must match all branches).
|
||||||
|
func TestGitLiveInvariants(t *testing.T) {
|
||||||
|
dsn := testDSNString(t)
|
||||||
|
db, err := database.NewConnection(dsn)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("connect: %v", err)
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
ctx := context.Background()
|
||||||
|
q := sqlcdb.New(db.DB)
|
||||||
|
|
||||||
|
userID := uuid.New()
|
||||||
|
projectID := uuid.New()
|
||||||
|
envID := uuid.New()
|
||||||
|
providerID := uuid.New()
|
||||||
|
repoID := uuid.New()
|
||||||
|
webhookID := uuid.New()
|
||||||
|
svcMain := uuid.New()
|
||||||
|
svcDev := uuid.New()
|
||||||
|
suffix := strings.ReplaceAll(userID.String()[:8], "-", "")
|
||||||
|
must := func(err error) {
|
||||||
|
t.Helper()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("setup: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = db.Exec(`INSERT INTO users (id, email, password_hash, name) VALUES ($1, $2, 'x', 'test')`,
|
||||||
|
userID, "git-"+suffix+"@test.local")
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO projects (id, name, owner_id, is_approved) VALUES ($1, $2, $3, true)`,
|
||||||
|
projectID, "git-"+suffix, userID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO environments (id, name, project_id) VALUES ($1, 'production', $2)`,
|
||||||
|
envID, projectID)
|
||||||
|
must(err)
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM users WHERE id = $1`, userID) }()
|
||||||
|
|
||||||
|
// Provider lifecycle.
|
||||||
|
must(q.InsertGitProvider(ctx, sqlcdb.InsertGitProviderParams{
|
||||||
|
ID: providerID, Name: "github", DisplayName: "GitHub",
|
||||||
|
ApiUrl: "https://api.github.com", WebhookUrl: "https://hooks.test",
|
||||||
|
AccessToken: "tok", UserID: userID,
|
||||||
|
}))
|
||||||
|
providers, err := q.ListGitProviders(ctx, userID)
|
||||||
|
if err != nil || len(providers) == 0 {
|
||||||
|
t.Fatalf("providers: %v %v", err, providers)
|
||||||
|
}
|
||||||
|
name, err := q.GetGitProviderNameByID(ctx, providerID)
|
||||||
|
if err != nil || name != "github" {
|
||||||
|
t.Fatalf("provider name: %v %q", err, name)
|
||||||
|
}
|
||||||
|
fetch, err := q.GetGitProviderForFetch(ctx, sqlcdb.GetGitProviderForFetchParams{ID: providerID, UserID: userID})
|
||||||
|
if err != nil || fetch.AccessToken != "tok" || fetch.ApiUrl != "https://api.github.com" {
|
||||||
|
t.Fatalf("fetch: %v %+v", err, fetch)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Repository insert + search.
|
||||||
|
cloneURL := "https://github.com/acme/svc-" + suffix + ".git"
|
||||||
|
fullName := "acme/svc-" + suffix
|
||||||
|
_, err = q.InsertGitRepository(ctx, sqlcdb.InsertGitRepositoryParams{
|
||||||
|
ID: repoID, ProviderID: providerID, Name: "svc-" + suffix, FullName: fullName,
|
||||||
|
Description: sql.NullString{String: "", Valid: true},
|
||||||
|
CloneUrl: cloneURL,
|
||||||
|
DefaultBranch: sql.NullString{String: "main", Valid: true},
|
||||||
|
IsPrivate: sql.NullBool{Bool: false, Valid: true},
|
||||||
|
UserID: userID,
|
||||||
|
})
|
||||||
|
must(err)
|
||||||
|
byName, err := q.GetGitRepositoryByFullName(ctx, sqlcdb.GetGitRepositoryByFullNameParams{
|
||||||
|
ProviderID: providerID, FullName: fullName, UserID: userID,
|
||||||
|
})
|
||||||
|
if err != nil || byName.ID != repoID || byName.CloneUrl != cloneURL {
|
||||||
|
t.Fatalf("repo by full name: %v %+v", err, byName)
|
||||||
|
}
|
||||||
|
all, err := q.ListGitRepositories(ctx, sqlcdb.ListGitRepositoriesParams{
|
||||||
|
ProviderID: providerID, UserID: userID, Column3: "", Limit: 50, Offset: 0,
|
||||||
|
})
|
||||||
|
if err != nil || len(all) == 0 {
|
||||||
|
t.Fatalf("repo list (empty search): %v %v", err, all)
|
||||||
|
}
|
||||||
|
hit, err := q.ListGitRepositories(ctx, sqlcdb.ListGitRepositoriesParams{
|
||||||
|
ProviderID: providerID, UserID: userID, Column3: "svc-" + suffix, Limit: 50, Offset: 0,
|
||||||
|
})
|
||||||
|
if err != nil || len(hit) != 1 || hit[0].ID != repoID {
|
||||||
|
t.Fatalf("repo list (matching search): %v %v", err, hit)
|
||||||
|
}
|
||||||
|
miss, err := q.ListGitRepositories(ctx, sqlcdb.ListGitRepositoriesParams{
|
||||||
|
ProviderID: providerID, UserID: userID, Column3: "zzz-no-match", Limit: 50, Offset: 0,
|
||||||
|
})
|
||||||
|
if err != nil || len(miss) != 0 {
|
||||||
|
t.Fatalf("repo list (non-matching search): %v %v", err, miss)
|
||||||
|
}
|
||||||
|
count, err := q.CountGitRepositories(ctx, sqlcdb.CountGitRepositoriesParams{
|
||||||
|
ProviderID: providerID, UserID: userID, Column3: "",
|
||||||
|
})
|
||||||
|
if err != nil || count < 1 {
|
||||||
|
t.Fatalf("repo count: %v %d", err, count)
|
||||||
|
}
|
||||||
|
// Upsert on (provider_id, full_name) conflict — updates rather than errors.
|
||||||
|
must(q.UpsertGitRepository(ctx, sqlcdb.UpsertGitRepositoryParams{
|
||||||
|
ID: uuid.New(), ProviderID: providerID, Name: "renamed-" + suffix, FullName: fullName,
|
||||||
|
Description: sql.NullString{String: "d", Valid: true},
|
||||||
|
CloneUrl: cloneURL,
|
||||||
|
DefaultBranch: sql.NullString{String: "main", Valid: true},
|
||||||
|
IsPrivate: sql.NullBool{Bool: true, Valid: true},
|
||||||
|
UserID: userID,
|
||||||
|
}))
|
||||||
|
byName, err = q.GetGitRepositoryByFullName(ctx, sqlcdb.GetGitRepositoryByFullNameParams{
|
||||||
|
ProviderID: providerID, FullName: fullName, UserID: userID,
|
||||||
|
})
|
||||||
|
if err != nil || byName.Name != "renamed-"+suffix {
|
||||||
|
t.Fatalf("repo after upsert: %v %+v", err, byName)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Connected-repos join + webhook lifecycle.
|
||||||
|
connected, err := q.ListConnectedRepositories(ctx, sqlcdb.ListConnectedRepositoriesParams{
|
||||||
|
UserID: userID, Limit: 50, Offset: 0,
|
||||||
|
})
|
||||||
|
if err != nil || len(connected) == 0 {
|
||||||
|
t.Fatalf("connected repos: %v %v", err, connected)
|
||||||
|
}
|
||||||
|
wh, err := q.UpsertGitWebhook(ctx, sqlcdb.UpsertGitWebhookParams{
|
||||||
|
ID: webhookID, RepoID: repoID, ProviderID: providerID,
|
||||||
|
Events: "push", WebhookSecret: "sec-" + suffix,
|
||||||
|
BranchFilter: sql.NullString{String: "", Valid: true},
|
||||||
|
})
|
||||||
|
must(err)
|
||||||
|
if !wh.Active.Bool {
|
||||||
|
t.Fatalf("webhook inactive after upsert: %+v", wh)
|
||||||
|
}
|
||||||
|
push, err := q.GetGitWebhookForPush(ctx, webhookID)
|
||||||
|
if err != nil || push.WebhookSecret != "sec-"+suffix || push.ProviderID != providerID {
|
||||||
|
t.Fatalf("webhook for push: %v %+v", err, push)
|
||||||
|
}
|
||||||
|
pushRepo, err := q.GetGitRepoForPush(ctx, repoID)
|
||||||
|
if err != nil || pushRepo.CloneUrl != cloneURL || pushRepo.FullName != fullName || pushRepo.UserID != userID {
|
||||||
|
t.Fatalf("repo for push: %v %+v", err, pushRepo)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Push-dispatch service matcher: branch filter and empty-branch match-all.
|
||||||
|
for id, branch := range map[uuid.UUID]string{svcMain: "main", svcDev: "dev"} {
|
||||||
|
_, err = db.Exec(`INSERT INTO services (id, name, project_id, environment_id, service_type, source_type, git_repo, git_branch)
|
||||||
|
VALUES ($1, $2, $3, $4, 'web', 'git', $5, $6)`,
|
||||||
|
id, "svc-"+suffix+"-"+branch, projectID, envID, cloneURL, branch)
|
||||||
|
must(err)
|
||||||
|
}
|
||||||
|
servicesFor := func(branch string) []sqlcdb.ListServicesForPushRow {
|
||||||
|
rows, err := q.ListServicesForPush(ctx, sqlcdb.ListServicesForPushParams{
|
||||||
|
GitBranch: sql.NullString{String: branch, Valid: true},
|
||||||
|
GitRepo: sql.NullString{String: cloneURL, Valid: true},
|
||||||
|
GitRepo_2: sql.NullString{String: fullName, Valid: true},
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("services for push %q: %v", branch, err)
|
||||||
|
}
|
||||||
|
return rows
|
||||||
|
}
|
||||||
|
mainRows := servicesFor("main")
|
||||||
|
if len(mainRows) != 1 || mainRows[0].ID != svcMain || mainRows[0].GitBranch != "main" {
|
||||||
|
t.Fatalf("main branch dispatch: %+v", mainRows)
|
||||||
|
}
|
||||||
|
devRows := servicesFor("dev")
|
||||||
|
if len(devRows) != 1 || devRows[0].ID != svcDev {
|
||||||
|
t.Fatalf("dev branch dispatch: %+v", devRows)
|
||||||
|
}
|
||||||
|
allRows := servicesFor("")
|
||||||
|
if len(allRows) != 2 {
|
||||||
|
t.Fatalf("empty-branch dispatch should match all: %+v", allRows)
|
||||||
|
}
|
||||||
|
// Repo may also be referenced by its full_name.
|
||||||
|
fullRows, err := q.ListServicesForPush(ctx, sqlcdb.ListServicesForPushParams{
|
||||||
|
GitBranch: sql.NullString{String: "main", Valid: true},
|
||||||
|
GitRepo: sql.NullString{String: "[email protected]:other/none.git", Valid: true},
|
||||||
|
GitRepo_2: sql.NullString{String: cloneURL, Valid: true},
|
||||||
|
})
|
||||||
|
if err != nil || len(fullRows) != 1 {
|
||||||
|
t.Fatalf("full-name repo match: %v %+v", err, fullRows)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Provider delete is owner-scoped.
|
||||||
|
other, err := q.DeleteGitProvider(ctx, sqlcdb.DeleteGitProviderParams{ID: providerID, UserID: uuid.New()})
|
||||||
|
if err != nil || other != 0 {
|
||||||
|
t.Fatalf("foreign delete affected %d rows: %v", other, err)
|
||||||
|
}
|
||||||
|
deleted, err := q.DeleteGitProvider(ctx, sqlcdb.DeleteGitProviderParams{ID: providerID, UserID: userID})
|
||||||
|
if err != nil || deleted != 1 {
|
||||||
|
t.Fatalf("owner delete affected %d rows: %v", deleted, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -40,6 +40,68 @@ type PreviewEnvironment struct {
|
|||||||
DeploymentID *uuid.UUID `json:"deployment_id,omitempty"`
|
DeploymentID *uuid.UUID `json:"deployment_id,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// previewRowShape is the common column set the generated preview rows
|
||||||
|
// carry — ListPreviewEnvironmentsForProjectRow and GetPreviewEnvironmentRow
|
||||||
|
// share it.
|
||||||
|
type previewRowShape struct {
|
||||||
|
ID uuid.UUID
|
||||||
|
ProjectID uuid.UUID
|
||||||
|
ServiceID uuid.UUID
|
||||||
|
PreviewServiceID uuid.NullUUID
|
||||||
|
BranchName string
|
||||||
|
PrNumber sql.NullInt32
|
||||||
|
Environment string
|
||||||
|
Status string
|
||||||
|
Url sql.NullString
|
||||||
|
ExpiresAt time.Time
|
||||||
|
CreatedAt sql.NullTime
|
||||||
|
UpdatedAt sql.NullTime
|
||||||
|
SvcID uuid.NullUUID
|
||||||
|
ServiceName sql.NullString
|
||||||
|
ServiceType sql.NullString
|
||||||
|
}
|
||||||
|
|
||||||
|
func previewFromRow(r previewRowShape) PreviewEnvironment {
|
||||||
|
env := PreviewEnvironment{
|
||||||
|
ID: r.ID,
|
||||||
|
ProjectID: r.ProjectID,
|
||||||
|
ServiceID: r.ServiceID,
|
||||||
|
BranchName: r.BranchName,
|
||||||
|
Environment: r.Environment,
|
||||||
|
Status: r.Status,
|
||||||
|
URL: r.Url.String,
|
||||||
|
CreatedAt: r.CreatedAt.Time,
|
||||||
|
UpdatedAt: r.UpdatedAt.Time,
|
||||||
|
}
|
||||||
|
if r.PreviewServiceID.Valid {
|
||||||
|
id := r.PreviewServiceID.UUID
|
||||||
|
env.PreviewServiceID = &id
|
||||||
|
}
|
||||||
|
if r.PrNumber.Valid {
|
||||||
|
n := int(r.PrNumber.Int32)
|
||||||
|
env.PRNumber = &n
|
||||||
|
}
|
||||||
|
if !r.ExpiresAt.IsZero() {
|
||||||
|
t := r.ExpiresAt
|
||||||
|
env.ExpiresAt = &t
|
||||||
|
}
|
||||||
|
if r.SvcID.Valid {
|
||||||
|
env.Service = &Service{
|
||||||
|
ID: r.SvcID.UUID,
|
||||||
|
Name: r.ServiceName.String,
|
||||||
|
Type: r.ServiceType.String,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return env
|
||||||
|
}
|
||||||
|
|
||||||
|
func prNumberSQL(n *int) sql.NullInt32 {
|
||||||
|
if n == nil {
|
||||||
|
return sql.NullInt32{}
|
||||||
|
}
|
||||||
|
return sql.NullInt32{Int32: int32(*n), Valid: true}
|
||||||
|
}
|
||||||
|
|
||||||
// CreatePreviewEnvironmentRequest represents a request to create a preview environment
|
// CreatePreviewEnvironmentRequest represents a request to create a preview environment
|
||||||
type CreatePreviewEnvironmentRequest struct {
|
type CreatePreviewEnvironmentRequest struct {
|
||||||
ProjectID uuid.UUID `json:"project_id"`
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
@@ -86,51 +148,21 @@ func handleGetPreviewEnvironments(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Get preview environments for the project with service info
|
// Get preview environments for the project with service info
|
||||||
rows, err := db.(*database.DB).Query(
|
rows, err := sqlcdb.New(db.(*database.DB).DB).ListPreviewEnvironmentsForProject(c.Request.Context(), projectID)
|
||||||
`SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.pr_number,
|
|
||||||
pe.environment, pe.status, pe.url, pe.expires_at, pe.created_at, pe.updated_at,
|
|
||||||
s.id as service_id, s.name as service_name, s.type as service_type
|
|
||||||
FROM preview_environments pe
|
|
||||||
LEFT JOIN services s ON pe.service_id = s.id
|
|
||||||
WHERE pe.project_id = $1
|
|
||||||
ORDER BY pe.created_at DESC`,
|
|
||||||
projectID,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to retrieve preview environments"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to retrieve preview environments"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var environments []PreviewEnvironment
|
environments := make([]PreviewEnvironment, 0, len(rows))
|
||||||
for rows.Next() {
|
for _, r := range rows {
|
||||||
var env PreviewEnvironment
|
environments = append(environments, previewFromRow(previewRowShape{
|
||||||
var serviceID sql.NullString
|
ID: r.ID, ProjectID: r.ProjectID, ServiceID: r.ServiceID,
|
||||||
var serviceName sql.NullString
|
PreviewServiceID: r.PreviewServiceID, BranchName: r.BranchName,
|
||||||
var serviceType sql.NullString
|
PrNumber: r.PrNumber, Environment: r.Environment, Status: r.Status,
|
||||||
|
Url: r.Url, ExpiresAt: r.ExpiresAt, CreatedAt: r.CreatedAt, UpdatedAt: r.UpdatedAt,
|
||||||
err := rows.Scan(
|
SvcID: r.SvcID, ServiceName: r.ServiceName, ServiceType: r.ServiceType,
|
||||||
&env.ID, &env.ProjectID, &env.ServiceID, &env.PreviewServiceID, &env.BranchName, &env.PRNumber,
|
}))
|
||||||
&env.Environment, &env.Status, &env.URL, &env.ExpiresAt, &env.CreatedAt, &env.UpdatedAt,
|
|
||||||
&serviceID, &serviceName, &serviceType,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to scan preview environment"})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if serviceID.Valid {
|
|
||||||
parsedServiceID, parseErr := uuid.Parse(serviceID.String)
|
|
||||||
if parseErr == nil {
|
|
||||||
env.Service = &Service{
|
|
||||||
ID: parsedServiceID,
|
|
||||||
Name: serviceName.String,
|
|
||||||
Type: serviceType.String,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
environments = append(environments, env)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
c.JSON(http.StatusOK, gin.H{"preview_environments": environments})
|
c.JSON(http.StatusOK, gin.H{"preview_environments": environments})
|
||||||
@@ -171,48 +203,37 @@ func handleCreatePreviewEnvironment(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check if project exists and user has access
|
ctx := c.Request.Context()
|
||||||
var project Project
|
q := sqlcdb.New(db.(*database.DB).DB)
|
||||||
err = db.(*database.DB).QueryRow(
|
|
||||||
"SELECT id, name, owner_id FROM projects WHERE id = $1",
|
|
||||||
req.ProjectID,
|
|
||||||
).Scan(&project.ID, &project.Name, &project.OwnerID)
|
|
||||||
|
|
||||||
|
// Check if project exists and user has access
|
||||||
|
project, err := q.GetProjectBrief(ctx, req.ProjectID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Project not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Project not found"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check if user owns the project
|
// Check if user owns the project
|
||||||
if project.OwnerID != userID.(string) && !contextIsAdmin(c) {
|
if project.OwnerID.String() != userID.(string) && !contextIsAdmin(c) {
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check if service exists and belongs to the project
|
// Check if service exists and belongs to the project
|
||||||
var service Service
|
service, err := q.GetServiceTypeBrief(ctx, sqlcdb.GetServiceTypeBriefParams{
|
||||||
err = db.(*database.DB).QueryRow(
|
ID: req.ServiceID, ProjectID: req.ProjectID})
|
||||||
"SELECT id, name, COALESCE(type, service_type, '') FROM services WHERE id = $1 AND project_id = $2",
|
|
||||||
req.ServiceID, req.ProjectID,
|
|
||||||
).Scan(&service.ID, &service.Name, &service.Type)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found or doesn't belong to this project"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found or doesn't belong to this project"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check if preview environment already exists for this branch and service
|
// Check if preview environment already exists for this branch and service
|
||||||
var count int
|
count, err := q.CountActivePreviewsForBranch(ctx, sqlcdb.CountActivePreviewsForBranchParams{
|
||||||
err = db.(*database.DB).QueryRow(
|
ServiceID: req.ServiceID, BranchName: req.BranchName})
|
||||||
"SELECT COUNT(*) FROM preview_environments WHERE service_id = $1 AND branch_name = $2 AND status NOT IN ('expired', 'stopped')",
|
|
||||||
req.ServiceID, req.BranchName,
|
|
||||||
).Scan(&count)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to check existing preview environment"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to check existing preview environment"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if count > 0 {
|
if count > 0 {
|
||||||
c.JSON(http.StatusConflict, gin.H{"error": "Preview environment already exists for this branch and service"})
|
c.JSON(http.StatusConflict, gin.H{"error": "Preview environment already exists for this branch and service"})
|
||||||
return
|
return
|
||||||
@@ -237,29 +258,23 @@ func handleCreatePreviewEnvironment(c *gin.Context) {
|
|||||||
|
|
||||||
// The clone's branch override is what makes it a preview — everything
|
// The clone's branch override is what makes it a preview — everything
|
||||||
// else (env vars, volumes, builder, resources) stays identical.
|
// else (env vars, volumes, builder, resources) stays identical.
|
||||||
if _, err := db.(*database.DB).Exec(
|
if err := q.UpdateServicePreviewBranch(ctx, sqlcdb.UpdateServicePreviewBranchParams{
|
||||||
`UPDATE services SET git_branch = $1, domain = '', updated_at = NOW() WHERE id = $2`,
|
GitBranch: sql.NullString{String: req.BranchName, Valid: true}, ID: cloneID}); err != nil {
|
||||||
req.BranchName, cloneID,
|
_ = q.DeleteServiceByID(ctx, cloneID)
|
||||||
); err != nil {
|
|
||||||
_, _ = db.(*database.DB).Exec(`DELETE FROM services WHERE id = $1`, cloneID)
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to configure preview service"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to configure preview service"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
// Cloned service_domains would collide with the parent's Traefik
|
// Cloned service_domains would collide with the parent's Traefik
|
||||||
// router — previews get their own hostname or none.
|
// router — previews get their own hostname or none.
|
||||||
if _, err := db.(*database.DB).Exec(
|
if err := q.DeleteAllServiceDomains(ctx, cloneID); err != nil {
|
||||||
`DELETE FROM service_domains WHERE service_id = $1`, cloneID,
|
_ = q.DeleteServiceByID(ctx, cloneID)
|
||||||
); err != nil {
|
|
||||||
_, _ = db.(*database.DB).Exec(`DELETE FROM services WHERE id = $1`, cloneID)
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to configure preview domains"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to configure preview domains"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
previewHost := resolvePreviewHost(db.(*database.DB), cloneID, envName)
|
previewHost := resolvePreviewHost(db.(*database.DB), cloneID, envName)
|
||||||
if previewHost != "" {
|
if previewHost != "" {
|
||||||
_, _ = db.(*database.DB).Exec(
|
_ = q.CreatePreviewDomain(ctx, sqlcdb.CreatePreviewDomainParams{
|
||||||
`INSERT INTO service_domains (service_id, domain, is_default, cert_type) VALUES ($1, $2, true, 'tls')`,
|
ServiceID: cloneID, Domain: previewHost})
|
||||||
cloneID, previewHost,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
expiresAt := time.Now().Add(time.Duration(ttlHours) * time.Hour)
|
expiresAt := time.Now().Add(time.Duration(ttlHours) * time.Hour)
|
||||||
@@ -279,30 +294,36 @@ func handleCreatePreviewEnvironment(c *gin.Context) {
|
|||||||
env.URL = "https://" + previewHost
|
env.URL = "https://" + previewHost
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err = db.(*database.DB).Exec(
|
err = q.CreatePreviewEnvironment(ctx, sqlcdb.CreatePreviewEnvironmentParams{
|
||||||
`INSERT INTO preview_environments
|
ID: env.ID,
|
||||||
(id, project_id, service_id, preview_service_id, branch_name, pr_number, environment,
|
ProjectID: env.ProjectID,
|
||||||
status, url, expires_at, created_at, updated_at)
|
ServiceID: env.ServiceID,
|
||||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)`,
|
PreviewServiceID: uuid.NullUUID{UUID: cloneID, Valid: true},
|
||||||
env.ID, env.ProjectID, env.ServiceID, cloneID, env.BranchName, env.PRNumber,
|
BranchName: env.BranchName,
|
||||||
env.Environment, env.Status, env.URL, env.ExpiresAt, env.CreatedAt, env.UpdatedAt,
|
PrNumber: prNumberSQL(env.PRNumber),
|
||||||
)
|
Environment: env.Environment,
|
||||||
|
Status: env.Status,
|
||||||
|
Url: sql.NullString{String: env.URL, Valid: env.URL != ""},
|
||||||
|
ExpiresAt: *env.ExpiresAt,
|
||||||
|
CreatedAt: sql.NullTime{Time: env.CreatedAt, Valid: true},
|
||||||
|
UpdatedAt: sql.NullTime{Time: env.UpdatedAt, Valid: true},
|
||||||
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
_, _ = db.(*database.DB).Exec(`DELETE FROM services WHERE id = $1`, cloneID)
|
_ = q.DeleteServiceByID(ctx, cloneID)
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create preview environment"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create preview environment"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
deploymentID := uuid.New()
|
deploymentID := uuid.New()
|
||||||
_, err = db.(*database.DB).Exec(
|
err = q.InsertPendingDeployment(ctx, sqlcdb.InsertPendingDeploymentParams{
|
||||||
`INSERT INTO deployments
|
ID: deploymentID,
|
||||||
(id, service_id, version, commit_hash, status, created_at, updated_at)
|
ServiceID: cloneID,
|
||||||
VALUES ($1, $2, $3, $4, 'pending', NOW(), NOW())`,
|
Version: "preview-" + env.Environment,
|
||||||
deploymentID, cloneID, "preview-"+env.Environment, req.BranchName,
|
CommitHash: sql.NullString{String: req.BranchName, Valid: true},
|
||||||
)
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
_, _ = db.(*database.DB).Exec(`DELETE FROM preview_environments WHERE id = $1`, env.ID)
|
_ = q.DeletePreviewEnvironment(ctx, env.ID)
|
||||||
_, _ = db.(*database.DB).Exec(`DELETE FROM services WHERE id = $1`, cloneID)
|
_ = q.DeleteServiceByID(ctx, cloneID)
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to provision preview deployment"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to provision preview deployment"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -349,43 +370,20 @@ func handleGetPreviewEnvironment(c *gin.Context) {
|
|||||||
|
|
||||||
// Public when the parent project is approved; otherwise owner/member/admin.
|
// Public when the parent project is approved; otherwise owner/member/admin.
|
||||||
userID := optionalUserUUID(c)
|
userID := optionalUserUUID(c)
|
||||||
var env PreviewEnvironment
|
r, err := sqlcdb.New(db.(*database.DB).DB).GetPreviewEnvironment(
|
||||||
var serviceID sql.NullString
|
c.Request.Context(), sqlcdb.GetPreviewEnvironmentParams{
|
||||||
var serviceName sql.NullString
|
ID: envID, OwnerID: userID, Column3: contextIsAdmin(c)})
|
||||||
var serviceType sql.NullString
|
|
||||||
err = db.(*database.DB).QueryRow(
|
|
||||||
`SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.pr_number,
|
|
||||||
pe.environment, pe.status, pe.url, pe.expires_at, pe.created_at, pe.updated_at,
|
|
||||||
s.id as service_id, s.name as service_name, s.type as service_type
|
|
||||||
FROM preview_environments pe
|
|
||||||
LEFT JOIN services s ON pe.service_id = s.id
|
|
||||||
JOIN projects p ON pe.project_id = p.id
|
|
||||||
WHERE pe.id = $1 AND (p.is_approved OR p.owner_id = $2 OR $3::bool
|
|
||||||
OR EXISTS (SELECT 1 FROM project_members pm WHERE pm.project_id = p.id AND pm.user_id = $2))`,
|
|
||||||
envID, userID, contextIsAdmin(c),
|
|
||||||
).Scan(
|
|
||||||
&env.ID, &env.ProjectID, &env.ServiceID, &env.PreviewServiceID, &env.BranchName, &env.PRNumber,
|
|
||||||
&env.Environment, &env.Status, &env.URL, &env.ExpiresAt, &env.CreatedAt, &env.UpdatedAt,
|
|
||||||
&serviceID, &serviceName, &serviceType,
|
|
||||||
)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Preview environment not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Preview environment not found"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
env := previewFromRow(previewRowShape{
|
||||||
// Populate service info if available
|
ID: r.ID, ProjectID: r.ProjectID, ServiceID: r.ServiceID,
|
||||||
if serviceID.Valid {
|
PreviewServiceID: r.PreviewServiceID, BranchName: r.BranchName,
|
||||||
parsedServiceID, parseErr := uuid.Parse(serviceID.String)
|
PrNumber: r.PrNumber, Environment: r.Environment, Status: r.Status,
|
||||||
if parseErr == nil {
|
Url: r.Url, ExpiresAt: r.ExpiresAt, CreatedAt: r.CreatedAt, UpdatedAt: r.UpdatedAt,
|
||||||
env.Service = &Service{
|
SvcID: r.SvcID, ServiceName: r.ServiceName, ServiceType: r.ServiceType,
|
||||||
ID: parsedServiceID,
|
})
|
||||||
Name: serviceName.String,
|
|
||||||
Type: serviceType.String,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
c.JSON(http.StatusOK, gin.H{"preview_environment": env})
|
c.JSON(http.StatusOK, gin.H{"preview_environment": env})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -418,24 +416,21 @@ func handleUpdatePreviewEnvironment(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Check if preview environment exists and user has access
|
// Check if preview environment exists and user has access
|
||||||
var existingEnv PreviewEnvironment
|
uid, _ := uuid.Parse(fmt.Sprint(userID))
|
||||||
err = db.(*database.DB).QueryRow(
|
q := sqlcdb.New(db.(*database.DB).DB)
|
||||||
`SELECT pe.id, pe.project_id, pe.service_id, pe.branch_name, pe.pr_number,
|
r, err := q.GetPreviewEnvironmentForWrite(
|
||||||
pe.environment, pe.status, pe.url, pe.expires_at, pe.created_at, pe.updated_at
|
c.Request.Context(), sqlcdb.GetPreviewEnvironmentForWriteParams{
|
||||||
FROM preview_environments pe
|
ID: envID, OwnerID: uid, Column3: contextIsAdmin(c)})
|
||||||
JOIN projects p ON pe.project_id = p.id
|
|
||||||
WHERE pe.id = $1 AND (p.owner_id = $2 OR $3::bool)`,
|
|
||||||
envID, userID, contextIsAdmin(c),
|
|
||||||
).Scan(
|
|
||||||
&existingEnv.ID, &existingEnv.ProjectID, &existingEnv.ServiceID, &existingEnv.BranchName,
|
|
||||||
&existingEnv.PRNumber, &existingEnv.Environment, &existingEnv.Status, &existingEnv.URL,
|
|
||||||
&existingEnv.ExpiresAt, &existingEnv.CreatedAt, &existingEnv.UpdatedAt,
|
|
||||||
)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Preview environment not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Preview environment not found"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
existingEnv := previewFromRow(previewRowShape{
|
||||||
|
ID: r.ID, ProjectID: r.ProjectID, ServiceID: r.ServiceID,
|
||||||
|
BranchName: r.BranchName, PrNumber: r.PrNumber, Environment: r.Environment,
|
||||||
|
Status: r.Status, Url: r.Url, ExpiresAt: r.ExpiresAt,
|
||||||
|
CreatedAt: r.CreatedAt, UpdatedAt: r.UpdatedAt,
|
||||||
|
})
|
||||||
|
|
||||||
// Update fields if provided
|
// Update fields if provided
|
||||||
if req.Status != "" {
|
if req.Status != "" {
|
||||||
@@ -455,12 +450,17 @@ func handleUpdatePreviewEnvironment(c *gin.Context) {
|
|||||||
existingEnv.UpdatedAt = time.Now()
|
existingEnv.UpdatedAt = time.Now()
|
||||||
|
|
||||||
// Update preview environment in database
|
// Update preview environment in database
|
||||||
_, err = db.(*database.DB).Exec(
|
expiresAt := time.Time{}
|
||||||
`UPDATE preview_environments
|
if existingEnv.ExpiresAt != nil {
|
||||||
SET status = $1, url = $2, expires_at = $3, updated_at = $4
|
expiresAt = *existingEnv.ExpiresAt
|
||||||
WHERE id = $5`,
|
}
|
||||||
existingEnv.Status, existingEnv.URL, existingEnv.ExpiresAt, existingEnv.UpdatedAt, existingEnv.ID,
|
err = q.UpdatePreviewEnvironment(c.Request.Context(), sqlcdb.UpdatePreviewEnvironmentParams{
|
||||||
)
|
Status: existingEnv.Status,
|
||||||
|
Url: sql.NullString{String: existingEnv.URL, Valid: existingEnv.URL != ""},
|
||||||
|
ExpiresAt: expiresAt,
|
||||||
|
UpdatedAt: sql.NullTime{Time: existingEnv.UpdatedAt, Valid: true},
|
||||||
|
ID: existingEnv.ID,
|
||||||
|
})
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update preview environment"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update preview environment"})
|
||||||
@@ -492,46 +492,34 @@ func handleDeletePreviewEnvironment(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check if preview environment exists and user has access
|
ctx := c.Request.Context()
|
||||||
var projectOwnerID string
|
q := sqlcdb.New(db.(*database.DB).DB)
|
||||||
err = db.(*database.DB).QueryRow(
|
|
||||||
`SELECT p.owner_id
|
|
||||||
FROM preview_environments pe
|
|
||||||
JOIN projects p ON pe.project_id = p.id
|
|
||||||
WHERE pe.id = $1`,
|
|
||||||
envID,
|
|
||||||
).Scan(&projectOwnerID)
|
|
||||||
|
|
||||||
|
// Check if preview environment exists and user has access
|
||||||
|
projectOwnerID, err := q.GetPreviewEnvironmentOwner(ctx, envID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Preview environment not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Preview environment not found"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check if user owns the project
|
// Check if user owns the project
|
||||||
if projectOwnerID != userID.(string) && !contextIsAdmin(c) {
|
if projectOwnerID.String() != userID.(string) && !contextIsAdmin(c) {
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Tear down the clone's runtime and row — the preview is the clone.
|
// Tear down the clone's runtime and row — the preview is the clone.
|
||||||
var cloneID *uuid.UUID
|
if cloneID, err := q.GetPreviewServiceID(ctx, envID); err == nil && cloneID.Valid {
|
||||||
_ = db.(*database.DB).QueryRow(
|
|
||||||
`SELECT preview_service_id FROM preview_environments WHERE id = $1`, envID,
|
|
||||||
).Scan(&cloneID)
|
|
||||||
if cloneID != nil {
|
|
||||||
if engineValue, exists := c.Get("deployment_engine"); exists && engineValue != nil {
|
if engineValue, exists := c.Get("deployment_engine"); exists && engineValue != nil {
|
||||||
if engine, ok := engineValue.(*deployment.DeploymentEngine); ok {
|
if engine, ok := engineValue.(*deployment.DeploymentEngine); ok {
|
||||||
removeServiceRuntime(c.Request.Context(), db.(*database.DB), engine, *cloneID)
|
removeServiceRuntime(ctx, db.(*database.DB), engine, cloneID.UUID)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_, _ = db.(*database.DB).Exec(`DELETE FROM services WHERE id = $1`, *cloneID)
|
_ = q.DeleteServiceByID(ctx, cloneID.UUID)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Delete preview environment
|
// Delete preview environment
|
||||||
_, err = db.(*database.DB).Exec(
|
err = q.DeletePreviewEnvironment(ctx, envID)
|
||||||
"DELETE FROM preview_environments WHERE id = $1",
|
|
||||||
envID,
|
|
||||||
)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to delete preview environment"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to delete preview environment"})
|
||||||
@@ -569,22 +557,30 @@ func handlePromotePreviewEnvironment(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get preview environment details
|
ctx := c.Request.Context()
|
||||||
var env PreviewEnvironment
|
q := sqlcdb.New(db.(*database.DB).DB)
|
||||||
err = db.(*database.DB).QueryRow(
|
|
||||||
`SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.environment, pe.status
|
|
||||||
FROM preview_environments pe
|
|
||||||
JOIN projects p ON pe.project_id = p.id
|
|
||||||
WHERE pe.id = $1 AND (p.owner_id = $2 OR $3::bool)`,
|
|
||||||
envID, userID, contextIsAdmin(c),
|
|
||||||
).Scan(
|
|
||||||
&env.ID, &env.ProjectID, &env.ServiceID, &env.PreviewServiceID, &env.BranchName, &env.Environment, &env.Status,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
// Get preview environment details
|
||||||
|
uid, _ := uuid.Parse(fmt.Sprint(userID))
|
||||||
|
r, err := q.GetPreviewEnvironmentForPromote(
|
||||||
|
ctx, sqlcdb.GetPreviewEnvironmentForPromoteParams{
|
||||||
|
ID: envID, OwnerID: uid, Column3: contextIsAdmin(c)})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Preview environment not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Preview environment not found"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
env := PreviewEnvironment{
|
||||||
|
ID: r.ID,
|
||||||
|
ProjectID: r.ProjectID,
|
||||||
|
ServiceID: r.ServiceID,
|
||||||
|
BranchName: r.BranchName,
|
||||||
|
Environment: r.Environment,
|
||||||
|
Status: r.Status,
|
||||||
|
}
|
||||||
|
if r.PreviewServiceID.Valid {
|
||||||
|
id := r.PreviewServiceID.UUID
|
||||||
|
env.PreviewServiceID = &id
|
||||||
|
}
|
||||||
|
|
||||||
// Check if preview environment is in a state that can be promoted
|
// Check if preview environment is in a state that can be promoted
|
||||||
if env.Status != "running" {
|
if env.Status != "running" {
|
||||||
@@ -596,11 +592,12 @@ func handlePromotePreviewEnvironment(c *gin.Context) {
|
|||||||
// branch onto its own deployment track.
|
// branch onto its own deployment track.
|
||||||
deploymentID := uuid.New()
|
deploymentID := uuid.New()
|
||||||
promotionVersion := fmt.Sprintf("promote-%s-%d", strings.ReplaceAll(env.BranchName, "/", "-"), time.Now().Unix())
|
promotionVersion := fmt.Sprintf("promote-%s-%d", strings.ReplaceAll(env.BranchName, "/", "-"), time.Now().Unix())
|
||||||
_, err = db.(*database.DB).Exec(
|
err = q.InsertPendingDeployment(ctx, sqlcdb.InsertPendingDeploymentParams{
|
||||||
`INSERT INTO deployments (id, service_id, version, commit_hash, status, created_at, updated_at)
|
ID: deploymentID,
|
||||||
VALUES ($1, $2, $3, $4, 'pending', NOW(), NOW())`,
|
ServiceID: env.ServiceID,
|
||||||
deploymentID, env.ServiceID, promotionVersion, env.BranchName,
|
Version: promotionVersion,
|
||||||
)
|
CommitHash: sql.NullString{String: env.BranchName, Valid: true},
|
||||||
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create promotion deployment"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create promotion deployment"})
|
||||||
return
|
return
|
||||||
@@ -628,18 +625,14 @@ func handlePromotePreviewEnvironment(c *gin.Context) {
|
|||||||
if env.PreviewServiceID != nil {
|
if env.PreviewServiceID != nil {
|
||||||
if engineValue, exists := c.Get("deployment_engine"); exists && engineValue != nil {
|
if engineValue, exists := c.Get("deployment_engine"); exists && engineValue != nil {
|
||||||
if engine, ok := engineValue.(*deployment.DeploymentEngine); ok {
|
if engine, ok := engineValue.(*deployment.DeploymentEngine); ok {
|
||||||
removeServiceRuntime(c.Request.Context(), db.(*database.DB), engine, *env.PreviewServiceID)
|
removeServiceRuntime(ctx, db.(*database.DB), engine, *env.PreviewServiceID)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_, _ = db.(*database.DB).Exec(`DELETE FROM services WHERE id = $1`, *env.PreviewServiceID)
|
_ = q.DeleteServiceByID(ctx, *env.PreviewServiceID)
|
||||||
}
|
}
|
||||||
|
|
||||||
if _, err := db.(*database.DB).Exec(
|
if err := q.MarkPreviewEnvironmentStatus(ctx, sqlcdb.MarkPreviewEnvironmentStatusParams{
|
||||||
`UPDATE preview_environments
|
Status: "stopped", ID: env.ID}); err != nil {
|
||||||
SET status = 'stopped', updated_at = NOW()
|
|
||||||
WHERE id = $1`,
|
|
||||||
env.ID,
|
|
||||||
); err != nil {
|
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update preview environment status"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update preview environment status"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -682,28 +675,27 @@ func handleCleanupExpiredPreviewEnvironments(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
ctx := c.Request.Context()
|
||||||
|
q := sqlcdb.New(db.(*database.DB).DB)
|
||||||
|
uid, _ := uuid.Parse(fmt.Sprint(userID))
|
||||||
|
|
||||||
// Find expired preview environments for user's projects
|
// Find expired preview environments for user's projects
|
||||||
rows, err := db.(*database.DB).Query(
|
rows, err := q.ListExpiredPreviewsForUser(ctx, sqlcdb.ListExpiredPreviewsForUserParams{
|
||||||
`SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.environment
|
OwnerID: uid, Column2: contextIsAdmin(c)})
|
||||||
FROM preview_environments pe
|
|
||||||
JOIN projects p ON pe.project_id = p.id
|
|
||||||
WHERE (p.owner_id = $1 OR $2::bool) AND pe.expires_at < NOW() AND pe.status != 'expired'`,
|
|
||||||
userID, contextIsAdmin(c),
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to find expired preview environments"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to find expired preview environments"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var expiredEnvs []PreviewEnvironment
|
expiredEnvs := make([]PreviewEnvironment, 0, len(rows))
|
||||||
for rows.Next() {
|
for _, r := range rows {
|
||||||
var env PreviewEnvironment
|
env := PreviewEnvironment{
|
||||||
err := rows.Scan(
|
ID: r.ID, ProjectID: r.ProjectID, ServiceID: r.ServiceID,
|
||||||
&env.ID, &env.ProjectID, &env.ServiceID, &env.PreviewServiceID, &env.BranchName, &env.Environment,
|
BranchName: r.BranchName, Environment: r.Environment,
|
||||||
)
|
}
|
||||||
if err != nil {
|
if r.PreviewServiceID.Valid {
|
||||||
continue
|
id := r.PreviewServiceID.UUID
|
||||||
|
env.PreviewServiceID = &id
|
||||||
}
|
}
|
||||||
expiredEnvs = append(expiredEnvs, env)
|
expiredEnvs = append(expiredEnvs, env)
|
||||||
}
|
}
|
||||||
@@ -711,20 +703,17 @@ func handleCleanupExpiredPreviewEnvironments(c *gin.Context) {
|
|||||||
// Mark expired environments as expired and tear down their clones
|
// Mark expired environments as expired and tear down their clones
|
||||||
cleanupCount := 0
|
cleanupCount := 0
|
||||||
for _, env := range expiredEnvs {
|
for _, env := range expiredEnvs {
|
||||||
_, err := db.(*database.DB).Exec(
|
if err := q.MarkPreviewEnvironmentStatus(ctx, sqlcdb.MarkPreviewEnvironmentStatusParams{
|
||||||
"UPDATE preview_environments SET status = 'expired', updated_at = NOW() WHERE id = $1",
|
Status: "expired", ID: env.ID}); err != nil {
|
||||||
env.ID,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if env.PreviewServiceID != nil {
|
if env.PreviewServiceID != nil {
|
||||||
if engineValue, exists := c.Get("deployment_engine"); exists && engineValue != nil {
|
if engineValue, exists := c.Get("deployment_engine"); exists && engineValue != nil {
|
||||||
if engine, ok := engineValue.(*deployment.DeploymentEngine); ok {
|
if engine, ok := engineValue.(*deployment.DeploymentEngine); ok {
|
||||||
removeServiceRuntime(c.Request.Context(), db.(*database.DB), engine, *env.PreviewServiceID)
|
removeServiceRuntime(ctx, db.(*database.DB), engine, *env.PreviewServiceID)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_, _ = db.(*database.DB).Exec(`DELETE FROM services WHERE id = $1`, *env.PreviewServiceID)
|
_ = q.DeleteServiceByID(ctx, *env.PreviewServiceID)
|
||||||
}
|
}
|
||||||
cleanupCount++
|
cleanupCount++
|
||||||
}
|
}
|
||||||
@@ -754,15 +743,13 @@ func resolvePreviewHost(db *database.DB, cloneID uuid.UUID, envName string) stri
|
|||||||
if label == "" {
|
if label == "" {
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
var nodeID sql.NullString
|
|
||||||
var spread bool
|
|
||||||
if err := db.QueryRow(
|
|
||||||
`SELECT node_id, COALESCE(spread, false) FROM services WHERE id = $1`, cloneID,
|
|
||||||
).Scan(&nodeID, &spread); err != nil {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
q := sqlcdb.New(db.DB)
|
q := sqlcdb.New(db.DB)
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
|
brief, err := q.GetServicePlacementBrief(ctx, cloneID)
|
||||||
|
if err != nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
nodeID, spread := brief.NodeID, brief.Spread
|
||||||
var base string
|
var base string
|
||||||
switch {
|
switch {
|
||||||
case nodeID.Valid && nodeID.String != "":
|
case nodeID.Valid && nodeID.String != "":
|
||||||
@@ -788,31 +775,27 @@ func resolvePreviewHost(db *database.DB, cloneID uuid.UUID, envName string) stri
|
|||||||
|
|
||||||
// loadServiceForDeploy hydrates the Service fields runDeploymentAndSync
|
// loadServiceForDeploy hydrates the Service fields runDeploymentAndSync
|
||||||
// needs — mirrors the loader in handleDeployService.
|
// needs — mirrors the loader in handleDeployService.
|
||||||
|
func serviceForDeployFromRow(r sqlcdb.GetServiceForDeployRow) Service {
|
||||||
|
return Service{
|
||||||
|
ID: r.ID, ProjectID: r.ProjectID, Name: r.Name, Type: r.Type.String,
|
||||||
|
Status: r.Status.String, Image: r.Image.String, Command: r.Command.String,
|
||||||
|
Environment: r.Environment.String, GitRepo: r.GitRepo.String,
|
||||||
|
GitBranch: r.GitBranch.String, BuildPath: r.BuildPath.String,
|
||||||
|
CPU: r.Cpu.String, Memory: r.Memory.String,
|
||||||
|
Replicas: int(r.Replicas), Port: int(r.Port), Domain: r.Domain,
|
||||||
|
HealthCheckPath: r.HealthcheckPath, RestartPolicy: r.RestartPolicy,
|
||||||
|
Builder: r.Builder, CPUReserve: r.CpuReserve, MemoryReserve: r.MemoryReserve,
|
||||||
|
StaticBuildCmd: r.StaticBuildCmd, StaticDir: r.StaticDir,
|
||||||
|
CreatedAt: r.CreatedAt.Time, UpdatedAt: r.UpdatedAt.Time,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func loadServiceForDeploy(db *database.DB, serviceID uuid.UUID) (Service, error) {
|
func loadServiceForDeploy(db *database.DB, serviceID uuid.UUID) (Service, error) {
|
||||||
var service Service
|
r, err := sqlcdb.New(db.DB).GetServiceForDeploy(context.Background(), serviceID)
|
||||||
err := db.QueryRow(
|
if err != nil {
|
||||||
`SELECT s.id, s.project_id, s.name, s.type, s.status, s.image, s.command,
|
return Service{}, err
|
||||||
s.environment, s.git_repo, s.git_branch, s.build_path, s.cpu, s.memory,
|
}
|
||||||
COALESCE(s.replicas, 1), COALESCE(s.port, 0),
|
return serviceForDeployFromRow(r), nil
|
||||||
COALESCE(s.domain, ''), COALESCE(s.healthcheck_path, ''),
|
|
||||||
COALESCE(s.restart_policy, 'unless-stopped'),
|
|
||||||
COALESCE(s.builder, 'auto'), COALESCE(s.cpu_reserve, ''),
|
|
||||||
COALESCE(s.memory_reserve, ''), COALESCE(s.static_build_cmd, ''),
|
|
||||||
COALESCE(s.static_dir, ''),
|
|
||||||
s.created_at, s.updated_at
|
|
||||||
FROM services s
|
|
||||||
WHERE s.id = $1`,
|
|
||||||
serviceID,
|
|
||||||
).Scan(
|
|
||||||
&service.ID, &service.ProjectID, &service.Name, &service.Type, &service.Status,
|
|
||||||
&service.Image, &service.Command, &service.Environment, &service.GitRepo,
|
|
||||||
&service.GitBranch, &service.BuildPath, &service.CPU, &service.Memory,
|
|
||||||
&service.Replicas, &service.Port, &service.Domain, &service.HealthCheckPath,
|
|
||||||
&service.RestartPolicy, &service.Builder, &service.CPUReserve,
|
|
||||||
&service.MemoryReserve, &service.StaticBuildCmd, &service.StaticDir,
|
|
||||||
&service.CreatedAt, &service.UpdatedAt,
|
|
||||||
)
|
|
||||||
return service, err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// StartPreviewSweeper expires preview environments past their TTL — tears
|
// StartPreviewSweeper expires preview environments past their TTL — tears
|
||||||
@@ -838,54 +821,27 @@ func StartPreviewSweeper(ctx context.Context, db *database.DB, engine *deploymen
|
|||||||
|
|
||||||
// sweepExpiredPreviews retires previews whose TTL elapsed.
|
// sweepExpiredPreviews retires previews whose TTL elapsed.
|
||||||
func sweepExpiredPreviews(ctx context.Context, db *database.DB, engine *deployment.DeploymentEngine) {
|
func sweepExpiredPreviews(ctx context.Context, db *database.DB, engine *deployment.DeploymentEngine) {
|
||||||
rows, err := db.QueryContext(ctx,
|
q := sqlcdb.New(db.DB)
|
||||||
`SELECT id, preview_service_id FROM preview_environments
|
rows, err := q.ListSweepablePreviews(ctx)
|
||||||
WHERE expires_at < NOW() AND status NOT IN ('expired', 'stopped')`)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
var ids []uuid.UUID
|
var ids []uuid.UUID
|
||||||
var clones []uuid.UUID
|
for _, r := range rows {
|
||||||
for rows.Next() {
|
ids = append(ids, r.ID)
|
||||||
var id uuid.UUID
|
if r.PreviewServiceID.Valid {
|
||||||
var cloneID *uuid.UUID
|
removeServiceRuntime(ctx, db, engine, r.PreviewServiceID.UUID)
|
||||||
if err := rows.Scan(&id, &cloneID); err == nil {
|
_ = q.DeleteServiceByID(ctx, r.PreviewServiceID.UUID)
|
||||||
ids = append(ids, id)
|
|
||||||
if cloneID != nil {
|
|
||||||
clones = append(clones, *cloneID)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
rows.Close()
|
|
||||||
for _, cloneID := range clones {
|
|
||||||
removeServiceRuntime(ctx, db, engine, cloneID)
|
|
||||||
_, _ = db.ExecContext(ctx, `DELETE FROM services WHERE id = $1`, cloneID)
|
|
||||||
}
|
|
||||||
for _, id := range ids {
|
for _, id := range ids {
|
||||||
_, _ = db.ExecContext(ctx,
|
_ = q.MarkPreviewEnvironmentStatus(ctx, sqlcdb.MarkPreviewEnvironmentStatusParams{
|
||||||
`UPDATE preview_environments SET status = 'expired', updated_at = NOW() WHERE id = $1`, id)
|
Status: "expired", ID: id})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// syncPreviewStatuses folds the clone's live service status into the
|
// syncPreviewStatuses folds the clone's live service status into the
|
||||||
// preview row so list/detail endpoints reflect the real deploy state.
|
// preview row so list/detail endpoints reflect the real deploy state.
|
||||||
func syncPreviewStatuses(ctx context.Context, db *database.DB) {
|
func syncPreviewStatuses(ctx context.Context, db *database.DB) {
|
||||||
_, _ = db.ExecContext(ctx, `
|
_ = sqlcdb.New(db.DB).SyncPreviewStatuses(ctx)
|
||||||
UPDATE preview_environments pe
|
|
||||||
SET status = derived.st, updated_at = NOW()
|
|
||||||
FROM (
|
|
||||||
SELECT pe2.id,
|
|
||||||
CASE
|
|
||||||
WHEN s.status IN ('running', 'deployed') THEN 'running'
|
|
||||||
WHEN s.status IN ('failed', 'error') THEN 'failed'
|
|
||||||
WHEN s.status IN ('building', 'deploying', 'queued', 'pending', 'cloning') THEN 'building'
|
|
||||||
ELSE pe2.status
|
|
||||||
END AS st
|
|
||||||
FROM preview_environments pe2
|
|
||||||
JOIN services s ON pe2.preview_service_id = s.id
|
|
||||||
) derived
|
|
||||||
WHERE pe.id = derived.id
|
|
||||||
AND pe.status IN ('building', 'running', 'failed')
|
|
||||||
AND pe.status <> derived.st
|
|
||||||
`)
|
|
||||||
}
|
}
|
||||||
+103
-197
@@ -2,6 +2,7 @@ package api
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"database/sql"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"regexp"
|
"regexp"
|
||||||
@@ -10,6 +11,7 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"containr/internal/database"
|
"containr/internal/database"
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
"containr/internal/deployment"
|
"containr/internal/deployment"
|
||||||
"containr/internal/deployqueue"
|
"containr/internal/deployqueue"
|
||||||
"containr/internal/secrets"
|
"containr/internal/secrets"
|
||||||
@@ -22,22 +24,17 @@ import (
|
|||||||
// service — covers pinned placements, spread replicas, and leftovers from a
|
// service — covers pinned placements, spread replicas, and leftovers from a
|
||||||
// since-cleared pin.
|
// since-cleared pin.
|
||||||
func remoteAgentsForService(ctx context.Context, db *database.DB, serviceID uuid.UUID) []string {
|
func remoteAgentsForService(ctx context.Context, db *database.DB, serviceID uuid.UUID) []string {
|
||||||
rows, err := db.Query(
|
ids, err := sqlcdb.New(db.DB).ListNodeAgentsForService(ctx, serviceID.String())
|
||||||
`SELECT DISTINCT node_agent_id FROM container_instances WHERE service_id = $1`,
|
|
||||||
serviceID,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
out := make([]string, 0, len(ids))
|
||||||
var ids []string
|
for _, id := range ids {
|
||||||
for rows.Next() {
|
if id != "" {
|
||||||
var id string
|
out = append(out, id)
|
||||||
if rows.Scan(&id) == nil && id != "" {
|
|
||||||
ids = append(ids, id)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return ids
|
return out
|
||||||
}
|
}
|
||||||
|
|
||||||
// serviceLifecycle routes a start/stop/restart action to every agent holding
|
// serviceLifecycle routes a start/stop/restart action to every agent holding
|
||||||
@@ -99,7 +96,9 @@ func serviceRuntimeSpec(db *database.DB, service Service) (deployment.RuntimeSpe
|
|||||||
// Best effort: reuse the host port from the last live deployment so the
|
// Best effort: reuse the host port from the last live deployment so the
|
||||||
// public URL survives restarts. Column exists post-migration; older DBs
|
// public URL survives restarts. Column exists post-migration; older DBs
|
||||||
// silently get ephemeral ports.
|
// silently get ephemeral ports.
|
||||||
_ = db.QueryRow(`SELECT published_port FROM services WHERE id = $1`, service.ID).Scan(&spec.PublishedPort)
|
if p, err := sqlcdb.New(db.DB).GetServicePublishedPort(context.Background(), service.ID); err == nil {
|
||||||
|
spec.PublishedPort = p
|
||||||
|
}
|
||||||
if cmd := strings.TrimSpace(service.Command); cmd != "" {
|
if cmd := strings.TrimSpace(service.Command); cmd != "" {
|
||||||
spec.Command = strings.Fields(cmd)
|
spec.Command = strings.Fields(cmd)
|
||||||
}
|
}
|
||||||
@@ -137,42 +136,23 @@ var refVarPattern = regexp.MustCompile(`\$\{\{\s*([A-Za-z0-9_-]+)\.([A-Za-z_][A-
|
|||||||
// secrets) and expands ${{service.KEY}} references. KEY may be a variable on
|
// secrets) and expands ${{service.KEY}} references. KEY may be a variable on
|
||||||
// the referenced service, or the builtins HOST (service name alias) and PORT.
|
// the referenced service, or the builtins HOST (service name alias) and PORT.
|
||||||
func resolveServiceEnv(db *database.DB, service Service) (map[string]string, error) {
|
func resolveServiceEnv(db *database.DB, service Service) (map[string]string, error) {
|
||||||
rows, err := db.Query(
|
q := sqlcdb.New(db.DB)
|
||||||
`SELECT key, value FROM environment_variables WHERE service_id = $1`,
|
vars, err := q.ListServiceVariableValues(context.Background(), service.ID)
|
||||||
service.ID,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
env := map[string]string{}
|
env := map[string]string{}
|
||||||
for rows.Next() {
|
for _, v := range vars {
|
||||||
var k, v string
|
env[v.Key] = secrets.Decrypt(v.Value)
|
||||||
if err := rows.Scan(&k, &v); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
env[k] = secrets.Decrypt(v)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var refServices []struct{ ID, Name string }
|
refServices, err := q.ListProjectServiceRefs(context.Background(), service.ProjectID)
|
||||||
svcRows, err := db.Query(
|
|
||||||
`SELECT id, name FROM services WHERE project_id = $1`,
|
|
||||||
service.ProjectID,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
defer svcRows.Close()
|
|
||||||
for svcRows.Next() {
|
|
||||||
var s struct{ ID, Name string }
|
|
||||||
if err := svcRows.Scan(&s.ID, &s.Name); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
refServices = append(refServices, s)
|
|
||||||
}
|
|
||||||
|
|
||||||
byName := map[string]string{}
|
byName := map[string]uuid.UUID{}
|
||||||
for _, s := range refServices {
|
for _, s := range refServices {
|
||||||
byName[strings.ToLower(s.Name)] = s.ID
|
byName[strings.ToLower(s.Name)] = s.ID
|
||||||
}
|
}
|
||||||
@@ -189,17 +169,15 @@ func resolveServiceEnv(db *database.DB, service Service) (map[string]string, err
|
|||||||
case "HOST":
|
case "HOST":
|
||||||
return serviceName, true
|
return serviceName, true
|
||||||
case "PORT":
|
case "PORT":
|
||||||
var port int
|
if port, err := q.GetServicePort(context.Background(), id); err == nil && port > 0 {
|
||||||
if err := db.QueryRow(`SELECT COALESCE(port, 0) FROM services WHERE id = $1`, id).Scan(&port); err == nil && port > 0 {
|
return strconv.Itoa(int(port)), true
|
||||||
return strconv.Itoa(port), true
|
|
||||||
}
|
}
|
||||||
return "", false
|
return "", false
|
||||||
}
|
}
|
||||||
var value string
|
value, err := q.GetServiceVariableValue(context.Background(), sqlcdb.GetServiceVariableValueParams{
|
||||||
err := db.QueryRow(
|
ServiceID: id,
|
||||||
`SELECT value FROM environment_variables WHERE service_id = $1 AND key = $2`,
|
Key: key,
|
||||||
id, key,
|
})
|
||||||
).Scan(&value)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", false
|
return "", false
|
||||||
}
|
}
|
||||||
@@ -272,6 +250,15 @@ func parseMemoryLimit(mem string) int64 {
|
|||||||
return 0
|
return 0
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// setServiceStatusNow writes the stored status with a fresh updated_at.
|
||||||
|
func setServiceStatusNow(db *database.DB, id uuid.UUID, status string) {
|
||||||
|
_ = sqlcdb.New(db.DB).SetServiceStatus(context.Background(), sqlcdb.SetServiceStatusParams{
|
||||||
|
Status: sql.NullString{String: status, Valid: true},
|
||||||
|
UpdatedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
|
ID: id,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
func getRuntimeEngine(c *gin.Context) (*deployment.DeploymentEngine, *database.DB, bool) {
|
func getRuntimeEngine(c *gin.Context) (*deployment.DeploymentEngine, *database.DB, bool) {
|
||||||
dbValue, exists := c.Get("db")
|
dbValue, exists := c.Get("db")
|
||||||
if !exists {
|
if !exists {
|
||||||
@@ -286,6 +273,22 @@ func getRuntimeEngine(c *gin.Context) (*deployment.DeploymentEngine, *database.D
|
|||||||
return engineValue.(*deployment.DeploymentEngine), dbValue.(*database.DB), true
|
return engineValue.(*deployment.DeploymentEngine), dbValue.(*database.DB), true
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// serviceFromRuntimeRow maps the sqlc runtime row (with legacy-column
|
||||||
|
// fallbacks applied in SQL) onto the API Service model.
|
||||||
|
func serviceFromRuntimeRow(r sqlcdb.GetServiceRuntimeWithOwnerRow) Service {
|
||||||
|
return Service{
|
||||||
|
ID: r.ID, ProjectID: r.ProjectID, Name: r.Name, Type: r.Type,
|
||||||
|
Status: r.Status, Image: r.Image, Command: r.Command,
|
||||||
|
Environment: r.Environment, GitRepo: r.GitRepo, GitBranch: r.GitBranch,
|
||||||
|
BuildPath: r.BuildPath, CPU: r.Cpu, Memory: r.Memory,
|
||||||
|
Replicas: int(r.Replicas), Port: int(r.Port), Domain: r.Domain,
|
||||||
|
HealthCheckPath: r.HealthcheckPath, RestartPolicy: r.RestartPolicy,
|
||||||
|
Builder: r.Builder, CPUReserve: r.CpuReserve,
|
||||||
|
MemoryReserve: r.MemoryReserve, StaticBuildCmd: r.StaticBuildCmd,
|
||||||
|
StaticDir: r.StaticDir, CreatedAt: r.CreatedAt.Time, UpdatedAt: r.UpdatedAt.Time,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func loadOwnedService(c *gin.Context, db *database.DB) (Service, bool) {
|
func loadOwnedService(c *gin.Context, db *database.DB) (Service, bool) {
|
||||||
serviceID, err := uuid.Parse(c.Param("id"))
|
serviceID, err := uuid.Parse(c.Param("id"))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -298,43 +301,16 @@ func loadOwnedService(c *gin.Context, db *database.DB) (Service, bool) {
|
|||||||
return Service{}, false
|
return Service{}, false
|
||||||
}
|
}
|
||||||
|
|
||||||
var service Service
|
row, err := sqlcdb.New(db.DB).GetServiceRuntimeWithOwner(context.Background(), serviceID)
|
||||||
var owner string
|
|
||||||
err = db.QueryRow(
|
|
||||||
`SELECT s.id, s.project_id, s.name,
|
|
||||||
COALESCE(s.type, s.service_type, ''), COALESCE(s.status, ''),
|
|
||||||
COALESCE(s.image, s.image_name, ''), COALESCE(s.command, s.start_command, ''),
|
|
||||||
COALESCE(s.environment, ''), COALESCE(s.git_repo, s.source_url, ''),
|
|
||||||
COALESCE(s.git_branch, ''), COALESCE(s.build_path, ''),
|
|
||||||
COALESCE(s.cpu, ''), COALESCE(s.memory, ''),
|
|
||||||
COALESCE(s.replicas, 1), COALESCE(s.port, 0),
|
|
||||||
COALESCE(s.domain, ''), COALESCE(s.healthcheck_path, ''),
|
|
||||||
COALESCE(s.restart_policy, 'unless-stopped'),
|
|
||||||
COALESCE(s.builder, 'auto'), COALESCE(s.cpu_reserve, ''),
|
|
||||||
COALESCE(s.memory_reserve, ''), COALESCE(s.static_build_cmd, ''),
|
|
||||||
COALESCE(s.static_dir, ''),
|
|
||||||
s.created_at, s.updated_at, p.owner_id
|
|
||||||
FROM services s JOIN projects p ON s.project_id = p.id
|
|
||||||
WHERE s.id = $1`,
|
|
||||||
serviceID,
|
|
||||||
).Scan(
|
|
||||||
&service.ID, &service.ProjectID, &service.Name, &service.Type, &service.Status,
|
|
||||||
&service.Image, &service.Command, &service.Environment, &service.GitRepo,
|
|
||||||
&service.GitBranch, &service.BuildPath, &service.CPU, &service.Memory,
|
|
||||||
&service.Replicas, &service.Port, &service.Domain, &service.HealthCheckPath,
|
|
||||||
&service.RestartPolicy, &service.Builder, &service.CPUReserve,
|
|
||||||
&service.MemoryReserve, &service.StaticBuildCmd, &service.StaticDir,
|
|
||||||
&service.CreatedAt, &service.UpdatedAt, &owner,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found"})
|
||||||
return Service{}, false
|
return Service{}, false
|
||||||
}
|
}
|
||||||
if owner != userID.(string) && !isAdminUser(db, userID.(string)) {
|
if row.OwnerID.String() != userID.(string) && !isAdminUser(db, userID.(string)) {
|
||||||
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
c.JSON(http.StatusForbidden, gin.H{"error": "Access denied"})
|
||||||
return Service{}, false
|
return Service{}, false
|
||||||
}
|
}
|
||||||
return service, true
|
return serviceFromRuntimeRow(row), true
|
||||||
}
|
}
|
||||||
|
|
||||||
// loadReadableService loads a service when the caller may read its project:
|
// loadReadableService loads a service when the caller may read its project:
|
||||||
@@ -346,39 +322,13 @@ func loadReadableService(c *gin.Context, db *database.DB) (Service, bool) {
|
|||||||
return Service{}, false
|
return Service{}, false
|
||||||
}
|
}
|
||||||
|
|
||||||
var service Service
|
row, err := sqlcdb.New(db.DB).GetServiceRuntimeWithOwner(context.Background(), serviceID)
|
||||||
var projectID uuid.UUID
|
|
||||||
err = db.QueryRow(
|
|
||||||
`SELECT s.id, s.project_id, s.name,
|
|
||||||
COALESCE(s.type, s.service_type, ''), COALESCE(s.status, ''),
|
|
||||||
COALESCE(s.image, s.image_name, ''), COALESCE(s.command, s.start_command, ''),
|
|
||||||
COALESCE(s.environment, ''), COALESCE(s.git_repo, s.source_url, ''),
|
|
||||||
COALESCE(s.git_branch, ''), COALESCE(s.build_path, ''),
|
|
||||||
COALESCE(s.cpu, ''), COALESCE(s.memory, ''),
|
|
||||||
COALESCE(s.replicas, 1), COALESCE(s.port, 0),
|
|
||||||
COALESCE(s.domain, ''), COALESCE(s.healthcheck_path, ''),
|
|
||||||
COALESCE(s.restart_policy, 'unless-stopped'),
|
|
||||||
COALESCE(s.builder, 'auto'), COALESCE(s.cpu_reserve, ''),
|
|
||||||
COALESCE(s.memory_reserve, ''), COALESCE(s.static_build_cmd, ''),
|
|
||||||
COALESCE(s.static_dir, ''),
|
|
||||||
s.created_at, s.updated_at, s.project_id
|
|
||||||
FROM services s
|
|
||||||
WHERE s.id = $1`,
|
|
||||||
serviceID,
|
|
||||||
).Scan(
|
|
||||||
&service.ID, &service.ProjectID, &service.Name, &service.Type, &service.Status,
|
|
||||||
&service.Image, &service.Command, &service.Environment, &service.GitRepo,
|
|
||||||
&service.GitBranch, &service.BuildPath, &service.CPU, &service.Memory,
|
|
||||||
&service.Replicas, &service.Port, &service.Domain, &service.HealthCheckPath,
|
|
||||||
&service.RestartPolicy, &service.Builder, &service.CPUReserve,
|
|
||||||
&service.MemoryReserve, &service.StaticBuildCmd, &service.StaticDir,
|
|
||||||
&service.CreatedAt, &service.UpdatedAt, &projectID,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found"})
|
||||||
return Service{}, false
|
return Service{}, false
|
||||||
}
|
}
|
||||||
if _, allowed := projectReadAccess(c, db, projectID); !allowed {
|
service := serviceFromRuntimeRow(row)
|
||||||
|
if _, allowed := projectReadAccess(c, db, service.ProjectID); !allowed {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found"})
|
||||||
return Service{}, false
|
return Service{}, false
|
||||||
}
|
}
|
||||||
@@ -429,7 +379,7 @@ func reconcileNow(c *gin.Context, engine *deployment.DeploymentEngine, db *datab
|
|||||||
status = state.Status
|
status = state.Status
|
||||||
persistPublishedPort(db, service.ID, state.Ports)
|
persistPublishedPort(db, service.ID, state.Ports)
|
||||||
}
|
}
|
||||||
_, _ = db.Exec(`UPDATE services SET status = $1, updated_at = $2 WHERE id = $3`, status, time.Now(), service.ID)
|
setServiceStatusNow(db, service.ID, status)
|
||||||
c.JSON(http.StatusOK, gin.H{"runtime": state})
|
c.JSON(http.StatusOK, gin.H{"runtime": state})
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
@@ -448,7 +398,7 @@ func handleServiceStart(c *gin.Context) {
|
|||||||
c.JSON(http.StatusBadGateway, gin.H{"error": err.Error()})
|
c.JSON(http.StatusBadGateway, gin.H{"error": err.Error()})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
_, _ = db.Exec(`UPDATE services SET status = 'running', updated_at = $1 WHERE id = $2`, time.Now(), service.ID)
|
setServiceStatusNow(db, service.ID, "running")
|
||||||
c.JSON(http.StatusOK, gin.H{"status": "running"})
|
c.JSON(http.StatusOK, gin.H{"status": "running"})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -466,7 +416,7 @@ func handleServiceStop(c *gin.Context) {
|
|||||||
c.JSON(http.StatusBadGateway, gin.H{"error": err.Error()})
|
c.JSON(http.StatusBadGateway, gin.H{"error": err.Error()})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
_, _ = db.Exec(`UPDATE services SET status = 'stopped', updated_at = $1 WHERE id = $2`, time.Now(), service.ID)
|
setServiceStatusNow(db, service.ID, "stopped")
|
||||||
c.JSON(http.StatusOK, gin.H{"status": "stopped"})
|
c.JSON(http.StatusOK, gin.H{"status": "stopped"})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -485,7 +435,7 @@ func handleServiceRestart(c *gin.Context) {
|
|||||||
c.JSON(http.StatusBadGateway, gin.H{"error": err.Error()})
|
c.JSON(http.StatusBadGateway, gin.H{"error": err.Error()})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
_, _ = db.Exec(`UPDATE services SET status = 'running', updated_at = $1 WHERE id = $2`, time.Now(), service.ID)
|
setServiceStatusNow(db, service.ID, "running")
|
||||||
c.JSON(http.StatusOK, gin.H{"status": "running"})
|
c.JSON(http.StatusOK, gin.H{"status": "running"})
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -505,12 +455,8 @@ func handleServiceRedeploy(c *gin.Context) {
|
|||||||
image := service.Image
|
image := service.Image
|
||||||
if image == "" {
|
if image == "" {
|
||||||
// Git-sourced service: reuse the last successful deployment image.
|
// Git-sourced service: reuse the last successful deployment image.
|
||||||
err := db.QueryRow(
|
var err error
|
||||||
`SELECT image_name || ':' || image_tag FROM deployments
|
image, err = sqlcdb.New(db.DB).GetLastDeployedImage(context.Background(), service.ID)
|
||||||
WHERE service_id = $1 AND status = 'deployed' AND image_name <> ''
|
|
||||||
ORDER BY created_at DESC LIMIT 1`,
|
|
||||||
service.ID,
|
|
||||||
).Scan(&image)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": "No deployable image yet — run a deployment first"})
|
c.JSON(http.StatusBadRequest, gin.H{"error": "No deployable image yet — run a deployment first"})
|
||||||
return
|
return
|
||||||
@@ -556,7 +502,7 @@ func reconcileServiceJob(ctx context.Context, db *database.DB, engine *deploymen
|
|||||||
if state.Status != "" {
|
if state.Status != "" {
|
||||||
status = state.Status
|
status = state.Status
|
||||||
}
|
}
|
||||||
_, _ = db.Exec(`UPDATE services SET status = $1, updated_at = $2 WHERE id = $3`, status, time.Now(), service.ID)
|
setServiceStatusNow(db, service.ID, status)
|
||||||
}
|
}
|
||||||
return state, nil
|
return state, nil
|
||||||
}
|
}
|
||||||
@@ -575,12 +521,9 @@ func enqueueServiceRedeploy(c *gin.Context, db *database.DB, service Service) bo
|
|||||||
}
|
}
|
||||||
image := service.Image
|
image := service.Image
|
||||||
if image == "" {
|
if image == "" {
|
||||||
if err := db.QueryRow(
|
var err error
|
||||||
`SELECT image_name || ':' || image_tag FROM deployments
|
image, err = sqlcdb.New(db.DB).GetLastDeployedImage(context.Background(), service.ID)
|
||||||
WHERE service_id = $1 AND status = 'deployed' AND image_name <> ''
|
if err != nil {
|
||||||
ORDER BY created_at DESC LIMIT 1`,
|
|
||||||
service.ID,
|
|
||||||
).Scan(&image); err != nil {
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -628,10 +571,10 @@ func persistPublishedPort(db *database.DB, serviceID uuid.UUID, ports []uint16)
|
|||||||
if len(ports) == 0 {
|
if len(ports) == 0 {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
_, _ = db.Exec(
|
_ = sqlcdb.New(db.DB).SetServicePublishedPort(context.Background(), sqlcdb.SetServicePublishedPortParams{
|
||||||
`UPDATE services SET published_port = $1 WHERE id = $2 AND published_port <> $1`,
|
PublishedPort: int32(ports[0]),
|
||||||
int(ports[0]), serviceID,
|
ID: serviceID,
|
||||||
)
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// liveServiceStatus reconciles the stored status with real container state.
|
// liveServiceStatus reconciles the stored status with real container state.
|
||||||
@@ -654,7 +597,7 @@ func liveServiceStatus(c *gin.Context, db *database.DB, service *Service) {
|
|||||||
}
|
}
|
||||||
if state.Status != service.Status {
|
if state.Status != service.Status {
|
||||||
service.Status = state.Status
|
service.Status = state.Status
|
||||||
_, _ = db.Exec(`UPDATE services SET status = $1, updated_at = $2 WHERE id = $3`, state.Status, time.Now(), service.ID)
|
setServiceStatusNow(db, service.ID, state.Status)
|
||||||
}
|
}
|
||||||
persistPublishedPort(db, service.ID, state.Ports)
|
persistPublishedPort(db, service.ID, state.Ports)
|
||||||
if service.Domain != "" {
|
if service.Domain != "" {
|
||||||
@@ -676,44 +619,22 @@ func runtimeScaleService(c *gin.Context, serviceID string, replicas int) error {
|
|||||||
return fmt.Errorf("runtime unavailable")
|
return fmt.Errorf("runtime unavailable")
|
||||||
}
|
}
|
||||||
|
|
||||||
var service Service
|
q := sqlcdb.New(db.DB)
|
||||||
var owner string
|
sid, err := uuid.Parse(serviceID)
|
||||||
err := db.QueryRow(
|
|
||||||
`SELECT s.id, s.project_id, s.name,
|
|
||||||
COALESCE(s.type, s.service_type, ''), COALESCE(s.status, ''),
|
|
||||||
COALESCE(s.image, s.image_name, ''), COALESCE(s.command, s.start_command, ''),
|
|
||||||
COALESCE(s.environment, ''), COALESCE(s.git_repo, s.source_url, ''),
|
|
||||||
COALESCE(s.git_branch, ''), COALESCE(s.build_path, ''),
|
|
||||||
COALESCE(s.cpu, ''), COALESCE(s.memory, ''),
|
|
||||||
COALESCE(s.replicas, 1), COALESCE(s.port, 0),
|
|
||||||
COALESCE(s.domain, ''), COALESCE(s.healthcheck_path, ''),
|
|
||||||
COALESCE(s.restart_policy, 'unless-stopped'),
|
|
||||||
COALESCE(s.builder, 'auto'), COALESCE(s.cpu_reserve, ''),
|
|
||||||
COALESCE(s.memory_reserve, ''), COALESCE(s.static_build_cmd, ''),
|
|
||||||
COALESCE(s.static_dir, ''),
|
|
||||||
s.created_at, s.updated_at, p.owner_id
|
|
||||||
FROM services s JOIN projects p ON s.project_id = p.id
|
|
||||||
WHERE s.id = $1`, serviceID,
|
|
||||||
).Scan(
|
|
||||||
&service.ID, &service.ProjectID, &service.Name, &service.Type, &service.Status,
|
|
||||||
&service.Image, &service.Command, &service.Environment, &service.GitRepo,
|
|
||||||
&service.GitBranch, &service.BuildPath, &service.CPU, &service.Memory,
|
|
||||||
&service.Replicas, &service.Port, &service.Domain, &service.HealthCheckPath,
|
|
||||||
&service.RestartPolicy, &service.Builder, &service.CPUReserve,
|
|
||||||
&service.MemoryReserve, &service.StaticBuildCmd, &service.StaticDir,
|
|
||||||
&service.CreatedAt, &service.UpdatedAt, &owner,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
row, err := q.GetServiceRuntimeWithOwner(context.Background(), sid)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
service := serviceFromRuntimeRow(row)
|
||||||
if service.Image == "" {
|
if service.Image == "" {
|
||||||
if err := db.QueryRow(
|
image, err := q.GetLastDeployedImage(context.Background(), service.ID)
|
||||||
`SELECT image_name || ':' || image_tag FROM deployments
|
if err != nil {
|
||||||
WHERE service_id = $1 AND status = 'deployed' AND image_name <> ''
|
|
||||||
ORDER BY created_at DESC LIMIT 1`, service.ID,
|
|
||||||
).Scan(&service.Image); err != nil {
|
|
||||||
return fmt.Errorf("no deployed image to scale")
|
return fmt.Errorf("no deployed image to scale")
|
||||||
}
|
}
|
||||||
|
service.Image = image
|
||||||
}
|
}
|
||||||
|
|
||||||
spec, err := serviceRuntimeSpec(db, service)
|
spec, err := serviceRuntimeSpec(db, service)
|
||||||
@@ -731,7 +652,11 @@ func runtimeScaleService(c *gin.Context, serviceID string, replicas int) error {
|
|||||||
if state != nil {
|
if state != nil {
|
||||||
persistPublishedPort(db, service.ID, state.Ports)
|
persistPublishedPort(db, service.ID, state.Ports)
|
||||||
}
|
}
|
||||||
_, _ = db.Exec(`UPDATE services SET replicas = $1, updated_at = $2 WHERE id = $3`, replicas, time.Now(), service.ID)
|
_ = q.SetServiceReplicas(context.Background(), sqlcdb.SetServiceReplicasParams{
|
||||||
|
Replicas: int32(replicas),
|
||||||
|
UpdatedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
|
ID: service.ID,
|
||||||
|
})
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -750,47 +675,34 @@ func handleServiceEnvCheck(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
rows, err := db.Query(
|
q := sqlcdb.New(db.DB)
|
||||||
`SELECT key, value, COALESCE(is_secret, false) FROM environment_variables WHERE service_id = $1`,
|
vars, err := q.ListServiceVariablesWithSecret(context.Background(), service.ID)
|
||||||
service.ID,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var unresolved, empty, unreadable []string
|
var unresolved, empty, unreadable []string
|
||||||
var raw []struct{ key, val string }
|
var raw []struct{ key, val string }
|
||||||
for rows.Next() {
|
for _, v := range vars {
|
||||||
var k, v string
|
decrypted := secrets.Decrypt(v.Value)
|
||||||
var secret bool
|
if v.IsSecret && secrets.IsEncrypted(v.Value) && decrypted == v.Value {
|
||||||
if err := rows.Scan(&k, &v, &secret); err != nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
decrypted := secrets.Decrypt(v)
|
|
||||||
if secret && secrets.IsEncrypted(v) && decrypted == v {
|
|
||||||
// Ciphertext failed to decrypt — Decrypt returns the input.
|
// Ciphertext failed to decrypt — Decrypt returns the input.
|
||||||
unreadable = append(unreadable, k)
|
unreadable = append(unreadable, v.Key)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if strings.TrimSpace(decrypted) == "" {
|
if strings.TrimSpace(decrypted) == "" {
|
||||||
empty = append(empty, k)
|
empty = append(empty, v.Key)
|
||||||
}
|
}
|
||||||
raw = append(raw, struct{ key, val string }{k, decrypted})
|
raw = append(raw, struct{ key, val string }{v.Key, decrypted})
|
||||||
}
|
}
|
||||||
|
|
||||||
// Project siblings for ${{name.KEY}} resolution.
|
// Project siblings for ${{name.KEY}} resolution.
|
||||||
siblingIDs := map[string]string{}
|
siblingIDs := map[string]uuid.UUID{}
|
||||||
svcRows, err := db.Query(`SELECT id, name FROM services WHERE project_id = $1`, service.ProjectID)
|
if refs, err := q.ListProjectServiceRefs(context.Background(), service.ProjectID); err == nil {
|
||||||
if err == nil {
|
for _, s := range refs {
|
||||||
for svcRows.Next() {
|
siblingIDs[strings.ToLower(s.Name)] = s.ID
|
||||||
var id, name string
|
|
||||||
if err := svcRows.Scan(&id, &name); err == nil {
|
|
||||||
siblingIDs[strings.ToLower(name)] = id
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
svcRows.Close()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
resolvable := func(serviceName, key string) bool {
|
resolvable := func(serviceName, key string) bool {
|
||||||
@@ -802,20 +714,14 @@ func handleServiceEnvCheck(c *gin.Context) {
|
|||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
if strings.EqualFold(key, "PORT") {
|
if strings.EqualFold(key, "PORT") {
|
||||||
var port int
|
port, err := q.GetServicePort(context.Background(), id)
|
||||||
if err := db.QueryRow(`SELECT COALESCE(port, 0) FROM services WHERE id = $1`, id).Scan(&port); err == nil {
|
return err == nil && port > 0
|
||||||
return port > 0
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
}
|
||||||
var n int
|
_, err := q.GetServiceVariableValue(context.Background(), sqlcdb.GetServiceVariableValueParams{
|
||||||
if err := db.QueryRow(
|
ServiceID: id,
|
||||||
`SELECT COUNT(*) FROM environment_variables WHERE service_id = $1 AND key = $2`,
|
Key: key,
|
||||||
id, key,
|
})
|
||||||
).Scan(&n); err == nil {
|
return err == nil
|
||||||
return n > 0
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, kv := range raw {
|
for _, kv := range raw {
|
||||||
|
|||||||
@@ -2,7 +2,9 @@ package api
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"containr/internal/database"
|
"containr/internal/database"
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
"containr/internal/security"
|
"containr/internal/security"
|
||||||
|
"context"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
@@ -33,8 +35,8 @@ func NewSecurityHandler(db *database.DB, encryptionKey string) *SecurityHandler
|
|||||||
// Notify the project owner when a background scan completes — criticals
|
// Notify the project owner when a background scan completes — criticals
|
||||||
// earn a distinct title so they stand out in the notification list.
|
// earn a distinct title so they stand out in the notification list.
|
||||||
scanner.OnScanComplete = func(scan *security.SecurityScan) {
|
scanner.OnScanComplete = func(scan *security.SecurityScan) {
|
||||||
var ownerID string
|
ownerID, err := sqlcdb.New(db.DB).GetProjectOwnerText(context.Background(), uuid.MustParse(scan.ProjectID))
|
||||||
if err := db.QueryRow(`SELECT owner_id::text FROM projects WHERE id = $1`, scan.ProjectID).Scan(&ownerID); err != nil || ownerID == "" {
|
if err != nil || ownerID == "" {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
title := "Security scan completed"
|
title := "Security scan completed"
|
||||||
@@ -81,14 +83,10 @@ func (sh *SecurityHandler) StartSecurityScan(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var serviceExists bool
|
serviceExists, err := sqlcdb.New(sh.db.DB).ServiceExistsInProject(context.Background(), sqlcdb.ServiceExistsInProjectParams{
|
||||||
err := sh.db.QueryRow(
|
ID: uuid.MustParse(req.ServiceID),
|
||||||
`SELECT EXISTS(
|
ProjectID: uuid.MustParse(req.ProjectID),
|
||||||
SELECT 1 FROM services WHERE id = $1 AND project_id = $2
|
})
|
||||||
)`,
|
|
||||||
req.ServiceID,
|
|
||||||
req.ProjectID,
|
|
||||||
).Scan(&serviceExists)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to validate service"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to validate service"})
|
||||||
return
|
return
|
||||||
@@ -164,40 +162,30 @@ func (sh *SecurityHandler) GetVulnerabilities(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Query vulnerabilities
|
// Query vulnerabilities
|
||||||
rows, err := sh.db.Query(`
|
rows, err := sqlcdb.New(sh.db.DB).ListVulnerabilities(context.Background(), uuid.MustParse(projectID))
|
||||||
SELECT id, type, severity, title, description, service_id, status, found_at, resolved_at
|
|
||||||
FROM vulnerabilities
|
|
||||||
WHERE project_id = $1
|
|
||||||
ORDER BY
|
|
||||||
CASE severity
|
|
||||||
WHEN 'critical' THEN 1
|
|
||||||
WHEN 'high' THEN 2
|
|
||||||
WHEN 'medium' THEN 3
|
|
||||||
WHEN 'low' THEN 4
|
|
||||||
END,
|
|
||||||
found_at DESC
|
|
||||||
`, projectID)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to get vulnerabilities"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to get vulnerabilities"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var vulnerabilities []security.Vulnerability
|
var vulnerabilities []security.Vulnerability
|
||||||
for rows.Next() {
|
for _, row := range rows {
|
||||||
var vuln security.Vulnerability
|
// NULL description/service_id failed the original scan and dropped
|
||||||
var resolvedAt *time.Time
|
// the row — keep the same filter.
|
||||||
|
if !row.Description.Valid || !row.ServiceID.Valid {
|
||||||
err := rows.Scan(&vuln.ID, &vuln.Type, &vuln.Severity, &vuln.Title, &vuln.Description,
|
|
||||||
&vuln.ServiceID, &vuln.Status, &vuln.FoundAt, &resolvedAt)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
vulnerabilities = append(vulnerabilities, security.Vulnerability{
|
||||||
vuln.ResolvedAt = resolvedAt
|
ID: row.ID.String(),
|
||||||
vulnerabilities = append(vulnerabilities, vuln)
|
Type: row.Type,
|
||||||
|
Severity: row.Severity,
|
||||||
|
Title: row.Title,
|
||||||
|
Description: row.Description.String,
|
||||||
|
ServiceID: row.ServiceID.UUID.String(),
|
||||||
|
Status: row.Status,
|
||||||
|
FoundAt: row.FoundAt,
|
||||||
|
ResolvedAt: timePtrNT(row.ResolvedAt),
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
c.JSON(http.StatusOK, gin.H{"vulnerabilities": vulnerabilities})
|
c.JSON(http.StatusOK, gin.H{"vulnerabilities": vulnerabilities})
|
||||||
@@ -227,11 +215,11 @@ func (sh *SecurityHandler) UpdateVulnerability(c *gin.Context) {
|
|||||||
resolvedAt = &now
|
resolvedAt = &now
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err := sh.db.Exec(`
|
err := sqlcdb.New(sh.db.DB).UpdateVulnerabilityStatus(context.Background(), sqlcdb.UpdateVulnerabilityStatusParams{
|
||||||
UPDATE vulnerabilities
|
Status: req.Status,
|
||||||
SET status = $1, resolved_at = $2
|
ResolvedAt: ntPtr(resolvedAt),
|
||||||
WHERE id = $3
|
ID: uuid.MustParse(vulnID),
|
||||||
`, req.Status, resolvedAt, vulnID)
|
})
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update vulnerability"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to update vulnerability"})
|
||||||
@@ -271,13 +259,7 @@ func (sh *SecurityHandler) StartComplianceAssessment(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var frameworkExists bool
|
frameworkExists, err := sqlcdb.New(sh.db.DB).ComplianceFrameworkExists(context.Background(), uuid.MustParse(req.FrameworkID))
|
||||||
err := sh.db.QueryRow(
|
|
||||||
`SELECT EXISTS(
|
|
||||||
SELECT 1 FROM compliance_frameworks WHERE id = $1
|
|
||||||
)`,
|
|
||||||
req.FrameworkID,
|
|
||||||
).Scan(&frameworkExists)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to validate framework"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to validate framework"})
|
||||||
return
|
return
|
||||||
@@ -321,30 +303,25 @@ func (sh *SecurityHandler) GetComplianceReport(c *gin.Context) {
|
|||||||
|
|
||||||
// GetComplianceFrameworks retrieves available compliance frameworks
|
// GetComplianceFrameworks retrieves available compliance frameworks
|
||||||
func (sh *SecurityHandler) GetComplianceFrameworks(c *gin.Context) {
|
func (sh *SecurityHandler) GetComplianceFrameworks(c *gin.Context) {
|
||||||
rows, err := sh.db.Query(`
|
rows, err := sqlcdb.New(sh.db.DB).ListComplianceFrameworks(context.Background())
|
||||||
SELECT id, name, description, version, enabled, created_at
|
|
||||||
FROM compliance_frameworks
|
|
||||||
WHERE enabled = true
|
|
||||||
ORDER BY name
|
|
||||||
`)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to get compliance frameworks"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to get compliance frameworks"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var frameworks []security.ComplianceFramework
|
var frameworks []security.ComplianceFramework
|
||||||
for rows.Next() {
|
for _, row := range rows {
|
||||||
var framework security.ComplianceFramework
|
if !row.Description.Valid {
|
||||||
err := rows.Scan(&framework.ID, &framework.Name, &framework.Description,
|
|
||||||
&framework.Version, &framework.Enabled, &framework.CreatedAt)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
frameworks = append(frameworks, security.ComplianceFramework{
|
||||||
frameworks = append(frameworks, framework)
|
ID: row.ID.String(),
|
||||||
|
Name: row.Name,
|
||||||
|
Description: row.Description.String,
|
||||||
|
Version: row.Version,
|
||||||
|
Enabled: row.Enabled,
|
||||||
|
CreatedAt: row.CreatedAt,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
c.JSON(http.StatusOK, gin.H{"frameworks": frameworks})
|
c.JSON(http.StatusOK, gin.H{"frameworks": frameworks})
|
||||||
@@ -385,23 +362,19 @@ func (sh *SecurityHandler) GetSecurityMetrics(c *gin.Context) {
|
|||||||
Resolved int `json:"resolved"`
|
Resolved int `json:"resolved"`
|
||||||
}
|
}
|
||||||
|
|
||||||
err := sh.db.QueryRow(`
|
q := sqlcdb.New(sh.db.DB)
|
||||||
SELECT
|
mrow, err := q.GetVulnerabilityMetrics(context.Background(), uuid.MustParse(projectID))
|
||||||
COUNT(*) as total,
|
|
||||||
COUNT(*) FILTER (WHERE severity = 'critical') as critical,
|
|
||||||
COUNT(*) FILTER (WHERE severity = 'high') as high,
|
|
||||||
COUNT(*) FILTER (WHERE severity = 'medium') as medium,
|
|
||||||
COUNT(*) FILTER (WHERE severity = 'low') as low,
|
|
||||||
COUNT(*) FILTER (WHERE status = 'open') as open,
|
|
||||||
COUNT(*) FILTER (WHERE status = 'resolved') as resolved
|
|
||||||
FROM vulnerabilities
|
|
||||||
WHERE project_id = $1
|
|
||||||
`, projectID).Scan(&vulnMetrics.Total, &vulnMetrics.Critical, &vulnMetrics.High,
|
|
||||||
&vulnMetrics.Medium, &vulnMetrics.Low, &vulnMetrics.Open, &vulnMetrics.Resolved)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to get vulnerability metrics"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to get vulnerability metrics"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
vulnMetrics.Total = int(mrow.Total)
|
||||||
|
vulnMetrics.Critical = int(mrow.Critical)
|
||||||
|
vulnMetrics.High = int(mrow.High)
|
||||||
|
vulnMetrics.Medium = int(mrow.Medium)
|
||||||
|
vulnMetrics.Low = int(mrow.Low)
|
||||||
|
vulnMetrics.Open = int(mrow.Open)
|
||||||
|
vulnMetrics.Resolved = int(mrow.Resolved)
|
||||||
|
|
||||||
// Get latest scan
|
// Get latest scan
|
||||||
var latestScan struct {
|
var latestScan struct {
|
||||||
@@ -411,13 +384,11 @@ func (sh *SecurityHandler) GetSecurityMetrics(c *gin.Context) {
|
|||||||
Status string `json:"status"`
|
Status string `json:"status"`
|
||||||
}
|
}
|
||||||
|
|
||||||
err = sh.db.QueryRow(`
|
srow, err := q.GetLatestSecurityScan(context.Background(), uuid.MustParse(projectID))
|
||||||
SELECT id, COALESCE((summary->>'score')::int, 0), started_at as scanned_at, status
|
latestScan.ID = srow.ID.String()
|
||||||
FROM security_scans
|
latestScan.Score = int(srow.Score)
|
||||||
WHERE project_id = $1
|
latestScan.ScannedAt = srow.ScannedAt
|
||||||
ORDER BY started_at DESC
|
latestScan.Status = srow.Status
|
||||||
LIMIT 1
|
|
||||||
`, projectID).Scan(&latestScan.ID, &latestScan.Score, &latestScan.ScannedAt, &latestScan.Status)
|
|
||||||
|
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
latestScan = struct {
|
latestScan = struct {
|
||||||
@@ -438,13 +409,10 @@ func (sh *SecurityHandler) GetSecurityMetrics(c *gin.Context) {
|
|||||||
LastAssessed *time.Time `json:"last_assessed"`
|
LastAssessed *time.Time `json:"last_assessed"`
|
||||||
}
|
}
|
||||||
|
|
||||||
err = sh.db.QueryRow(`
|
crow, err := q.GetLatestComplianceReport(context.Background(), uuid.MustParse(projectID))
|
||||||
SELECT overall_status, score, assessment_date
|
complianceStatus.OverallStatus = crow.OverallStatus
|
||||||
FROM compliance_reports
|
complianceStatus.Score = int(crow.Score)
|
||||||
WHERE project_id = $1
|
complianceStatus.LastAssessed = &crow.AssessmentDate
|
||||||
ORDER BY assessment_date DESC
|
|
||||||
LIMIT 1
|
|
||||||
`, projectID).Scan(&complianceStatus.OverallStatus, &complianceStatus.Score, &complianceStatus.LastAssessed)
|
|
||||||
|
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
complianceStatus = struct {
|
complianceStatus = struct {
|
||||||
@@ -614,21 +582,11 @@ func (sh *SecurityHandler) requireProjectAccess(c *gin.Context, projectID string
|
|||||||
return "", false
|
return "", false
|
||||||
}
|
}
|
||||||
|
|
||||||
var hasAccess bool
|
hasAccess, err := sqlcdb.New(sh.db.DB).CheckSecurityProjectAccess(context.Background(), sqlcdb.CheckSecurityProjectAccessParams{
|
||||||
err := sh.db.QueryRow(
|
ID: uuid.MustParse(projectID),
|
||||||
`SELECT EXISTS (
|
OwnerID: optionalUserUUID(c),
|
||||||
SELECT 1
|
Column3: contextIsAdmin(c),
|
||||||
FROM projects p
|
})
|
||||||
WHERE p.id = $1
|
|
||||||
AND (p.is_approved
|
|
||||||
OR $3::bool
|
|
||||||
OR p.owner_id = $2 OR EXISTS (
|
|
||||||
SELECT 1 FROM project_members pm
|
|
||||||
WHERE pm.project_id = p.id AND pm.user_id = $2
|
|
||||||
))
|
|
||||||
)`,
|
|
||||||
projectID, optionalUserUUID(c), contextIsAdmin(c),
|
|
||||||
).Scan(&hasAccess)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to verify project access"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to verify project access"})
|
||||||
return "", false
|
return "", false
|
||||||
@@ -649,7 +607,8 @@ func (sh *SecurityHandler) requireSecurityScanAccess(c *gin.Context, scanID stri
|
|||||||
}
|
}
|
||||||
|
|
||||||
var projectID string
|
var projectID string
|
||||||
err := sh.db.QueryRow("SELECT project_id FROM security_scans WHERE id = $1", scanID).Scan(&projectID)
|
pid, err := sqlcdb.New(sh.db.DB).GetSecurityScanProjectID(context.Background(), uuid.MustParse(scanID))
|
||||||
|
projectID = pid.String()
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Security scan not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Security scan not found"})
|
||||||
return false
|
return false
|
||||||
@@ -670,7 +629,8 @@ func (sh *SecurityHandler) requireComplianceReportAccess(c *gin.Context, reportI
|
|||||||
}
|
}
|
||||||
|
|
||||||
var projectID string
|
var projectID string
|
||||||
err := sh.db.QueryRow("SELECT project_id FROM compliance_reports WHERE id = $1", reportID).Scan(&projectID)
|
pid, err := sqlcdb.New(sh.db.DB).GetComplianceReportProjectID(context.Background(), uuid.MustParse(reportID))
|
||||||
|
projectID = pid.String()
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Compliance report not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Compliance report not found"})
|
||||||
return false
|
return false
|
||||||
@@ -691,7 +651,8 @@ func (sh *SecurityHandler) requireVulnerabilityAccess(c *gin.Context, vulnID str
|
|||||||
}
|
}
|
||||||
|
|
||||||
var projectID string
|
var projectID string
|
||||||
err := sh.db.QueryRow("SELECT project_id FROM vulnerabilities WHERE id = $1", vulnID).Scan(&projectID)
|
pid, err := sqlcdb.New(sh.db.DB).GetVulnerabilityProjectID(context.Background(), uuid.MustParse(vulnID))
|
||||||
|
projectID = pid.String()
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Vulnerability not found"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Vulnerability not found"})
|
||||||
return "", false
|
return "", false
|
||||||
|
|||||||
@@ -0,0 +1,212 @@
|
|||||||
|
package api
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"containr/internal/database"
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Live invariant test against real Postgres — set METRICS_TEST_DSN (or
|
||||||
|
// DATABASE_URL). Covers the generated auth/security queries: user upsert
|
||||||
|
// conflict path, project-access EXISTS predicate, vulnerability ordering and
|
||||||
|
// status update, latest-scan/report lookups, and the framework list.
|
||||||
|
func TestSecurityLiveInvariants(t *testing.T) {
|
||||||
|
dsn := testDSNString(t)
|
||||||
|
db, err := database.NewConnection(dsn)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("connect: %v", err)
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
ctx := context.Background()
|
||||||
|
q := sqlcdb.New(db.DB)
|
||||||
|
|
||||||
|
userID := uuid.New()
|
||||||
|
otherID := uuid.New()
|
||||||
|
projectID := uuid.New()
|
||||||
|
envID := uuid.New()
|
||||||
|
serviceID := uuid.New()
|
||||||
|
scanID := uuid.New()
|
||||||
|
reportID := uuid.New()
|
||||||
|
frameworkID := uuid.New()
|
||||||
|
vulnCrit := uuid.New()
|
||||||
|
vulnLow := uuid.New()
|
||||||
|
suffix := strings.ReplaceAll(userID.String()[:8], "-", "")
|
||||||
|
must := func(err error) {
|
||||||
|
t.Helper()
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("setup: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_, err = db.Exec(`INSERT INTO users (id, email, password_hash, name) VALUES ($1, $2, 'x', 'test'), ($3, $4, 'x', 'other')`,
|
||||||
|
userID, "sec-"+suffix+"@test.local", otherID, "sec-"+suffix+"[email protected]")
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO projects (id, name, owner_id, is_approved) VALUES ($1, $2, $3, false)`,
|
||||||
|
projectID, "sec-"+suffix, userID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO environments (id, name, project_id) VALUES ($1, 'production', $2)`,
|
||||||
|
envID, projectID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO services (id, name, project_id, environment_id, service_type, source_type)
|
||||||
|
VALUES ($1, $2, $3, $4, 'web', 'image')`,
|
||||||
|
serviceID, "sec-"+suffix, projectID, envID)
|
||||||
|
must(err)
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM users WHERE id IN ($1, $2)`, userID, otherID) }()
|
||||||
|
|
||||||
|
// Auth queries: upsert conflicts update rather than error, counts see rows.
|
||||||
|
_, err = q.InsertUserAdmin(ctx, sqlcdb.InsertUserAdminParams{
|
||||||
|
Email: "ins-" + suffix + "@test.local", PasswordHash: "x", Name: "ins", IsAdmin: false,
|
||||||
|
})
|
||||||
|
must(err)
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM users WHERE email = $1`, "ins-"+suffix+"@test.local") }()
|
||||||
|
urow, err := q.UpsertLocalUser(ctx, sqlcdb.UpsertLocalUserParams{
|
||||||
|
Email: "ins-" + suffix + "@test.local", PasswordHash: "y", Name: "renamed",
|
||||||
|
Column4: "http://img", IsAdmin: false,
|
||||||
|
})
|
||||||
|
must(err)
|
||||||
|
if urow.Name != "renamed" || urow.AvatarUrl != "http://img" {
|
||||||
|
t.Fatalf("upsert: %+v", urow)
|
||||||
|
}
|
||||||
|
byEmail, err := q.GetUserByEmailForAuth(ctx, "ins-"+suffix+"@test.local")
|
||||||
|
must(err)
|
||||||
|
if byEmail.ID != urow.ID || byEmail.PasswordHash != "x" {
|
||||||
|
t.Fatalf("by email: %+v", byEmail)
|
||||||
|
}
|
||||||
|
count, err := q.CountUsersByEmail(ctx, "ins-"+suffix+"@test.local")
|
||||||
|
must(err)
|
||||||
|
if count != 1 {
|
||||||
|
t.Fatalf("count by email: %d", count)
|
||||||
|
}
|
||||||
|
byID, err := q.GetUserByID(ctx, urow.ID)
|
||||||
|
must(err)
|
||||||
|
if byID.Email != "ins-"+suffix+"@test.local" {
|
||||||
|
t.Fatalf("by id: %+v", byID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Access predicate: unapproved project is owner/admin/member only.
|
||||||
|
ownerAccess, err := q.CheckSecurityProjectAccess(ctx, sqlcdb.CheckSecurityProjectAccessParams{
|
||||||
|
ID: projectID, OwnerID: userID, Column3: false,
|
||||||
|
})
|
||||||
|
must(err)
|
||||||
|
if !ownerAccess {
|
||||||
|
t.Fatal("owner should have access")
|
||||||
|
}
|
||||||
|
strangerAccess, err := q.CheckSecurityProjectAccess(ctx, sqlcdb.CheckSecurityProjectAccessParams{
|
||||||
|
ID: projectID, OwnerID: otherID, Column3: false,
|
||||||
|
})
|
||||||
|
must(err)
|
||||||
|
if strangerAccess {
|
||||||
|
t.Fatal("stranger should not have access")
|
||||||
|
}
|
||||||
|
adminAccess, err := q.CheckSecurityProjectAccess(ctx, sqlcdb.CheckSecurityProjectAccessParams{
|
||||||
|
ID: projectID, OwnerID: otherID, Column3: true,
|
||||||
|
})
|
||||||
|
must(err)
|
||||||
|
if !adminAccess {
|
||||||
|
t.Fatal("admin should have access")
|
||||||
|
}
|
||||||
|
ownerText, err := q.GetProjectOwnerText(ctx, projectID)
|
||||||
|
must(err)
|
||||||
|
if ownerText != userID.String() {
|
||||||
|
t.Fatalf("owner text: %q", ownerText)
|
||||||
|
}
|
||||||
|
exists, err := q.ServiceExistsInProject(ctx, sqlcdb.ServiceExistsInProjectParams{
|
||||||
|
ID: serviceID, ProjectID: projectID,
|
||||||
|
})
|
||||||
|
must(err)
|
||||||
|
if !exists {
|
||||||
|
t.Fatal("service should exist in project")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Scan/report/vuln rows via raw inserts (fixtures), reads via sqlc.
|
||||||
|
_, err = db.Exec(`INSERT INTO security_scans (id, project_id, service_id, scan_type, status, summary, started_at)
|
||||||
|
VALUES ($1, $2, $3, 'dependency', 'completed', '{"score": 85}', NOW() - interval '1 hour')`,
|
||||||
|
scanID, projectID, serviceID)
|
||||||
|
must(err)
|
||||||
|
_, err = db.Exec(`INSERT INTO compliance_frameworks (id, name, description, version, enabled)
|
||||||
|
VALUES ($1, $2, 'desc', '1.0', true)`,
|
||||||
|
frameworkID, "fw-"+suffix)
|
||||||
|
must(err)
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM compliance_frameworks WHERE id = $1`, frameworkID) }()
|
||||||
|
_, err = db.Exec(`INSERT INTO compliance_reports (id, project_id, framework_id, overall_status, score, assessment_date)
|
||||||
|
VALUES ($1, $2, $3, 'compliant', 92, NOW() - interval '30 minutes')`,
|
||||||
|
reportID, projectID, frameworkID)
|
||||||
|
must(err)
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM compliance_reports WHERE id = $1`, reportID) }()
|
||||||
|
_, err = db.Exec(`INSERT INTO vulnerabilities (id, type, severity, title, description, service_id, project_id, status, found_at)
|
||||||
|
VALUES ($1, 'dependency', 'low', 'low-v', 'd', $3, $4, 'open', NOW() - interval '2 hours'),
|
||||||
|
($2, 'dependency', 'critical', 'crit-v', 'd', $3, $4, 'open', NOW() - interval '1 hour')`,
|
||||||
|
vulnLow, vulnCrit, serviceID, projectID)
|
||||||
|
must(err)
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM vulnerabilities WHERE project_id = $1`, projectID) }()
|
||||||
|
defer func() { _, _ = db.Exec(`DELETE FROM security_scans WHERE id = $1`, scanID) }()
|
||||||
|
|
||||||
|
// Severity ordering: critical first.
|
||||||
|
vulns, err := q.ListVulnerabilities(ctx, projectID)
|
||||||
|
must(err)
|
||||||
|
if len(vulns) != 2 || vulns[0].ID != vulnCrit {
|
||||||
|
t.Fatalf("vuln ordering: %+v", vulns)
|
||||||
|
}
|
||||||
|
metrics, err := q.GetVulnerabilityMetrics(ctx, projectID)
|
||||||
|
must(err)
|
||||||
|
if metrics.Total != 2 || metrics.Critical != 1 || metrics.Low != 1 || metrics.Open != 2 {
|
||||||
|
t.Fatalf("metrics: %+v", metrics)
|
||||||
|
}
|
||||||
|
must(q.UpdateVulnerabilityStatus(ctx, sqlcdb.UpdateVulnerabilityStatusParams{
|
||||||
|
Status: "resolved",
|
||||||
|
ResolvedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
|
ID: vulnCrit,
|
||||||
|
}))
|
||||||
|
metrics, err = q.GetVulnerabilityMetrics(ctx, projectID)
|
||||||
|
must(err)
|
||||||
|
if metrics.Open != 1 || metrics.Resolved != 1 {
|
||||||
|
t.Fatalf("metrics after resolve: %+v", metrics)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Latest scan/report lookups.
|
||||||
|
scan, err := q.GetLatestSecurityScan(ctx, projectID)
|
||||||
|
must(err)
|
||||||
|
if scan.ID != scanID || scan.Score != 85 || scan.Status != "completed" {
|
||||||
|
t.Fatalf("latest scan: %+v", scan)
|
||||||
|
}
|
||||||
|
report, err := q.GetLatestComplianceReport(ctx, projectID)
|
||||||
|
must(err)
|
||||||
|
if report.OverallStatus != "compliant" || report.Score != 92 {
|
||||||
|
t.Fatalf("latest report: %+v", report)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Project-ID lookups used by the access guards.
|
||||||
|
scanPID, err := q.GetSecurityScanProjectID(ctx, scanID)
|
||||||
|
must(err)
|
||||||
|
reportPID, err := q.GetComplianceReportProjectID(ctx, reportID)
|
||||||
|
must(err)
|
||||||
|
vulnPID, err := q.GetVulnerabilityProjectID(ctx, vulnCrit)
|
||||||
|
must(err)
|
||||||
|
if scanPID != projectID || reportPID != projectID || vulnPID != projectID {
|
||||||
|
t.Fatalf("project id lookups: %v %v %v", scanPID, reportPID, vulnPID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Framework list only returns enabled rows.
|
||||||
|
frameworks, err := q.ListComplianceFrameworks(ctx)
|
||||||
|
must(err)
|
||||||
|
found := false
|
||||||
|
for _, f := range frameworks {
|
||||||
|
if f.ID == frameworkID && f.Enabled {
|
||||||
|
found = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !found {
|
||||||
|
t.Fatal("enabled framework missing from list")
|
||||||
|
}
|
||||||
|
frameworkExists, err := q.ComplianceFrameworkExists(ctx, frameworkID)
|
||||||
|
must(err)
|
||||||
|
if !frameworkExists {
|
||||||
|
t.Fatal("framework should exist")
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,11 +1,14 @@
|
|||||||
package api
|
package api
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"containr/internal/database"
|
"containr/internal/database"
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
@@ -53,11 +56,8 @@ func handleCloneService(c *gin.Context) {
|
|||||||
if name == "" {
|
if name == "" {
|
||||||
name = source.Name + "-copy"
|
name = source.Name + "-copy"
|
||||||
}
|
}
|
||||||
var count int
|
count, _ := sqlcdb.New(db.(*database.DB).DB).CountServicesByName(
|
||||||
_ = db.(*database.DB).QueryRow(
|
c.Request.Context(), sqlcdb.CountServicesByNameParams{ProjectID: targetProjectID, Name: name})
|
||||||
`SELECT COUNT(*) FROM services WHERE project_id = $1 AND name = $2`,
|
|
||||||
targetProjectID, name,
|
|
||||||
).Scan(&count)
|
|
||||||
if count > 0 {
|
if count > 0 {
|
||||||
c.JSON(http.StatusConflict, gin.H{"error": "Service name already exists in target project", "code": "CONFLICT"})
|
c.JSON(http.StatusConflict, gin.H{"error": "Service name already exists in target project", "code": "CONFLICT"})
|
||||||
return
|
return
|
||||||
@@ -84,8 +84,11 @@ func handleCloneService(c *gin.Context) {
|
|||||||
// inherits the source's. Used by the clone endpoint and preview
|
// inherits the source's. Used by the clone endpoint and preview
|
||||||
// environments, which additionally override git_branch and domains.
|
// environments, which additionally override git_branch and domains.
|
||||||
func cloneServiceRow(db *database.DB, sourceID uuid.UUID, targetProjectID uuid.UUID, name, environment string) (uuid.UUID, error) {
|
func cloneServiceRow(db *database.DB, sourceID uuid.UUID, targetProjectID uuid.UUID, name, environment string) (uuid.UUID, error) {
|
||||||
var sourceEnv string
|
ctx := context.Background()
|
||||||
if err := db.QueryRow(`SELECT COALESCE(environment, '') FROM services WHERE id = $1`, sourceID).Scan(&sourceEnv); err != nil {
|
q := sqlcdb.New(db.DB)
|
||||||
|
|
||||||
|
sourceEnv, err := q.GetServiceEnvironment(ctx, sourceID)
|
||||||
|
if err != nil {
|
||||||
return uuid.Nil, err
|
return uuid.Nil, err
|
||||||
}
|
}
|
||||||
if environment == "" {
|
if environment == "" {
|
||||||
@@ -102,46 +105,31 @@ func cloneServiceRow(db *database.DB, sourceID uuid.UUID, targetProjectID uuid.U
|
|||||||
maintenance, basicAuth := serviceAccess(db, sourceID)
|
maintenance, basicAuth := serviceAccess(db, sourceID)
|
||||||
newID := uuid.New()
|
newID := uuid.New()
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
// string() so pq sends text — []byte would go as bytea and fail ::jsonb.
|
|
||||||
volumesJSON := string(loadServiceVolumesJSON(db, sourceID))
|
|
||||||
|
|
||||||
_, err = db.Exec(
|
err = q.CloneServiceRow(ctx, sqlcdb.CloneServiceRowParams{
|
||||||
`INSERT INTO services
|
ID: newID,
|
||||||
(id, project_id, name, environment_id, service_type, source_type, source_url, image_name,
|
ProjectID: targetProjectID,
|
||||||
build_command, start_command, type, status, image, command, environment,
|
Name: name,
|
||||||
git_repo, git_branch, build_path, cpu, memory, replicas, port, domain,
|
EnvironmentID: environmentID,
|
||||||
healthcheck_path, restart_policy, volumes, maintenance_mode, basic_auth_users,
|
Environment: sql.NullString{String: environment, Valid: true},
|
||||||
builder, cpu_reserve, memory_reserve, static_build_cmd, static_dir,
|
MaintenanceMode: maintenance,
|
||||||
created_at, updated_at)
|
BasicAuthUsers: basicAuth,
|
||||||
SELECT $1, $2, $3, $4, service_type, source_type, source_url, image_name,
|
CreatedAt: sql.NullTime{Time: now, Valid: true},
|
||||||
build_command, start_command, type, 'stopped', image, command, $5,
|
ID_2: sourceID,
|
||||||
git_repo, git_branch, build_path, cpu, memory, replicas, port, domain,
|
})
|
||||||
healthcheck_path, restart_policy, $6::jsonb, $7, $8,
|
|
||||||
builder, cpu_reserve, memory_reserve, static_build_cmd, static_dir,
|
|
||||||
$9, $9
|
|
||||||
FROM services WHERE id = $10`,
|
|
||||||
newID, targetProjectID, name, environmentID, environment, volumesJSON,
|
|
||||||
maintenance, basicAuth, now, sourceID,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return uuid.Nil, err
|
return uuid.Nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if _, err := db.Exec(
|
if err := q.CloneServiceDomains(ctx, sqlcdb.CloneServiceDomainsParams{
|
||||||
`INSERT INTO service_domains (service_id, domain, is_default, cert_type)
|
ServiceID: newID, ServiceID_2: sourceID}); err != nil {
|
||||||
SELECT $1, domain, is_default, cert_type FROM service_domains WHERE service_id = $2`,
|
_ = q.DeleteServiceByID(ctx, newID)
|
||||||
newID, sourceID,
|
|
||||||
); err != nil {
|
|
||||||
_, _ = db.Exec(`DELETE FROM services WHERE id = $1`, newID)
|
|
||||||
return uuid.Nil, err
|
return uuid.Nil, err
|
||||||
}
|
}
|
||||||
// Variable values are copied as stored — ciphertext stays ciphertext.
|
// Variable values are copied as stored — ciphertext stays ciphertext.
|
||||||
if _, err := db.Exec(
|
if err := q.CloneServiceVariables(ctx, sqlcdb.CloneServiceVariablesParams{
|
||||||
`INSERT INTO environment_variables (id, service_id, key, value, is_secret)
|
ServiceID: newID, ServiceID_2: sourceID}); err != nil {
|
||||||
SELECT gen_random_uuid(), $1, key, value, is_secret FROM environment_variables WHERE service_id = $2`,
|
_ = q.DeleteServiceByID(ctx, newID)
|
||||||
newID, sourceID,
|
|
||||||
); err != nil {
|
|
||||||
_, _ = db.Exec(`DELETE FROM services WHERE id = $1`, newID)
|
|
||||||
return uuid.Nil, err
|
return uuid.Nil, err
|
||||||
}
|
}
|
||||||
return newID, nil
|
return newID, nil
|
||||||
@@ -187,11 +175,8 @@ func handleMoveService(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var count int
|
count, _ := sqlcdb.New(db.(*database.DB).DB).CountServicesByName(
|
||||||
_ = db.(*database.DB).QueryRow(
|
c.Request.Context(), sqlcdb.CountServicesByNameParams{ProjectID: req.ProjectID, Name: service.Name})
|
||||||
`SELECT COUNT(*) FROM services WHERE project_id = $1 AND name = $2`,
|
|
||||||
req.ProjectID, service.Name,
|
|
||||||
).Scan(&count)
|
|
||||||
if count > 0 {
|
if count > 0 {
|
||||||
c.JSON(http.StatusConflict, gin.H{"error": "A service with this name already exists in the target project", "code": "CONFLICT"})
|
c.JSON(http.StatusConflict, gin.H{"error": "A service with this name already exists in the target project", "code": "CONFLICT"})
|
||||||
return
|
return
|
||||||
@@ -207,15 +192,18 @@ func handleMoveService(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
res, err := db.(*database.DB).Exec(
|
n, err := sqlcdb.New(db.(*database.DB).DB).MoveServiceProject(
|
||||||
`UPDATE services SET project_id = $1, environment_id = $2, updated_at = $3 WHERE id = $4`,
|
c.Request.Context(), sqlcdb.MoveServiceProjectParams{
|
||||||
req.ProjectID, environmentID, time.Now(), serviceID,
|
ProjectID: req.ProjectID,
|
||||||
)
|
EnvironmentID: environmentID,
|
||||||
|
UpdatedAt: sql.NullTime{Time: time.Now(), Valid: true},
|
||||||
|
ID: serviceID,
|
||||||
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to move service"})
|
c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to move service"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if n, _ := res.RowsAffected(); n == 0 {
|
if n == 0 {
|
||||||
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found", "code": "NOT_FOUND"})
|
c.JSON(http.StatusNotFound, gin.H{"error": "Service not found", "code": "NOT_FOUND"})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -229,16 +217,10 @@ func ownsProject(db *database.DB, projectID uuid.UUID, userID string, isAdmin bo
|
|||||||
if isAdmin {
|
if isAdmin {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
var owner string
|
uid, err := uuid.Parse(userID)
|
||||||
err := db.QueryRow(`SELECT owner_id FROM projects WHERE id = $1`, projectID).Scan(&owner)
|
if err != nil {
|
||||||
return err == nil && owner == userID
|
return false
|
||||||
}
|
|
||||||
|
|
||||||
// loadServiceVolumesJSON returns the raw stored mounts for a verbatim copy.
|
|
||||||
func loadServiceVolumesJSON(db *database.DB, serviceID uuid.UUID) []byte {
|
|
||||||
var raw []byte
|
|
||||||
if err := db.QueryRow(`SELECT volumes FROM services WHERE id = $1`, serviceID).Scan(&raw); err != nil || len(raw) == 0 {
|
|
||||||
return []byte("[]")
|
|
||||||
}
|
}
|
||||||
return raw
|
owner, err := sqlcdb.New(db.DB).GetProjectOwner(context.Background(), projectID)
|
||||||
|
return err == nil && owner == uid
|
||||||
}
|
}
|
||||||
@@ -86,20 +86,17 @@ type sleepCandidate struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func sweepSleeping(ctx context.Context, db *database.DB, dc *docker.Client, engine *deployment.DeploymentEngine, apiPort int) {
|
func sweepSleeping(ctx context.Context, db *database.DB, dc *docker.Client, engine *deployment.DeploymentEngine, apiPort int) {
|
||||||
rows, err := db.QueryContext(ctx,
|
sweepRows, err := sqlcdb.New(db.DB).ListSleepCandidates(ctx)
|
||||||
`SELECT id, project_id, name, COALESCE(status,''), COALESCE(sleep_idle_minutes,15)
|
|
||||||
FROM services WHERE sleep_enabled AND COALESCE(status,'') IN ('running','deployed','degraded','starting')`)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
var candidates []sleepCandidate
|
var candidates []sleepCandidate
|
||||||
for rows.Next() {
|
for _, r := range sweepRows {
|
||||||
var s sleepCandidate
|
candidates = append(candidates, sleepCandidate{
|
||||||
if err := rows.Scan(&s.ID, &s.ProjectID, &s.Name, &s.Status, &s.IdleMinutes); err == nil {
|
ID: r.ID, ProjectID: r.ProjectID, Name: r.Name,
|
||||||
candidates = append(candidates, s)
|
Status: r.Status, IdleMinutes: int(r.SleepIdleMinutes),
|
||||||
}
|
})
|
||||||
}
|
}
|
||||||
rows.Close()
|
|
||||||
|
|
||||||
q := sqlcdb.New(db.DB)
|
q := sqlcdb.New(db.DB)
|
||||||
for _, svc := range candidates {
|
for _, svc := range candidates {
|
||||||
@@ -187,11 +184,10 @@ func putToSleepRemote(ctx context.Context, db *database.DB, q *sqlcdb.Queries, s
|
|||||||
tracker.mu.Lock()
|
tracker.mu.Lock()
|
||||||
delete(tracker.activity, svc.ID)
|
delete(tracker.activity, svc.ID)
|
||||||
tracker.mu.Unlock()
|
tracker.mu.Unlock()
|
||||||
_, _ = db.Exec(`UPDATE services SET status = 'sleeping', updated_at = $1 WHERE id = $2`, time.Now(), svc.ID)
|
setServiceStatusNow(db, svc.ID, "sleeping")
|
||||||
log.Printf("sleep: %s (%s) remote replicas removed after idle timeout", svc.Name, svc.ID)
|
log.Printf("sleep: %s (%s) remote replicas removed after idle timeout", svc.Name, svc.ID)
|
||||||
|
|
||||||
var domain string
|
domain, _ := q.GetServiceDomainText(ctx, svc.ID)
|
||||||
_ = db.QueryRow(`SELECT COALESCE(domain,'') FROM services WHERE id = $1`, svc.ID).Scan(&domain)
|
|
||||||
domains := serviceDomainNames(db, svc.ID, domain)
|
domains := serviceDomainNames(db, svc.ID, domain)
|
||||||
if len(domains) > 0 {
|
if len(domains) > 0 {
|
||||||
writeWakeIngress(svc.ID.String(), domains, apiPort)
|
writeWakeIngress(svc.ID.String(), domains, apiPort)
|
||||||
@@ -250,11 +246,10 @@ func putToSleep(ctx context.Context, db *database.DB, dc *docker.Client, engine
|
|||||||
tracker.mu.Lock()
|
tracker.mu.Lock()
|
||||||
delete(tracker.activity, svc.ID)
|
delete(tracker.activity, svc.ID)
|
||||||
tracker.mu.Unlock()
|
tracker.mu.Unlock()
|
||||||
_, _ = db.Exec(`UPDATE services SET status = 'sleeping', updated_at = $1 WHERE id = $2`, time.Now(), svc.ID)
|
setServiceStatusNow(db, svc.ID, "sleeping")
|
||||||
log.Printf("sleep: %s (%s) scaled to zero after idle timeout", svc.Name, svc.ID)
|
log.Printf("sleep: %s (%s) scaled to zero after idle timeout", svc.Name, svc.ID)
|
||||||
|
|
||||||
var domain string
|
domain, _ := sqlcdb.New(db.DB).GetServiceDomainText(ctx, svc.ID)
|
||||||
_ = db.QueryRow(`SELECT COALESCE(domain,'') FROM services WHERE id = $1`, svc.ID).Scan(&domain)
|
|
||||||
domains := serviceDomainNames(db, svc.ID, domain)
|
domains := serviceDomainNames(db, svc.ID, domain)
|
||||||
if len(domains) == 0 {
|
if len(domains) == 0 {
|
||||||
return
|
return
|
||||||
@@ -444,7 +439,9 @@ func pruneWakeContainers(ctx context.Context, db *database.DB, dc *docker.Client
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
var status string
|
var status string
|
||||||
_ = db.QueryRow(`SELECT COALESCE(status,'') FROM services WHERE id = $1`, svcID).Scan(&status)
|
if sid, err := uuid.Parse(svcID); err == nil {
|
||||||
|
status, _ = sqlcdb.New(db.DB).GetServiceStatusText(ctx, sid)
|
||||||
|
}
|
||||||
if live > 0 || status == "stopped" || status == "" {
|
if live > 0 || status == "stopped" || status == "" {
|
||||||
_ = dc.RemoveContainer(ctx, c.ID, true)
|
_ = dc.RemoveContainer(ctx, c.ID, true)
|
||||||
}
|
}
|
||||||
@@ -491,34 +488,11 @@ func triggerServiceWake(c *gin.Context) (int, gin.H) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return http.StatusBadRequest, gin.H{"error": "invalid service id"}
|
return http.StatusBadRequest, gin.H{"error": "invalid service id"}
|
||||||
}
|
}
|
||||||
var service Service
|
row, err := sqlcdb.New(db.DB).GetServiceRuntimeWithOwner(context.Background(), id)
|
||||||
err = db.QueryRow(
|
|
||||||
`SELECT s.id, s.project_id, s.name,
|
|
||||||
COALESCE(s.type, s.service_type, ''), COALESCE(s.status, ''),
|
|
||||||
COALESCE(s.image, s.image_name, ''), COALESCE(s.command, s.start_command, ''),
|
|
||||||
COALESCE(s.environment, ''), COALESCE(s.git_repo, s.source_url, ''),
|
|
||||||
COALESCE(s.git_branch, ''), COALESCE(s.build_path, ''),
|
|
||||||
COALESCE(s.cpu, ''), COALESCE(s.memory, ''),
|
|
||||||
COALESCE(s.replicas, 1), COALESCE(s.port, 0),
|
|
||||||
COALESCE(s.domain, ''), COALESCE(s.healthcheck_path, ''),
|
|
||||||
COALESCE(s.restart_policy, 'unless-stopped'),
|
|
||||||
COALESCE(s.builder, 'auto'), COALESCE(s.cpu_reserve, ''),
|
|
||||||
COALESCE(s.memory_reserve, ''), COALESCE(s.static_build_cmd, ''),
|
|
||||||
COALESCE(s.static_dir, ''),
|
|
||||||
s.created_at, s.updated_at
|
|
||||||
FROM services s WHERE s.id = $1`, id,
|
|
||||||
).Scan(
|
|
||||||
&service.ID, &service.ProjectID, &service.Name, &service.Type, &service.Status,
|
|
||||||
&service.Image, &service.Command, &service.Environment, &service.GitRepo,
|
|
||||||
&service.GitBranch, &service.BuildPath, &service.CPU, &service.Memory,
|
|
||||||
&service.Replicas, &service.Port, &service.Domain, &service.HealthCheckPath,
|
|
||||||
&service.RestartPolicy, &service.Builder, &service.CPUReserve,
|
|
||||||
&service.MemoryReserve, &service.StaticBuildCmd, &service.StaticDir,
|
|
||||||
&service.CreatedAt, &service.UpdatedAt,
|
|
||||||
)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return http.StatusNotFound, gin.H{"error": "service not found"}
|
return http.StatusNotFound, gin.H{"error": "service not found"}
|
||||||
}
|
}
|
||||||
|
service := serviceFromRuntimeRow(row)
|
||||||
if service.Status == "running" || service.Status == "deployed" {
|
if service.Status == "running" || service.Status == "deployed" {
|
||||||
return http.StatusOK, gin.H{"status": service.Status}
|
return http.StatusOK, gin.H{"status": service.Status}
|
||||||
}
|
}
|
||||||
@@ -528,13 +502,13 @@ func triggerServiceWake(c *gin.Context) (int, gin.H) {
|
|||||||
spec, err := serviceRuntimeSpec(db, service)
|
spec, err := serviceRuntimeSpec(db, service)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("wake: runtime spec for %s failed: %v", service.ID, err)
|
log.Printf("wake: runtime spec for %s failed: %v", service.ID, err)
|
||||||
_, _ = db.Exec(`UPDATE services SET status = 'failed', updated_at = $1 WHERE id = $2`, time.Now(), service.ID)
|
setServiceStatusNow(db, service.ID, "failed")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
state, err := engine.ReconcileService(ctx, spec)
|
state, err := engine.ReconcileService(ctx, spec)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("wake: reconcile for %s failed: %v", service.ID, err)
|
log.Printf("wake: reconcile for %s failed: %v", service.ID, err)
|
||||||
_, _ = db.Exec(`UPDATE services SET status = 'failed', updated_at = $1 WHERE id = $2`, time.Now(), service.ID)
|
setServiceStatusNow(db, service.ID, "failed")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
status := "running"
|
status := "running"
|
||||||
@@ -542,7 +516,7 @@ func triggerServiceWake(c *gin.Context) (int, gin.H) {
|
|||||||
status = state.Status
|
status = state.Status
|
||||||
persistPublishedPort(db, service.ID, state.Ports)
|
persistPublishedPort(db, service.ID, state.Ports)
|
||||||
}
|
}
|
||||||
_, _ = db.Exec(`UPDATE services SET status = $1, updated_at = $2 WHERE id = $3`, status, time.Now(), service.ID)
|
setServiceStatusNow(db, service.ID, status)
|
||||||
}()
|
}()
|
||||||
return http.StatusAccepted, gin.H{"status": "waking"}
|
return http.StatusAccepted, gin.H{"status": "waking"}
|
||||||
}
|
}
|
||||||
@@ -563,8 +537,6 @@ func handleServiceSleep(c *gin.Context) {
|
|||||||
if !ok {
|
if !ok {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
var port int
|
|
||||||
_ = db.QueryRow(`SELECT COALESCE(port,0) FROM services WHERE id = $1`, service.ID).Scan(&port)
|
|
||||||
svc := sleepCandidate{
|
svc := sleepCandidate{
|
||||||
ID: service.ID,
|
ID: service.ID,
|
||||||
ProjectID: service.ProjectID,
|
ProjectID: service.ProjectID,
|
||||||
|
|||||||
@@ -31,3 +31,33 @@ func TestWebhookEventMatches(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestValidWebhookURL(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
name string
|
||||||
|
url string
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{"https", "https://hooks.example.com/wh/abc", false},
|
||||||
|
{"http", "http://internal.lan:8080/notify", false},
|
||||||
|
{"whitespace trimmed", " https://example.com/hook ", false},
|
||||||
|
{"with query", "https://example.com/hook?token=x", false},
|
||||||
|
{"ftp rejected", "ftp://files.example.com/x", true},
|
||||||
|
{"no scheme", "example.com/hook", true},
|
||||||
|
{"bare scheme", "https://", true},
|
||||||
|
{"empty", "", true},
|
||||||
|
{"javascript rejected", "javascript:alert(1)", true},
|
||||||
|
{"host empty", "http:///path-only", true},
|
||||||
|
}
|
||||||
|
for _, tc := range cases {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
err := validWebhookURL(tc.url)
|
||||||
|
if tc.wantErr && err == nil {
|
||||||
|
t.Errorf("validWebhookURL(%q) = nil, want error", tc.url)
|
||||||
|
}
|
||||||
|
if !tc.wantErr && err != nil {
|
||||||
|
t.Errorf("validWebhookURL(%q) = %v, want nil", tc.url, err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,233 @@
|
|||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.31.1
|
||||||
|
// source: auth.sql
|
||||||
|
|
||||||
|
package sqlcdb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
const countUsers = `-- name: CountUsers :one
|
||||||
|
SELECT COUNT(*) FROM users
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) CountUsers(ctx context.Context) (int64, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, countUsers)
|
||||||
|
var count int64
|
||||||
|
err := row.Scan(&count)
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const countUsersByEmail = `-- name: CountUsersByEmail :one
|
||||||
|
SELECT COUNT(*) FROM users WHERE email = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) CountUsersByEmail(ctx context.Context, email string) (int64, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, countUsersByEmail, email)
|
||||||
|
var count int64
|
||||||
|
err := row.Scan(&count)
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getUserByEmailForAuth = `-- name: GetUserByEmailForAuth :one
|
||||||
|
SELECT id, email, password_hash, name, COALESCE(avatar_url, '') AS avatar_url, is_admin, created_at
|
||||||
|
FROM users WHERE email = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetUserByEmailForAuthRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Email string `json:"email"`
|
||||||
|
PasswordHash string `json:"password_hash"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
AvatarUrl string `json:"avatar_url"`
|
||||||
|
IsAdmin bool `json:"is_admin"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetUserByEmailForAuth(ctx context.Context, email string) (GetUserByEmailForAuthRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getUserByEmailForAuth, email)
|
||||||
|
var i GetUserByEmailForAuthRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.Email,
|
||||||
|
&i.PasswordHash,
|
||||||
|
&i.Name,
|
||||||
|
&i.AvatarUrl,
|
||||||
|
&i.IsAdmin,
|
||||||
|
&i.CreatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getUserByID = `-- name: GetUserByID :one
|
||||||
|
SELECT id, email, name, COALESCE(avatar_url, '') AS avatar_url, is_admin, created_at
|
||||||
|
FROM users WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetUserByIDRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Email string `json:"email"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
AvatarUrl string `json:"avatar_url"`
|
||||||
|
IsAdmin bool `json:"is_admin"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetUserByID(ctx context.Context, id uuid.UUID) (GetUserByIDRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getUserByID, id)
|
||||||
|
var i GetUserByIDRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.Email,
|
||||||
|
&i.Name,
|
||||||
|
&i.AvatarUrl,
|
||||||
|
&i.IsAdmin,
|
||||||
|
&i.CreatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const insertUser = `-- name: InsertUser :one
|
||||||
|
INSERT INTO users (email, password_hash, name)
|
||||||
|
VALUES ($1, $2, $3)
|
||||||
|
RETURNING id, email, name, COALESCE(avatar_url, '') AS avatar_url, is_admin, created_at
|
||||||
|
`
|
||||||
|
|
||||||
|
type InsertUserParams struct {
|
||||||
|
Email string `json:"email"`
|
||||||
|
PasswordHash string `json:"password_hash"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type InsertUserRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Email string `json:"email"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
AvatarUrl string `json:"avatar_url"`
|
||||||
|
IsAdmin bool `json:"is_admin"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) InsertUser(ctx context.Context, arg InsertUserParams) (InsertUserRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, insertUser, arg.Email, arg.PasswordHash, arg.Name)
|
||||||
|
var i InsertUserRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.Email,
|
||||||
|
&i.Name,
|
||||||
|
&i.AvatarUrl,
|
||||||
|
&i.IsAdmin,
|
||||||
|
&i.CreatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const insertUserAdmin = `-- name: InsertUserAdmin :one
|
||||||
|
INSERT INTO users (email, password_hash, name, is_admin)
|
||||||
|
VALUES ($1, $2, $3, $4)
|
||||||
|
RETURNING id, email, name, COALESCE(avatar_url, '') AS avatar_url, is_admin, created_at
|
||||||
|
`
|
||||||
|
|
||||||
|
type InsertUserAdminParams struct {
|
||||||
|
Email string `json:"email"`
|
||||||
|
PasswordHash string `json:"password_hash"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
IsAdmin bool `json:"is_admin"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type InsertUserAdminRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Email string `json:"email"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
AvatarUrl string `json:"avatar_url"`
|
||||||
|
IsAdmin bool `json:"is_admin"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) InsertUserAdmin(ctx context.Context, arg InsertUserAdminParams) (InsertUserAdminRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, insertUserAdmin,
|
||||||
|
arg.Email,
|
||||||
|
arg.PasswordHash,
|
||||||
|
arg.Name,
|
||||||
|
arg.IsAdmin,
|
||||||
|
)
|
||||||
|
var i InsertUserAdminRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.Email,
|
||||||
|
&i.Name,
|
||||||
|
&i.AvatarUrl,
|
||||||
|
&i.IsAdmin,
|
||||||
|
&i.CreatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const updateUserProfile = `-- name: UpdateUserProfile :exec
|
||||||
|
UPDATE users
|
||||||
|
SET name = COALESCE($1::varchar, name), avatar_url = COALESCE($2::varchar, avatar_url)
|
||||||
|
WHERE id = $3
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpdateUserProfileParams struct {
|
||||||
|
Column1 string `json:"column_1"`
|
||||||
|
Column2 string `json:"column_2"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpdateUserProfile(ctx context.Context, arg UpdateUserProfileParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, updateUserProfile, arg.Column1, arg.Column2, arg.ID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const upsertLocalUser = `-- name: UpsertLocalUser :one
|
||||||
|
INSERT INTO users (email, password_hash, name, avatar_url, is_admin)
|
||||||
|
VALUES ($1, $2, $3, NULLIF($4::varchar, ''), $5)
|
||||||
|
ON CONFLICT (email) DO UPDATE
|
||||||
|
SET name = EXCLUDED.name,
|
||||||
|
avatar_url = COALESCE(EXCLUDED.avatar_url, users.avatar_url),
|
||||||
|
updated_at = NOW()
|
||||||
|
RETURNING id, email, name, COALESCE(avatar_url, '') AS avatar_url, is_admin, created_at
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpsertLocalUserParams struct {
|
||||||
|
Email string `json:"email"`
|
||||||
|
PasswordHash string `json:"password_hash"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Column4 string `json:"column_4"`
|
||||||
|
IsAdmin bool `json:"is_admin"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpsertLocalUserRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Email string `json:"email"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
AvatarUrl string `json:"avatar_url"`
|
||||||
|
IsAdmin bool `json:"is_admin"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpsertLocalUser(ctx context.Context, arg UpsertLocalUserParams) (UpsertLocalUserRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, upsertLocalUser,
|
||||||
|
arg.Email,
|
||||||
|
arg.PasswordHash,
|
||||||
|
arg.Name,
|
||||||
|
arg.Column4,
|
||||||
|
arg.IsAdmin,
|
||||||
|
)
|
||||||
|
var i UpsertLocalUserRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.Email,
|
||||||
|
&i.Name,
|
||||||
|
&i.AvatarUrl,
|
||||||
|
&i.IsAdmin,
|
||||||
|
&i.CreatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
@@ -0,0 +1,332 @@
|
|||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.31.1
|
||||||
|
// source: cron.sql
|
||||||
|
|
||||||
|
package sqlcdb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
const deleteCronJob = `-- name: DeleteCronJob :exec
|
||||||
|
DELETE FROM cron_jobs WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) DeleteCronJob(ctx context.Context, id uuid.UUID) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, deleteCronJob, id)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const finishCronExecution = `-- name: FinishCronExecution :exec
|
||||||
|
UPDATE cron_executions SET finished_at = $1, status = $2, output = $3, error = $4 WHERE id = $5
|
||||||
|
`
|
||||||
|
|
||||||
|
type FinishCronExecutionParams struct {
|
||||||
|
FinishedAt sql.NullTime `json:"finished_at"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
Output sql.NullString `json:"output"`
|
||||||
|
Error sql.NullString `json:"error"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) FinishCronExecution(ctx context.Context, arg FinishCronExecutionParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, finishCronExecution,
|
||||||
|
arg.FinishedAt,
|
||||||
|
arg.Status,
|
||||||
|
arg.Output,
|
||||||
|
arg.Error,
|
||||||
|
arg.ID,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getCronJob = `-- name: GetCronJob :one
|
||||||
|
SELECT id, project_id, service_id, name, schedule, command, timezone,
|
||||||
|
enabled, last_run_at, next_run_at, last_status, last_output,
|
||||||
|
retention, created_at, updated_at
|
||||||
|
FROM cron_jobs WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetCronJob(ctx context.Context, id uuid.UUID) (CronJob, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getCronJob, id)
|
||||||
|
var i CronJob
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.Name,
|
||||||
|
&i.Schedule,
|
||||||
|
&i.Command,
|
||||||
|
&i.Timezone,
|
||||||
|
&i.Enabled,
|
||||||
|
&i.LastRunAt,
|
||||||
|
&i.NextRunAt,
|
||||||
|
&i.LastStatus,
|
||||||
|
&i.LastOutput,
|
||||||
|
&i.Retention,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getCronJobForTrigger = `-- name: GetCronJobForTrigger :one
|
||||||
|
SELECT cj.service_id, cj.command, cj.schedule, cj.timezone, cj.retention, p.owner_id
|
||||||
|
FROM cron_jobs cj
|
||||||
|
JOIN projects p ON cj.project_id = p.id
|
||||||
|
WHERE cj.id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetCronJobForTriggerRow struct {
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
Command string `json:"command"`
|
||||||
|
Schedule string `json:"schedule"`
|
||||||
|
Timezone sql.NullString `json:"timezone"`
|
||||||
|
Retention sql.NullInt32 `json:"retention"`
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetCronJobForTrigger(ctx context.Context, id uuid.UUID) (GetCronJobForTriggerRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getCronJobForTrigger, id)
|
||||||
|
var i GetCronJobForTriggerRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.Command,
|
||||||
|
&i.Schedule,
|
||||||
|
&i.Timezone,
|
||||||
|
&i.Retention,
|
||||||
|
&i.OwnerID,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getCronJobOwner = `-- name: GetCronJobOwner :one
|
||||||
|
SELECT p.owner_id FROM cron_jobs cj
|
||||||
|
JOIN projects p ON cj.project_id = p.id
|
||||||
|
WHERE cj.id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetCronJobOwner(ctx context.Context, id uuid.UUID) (uuid.UUID, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getCronJobOwner, id)
|
||||||
|
var owner_id uuid.UUID
|
||||||
|
err := row.Scan(&owner_id)
|
||||||
|
return owner_id, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getCronJobProjectID = `-- name: GetCronJobProjectID :one
|
||||||
|
SELECT project_id FROM cron_jobs WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetCronJobProjectID(ctx context.Context, id uuid.UUID) (uuid.UUID, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getCronJobProjectID, id)
|
||||||
|
var project_id uuid.UUID
|
||||||
|
err := row.Scan(&project_id)
|
||||||
|
return project_id, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getCronOwnerViaService = `-- name: GetCronOwnerViaService :one
|
||||||
|
SELECT p.owner_id FROM projects p
|
||||||
|
JOIN services s ON s.project_id = p.id
|
||||||
|
WHERE s.id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetCronOwnerViaService(ctx context.Context, id uuid.UUID) (uuid.UUID, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getCronOwnerViaService, id)
|
||||||
|
var owner_id uuid.UUID
|
||||||
|
err := row.Scan(&owner_id)
|
||||||
|
return owner_id, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const insertCronExecution = `-- name: InsertCronExecution :exec
|
||||||
|
INSERT INTO cron_executions (id, cron_job_id, started_at, status)
|
||||||
|
VALUES ($1, $2, $3, $4)
|
||||||
|
`
|
||||||
|
|
||||||
|
type InsertCronExecutionParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
CronJobID uuid.UUID `json:"cron_job_id"`
|
||||||
|
StartedAt time.Time `json:"started_at"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) InsertCronExecution(ctx context.Context, arg InsertCronExecutionParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, insertCronExecution,
|
||||||
|
arg.ID,
|
||||||
|
arg.CronJobID,
|
||||||
|
arg.StartedAt,
|
||||||
|
arg.Status,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const insertCronJob = `-- name: InsertCronJob :exec
|
||||||
|
INSERT INTO cron_jobs (id, project_id, service_id, name, schedule, command, timezone, enabled, next_run_at, retention, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)
|
||||||
|
`
|
||||||
|
|
||||||
|
type InsertCronJobParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Schedule string `json:"schedule"`
|
||||||
|
Command string `json:"command"`
|
||||||
|
Timezone sql.NullString `json:"timezone"`
|
||||||
|
Enabled sql.NullBool `json:"enabled"`
|
||||||
|
NextRunAt sql.NullTime `json:"next_run_at"`
|
||||||
|
Retention sql.NullInt32 `json:"retention"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) InsertCronJob(ctx context.Context, arg InsertCronJobParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, insertCronJob,
|
||||||
|
arg.ID,
|
||||||
|
arg.ProjectID,
|
||||||
|
arg.ServiceID,
|
||||||
|
arg.Name,
|
||||||
|
arg.Schedule,
|
||||||
|
arg.Command,
|
||||||
|
arg.Timezone,
|
||||||
|
arg.Enabled,
|
||||||
|
arg.NextRunAt,
|
||||||
|
arg.Retention,
|
||||||
|
arg.CreatedAt,
|
||||||
|
arg.UpdatedAt,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const listCronExecutions = `-- name: ListCronExecutions :many
|
||||||
|
SELECT id, cron_job_id, started_at, finished_at, status, output, error
|
||||||
|
FROM cron_executions
|
||||||
|
WHERE cron_job_id = $1
|
||||||
|
ORDER BY started_at DESC
|
||||||
|
LIMIT 100
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) ListCronExecutions(ctx context.Context, cronJobID uuid.UUID) ([]CronExecution, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listCronExecutions, cronJobID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []CronExecution{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i CronExecution
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.CronJobID,
|
||||||
|
&i.StartedAt,
|
||||||
|
&i.FinishedAt,
|
||||||
|
&i.Status,
|
||||||
|
&i.Output,
|
||||||
|
&i.Error,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listDueCronJobs = `-- name: ListDueCronJobs :many
|
||||||
|
SELECT id, project_id, service_id, name, schedule, command, timezone, enabled, retention
|
||||||
|
FROM cron_jobs
|
||||||
|
WHERE enabled = TRUE AND next_run_at IS NOT NULL AND next_run_at <= NOW()
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListDueCronJobsRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Schedule string `json:"schedule"`
|
||||||
|
Command string `json:"command"`
|
||||||
|
Timezone sql.NullString `json:"timezone"`
|
||||||
|
Enabled sql.NullBool `json:"enabled"`
|
||||||
|
Retention sql.NullInt32 `json:"retention"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListDueCronJobs(ctx context.Context) ([]ListDueCronJobsRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listDueCronJobs)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListDueCronJobsRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListDueCronJobsRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.Name,
|
||||||
|
&i.Schedule,
|
||||||
|
&i.Command,
|
||||||
|
&i.Timezone,
|
||||||
|
&i.Enabled,
|
||||||
|
&i.Retention,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const trimCronExecutions = `-- name: TrimCronExecutions :exec
|
||||||
|
DELETE FROM cron_executions WHERE cron_executions.cron_job_id = $1 AND cron_executions.id NOT IN (
|
||||||
|
SELECT ce.id FROM cron_executions ce WHERE ce.cron_job_id = $1
|
||||||
|
ORDER BY ce.started_at DESC LIMIT $2)
|
||||||
|
`
|
||||||
|
|
||||||
|
type TrimCronExecutionsParams struct {
|
||||||
|
CronJobID uuid.UUID `json:"cron_job_id"`
|
||||||
|
Limit int32 `json:"limit"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) TrimCronExecutions(ctx context.Context, arg TrimCronExecutionsParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, trimCronExecutions, arg.CronJobID, arg.Limit)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const updateCronJobRun = `-- name: UpdateCronJobRun :exec
|
||||||
|
UPDATE cron_jobs SET last_run_at = $1, last_status = $2, last_output = $3, next_run_at = $4 WHERE id = $5
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpdateCronJobRunParams struct {
|
||||||
|
LastRunAt sql.NullTime `json:"last_run_at"`
|
||||||
|
LastStatus sql.NullString `json:"last_status"`
|
||||||
|
LastOutput sql.NullString `json:"last_output"`
|
||||||
|
NextRunAt sql.NullTime `json:"next_run_at"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpdateCronJobRun(ctx context.Context, arg UpdateCronJobRunParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, updateCronJobRun,
|
||||||
|
arg.LastRunAt,
|
||||||
|
arg.LastStatus,
|
||||||
|
arg.LastOutput,
|
||||||
|
arg.NextRunAt,
|
||||||
|
arg.ID,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
@@ -0,0 +1,536 @@
|
|||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.31.1
|
||||||
|
// source: deployments.sql
|
||||||
|
|
||||||
|
package sqlcdb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
const completeDeployment = `-- name: CompleteDeployment :exec
|
||||||
|
UPDATE deployments
|
||||||
|
SET status = $1, error = $2, completed_at = $3, updated_at = $3
|
||||||
|
WHERE id = $4
|
||||||
|
`
|
||||||
|
|
||||||
|
type CompleteDeploymentParams struct {
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
Error sql.NullString `json:"error"`
|
||||||
|
CompletedAt sql.NullTime `json:"completed_at"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CompleteDeployment(ctx context.Context, arg CompleteDeploymentParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, completeDeployment,
|
||||||
|
arg.Status,
|
||||||
|
arg.Error,
|
||||||
|
arg.CompletedAt,
|
||||||
|
arg.ID,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const countAccessibleDeployments = `-- name: CountAccessibleDeployments :one
|
||||||
|
SELECT COUNT(*)
|
||||||
|
FROM deployments d
|
||||||
|
JOIN services s ON s.id = d.service_id
|
||||||
|
JOIN projects p ON p.id = s.project_id
|
||||||
|
WHERE p.is_approved OR p.owner_id = $1 OR $2::bool
|
||||||
|
OR EXISTS (SELECT 1 FROM project_members pm WHERE pm.project_id = p.id AND pm.user_id = $1)
|
||||||
|
`
|
||||||
|
|
||||||
|
type CountAccessibleDeploymentsParams struct {
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
Column2 bool `json:"column_2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CountAccessibleDeployments(ctx context.Context, arg CountAccessibleDeploymentsParams) (int64, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, countAccessibleDeployments, arg.OwnerID, arg.Column2)
|
||||||
|
var count int64
|
||||||
|
err := row.Scan(&count)
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const countDeploymentsForService = `-- name: CountDeploymentsForService :one
|
||||||
|
SELECT COUNT(*) FROM deployments WHERE service_id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) CountDeploymentsForService(ctx context.Context, serviceID uuid.UUID) (int64, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, countDeploymentsForService, serviceID)
|
||||||
|
var count int64
|
||||||
|
err := row.Scan(&count)
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const failDeployment = `-- name: FailDeployment :exec
|
||||||
|
UPDATE deployments
|
||||||
|
SET status = 'failed', error = $1, completed_at = $2, updated_at = $2
|
||||||
|
WHERE id = $3
|
||||||
|
`
|
||||||
|
|
||||||
|
type FailDeploymentParams struct {
|
||||||
|
Error sql.NullString `json:"error"`
|
||||||
|
CompletedAt sql.NullTime `json:"completed_at"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) FailDeployment(ctx context.Context, arg FailDeploymentParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, failDeployment, arg.Error, arg.CompletedAt, arg.ID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getDeploymentAccess = `-- name: GetDeploymentAccess :one
|
||||||
|
SELECT d.status, d.service_id, p.owner_id
|
||||||
|
FROM deployments d
|
||||||
|
JOIN services s ON d.service_id = s.id
|
||||||
|
JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE d.id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetDeploymentAccessRow struct {
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetDeploymentAccess(ctx context.Context, id uuid.UUID) (GetDeploymentAccessRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getDeploymentAccess, id)
|
||||||
|
var i GetDeploymentAccessRow
|
||||||
|
err := row.Scan(&i.Status, &i.ServiceID, &i.OwnerID)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getDeploymentForRollback = `-- name: GetDeploymentForRollback :one
|
||||||
|
SELECT d.id, d.service_id, d.commit_hash, d.status, d.image_name, d.image_tag,
|
||||||
|
d.build_log, d.runtime_log, d.error, d.started_at, d.completed_at,
|
||||||
|
d.created_at, d.updated_at, p.owner_id
|
||||||
|
FROM deployments d
|
||||||
|
JOIN services s ON d.service_id = s.id
|
||||||
|
JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE d.id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetDeploymentForRollbackRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
CommitHash sql.NullString `json:"commit_hash"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
ImageName sql.NullString `json:"image_name"`
|
||||||
|
ImageTag sql.NullString `json:"image_tag"`
|
||||||
|
BuildLog sql.NullString `json:"build_log"`
|
||||||
|
RuntimeLog sql.NullString `json:"runtime_log"`
|
||||||
|
Error sql.NullString `json:"error"`
|
||||||
|
StartedAt sql.NullTime `json:"started_at"`
|
||||||
|
CompletedAt sql.NullTime `json:"completed_at"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetDeploymentForRollback(ctx context.Context, id uuid.UUID) (GetDeploymentForRollbackRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getDeploymentForRollback, id)
|
||||||
|
var i GetDeploymentForRollbackRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.CommitHash,
|
||||||
|
&i.Status,
|
||||||
|
&i.ImageName,
|
||||||
|
&i.ImageTag,
|
||||||
|
&i.BuildLog,
|
||||||
|
&i.RuntimeLog,
|
||||||
|
&i.Error,
|
||||||
|
&i.StartedAt,
|
||||||
|
&i.CompletedAt,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
&i.OwnerID,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getDeploymentWithProject = `-- name: GetDeploymentWithProject :one
|
||||||
|
SELECT d.id, d.service_id, d.commit_hash, d.status, d.image_name, d.image_tag,
|
||||||
|
d.build_log, d.runtime_log, d.error, d.started_at, d.completed_at,
|
||||||
|
d.created_at, d.updated_at, p.id AS project_id
|
||||||
|
FROM deployments d
|
||||||
|
JOIN services s ON d.service_id = s.id
|
||||||
|
JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE d.id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetDeploymentWithProjectRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
CommitHash sql.NullString `json:"commit_hash"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
ImageName sql.NullString `json:"image_name"`
|
||||||
|
ImageTag sql.NullString `json:"image_tag"`
|
||||||
|
BuildLog sql.NullString `json:"build_log"`
|
||||||
|
RuntimeLog sql.NullString `json:"runtime_log"`
|
||||||
|
Error sql.NullString `json:"error"`
|
||||||
|
StartedAt sql.NullTime `json:"started_at"`
|
||||||
|
CompletedAt sql.NullTime `json:"completed_at"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetDeploymentWithProject(ctx context.Context, id uuid.UUID) (GetDeploymentWithProjectRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getDeploymentWithProject, id)
|
||||||
|
var i GetDeploymentWithProjectRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.CommitHash,
|
||||||
|
&i.Status,
|
||||||
|
&i.ImageName,
|
||||||
|
&i.ImageTag,
|
||||||
|
&i.BuildLog,
|
||||||
|
&i.RuntimeLog,
|
||||||
|
&i.Error,
|
||||||
|
&i.StartedAt,
|
||||||
|
&i.CompletedAt,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
&i.ProjectID,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServiceForDeployWithOwner = `-- name: GetServiceForDeployWithOwner :one
|
||||||
|
SELECT s.id, s.project_id, s.name, s.type, s.status, s.image, s.command,
|
||||||
|
s.environment, s.git_repo, s.git_branch, s.build_path, s.cpu, s.memory,
|
||||||
|
COALESCE(s.replicas, 1) AS replicas, COALESCE(s.port, 0) AS port,
|
||||||
|
COALESCE(s.domain, '') AS domain, COALESCE(s.healthcheck_path, '') AS healthcheck_path,
|
||||||
|
COALESCE(s.restart_policy, 'unless-stopped') AS restart_policy,
|
||||||
|
COALESCE(s.builder, 'auto') AS builder, COALESCE(s.cpu_reserve, '') AS cpu_reserve,
|
||||||
|
COALESCE(s.memory_reserve, '') AS memory_reserve, COALESCE(s.static_build_cmd, '') AS static_build_cmd,
|
||||||
|
COALESCE(s.static_dir, '') AS static_dir,
|
||||||
|
s.created_at, s.updated_at, p.owner_id
|
||||||
|
FROM services s
|
||||||
|
JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE s.id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetServiceForDeployWithOwnerRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Type sql.NullString `json:"type"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
Image sql.NullString `json:"image"`
|
||||||
|
Command sql.NullString `json:"command"`
|
||||||
|
Environment sql.NullString `json:"environment"`
|
||||||
|
GitRepo sql.NullString `json:"git_repo"`
|
||||||
|
GitBranch sql.NullString `json:"git_branch"`
|
||||||
|
BuildPath sql.NullString `json:"build_path"`
|
||||||
|
Cpu sql.NullString `json:"cpu"`
|
||||||
|
Memory sql.NullString `json:"memory"`
|
||||||
|
Replicas int32 `json:"replicas"`
|
||||||
|
Port int32 `json:"port"`
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
HealthcheckPath string `json:"healthcheck_path"`
|
||||||
|
RestartPolicy string `json:"restart_policy"`
|
||||||
|
Builder string `json:"builder"`
|
||||||
|
CpuReserve string `json:"cpu_reserve"`
|
||||||
|
MemoryReserve string `json:"memory_reserve"`
|
||||||
|
StaticBuildCmd string `json:"static_build_cmd"`
|
||||||
|
StaticDir string `json:"static_dir"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetServiceForDeployWithOwner(ctx context.Context, id uuid.UUID) (GetServiceForDeployWithOwnerRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServiceForDeployWithOwner, id)
|
||||||
|
var i GetServiceForDeployWithOwnerRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.Name,
|
||||||
|
&i.Type,
|
||||||
|
&i.Status,
|
||||||
|
&i.Image,
|
||||||
|
&i.Command,
|
||||||
|
&i.Environment,
|
||||||
|
&i.GitRepo,
|
||||||
|
&i.GitBranch,
|
||||||
|
&i.BuildPath,
|
||||||
|
&i.Cpu,
|
||||||
|
&i.Memory,
|
||||||
|
&i.Replicas,
|
||||||
|
&i.Port,
|
||||||
|
&i.Domain,
|
||||||
|
&i.HealthcheckPath,
|
||||||
|
&i.RestartPolicy,
|
||||||
|
&i.Builder,
|
||||||
|
&i.CpuReserve,
|
||||||
|
&i.MemoryReserve,
|
||||||
|
&i.StaticBuildCmd,
|
||||||
|
&i.StaticDir,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
&i.OwnerID,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServicePublishedPort = `-- name: GetServicePublishedPort :one
|
||||||
|
SELECT published_port FROM services WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetServicePublishedPort(ctx context.Context, id uuid.UUID) (int32, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServicePublishedPort, id)
|
||||||
|
var published_port int32
|
||||||
|
err := row.Scan(&published_port)
|
||||||
|
return published_port, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const insertDeployment = `-- name: InsertDeployment :exec
|
||||||
|
INSERT INTO deployments
|
||||||
|
(id, service_id, version, commit_hash, status, image_name, image_tag, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)
|
||||||
|
`
|
||||||
|
|
||||||
|
type InsertDeploymentParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
Version string `json:"version"`
|
||||||
|
CommitHash sql.NullString `json:"commit_hash"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
ImageName sql.NullString `json:"image_name"`
|
||||||
|
ImageTag sql.NullString `json:"image_tag"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) InsertDeployment(ctx context.Context, arg InsertDeploymentParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, insertDeployment,
|
||||||
|
arg.ID,
|
||||||
|
arg.ServiceID,
|
||||||
|
arg.Version,
|
||||||
|
arg.CommitHash,
|
||||||
|
arg.Status,
|
||||||
|
arg.ImageName,
|
||||||
|
arg.ImageTag,
|
||||||
|
arg.CreatedAt,
|
||||||
|
arg.UpdatedAt,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const listDeploymentsForService = `-- name: ListDeploymentsForService :many
|
||||||
|
SELECT id, service_id, commit_hash, status, image_name, image_tag,
|
||||||
|
build_log, runtime_log, error, started_at, completed_at, created_at, updated_at
|
||||||
|
FROM deployments
|
||||||
|
WHERE service_id = $1
|
||||||
|
ORDER BY created_at DESC
|
||||||
|
LIMIT $2 OFFSET $3
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListDeploymentsForServiceParams struct {
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
Limit int32 `json:"limit"`
|
||||||
|
Offset int32 `json:"offset"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ListDeploymentsForServiceRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
CommitHash sql.NullString `json:"commit_hash"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
ImageName sql.NullString `json:"image_name"`
|
||||||
|
ImageTag sql.NullString `json:"image_tag"`
|
||||||
|
BuildLog sql.NullString `json:"build_log"`
|
||||||
|
RuntimeLog sql.NullString `json:"runtime_log"`
|
||||||
|
Error sql.NullString `json:"error"`
|
||||||
|
StartedAt sql.NullTime `json:"started_at"`
|
||||||
|
CompletedAt sql.NullTime `json:"completed_at"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListDeploymentsForService(ctx context.Context, arg ListDeploymentsForServiceParams) ([]ListDeploymentsForServiceRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listDeploymentsForService, arg.ServiceID, arg.Limit, arg.Offset)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListDeploymentsForServiceRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListDeploymentsForServiceRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.CommitHash,
|
||||||
|
&i.Status,
|
||||||
|
&i.ImageName,
|
||||||
|
&i.ImageTag,
|
||||||
|
&i.BuildLog,
|
||||||
|
&i.RuntimeLog,
|
||||||
|
&i.Error,
|
||||||
|
&i.StartedAt,
|
||||||
|
&i.CompletedAt,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listRecentAccessibleDeployments = `-- name: ListRecentAccessibleDeployments :many
|
||||||
|
SELECT d.id, d.service_id, s.name AS service_name, p.name AS project_name, d.status,
|
||||||
|
COALESCE(d.image_name, '') AS image_name, d.started_at, d.completed_at, d.created_at
|
||||||
|
FROM deployments d
|
||||||
|
JOIN services s ON s.id = d.service_id
|
||||||
|
JOIN projects p ON p.id = s.project_id
|
||||||
|
WHERE p.is_approved OR p.owner_id = $1 OR $2::bool
|
||||||
|
OR EXISTS (SELECT 1 FROM project_members pm WHERE pm.project_id = p.id AND pm.user_id = $1)
|
||||||
|
ORDER BY d.created_at DESC
|
||||||
|
LIMIT $3 OFFSET $4
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListRecentAccessibleDeploymentsParams struct {
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
Column2 bool `json:"column_2"`
|
||||||
|
Limit int32 `json:"limit"`
|
||||||
|
Offset int32 `json:"offset"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ListRecentAccessibleDeploymentsRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
ServiceName string `json:"service_name"`
|
||||||
|
ProjectName string `json:"project_name"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
ImageName string `json:"image_name"`
|
||||||
|
StartedAt sql.NullTime `json:"started_at"`
|
||||||
|
CompletedAt sql.NullTime `json:"completed_at"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListRecentAccessibleDeployments(ctx context.Context, arg ListRecentAccessibleDeploymentsParams) ([]ListRecentAccessibleDeploymentsRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listRecentAccessibleDeployments,
|
||||||
|
arg.OwnerID,
|
||||||
|
arg.Column2,
|
||||||
|
arg.Limit,
|
||||||
|
arg.Offset,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListRecentAccessibleDeploymentsRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListRecentAccessibleDeploymentsRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.ServiceName,
|
||||||
|
&i.ProjectName,
|
||||||
|
&i.Status,
|
||||||
|
&i.ImageName,
|
||||||
|
&i.StartedAt,
|
||||||
|
&i.CompletedAt,
|
||||||
|
&i.CreatedAt,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const setDeploymentStatus = `-- name: SetDeploymentStatus :exec
|
||||||
|
UPDATE deployments SET status = $1, updated_at = $2 WHERE id = $3
|
||||||
|
`
|
||||||
|
|
||||||
|
type SetDeploymentStatusParams struct {
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) SetDeploymentStatus(ctx context.Context, arg SetDeploymentStatusParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, setDeploymentStatus, arg.Status, arg.UpdatedAt, arg.ID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const setServiceStatus = `-- name: SetServiceStatus :exec
|
||||||
|
UPDATE services SET status = $1, updated_at = $2 WHERE id = $3
|
||||||
|
`
|
||||||
|
|
||||||
|
type SetServiceStatusParams struct {
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) SetServiceStatus(ctx context.Context, arg SetServiceStatusParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, setServiceStatus, arg.Status, arg.UpdatedAt, arg.ID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const syncDeploymentProgress = `-- name: SyncDeploymentProgress :exec
|
||||||
|
UPDATE deployments
|
||||||
|
SET status = $1,
|
||||||
|
image_name = $2,
|
||||||
|
image_tag = $3,
|
||||||
|
build_log = $4,
|
||||||
|
runtime_log = $5,
|
||||||
|
error = $6,
|
||||||
|
started_at = $7,
|
||||||
|
completed_at = $8,
|
||||||
|
updated_at = $9
|
||||||
|
WHERE id = $10
|
||||||
|
`
|
||||||
|
|
||||||
|
type SyncDeploymentProgressParams struct {
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
ImageName sql.NullString `json:"image_name"`
|
||||||
|
ImageTag sql.NullString `json:"image_tag"`
|
||||||
|
BuildLog sql.NullString `json:"build_log"`
|
||||||
|
RuntimeLog sql.NullString `json:"runtime_log"`
|
||||||
|
Error sql.NullString `json:"error"`
|
||||||
|
StartedAt sql.NullTime `json:"started_at"`
|
||||||
|
CompletedAt sql.NullTime `json:"completed_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) SyncDeploymentProgress(ctx context.Context, arg SyncDeploymentProgressParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, syncDeploymentProgress,
|
||||||
|
arg.Status,
|
||||||
|
arg.ImageName,
|
||||||
|
arg.ImageTag,
|
||||||
|
arg.BuildLog,
|
||||||
|
arg.RuntimeLog,
|
||||||
|
arg.Error,
|
||||||
|
arg.StartedAt,
|
||||||
|
arg.CompletedAt,
|
||||||
|
arg.UpdatedAt,
|
||||||
|
arg.ID,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
@@ -0,0 +1,242 @@
|
|||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.31.1
|
||||||
|
// source: domains.sql
|
||||||
|
|
||||||
|
package sqlcdb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
const clearServiceDomainDefault = `-- name: ClearServiceDomainDefault :exec
|
||||||
|
UPDATE service_domains SET is_default = false WHERE service_id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) ClearServiceDomainDefault(ctx context.Context, serviceID uuid.UUID) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, clearServiceDomainDefault, serviceID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const countServiceDomains = `-- name: CountServiceDomains :one
|
||||||
|
SELECT COUNT(*) FROM service_domains WHERE service_id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) CountServiceDomains(ctx context.Context, serviceID uuid.UUID) (int64, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, countServiceDomains, serviceID)
|
||||||
|
var count int64
|
||||||
|
err := row.Scan(&count)
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const createServiceDomain = `-- name: CreateServiceDomain :one
|
||||||
|
INSERT INTO service_domains (service_id, domain, is_default)
|
||||||
|
VALUES ($1, $2, $3)
|
||||||
|
RETURNING id, service_id, domain, is_default, cert_type, cert_status, created_at
|
||||||
|
`
|
||||||
|
|
||||||
|
type CreateServiceDomainParams struct {
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
IsDefault bool `json:"is_default"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type CreateServiceDomainRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
IsDefault bool `json:"is_default"`
|
||||||
|
CertType string `json:"cert_type"`
|
||||||
|
CertStatus string `json:"cert_status"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CreateServiceDomain(ctx context.Context, arg CreateServiceDomainParams) (CreateServiceDomainRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, createServiceDomain, arg.ServiceID, arg.Domain, arg.IsDefault)
|
||||||
|
var i CreateServiceDomainRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.Domain,
|
||||||
|
&i.IsDefault,
|
||||||
|
&i.CertType,
|
||||||
|
&i.CertStatus,
|
||||||
|
&i.CreatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const deleteServiceDomain = `-- name: DeleteServiceDomain :execrows
|
||||||
|
DELETE FROM service_domains WHERE id = $1 AND service_id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type DeleteServiceDomainParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) DeleteServiceDomain(ctx context.Context, arg DeleteServiceDomainParams) (int64, error) {
|
||||||
|
result, err := q.db.ExecContext(ctx, deleteServiceDomain, arg.ID, arg.ServiceID)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
return result.RowsAffected()
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServiceAccess = `-- name: GetServiceAccess :one
|
||||||
|
SELECT maintenance_mode, basic_auth_users FROM services WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetServiceAccessRow struct {
|
||||||
|
MaintenanceMode bool `json:"maintenance_mode"`
|
||||||
|
BasicAuthUsers string `json:"basic_auth_users"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetServiceAccess(ctx context.Context, id uuid.UUID) (GetServiceAccessRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServiceAccess, id)
|
||||||
|
var i GetServiceAccessRow
|
||||||
|
err := row.Scan(&i.MaintenanceMode, &i.BasicAuthUsers)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServiceDomainDefault = `-- name: GetServiceDomainDefault :one
|
||||||
|
SELECT domain FROM service_domains WHERE service_id = $1 AND is_default
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetServiceDomainDefault(ctx context.Context, serviceID uuid.UUID) (string, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServiceDomainDefault, serviceID)
|
||||||
|
var domain string
|
||||||
|
err := row.Scan(&domain)
|
||||||
|
return domain, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServiceDomainFirst = `-- name: GetServiceDomainFirst :one
|
||||||
|
SELECT domain FROM service_domains WHERE service_id = $1 ORDER BY created_at ASC LIMIT 1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetServiceDomainFirst(ctx context.Context, serviceID uuid.UUID) (string, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServiceDomainFirst, serviceID)
|
||||||
|
var domain string
|
||||||
|
err := row.Scan(&domain)
|
||||||
|
return domain, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServicePlacementBrief = `-- name: GetServicePlacementBrief :one
|
||||||
|
SELECT name, node_id, COALESCE(spread, false) AS spread FROM services WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetServicePlacementBriefRow struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
NodeID sql.NullString `json:"node_id"`
|
||||||
|
Spread bool `json:"spread"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetServicePlacementBrief(ctx context.Context, id uuid.UUID) (GetServicePlacementBriefRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServicePlacementBrief, id)
|
||||||
|
var i GetServicePlacementBriefRow
|
||||||
|
err := row.Scan(&i.Name, &i.NodeID, &i.Spread)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const listServiceDomains = `-- name: ListServiceDomains :many
|
||||||
|
SELECT id, service_id, domain, is_default, cert_type, cert_status, last_checked_at, created_at
|
||||||
|
FROM service_domains
|
||||||
|
WHERE service_id = $1
|
||||||
|
ORDER BY is_default DESC, created_at ASC
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) ListServiceDomains(ctx context.Context, serviceID uuid.UUID) ([]ServiceDomain, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listServiceDomains, serviceID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ServiceDomain{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ServiceDomain
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.Domain,
|
||||||
|
&i.IsDefault,
|
||||||
|
&i.CertType,
|
||||||
|
&i.CertStatus,
|
||||||
|
&i.LastCheckedAt,
|
||||||
|
&i.CreatedAt,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const serviceWriteAccess = `-- name: ServiceWriteAccess :one
|
||||||
|
SELECT EXISTS(
|
||||||
|
SELECT 1 FROM services s JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE s.id = $1 AND (p.owner_id = $2 OR $3::bool)
|
||||||
|
) AS allowed
|
||||||
|
`
|
||||||
|
|
||||||
|
type ServiceWriteAccessParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
Column3 bool `json:"column_3"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ServiceWriteAccess(ctx context.Context, arg ServiceWriteAccessParams) (bool, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, serviceWriteAccess, arg.ID, arg.OwnerID, arg.Column3)
|
||||||
|
var allowed bool
|
||||||
|
err := row.Scan(&allowed)
|
||||||
|
return allowed, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const setServiceDomainDefault = `-- name: SetServiceDomainDefault :execrows
|
||||||
|
UPDATE service_domains SET is_default = (id = $1) WHERE service_id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type SetServiceDomainDefaultParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) SetServiceDomainDefault(ctx context.Context, arg SetServiceDomainDefaultParams) (int64, error) {
|
||||||
|
result, err := q.db.ExecContext(ctx, setServiceDomainDefault, arg.ID, arg.ServiceID)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
return result.RowsAffected()
|
||||||
|
}
|
||||||
|
|
||||||
|
const setServiceLegacyDomain = `-- name: SetServiceLegacyDomain :exec
|
||||||
|
UPDATE services SET domain = $1 WHERE id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type SetServiceLegacyDomainParams struct {
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) SetServiceLegacyDomain(ctx context.Context, arg SetServiceLegacyDomainParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, setServiceLegacyDomain, arg.Domain, arg.ID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const touchServiceDomainChecked = `-- name: TouchServiceDomainChecked :exec
|
||||||
|
UPDATE service_domains SET last_checked_at = NOW() WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) TouchServiceDomainChecked(ctx context.Context, id uuid.UUID) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, touchServiceDomainChecked, id)
|
||||||
|
return err
|
||||||
|
}
|
||||||
@@ -0,0 +1,737 @@
|
|||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.31.1
|
||||||
|
// source: git.sql
|
||||||
|
|
||||||
|
package sqlcdb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
const countGitRepositories = `-- name: CountGitRepositories :one
|
||||||
|
SELECT COUNT(*)
|
||||||
|
FROM git_repositories
|
||||||
|
WHERE provider_id = $1 AND user_id = $2
|
||||||
|
AND ($3::text IS NULL OR full_name ILIKE ('%' || $3::text || '%') OR name ILIKE ('%' || $3::text || '%'))
|
||||||
|
`
|
||||||
|
|
||||||
|
type CountGitRepositoriesParams struct {
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
Column3 string `json:"column_3"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CountGitRepositories(ctx context.Context, arg CountGitRepositoriesParams) (int64, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, countGitRepositories, arg.ProviderID, arg.UserID, arg.Column3)
|
||||||
|
var count int64
|
||||||
|
err := row.Scan(&count)
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const countUserRepositories = `-- name: CountUserRepositories :one
|
||||||
|
SELECT COUNT(*) FROM git_repositories WHERE user_id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) CountUserRepositories(ctx context.Context, userID uuid.UUID) (int64, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, countUserRepositories, userID)
|
||||||
|
var count int64
|
||||||
|
err := row.Scan(&count)
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const deleteGitProvider = `-- name: DeleteGitProvider :execrows
|
||||||
|
DELETE FROM git_providers WHERE id = $1 AND user_id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type DeleteGitProviderParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) DeleteGitProvider(ctx context.Context, arg DeleteGitProviderParams) (int64, error) {
|
||||||
|
result, err := q.db.ExecContext(ctx, deleteGitProvider, arg.ID, arg.UserID)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
return result.RowsAffected()
|
||||||
|
}
|
||||||
|
|
||||||
|
const getGitProviderForFetch = `-- name: GetGitProviderForFetch :one
|
||||||
|
SELECT name, api_url, access_token
|
||||||
|
FROM git_providers
|
||||||
|
WHERE id = $1 AND user_id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetGitProviderForFetchParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetGitProviderForFetchRow struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
ApiUrl string `json:"api_url"`
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetGitProviderForFetch(ctx context.Context, arg GetGitProviderForFetchParams) (GetGitProviderForFetchRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getGitProviderForFetch, arg.ID, arg.UserID)
|
||||||
|
var i GetGitProviderForFetchRow
|
||||||
|
err := row.Scan(&i.Name, &i.ApiUrl, &i.AccessToken)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getGitProviderForRepos = `-- name: GetGitProviderForRepos :one
|
||||||
|
SELECT name, display_name, access_token, api_url
|
||||||
|
FROM git_providers
|
||||||
|
WHERE id = $1 AND user_id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetGitProviderForReposParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetGitProviderForReposRow struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
DisplayName string `json:"display_name"`
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
ApiUrl string `json:"api_url"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetGitProviderForRepos(ctx context.Context, arg GetGitProviderForReposParams) (GetGitProviderForReposRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getGitProviderForRepos, arg.ID, arg.UserID)
|
||||||
|
var i GetGitProviderForReposRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.Name,
|
||||||
|
&i.DisplayName,
|
||||||
|
&i.AccessToken,
|
||||||
|
&i.ApiUrl,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getGitProviderName = `-- name: GetGitProviderName :one
|
||||||
|
SELECT name FROM git_providers WHERE id = $1 AND user_id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetGitProviderNameParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetGitProviderName(ctx context.Context, arg GetGitProviderNameParams) (string, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getGitProviderName, arg.ID, arg.UserID)
|
||||||
|
var name string
|
||||||
|
err := row.Scan(&name)
|
||||||
|
return name, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getGitProviderNameByID = `-- name: GetGitProviderNameByID :one
|
||||||
|
SELECT name FROM git_providers WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetGitProviderNameByID(ctx context.Context, id uuid.UUID) (string, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getGitProviderNameByID, id)
|
||||||
|
var name string
|
||||||
|
err := row.Scan(&name)
|
||||||
|
return name, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getGitProviderTimestamps = `-- name: GetGitProviderTimestamps :one
|
||||||
|
SELECT created_at, updated_at FROM git_providers WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetGitProviderTimestampsRow struct {
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetGitProviderTimestamps(ctx context.Context, id uuid.UUID) (GetGitProviderTimestampsRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getGitProviderTimestamps, id)
|
||||||
|
var i GetGitProviderTimestampsRow
|
||||||
|
err := row.Scan(&i.CreatedAt, &i.UpdatedAt)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getGitRepoForPush = `-- name: GetGitRepoForPush :one
|
||||||
|
SELECT clone_url, full_name, user_id FROM git_repositories WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetGitRepoForPushRow struct {
|
||||||
|
CloneUrl string `json:"clone_url"`
|
||||||
|
FullName string `json:"full_name"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetGitRepoForPush(ctx context.Context, id uuid.UUID) (GetGitRepoForPushRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getGitRepoForPush, id)
|
||||||
|
var i GetGitRepoForPushRow
|
||||||
|
err := row.Scan(&i.CloneUrl, &i.FullName, &i.UserID)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getGitRepositoryByFullName = `-- name: GetGitRepositoryByFullName :one
|
||||||
|
SELECT id, provider_id, name, full_name, COALESCE(description, '') AS description,
|
||||||
|
clone_url, COALESCE(webhook_url, '') AS webhook_url,
|
||||||
|
default_branch, is_private, user_id, created_at, updated_at
|
||||||
|
FROM git_repositories
|
||||||
|
WHERE provider_id = $1 AND full_name = $2 AND user_id = $3
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetGitRepositoryByFullNameParams struct {
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
FullName string `json:"full_name"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetGitRepositoryByFullNameRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
FullName string `json:"full_name"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
CloneUrl string `json:"clone_url"`
|
||||||
|
WebhookUrl string `json:"webhook_url"`
|
||||||
|
DefaultBranch sql.NullString `json:"default_branch"`
|
||||||
|
IsPrivate sql.NullBool `json:"is_private"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetGitRepositoryByFullName(ctx context.Context, arg GetGitRepositoryByFullNameParams) (GetGitRepositoryByFullNameRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getGitRepositoryByFullName, arg.ProviderID, arg.FullName, arg.UserID)
|
||||||
|
var i GetGitRepositoryByFullNameRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProviderID,
|
||||||
|
&i.Name,
|
||||||
|
&i.FullName,
|
||||||
|
&i.Description,
|
||||||
|
&i.CloneUrl,
|
||||||
|
&i.WebhookUrl,
|
||||||
|
&i.DefaultBranch,
|
||||||
|
&i.IsPrivate,
|
||||||
|
&i.UserID,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getGitRepositoryProviderID = `-- name: GetGitRepositoryProviderID :one
|
||||||
|
SELECT provider_id FROM git_repositories WHERE id = $1 AND user_id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetGitRepositoryProviderIDParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetGitRepositoryProviderID(ctx context.Context, arg GetGitRepositoryProviderIDParams) (uuid.UUID, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getGitRepositoryProviderID, arg.ID, arg.UserID)
|
||||||
|
var provider_id uuid.UUID
|
||||||
|
err := row.Scan(&provider_id)
|
||||||
|
return provider_id, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getGitWebhookForPush = `-- name: GetGitWebhookForPush :one
|
||||||
|
SELECT webhook_secret, repo_id, provider_id, COALESCE(branch_filter, '') AS branch_filter, active
|
||||||
|
FROM git_webhooks WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetGitWebhookForPushRow struct {
|
||||||
|
WebhookSecret string `json:"webhook_secret"`
|
||||||
|
RepoID uuid.UUID `json:"repo_id"`
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
BranchFilter string `json:"branch_filter"`
|
||||||
|
Active sql.NullBool `json:"active"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetGitWebhookForPush(ctx context.Context, id uuid.UUID) (GetGitWebhookForPushRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getGitWebhookForPush, id)
|
||||||
|
var i GetGitWebhookForPushRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.WebhookSecret,
|
||||||
|
&i.RepoID,
|
||||||
|
&i.ProviderID,
|
||||||
|
&i.BranchFilter,
|
||||||
|
&i.Active,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const insertGitProvider = `-- name: InsertGitProvider :exec
|
||||||
|
INSERT INTO git_providers (id, name, display_name, api_url, webhook_url, access_token, user_id, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, NOW(), NOW())
|
||||||
|
`
|
||||||
|
|
||||||
|
type InsertGitProviderParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
DisplayName string `json:"display_name"`
|
||||||
|
ApiUrl string `json:"api_url"`
|
||||||
|
WebhookUrl string `json:"webhook_url"`
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) InsertGitProvider(ctx context.Context, arg InsertGitProviderParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, insertGitProvider,
|
||||||
|
arg.ID,
|
||||||
|
arg.Name,
|
||||||
|
arg.DisplayName,
|
||||||
|
arg.ApiUrl,
|
||||||
|
arg.WebhookUrl,
|
||||||
|
arg.AccessToken,
|
||||||
|
arg.UserID,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const insertGitRepository = `-- name: InsertGitRepository :one
|
||||||
|
INSERT INTO git_repositories (id, provider_id, name, full_name, description, clone_url, default_branch, is_private, user_id, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, NOW(), NOW())
|
||||||
|
RETURNING created_at, updated_at
|
||||||
|
`
|
||||||
|
|
||||||
|
type InsertGitRepositoryParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
FullName string `json:"full_name"`
|
||||||
|
Description sql.NullString `json:"description"`
|
||||||
|
CloneUrl string `json:"clone_url"`
|
||||||
|
DefaultBranch sql.NullString `json:"default_branch"`
|
||||||
|
IsPrivate sql.NullBool `json:"is_private"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type InsertGitRepositoryRow struct {
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) InsertGitRepository(ctx context.Context, arg InsertGitRepositoryParams) (InsertGitRepositoryRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, insertGitRepository,
|
||||||
|
arg.ID,
|
||||||
|
arg.ProviderID,
|
||||||
|
arg.Name,
|
||||||
|
arg.FullName,
|
||||||
|
arg.Description,
|
||||||
|
arg.CloneUrl,
|
||||||
|
arg.DefaultBranch,
|
||||||
|
arg.IsPrivate,
|
||||||
|
arg.UserID,
|
||||||
|
)
|
||||||
|
var i InsertGitRepositoryRow
|
||||||
|
err := row.Scan(&i.CreatedAt, &i.UpdatedAt)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const listConnectedRepositories = `-- name: ListConnectedRepositories :many
|
||||||
|
SELECT r.id, r.provider_id, r.name, r.full_name, COALESCE(r.description, '') AS description, r.clone_url,
|
||||||
|
r.default_branch, r.is_private, r.user_id, r.created_at, r.updated_at,
|
||||||
|
p.name AS provider_name, p.display_name
|
||||||
|
FROM git_repositories r
|
||||||
|
JOIN git_providers p ON r.provider_id = p.id
|
||||||
|
WHERE r.user_id = $1
|
||||||
|
ORDER BY r.updated_at DESC
|
||||||
|
LIMIT $2 OFFSET $3
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListConnectedRepositoriesParams struct {
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
Limit int32 `json:"limit"`
|
||||||
|
Offset int32 `json:"offset"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ListConnectedRepositoriesRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
FullName string `json:"full_name"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
CloneUrl string `json:"clone_url"`
|
||||||
|
DefaultBranch sql.NullString `json:"default_branch"`
|
||||||
|
IsPrivate sql.NullBool `json:"is_private"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
ProviderName string `json:"provider_name"`
|
||||||
|
DisplayName string `json:"display_name"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListConnectedRepositories(ctx context.Context, arg ListConnectedRepositoriesParams) ([]ListConnectedRepositoriesRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listConnectedRepositories, arg.UserID, arg.Limit, arg.Offset)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListConnectedRepositoriesRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListConnectedRepositoriesRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProviderID,
|
||||||
|
&i.Name,
|
||||||
|
&i.FullName,
|
||||||
|
&i.Description,
|
||||||
|
&i.CloneUrl,
|
||||||
|
&i.DefaultBranch,
|
||||||
|
&i.IsPrivate,
|
||||||
|
&i.UserID,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
&i.ProviderName,
|
||||||
|
&i.DisplayName,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listGitProviders = `-- name: ListGitProviders :many
|
||||||
|
SELECT id, name, display_name, api_url, webhook_url, user_id, created_at, updated_at
|
||||||
|
FROM git_providers
|
||||||
|
WHERE user_id = $1
|
||||||
|
ORDER BY created_at DESC
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListGitProvidersRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
DisplayName string `json:"display_name"`
|
||||||
|
ApiUrl string `json:"api_url"`
|
||||||
|
WebhookUrl string `json:"webhook_url"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListGitProviders(ctx context.Context, userID uuid.UUID) ([]ListGitProvidersRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listGitProviders, userID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListGitProvidersRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListGitProvidersRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.Name,
|
||||||
|
&i.DisplayName,
|
||||||
|
&i.ApiUrl,
|
||||||
|
&i.WebhookUrl,
|
||||||
|
&i.UserID,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listGitRepositories = `-- name: ListGitRepositories :many
|
||||||
|
SELECT id, provider_id, name, full_name, COALESCE(description, '') AS description,
|
||||||
|
clone_url, COALESCE(webhook_url, '') AS webhook_url,
|
||||||
|
default_branch, is_private, user_id, created_at, updated_at
|
||||||
|
FROM git_repositories
|
||||||
|
WHERE provider_id = $1 AND user_id = $2
|
||||||
|
AND ($3::text IS NULL OR full_name ILIKE ('%' || $3::text || '%') OR name ILIKE ('%' || $3::text || '%'))
|
||||||
|
ORDER BY updated_at DESC
|
||||||
|
LIMIT $4 OFFSET $5
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListGitRepositoriesParams struct {
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
Column3 string `json:"column_3"`
|
||||||
|
Limit int32 `json:"limit"`
|
||||||
|
Offset int32 `json:"offset"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ListGitRepositoriesRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
FullName string `json:"full_name"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
CloneUrl string `json:"clone_url"`
|
||||||
|
WebhookUrl string `json:"webhook_url"`
|
||||||
|
DefaultBranch sql.NullString `json:"default_branch"`
|
||||||
|
IsPrivate sql.NullBool `json:"is_private"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListGitRepositories(ctx context.Context, arg ListGitRepositoriesParams) ([]ListGitRepositoriesRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listGitRepositories,
|
||||||
|
arg.ProviderID,
|
||||||
|
arg.UserID,
|
||||||
|
arg.Column3,
|
||||||
|
arg.Limit,
|
||||||
|
arg.Offset,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListGitRepositoriesRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListGitRepositoriesRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProviderID,
|
||||||
|
&i.Name,
|
||||||
|
&i.FullName,
|
||||||
|
&i.Description,
|
||||||
|
&i.CloneUrl,
|
||||||
|
&i.WebhookUrl,
|
||||||
|
&i.DefaultBranch,
|
||||||
|
&i.IsPrivate,
|
||||||
|
&i.UserID,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listServicesForPush = `-- name: ListServicesForPush :many
|
||||||
|
SELECT s.id, s.project_id, s.name,
|
||||||
|
COALESCE(s.type, s.service_type, '') AS type, COALESCE(s.status, '') AS status,
|
||||||
|
COALESCE(s.image, s.image_name, '') AS image, COALESCE(s.command, s.start_command, '') AS command,
|
||||||
|
COALESCE(s.environment, '') AS environment, COALESCE(s.git_repo, s.source_url, '') AS git_repo,
|
||||||
|
COALESCE(s.git_branch, '') AS git_branch, COALESCE(s.build_path, '') AS build_path,
|
||||||
|
COALESCE(s.cpu, '') AS cpu, COALESCE(s.memory, '') AS memory,
|
||||||
|
s.created_at, s.updated_at
|
||||||
|
FROM services s
|
||||||
|
WHERE (s.git_branch = $1 OR $1 = '') AND (s.git_repo = $2 OR s.git_repo = $3)
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListServicesForPushParams struct {
|
||||||
|
GitBranch sql.NullString `json:"git_branch"`
|
||||||
|
GitRepo sql.NullString `json:"git_repo"`
|
||||||
|
GitRepo_2 sql.NullString `json:"git_repo_2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ListServicesForPushRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Type string `json:"type"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
Image string `json:"image"`
|
||||||
|
Command string `json:"command"`
|
||||||
|
Environment string `json:"environment"`
|
||||||
|
GitRepo string `json:"git_repo"`
|
||||||
|
GitBranch string `json:"git_branch"`
|
||||||
|
BuildPath string `json:"build_path"`
|
||||||
|
Cpu string `json:"cpu"`
|
||||||
|
Memory string `json:"memory"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListServicesForPush(ctx context.Context, arg ListServicesForPushParams) ([]ListServicesForPushRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listServicesForPush, arg.GitBranch, arg.GitRepo, arg.GitRepo_2)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListServicesForPushRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListServicesForPushRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.Name,
|
||||||
|
&i.Type,
|
||||||
|
&i.Status,
|
||||||
|
&i.Image,
|
||||||
|
&i.Command,
|
||||||
|
&i.Environment,
|
||||||
|
&i.GitRepo,
|
||||||
|
&i.GitBranch,
|
||||||
|
&i.BuildPath,
|
||||||
|
&i.Cpu,
|
||||||
|
&i.Memory,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const upsertGitHubAppProvider = `-- name: UpsertGitHubAppProvider :one
|
||||||
|
INSERT INTO git_providers (id, name, display_name, api_url, webhook_url, access_token, user_id, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, NOW(), NOW())
|
||||||
|
ON CONFLICT (name, user_id)
|
||||||
|
DO UPDATE SET
|
||||||
|
display_name = EXCLUDED.display_name,
|
||||||
|
api_url = EXCLUDED.api_url,
|
||||||
|
webhook_url = EXCLUDED.webhook_url,
|
||||||
|
access_token = EXCLUDED.access_token,
|
||||||
|
updated_at = NOW()
|
||||||
|
RETURNING id, created_at, updated_at
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpsertGitHubAppProviderParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
DisplayName string `json:"display_name"`
|
||||||
|
ApiUrl string `json:"api_url"`
|
||||||
|
WebhookUrl string `json:"webhook_url"`
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpsertGitHubAppProviderRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpsertGitHubAppProvider(ctx context.Context, arg UpsertGitHubAppProviderParams) (UpsertGitHubAppProviderRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, upsertGitHubAppProvider,
|
||||||
|
arg.ID,
|
||||||
|
arg.Name,
|
||||||
|
arg.DisplayName,
|
||||||
|
arg.ApiUrl,
|
||||||
|
arg.WebhookUrl,
|
||||||
|
arg.AccessToken,
|
||||||
|
arg.UserID,
|
||||||
|
)
|
||||||
|
var i UpsertGitHubAppProviderRow
|
||||||
|
err := row.Scan(&i.ID, &i.CreatedAt, &i.UpdatedAt)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const upsertGitRepository = `-- name: UpsertGitRepository :exec
|
||||||
|
INSERT INTO git_repositories (id, provider_id, name, full_name, description, clone_url, default_branch, is_private, user_id, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, NOW(), NOW())
|
||||||
|
ON CONFLICT (provider_id, full_name)
|
||||||
|
DO UPDATE SET
|
||||||
|
name = EXCLUDED.name,
|
||||||
|
description = EXCLUDED.description,
|
||||||
|
clone_url = EXCLUDED.clone_url,
|
||||||
|
default_branch = EXCLUDED.default_branch,
|
||||||
|
is_private = EXCLUDED.is_private,
|
||||||
|
updated_at = NOW()
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpsertGitRepositoryParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
FullName string `json:"full_name"`
|
||||||
|
Description sql.NullString `json:"description"`
|
||||||
|
CloneUrl string `json:"clone_url"`
|
||||||
|
DefaultBranch sql.NullString `json:"default_branch"`
|
||||||
|
IsPrivate sql.NullBool `json:"is_private"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpsertGitRepository(ctx context.Context, arg UpsertGitRepositoryParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, upsertGitRepository,
|
||||||
|
arg.ID,
|
||||||
|
arg.ProviderID,
|
||||||
|
arg.Name,
|
||||||
|
arg.FullName,
|
||||||
|
arg.Description,
|
||||||
|
arg.CloneUrl,
|
||||||
|
arg.DefaultBranch,
|
||||||
|
arg.IsPrivate,
|
||||||
|
arg.UserID,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const upsertGitWebhook = `-- name: UpsertGitWebhook :one
|
||||||
|
INSERT INTO git_webhooks (id, repo_id, provider_id, events, webhook_secret, active, branch_filter, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, TRUE, $6, NOW(), NOW())
|
||||||
|
ON CONFLICT (repo_id, provider_id)
|
||||||
|
DO UPDATE SET
|
||||||
|
events = EXCLUDED.events,
|
||||||
|
webhook_secret = EXCLUDED.webhook_secret,
|
||||||
|
active = TRUE,
|
||||||
|
branch_filter = EXCLUDED.branch_filter,
|
||||||
|
updated_at = NOW()
|
||||||
|
RETURNING id, active, created_at, updated_at
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpsertGitWebhookParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
RepoID uuid.UUID `json:"repo_id"`
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
Events string `json:"events"`
|
||||||
|
WebhookSecret string `json:"webhook_secret"`
|
||||||
|
BranchFilter sql.NullString `json:"branch_filter"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpsertGitWebhookRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Active sql.NullBool `json:"active"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpsertGitWebhook(ctx context.Context, arg UpsertGitWebhookParams) (UpsertGitWebhookRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, upsertGitWebhook,
|
||||||
|
arg.ID,
|
||||||
|
arg.RepoID,
|
||||||
|
arg.ProviderID,
|
||||||
|
arg.Events,
|
||||||
|
arg.WebhookSecret,
|
||||||
|
arg.BranchFilter,
|
||||||
|
)
|
||||||
|
var i UpsertGitWebhookRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.Active,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
@@ -0,0 +1,623 @@
|
|||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.31.1
|
||||||
|
// source: metrics.sql
|
||||||
|
|
||||||
|
package sqlcdb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const aggregateNodeMetrics = `-- name: AggregateNodeMetrics :many
|
||||||
|
SELECT date_bin($1::interval, timestamp, '2000-01-01'::timestamptz) AS bucket,
|
||||||
|
AVG(cpu_usage)::float8 AS avg_cpu,
|
||||||
|
AVG(memory_usage_percent)::float8 AS avg_memory,
|
||||||
|
AVG(storage_usage_percent)::float8 AS avg_storage
|
||||||
|
FROM node_metrics
|
||||||
|
WHERE node_id = $2 AND timestamp BETWEEN $3 AND $4
|
||||||
|
GROUP BY bucket
|
||||||
|
ORDER BY bucket ASC
|
||||||
|
`
|
||||||
|
|
||||||
|
type AggregateNodeMetricsParams struct {
|
||||||
|
Column1 int64 `json:"column_1"`
|
||||||
|
NodeID string `json:"node_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
Timestamp_2 time.Time `json:"timestamp_2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type AggregateNodeMetricsRow struct {
|
||||||
|
Bucket interface{} `json:"bucket"`
|
||||||
|
AvgCpu float64 `json:"avg_cpu"`
|
||||||
|
AvgMemory float64 `json:"avg_memory"`
|
||||||
|
AvgStorage float64 `json:"avg_storage"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Portable bucket bucketing (no TimescaleDB): floor epoch seconds into
|
||||||
|
// bucket-sized buckets, then back to timestamptz.
|
||||||
|
func (q *Queries) AggregateNodeMetrics(ctx context.Context, arg AggregateNodeMetricsParams) ([]AggregateNodeMetricsRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, aggregateNodeMetrics,
|
||||||
|
arg.Column1,
|
||||||
|
arg.NodeID,
|
||||||
|
arg.Timestamp,
|
||||||
|
arg.Timestamp_2,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []AggregateNodeMetricsRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i AggregateNodeMetricsRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.Bucket,
|
||||||
|
&i.AvgCpu,
|
||||||
|
&i.AvgMemory,
|
||||||
|
&i.AvgStorage,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listInstanceMetrics = `-- name: ListInstanceMetrics :many
|
||||||
|
SELECT instance_id, node_id, status, cpu, memory,
|
||||||
|
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
||||||
|
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
||||||
|
start_time, last_seen, health_status, health_last_check, health_check_count, health_failure_count
|
||||||
|
FROM instance_metrics
|
||||||
|
WHERE service_id = $1 AND timestamp = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListInstanceMetricsParams struct {
|
||||||
|
ServiceID string `json:"service_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ListInstanceMetricsRow struct {
|
||||||
|
InstanceID string `json:"instance_id"`
|
||||||
|
NodeID sql.NullString `json:"node_id"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
Cpu sql.NullString `json:"cpu"`
|
||||||
|
Memory sql.NullInt64 `json:"memory"`
|
||||||
|
NetworkBytesIn sql.NullInt64 `json:"network_bytes_in"`
|
||||||
|
NetworkBytesOut sql.NullInt64 `json:"network_bytes_out"`
|
||||||
|
NetworkPacketsIn sql.NullInt64 `json:"network_packets_in"`
|
||||||
|
NetworkPacketsOut sql.NullInt64 `json:"network_packets_out"`
|
||||||
|
NetworkConnectionsIn sql.NullInt32 `json:"network_connections_in"`
|
||||||
|
NetworkConnectionsOut sql.NullInt32 `json:"network_connections_out"`
|
||||||
|
NetworkErrorsIn sql.NullInt64 `json:"network_errors_in"`
|
||||||
|
NetworkErrorsOut sql.NullInt64 `json:"network_errors_out"`
|
||||||
|
StartTime sql.NullTime `json:"start_time"`
|
||||||
|
LastSeen sql.NullTime `json:"last_seen"`
|
||||||
|
HealthStatus sql.NullString `json:"health_status"`
|
||||||
|
HealthLastCheck sql.NullTime `json:"health_last_check"`
|
||||||
|
HealthCheckCount sql.NullInt32 `json:"health_check_count"`
|
||||||
|
HealthFailureCount sql.NullInt32 `json:"health_failure_count"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListInstanceMetrics(ctx context.Context, arg ListInstanceMetricsParams) ([]ListInstanceMetricsRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listInstanceMetrics, arg.ServiceID, arg.Timestamp)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListInstanceMetricsRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListInstanceMetricsRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.InstanceID,
|
||||||
|
&i.NodeID,
|
||||||
|
&i.Status,
|
||||||
|
&i.Cpu,
|
||||||
|
&i.Memory,
|
||||||
|
&i.NetworkBytesIn,
|
||||||
|
&i.NetworkBytesOut,
|
||||||
|
&i.NetworkPacketsIn,
|
||||||
|
&i.NetworkPacketsOut,
|
||||||
|
&i.NetworkConnectionsIn,
|
||||||
|
&i.NetworkConnectionsOut,
|
||||||
|
&i.NetworkErrorsIn,
|
||||||
|
&i.NetworkErrorsOut,
|
||||||
|
&i.StartTime,
|
||||||
|
&i.LastSeen,
|
||||||
|
&i.HealthStatus,
|
||||||
|
&i.HealthLastCheck,
|
||||||
|
&i.HealthCheckCount,
|
||||||
|
&i.HealthFailureCount,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listNodeMetrics = `-- name: ListNodeMetrics :many
|
||||||
|
SELECT node_id, timestamp, cpu_usage, cpu_cores, load_avg_1, load_avg_5, load_avg_15,
|
||||||
|
memory_total, memory_used, memory_available, memory_usage_percent,
|
||||||
|
storage_total, storage_used, storage_available, storage_usage_percent,
|
||||||
|
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
||||||
|
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
||||||
|
coalesce(extract(epoch from uptime), 0)::bigint AS uptime_seconds,
|
||||||
|
processes, os, kernel, architecture
|
||||||
|
FROM node_metrics
|
||||||
|
WHERE node_id = $1 AND timestamp BETWEEN $2 AND $3
|
||||||
|
ORDER BY timestamp ASC
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListNodeMetricsParams struct {
|
||||||
|
NodeID string `json:"node_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
Timestamp_2 time.Time `json:"timestamp_2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ListNodeMetricsRow struct {
|
||||||
|
NodeID string `json:"node_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
CpuUsage sql.NullString `json:"cpu_usage"`
|
||||||
|
CpuCores sql.NullString `json:"cpu_cores"`
|
||||||
|
LoadAvg1 sql.NullString `json:"load_avg_1"`
|
||||||
|
LoadAvg5 sql.NullString `json:"load_avg_5"`
|
||||||
|
LoadAvg15 sql.NullString `json:"load_avg_15"`
|
||||||
|
MemoryTotal sql.NullInt64 `json:"memory_total"`
|
||||||
|
MemoryUsed sql.NullInt64 `json:"memory_used"`
|
||||||
|
MemoryAvailable sql.NullInt64 `json:"memory_available"`
|
||||||
|
MemoryUsagePercent sql.NullString `json:"memory_usage_percent"`
|
||||||
|
StorageTotal sql.NullInt64 `json:"storage_total"`
|
||||||
|
StorageUsed sql.NullInt64 `json:"storage_used"`
|
||||||
|
StorageAvailable sql.NullInt64 `json:"storage_available"`
|
||||||
|
StorageUsagePercent sql.NullString `json:"storage_usage_percent"`
|
||||||
|
NetworkBytesIn sql.NullInt64 `json:"network_bytes_in"`
|
||||||
|
NetworkBytesOut sql.NullInt64 `json:"network_bytes_out"`
|
||||||
|
NetworkPacketsIn sql.NullInt64 `json:"network_packets_in"`
|
||||||
|
NetworkPacketsOut sql.NullInt64 `json:"network_packets_out"`
|
||||||
|
NetworkConnectionsIn sql.NullInt32 `json:"network_connections_in"`
|
||||||
|
NetworkConnectionsOut sql.NullInt32 `json:"network_connections_out"`
|
||||||
|
NetworkErrorsIn sql.NullInt64 `json:"network_errors_in"`
|
||||||
|
NetworkErrorsOut sql.NullInt64 `json:"network_errors_out"`
|
||||||
|
UptimeSeconds int64 `json:"uptime_seconds"`
|
||||||
|
Processes sql.NullInt32 `json:"processes"`
|
||||||
|
Os sql.NullString `json:"os"`
|
||||||
|
Kernel sql.NullString `json:"kernel"`
|
||||||
|
Architecture sql.NullString `json:"architecture"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListNodeMetrics(ctx context.Context, arg ListNodeMetricsParams) ([]ListNodeMetricsRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listNodeMetrics, arg.NodeID, arg.Timestamp, arg.Timestamp_2)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListNodeMetricsRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListNodeMetricsRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.NodeID,
|
||||||
|
&i.Timestamp,
|
||||||
|
&i.CpuUsage,
|
||||||
|
&i.CpuCores,
|
||||||
|
&i.LoadAvg1,
|
||||||
|
&i.LoadAvg5,
|
||||||
|
&i.LoadAvg15,
|
||||||
|
&i.MemoryTotal,
|
||||||
|
&i.MemoryUsed,
|
||||||
|
&i.MemoryAvailable,
|
||||||
|
&i.MemoryUsagePercent,
|
||||||
|
&i.StorageTotal,
|
||||||
|
&i.StorageUsed,
|
||||||
|
&i.StorageAvailable,
|
||||||
|
&i.StorageUsagePercent,
|
||||||
|
&i.NetworkBytesIn,
|
||||||
|
&i.NetworkBytesOut,
|
||||||
|
&i.NetworkPacketsIn,
|
||||||
|
&i.NetworkPacketsOut,
|
||||||
|
&i.NetworkConnectionsIn,
|
||||||
|
&i.NetworkConnectionsOut,
|
||||||
|
&i.NetworkErrorsIn,
|
||||||
|
&i.NetworkErrorsOut,
|
||||||
|
&i.UptimeSeconds,
|
||||||
|
&i.Processes,
|
||||||
|
&i.Os,
|
||||||
|
&i.Kernel,
|
||||||
|
&i.Architecture,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listServiceMetrics = `-- name: ListServiceMetrics :many
|
||||||
|
SELECT service_id, service_name, project_id, timestamp,
|
||||||
|
requests_total, requests_success, requests_errors, requests_avg_latency,
|
||||||
|
requests_p95_latency, requests_p99_latency, requests_throughput,
|
||||||
|
errors_total, errors_rate, performance_response_time, performance_throughput,
|
||||||
|
performance_concurrency, performance_saturation, performance_utilization,
|
||||||
|
resource_cpu_usage, resource_memory_usage, resource_storage_usage,
|
||||||
|
resource_network_usage, resource_score
|
||||||
|
FROM service_metrics
|
||||||
|
WHERE service_id = $1 AND timestamp BETWEEN $2 AND $3
|
||||||
|
ORDER BY timestamp ASC
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListServiceMetricsParams struct {
|
||||||
|
ServiceID string `json:"service_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
Timestamp_2 time.Time `json:"timestamp_2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ListServiceMetricsRow struct {
|
||||||
|
ServiceID string `json:"service_id"`
|
||||||
|
ServiceName string `json:"service_name"`
|
||||||
|
ProjectID string `json:"project_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
RequestsTotal sql.NullInt64 `json:"requests_total"`
|
||||||
|
RequestsSuccess sql.NullInt64 `json:"requests_success"`
|
||||||
|
RequestsErrors sql.NullInt64 `json:"requests_errors"`
|
||||||
|
RequestsAvgLatency sql.NullString `json:"requests_avg_latency"`
|
||||||
|
RequestsP95Latency sql.NullString `json:"requests_p95_latency"`
|
||||||
|
RequestsP99Latency sql.NullString `json:"requests_p99_latency"`
|
||||||
|
RequestsThroughput sql.NullString `json:"requests_throughput"`
|
||||||
|
ErrorsTotal sql.NullInt64 `json:"errors_total"`
|
||||||
|
ErrorsRate sql.NullString `json:"errors_rate"`
|
||||||
|
PerformanceResponseTime sql.NullString `json:"performance_response_time"`
|
||||||
|
PerformanceThroughput sql.NullString `json:"performance_throughput"`
|
||||||
|
PerformanceConcurrency sql.NullInt64 `json:"performance_concurrency"`
|
||||||
|
PerformanceSaturation sql.NullString `json:"performance_saturation"`
|
||||||
|
PerformanceUtilization sql.NullString `json:"performance_utilization"`
|
||||||
|
ResourceCpuUsage sql.NullString `json:"resource_cpu_usage"`
|
||||||
|
ResourceMemoryUsage sql.NullInt64 `json:"resource_memory_usage"`
|
||||||
|
ResourceStorageUsage sql.NullInt64 `json:"resource_storage_usage"`
|
||||||
|
ResourceNetworkUsage sql.NullInt64 `json:"resource_network_usage"`
|
||||||
|
ResourceScore sql.NullString `json:"resource_score"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListServiceMetrics(ctx context.Context, arg ListServiceMetricsParams) ([]ListServiceMetricsRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listServiceMetrics, arg.ServiceID, arg.Timestamp, arg.Timestamp_2)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListServiceMetricsRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListServiceMetricsRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.ServiceName,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.Timestamp,
|
||||||
|
&i.RequestsTotal,
|
||||||
|
&i.RequestsSuccess,
|
||||||
|
&i.RequestsErrors,
|
||||||
|
&i.RequestsAvgLatency,
|
||||||
|
&i.RequestsP95Latency,
|
||||||
|
&i.RequestsP99Latency,
|
||||||
|
&i.RequestsThroughput,
|
||||||
|
&i.ErrorsTotal,
|
||||||
|
&i.ErrorsRate,
|
||||||
|
&i.PerformanceResponseTime,
|
||||||
|
&i.PerformanceThroughput,
|
||||||
|
&i.PerformanceConcurrency,
|
||||||
|
&i.PerformanceSaturation,
|
||||||
|
&i.PerformanceUtilization,
|
||||||
|
&i.ResourceCpuUsage,
|
||||||
|
&i.ResourceMemoryUsage,
|
||||||
|
&i.ResourceStorageUsage,
|
||||||
|
&i.ResourceNetworkUsage,
|
||||||
|
&i.ResourceScore,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const upsertInstanceMetrics = `-- name: UpsertInstanceMetrics :exec
|
||||||
|
INSERT INTO instance_metrics (
|
||||||
|
service_id, timestamp, instance_id, node_id, status, cpu, memory,
|
||||||
|
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
||||||
|
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
||||||
|
start_time, last_seen, health_status, health_last_check, health_check_count, health_failure_count
|
||||||
|
) VALUES (
|
||||||
|
$1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15,
|
||||||
|
$16, $17, $18, $19, $20, $21
|
||||||
|
)
|
||||||
|
ON CONFLICT (service_id, timestamp, instance_id) DO UPDATE SET
|
||||||
|
node_id = EXCLUDED.node_id,
|
||||||
|
status = EXCLUDED.status,
|
||||||
|
cpu = EXCLUDED.cpu,
|
||||||
|
memory = EXCLUDED.memory,
|
||||||
|
network_bytes_in = EXCLUDED.network_bytes_in,
|
||||||
|
network_bytes_out = EXCLUDED.network_bytes_out,
|
||||||
|
network_packets_in = EXCLUDED.network_packets_in,
|
||||||
|
network_packets_out = EXCLUDED.network_packets_out,
|
||||||
|
network_connections_in = EXCLUDED.network_connections_in,
|
||||||
|
network_connections_out = EXCLUDED.network_connections_out,
|
||||||
|
network_errors_in = EXCLUDED.network_errors_in,
|
||||||
|
network_errors_out = EXCLUDED.network_errors_out,
|
||||||
|
start_time = EXCLUDED.start_time,
|
||||||
|
last_seen = EXCLUDED.last_seen,
|
||||||
|
health_status = EXCLUDED.health_status,
|
||||||
|
health_last_check = EXCLUDED.health_last_check,
|
||||||
|
health_check_count = EXCLUDED.health_check_count,
|
||||||
|
health_failure_count = EXCLUDED.health_failure_count
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpsertInstanceMetricsParams struct {
|
||||||
|
ServiceID string `json:"service_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
InstanceID string `json:"instance_id"`
|
||||||
|
NodeID sql.NullString `json:"node_id"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
Cpu sql.NullString `json:"cpu"`
|
||||||
|
Memory sql.NullInt64 `json:"memory"`
|
||||||
|
NetworkBytesIn sql.NullInt64 `json:"network_bytes_in"`
|
||||||
|
NetworkBytesOut sql.NullInt64 `json:"network_bytes_out"`
|
||||||
|
NetworkPacketsIn sql.NullInt64 `json:"network_packets_in"`
|
||||||
|
NetworkPacketsOut sql.NullInt64 `json:"network_packets_out"`
|
||||||
|
NetworkConnectionsIn sql.NullInt32 `json:"network_connections_in"`
|
||||||
|
NetworkConnectionsOut sql.NullInt32 `json:"network_connections_out"`
|
||||||
|
NetworkErrorsIn sql.NullInt64 `json:"network_errors_in"`
|
||||||
|
NetworkErrorsOut sql.NullInt64 `json:"network_errors_out"`
|
||||||
|
StartTime sql.NullTime `json:"start_time"`
|
||||||
|
LastSeen sql.NullTime `json:"last_seen"`
|
||||||
|
HealthStatus sql.NullString `json:"health_status"`
|
||||||
|
HealthLastCheck sql.NullTime `json:"health_last_check"`
|
||||||
|
HealthCheckCount sql.NullInt32 `json:"health_check_count"`
|
||||||
|
HealthFailureCount sql.NullInt32 `json:"health_failure_count"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpsertInstanceMetrics(ctx context.Context, arg UpsertInstanceMetricsParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, upsertInstanceMetrics,
|
||||||
|
arg.ServiceID,
|
||||||
|
arg.Timestamp,
|
||||||
|
arg.InstanceID,
|
||||||
|
arg.NodeID,
|
||||||
|
arg.Status,
|
||||||
|
arg.Cpu,
|
||||||
|
arg.Memory,
|
||||||
|
arg.NetworkBytesIn,
|
||||||
|
arg.NetworkBytesOut,
|
||||||
|
arg.NetworkPacketsIn,
|
||||||
|
arg.NetworkPacketsOut,
|
||||||
|
arg.NetworkConnectionsIn,
|
||||||
|
arg.NetworkConnectionsOut,
|
||||||
|
arg.NetworkErrorsIn,
|
||||||
|
arg.NetworkErrorsOut,
|
||||||
|
arg.StartTime,
|
||||||
|
arg.LastSeen,
|
||||||
|
arg.HealthStatus,
|
||||||
|
arg.HealthLastCheck,
|
||||||
|
arg.HealthCheckCount,
|
||||||
|
arg.HealthFailureCount,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const upsertNodeMetrics = `-- name: UpsertNodeMetrics :exec
|
||||||
|
INSERT INTO node_metrics (
|
||||||
|
node_id, timestamp, cpu_usage, cpu_cores, load_avg_1, load_avg_5, load_avg_15,
|
||||||
|
memory_total, memory_used, memory_available, memory_usage_percent,
|
||||||
|
storage_total, storage_used, storage_available, storage_usage_percent,
|
||||||
|
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
||||||
|
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
||||||
|
uptime, processes, os, kernel, architecture
|
||||||
|
) VALUES (
|
||||||
|
$1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15,
|
||||||
|
$16, $17, $18, $19, $20, $21, $22, $23, $24, $25, $26, $27, $28
|
||||||
|
)
|
||||||
|
ON CONFLICT (node_id, timestamp) DO UPDATE SET
|
||||||
|
cpu_usage = EXCLUDED.cpu_usage,
|
||||||
|
cpu_cores = EXCLUDED.cpu_cores,
|
||||||
|
load_avg_1 = EXCLUDED.load_avg_1,
|
||||||
|
load_avg_5 = EXCLUDED.load_avg_5,
|
||||||
|
load_avg_15 = EXCLUDED.load_avg_15,
|
||||||
|
memory_total = EXCLUDED.memory_total,
|
||||||
|
memory_used = EXCLUDED.memory_used,
|
||||||
|
memory_available = EXCLUDED.memory_available,
|
||||||
|
memory_usage_percent = EXCLUDED.memory_usage_percent,
|
||||||
|
storage_total = EXCLUDED.storage_total,
|
||||||
|
storage_used = EXCLUDED.storage_used,
|
||||||
|
storage_available = EXCLUDED.storage_available,
|
||||||
|
storage_usage_percent = EXCLUDED.storage_usage_percent,
|
||||||
|
network_bytes_in = EXCLUDED.network_bytes_in,
|
||||||
|
network_bytes_out = EXCLUDED.network_bytes_out,
|
||||||
|
network_packets_in = EXCLUDED.network_packets_in,
|
||||||
|
network_packets_out = EXCLUDED.network_packets_out,
|
||||||
|
network_connections_in = EXCLUDED.network_connections_in,
|
||||||
|
network_connections_out = EXCLUDED.network_connections_out,
|
||||||
|
network_errors_in = EXCLUDED.network_errors_in,
|
||||||
|
network_errors_out = EXCLUDED.network_errors_out,
|
||||||
|
uptime = EXCLUDED.uptime,
|
||||||
|
processes = EXCLUDED.processes,
|
||||||
|
os = EXCLUDED.os,
|
||||||
|
kernel = EXCLUDED.kernel,
|
||||||
|
architecture = EXCLUDED.architecture
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpsertNodeMetricsParams struct {
|
||||||
|
NodeID string `json:"node_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
CpuUsage sql.NullString `json:"cpu_usage"`
|
||||||
|
CpuCores sql.NullString `json:"cpu_cores"`
|
||||||
|
LoadAvg1 sql.NullString `json:"load_avg_1"`
|
||||||
|
LoadAvg5 sql.NullString `json:"load_avg_5"`
|
||||||
|
LoadAvg15 sql.NullString `json:"load_avg_15"`
|
||||||
|
MemoryTotal sql.NullInt64 `json:"memory_total"`
|
||||||
|
MemoryUsed sql.NullInt64 `json:"memory_used"`
|
||||||
|
MemoryAvailable sql.NullInt64 `json:"memory_available"`
|
||||||
|
MemoryUsagePercent sql.NullString `json:"memory_usage_percent"`
|
||||||
|
StorageTotal sql.NullInt64 `json:"storage_total"`
|
||||||
|
StorageUsed sql.NullInt64 `json:"storage_used"`
|
||||||
|
StorageAvailable sql.NullInt64 `json:"storage_available"`
|
||||||
|
StorageUsagePercent sql.NullString `json:"storage_usage_percent"`
|
||||||
|
NetworkBytesIn sql.NullInt64 `json:"network_bytes_in"`
|
||||||
|
NetworkBytesOut sql.NullInt64 `json:"network_bytes_out"`
|
||||||
|
NetworkPacketsIn sql.NullInt64 `json:"network_packets_in"`
|
||||||
|
NetworkPacketsOut sql.NullInt64 `json:"network_packets_out"`
|
||||||
|
NetworkConnectionsIn sql.NullInt32 `json:"network_connections_in"`
|
||||||
|
NetworkConnectionsOut sql.NullInt32 `json:"network_connections_out"`
|
||||||
|
NetworkErrorsIn sql.NullInt64 `json:"network_errors_in"`
|
||||||
|
NetworkErrorsOut sql.NullInt64 `json:"network_errors_out"`
|
||||||
|
Uptime sql.NullInt64 `json:"uptime"`
|
||||||
|
Processes sql.NullInt32 `json:"processes"`
|
||||||
|
Os sql.NullString `json:"os"`
|
||||||
|
Kernel sql.NullString `json:"kernel"`
|
||||||
|
Architecture sql.NullString `json:"architecture"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpsertNodeMetrics(ctx context.Context, arg UpsertNodeMetricsParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, upsertNodeMetrics,
|
||||||
|
arg.NodeID,
|
||||||
|
arg.Timestamp,
|
||||||
|
arg.CpuUsage,
|
||||||
|
arg.CpuCores,
|
||||||
|
arg.LoadAvg1,
|
||||||
|
arg.LoadAvg5,
|
||||||
|
arg.LoadAvg15,
|
||||||
|
arg.MemoryTotal,
|
||||||
|
arg.MemoryUsed,
|
||||||
|
arg.MemoryAvailable,
|
||||||
|
arg.MemoryUsagePercent,
|
||||||
|
arg.StorageTotal,
|
||||||
|
arg.StorageUsed,
|
||||||
|
arg.StorageAvailable,
|
||||||
|
arg.StorageUsagePercent,
|
||||||
|
arg.NetworkBytesIn,
|
||||||
|
arg.NetworkBytesOut,
|
||||||
|
arg.NetworkPacketsIn,
|
||||||
|
arg.NetworkPacketsOut,
|
||||||
|
arg.NetworkConnectionsIn,
|
||||||
|
arg.NetworkConnectionsOut,
|
||||||
|
arg.NetworkErrorsIn,
|
||||||
|
arg.NetworkErrorsOut,
|
||||||
|
arg.Uptime,
|
||||||
|
arg.Processes,
|
||||||
|
arg.Os,
|
||||||
|
arg.Kernel,
|
||||||
|
arg.Architecture,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const upsertServiceMetrics = `-- name: UpsertServiceMetrics :exec
|
||||||
|
INSERT INTO service_metrics (
|
||||||
|
service_id, service_name, project_id, timestamp,
|
||||||
|
requests_total, requests_success, requests_errors, requests_avg_latency,
|
||||||
|
requests_p95_latency, requests_p99_latency, requests_throughput,
|
||||||
|
errors_total, errors_rate, performance_response_time, performance_throughput,
|
||||||
|
performance_concurrency, performance_saturation, performance_utilization,
|
||||||
|
resource_cpu_usage, resource_memory_usage, resource_storage_usage,
|
||||||
|
resource_network_usage, resource_score
|
||||||
|
) VALUES (
|
||||||
|
$1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15,
|
||||||
|
$16, $17, $18, $19, $20, $21, $22, $23
|
||||||
|
)
|
||||||
|
ON CONFLICT (service_id, timestamp) DO UPDATE SET
|
||||||
|
requests_total = EXCLUDED.requests_total,
|
||||||
|
requests_success = EXCLUDED.requests_success,
|
||||||
|
requests_errors = EXCLUDED.requests_errors,
|
||||||
|
requests_avg_latency = EXCLUDED.requests_avg_latency,
|
||||||
|
requests_p95_latency = EXCLUDED.requests_p95_latency,
|
||||||
|
requests_p99_latency = EXCLUDED.requests_p99_latency,
|
||||||
|
requests_throughput = EXCLUDED.requests_throughput,
|
||||||
|
errors_total = EXCLUDED.errors_total,
|
||||||
|
errors_rate = EXCLUDED.errors_rate,
|
||||||
|
performance_response_time = EXCLUDED.performance_response_time,
|
||||||
|
performance_throughput = EXCLUDED.performance_throughput,
|
||||||
|
performance_concurrency = EXCLUDED.performance_concurrency,
|
||||||
|
performance_saturation = EXCLUDED.performance_saturation,
|
||||||
|
performance_utilization = EXCLUDED.performance_utilization,
|
||||||
|
resource_cpu_usage = EXCLUDED.resource_cpu_usage,
|
||||||
|
resource_memory_usage = EXCLUDED.resource_memory_usage,
|
||||||
|
resource_storage_usage = EXCLUDED.resource_storage_usage,
|
||||||
|
resource_network_usage = EXCLUDED.resource_network_usage,
|
||||||
|
resource_score = EXCLUDED.resource_score
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpsertServiceMetricsParams struct {
|
||||||
|
ServiceID string `json:"service_id"`
|
||||||
|
ServiceName string `json:"service_name"`
|
||||||
|
ProjectID string `json:"project_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
RequestsTotal sql.NullInt64 `json:"requests_total"`
|
||||||
|
RequestsSuccess sql.NullInt64 `json:"requests_success"`
|
||||||
|
RequestsErrors sql.NullInt64 `json:"requests_errors"`
|
||||||
|
RequestsAvgLatency sql.NullString `json:"requests_avg_latency"`
|
||||||
|
RequestsP95Latency sql.NullString `json:"requests_p95_latency"`
|
||||||
|
RequestsP99Latency sql.NullString `json:"requests_p99_latency"`
|
||||||
|
RequestsThroughput sql.NullString `json:"requests_throughput"`
|
||||||
|
ErrorsTotal sql.NullInt64 `json:"errors_total"`
|
||||||
|
ErrorsRate sql.NullString `json:"errors_rate"`
|
||||||
|
PerformanceResponseTime sql.NullString `json:"performance_response_time"`
|
||||||
|
PerformanceThroughput sql.NullString `json:"performance_throughput"`
|
||||||
|
PerformanceConcurrency sql.NullInt64 `json:"performance_concurrency"`
|
||||||
|
PerformanceSaturation sql.NullString `json:"performance_saturation"`
|
||||||
|
PerformanceUtilization sql.NullString `json:"performance_utilization"`
|
||||||
|
ResourceCpuUsage sql.NullString `json:"resource_cpu_usage"`
|
||||||
|
ResourceMemoryUsage sql.NullInt64 `json:"resource_memory_usage"`
|
||||||
|
ResourceStorageUsage sql.NullInt64 `json:"resource_storage_usage"`
|
||||||
|
ResourceNetworkUsage sql.NullInt64 `json:"resource_network_usage"`
|
||||||
|
ResourceScore sql.NullString `json:"resource_score"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpsertServiceMetrics(ctx context.Context, arg UpsertServiceMetricsParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, upsertServiceMetrics,
|
||||||
|
arg.ServiceID,
|
||||||
|
arg.ServiceName,
|
||||||
|
arg.ProjectID,
|
||||||
|
arg.Timestamp,
|
||||||
|
arg.RequestsTotal,
|
||||||
|
arg.RequestsSuccess,
|
||||||
|
arg.RequestsErrors,
|
||||||
|
arg.RequestsAvgLatency,
|
||||||
|
arg.RequestsP95Latency,
|
||||||
|
arg.RequestsP99Latency,
|
||||||
|
arg.RequestsThroughput,
|
||||||
|
arg.ErrorsTotal,
|
||||||
|
arg.ErrorsRate,
|
||||||
|
arg.PerformanceResponseTime,
|
||||||
|
arg.PerformanceThroughput,
|
||||||
|
arg.PerformanceConcurrency,
|
||||||
|
arg.PerformanceSaturation,
|
||||||
|
arg.PerformanceUtilization,
|
||||||
|
arg.ResourceCpuUsage,
|
||||||
|
arg.ResourceMemoryUsage,
|
||||||
|
arg.ResourceStorageUsage,
|
||||||
|
arg.ResourceNetworkUsage,
|
||||||
|
arg.ResourceScore,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
@@ -78,6 +78,28 @@ type Banner struct {
|
|||||||
UpdatedAt sql.NullTime `json:"updated_at"`
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type ComplianceFramework struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Description sql.NullString `json:"description"`
|
||||||
|
Version string `json:"version"`
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ComplianceReport struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
FrameworkID uuid.UUID `json:"framework_id"`
|
||||||
|
AssessmentDate time.Time `json:"assessment_date"`
|
||||||
|
Assessor sql.NullString `json:"assessor"`
|
||||||
|
OverallStatus string `json:"overall_status"`
|
||||||
|
Score int32 `json:"score"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
type ContainerInstance struct {
|
type ContainerInstance struct {
|
||||||
ID string `json:"id"`
|
ID string `json:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
@@ -98,6 +120,34 @@ type ContainerInstance struct {
|
|||||||
UpdatedAt sql.NullTime `json:"updated_at"`
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type CronExecution struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
CronJobID uuid.UUID `json:"cron_job_id"`
|
||||||
|
StartedAt time.Time `json:"started_at"`
|
||||||
|
FinishedAt sql.NullTime `json:"finished_at"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
Output sql.NullString `json:"output"`
|
||||||
|
Error sql.NullString `json:"error"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type CronJob struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Schedule string `json:"schedule"`
|
||||||
|
Command string `json:"command"`
|
||||||
|
Timezone sql.NullString `json:"timezone"`
|
||||||
|
Enabled sql.NullBool `json:"enabled"`
|
||||||
|
LastRunAt sql.NullTime `json:"last_run_at"`
|
||||||
|
NextRunAt sql.NullTime `json:"next_run_at"`
|
||||||
|
LastStatus sql.NullString `json:"last_status"`
|
||||||
|
LastOutput sql.NullString `json:"last_output"`
|
||||||
|
Retention sql.NullInt32 `json:"retention"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
type DatabaseBackup struct {
|
type DatabaseBackup struct {
|
||||||
ID string `json:"id"`
|
ID string `json:"id"`
|
||||||
DatabaseID string `json:"database_id"`
|
DatabaseID string `json:"database_id"`
|
||||||
@@ -136,11 +186,22 @@ type DatabaseService struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type Deployment struct {
|
type Deployment struct {
|
||||||
ID uuid.UUID `json:"id"`
|
ID uuid.UUID `json:"id"`
|
||||||
ServiceID uuid.UUID `json:"service_id"`
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
Status sql.NullString `json:"status"`
|
Version string `json:"version"`
|
||||||
CreatedAt sql.NullTime `json:"created_at"`
|
CommitHash sql.NullString `json:"commit_hash"`
|
||||||
UpdatedAt sql.NullTime `json:"updated_at"`
|
ImageDigest sql.NullString `json:"image_digest"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
BuildLog sql.NullString `json:"build_log"`
|
||||||
|
DeploymentLog sql.NullString `json:"deployment_log"`
|
||||||
|
StartedAt sql.NullTime `json:"started_at"`
|
||||||
|
CompletedAt sql.NullTime `json:"completed_at"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
ImageName sql.NullString `json:"image_name"`
|
||||||
|
ImageTag sql.NullString `json:"image_tag"`
|
||||||
|
RuntimeLog sql.NullString `json:"runtime_log"`
|
||||||
|
Error sql.NullString `json:"error"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type Environment struct {
|
type Environment struct {
|
||||||
@@ -161,6 +222,71 @@ type EnvironmentVariable struct {
|
|||||||
UpdatedAt sql.NullTime `json:"updated_at"`
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type GitProvider struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
DisplayName string `json:"display_name"`
|
||||||
|
ApiUrl string `json:"api_url"`
|
||||||
|
WebhookUrl string `json:"webhook_url"`
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GitRepository struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
FullName string `json:"full_name"`
|
||||||
|
Description sql.NullString `json:"description"`
|
||||||
|
CloneUrl string `json:"clone_url"`
|
||||||
|
WebhookUrl sql.NullString `json:"webhook_url"`
|
||||||
|
DefaultBranch sql.NullString `json:"default_branch"`
|
||||||
|
IsPrivate sql.NullBool `json:"is_private"`
|
||||||
|
UserID uuid.UUID `json:"user_id"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GitWebhook struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
RepoID uuid.UUID `json:"repo_id"`
|
||||||
|
ProviderID uuid.UUID `json:"provider_id"`
|
||||||
|
Events string `json:"events"`
|
||||||
|
WebhookSecret string `json:"webhook_secret"`
|
||||||
|
RemoteWebhookID sql.NullString `json:"remote_webhook_id"`
|
||||||
|
Active sql.NullBool `json:"active"`
|
||||||
|
BranchFilter sql.NullString `json:"branch_filter"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type InstanceMetric struct {
|
||||||
|
ServiceID string `json:"service_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
InstanceID string `json:"instance_id"`
|
||||||
|
NodeID sql.NullString `json:"node_id"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
Cpu sql.NullString `json:"cpu"`
|
||||||
|
Memory sql.NullInt64 `json:"memory"`
|
||||||
|
NetworkBytesIn sql.NullInt64 `json:"network_bytes_in"`
|
||||||
|
NetworkBytesOut sql.NullInt64 `json:"network_bytes_out"`
|
||||||
|
NetworkPacketsIn sql.NullInt64 `json:"network_packets_in"`
|
||||||
|
NetworkPacketsOut sql.NullInt64 `json:"network_packets_out"`
|
||||||
|
NetworkConnectionsIn sql.NullInt32 `json:"network_connections_in"`
|
||||||
|
NetworkConnectionsOut sql.NullInt32 `json:"network_connections_out"`
|
||||||
|
NetworkErrorsIn sql.NullInt64 `json:"network_errors_in"`
|
||||||
|
NetworkErrorsOut sql.NullInt64 `json:"network_errors_out"`
|
||||||
|
StartTime sql.NullTime `json:"start_time"`
|
||||||
|
LastSeen sql.NullTime `json:"last_seen"`
|
||||||
|
HealthStatus sql.NullString `json:"health_status"`
|
||||||
|
HealthLastCheck sql.NullTime `json:"health_last_check"`
|
||||||
|
HealthCheckCount sql.NullInt32 `json:"health_check_count"`
|
||||||
|
HealthFailureCount sql.NullInt32 `json:"health_failure_count"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
type NodeAgent struct {
|
type NodeAgent struct {
|
||||||
ID string `json:"id"`
|
ID string `json:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
@@ -181,6 +307,38 @@ type NodeAgent struct {
|
|||||||
DefaultDomain string `json:"default_domain"`
|
DefaultDomain string `json:"default_domain"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type NodeMetric struct {
|
||||||
|
NodeID string `json:"node_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
CpuUsage sql.NullString `json:"cpu_usage"`
|
||||||
|
CpuCores sql.NullString `json:"cpu_cores"`
|
||||||
|
LoadAvg1 sql.NullString `json:"load_avg_1"`
|
||||||
|
LoadAvg5 sql.NullString `json:"load_avg_5"`
|
||||||
|
LoadAvg15 sql.NullString `json:"load_avg_15"`
|
||||||
|
MemoryTotal sql.NullInt64 `json:"memory_total"`
|
||||||
|
MemoryUsed sql.NullInt64 `json:"memory_used"`
|
||||||
|
MemoryAvailable sql.NullInt64 `json:"memory_available"`
|
||||||
|
MemoryUsagePercent sql.NullString `json:"memory_usage_percent"`
|
||||||
|
StorageTotal sql.NullInt64 `json:"storage_total"`
|
||||||
|
StorageUsed sql.NullInt64 `json:"storage_used"`
|
||||||
|
StorageAvailable sql.NullInt64 `json:"storage_available"`
|
||||||
|
StorageUsagePercent sql.NullString `json:"storage_usage_percent"`
|
||||||
|
NetworkBytesIn sql.NullInt64 `json:"network_bytes_in"`
|
||||||
|
NetworkBytesOut sql.NullInt64 `json:"network_bytes_out"`
|
||||||
|
NetworkPacketsIn sql.NullInt64 `json:"network_packets_in"`
|
||||||
|
NetworkPacketsOut sql.NullInt64 `json:"network_packets_out"`
|
||||||
|
NetworkConnectionsIn sql.NullInt32 `json:"network_connections_in"`
|
||||||
|
NetworkConnectionsOut sql.NullInt32 `json:"network_connections_out"`
|
||||||
|
NetworkErrorsIn sql.NullInt64 `json:"network_errors_in"`
|
||||||
|
NetworkErrorsOut sql.NullInt64 `json:"network_errors_out"`
|
||||||
|
Uptime sql.NullInt64 `json:"uptime"`
|
||||||
|
Processes sql.NullInt32 `json:"processes"`
|
||||||
|
Os sql.NullString `json:"os"`
|
||||||
|
Kernel sql.NullString `json:"kernel"`
|
||||||
|
Architecture sql.NullString `json:"architecture"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
type Notification struct {
|
type Notification struct {
|
||||||
ID uuid.UUID `json:"id"`
|
ID uuid.UUID `json:"id"`
|
||||||
UserID uuid.UUID `json:"user_id"`
|
UserID uuid.UUID `json:"user_id"`
|
||||||
@@ -217,6 +375,21 @@ type OutboundWebhook struct {
|
|||||||
UpdatedAt sql.NullTime `json:"updated_at"`
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type PreviewEnvironment struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
BranchName string `json:"branch_name"`
|
||||||
|
PrNumber sql.NullInt32 `json:"pr_number"`
|
||||||
|
Environment string `json:"environment"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
Url sql.NullString `json:"url"`
|
||||||
|
ExpiresAt time.Time `json:"expires_at"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
PreviewServiceID uuid.NullUUID `json:"preview_service_id"`
|
||||||
|
}
|
||||||
|
|
||||||
type Project struct {
|
type Project struct {
|
||||||
ID uuid.UUID `json:"id"`
|
ID uuid.UUID `json:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
@@ -246,52 +419,67 @@ type Registry struct {
|
|||||||
UpdatedAt time.Time `json:"updated_at"`
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type SecurityScan struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
ServiceID uuid.NullUUID `json:"service_id"`
|
||||||
|
ScanType string `json:"scan_type"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
StartedAt time.Time `json:"started_at"`
|
||||||
|
CompletedAt sql.NullTime `json:"completed_at"`
|
||||||
|
Summary pqtype.NullRawMessage `json:"summary"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
type Service struct {
|
type Service struct {
|
||||||
ID uuid.UUID `json:"id"`
|
ID uuid.UUID `json:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
Description sql.NullString `json:"description"`
|
Description sql.NullString `json:"description"`
|
||||||
ProjectID uuid.UUID `json:"project_id"`
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
EnvironmentID uuid.UUID `json:"environment_id"`
|
EnvironmentID uuid.UUID `json:"environment_id"`
|
||||||
ServiceType string `json:"service_type"`
|
ServiceType string `json:"service_type"`
|
||||||
SourceType string `json:"source_type"`
|
SourceType string `json:"source_type"`
|
||||||
SourceUrl sql.NullString `json:"source_url"`
|
SourceUrl sql.NullString `json:"source_url"`
|
||||||
ImageName sql.NullString `json:"image_name"`
|
ImageName sql.NullString `json:"image_name"`
|
||||||
BuildCommand sql.NullString `json:"build_command"`
|
BuildCommand sql.NullString `json:"build_command"`
|
||||||
StartCommand sql.NullString `json:"start_command"`
|
StartCommand sql.NullString `json:"start_command"`
|
||||||
CpuLimit sql.NullInt32 `json:"cpu_limit"`
|
CpuLimit sql.NullInt32 `json:"cpu_limit"`
|
||||||
MemoryLimit sql.NullInt32 `json:"memory_limit"`
|
MemoryLimit sql.NullInt32 `json:"memory_limit"`
|
||||||
PublicUrl sql.NullString `json:"public_url"`
|
PublicUrl sql.NullString `json:"public_url"`
|
||||||
HealthCheckUrl sql.NullString `json:"health_check_url"`
|
HealthCheckUrl sql.NullString `json:"health_check_url"`
|
||||||
Status sql.NullString `json:"status"`
|
Status sql.NullString `json:"status"`
|
||||||
CreatedAt sql.NullTime `json:"created_at"`
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
UpdatedAt sql.NullTime `json:"updated_at"`
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
Type sql.NullString `json:"type"`
|
Type sql.NullString `json:"type"`
|
||||||
Image sql.NullString `json:"image"`
|
Image sql.NullString `json:"image"`
|
||||||
Command sql.NullString `json:"command"`
|
Command sql.NullString `json:"command"`
|
||||||
Environment sql.NullString `json:"environment"`
|
Environment sql.NullString `json:"environment"`
|
||||||
GitRepo sql.NullString `json:"git_repo"`
|
GitRepo sql.NullString `json:"git_repo"`
|
||||||
GitBranch sql.NullString `json:"git_branch"`
|
GitBranch sql.NullString `json:"git_branch"`
|
||||||
BuildPath sql.NullString `json:"build_path"`
|
BuildPath sql.NullString `json:"build_path"`
|
||||||
Cpu sql.NullString `json:"cpu"`
|
Cpu sql.NullString `json:"cpu"`
|
||||||
Memory sql.NullString `json:"memory"`
|
Memory sql.NullString `json:"memory"`
|
||||||
Replicas int32 `json:"replicas"`
|
Replicas int32 `json:"replicas"`
|
||||||
Port int32 `json:"port"`
|
Port int32 `json:"port"`
|
||||||
Domain string `json:"domain"`
|
Domain string `json:"domain"`
|
||||||
HealthcheckPath string `json:"healthcheck_path"`
|
HealthcheckPath string `json:"healthcheck_path"`
|
||||||
RestartPolicy string `json:"restart_policy"`
|
RestartPolicy string `json:"restart_policy"`
|
||||||
PublishedPort int32 `json:"published_port"`
|
PublishedPort int32 `json:"published_port"`
|
||||||
Volumes json.RawMessage `json:"volumes"`
|
Volumes json.RawMessage `json:"volumes"`
|
||||||
MaintenanceMode bool `json:"maintenance_mode"`
|
MaintenanceMode bool `json:"maintenance_mode"`
|
||||||
BasicAuthUsers string `json:"basic_auth_users"`
|
BasicAuthUsers string `json:"basic_auth_users"`
|
||||||
Builder string `json:"builder"`
|
Builder string `json:"builder"`
|
||||||
CpuReserve string `json:"cpu_reserve"`
|
CpuReserve string `json:"cpu_reserve"`
|
||||||
MemoryReserve string `json:"memory_reserve"`
|
MemoryReserve string `json:"memory_reserve"`
|
||||||
StaticBuildCmd string `json:"static_build_cmd"`
|
StaticBuildCmd string `json:"static_build_cmd"`
|
||||||
StaticDir string `json:"static_dir"`
|
StaticDir string `json:"static_dir"`
|
||||||
NodeID sql.NullString `json:"node_id"`
|
NodeID sql.NullString `json:"node_id"`
|
||||||
Spread bool `json:"spread"`
|
Spread bool `json:"spread"`
|
||||||
PlacementTags json.RawMessage `json:"placement_tags"`
|
SleepEnabled bool `json:"sleep_enabled"`
|
||||||
TraefikLabels json.RawMessage `json:"traefik_labels"`
|
SleepIdleMinutes int32 `json:"sleep_idle_minutes"`
|
||||||
|
PlacementTags json.RawMessage `json:"placement_tags"`
|
||||||
|
TraefikLabels json.RawMessage `json:"traefik_labels"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type ServiceDomain struct {
|
type ServiceDomain struct {
|
||||||
@@ -305,6 +493,33 @@ type ServiceDomain struct {
|
|||||||
CreatedAt time.Time `json:"created_at"`
|
CreatedAt time.Time `json:"created_at"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type ServiceMetric struct {
|
||||||
|
ServiceID string `json:"service_id"`
|
||||||
|
ServiceName string `json:"service_name"`
|
||||||
|
ProjectID string `json:"project_id"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
RequestsTotal sql.NullInt64 `json:"requests_total"`
|
||||||
|
RequestsSuccess sql.NullInt64 `json:"requests_success"`
|
||||||
|
RequestsErrors sql.NullInt64 `json:"requests_errors"`
|
||||||
|
RequestsAvgLatency sql.NullString `json:"requests_avg_latency"`
|
||||||
|
RequestsP95Latency sql.NullString `json:"requests_p95_latency"`
|
||||||
|
RequestsP99Latency sql.NullString `json:"requests_p99_latency"`
|
||||||
|
RequestsThroughput sql.NullString `json:"requests_throughput"`
|
||||||
|
ErrorsTotal sql.NullInt64 `json:"errors_total"`
|
||||||
|
ErrorsRate sql.NullString `json:"errors_rate"`
|
||||||
|
PerformanceResponseTime sql.NullString `json:"performance_response_time"`
|
||||||
|
PerformanceThroughput sql.NullString `json:"performance_throughput"`
|
||||||
|
PerformanceConcurrency sql.NullInt64 `json:"performance_concurrency"`
|
||||||
|
PerformanceSaturation sql.NullString `json:"performance_saturation"`
|
||||||
|
PerformanceUtilization sql.NullString `json:"performance_utilization"`
|
||||||
|
ResourceCpuUsage sql.NullString `json:"resource_cpu_usage"`
|
||||||
|
ResourceMemoryUsage sql.NullInt64 `json:"resource_memory_usage"`
|
||||||
|
ResourceStorageUsage sql.NullInt64 `json:"resource_storage_usage"`
|
||||||
|
ResourceNetworkUsage sql.NullInt64 `json:"resource_network_usage"`
|
||||||
|
ResourceScore sql.NullString `json:"resource_score"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
type ServiceTemplate struct {
|
type ServiceTemplate struct {
|
||||||
ID string `json:"id"`
|
ID string `json:"id"`
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
@@ -320,6 +535,17 @@ type ServiceTemplate struct {
|
|||||||
UpdatedAt sql.NullTime `json:"updated_at"`
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type User struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Email string `json:"email"`
|
||||||
|
PasswordHash string `json:"password_hash"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
AvatarUrl sql.NullString `json:"avatar_url"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
IsAdmin bool `json:"is_admin"`
|
||||||
|
}
|
||||||
|
|
||||||
type UserInvite struct {
|
type UserInvite struct {
|
||||||
ID uuid.UUID `json:"id"`
|
ID uuid.UUID `json:"id"`
|
||||||
TokenHash string `json:"token_hash"`
|
TokenHash string `json:"token_hash"`
|
||||||
@@ -344,6 +570,21 @@ type UserToken struct {
|
|||||||
CreatedAt sql.NullTime `json:"created_at"`
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type Vulnerability struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Type string `json:"type"`
|
||||||
|
Severity string `json:"severity"`
|
||||||
|
Title string `json:"title"`
|
||||||
|
Description sql.NullString `json:"description"`
|
||||||
|
ServiceID uuid.NullUUID `json:"service_id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
FoundAt time.Time `json:"found_at"`
|
||||||
|
ResolvedAt sql.NullTime `json:"resolved_at"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
type WebhookDelivery struct {
|
type WebhookDelivery struct {
|
||||||
ID uuid.UUID `json:"id"`
|
ID uuid.UUID `json:"id"`
|
||||||
WebhookID uuid.UUID `json:"webhook_id"`
|
WebhookID uuid.UUID `json:"webhook_id"`
|
||||||
|
|||||||
@@ -0,0 +1,635 @@
|
|||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.31.1
|
||||||
|
// source: preview_environments.sql
|
||||||
|
|
||||||
|
package sqlcdb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
const countActivePreviewsForBranch = `-- name: CountActivePreviewsForBranch :one
|
||||||
|
SELECT COUNT(*) FROM preview_environments
|
||||||
|
WHERE service_id = $1 AND branch_name = $2 AND status NOT IN ('expired', 'stopped')
|
||||||
|
`
|
||||||
|
|
||||||
|
type CountActivePreviewsForBranchParams struct {
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
BranchName string `json:"branch_name"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CountActivePreviewsForBranch(ctx context.Context, arg CountActivePreviewsForBranchParams) (int64, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, countActivePreviewsForBranch, arg.ServiceID, arg.BranchName)
|
||||||
|
var count int64
|
||||||
|
err := row.Scan(&count)
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const createPreviewDomain = `-- name: CreatePreviewDomain :exec
|
||||||
|
INSERT INTO service_domains (service_id, domain, is_default, cert_type) VALUES ($1, $2, true, 'tls')
|
||||||
|
`
|
||||||
|
|
||||||
|
type CreatePreviewDomainParams struct {
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CreatePreviewDomain(ctx context.Context, arg CreatePreviewDomainParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, createPreviewDomain, arg.ServiceID, arg.Domain)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const createPreviewEnvironment = `-- name: CreatePreviewEnvironment :exec
|
||||||
|
INSERT INTO preview_environments
|
||||||
|
(id, project_id, service_id, preview_service_id, branch_name, pr_number, environment,
|
||||||
|
status, url, expires_at, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)
|
||||||
|
`
|
||||||
|
|
||||||
|
type CreatePreviewEnvironmentParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
PreviewServiceID uuid.NullUUID `json:"preview_service_id"`
|
||||||
|
BranchName string `json:"branch_name"`
|
||||||
|
PrNumber sql.NullInt32 `json:"pr_number"`
|
||||||
|
Environment string `json:"environment"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
Url sql.NullString `json:"url"`
|
||||||
|
ExpiresAt time.Time `json:"expires_at"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CreatePreviewEnvironment(ctx context.Context, arg CreatePreviewEnvironmentParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, createPreviewEnvironment,
|
||||||
|
arg.ID,
|
||||||
|
arg.ProjectID,
|
||||||
|
arg.ServiceID,
|
||||||
|
arg.PreviewServiceID,
|
||||||
|
arg.BranchName,
|
||||||
|
arg.PrNumber,
|
||||||
|
arg.Environment,
|
||||||
|
arg.Status,
|
||||||
|
arg.Url,
|
||||||
|
arg.ExpiresAt,
|
||||||
|
arg.CreatedAt,
|
||||||
|
arg.UpdatedAt,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const deleteAllServiceDomains = `-- name: DeleteAllServiceDomains :exec
|
||||||
|
DELETE FROM service_domains WHERE service_id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) DeleteAllServiceDomains(ctx context.Context, serviceID uuid.UUID) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, deleteAllServiceDomains, serviceID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const deletePreviewEnvironment = `-- name: DeletePreviewEnvironment :exec
|
||||||
|
DELETE FROM preview_environments WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) DeletePreviewEnvironment(ctx context.Context, id uuid.UUID) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, deletePreviewEnvironment, id)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getPreviewEnvironment = `-- name: GetPreviewEnvironment :one
|
||||||
|
SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.pr_number,
|
||||||
|
pe.environment, pe.status, pe.url, pe.expires_at, pe.created_at, pe.updated_at,
|
||||||
|
s.id AS svc_id, s.name AS service_name, s.type AS service_type
|
||||||
|
FROM preview_environments pe
|
||||||
|
LEFT JOIN services s ON pe.service_id = s.id
|
||||||
|
JOIN projects p ON pe.project_id = p.id
|
||||||
|
WHERE pe.id = $1 AND (p.is_approved OR p.owner_id = $2 OR $3::bool
|
||||||
|
OR EXISTS (SELECT 1 FROM project_members pm WHERE pm.project_id = p.id AND pm.user_id = $2))
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetPreviewEnvironmentParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
Column3 bool `json:"column_3"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetPreviewEnvironmentRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
PreviewServiceID uuid.NullUUID `json:"preview_service_id"`
|
||||||
|
BranchName string `json:"branch_name"`
|
||||||
|
PrNumber sql.NullInt32 `json:"pr_number"`
|
||||||
|
Environment string `json:"environment"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
Url sql.NullString `json:"url"`
|
||||||
|
ExpiresAt time.Time `json:"expires_at"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
SvcID uuid.NullUUID `json:"svc_id"`
|
||||||
|
ServiceName sql.NullString `json:"service_name"`
|
||||||
|
ServiceType sql.NullString `json:"service_type"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetPreviewEnvironment(ctx context.Context, arg GetPreviewEnvironmentParams) (GetPreviewEnvironmentRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getPreviewEnvironment, arg.ID, arg.OwnerID, arg.Column3)
|
||||||
|
var i GetPreviewEnvironmentRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.PreviewServiceID,
|
||||||
|
&i.BranchName,
|
||||||
|
&i.PrNumber,
|
||||||
|
&i.Environment,
|
||||||
|
&i.Status,
|
||||||
|
&i.Url,
|
||||||
|
&i.ExpiresAt,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
&i.SvcID,
|
||||||
|
&i.ServiceName,
|
||||||
|
&i.ServiceType,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getPreviewEnvironmentForPromote = `-- name: GetPreviewEnvironmentForPromote :one
|
||||||
|
SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.environment, pe.status
|
||||||
|
FROM preview_environments pe
|
||||||
|
JOIN projects p ON pe.project_id = p.id
|
||||||
|
WHERE pe.id = $1 AND (p.owner_id = $2 OR $3::bool)
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetPreviewEnvironmentForPromoteParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
Column3 bool `json:"column_3"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetPreviewEnvironmentForPromoteRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
PreviewServiceID uuid.NullUUID `json:"preview_service_id"`
|
||||||
|
BranchName string `json:"branch_name"`
|
||||||
|
Environment string `json:"environment"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetPreviewEnvironmentForPromote(ctx context.Context, arg GetPreviewEnvironmentForPromoteParams) (GetPreviewEnvironmentForPromoteRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getPreviewEnvironmentForPromote, arg.ID, arg.OwnerID, arg.Column3)
|
||||||
|
var i GetPreviewEnvironmentForPromoteRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.PreviewServiceID,
|
||||||
|
&i.BranchName,
|
||||||
|
&i.Environment,
|
||||||
|
&i.Status,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getPreviewEnvironmentForWrite = `-- name: GetPreviewEnvironmentForWrite :one
|
||||||
|
SELECT pe.id, pe.project_id, pe.service_id, pe.branch_name, pe.pr_number,
|
||||||
|
pe.environment, pe.status, pe.url, pe.expires_at, pe.created_at, pe.updated_at
|
||||||
|
FROM preview_environments pe
|
||||||
|
JOIN projects p ON pe.project_id = p.id
|
||||||
|
WHERE pe.id = $1 AND (p.owner_id = $2 OR $3::bool)
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetPreviewEnvironmentForWriteParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
Column3 bool `json:"column_3"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetPreviewEnvironmentForWriteRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
BranchName string `json:"branch_name"`
|
||||||
|
PrNumber sql.NullInt32 `json:"pr_number"`
|
||||||
|
Environment string `json:"environment"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
Url sql.NullString `json:"url"`
|
||||||
|
ExpiresAt time.Time `json:"expires_at"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetPreviewEnvironmentForWrite(ctx context.Context, arg GetPreviewEnvironmentForWriteParams) (GetPreviewEnvironmentForWriteRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getPreviewEnvironmentForWrite, arg.ID, arg.OwnerID, arg.Column3)
|
||||||
|
var i GetPreviewEnvironmentForWriteRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.BranchName,
|
||||||
|
&i.PrNumber,
|
||||||
|
&i.Environment,
|
||||||
|
&i.Status,
|
||||||
|
&i.Url,
|
||||||
|
&i.ExpiresAt,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getPreviewEnvironmentOwner = `-- name: GetPreviewEnvironmentOwner :one
|
||||||
|
SELECT p.owner_id
|
||||||
|
FROM preview_environments pe
|
||||||
|
JOIN projects p ON pe.project_id = p.id
|
||||||
|
WHERE pe.id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetPreviewEnvironmentOwner(ctx context.Context, id uuid.UUID) (uuid.UUID, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getPreviewEnvironmentOwner, id)
|
||||||
|
var owner_id uuid.UUID
|
||||||
|
err := row.Scan(&owner_id)
|
||||||
|
return owner_id, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getPreviewServiceID = `-- name: GetPreviewServiceID :one
|
||||||
|
SELECT preview_service_id FROM preview_environments WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetPreviewServiceID(ctx context.Context, id uuid.UUID) (uuid.NullUUID, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getPreviewServiceID, id)
|
||||||
|
var preview_service_id uuid.NullUUID
|
||||||
|
err := row.Scan(&preview_service_id)
|
||||||
|
return preview_service_id, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getProjectBrief = `-- name: GetProjectBrief :one
|
||||||
|
SELECT id, name, owner_id FROM projects WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetProjectBriefRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetProjectBrief(ctx context.Context, id uuid.UUID) (GetProjectBriefRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getProjectBrief, id)
|
||||||
|
var i GetProjectBriefRow
|
||||||
|
err := row.Scan(&i.ID, &i.Name, &i.OwnerID)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServiceForDeploy = `-- name: GetServiceForDeploy :one
|
||||||
|
SELECT s.id, s.project_id, s.name, s.type, s.status, s.image, s.command,
|
||||||
|
s.environment, s.git_repo, s.git_branch, s.build_path, s.cpu, s.memory,
|
||||||
|
COALESCE(s.replicas, 1) AS replicas, COALESCE(s.port, 0) AS port,
|
||||||
|
COALESCE(s.domain, '') AS domain, COALESCE(s.healthcheck_path, '') AS healthcheck_path,
|
||||||
|
COALESCE(s.restart_policy, 'unless-stopped') AS restart_policy,
|
||||||
|
COALESCE(s.builder, 'auto') AS builder, COALESCE(s.cpu_reserve, '') AS cpu_reserve,
|
||||||
|
COALESCE(s.memory_reserve, '') AS memory_reserve, COALESCE(s.static_build_cmd, '') AS static_build_cmd,
|
||||||
|
COALESCE(s.static_dir, '') AS static_dir,
|
||||||
|
s.created_at, s.updated_at
|
||||||
|
FROM services s
|
||||||
|
WHERE s.id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetServiceForDeployRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Type sql.NullString `json:"type"`
|
||||||
|
Status sql.NullString `json:"status"`
|
||||||
|
Image sql.NullString `json:"image"`
|
||||||
|
Command sql.NullString `json:"command"`
|
||||||
|
Environment sql.NullString `json:"environment"`
|
||||||
|
GitRepo sql.NullString `json:"git_repo"`
|
||||||
|
GitBranch sql.NullString `json:"git_branch"`
|
||||||
|
BuildPath sql.NullString `json:"build_path"`
|
||||||
|
Cpu sql.NullString `json:"cpu"`
|
||||||
|
Memory sql.NullString `json:"memory"`
|
||||||
|
Replicas int32 `json:"replicas"`
|
||||||
|
Port int32 `json:"port"`
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
HealthcheckPath string `json:"healthcheck_path"`
|
||||||
|
RestartPolicy string `json:"restart_policy"`
|
||||||
|
Builder string `json:"builder"`
|
||||||
|
CpuReserve string `json:"cpu_reserve"`
|
||||||
|
MemoryReserve string `json:"memory_reserve"`
|
||||||
|
StaticBuildCmd string `json:"static_build_cmd"`
|
||||||
|
StaticDir string `json:"static_dir"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetServiceForDeploy(ctx context.Context, id uuid.UUID) (GetServiceForDeployRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServiceForDeploy, id)
|
||||||
|
var i GetServiceForDeployRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.Name,
|
||||||
|
&i.Type,
|
||||||
|
&i.Status,
|
||||||
|
&i.Image,
|
||||||
|
&i.Command,
|
||||||
|
&i.Environment,
|
||||||
|
&i.GitRepo,
|
||||||
|
&i.GitBranch,
|
||||||
|
&i.BuildPath,
|
||||||
|
&i.Cpu,
|
||||||
|
&i.Memory,
|
||||||
|
&i.Replicas,
|
||||||
|
&i.Port,
|
||||||
|
&i.Domain,
|
||||||
|
&i.HealthcheckPath,
|
||||||
|
&i.RestartPolicy,
|
||||||
|
&i.Builder,
|
||||||
|
&i.CpuReserve,
|
||||||
|
&i.MemoryReserve,
|
||||||
|
&i.StaticBuildCmd,
|
||||||
|
&i.StaticDir,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServiceTypeBrief = `-- name: GetServiceTypeBrief :one
|
||||||
|
SELECT id, name, COALESCE(type, service_type, '') AS service_type FROM services WHERE id = $1 AND project_id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetServiceTypeBriefParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type GetServiceTypeBriefRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
ServiceType string `json:"service_type"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetServiceTypeBrief(ctx context.Context, arg GetServiceTypeBriefParams) (GetServiceTypeBriefRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServiceTypeBrief, arg.ID, arg.ProjectID)
|
||||||
|
var i GetServiceTypeBriefRow
|
||||||
|
err := row.Scan(&i.ID, &i.Name, &i.ServiceType)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const insertPendingDeployment = `-- name: InsertPendingDeployment :exec
|
||||||
|
INSERT INTO deployments (id, service_id, version, commit_hash, status, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, 'pending', NOW(), NOW())
|
||||||
|
`
|
||||||
|
|
||||||
|
type InsertPendingDeploymentParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
Version string `json:"version"`
|
||||||
|
CommitHash sql.NullString `json:"commit_hash"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) InsertPendingDeployment(ctx context.Context, arg InsertPendingDeploymentParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, insertPendingDeployment,
|
||||||
|
arg.ID,
|
||||||
|
arg.ServiceID,
|
||||||
|
arg.Version,
|
||||||
|
arg.CommitHash,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const listExpiredPreviewsForUser = `-- name: ListExpiredPreviewsForUser :many
|
||||||
|
SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.environment
|
||||||
|
FROM preview_environments pe
|
||||||
|
JOIN projects p ON pe.project_id = p.id
|
||||||
|
WHERE (p.owner_id = $1 OR $2::bool) AND pe.expires_at < NOW() AND pe.status != 'expired'
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListExpiredPreviewsForUserParams struct {
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
Column2 bool `json:"column_2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ListExpiredPreviewsForUserRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
PreviewServiceID uuid.NullUUID `json:"preview_service_id"`
|
||||||
|
BranchName string `json:"branch_name"`
|
||||||
|
Environment string `json:"environment"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListExpiredPreviewsForUser(ctx context.Context, arg ListExpiredPreviewsForUserParams) ([]ListExpiredPreviewsForUserRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listExpiredPreviewsForUser, arg.OwnerID, arg.Column2)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListExpiredPreviewsForUserRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListExpiredPreviewsForUserRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.PreviewServiceID,
|
||||||
|
&i.BranchName,
|
||||||
|
&i.Environment,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listPreviewEnvironmentsForProject = `-- name: ListPreviewEnvironmentsForProject :many
|
||||||
|
SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.pr_number,
|
||||||
|
pe.environment, pe.status, pe.url, pe.expires_at, pe.created_at, pe.updated_at,
|
||||||
|
s.id AS svc_id, s.name AS service_name, s.type AS service_type
|
||||||
|
FROM preview_environments pe
|
||||||
|
LEFT JOIN services s ON pe.service_id = s.id
|
||||||
|
WHERE pe.project_id = $1
|
||||||
|
ORDER BY pe.created_at DESC
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListPreviewEnvironmentsForProjectRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
PreviewServiceID uuid.NullUUID `json:"preview_service_id"`
|
||||||
|
BranchName string `json:"branch_name"`
|
||||||
|
PrNumber sql.NullInt32 `json:"pr_number"`
|
||||||
|
Environment string `json:"environment"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
Url sql.NullString `json:"url"`
|
||||||
|
ExpiresAt time.Time `json:"expires_at"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
SvcID uuid.NullUUID `json:"svc_id"`
|
||||||
|
ServiceName sql.NullString `json:"service_name"`
|
||||||
|
ServiceType sql.NullString `json:"service_type"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListPreviewEnvironmentsForProject(ctx context.Context, projectID uuid.UUID) ([]ListPreviewEnvironmentsForProjectRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listPreviewEnvironmentsForProject, projectID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListPreviewEnvironmentsForProjectRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListPreviewEnvironmentsForProjectRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.PreviewServiceID,
|
||||||
|
&i.BranchName,
|
||||||
|
&i.PrNumber,
|
||||||
|
&i.Environment,
|
||||||
|
&i.Status,
|
||||||
|
&i.Url,
|
||||||
|
&i.ExpiresAt,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
&i.SvcID,
|
||||||
|
&i.ServiceName,
|
||||||
|
&i.ServiceType,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listSweepablePreviews = `-- name: ListSweepablePreviews :many
|
||||||
|
SELECT id, preview_service_id FROM preview_environments
|
||||||
|
WHERE expires_at < NOW() AND status NOT IN ('expired', 'stopped')
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListSweepablePreviewsRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
PreviewServiceID uuid.NullUUID `json:"preview_service_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListSweepablePreviews(ctx context.Context) ([]ListSweepablePreviewsRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listSweepablePreviews)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListSweepablePreviewsRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListSweepablePreviewsRow
|
||||||
|
if err := rows.Scan(&i.ID, &i.PreviewServiceID); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const markPreviewEnvironmentStatus = `-- name: MarkPreviewEnvironmentStatus :exec
|
||||||
|
UPDATE preview_environments SET status = $1, updated_at = NOW() WHERE id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type MarkPreviewEnvironmentStatusParams struct {
|
||||||
|
Status string `json:"status"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) MarkPreviewEnvironmentStatus(ctx context.Context, arg MarkPreviewEnvironmentStatusParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, markPreviewEnvironmentStatus, arg.Status, arg.ID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const syncPreviewStatuses = `-- name: SyncPreviewStatuses :exec
|
||||||
|
UPDATE preview_environments pe
|
||||||
|
SET status = derived.st, updated_at = NOW()
|
||||||
|
FROM (
|
||||||
|
SELECT pe2.id,
|
||||||
|
CASE
|
||||||
|
WHEN s.status IN ('running', 'deployed') THEN 'running'
|
||||||
|
WHEN s.status IN ('failed', 'error') THEN 'failed'
|
||||||
|
WHEN s.status IN ('building', 'deploying', 'queued', 'pending', 'cloning') THEN 'building'
|
||||||
|
ELSE pe2.status
|
||||||
|
END AS st
|
||||||
|
FROM preview_environments pe2
|
||||||
|
JOIN services s ON pe2.preview_service_id = s.id
|
||||||
|
) derived
|
||||||
|
WHERE pe.id = derived.id
|
||||||
|
AND pe.status IN ('building', 'running', 'failed')
|
||||||
|
AND pe.status <> derived.st
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) SyncPreviewStatuses(ctx context.Context) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, syncPreviewStatuses)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const updatePreviewEnvironment = `-- name: UpdatePreviewEnvironment :exec
|
||||||
|
UPDATE preview_environments
|
||||||
|
SET status = $1, url = $2, expires_at = $3, updated_at = $4
|
||||||
|
WHERE id = $5
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpdatePreviewEnvironmentParams struct {
|
||||||
|
Status string `json:"status"`
|
||||||
|
Url sql.NullString `json:"url"`
|
||||||
|
ExpiresAt time.Time `json:"expires_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpdatePreviewEnvironment(ctx context.Context, arg UpdatePreviewEnvironmentParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, updatePreviewEnvironment,
|
||||||
|
arg.Status,
|
||||||
|
arg.Url,
|
||||||
|
arg.ExpiresAt,
|
||||||
|
arg.UpdatedAt,
|
||||||
|
arg.ID,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const updateServicePreviewBranch = `-- name: UpdateServicePreviewBranch :exec
|
||||||
|
UPDATE services SET git_branch = $1, domain = '', updated_at = NOW() WHERE id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpdateServicePreviewBranchParams struct {
|
||||||
|
GitBranch sql.NullString `json:"git_branch"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpdateServicePreviewBranch(ctx context.Context, arg UpdateServicePreviewBranchParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, updateServicePreviewBranch, arg.GitBranch, arg.ID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
@@ -13,14 +13,33 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Querier interface {
|
type Querier interface {
|
||||||
|
// Portable bucket bucketing (no TimescaleDB): floor epoch seconds into
|
||||||
|
// bucket-sized buckets, then back to timestamptz.
|
||||||
|
AggregateNodeMetrics(ctx context.Context, arg AggregateNodeMetricsParams) ([]AggregateNodeMetricsRow, error)
|
||||||
|
CheckSecurityProjectAccess(ctx context.Context, arg CheckSecurityProjectAccessParams) (bool, error)
|
||||||
|
ClearServiceDomainDefault(ctx context.Context, serviceID uuid.UUID) error
|
||||||
ClearServiceNodePins(ctx context.Context, nodeID sql.NullString) (int64, error)
|
ClearServiceNodePins(ctx context.Context, nodeID sql.NullString) (int64, error)
|
||||||
|
CloneServiceDomains(ctx context.Context, arg CloneServiceDomainsParams) error
|
||||||
|
CloneServiceRow(ctx context.Context, arg CloneServiceRowParams) error
|
||||||
|
CloneServiceVariables(ctx context.Context, arg CloneServiceVariablesParams) error
|
||||||
CompleteCommand(ctx context.Context, arg CompleteCommandParams) (AgentCommand, error)
|
CompleteCommand(ctx context.Context, arg CompleteCommandParams) (AgentCommand, error)
|
||||||
|
CompleteDeployment(ctx context.Context, arg CompleteDeploymentParams) error
|
||||||
|
ComplianceFrameworkExists(ctx context.Context, id uuid.UUID) (bool, error)
|
||||||
|
CountAccessibleDeployments(ctx context.Context, arg CountAccessibleDeploymentsParams) (int64, error)
|
||||||
|
CountActivePreviewsForBranch(ctx context.Context, arg CountActivePreviewsForBranchParams) (int64, error)
|
||||||
CountDatabaseServicesByUserAndName(ctx context.Context, arg CountDatabaseServicesByUserAndNameParams) (int64, error)
|
CountDatabaseServicesByUserAndName(ctx context.Context, arg CountDatabaseServicesByUserAndNameParams) (int64, error)
|
||||||
CountDatabasesUsingBackupTarget(ctx context.Context, targetID sql.NullString) (int64, error)
|
CountDatabasesUsingBackupTarget(ctx context.Context, targetID sql.NullString) (int64, error)
|
||||||
|
CountDeploymentsForService(ctx context.Context, serviceID uuid.UUID) (int64, error)
|
||||||
CountEnvironmentServices(ctx context.Context, environmentID uuid.UUID) (int64, error)
|
CountEnvironmentServices(ctx context.Context, environmentID uuid.UUID) (int64, error)
|
||||||
|
CountGitRepositories(ctx context.Context, arg CountGitRepositoriesParams) (int64, error)
|
||||||
CountProjectsByUser(ctx context.Context, arg CountProjectsByUserParams) (int64, error)
|
CountProjectsByUser(ctx context.Context, arg CountProjectsByUserParams) (int64, error)
|
||||||
|
CountServiceDomains(ctx context.Context, serviceID uuid.UUID) (int64, error)
|
||||||
|
CountServicesByName(ctx context.Context, arg CountServicesByNameParams) (int64, error)
|
||||||
CountServicesByProjectAndName(ctx context.Context, arg CountServicesByProjectAndNameParams) (int64, error)
|
CountServicesByProjectAndName(ctx context.Context, arg CountServicesByProjectAndNameParams) (int64, error)
|
||||||
CountUnreadNotificationsByUser(ctx context.Context, userID uuid.UUID) (int64, error)
|
CountUnreadNotificationsByUser(ctx context.Context, userID uuid.UUID) (int64, error)
|
||||||
|
CountUserRepositories(ctx context.Context, userID uuid.UUID) (int64, error)
|
||||||
|
CountUsers(ctx context.Context) (int64, error)
|
||||||
|
CountUsersByEmail(ctx context.Context, email string) (int64, error)
|
||||||
CreateAgent(ctx context.Context, arg CreateAgentParams) (NodeAgent, error)
|
CreateAgent(ctx context.Context, arg CreateAgentParams) (NodeAgent, error)
|
||||||
CreateAgentAuthToken(ctx context.Context, arg CreateAgentAuthTokenParams) (AgentAuthToken, error)
|
CreateAgentAuthToken(ctx context.Context, arg CreateAgentAuthTokenParams) (AgentAuthToken, error)
|
||||||
CreateBackupTarget(ctx context.Context, arg CreateBackupTargetParams) error
|
CreateBackupTarget(ctx context.Context, arg CreateBackupTargetParams) error
|
||||||
@@ -32,7 +51,10 @@ type Querier interface {
|
|||||||
CreateExternalDatabaseService(ctx context.Context, arg CreateExternalDatabaseServiceParams) error
|
CreateExternalDatabaseService(ctx context.Context, arg CreateExternalDatabaseServiceParams) error
|
||||||
CreateNotification(ctx context.Context, arg CreateNotificationParams) error
|
CreateNotification(ctx context.Context, arg CreateNotificationParams) error
|
||||||
CreateOutboundWebhook(ctx context.Context, arg CreateOutboundWebhookParams) error
|
CreateOutboundWebhook(ctx context.Context, arg CreateOutboundWebhookParams) error
|
||||||
|
CreatePreviewDomain(ctx context.Context, arg CreatePreviewDomainParams) error
|
||||||
|
CreatePreviewEnvironment(ctx context.Context, arg CreatePreviewEnvironmentParams) error
|
||||||
CreateProject(ctx context.Context, arg CreateProjectParams) (Project, error)
|
CreateProject(ctx context.Context, arg CreateProjectParams) (Project, error)
|
||||||
|
CreateServiceDomain(ctx context.Context, arg CreateServiceDomainParams) (CreateServiceDomainRow, error)
|
||||||
CreateServiceFromTemplate(ctx context.Context, arg CreateServiceFromTemplateParams) error
|
CreateServiceFromTemplate(ctx context.Context, arg CreateServiceFromTemplateParams) error
|
||||||
CreateUserInvite(ctx context.Context, arg CreateUserInviteParams) (UserInvite, error)
|
CreateUserInvite(ctx context.Context, arg CreateUserInviteParams) (UserInvite, error)
|
||||||
CreateUserTemplate(ctx context.Context, arg CreateUserTemplateParams) error
|
CreateUserTemplate(ctx context.Context, arg CreateUserTemplateParams) error
|
||||||
@@ -40,17 +62,25 @@ type Querier interface {
|
|||||||
CreateWebhookDelivery(ctx context.Context, arg CreateWebhookDeliveryParams) error
|
CreateWebhookDelivery(ctx context.Context, arg CreateWebhookDeliveryParams) error
|
||||||
DatabaseServiceExistsByIDAndUser(ctx context.Context, arg DatabaseServiceExistsByIDAndUserParams) (bool, error)
|
DatabaseServiceExistsByIDAndUser(ctx context.Context, arg DatabaseServiceExistsByIDAndUserParams) (bool, error)
|
||||||
DeleteAgent(ctx context.Context, id string) error
|
DeleteAgent(ctx context.Context, id string) error
|
||||||
|
DeleteAllServiceDomains(ctx context.Context, serviceID uuid.UUID) error
|
||||||
DeleteBackupTargetByIDAndUser(ctx context.Context, arg DeleteBackupTargetByIDAndUserParams) error
|
DeleteBackupTargetByIDAndUser(ctx context.Context, arg DeleteBackupTargetByIDAndUserParams) error
|
||||||
DeleteBanner(ctx context.Context, id uuid.UUID) error
|
DeleteBanner(ctx context.Context, id uuid.UUID) error
|
||||||
|
DeleteCronJob(ctx context.Context, id uuid.UUID) error
|
||||||
DeleteDatabaseServiceByID(ctx context.Context, id string) error
|
DeleteDatabaseServiceByID(ctx context.Context, id string) error
|
||||||
DeleteDatabaseServiceByIDAndUser(ctx context.Context, arg DeleteDatabaseServiceByIDAndUserParams) error
|
DeleteDatabaseServiceByIDAndUser(ctx context.Context, arg DeleteDatabaseServiceByIDAndUserParams) error
|
||||||
|
DeleteGitProvider(ctx context.Context, arg DeleteGitProviderParams) (int64, error)
|
||||||
DeleteNotificationChannel(ctx context.Context, arg DeleteNotificationChannelParams) (int64, error)
|
DeleteNotificationChannel(ctx context.Context, arg DeleteNotificationChannelParams) (int64, error)
|
||||||
DeleteOutboundWebhookByIDAndUser(ctx context.Context, arg DeleteOutboundWebhookByIDAndUserParams) error
|
DeleteOutboundWebhookByIDAndUser(ctx context.Context, arg DeleteOutboundWebhookByIDAndUserParams) error
|
||||||
|
DeletePreviewEnvironment(ctx context.Context, id uuid.UUID) error
|
||||||
DeleteProjectByID(ctx context.Context, projectID uuid.UUID) (int64, error)
|
DeleteProjectByID(ctx context.Context, projectID uuid.UUID) (int64, error)
|
||||||
DeleteProjectEnvironment(ctx context.Context, arg DeleteProjectEnvironmentParams) error
|
DeleteProjectEnvironment(ctx context.Context, arg DeleteProjectEnvironmentParams) error
|
||||||
|
DeleteServiceByID(ctx context.Context, id uuid.UUID) error
|
||||||
DeleteServiceContainersOnAgent(ctx context.Context, arg DeleteServiceContainersOnAgentParams) error
|
DeleteServiceContainersOnAgent(ctx context.Context, arg DeleteServiceContainersOnAgentParams) error
|
||||||
|
DeleteServiceDomain(ctx context.Context, arg DeleteServiceDomainParams) (int64, error)
|
||||||
DeleteUserInvite(ctx context.Context, id uuid.UUID) error
|
DeleteUserInvite(ctx context.Context, id uuid.UUID) error
|
||||||
DeleteUserTemplate(ctx context.Context, arg DeleteUserTemplateParams) (int64, error)
|
DeleteUserTemplate(ctx context.Context, arg DeleteUserTemplateParams) (int64, error)
|
||||||
|
FailDeployment(ctx context.Context, arg FailDeploymentParams) error
|
||||||
|
FinishCronExecution(ctx context.Context, arg FinishCronExecutionParams) error
|
||||||
GetActiveAgentAuthTokenByHash(ctx context.Context, tokenHash string) (AgentAuthToken, error)
|
GetActiveAgentAuthTokenByHash(ctx context.Context, tokenHash string) (AgentAuthToken, error)
|
||||||
GetAgent(ctx context.Context, id string) (NodeAgent, error)
|
GetAgent(ctx context.Context, id string) (NodeAgent, error)
|
||||||
GetAgentByHostAndIP(ctx context.Context, arg GetAgentByHostAndIPParams) (NodeAgent, error)
|
GetAgentByHostAndIP(ctx context.Context, arg GetAgentByHostAndIPParams) (NodeAgent, error)
|
||||||
@@ -58,26 +88,82 @@ type Querier interface {
|
|||||||
GetBackupTargetByIDAndUser(ctx context.Context, arg GetBackupTargetByIDAndUserParams) (BackupTarget, error)
|
GetBackupTargetByIDAndUser(ctx context.Context, arg GetBackupTargetByIDAndUserParams) (BackupTarget, error)
|
||||||
GetBannerByID(ctx context.Context, id uuid.UUID) (Banner, error)
|
GetBannerByID(ctx context.Context, id uuid.UUID) (Banner, error)
|
||||||
GetCommandForAgent(ctx context.Context, arg GetCommandForAgentParams) (AgentCommand, error)
|
GetCommandForAgent(ctx context.Context, arg GetCommandForAgentParams) (AgentCommand, error)
|
||||||
|
GetComplianceReportProjectID(ctx context.Context, id uuid.UUID) (uuid.UUID, error)
|
||||||
GetContainer(ctx context.Context, id string) (ContainerInstance, error)
|
GetContainer(ctx context.Context, id string) (ContainerInstance, error)
|
||||||
GetContainerForAgent(ctx context.Context, arg GetContainerForAgentParams) (ContainerInstance, error)
|
GetContainerForAgent(ctx context.Context, arg GetContainerForAgentParams) (ContainerInstance, error)
|
||||||
|
GetCronJob(ctx context.Context, id uuid.UUID) (CronJob, error)
|
||||||
|
GetCronJobForTrigger(ctx context.Context, id uuid.UUID) (GetCronJobForTriggerRow, error)
|
||||||
|
GetCronJobOwner(ctx context.Context, id uuid.UUID) (uuid.UUID, error)
|
||||||
|
GetCronJobProjectID(ctx context.Context, id uuid.UUID) (uuid.UUID, error)
|
||||||
|
GetCronOwnerViaService(ctx context.Context, id uuid.UUID) (uuid.UUID, error)
|
||||||
GetDatabaseBackupByIDAndDatabaseAndUser(ctx context.Context, arg GetDatabaseBackupByIDAndDatabaseAndUserParams) (DatabaseBackup, error)
|
GetDatabaseBackupByIDAndDatabaseAndUser(ctx context.Context, arg GetDatabaseBackupByIDAndDatabaseAndUserParams) (DatabaseBackup, error)
|
||||||
GetDatabaseServiceByID(ctx context.Context, id string) (DatabaseService, error)
|
GetDatabaseServiceByID(ctx context.Context, id string) (DatabaseService, error)
|
||||||
GetDatabaseServiceByIDAndUser(ctx context.Context, arg GetDatabaseServiceByIDAndUserParams) (DatabaseService, error)
|
GetDatabaseServiceByIDAndUser(ctx context.Context, arg GetDatabaseServiceByIDAndUserParams) (DatabaseService, error)
|
||||||
GetDatabaseServiceOwnerID(ctx context.Context, id string) (string, error)
|
GetDatabaseServiceOwnerID(ctx context.Context, id string) (string, error)
|
||||||
|
GetDeploymentAccess(ctx context.Context, id uuid.UUID) (GetDeploymentAccessRow, error)
|
||||||
|
GetDeploymentForRollback(ctx context.Context, id uuid.UUID) (GetDeploymentForRollbackRow, error)
|
||||||
|
GetDeploymentWithProject(ctx context.Context, id uuid.UUID) (GetDeploymentWithProjectRow, error)
|
||||||
|
GetGitProviderForFetch(ctx context.Context, arg GetGitProviderForFetchParams) (GetGitProviderForFetchRow, error)
|
||||||
|
GetGitProviderForRepos(ctx context.Context, arg GetGitProviderForReposParams) (GetGitProviderForReposRow, error)
|
||||||
|
GetGitProviderName(ctx context.Context, arg GetGitProviderNameParams) (string, error)
|
||||||
|
GetGitProviderNameByID(ctx context.Context, id uuid.UUID) (string, error)
|
||||||
|
GetGitProviderTimestamps(ctx context.Context, id uuid.UUID) (GetGitProviderTimestampsRow, error)
|
||||||
|
GetGitRepoForPush(ctx context.Context, id uuid.UUID) (GetGitRepoForPushRow, error)
|
||||||
|
GetGitRepositoryByFullName(ctx context.Context, arg GetGitRepositoryByFullNameParams) (GetGitRepositoryByFullNameRow, error)
|
||||||
|
GetGitRepositoryProviderID(ctx context.Context, arg GetGitRepositoryProviderIDParams) (uuid.UUID, error)
|
||||||
|
GetGitWebhookForPush(ctx context.Context, id uuid.UUID) (GetGitWebhookForPushRow, error)
|
||||||
GetLastAgentCommandByType(ctx context.Context, arg GetLastAgentCommandByTypeParams) (AgentCommand, error)
|
GetLastAgentCommandByType(ctx context.Context, arg GetLastAgentCommandByTypeParams) (AgentCommand, error)
|
||||||
|
GetLastDeployedImage(ctx context.Context, serviceID uuid.UUID) (string, error)
|
||||||
|
GetLatestComplianceReport(ctx context.Context, projectID uuid.UUID) (GetLatestComplianceReportRow, error)
|
||||||
|
GetLatestSecurityScan(ctx context.Context, projectID uuid.UUID) (GetLatestSecurityScanRow, error)
|
||||||
GetOutboundWebhookByID(ctx context.Context, id uuid.UUID) (OutboundWebhook, error)
|
GetOutboundWebhookByID(ctx context.Context, id uuid.UUID) (OutboundWebhook, error)
|
||||||
GetOutboundWebhookByIDAndUser(ctx context.Context, arg GetOutboundWebhookByIDAndUserParams) (OutboundWebhook, error)
|
GetOutboundWebhookByIDAndUser(ctx context.Context, arg GetOutboundWebhookByIDAndUserParams) (OutboundWebhook, error)
|
||||||
|
GetPreviewEnvironment(ctx context.Context, arg GetPreviewEnvironmentParams) (GetPreviewEnvironmentRow, error)
|
||||||
|
GetPreviewEnvironmentForPromote(ctx context.Context, arg GetPreviewEnvironmentForPromoteParams) (GetPreviewEnvironmentForPromoteRow, error)
|
||||||
|
GetPreviewEnvironmentForWrite(ctx context.Context, arg GetPreviewEnvironmentForWriteParams) (GetPreviewEnvironmentForWriteRow, error)
|
||||||
|
GetPreviewEnvironmentOwner(ctx context.Context, id uuid.UUID) (uuid.UUID, error)
|
||||||
|
GetPreviewServiceID(ctx context.Context, id uuid.UUID) (uuid.NullUUID, error)
|
||||||
|
GetProjectBrief(ctx context.Context, id uuid.UUID) (GetProjectBriefRow, error)
|
||||||
GetProjectByIDForUser(ctx context.Context, arg GetProjectByIDForUserParams) (Project, error)
|
GetProjectByIDForUser(ctx context.Context, arg GetProjectByIDForUserParams) (Project, error)
|
||||||
GetProjectEnvironmentByName(ctx context.Context, arg GetProjectEnvironmentByNameParams) (Environment, error)
|
GetProjectEnvironmentByName(ctx context.Context, arg GetProjectEnvironmentByNameParams) (Environment, error)
|
||||||
|
GetProjectOwner(ctx context.Context, id uuid.UUID) (uuid.UUID, error)
|
||||||
GetProjectOwnerByID(ctx context.Context, projectID uuid.UUID) (uuid.UUID, error)
|
GetProjectOwnerByID(ctx context.Context, projectID uuid.UUID) (uuid.UUID, error)
|
||||||
GetProjectOwnerID(ctx context.Context, id uuid.UUID) (uuid.UUID, error)
|
GetProjectOwnerID(ctx context.Context, id uuid.UUID) (uuid.UUID, error)
|
||||||
|
GetProjectOwnerText(ctx context.Context, id uuid.UUID) (string, error)
|
||||||
GetProjectRoleForUser(ctx context.Context, arg GetProjectRoleForUserParams) (string, error)
|
GetProjectRoleForUser(ctx context.Context, arg GetProjectRoleForUserParams) (string, error)
|
||||||
|
GetSecurityScanProjectID(ctx context.Context, id uuid.UUID) (uuid.UUID, error)
|
||||||
|
GetServiceAccess(ctx context.Context, id uuid.UUID) (GetServiceAccessRow, error)
|
||||||
|
GetServiceDomainDefault(ctx context.Context, serviceID uuid.UUID) (string, error)
|
||||||
|
GetServiceDomainFirst(ctx context.Context, serviceID uuid.UUID) (string, error)
|
||||||
|
GetServiceDomainText(ctx context.Context, id uuid.UUID) (string, error)
|
||||||
|
GetServiceEnvironment(ctx context.Context, id uuid.UUID) (string, error)
|
||||||
|
GetServiceForDeploy(ctx context.Context, id uuid.UUID) (GetServiceForDeployRow, error)
|
||||||
|
GetServiceForDeployWithOwner(ctx context.Context, id uuid.UUID) (GetServiceForDeployWithOwnerRow, error)
|
||||||
|
GetServicePlacementBrief(ctx context.Context, id uuid.UUID) (GetServicePlacementBriefRow, error)
|
||||||
|
GetServicePort(ctx context.Context, id uuid.UUID) (int32, error)
|
||||||
|
GetServicePublishedPort(ctx context.Context, id uuid.UUID) (int32, error)
|
||||||
|
GetServiceRuntimeWithOwner(ctx context.Context, id uuid.UUID) (GetServiceRuntimeWithOwnerRow, error)
|
||||||
|
GetServiceStatusText(ctx context.Context, id uuid.UUID) (string, error)
|
||||||
GetServiceTemplateByID(ctx context.Context, id string) (ServiceTemplate, error)
|
GetServiceTemplateByID(ctx context.Context, id string) (ServiceTemplate, error)
|
||||||
|
GetServiceTypeBrief(ctx context.Context, arg GetServiceTypeBriefParams) (GetServiceTypeBriefRow, error)
|
||||||
|
GetServiceVariableValue(ctx context.Context, arg GetServiceVariableValueParams) (string, error)
|
||||||
|
GetUserByEmailForAuth(ctx context.Context, email string) (GetUserByEmailForAuthRow, error)
|
||||||
|
GetUserByID(ctx context.Context, id uuid.UUID) (GetUserByIDRow, error)
|
||||||
GetUserInviteByTokenHash(ctx context.Context, tokenHash string) (UserInvite, error)
|
GetUserInviteByTokenHash(ctx context.Context, tokenHash string) (UserInvite, error)
|
||||||
GetUserTokenByHash(ctx context.Context, tokenHash string) (UserToken, error)
|
GetUserTokenByHash(ctx context.Context, tokenHash string) (UserToken, error)
|
||||||
|
GetVulnerabilityMetrics(ctx context.Context, projectID uuid.UUID) (GetVulnerabilityMetricsRow, error)
|
||||||
|
GetVulnerabilityProjectID(ctx context.Context, id uuid.UUID) (uuid.UUID, error)
|
||||||
InsertAgentHeartbeat(ctx context.Context, arg InsertAgentHeartbeatParams) error
|
InsertAgentHeartbeat(ctx context.Context, arg InsertAgentHeartbeatParams) error
|
||||||
|
InsertCronExecution(ctx context.Context, arg InsertCronExecutionParams) error
|
||||||
|
InsertCronJob(ctx context.Context, arg InsertCronJobParams) error
|
||||||
|
InsertDeployment(ctx context.Context, arg InsertDeploymentParams) error
|
||||||
|
InsertGitProvider(ctx context.Context, arg InsertGitProviderParams) error
|
||||||
|
InsertGitRepository(ctx context.Context, arg InsertGitRepositoryParams) (InsertGitRepositoryRow, error)
|
||||||
InsertNotificationChannel(ctx context.Context, arg InsertNotificationChannelParams) (NotificationChannel, error)
|
InsertNotificationChannel(ctx context.Context, arg InsertNotificationChannelParams) (NotificationChannel, error)
|
||||||
|
InsertPendingDeployment(ctx context.Context, arg InsertPendingDeploymentParams) error
|
||||||
InsertProjectEnvironment(ctx context.Context, arg InsertProjectEnvironmentParams) error
|
InsertProjectEnvironment(ctx context.Context, arg InsertProjectEnvironmentParams) error
|
||||||
|
InsertUser(ctx context.Context, arg InsertUserParams) (InsertUserRow, error)
|
||||||
|
InsertUserAdmin(ctx context.Context, arg InsertUserAdminParams) (InsertUserAdminRow, error)
|
||||||
ListActiveBanners(ctx context.Context) ([]Banner, error)
|
ListActiveBanners(ctx context.Context) ([]Banner, error)
|
||||||
ListAgentAuthTokens(ctx context.Context) ([]AgentAuthToken, error)
|
ListAgentAuthTokens(ctx context.Context) ([]AgentAuthToken, error)
|
||||||
ListAgentHeartbeatsSince(ctx context.Context, arg ListAgentHeartbeatsSinceParams) ([]AgentHeartbeat, error)
|
ListAgentHeartbeatsSince(ctx context.Context, arg ListAgentHeartbeatsSinceParams) ([]AgentHeartbeat, error)
|
||||||
@@ -87,22 +173,36 @@ type Querier interface {
|
|||||||
ListBackupTargetsByUser(ctx context.Context, userID string) ([]BackupTarget, error)
|
ListBackupTargetsByUser(ctx context.Context, userID string) ([]BackupTarget, error)
|
||||||
ListBanners(ctx context.Context) ([]Banner, error)
|
ListBanners(ctx context.Context) ([]Banner, error)
|
||||||
ListCommandsForAgent(ctx context.Context, nodeAgentID string) ([]AgentCommand, error)
|
ListCommandsForAgent(ctx context.Context, nodeAgentID string) ([]AgentCommand, error)
|
||||||
|
ListComplianceFrameworks(ctx context.Context) ([]ListComplianceFrameworksRow, error)
|
||||||
|
ListConnectedRepositories(ctx context.Context, arg ListConnectedRepositoriesParams) ([]ListConnectedRepositoriesRow, error)
|
||||||
ListContainersForAgent(ctx context.Context, nodeAgentID string) ([]ContainerInstance, error)
|
ListContainersForAgent(ctx context.Context, nodeAgentID string) ([]ContainerInstance, error)
|
||||||
|
ListCronExecutions(ctx context.Context, cronJobID uuid.UUID) ([]CronExecution, error)
|
||||||
ListDatabaseBackupsByDatabaseAndUser(ctx context.Context, arg ListDatabaseBackupsByDatabaseAndUserParams) ([]DatabaseBackup, error)
|
ListDatabaseBackupsByDatabaseAndUser(ctx context.Context, arg ListDatabaseBackupsByDatabaseAndUserParams) ([]DatabaseBackup, error)
|
||||||
ListDatabaseServicesByProject(ctx context.Context, projectID uuid.NullUUID) ([]DatabaseService, error)
|
ListDatabaseServicesByProject(ctx context.Context, projectID uuid.NullUUID) ([]DatabaseService, error)
|
||||||
ListDatabaseServicesByUser(ctx context.Context, userID string) ([]DatabaseService, error)
|
ListDatabaseServicesByUser(ctx context.Context, userID string) ([]DatabaseService, error)
|
||||||
|
ListDeploymentsForService(ctx context.Context, arg ListDeploymentsForServiceParams) ([]ListDeploymentsForServiceRow, error)
|
||||||
|
ListDueCronJobs(ctx context.Context) ([]ListDueCronJobsRow, error)
|
||||||
ListDueDatabaseBackups(ctx context.Context) ([]ListDueDatabaseBackupsRow, error)
|
ListDueDatabaseBackups(ctx context.Context) ([]ListDueDatabaseBackupsRow, error)
|
||||||
ListEnabledOutboundWebhooks(ctx context.Context) ([]OutboundWebhook, error)
|
ListEnabledOutboundWebhooks(ctx context.Context) ([]OutboundWebhook, error)
|
||||||
|
ListExpiredPreviewsForUser(ctx context.Context, arg ListExpiredPreviewsForUserParams) ([]ListExpiredPreviewsForUserRow, error)
|
||||||
|
ListGitProviders(ctx context.Context, userID uuid.UUID) ([]ListGitProvidersRow, error)
|
||||||
|
ListGitRepositories(ctx context.Context, arg ListGitRepositoriesParams) ([]ListGitRepositoriesRow, error)
|
||||||
|
ListInstanceMetrics(ctx context.Context, arg ListInstanceMetricsParams) ([]ListInstanceMetricsRow, error)
|
||||||
|
ListNodeAgentsForService(ctx context.Context, serviceID string) ([]string, error)
|
||||||
|
ListNodeMetrics(ctx context.Context, arg ListNodeMetricsParams) ([]ListNodeMetricsRow, error)
|
||||||
ListNotificationChannelsByUser(ctx context.Context, userID uuid.UUID) ([]NotificationChannel, error)
|
ListNotificationChannelsByUser(ctx context.Context, userID uuid.UUID) ([]NotificationChannel, error)
|
||||||
ListNotificationsByUser(ctx context.Context, arg ListNotificationsByUserParams) ([]Notification, error)
|
ListNotificationsByUser(ctx context.Context, arg ListNotificationsByUserParams) ([]Notification, error)
|
||||||
ListOutboundWebhooksByUser(ctx context.Context, userID uuid.UUID) ([]OutboundWebhook, error)
|
ListOutboundWebhooksByUser(ctx context.Context, userID uuid.UUID) ([]OutboundWebhook, error)
|
||||||
ListPendingCommands(ctx context.Context, nodeAgentID string) ([]AgentCommand, error)
|
ListPendingCommands(ctx context.Context, nodeAgentID string) ([]AgentCommand, error)
|
||||||
ListPendingProjects(ctx context.Context) ([]Project, error)
|
ListPendingProjects(ctx context.Context) ([]Project, error)
|
||||||
|
ListPreviewEnvironmentsForProject(ctx context.Context, projectID uuid.UUID) ([]ListPreviewEnvironmentsForProjectRow, error)
|
||||||
ListProjectEnvironments(ctx context.Context, projectID uuid.UUID) ([]ListProjectEnvironmentsRow, error)
|
ListProjectEnvironments(ctx context.Context, projectID uuid.UUID) ([]ListProjectEnvironmentsRow, error)
|
||||||
|
ListProjectServiceRefs(ctx context.Context, projectID uuid.UUID) ([]ListProjectServiceRefsRow, error)
|
||||||
// Visibility model: a project is readable when it is approved for public
|
// Visibility model: a project is readable when it is approved for public
|
||||||
// display, when the caller owns or is a member of it, or when the caller is a
|
// display, when the caller owns or is a member of it, or when the caller is a
|
||||||
// platform admin. Anonymous callers pass user_id = uuid.Nil, is_admin = false.
|
// platform admin. Anonymous callers pass user_id = uuid.Nil, is_admin = false.
|
||||||
ListProjectsWithStatsByUser(ctx context.Context, arg ListProjectsWithStatsByUserParams) ([]ListProjectsWithStatsByUserRow, error)
|
ListProjectsWithStatsByUser(ctx context.Context, arg ListProjectsWithStatsByUserParams) ([]ListProjectsWithStatsByUserRow, error)
|
||||||
|
ListRecentAccessibleDeployments(ctx context.Context, arg ListRecentAccessibleDeploymentsParams) ([]ListRecentAccessibleDeploymentsRow, error)
|
||||||
// Base domains of every online, schedulable agent carrying the required
|
// Base domains of every online, schedulable agent carrying the required
|
||||||
// placement tags — auto-domain candidates for spread services.
|
// placement tags — auto-domain candidates for spread services.
|
||||||
ListSchedulableAgentDomains(ctx context.Context, dollar_1 json.RawMessage) ([]string, error)
|
ListSchedulableAgentDomains(ctx context.Context, dollar_1 json.RawMessage) ([]string, error)
|
||||||
@@ -112,14 +212,24 @@ type Querier interface {
|
|||||||
// Distinct agents holding inventory rows for a service.
|
// Distinct agents holding inventory rows for a service.
|
||||||
ListServiceAgents(ctx context.Context, serviceID string) ([]string, error)
|
ListServiceAgents(ctx context.Context, serviceID string) ([]string, error)
|
||||||
ListServiceContainers(ctx context.Context, serviceID string) ([]ListServiceContainersRow, error)
|
ListServiceContainers(ctx context.Context, serviceID string) ([]ListServiceContainersRow, error)
|
||||||
|
ListServiceDomains(ctx context.Context, serviceID uuid.UUID) ([]ServiceDomain, error)
|
||||||
|
ListServiceMetrics(ctx context.Context, arg ListServiceMetricsParams) ([]ListServiceMetricsRow, error)
|
||||||
ListServiceTemplatesByCategoryForUser(ctx context.Context, arg ListServiceTemplatesByCategoryForUserParams) ([]ServiceTemplate, error)
|
ListServiceTemplatesByCategoryForUser(ctx context.Context, arg ListServiceTemplatesByCategoryForUserParams) ([]ServiceTemplate, error)
|
||||||
ListServiceTemplatesForUser(ctx context.Context, ownerID uuid.NullUUID) ([]ServiceTemplate, error)
|
ListServiceTemplatesForUser(ctx context.Context, ownerID uuid.NullUUID) ([]ServiceTemplate, error)
|
||||||
|
ListServiceVariableValues(ctx context.Context, serviceID uuid.UUID) ([]ListServiceVariableValuesRow, error)
|
||||||
|
ListServiceVariablesWithSecret(ctx context.Context, serviceID uuid.UUID) ([]ListServiceVariablesWithSecretRow, error)
|
||||||
|
ListServicesForPush(ctx context.Context, arg ListServicesForPushParams) ([]ListServicesForPushRow, error)
|
||||||
|
ListSleepCandidates(ctx context.Context) ([]ListSleepCandidatesRow, error)
|
||||||
|
ListSweepablePreviews(ctx context.Context) ([]ListSweepablePreviewsRow, error)
|
||||||
ListUserInvites(ctx context.Context) ([]UserInvite, error)
|
ListUserInvites(ctx context.Context) ([]UserInvite, error)
|
||||||
ListUserTokens(ctx context.Context, userID uuid.UUID) ([]UserToken, error)
|
ListUserTokens(ctx context.Context, userID uuid.UUID) ([]UserToken, error)
|
||||||
|
ListVulnerabilities(ctx context.Context, projectID uuid.UUID) ([]ListVulnerabilitiesRow, error)
|
||||||
ListWebhookDeliveriesByWebhook(ctx context.Context, webhookID uuid.UUID) ([]WebhookDelivery, error)
|
ListWebhookDeliveriesByWebhook(ctx context.Context, webhookID uuid.UUID) ([]WebhookDelivery, error)
|
||||||
MarkAllNotificationsReadByUser(ctx context.Context, arg MarkAllNotificationsReadByUserParams) error
|
MarkAllNotificationsReadByUser(ctx context.Context, arg MarkAllNotificationsReadByUserParams) error
|
||||||
MarkNotificationReadByIDAndUser(ctx context.Context, arg MarkNotificationReadByIDAndUserParams) error
|
MarkNotificationReadByIDAndUser(ctx context.Context, arg MarkNotificationReadByIDAndUserParams) error
|
||||||
|
MarkPreviewEnvironmentStatus(ctx context.Context, arg MarkPreviewEnvironmentStatusParams) error
|
||||||
MarkUserInviteUsed(ctx context.Context, arg MarkUserInviteUsedParams) error
|
MarkUserInviteUsed(ctx context.Context, arg MarkUserInviteUsedParams) error
|
||||||
|
MoveServiceProject(ctx context.Context, arg MoveServiceProjectParams) (int64, error)
|
||||||
// Resource-aware: lowest memory utilisation first, then cpu, then container
|
// Resource-aware: lowest memory utilisation first, then cpu, then container
|
||||||
// count. Agents without telemetry sort last (NULL → worst score).
|
// count. Agents without telemetry sort last (NULL → worst score).
|
||||||
PickLeastLoadedAgent(ctx context.Context) (string, error)
|
PickLeastLoadedAgent(ctx context.Context) (string, error)
|
||||||
@@ -129,6 +239,8 @@ type Querier interface {
|
|||||||
RevokeAgentAuthToken(ctx context.Context, id uuid.UUID) (AgentAuthToken, error)
|
RevokeAgentAuthToken(ctx context.Context, id uuid.UUID) (AgentAuthToken, error)
|
||||||
RevokeUserToken(ctx context.Context, arg RevokeUserTokenParams) (int64, error)
|
RevokeUserToken(ctx context.Context, arg RevokeUserTokenParams) (int64, error)
|
||||||
ScrubCommandPayload(ctx context.Context, arg ScrubCommandPayloadParams) error
|
ScrubCommandPayload(ctx context.Context, arg ScrubCommandPayloadParams) error
|
||||||
|
ServiceExistsInProject(ctx context.Context, arg ServiceExistsInProjectParams) (bool, error)
|
||||||
|
ServiceWriteAccess(ctx context.Context, arg ServiceWriteAccessParams) (bool, error)
|
||||||
SetAgentDefaultDomain(ctx context.Context, arg SetAgentDefaultDomainParams) error
|
SetAgentDefaultDomain(ctx context.Context, arg SetAgentDefaultDomainParams) error
|
||||||
SetAgentSchedulable(ctx context.Context, arg SetAgentSchedulableParams) error
|
SetAgentSchedulable(ctx context.Context, arg SetAgentSchedulableParams) error
|
||||||
SetAgentTags(ctx context.Context, arg SetAgentTagsParams) error
|
SetAgentTags(ctx context.Context, arg SetAgentTagsParams) error
|
||||||
@@ -141,9 +253,19 @@ type Querier interface {
|
|||||||
SetDatabaseServiceStatusAndConnectionByID(ctx context.Context, arg SetDatabaseServiceStatusAndConnectionByIDParams) error
|
SetDatabaseServiceStatusAndConnectionByID(ctx context.Context, arg SetDatabaseServiceStatusAndConnectionByIDParams) error
|
||||||
SetDatabaseServiceStatusByID(ctx context.Context, arg SetDatabaseServiceStatusByIDParams) error
|
SetDatabaseServiceStatusByID(ctx context.Context, arg SetDatabaseServiceStatusByIDParams) error
|
||||||
SetDatabaseServiceStatusByIDAndUser(ctx context.Context, arg SetDatabaseServiceStatusByIDAndUserParams) error
|
SetDatabaseServiceStatusByIDAndUser(ctx context.Context, arg SetDatabaseServiceStatusByIDAndUserParams) error
|
||||||
|
SetDeploymentStatus(ctx context.Context, arg SetDeploymentStatusParams) error
|
||||||
SetProjectApproved(ctx context.Context, arg SetProjectApprovedParams) (int64, error)
|
SetProjectApproved(ctx context.Context, arg SetProjectApprovedParams) (int64, error)
|
||||||
|
SetServiceDomainDefault(ctx context.Context, arg SetServiceDomainDefaultParams) (int64, error)
|
||||||
|
SetServiceLegacyDomain(ctx context.Context, arg SetServiceLegacyDomainParams) error
|
||||||
|
SetServicePublishedPort(ctx context.Context, arg SetServicePublishedPortParams) error
|
||||||
|
SetServiceReplicas(ctx context.Context, arg SetServiceReplicasParams) error
|
||||||
|
SetServiceStatus(ctx context.Context, arg SetServiceStatusParams) error
|
||||||
|
SyncDeploymentProgress(ctx context.Context, arg SyncDeploymentProgressParams) error
|
||||||
|
SyncPreviewStatuses(ctx context.Context) error
|
||||||
TouchAgentAuthToken(ctx context.Context, tokenHash string) error
|
TouchAgentAuthToken(ctx context.Context, tokenHash string) error
|
||||||
|
TouchServiceDomainChecked(ctx context.Context, id uuid.UUID) error
|
||||||
TouchUserToken(ctx context.Context, id uuid.UUID) error
|
TouchUserToken(ctx context.Context, id uuid.UUID) error
|
||||||
|
TrimCronExecutions(ctx context.Context, arg TrimCronExecutionsParams) error
|
||||||
UpdateAgent(ctx context.Context, arg UpdateAgentParams) (NodeAgent, error)
|
UpdateAgent(ctx context.Context, arg UpdateAgentParams) (NodeAgent, error)
|
||||||
UpdateAgentHeartbeat(ctx context.Context, arg UpdateAgentHeartbeatParams) error
|
UpdateAgentHeartbeat(ctx context.Context, arg UpdateAgentHeartbeatParams) error
|
||||||
UpdateBackupTargetByIDAndUser(ctx context.Context, arg UpdateBackupTargetByIDAndUserParams) error
|
UpdateBackupTargetByIDAndUser(ctx context.Context, arg UpdateBackupTargetByIDAndUserParams) error
|
||||||
@@ -156,15 +278,27 @@ type Querier interface {
|
|||||||
// removed replica. $4 is the cumulative rx+tx byte counter.
|
// removed replica. $4 is the cumulative rx+tx byte counter.
|
||||||
UpdateContainerStateByName(ctx context.Context, arg UpdateContainerStateByNameParams) error
|
UpdateContainerStateByName(ctx context.Context, arg UpdateContainerStateByNameParams) error
|
||||||
UpdateContainerStatus(ctx context.Context, arg UpdateContainerStatusParams) error
|
UpdateContainerStatus(ctx context.Context, arg UpdateContainerStatusParams) error
|
||||||
|
UpdateCronJobRun(ctx context.Context, arg UpdateCronJobRunParams) error
|
||||||
UpdateDatabaseServiceNameAndPlanByIDAndUser(ctx context.Context, arg UpdateDatabaseServiceNameAndPlanByIDAndUserParams) error
|
UpdateDatabaseServiceNameAndPlanByIDAndUser(ctx context.Context, arg UpdateDatabaseServiceNameAndPlanByIDAndUserParams) error
|
||||||
UpdateDatabaseServiceNameByIDAndUser(ctx context.Context, arg UpdateDatabaseServiceNameByIDAndUserParams) error
|
UpdateDatabaseServiceNameByIDAndUser(ctx context.Context, arg UpdateDatabaseServiceNameByIDAndUserParams) error
|
||||||
UpdateDatabaseServicePlanByIDAndUser(ctx context.Context, arg UpdateDatabaseServicePlanByIDAndUserParams) error
|
UpdateDatabaseServicePlanByIDAndUser(ctx context.Context, arg UpdateDatabaseServicePlanByIDAndUserParams) error
|
||||||
UpdateOutboundWebhook(ctx context.Context, arg UpdateOutboundWebhookParams) error
|
UpdateOutboundWebhook(ctx context.Context, arg UpdateOutboundWebhookParams) error
|
||||||
|
UpdatePreviewEnvironment(ctx context.Context, arg UpdatePreviewEnvironmentParams) error
|
||||||
UpdateProjectByID(ctx context.Context, arg UpdateProjectByIDParams) (int64, error)
|
UpdateProjectByID(ctx context.Context, arg UpdateProjectByIDParams) (int64, error)
|
||||||
|
UpdateServicePreviewBranch(ctx context.Context, arg UpdateServicePreviewBranchParams) error
|
||||||
|
UpdateUserProfile(ctx context.Context, arg UpdateUserProfileParams) error
|
||||||
UpdateUserTemplate(ctx context.Context, arg UpdateUserTemplateParams) (int64, error)
|
UpdateUserTemplate(ctx context.Context, arg UpdateUserTemplateParams) (int64, error)
|
||||||
|
UpdateVulnerabilityStatus(ctx context.Context, arg UpdateVulnerabilityStatusParams) error
|
||||||
UpdateWebhookDeliveryResult(ctx context.Context, arg UpdateWebhookDeliveryResultParams) error
|
UpdateWebhookDeliveryResult(ctx context.Context, arg UpdateWebhookDeliveryResultParams) error
|
||||||
UpsertEnvironmentVariable(ctx context.Context, arg UpsertEnvironmentVariableParams) error
|
UpsertEnvironmentVariable(ctx context.Context, arg UpsertEnvironmentVariableParams) error
|
||||||
|
UpsertGitHubAppProvider(ctx context.Context, arg UpsertGitHubAppProviderParams) (UpsertGitHubAppProviderRow, error)
|
||||||
|
UpsertGitRepository(ctx context.Context, arg UpsertGitRepositoryParams) error
|
||||||
|
UpsertGitWebhook(ctx context.Context, arg UpsertGitWebhookParams) (UpsertGitWebhookRow, error)
|
||||||
|
UpsertInstanceMetrics(ctx context.Context, arg UpsertInstanceMetricsParams) error
|
||||||
|
UpsertLocalUser(ctx context.Context, arg UpsertLocalUserParams) (UpsertLocalUserRow, error)
|
||||||
|
UpsertNodeMetrics(ctx context.Context, arg UpsertNodeMetricsParams) error
|
||||||
UpsertServiceContainer(ctx context.Context, arg UpsertServiceContainerParams) error
|
UpsertServiceContainer(ctx context.Context, arg UpsertServiceContainerParams) error
|
||||||
|
UpsertServiceMetrics(ctx context.Context, arg UpsertServiceMetricsParams) error
|
||||||
UpsertServiceTemplate(ctx context.Context, arg UpsertServiceTemplateParams) error
|
UpsertServiceTemplate(ctx context.Context, arg UpsertServiceTemplateParams) error
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,355 @@
|
|||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.31.1
|
||||||
|
// source: runtime.sql
|
||||||
|
|
||||||
|
package sqlcdb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
const getLastDeployedImage = `-- name: GetLastDeployedImage :one
|
||||||
|
SELECT (image_name || ':' || image_tag)::text AS image FROM deployments
|
||||||
|
WHERE service_id = $1 AND status = 'deployed' AND image_name <> ''
|
||||||
|
ORDER BY created_at DESC LIMIT 1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetLastDeployedImage(ctx context.Context, serviceID uuid.UUID) (string, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getLastDeployedImage, serviceID)
|
||||||
|
var image string
|
||||||
|
err := row.Scan(&image)
|
||||||
|
return image, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServiceDomainText = `-- name: GetServiceDomainText :one
|
||||||
|
SELECT COALESCE(domain,'') AS domain FROM services WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetServiceDomainText(ctx context.Context, id uuid.UUID) (string, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServiceDomainText, id)
|
||||||
|
var domain string
|
||||||
|
err := row.Scan(&domain)
|
||||||
|
return domain, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServicePort = `-- name: GetServicePort :one
|
||||||
|
SELECT COALESCE(port, 0) AS port FROM services WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetServicePort(ctx context.Context, id uuid.UUID) (int32, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServicePort, id)
|
||||||
|
var port int32
|
||||||
|
err := row.Scan(&port)
|
||||||
|
return port, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServiceRuntimeWithOwner = `-- name: GetServiceRuntimeWithOwner :one
|
||||||
|
SELECT s.id, s.project_id, s.name,
|
||||||
|
COALESCE(s.type, s.service_type, '') AS type, COALESCE(s.status, '') AS status,
|
||||||
|
COALESCE(s.image, s.image_name, '') AS image, COALESCE(s.command, s.start_command, '') AS command,
|
||||||
|
COALESCE(s.environment, '') AS environment, COALESCE(s.git_repo, s.source_url, '') AS git_repo,
|
||||||
|
COALESCE(s.git_branch, '') AS git_branch, COALESCE(s.build_path, '') AS build_path,
|
||||||
|
COALESCE(s.cpu, '') AS cpu, COALESCE(s.memory, '') AS memory,
|
||||||
|
COALESCE(s.replicas, 1) AS replicas, COALESCE(s.port, 0) AS port,
|
||||||
|
COALESCE(s.domain, '') AS domain, COALESCE(s.healthcheck_path, '') AS healthcheck_path,
|
||||||
|
COALESCE(s.restart_policy, 'unless-stopped') AS restart_policy,
|
||||||
|
COALESCE(s.builder, 'auto') AS builder, COALESCE(s.cpu_reserve, '') AS cpu_reserve,
|
||||||
|
COALESCE(s.memory_reserve, '') AS memory_reserve, COALESCE(s.static_build_cmd, '') AS static_build_cmd,
|
||||||
|
COALESCE(s.static_dir, '') AS static_dir,
|
||||||
|
s.created_at, s.updated_at, p.owner_id
|
||||||
|
FROM services s
|
||||||
|
JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE s.id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetServiceRuntimeWithOwnerRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Type string `json:"type"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
Image string `json:"image"`
|
||||||
|
Command string `json:"command"`
|
||||||
|
Environment string `json:"environment"`
|
||||||
|
GitRepo string `json:"git_repo"`
|
||||||
|
GitBranch string `json:"git_branch"`
|
||||||
|
BuildPath string `json:"build_path"`
|
||||||
|
Cpu string `json:"cpu"`
|
||||||
|
Memory string `json:"memory"`
|
||||||
|
Replicas int32 `json:"replicas"`
|
||||||
|
Port int32 `json:"port"`
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
HealthcheckPath string `json:"healthcheck_path"`
|
||||||
|
RestartPolicy string `json:"restart_policy"`
|
||||||
|
Builder string `json:"builder"`
|
||||||
|
CpuReserve string `json:"cpu_reserve"`
|
||||||
|
MemoryReserve string `json:"memory_reserve"`
|
||||||
|
StaticBuildCmd string `json:"static_build_cmd"`
|
||||||
|
StaticDir string `json:"static_dir"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetServiceRuntimeWithOwner(ctx context.Context, id uuid.UUID) (GetServiceRuntimeWithOwnerRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServiceRuntimeWithOwner, id)
|
||||||
|
var i GetServiceRuntimeWithOwnerRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.Name,
|
||||||
|
&i.Type,
|
||||||
|
&i.Status,
|
||||||
|
&i.Image,
|
||||||
|
&i.Command,
|
||||||
|
&i.Environment,
|
||||||
|
&i.GitRepo,
|
||||||
|
&i.GitBranch,
|
||||||
|
&i.BuildPath,
|
||||||
|
&i.Cpu,
|
||||||
|
&i.Memory,
|
||||||
|
&i.Replicas,
|
||||||
|
&i.Port,
|
||||||
|
&i.Domain,
|
||||||
|
&i.HealthcheckPath,
|
||||||
|
&i.RestartPolicy,
|
||||||
|
&i.Builder,
|
||||||
|
&i.CpuReserve,
|
||||||
|
&i.MemoryReserve,
|
||||||
|
&i.StaticBuildCmd,
|
||||||
|
&i.StaticDir,
|
||||||
|
&i.CreatedAt,
|
||||||
|
&i.UpdatedAt,
|
||||||
|
&i.OwnerID,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServiceStatusText = `-- name: GetServiceStatusText :one
|
||||||
|
SELECT COALESCE(status,'') AS status FROM services WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetServiceStatusText(ctx context.Context, id uuid.UUID) (string, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServiceStatusText, id)
|
||||||
|
var status string
|
||||||
|
err := row.Scan(&status)
|
||||||
|
return status, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServiceVariableValue = `-- name: GetServiceVariableValue :one
|
||||||
|
SELECT value FROM environment_variables WHERE service_id = $1 AND key = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetServiceVariableValueParams struct {
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
Key string `json:"key"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetServiceVariableValue(ctx context.Context, arg GetServiceVariableValueParams) (string, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServiceVariableValue, arg.ServiceID, arg.Key)
|
||||||
|
var value string
|
||||||
|
err := row.Scan(&value)
|
||||||
|
return value, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const listNodeAgentsForService = `-- name: ListNodeAgentsForService :many
|
||||||
|
SELECT DISTINCT node_agent_id FROM container_instances WHERE service_id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) ListNodeAgentsForService(ctx context.Context, serviceID string) ([]string, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listNodeAgentsForService, serviceID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []string{}
|
||||||
|
for rows.Next() {
|
||||||
|
var node_agent_id string
|
||||||
|
if err := rows.Scan(&node_agent_id); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, node_agent_id)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listProjectServiceRefs = `-- name: ListProjectServiceRefs :many
|
||||||
|
SELECT id, name FROM services WHERE project_id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListProjectServiceRefsRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListProjectServiceRefs(ctx context.Context, projectID uuid.UUID) ([]ListProjectServiceRefsRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listProjectServiceRefs, projectID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListProjectServiceRefsRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListProjectServiceRefsRow
|
||||||
|
if err := rows.Scan(&i.ID, &i.Name); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listServiceVariableValues = `-- name: ListServiceVariableValues :many
|
||||||
|
SELECT key, value FROM environment_variables WHERE service_id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListServiceVariableValuesRow struct {
|
||||||
|
Key string `json:"key"`
|
||||||
|
Value string `json:"value"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListServiceVariableValues(ctx context.Context, serviceID uuid.UUID) ([]ListServiceVariableValuesRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listServiceVariableValues, serviceID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListServiceVariableValuesRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListServiceVariableValuesRow
|
||||||
|
if err := rows.Scan(&i.Key, &i.Value); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listServiceVariablesWithSecret = `-- name: ListServiceVariablesWithSecret :many
|
||||||
|
SELECT key, value, COALESCE(is_secret, false) AS is_secret
|
||||||
|
FROM environment_variables WHERE service_id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListServiceVariablesWithSecretRow struct {
|
||||||
|
Key string `json:"key"`
|
||||||
|
Value string `json:"value"`
|
||||||
|
IsSecret bool `json:"is_secret"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListServiceVariablesWithSecret(ctx context.Context, serviceID uuid.UUID) ([]ListServiceVariablesWithSecretRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listServiceVariablesWithSecret, serviceID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListServiceVariablesWithSecretRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListServiceVariablesWithSecretRow
|
||||||
|
if err := rows.Scan(&i.Key, &i.Value, &i.IsSecret); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listSleepCandidates = `-- name: ListSleepCandidates :many
|
||||||
|
SELECT id, project_id, name, COALESCE(status,'') AS status,
|
||||||
|
COALESCE(sleep_idle_minutes, 15) AS sleep_idle_minutes
|
||||||
|
FROM services
|
||||||
|
WHERE sleep_enabled AND COALESCE(status,'') IN ('running','deployed','degraded','starting')
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListSleepCandidatesRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
SleepIdleMinutes int32 `json:"sleep_idle_minutes"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListSleepCandidates(ctx context.Context) ([]ListSleepCandidatesRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listSleepCandidates)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListSleepCandidatesRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListSleepCandidatesRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.ProjectID,
|
||||||
|
&i.Name,
|
||||||
|
&i.Status,
|
||||||
|
&i.SleepIdleMinutes,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const setServicePublishedPort = `-- name: SetServicePublishedPort :exec
|
||||||
|
UPDATE services SET published_port = $1 WHERE id = $2 AND published_port <> $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type SetServicePublishedPortParams struct {
|
||||||
|
PublishedPort int32 `json:"published_port"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) SetServicePublishedPort(ctx context.Context, arg SetServicePublishedPortParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, setServicePublishedPort, arg.PublishedPort, arg.ID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const setServiceReplicas = `-- name: SetServiceReplicas :exec
|
||||||
|
UPDATE services SET replicas = $1, updated_at = $2 WHERE id = $3
|
||||||
|
`
|
||||||
|
|
||||||
|
type SetServiceReplicasParams struct {
|
||||||
|
Replicas int32 `json:"replicas"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) SetServiceReplicas(ctx context.Context, arg SetServiceReplicasParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, setServiceReplicas, arg.Replicas, arg.UpdatedAt, arg.ID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
@@ -0,0 +1,318 @@
|
|||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.31.1
|
||||||
|
// source: security.sql
|
||||||
|
|
||||||
|
package sqlcdb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
const checkSecurityProjectAccess = `-- name: CheckSecurityProjectAccess :one
|
||||||
|
SELECT EXISTS (
|
||||||
|
SELECT 1
|
||||||
|
FROM projects p
|
||||||
|
WHERE p.id = $1
|
||||||
|
AND (p.is_approved
|
||||||
|
OR $3::bool
|
||||||
|
OR p.owner_id = $2 OR EXISTS (
|
||||||
|
SELECT 1 FROM project_members pm
|
||||||
|
WHERE pm.project_id = p.id AND pm.user_id = $2
|
||||||
|
))
|
||||||
|
)
|
||||||
|
`
|
||||||
|
|
||||||
|
type CheckSecurityProjectAccessParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
OwnerID uuid.UUID `json:"owner_id"`
|
||||||
|
Column3 bool `json:"column_3"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CheckSecurityProjectAccess(ctx context.Context, arg CheckSecurityProjectAccessParams) (bool, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, checkSecurityProjectAccess, arg.ID, arg.OwnerID, arg.Column3)
|
||||||
|
var exists bool
|
||||||
|
err := row.Scan(&exists)
|
||||||
|
return exists, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const complianceFrameworkExists = `-- name: ComplianceFrameworkExists :one
|
||||||
|
SELECT EXISTS(SELECT 1 FROM compliance_frameworks WHERE id = $1)
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) ComplianceFrameworkExists(ctx context.Context, id uuid.UUID) (bool, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, complianceFrameworkExists, id)
|
||||||
|
var exists bool
|
||||||
|
err := row.Scan(&exists)
|
||||||
|
return exists, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getComplianceReportProjectID = `-- name: GetComplianceReportProjectID :one
|
||||||
|
SELECT project_id FROM compliance_reports WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetComplianceReportProjectID(ctx context.Context, id uuid.UUID) (uuid.UUID, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getComplianceReportProjectID, id)
|
||||||
|
var project_id uuid.UUID
|
||||||
|
err := row.Scan(&project_id)
|
||||||
|
return project_id, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getLatestComplianceReport = `-- name: GetLatestComplianceReport :one
|
||||||
|
SELECT overall_status, score, assessment_date
|
||||||
|
FROM compliance_reports
|
||||||
|
WHERE project_id = $1
|
||||||
|
ORDER BY assessment_date DESC
|
||||||
|
LIMIT 1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetLatestComplianceReportRow struct {
|
||||||
|
OverallStatus string `json:"overall_status"`
|
||||||
|
Score int32 `json:"score"`
|
||||||
|
AssessmentDate time.Time `json:"assessment_date"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetLatestComplianceReport(ctx context.Context, projectID uuid.UUID) (GetLatestComplianceReportRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getLatestComplianceReport, projectID)
|
||||||
|
var i GetLatestComplianceReportRow
|
||||||
|
err := row.Scan(&i.OverallStatus, &i.Score, &i.AssessmentDate)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getLatestSecurityScan = `-- name: GetLatestSecurityScan :one
|
||||||
|
SELECT id, COALESCE((summary->>'score')::int, 0)::int AS score, started_at AS scanned_at, status
|
||||||
|
FROM security_scans
|
||||||
|
WHERE project_id = $1
|
||||||
|
ORDER BY started_at DESC
|
||||||
|
LIMIT 1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetLatestSecurityScanRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Score int32 `json:"score"`
|
||||||
|
ScannedAt time.Time `json:"scanned_at"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetLatestSecurityScan(ctx context.Context, projectID uuid.UUID) (GetLatestSecurityScanRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getLatestSecurityScan, projectID)
|
||||||
|
var i GetLatestSecurityScanRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.Score,
|
||||||
|
&i.ScannedAt,
|
||||||
|
&i.Status,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getProjectOwnerText = `-- name: GetProjectOwnerText :one
|
||||||
|
SELECT owner_id::text FROM projects WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetProjectOwnerText(ctx context.Context, id uuid.UUID) (string, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getProjectOwnerText, id)
|
||||||
|
var owner_id string
|
||||||
|
err := row.Scan(&owner_id)
|
||||||
|
return owner_id, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getSecurityScanProjectID = `-- name: GetSecurityScanProjectID :one
|
||||||
|
SELECT project_id FROM security_scans WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetSecurityScanProjectID(ctx context.Context, id uuid.UUID) (uuid.UUID, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getSecurityScanProjectID, id)
|
||||||
|
var project_id uuid.UUID
|
||||||
|
err := row.Scan(&project_id)
|
||||||
|
return project_id, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getVulnerabilityMetrics = `-- name: GetVulnerabilityMetrics :one
|
||||||
|
SELECT
|
||||||
|
COUNT(*) AS total,
|
||||||
|
COUNT(*) FILTER (WHERE severity = 'critical') AS critical,
|
||||||
|
COUNT(*) FILTER (WHERE severity = 'high') AS high,
|
||||||
|
COUNT(*) FILTER (WHERE severity = 'medium') AS medium,
|
||||||
|
COUNT(*) FILTER (WHERE severity = 'low') AS low,
|
||||||
|
COUNT(*) FILTER (WHERE status = 'open') AS open,
|
||||||
|
COUNT(*) FILTER (WHERE status = 'resolved') AS resolved
|
||||||
|
FROM vulnerabilities
|
||||||
|
WHERE project_id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
type GetVulnerabilityMetricsRow struct {
|
||||||
|
Total int64 `json:"total"`
|
||||||
|
Critical int64 `json:"critical"`
|
||||||
|
High int64 `json:"high"`
|
||||||
|
Medium int64 `json:"medium"`
|
||||||
|
Low int64 `json:"low"`
|
||||||
|
Open int64 `json:"open"`
|
||||||
|
Resolved int64 `json:"resolved"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) GetVulnerabilityMetrics(ctx context.Context, projectID uuid.UUID) (GetVulnerabilityMetricsRow, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getVulnerabilityMetrics, projectID)
|
||||||
|
var i GetVulnerabilityMetricsRow
|
||||||
|
err := row.Scan(
|
||||||
|
&i.Total,
|
||||||
|
&i.Critical,
|
||||||
|
&i.High,
|
||||||
|
&i.Medium,
|
||||||
|
&i.Low,
|
||||||
|
&i.Open,
|
||||||
|
&i.Resolved,
|
||||||
|
)
|
||||||
|
return i, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getVulnerabilityProjectID = `-- name: GetVulnerabilityProjectID :one
|
||||||
|
SELECT project_id FROM vulnerabilities WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetVulnerabilityProjectID(ctx context.Context, id uuid.UUID) (uuid.UUID, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getVulnerabilityProjectID, id)
|
||||||
|
var project_id uuid.UUID
|
||||||
|
err := row.Scan(&project_id)
|
||||||
|
return project_id, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const listComplianceFrameworks = `-- name: ListComplianceFrameworks :many
|
||||||
|
SELECT id, name, description, version, enabled, created_at
|
||||||
|
FROM compliance_frameworks
|
||||||
|
WHERE enabled = true
|
||||||
|
ORDER BY name
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListComplianceFrameworksRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Description sql.NullString `json:"description"`
|
||||||
|
Version string `json:"version"`
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListComplianceFrameworks(ctx context.Context) ([]ListComplianceFrameworksRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listComplianceFrameworks)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListComplianceFrameworksRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListComplianceFrameworksRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.Name,
|
||||||
|
&i.Description,
|
||||||
|
&i.Version,
|
||||||
|
&i.Enabled,
|
||||||
|
&i.CreatedAt,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const listVulnerabilities = `-- name: ListVulnerabilities :many
|
||||||
|
SELECT id, type, severity, title, description, service_id, status, found_at, resolved_at
|
||||||
|
FROM vulnerabilities
|
||||||
|
WHERE project_id = $1
|
||||||
|
ORDER BY
|
||||||
|
CASE severity
|
||||||
|
WHEN 'critical' THEN 1
|
||||||
|
WHEN 'high' THEN 2
|
||||||
|
WHEN 'medium' THEN 3
|
||||||
|
WHEN 'low' THEN 4
|
||||||
|
END,
|
||||||
|
found_at DESC
|
||||||
|
`
|
||||||
|
|
||||||
|
type ListVulnerabilitiesRow struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
Type string `json:"type"`
|
||||||
|
Severity string `json:"severity"`
|
||||||
|
Title string `json:"title"`
|
||||||
|
Description sql.NullString `json:"description"`
|
||||||
|
ServiceID uuid.NullUUID `json:"service_id"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
FoundAt time.Time `json:"found_at"`
|
||||||
|
ResolvedAt sql.NullTime `json:"resolved_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ListVulnerabilities(ctx context.Context, projectID uuid.UUID) ([]ListVulnerabilitiesRow, error) {
|
||||||
|
rows, err := q.db.QueryContext(ctx, listVulnerabilities, projectID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
items := []ListVulnerabilitiesRow{}
|
||||||
|
for rows.Next() {
|
||||||
|
var i ListVulnerabilitiesRow
|
||||||
|
if err := rows.Scan(
|
||||||
|
&i.ID,
|
||||||
|
&i.Type,
|
||||||
|
&i.Severity,
|
||||||
|
&i.Title,
|
||||||
|
&i.Description,
|
||||||
|
&i.ServiceID,
|
||||||
|
&i.Status,
|
||||||
|
&i.FoundAt,
|
||||||
|
&i.ResolvedAt,
|
||||||
|
); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
items = append(items, i)
|
||||||
|
}
|
||||||
|
if err := rows.Close(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rows.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const serviceExistsInProject = `-- name: ServiceExistsInProject :one
|
||||||
|
SELECT EXISTS(SELECT 1 FROM services WHERE id = $1 AND project_id = $2)
|
||||||
|
`
|
||||||
|
|
||||||
|
type ServiceExistsInProjectParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) ServiceExistsInProject(ctx context.Context, arg ServiceExistsInProjectParams) (bool, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, serviceExistsInProject, arg.ID, arg.ProjectID)
|
||||||
|
var exists bool
|
||||||
|
err := row.Scan(&exists)
|
||||||
|
return exists, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const updateVulnerabilityStatus = `-- name: UpdateVulnerabilityStatus :exec
|
||||||
|
UPDATE vulnerabilities SET status = $1, resolved_at = $2 WHERE id = $3
|
||||||
|
`
|
||||||
|
|
||||||
|
type UpdateVulnerabilityStatusParams struct {
|
||||||
|
Status string `json:"status"`
|
||||||
|
ResolvedAt sql.NullTime `json:"resolved_at"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) UpdateVulnerabilityStatus(ctx context.Context, arg UpdateVulnerabilityStatusParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, updateVulnerabilityStatus, arg.Status, arg.ResolvedAt, arg.ID)
|
||||||
|
return err
|
||||||
|
}
|
||||||
@@ -0,0 +1,158 @@
|
|||||||
|
// Code generated by sqlc. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// sqlc v1.31.1
|
||||||
|
// source: service_ops.sql
|
||||||
|
|
||||||
|
package sqlcdb
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
)
|
||||||
|
|
||||||
|
const cloneServiceDomains = `-- name: CloneServiceDomains :exec
|
||||||
|
INSERT INTO service_domains (service_id, domain, is_default, cert_type)
|
||||||
|
SELECT $1, domain, is_default, cert_type FROM service_domains sd WHERE sd.service_id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type CloneServiceDomainsParams struct {
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
ServiceID_2 uuid.UUID `json:"service_id_2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CloneServiceDomains(ctx context.Context, arg CloneServiceDomainsParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, cloneServiceDomains, arg.ServiceID, arg.ServiceID_2)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const cloneServiceRow = `-- name: CloneServiceRow :exec
|
||||||
|
INSERT INTO services
|
||||||
|
(id, project_id, name, environment_id, service_type, source_type, source_url, image_name,
|
||||||
|
build_command, start_command, type, status, image, command, environment,
|
||||||
|
git_repo, git_branch, build_path, cpu, memory, replicas, port, domain,
|
||||||
|
healthcheck_path, restart_policy, volumes, maintenance_mode, basic_auth_users,
|
||||||
|
builder, cpu_reserve, memory_reserve, static_build_cmd, static_dir,
|
||||||
|
traefik_labels, created_at, updated_at)
|
||||||
|
SELECT $1, $2, $3, $4, service_type, source_type, source_url, image_name,
|
||||||
|
build_command, start_command, type, 'stopped', image, command, $5,
|
||||||
|
git_repo, git_branch, build_path, cpu, memory, replicas, port, domain,
|
||||||
|
healthcheck_path, restart_policy, volumes, $6, $7,
|
||||||
|
builder, cpu_reserve, memory_reserve, static_build_cmd, static_dir,
|
||||||
|
traefik_labels, $8, $8
|
||||||
|
FROM services s WHERE s.id = $9
|
||||||
|
`
|
||||||
|
|
||||||
|
type CloneServiceRowParams struct {
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
EnvironmentID uuid.UUID `json:"environment_id"`
|
||||||
|
Environment sql.NullString `json:"environment"`
|
||||||
|
MaintenanceMode bool `json:"maintenance_mode"`
|
||||||
|
BasicAuthUsers string `json:"basic_auth_users"`
|
||||||
|
CreatedAt sql.NullTime `json:"created_at"`
|
||||||
|
ID_2 uuid.UUID `json:"id_2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CloneServiceRow(ctx context.Context, arg CloneServiceRowParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, cloneServiceRow,
|
||||||
|
arg.ID,
|
||||||
|
arg.ProjectID,
|
||||||
|
arg.Name,
|
||||||
|
arg.EnvironmentID,
|
||||||
|
arg.Environment,
|
||||||
|
arg.MaintenanceMode,
|
||||||
|
arg.BasicAuthUsers,
|
||||||
|
arg.CreatedAt,
|
||||||
|
arg.ID_2,
|
||||||
|
)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const cloneServiceVariables = `-- name: CloneServiceVariables :exec
|
||||||
|
INSERT INTO environment_variables (id, service_id, key, value, is_secret)
|
||||||
|
SELECT gen_random_uuid(), $1, key, value, is_secret FROM environment_variables ev WHERE ev.service_id = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type CloneServiceVariablesParams struct {
|
||||||
|
ServiceID uuid.UUID `json:"service_id"`
|
||||||
|
ServiceID_2 uuid.UUID `json:"service_id_2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CloneServiceVariables(ctx context.Context, arg CloneServiceVariablesParams) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, cloneServiceVariables, arg.ServiceID, arg.ServiceID_2)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const countServicesByName = `-- name: CountServicesByName :one
|
||||||
|
SELECT COUNT(*) FROM services WHERE project_id = $1 AND name = $2
|
||||||
|
`
|
||||||
|
|
||||||
|
type CountServicesByNameParams struct {
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) CountServicesByName(ctx context.Context, arg CountServicesByNameParams) (int64, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, countServicesByName, arg.ProjectID, arg.Name)
|
||||||
|
var count int64
|
||||||
|
err := row.Scan(&count)
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const deleteServiceByID = `-- name: DeleteServiceByID :exec
|
||||||
|
DELETE FROM services WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) DeleteServiceByID(ctx context.Context, id uuid.UUID) error {
|
||||||
|
_, err := q.db.ExecContext(ctx, deleteServiceByID, id)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getProjectOwner = `-- name: GetProjectOwner :one
|
||||||
|
SELECT owner_id FROM projects WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetProjectOwner(ctx context.Context, id uuid.UUID) (uuid.UUID, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getProjectOwner, id)
|
||||||
|
var owner_id uuid.UUID
|
||||||
|
err := row.Scan(&owner_id)
|
||||||
|
return owner_id, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const getServiceEnvironment = `-- name: GetServiceEnvironment :one
|
||||||
|
SELECT COALESCE(environment, '') FROM services WHERE id = $1
|
||||||
|
`
|
||||||
|
|
||||||
|
func (q *Queries) GetServiceEnvironment(ctx context.Context, id uuid.UUID) (string, error) {
|
||||||
|
row := q.db.QueryRowContext(ctx, getServiceEnvironment, id)
|
||||||
|
var environment string
|
||||||
|
err := row.Scan(&environment)
|
||||||
|
return environment, err
|
||||||
|
}
|
||||||
|
|
||||||
|
const moveServiceProject = `-- name: MoveServiceProject :execrows
|
||||||
|
UPDATE services SET project_id = $1, environment_id = $2, updated_at = $3 WHERE id = $4
|
||||||
|
`
|
||||||
|
|
||||||
|
type MoveServiceProjectParams struct {
|
||||||
|
ProjectID uuid.UUID `json:"project_id"`
|
||||||
|
EnvironmentID uuid.UUID `json:"environment_id"`
|
||||||
|
UpdatedAt sql.NullTime `json:"updated_at"`
|
||||||
|
ID uuid.UUID `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (q *Queries) MoveServiceProject(ctx context.Context, arg MoveServiceProjectParams) (int64, error) {
|
||||||
|
result, err := q.db.ExecContext(ctx, moveServiceProject,
|
||||||
|
arg.ProjectID,
|
||||||
|
arg.EnvironmentID,
|
||||||
|
arg.UpdatedAt,
|
||||||
|
arg.ID,
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
return result.RowsAffected()
|
||||||
|
}
|
||||||
@@ -4,293 +4,286 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"strconv"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"containr/internal/database/sqlcdb"
|
||||||
|
|
||||||
_ "github.com/lib/pq"
|
_ "github.com/lib/pq"
|
||||||
)
|
)
|
||||||
|
|
||||||
// PostgreSQLMetricsStorage implements MetricsStorage using PostgreSQL
|
// PostgreSQLMetricsStorage implements MetricsStorage using PostgreSQL
|
||||||
|
// through sqlc-generated queries (sqlc/queries/metrics.sql).
|
||||||
|
//
|
||||||
|
// Per-container node metrics are deliberately not persisted: the
|
||||||
|
// container_metrics table belongs to the container_instances model and
|
||||||
|
// has no place for the node/container samples collected here. Node and
|
||||||
|
// service/instance rows map cleanly, so those are stored.
|
||||||
type PostgreSQLMetricsStorage struct {
|
type PostgreSQLMetricsStorage struct {
|
||||||
db *sql.DB
|
q *sqlcdb.Queries
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewPostgreSQLMetricsStorage creates a new PostgreSQL metrics storage
|
// NewPostgreSQLMetricsStorage creates a new PostgreSQL metrics storage
|
||||||
func NewPostgreSQLMetricsStorage(db *sql.DB) *PostgreSQLMetricsStorage {
|
func NewPostgreSQLMetricsStorage(db *sql.DB) *PostgreSQLMetricsStorage {
|
||||||
return &PostgreSQLMetricsStorage{db: db}
|
return &PostgreSQLMetricsStorage{q: sqlcdb.New(db)}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func numStr(v float64) sql.NullString {
|
||||||
|
return sql.NullString{String: strconv.FormatFloat(v, 'f', -1, 64), Valid: true}
|
||||||
|
}
|
||||||
|
|
||||||
|
func int64N(v int64) sql.NullInt64 {
|
||||||
|
return sql.NullInt64{Int64: v, Valid: true}
|
||||||
|
}
|
||||||
|
|
||||||
|
func int32N(v int64) sql.NullInt32 {
|
||||||
|
return sql.NullInt32{Int32: int32(v), Valid: true}
|
||||||
|
}
|
||||||
|
|
||||||
|
func strN(v string) sql.NullString {
|
||||||
|
return sql.NullString{String: v, Valid: v != ""}
|
||||||
|
}
|
||||||
|
|
||||||
|
func timeN(t time.Time) sql.NullTime {
|
||||||
|
return sql.NullTime{Time: t, Valid: !t.IsZero()}
|
||||||
|
}
|
||||||
|
|
||||||
|
func f64v(n sql.NullString) float64 {
|
||||||
|
f, _ := strconv.ParseFloat(n.String, 64)
|
||||||
|
return f
|
||||||
|
}
|
||||||
|
|
||||||
|
func i64v(n sql.NullInt64) int64 { return n.Int64 }
|
||||||
|
|
||||||
|
func i32v(n sql.NullInt32) int64 { return int64(n.Int32) }
|
||||||
|
|
||||||
// StoreNodeMetrics stores node metrics in the database
|
// StoreNodeMetrics stores node metrics in the database
|
||||||
func (s *PostgreSQLMetricsStorage) StoreNodeMetrics(ctx context.Context, metrics *NodeMetrics) error {
|
func (s *PostgreSQLMetricsStorage) StoreNodeMetrics(ctx context.Context, metrics *NodeMetrics) error {
|
||||||
query := `
|
err := s.q.UpsertNodeMetrics(ctx, sqlcdb.UpsertNodeMetricsParams{
|
||||||
INSERT INTO node_metrics (
|
NodeID: metrics.NodeID,
|
||||||
node_id, timestamp, cpu_usage, cpu_cores, load_avg_1, load_avg_5, load_avg_15,
|
Timestamp: metrics.Timestamp,
|
||||||
memory_total, memory_used, memory_available, memory_usage_percent,
|
CpuUsage: numStr(metrics.CPU.UsagePercent),
|
||||||
storage_total, storage_used, storage_available, storage_usage_percent,
|
CpuCores: numStr(metrics.CPU.UsageCores),
|
||||||
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
LoadAvg1: numStr(metrics.CPU.LoadAverage1),
|
||||||
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
LoadAvg5: numStr(metrics.CPU.LoadAverage5),
|
||||||
uptime, processes, os, kernel, architecture
|
LoadAvg15: numStr(metrics.CPU.LoadAverage15),
|
||||||
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21, $22, $23, $24, $25, $26, $27, $28)
|
MemoryTotal: int64N(metrics.Memory.Total),
|
||||||
ON CONFLICT (node_id, timestamp) DO UPDATE SET
|
MemoryUsed: int64N(metrics.Memory.Used),
|
||||||
cpu_usage = EXCLUDED.cpu_usage,
|
MemoryAvailable: int64N(metrics.Memory.Available),
|
||||||
cpu_cores = EXCLUDED.cpu_cores,
|
MemoryUsagePercent: numStr(metrics.Memory.UsagePercent),
|
||||||
load_avg_1 = EXCLUDED.load_avg_1,
|
StorageTotal: int64N(metrics.Storage.Total),
|
||||||
load_avg_5 = EXCLUDED.load_avg_5,
|
StorageUsed: int64N(metrics.Storage.Used),
|
||||||
load_avg_15 = EXCLUDED.load_avg_15,
|
StorageAvailable: int64N(metrics.Storage.Available),
|
||||||
memory_total = EXCLUDED.memory_total,
|
StorageUsagePercent: numStr(metrics.Storage.UsagePercent),
|
||||||
memory_used = EXCLUDED.memory_used,
|
NetworkBytesIn: int64N(metrics.Network.BytesIn),
|
||||||
memory_available = EXCLUDED.memory_available,
|
NetworkBytesOut: int64N(metrics.Network.BytesOut),
|
||||||
memory_usage_percent = EXCLUDED.memory_usage_percent,
|
NetworkPacketsIn: int64N(metrics.Network.PacketsIn),
|
||||||
storage_total = EXCLUDED.storage_total,
|
NetworkPacketsOut: int64N(metrics.Network.PacketsOut),
|
||||||
storage_used = EXCLUDED.storage_used,
|
NetworkConnectionsIn: int32N(metrics.Network.ConnectionsIn),
|
||||||
storage_available = EXCLUDED.storage_available,
|
NetworkConnectionsOut: int32N(metrics.Network.ConnectionsOut),
|
||||||
storage_usage_percent = EXCLUDED.storage_usage_percent,
|
NetworkErrorsIn: int64N(metrics.Network.ErrorsIn),
|
||||||
network_bytes_in = EXCLUDED.network_bytes_in,
|
NetworkErrorsOut: int64N(metrics.Network.ErrorsOut),
|
||||||
network_bytes_out = EXCLUDED.network_bytes_out,
|
// INTERVAL param — Postgres casts the bigint as seconds.
|
||||||
network_packets_in = EXCLUDED.network_packets_in,
|
Uptime: int64N(int64(metrics.System.Uptime.Seconds())),
|
||||||
network_packets_out = EXCLUDED.network_packets_out,
|
Processes: sql.NullInt32{Int32: int32(metrics.System.Processes), Valid: true},
|
||||||
network_connections_in = EXCLUDED.network_connections_in,
|
Os: strN(metrics.System.OS),
|
||||||
network_connections_out = EXCLUDED.network_connections_out,
|
Kernel: strN(metrics.System.Kernel),
|
||||||
network_errors_in = EXCLUDED.network_errors_in,
|
Architecture: strN(metrics.System.Architecture),
|
||||||
network_errors_out = EXCLUDED.network_errors_out,
|
})
|
||||||
uptime = EXCLUDED.uptime,
|
|
||||||
processes = EXCLUDED.processes,
|
|
||||||
os = EXCLUDED.os,
|
|
||||||
kernel = EXCLUDED.kernel,
|
|
||||||
architecture = EXCLUDED.architecture
|
|
||||||
`
|
|
||||||
|
|
||||||
_, err := s.db.ExecContext(ctx, query,
|
|
||||||
metrics.NodeID, metrics.Timestamp, metrics.CPU.UsagePercent, metrics.CPU.UsageCores,
|
|
||||||
metrics.CPU.LoadAverage1, metrics.CPU.LoadAverage5, metrics.CPU.LoadAverage15,
|
|
||||||
metrics.Memory.Total, metrics.Memory.Used, metrics.Memory.Available, metrics.Memory.UsagePercent,
|
|
||||||
metrics.Storage.Total, metrics.Storage.Used, metrics.Storage.Available, metrics.Storage.UsagePercent,
|
|
||||||
metrics.Network.BytesIn, metrics.Network.BytesOut, metrics.Network.PacketsIn, metrics.Network.PacketsOut,
|
|
||||||
metrics.Network.ConnectionsIn, metrics.Network.ConnectionsOut, metrics.Network.ErrorsIn, metrics.Network.ErrorsOut,
|
|
||||||
metrics.System.Uptime, metrics.System.Processes, metrics.System.OS, metrics.System.Kernel, metrics.System.Architecture,
|
|
||||||
)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to store node metrics: %w", err)
|
return fmt.Errorf("failed to store node metrics: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Store container metrics
|
|
||||||
for _, container := range metrics.Containers {
|
|
||||||
if err := s.storeContainerMetrics(ctx, metrics.NodeID, metrics.Timestamp, container); err != nil {
|
|
||||||
return fmt.Errorf("failed to store container metrics: %w", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// StoreServiceMetrics stores service metrics in the database
|
// StoreServiceMetrics stores service metrics in the database
|
||||||
func (s *PostgreSQLMetricsStorage) StoreServiceMetrics(ctx context.Context, metrics *ServiceMetrics) error {
|
func (s *PostgreSQLMetricsStorage) StoreServiceMetrics(ctx context.Context, metrics *ServiceMetrics) error {
|
||||||
query := `
|
err := s.q.UpsertServiceMetrics(ctx, sqlcdb.UpsertServiceMetricsParams{
|
||||||
INSERT INTO service_metrics (
|
ServiceID: metrics.ServiceID,
|
||||||
service_id, service_name, project_id, timestamp,
|
ServiceName: metrics.ServiceName,
|
||||||
requests_total, requests_success, requests_errors, requests_avg_latency,
|
ProjectID: metrics.ProjectID,
|
||||||
requests_p95_latency, requests_p99_latency, requests_throughput,
|
Timestamp: metrics.Timestamp,
|
||||||
errors_total, errors_rate, performance_response_time, performance_throughput,
|
RequestsTotal: int64N(metrics.Requests.Total),
|
||||||
performance_concurrency, performance_saturation, performance_utilization,
|
RequestsSuccess: int64N(metrics.Requests.Success),
|
||||||
resource_cpu_usage, resource_memory_usage, resource_storage_usage,
|
RequestsErrors: int64N(metrics.Requests.Errors),
|
||||||
resource_network_usage, resource_score
|
RequestsAvgLatency: numStr(metrics.Requests.AvgLatency),
|
||||||
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21, $22, $23)
|
RequestsP95Latency: numStr(metrics.Requests.P95Latency),
|
||||||
ON CONFLICT (service_id, timestamp) DO UPDATE SET
|
RequestsP99Latency: numStr(metrics.Requests.P99Latency),
|
||||||
requests_total = EXCLUDED.requests_total,
|
RequestsThroughput: numStr(metrics.Requests.Throughput),
|
||||||
requests_success = EXCLUDED.requests_success,
|
ErrorsTotal: int64N(metrics.Errors.Total),
|
||||||
requests_errors = EXCLUDED.requests_errors,
|
ErrorsRate: numStr(metrics.Errors.Rate),
|
||||||
requests_avg_latency = EXCLUDED.requests_avg_latency,
|
PerformanceResponseTime: numStr(metrics.Performance.ResponseTime),
|
||||||
requests_p95_latency = EXCLUDED.requests_p95_latency,
|
PerformanceThroughput: numStr(metrics.Performance.Throughput),
|
||||||
requests_p99_latency = EXCLUDED.requests_p99_latency,
|
PerformanceConcurrency: int64N(metrics.Performance.Concurrency),
|
||||||
requests_throughput = EXCLUDED.requests_throughput,
|
PerformanceSaturation: numStr(metrics.Performance.Saturation),
|
||||||
errors_total = EXCLUDED.errors_total,
|
PerformanceUtilization: numStr(metrics.Performance.Utilization),
|
||||||
errors_rate = EXCLUDED.errors_rate,
|
ResourceCpuUsage: numStr(metrics.Resources.CPUUsage),
|
||||||
performance_response_time = EXCLUDED.performance_response_time,
|
ResourceMemoryUsage: int64N(metrics.Resources.MemoryUsage),
|
||||||
performance_throughput = EXCLUDED.performance_throughput,
|
ResourceStorageUsage: int64N(metrics.Resources.StorageUsage),
|
||||||
performance_concurrency = EXCLUDED.performance_concurrency,
|
ResourceNetworkUsage: int64N(metrics.Resources.NetworkUsage),
|
||||||
performance_saturation = EXCLUDED.performance_saturation,
|
ResourceScore: numStr(metrics.Resources.ResourceScore),
|
||||||
performance_utilization = EXCLUDED.performance_utilization,
|
})
|
||||||
resource_cpu_usage = EXCLUDED.resource_cpu_usage,
|
|
||||||
resource_memory_usage = EXCLUDED.resource_memory_usage,
|
|
||||||
resource_storage_usage = EXCLUDED.resource_storage_usage,
|
|
||||||
resource_network_usage = EXCLUDED.resource_network_usage,
|
|
||||||
resource_score = EXCLUDED.resource_score
|
|
||||||
`
|
|
||||||
|
|
||||||
_, err := s.db.ExecContext(ctx, query,
|
|
||||||
metrics.ServiceID, metrics.ServiceName, metrics.ProjectID, metrics.Timestamp,
|
|
||||||
metrics.Requests.Total, metrics.Requests.Success, metrics.Requests.Errors,
|
|
||||||
metrics.Requests.AvgLatency, metrics.Requests.P95Latency, metrics.Requests.P99Latency,
|
|
||||||
metrics.Requests.Throughput, metrics.Errors.Total, metrics.Errors.Rate,
|
|
||||||
metrics.Performance.ResponseTime, metrics.Performance.Throughput,
|
|
||||||
metrics.Performance.Concurrency, metrics.Performance.Saturation, metrics.Performance.Utilization,
|
|
||||||
metrics.Resources.CPUUsage, metrics.Resources.MemoryUsage, metrics.Resources.StorageUsage,
|
|
||||||
metrics.Resources.NetworkUsage, metrics.Resources.ResourceScore,
|
|
||||||
)
|
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to store service metrics: %w", err)
|
return fmt.Errorf("failed to store service metrics: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Store instance metrics
|
// The service row must land first — instance_metrics references
|
||||||
|
// service_metrics(service_id, timestamp).
|
||||||
for _, instance := range metrics.Instances {
|
for _, instance := range metrics.Instances {
|
||||||
if err := s.storeInstanceMetrics(ctx, metrics.ServiceID, metrics.Timestamp, instance); err != nil {
|
if err := s.storeInstanceMetrics(ctx, metrics.ServiceID, metrics.Timestamp, instance); err != nil {
|
||||||
return fmt.Errorf("failed to store instance metrics: %w", err)
|
return fmt.Errorf("failed to store instance metrics: %w", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetNodeMetrics retrieves node metrics from the database
|
// GetNodeMetrics retrieves node metrics from the database
|
||||||
func (s *PostgreSQLMetricsStorage) GetNodeMetrics(ctx context.Context, nodeID string, from, to time.Time) ([]*NodeMetrics, error) {
|
func (s *PostgreSQLMetricsStorage) GetNodeMetrics(ctx context.Context, nodeID string, from, to time.Time) ([]*NodeMetrics, error) {
|
||||||
query := `
|
rows, err := s.q.ListNodeMetrics(ctx, sqlcdb.ListNodeMetricsParams{
|
||||||
SELECT node_id, timestamp, cpu_usage, cpu_cores, load_avg_1, load_avg_5, load_avg_15,
|
NodeID: nodeID,
|
||||||
memory_total, memory_used, memory_available, memory_usage_percent,
|
Timestamp: from,
|
||||||
storage_total, storage_used, storage_available, storage_usage_percent,
|
Timestamp_2: to,
|
||||||
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
})
|
||||||
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
|
||||||
uptime, processes, os, kernel, architecture
|
|
||||||
FROM node_metrics
|
|
||||||
WHERE node_id = $1 AND timestamp BETWEEN $2 AND $3
|
|
||||||
ORDER BY timestamp ASC
|
|
||||||
`
|
|
||||||
|
|
||||||
rows, err := s.db.QueryContext(ctx, query, nodeID, from, to)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to query node metrics: %w", err)
|
return nil, fmt.Errorf("failed to query node metrics: %w", err)
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var metrics []*NodeMetrics
|
var metrics []*NodeMetrics
|
||||||
for rows.Next() {
|
for _, r := range rows {
|
||||||
var m NodeMetrics
|
metrics = append(metrics, &NodeMetrics{
|
||||||
err := rows.Scan(
|
NodeID: r.NodeID,
|
||||||
&m.NodeID, &m.Timestamp, &m.CPU.UsagePercent, &m.CPU.UsageCores,
|
Timestamp: r.Timestamp,
|
||||||
&m.CPU.LoadAverage1, &m.CPU.LoadAverage5, &m.CPU.LoadAverage15,
|
CPU: CPUMetrics{
|
||||||
&m.Memory.Total, &m.Memory.Used, &m.Memory.Available, &m.Memory.UsagePercent,
|
UsagePercent: f64v(r.CpuUsage),
|
||||||
&m.Storage.Total, &m.Storage.Used, &m.Storage.Available, &m.Storage.UsagePercent,
|
UsageCores: f64v(r.CpuCores),
|
||||||
&m.Network.BytesIn, &m.Network.BytesOut, &m.Network.PacketsIn, &m.Network.PacketsOut,
|
LoadAverage1: f64v(r.LoadAvg1),
|
||||||
&m.Network.ConnectionsIn, &m.Network.ConnectionsOut, &m.Network.ErrorsIn, &m.Network.ErrorsOut,
|
LoadAverage5: f64v(r.LoadAvg5),
|
||||||
&m.System.Uptime, &m.System.Processes, &m.System.OS, &m.System.Kernel, &m.System.Architecture,
|
LoadAverage15: f64v(r.LoadAvg15),
|
||||||
)
|
},
|
||||||
if err != nil {
|
Memory: MemoryMetrics{
|
||||||
return nil, fmt.Errorf("failed to scan node metrics: %w", err)
|
Total: i64v(r.MemoryTotal),
|
||||||
}
|
Used: i64v(r.MemoryUsed),
|
||||||
|
Available: i64v(r.MemoryAvailable),
|
||||||
// Get container metrics for this timestamp
|
UsagePercent: f64v(r.MemoryUsagePercent),
|
||||||
containers, err := s.getContainerMetrics(ctx, nodeID, m.Timestamp)
|
},
|
||||||
if err != nil {
|
Storage: StorageMetrics{
|
||||||
return nil, fmt.Errorf("failed to get container metrics: %w", err)
|
Total: i64v(r.StorageTotal),
|
||||||
}
|
Used: i64v(r.StorageUsed),
|
||||||
m.Containers = containers
|
Available: i64v(r.StorageAvailable),
|
||||||
|
UsagePercent: f64v(r.StorageUsagePercent),
|
||||||
metrics = append(metrics, &m)
|
},
|
||||||
|
Network: NetworkMetrics{
|
||||||
|
BytesIn: i64v(r.NetworkBytesIn),
|
||||||
|
BytesOut: i64v(r.NetworkBytesOut),
|
||||||
|
PacketsIn: i64v(r.NetworkPacketsIn),
|
||||||
|
PacketsOut: i64v(r.NetworkPacketsOut),
|
||||||
|
ConnectionsIn: i32v(r.NetworkConnectionsIn),
|
||||||
|
ConnectionsOut: i32v(r.NetworkConnectionsOut),
|
||||||
|
ErrorsIn: i64v(r.NetworkErrorsIn),
|
||||||
|
ErrorsOut: i64v(r.NetworkErrorsOut),
|
||||||
|
},
|
||||||
|
System: SystemMetrics{
|
||||||
|
Uptime: time.Duration(r.UptimeSeconds) * time.Second,
|
||||||
|
Processes: int(r.Processes.Int32),
|
||||||
|
OS: r.Os.String,
|
||||||
|
Kernel: r.Kernel.String,
|
||||||
|
Architecture: r.Architecture.String,
|
||||||
|
},
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
return metrics, nil
|
return metrics, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetServiceMetrics retrieves service metrics from the database
|
// GetServiceMetrics retrieves service metrics from the database
|
||||||
func (s *PostgreSQLMetricsStorage) GetServiceMetrics(ctx context.Context, serviceID string, from, to time.Time) ([]*ServiceMetrics, error) {
|
func (s *PostgreSQLMetricsStorage) GetServiceMetrics(ctx context.Context, serviceID string, from, to time.Time) ([]*ServiceMetrics, error) {
|
||||||
query := `
|
rows, err := s.q.ListServiceMetrics(ctx, sqlcdb.ListServiceMetricsParams{
|
||||||
SELECT service_id, service_name, project_id, timestamp,
|
ServiceID: serviceID,
|
||||||
requests_total, requests_success, requests_errors, requests_avg_latency,
|
Timestamp: from,
|
||||||
requests_p95_latency, requests_p99_latency, requests_throughput,
|
Timestamp_2: to,
|
||||||
errors_total, errors_rate, performance_response_time, performance_throughput,
|
})
|
||||||
performance_concurrency, performance_saturation, performance_utilization,
|
|
||||||
resource_cpu_usage, resource_memory_usage, resource_storage_usage,
|
|
||||||
resource_network_usage, resource_score
|
|
||||||
FROM service_metrics
|
|
||||||
WHERE service_id = $1 AND timestamp BETWEEN $2 AND $3
|
|
||||||
ORDER BY timestamp ASC
|
|
||||||
`
|
|
||||||
|
|
||||||
rows, err := s.db.QueryContext(ctx, query, serviceID, from, to)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to query service metrics: %w", err)
|
return nil, fmt.Errorf("failed to query service metrics: %w", err)
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var metrics []*ServiceMetrics
|
var metrics []*ServiceMetrics
|
||||||
for rows.Next() {
|
for _, r := range rows {
|
||||||
var m ServiceMetrics
|
m := &ServiceMetrics{
|
||||||
err := rows.Scan(
|
ServiceID: r.ServiceID,
|
||||||
&m.ServiceID, &m.ServiceName, &m.ProjectID, &m.Timestamp,
|
ServiceName: r.ServiceName,
|
||||||
&m.Requests.Total, &m.Requests.Success, &m.Requests.Errors,
|
ProjectID: r.ProjectID,
|
||||||
&m.Requests.AvgLatency, &m.Requests.P95Latency, &m.Requests.P99Latency,
|
Timestamp: r.Timestamp,
|
||||||
&m.Requests.Throughput, &m.Errors.Total, &m.Errors.Rate,
|
Requests: RequestMetrics{
|
||||||
&m.Performance.ResponseTime, &m.Performance.Throughput,
|
Total: i64v(r.RequestsTotal),
|
||||||
&m.Performance.Concurrency, &m.Performance.Saturation, &m.Performance.Utilization,
|
Success: i64v(r.RequestsSuccess),
|
||||||
&m.Resources.CPUUsage, &m.Resources.MemoryUsage, &m.Resources.StorageUsage,
|
Errors: i64v(r.RequestsErrors),
|
||||||
&m.Resources.NetworkUsage, &m.Resources.ResourceScore,
|
AvgLatency: f64v(r.RequestsAvgLatency),
|
||||||
)
|
P95Latency: f64v(r.RequestsP95Latency),
|
||||||
if err != nil {
|
P99Latency: f64v(r.RequestsP99Latency),
|
||||||
return nil, fmt.Errorf("failed to scan service metrics: %w", err)
|
Throughput: f64v(r.RequestsThroughput),
|
||||||
|
},
|
||||||
|
Errors: ErrorMetrics{
|
||||||
|
Total: i64v(r.ErrorsTotal),
|
||||||
|
Rate: f64v(r.ErrorsRate),
|
||||||
|
},
|
||||||
|
Performance: PerformanceMetrics{
|
||||||
|
ResponseTime: f64v(r.PerformanceResponseTime),
|
||||||
|
Throughput: f64v(r.PerformanceThroughput),
|
||||||
|
Concurrency: i64v(r.PerformanceConcurrency),
|
||||||
|
Saturation: f64v(r.PerformanceSaturation),
|
||||||
|
Utilization: f64v(r.PerformanceUtilization),
|
||||||
|
},
|
||||||
|
Resources: ResourceMetrics{
|
||||||
|
CPUUsage: f64v(r.ResourceCpuUsage),
|
||||||
|
MemoryUsage: i64v(r.ResourceMemoryUsage),
|
||||||
|
StorageUsage: i64v(r.ResourceStorageUsage),
|
||||||
|
NetworkUsage: i64v(r.ResourceNetworkUsage),
|
||||||
|
ResourceScore: f64v(r.ResourceScore),
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get instance metrics for this timestamp
|
instances, err := s.getInstanceMetrics(ctx, serviceID, r.Timestamp)
|
||||||
instances, err := s.getInstanceMetrics(ctx, serviceID, m.Timestamp)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to get instance metrics: %w", err)
|
return nil, fmt.Errorf("failed to get instance metrics: %w", err)
|
||||||
}
|
}
|
||||||
m.Instances = instances
|
m.Instances = instances
|
||||||
|
metrics = append(metrics, m)
|
||||||
metrics = append(metrics, &m)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return metrics, nil
|
return metrics, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetAggregatedMetrics retrieves aggregated metrics based on a query
|
// GetAggregatedMetrics retrieves aggregated metrics based on a query.
|
||||||
|
// Bucketing uses date_bin (PostgreSQL 14+) — no TimescaleDB needed.
|
||||||
func (s *PostgreSQLMetricsStorage) GetAggregatedMetrics(ctx context.Context, query MetricsQuery) (*AggregatedMetrics, error) {
|
func (s *PostgreSQLMetricsStorage) GetAggregatedMetrics(ctx context.Context, query MetricsQuery) (*AggregatedMetrics, error) {
|
||||||
// This is a simplified implementation
|
|
||||||
// In a real system, you'd build dynamic SQL based on the query
|
|
||||||
|
|
||||||
var timeSeries []TimeSeriesPoint
|
var timeSeries []TimeSeriesPoint
|
||||||
var summary map[string]MetricSummary
|
var summary map[string]MetricSummary
|
||||||
|
|
||||||
switch query.Type {
|
switch query.Type {
|
||||||
case "node":
|
case "node":
|
||||||
// Aggregate node metrics
|
seconds := int64(query.Interval.Seconds())
|
||||||
nodeQuery := `
|
if seconds <= 0 {
|
||||||
SELECT
|
seconds = 300
|
||||||
time_bucket($1, timestamp) AS bucket,
|
}
|
||||||
AVG(cpu_usage) as avg_cpu,
|
rows, err := s.q.AggregateNodeMetrics(ctx, sqlcdb.AggregateNodeMetricsParams{
|
||||||
AVG(memory_usage_percent) as avg_memory,
|
Column1: seconds,
|
||||||
AVG(storage_usage_percent) as avg_storage
|
NodeID: query.ID,
|
||||||
FROM node_metrics
|
Timestamp: query.From,
|
||||||
WHERE node_id = $2 AND timestamp BETWEEN $3 AND $4
|
Timestamp_2: query.To,
|
||||||
GROUP BY bucket
|
})
|
||||||
ORDER BY bucket ASC
|
|
||||||
`
|
|
||||||
|
|
||||||
rows, err := s.db.QueryContext(ctx, nodeQuery, query.Interval, query.ID, query.From, query.To)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to query aggregated node metrics: %w", err)
|
return nil, fmt.Errorf("failed to query aggregated node metrics: %w", err)
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
for _, r := range rows {
|
||||||
|
bucket, _ := r.Bucket.(time.Time)
|
||||||
for rows.Next() {
|
timeSeries = append(timeSeries, TimeSeriesPoint{
|
||||||
var bucket time.Time
|
|
||||||
var avgCPU, avgMemory, avgStorage float64
|
|
||||||
if err := rows.Scan(&bucket, &avgCPU, &avgMemory, &avgStorage); err != nil {
|
|
||||||
return nil, fmt.Errorf("failed to scan aggregated metrics: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
point := TimeSeriesPoint{
|
|
||||||
Timestamp: bucket,
|
Timestamp: bucket,
|
||||||
Values: map[string]float64{
|
Values: map[string]float64{
|
||||||
"cpu_usage": avgCPU,
|
"cpu_usage": r.AvgCpu,
|
||||||
"memory_usage": avgMemory,
|
"memory_usage": r.AvgMemory,
|
||||||
"storage_usage": avgStorage,
|
"storage_usage": r.AvgStorage,
|
||||||
},
|
},
|
||||||
}
|
})
|
||||||
timeSeries = append(timeSeries, point)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Calculate summary statistics
|
|
||||||
summary = map[string]MetricSummary{
|
summary = map[string]MetricSummary{
|
||||||
"cpu_usage": calculateSummary(timeSeries, "cpu_usage"),
|
"cpu_usage": calculateSummary(timeSeries, "cpu_usage"),
|
||||||
"memory_usage": calculateSummary(timeSeries, "memory_usage"),
|
"memory_usage": calculateSummary(timeSeries, "memory_usage"),
|
||||||
@@ -305,137 +298,69 @@ func (s *PostgreSQLMetricsStorage) GetAggregatedMetrics(ctx context.Context, que
|
|||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Helper methods
|
|
||||||
|
|
||||||
func (s *PostgreSQLMetricsStorage) storeContainerMetrics(ctx context.Context, nodeID string, timestamp time.Time, container ContainerMetrics) error {
|
|
||||||
query := `
|
|
||||||
INSERT INTO container_metrics (
|
|
||||||
node_id, timestamp, container_id, name, state, cpu, memory,
|
|
||||||
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out, start_time
|
|
||||||
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)
|
|
||||||
ON CONFLICT (node_id, timestamp, container_id) DO UPDATE SET
|
|
||||||
name = EXCLUDED.name,
|
|
||||||
state = EXCLUDED.state,
|
|
||||||
cpu = EXCLUDED.cpu,
|
|
||||||
memory = EXCLUDED.memory,
|
|
||||||
network_bytes_in = EXCLUDED.network_bytes_in,
|
|
||||||
network_bytes_out = EXCLUDED.network_bytes_out,
|
|
||||||
network_packets_in = EXCLUDED.network_packets_in,
|
|
||||||
network_packets_out = EXCLUDED.network_packets_out,
|
|
||||||
start_time = EXCLUDED.start_time
|
|
||||||
`
|
|
||||||
|
|
||||||
_, err := s.db.ExecContext(ctx, query,
|
|
||||||
nodeID, timestamp, container.ContainerID, container.Name, container.State,
|
|
||||||
container.CPU, container.Memory, container.Network.BytesIn, container.Network.BytesOut,
|
|
||||||
container.Network.PacketsIn, container.Network.PacketsOut, container.StartTime,
|
|
||||||
)
|
|
||||||
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *PostgreSQLMetricsStorage) storeInstanceMetrics(ctx context.Context, serviceID string, timestamp time.Time, instance InstanceMetrics) error {
|
func (s *PostgreSQLMetricsStorage) storeInstanceMetrics(ctx context.Context, serviceID string, timestamp time.Time, instance InstanceMetrics) error {
|
||||||
query := `
|
return s.q.UpsertInstanceMetrics(ctx, sqlcdb.UpsertInstanceMetricsParams{
|
||||||
INSERT INTO instance_metrics (
|
ServiceID: serviceID,
|
||||||
service_id, timestamp, instance_id, node_id, status, cpu, memory,
|
Timestamp: timestamp,
|
||||||
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
InstanceID: instance.InstanceID,
|
||||||
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
NodeID: strN(instance.NodeID),
|
||||||
start_time, last_seen, health_status, health_last_check, health_check_count, health_failure_count
|
Status: strN(instance.Status),
|
||||||
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21)
|
Cpu: numStr(instance.CPU),
|
||||||
ON CONFLICT (service_id, timestamp, instance_id) DO UPDATE SET
|
Memory: int64N(instance.Memory),
|
||||||
node_id = EXCLUDED.node_id,
|
NetworkBytesIn: int64N(instance.Network.BytesIn),
|
||||||
status = EXCLUDED.status,
|
NetworkBytesOut: int64N(instance.Network.BytesOut),
|
||||||
cpu = EXCLUDED.cpu,
|
NetworkPacketsIn: int64N(instance.Network.PacketsIn),
|
||||||
memory = EXCLUDED.memory,
|
NetworkPacketsOut: int64N(instance.Network.PacketsOut),
|
||||||
network_bytes_in = EXCLUDED.network_bytes_in,
|
NetworkConnectionsIn: int32N(instance.Network.ConnectionsIn),
|
||||||
network_bytes_out = EXCLUDED.network_bytes_out,
|
NetworkConnectionsOut: int32N(instance.Network.ConnectionsOut),
|
||||||
network_packets_in = EXCLUDED.network_packets_in,
|
NetworkErrorsIn: int64N(instance.Network.ErrorsIn),
|
||||||
network_packets_out = EXCLUDED.network_packets_out,
|
NetworkErrorsOut: int64N(instance.Network.ErrorsOut),
|
||||||
network_connections_in = EXCLUDED.network_connections_in,
|
StartTime: timeN(instance.StartTime),
|
||||||
network_connections_out = EXCLUDED.network_connections_out,
|
LastSeen: timeN(instance.LastSeen),
|
||||||
network_errors_in = EXCLUDED.network_errors_in,
|
HealthStatus: strN(instance.Health.Status),
|
||||||
network_errors_out = EXCLUDED.network_errors_out,
|
HealthLastCheck: timeN(instance.Health.LastCheck),
|
||||||
start_time = EXCLUDED.start_time,
|
HealthCheckCount: sql.NullInt32{Int32: int32(instance.Health.CheckCount), Valid: true},
|
||||||
last_seen = EXCLUDED.last_seen,
|
HealthFailureCount: sql.NullInt32{Int32: int32(instance.Health.FailureCount), Valid: true},
|
||||||
health_status = EXCLUDED.health_status,
|
})
|
||||||
health_last_check = EXCLUDED.health_last_check,
|
|
||||||
health_check_count = EXCLUDED.health_check_count,
|
|
||||||
health_failure_count = EXCLUDED.health_failure_count
|
|
||||||
`
|
|
||||||
|
|
||||||
_, err := s.db.ExecContext(ctx, query,
|
|
||||||
serviceID, timestamp, instance.InstanceID, instance.NodeID, instance.Status,
|
|
||||||
instance.CPU, instance.Memory, instance.Network.BytesIn, instance.Network.BytesOut,
|
|
||||||
instance.Network.PacketsIn, instance.Network.PacketsOut, instance.Network.ConnectionsIn,
|
|
||||||
instance.Network.ConnectionsOut, instance.Network.ErrorsIn, instance.Network.ErrorsOut,
|
|
||||||
instance.StartTime, instance.LastSeen, instance.Health.Status, instance.Health.LastCheck,
|
|
||||||
instance.Health.CheckCount, instance.Health.FailureCount,
|
|
||||||
)
|
|
||||||
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *PostgreSQLMetricsStorage) getContainerMetrics(ctx context.Context, nodeID string, timestamp time.Time) ([]ContainerMetrics, error) {
|
|
||||||
query := `
|
|
||||||
SELECT container_id, name, state, cpu, memory,
|
|
||||||
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out, start_time
|
|
||||||
FROM container_metrics
|
|
||||||
WHERE node_id = $1 AND timestamp = $2
|
|
||||||
`
|
|
||||||
|
|
||||||
rows, err := s.db.QueryContext(ctx, query, nodeID, timestamp)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var containers []ContainerMetrics
|
|
||||||
for rows.Next() {
|
|
||||||
var c ContainerMetrics
|
|
||||||
err := rows.Scan(
|
|
||||||
&c.ContainerID, &c.Name, &c.State, &c.CPU, &c.Memory,
|
|
||||||
&c.Network.BytesIn, &c.Network.BytesOut, &c.Network.PacketsIn, &c.Network.PacketsOut, &c.StartTime,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
containers = append(containers, c)
|
|
||||||
}
|
|
||||||
|
|
||||||
return containers, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *PostgreSQLMetricsStorage) getInstanceMetrics(ctx context.Context, serviceID string, timestamp time.Time) ([]InstanceMetrics, error) {
|
func (s *PostgreSQLMetricsStorage) getInstanceMetrics(ctx context.Context, serviceID string, timestamp time.Time) ([]InstanceMetrics, error) {
|
||||||
query := `
|
rows, err := s.q.ListInstanceMetrics(ctx, sqlcdb.ListInstanceMetricsParams{
|
||||||
SELECT instance_id, node_id, status, cpu, memory,
|
ServiceID: serviceID,
|
||||||
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
Timestamp: timestamp,
|
||||||
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
})
|
||||||
start_time, last_seen, health_status, health_last_check, health_check_count, health_failure_count
|
|
||||||
FROM instance_metrics
|
|
||||||
WHERE service_id = $1 AND timestamp = $2
|
|
||||||
`
|
|
||||||
|
|
||||||
rows, err := s.db.QueryContext(ctx, query, serviceID, timestamp)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
var instances []InstanceMetrics
|
var instances []InstanceMetrics
|
||||||
for rows.Next() {
|
for _, r := range rows {
|
||||||
var i InstanceMetrics
|
instances = append(instances, InstanceMetrics{
|
||||||
err := rows.Scan(
|
InstanceID: r.InstanceID,
|
||||||
&i.InstanceID, &i.NodeID, &i.Status, &i.CPU, &i.Memory,
|
NodeID: r.NodeID.String,
|
||||||
&i.Network.BytesIn, &i.Network.BytesOut, &i.Network.PacketsIn, &i.Network.PacketsOut,
|
Status: r.Status.String,
|
||||||
&i.Network.ConnectionsIn, &i.Network.ConnectionsOut, &i.Network.ErrorsIn, &i.Network.ErrorsOut,
|
CPU: f64v(r.Cpu),
|
||||||
&i.StartTime, &i.LastSeen, &i.Health.Status, &i.Health.LastCheck, &i.Health.CheckCount, &i.Health.FailureCount,
|
Memory: i64v(r.Memory),
|
||||||
)
|
Network: NetworkMetrics{
|
||||||
if err != nil {
|
BytesIn: i64v(r.NetworkBytesIn),
|
||||||
return nil, err
|
BytesOut: i64v(r.NetworkBytesOut),
|
||||||
}
|
PacketsIn: i64v(r.NetworkPacketsIn),
|
||||||
instances = append(instances, i)
|
PacketsOut: i64v(r.NetworkPacketsOut),
|
||||||
|
ConnectionsIn: i32v(r.NetworkConnectionsIn),
|
||||||
|
ConnectionsOut: i32v(r.NetworkConnectionsOut),
|
||||||
|
ErrorsIn: i64v(r.NetworkErrorsIn),
|
||||||
|
ErrorsOut: i64v(r.NetworkErrorsOut),
|
||||||
|
},
|
||||||
|
StartTime: r.StartTime.Time,
|
||||||
|
LastSeen: r.LastSeen.Time,
|
||||||
|
Health: HealthMetrics{
|
||||||
|
Status: r.HealthStatus.String,
|
||||||
|
LastCheck: r.HealthLastCheck.Time,
|
||||||
|
CheckCount: int(r.HealthCheckCount.Int32),
|
||||||
|
FailureCount: int(r.HealthFailureCount.Int32),
|
||||||
|
},
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
return instances, nil
|
return instances, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -544,7 +469,9 @@ func (s *InMemoryMetricsStorage) GetServiceMetrics(ctx context.Context, serviceI
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *InMemoryMetricsStorage) GetAggregatedMetrics(ctx context.Context, query MetricsQuery) (*AggregatedMetrics, error) {
|
func (s *InMemoryMetricsStorage) GetAggregatedMetrics(ctx context.Context, query MetricsQuery) (*AggregatedMetrics, error) {
|
||||||
// Simplified implementation
|
s.mu.RLock()
|
||||||
|
defer s.mu.RUnlock()
|
||||||
|
|
||||||
return &AggregatedMetrics{
|
return &AggregatedMetrics{
|
||||||
Query: query,
|
Query: query,
|
||||||
TimeSeries: []TimeSeriesPoint{},
|
TimeSeries: []TimeSeriesPoint{},
|
||||||
|
|||||||
@@ -0,0 +1,130 @@
|
|||||||
|
package metrics
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"database/sql"
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Live round-trip against a real Postgres — set METRICS_TEST_DSN (or
|
||||||
|
// DATABASE_URL) to enable. Skipped by default so unit runs stay hermetic.
|
||||||
|
func testDSN(t *testing.T) *sql.DB {
|
||||||
|
t.Helper()
|
||||||
|
dsn := os.Getenv("METRICS_TEST_DSN")
|
||||||
|
if dsn == "" {
|
||||||
|
dsn = os.Getenv("DATABASE_URL")
|
||||||
|
}
|
||||||
|
if dsn == "" {
|
||||||
|
t.Skip("METRICS_TEST_DSN not set")
|
||||||
|
}
|
||||||
|
db, err := sql.Open("postgres", dsn)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("open: %v", err)
|
||||||
|
}
|
||||||
|
if err := db.PingContext(context.Background()); err != nil {
|
||||||
|
t.Fatalf("ping: %v", err)
|
||||||
|
}
|
||||||
|
t.Cleanup(func() { db.Close() })
|
||||||
|
return db
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestPostgresStorageRoundTrip(t *testing.T) {
|
||||||
|
db := testDSN(t)
|
||||||
|
s := NewPostgreSQLMetricsStorage(db)
|
||||||
|
ctx := context.Background()
|
||||||
|
|
||||||
|
ts := time.Now().UTC().Truncate(time.Second)
|
||||||
|
nodeID := "test-node-" + ts.Format("150405")
|
||||||
|
svcID := "test-svc-" + ts.Format("150405")
|
||||||
|
|
||||||
|
// Clean slate + teardown
|
||||||
|
db.ExecContext(ctx, `DELETE FROM instance_metrics WHERE service_id = $1`, svcID)
|
||||||
|
db.ExecContext(ctx, `DELETE FROM service_metrics WHERE service_id = $1`, svcID)
|
||||||
|
db.ExecContext(ctx, `DELETE FROM node_metrics WHERE node_id = $1`, nodeID)
|
||||||
|
t.Cleanup(func() {
|
||||||
|
db.ExecContext(ctx, `DELETE FROM instance_metrics WHERE service_id = $1`, svcID)
|
||||||
|
db.ExecContext(ctx, `DELETE FROM service_metrics WHERE service_id = $1`, svcID)
|
||||||
|
db.ExecContext(ctx, `DELETE FROM node_metrics WHERE node_id = $1`, nodeID)
|
||||||
|
})
|
||||||
|
|
||||||
|
node := &NodeMetrics{
|
||||||
|
NodeID: nodeID,
|
||||||
|
Timestamp: ts,
|
||||||
|
CPU: CPUMetrics{UsagePercent: 42.5, UsageCores: 1.7, LoadAverage1: 0.5, LoadAverage5: 0.4, LoadAverage15: 0.3},
|
||||||
|
Memory: MemoryMetrics{Total: 16 << 30, Used: 8 << 30, Available: 8 << 30, UsagePercent: 50.0},
|
||||||
|
Storage: StorageMetrics{Total: 100 << 30, Used: 40 << 30, Available: 60 << 30, UsagePercent: 40.0},
|
||||||
|
Network: NetworkMetrics{BytesIn: 1000, BytesOut: 2000, PacketsIn: 10, PacketsOut: 20, ConnectionsIn: 3, ConnectionsOut: 4, ErrorsIn: 0, ErrorsOut: 1},
|
||||||
|
System: SystemMetrics{Uptime: 3600 * time.Second, Processes: 120, OS: "linux", Kernel: "6.8", Architecture: "amd64"},
|
||||||
|
}
|
||||||
|
if err := s.StoreNodeMetrics(ctx, node); err != nil {
|
||||||
|
t.Fatalf("store node: %v", err)
|
||||||
|
}
|
||||||
|
// Upsert idempotent on the same key.
|
||||||
|
if err := s.StoreNodeMetrics(ctx, node); err != nil {
|
||||||
|
t.Fatalf("store node (repeat): %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
svc := &ServiceMetrics{
|
||||||
|
ServiceID: svcID,
|
||||||
|
ServiceName: "svc",
|
||||||
|
ProjectID: "proj",
|
||||||
|
Timestamp: ts,
|
||||||
|
Requests: RequestMetrics{Total: 100, Success: 95, Errors: 5, AvgLatency: 12.5, P95Latency: 80, P99Latency: 120, Throughput: 3.3},
|
||||||
|
Errors: ErrorMetrics{Total: 5, Rate: 0.05},
|
||||||
|
Performance: PerformanceMetrics{ResponseTime: 12.5, Throughput: 3.3, Concurrency: 2, Saturation: 0.4, Utilization: 0.55},
|
||||||
|
Resources: ResourceMetrics{CPUUsage: 42.5, MemoryUsage: 512 << 20, StorageUsage: 1 << 30, NetworkUsage: 3000, ResourceScore: 0.7},
|
||||||
|
Instances: []InstanceMetrics{{
|
||||||
|
InstanceID: "inst-1",
|
||||||
|
NodeID: nodeID,
|
||||||
|
Status: "running",
|
||||||
|
CPU: 42.5,
|
||||||
|
Memory: 512 << 20,
|
||||||
|
Network: NetworkMetrics{BytesIn: 1000, BytesOut: 2000},
|
||||||
|
StartTime: ts.Add(-time.Hour),
|
||||||
|
LastSeen: ts,
|
||||||
|
Health: HealthMetrics{Status: "healthy", LastCheck: ts, CheckCount: 10, FailureCount: 0},
|
||||||
|
}},
|
||||||
|
}
|
||||||
|
if err := s.StoreServiceMetrics(ctx, svc); err != nil {
|
||||||
|
t.Fatalf("store service: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
from, to := ts.Add(-time.Minute), ts.Add(time.Minute)
|
||||||
|
gotNodes, err := s.GetNodeMetrics(ctx, nodeID, from, to)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("get nodes: %v", err)
|
||||||
|
}
|
||||||
|
if len(gotNodes) != 1 {
|
||||||
|
t.Fatalf("want 1 node row, got %d", len(gotNodes))
|
||||||
|
}
|
||||||
|
if got := gotNodes[0]; got.CPU.UsagePercent != 42.5 || got.System.Processes != 120 || got.System.Uptime != time.Hour {
|
||||||
|
t.Errorf("node round-trip mismatch: %+v", got)
|
||||||
|
}
|
||||||
|
|
||||||
|
gotSvcs, err := s.GetServiceMetrics(ctx, svcID, from, to)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("get services: %v", err)
|
||||||
|
}
|
||||||
|
if len(gotSvcs) != 1 {
|
||||||
|
t.Fatalf("want 1 service row, got %d", len(gotSvcs))
|
||||||
|
}
|
||||||
|
if gotSvcs[0].Requests.P99Latency != 120 {
|
||||||
|
t.Errorf("p99 mismatch: %v", gotSvcs[0].Requests.P99Latency)
|
||||||
|
}
|
||||||
|
if len(gotSvcs[0].Instances) != 1 || gotSvcs[0].Instances[0].Health.Status != "healthy" {
|
||||||
|
t.Errorf("instance join mismatch: %+v", gotSvcs[0].Instances)
|
||||||
|
}
|
||||||
|
|
||||||
|
agg, err := s.GetAggregatedMetrics(ctx, MetricsQuery{Type: "node", ID: nodeID, From: from, To: to, Interval: 5 * time.Minute})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("aggregate: %v", err)
|
||||||
|
}
|
||||||
|
if len(agg.TimeSeries) != 1 || agg.TimeSeries[0].Values["cpu_usage"] != 42.5 {
|
||||||
|
t.Errorf("aggregate mismatch: %+v", agg.TimeSeries)
|
||||||
|
}
|
||||||
|
if agg.Summary["cpu_usage"].Avg != 42.5 {
|
||||||
|
t.Errorf("summary mismatch: %+v", agg.Summary)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,37 @@
|
|||||||
|
-- name: GetUserByEmailForAuth :one
|
||||||
|
SELECT id, email, password_hash, name, COALESCE(avatar_url, '') AS avatar_url, is_admin, created_at
|
||||||
|
FROM users WHERE email = $1;
|
||||||
|
|
||||||
|
-- name: CountUsers :one
|
||||||
|
SELECT COUNT(*) FROM users;
|
||||||
|
|
||||||
|
-- name: CountUsersByEmail :one
|
||||||
|
SELECT COUNT(*) FROM users WHERE email = $1;
|
||||||
|
|
||||||
|
-- name: UpsertLocalUser :one
|
||||||
|
INSERT INTO users (email, password_hash, name, avatar_url, is_admin)
|
||||||
|
VALUES ($1, $2, $3, NULLIF($4::varchar, ''), $5)
|
||||||
|
ON CONFLICT (email) DO UPDATE
|
||||||
|
SET name = EXCLUDED.name,
|
||||||
|
avatar_url = COALESCE(EXCLUDED.avatar_url, users.avatar_url),
|
||||||
|
updated_at = NOW()
|
||||||
|
RETURNING id, email, name, COALESCE(avatar_url, '') AS avatar_url, is_admin, created_at;
|
||||||
|
|
||||||
|
-- name: InsertUserAdmin :one
|
||||||
|
INSERT INTO users (email, password_hash, name, is_admin)
|
||||||
|
VALUES ($1, $2, $3, $4)
|
||||||
|
RETURNING id, email, name, COALESCE(avatar_url, '') AS avatar_url, is_admin, created_at;
|
||||||
|
|
||||||
|
-- name: InsertUser :one
|
||||||
|
INSERT INTO users (email, password_hash, name)
|
||||||
|
VALUES ($1, $2, $3)
|
||||||
|
RETURNING id, email, name, COALESCE(avatar_url, '') AS avatar_url, is_admin, created_at;
|
||||||
|
|
||||||
|
-- name: GetUserByID :one
|
||||||
|
SELECT id, email, name, COALESCE(avatar_url, '') AS avatar_url, is_admin, created_at
|
||||||
|
FROM users WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: UpdateUserProfile :exec
|
||||||
|
UPDATE users
|
||||||
|
SET name = COALESCE($1::varchar, name), avatar_url = COALESCE($2::varchar, avatar_url)
|
||||||
|
WHERE id = $3;
|
||||||
@@ -0,0 +1,58 @@
|
|||||||
|
-- name: GetCronJobProjectID :one
|
||||||
|
SELECT project_id FROM cron_jobs WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetCronOwnerViaService :one
|
||||||
|
SELECT p.owner_id FROM projects p
|
||||||
|
JOIN services s ON s.project_id = p.id
|
||||||
|
WHERE s.id = $1;
|
||||||
|
|
||||||
|
-- name: InsertCronJob :exec
|
||||||
|
INSERT INTO cron_jobs (id, project_id, service_id, name, schedule, command, timezone, enabled, next_run_at, retention, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12);
|
||||||
|
|
||||||
|
-- name: GetCronJob :one
|
||||||
|
SELECT id, project_id, service_id, name, schedule, command, timezone,
|
||||||
|
enabled, last_run_at, next_run_at, last_status, last_output,
|
||||||
|
retention, created_at, updated_at
|
||||||
|
FROM cron_jobs WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetCronJobOwner :one
|
||||||
|
SELECT p.owner_id FROM cron_jobs cj
|
||||||
|
JOIN projects p ON cj.project_id = p.id
|
||||||
|
WHERE cj.id = $1;
|
||||||
|
|
||||||
|
-- name: DeleteCronJob :exec
|
||||||
|
DELETE FROM cron_jobs WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: ListCronExecutions :many
|
||||||
|
SELECT id, cron_job_id, started_at, finished_at, status, output, error
|
||||||
|
FROM cron_executions
|
||||||
|
WHERE cron_job_id = $1
|
||||||
|
ORDER BY started_at DESC
|
||||||
|
LIMIT 100;
|
||||||
|
|
||||||
|
-- name: GetCronJobForTrigger :one
|
||||||
|
SELECT cj.service_id, cj.command, cj.schedule, cj.timezone, cj.retention, p.owner_id
|
||||||
|
FROM cron_jobs cj
|
||||||
|
JOIN projects p ON cj.project_id = p.id
|
||||||
|
WHERE cj.id = $1;
|
||||||
|
|
||||||
|
-- name: InsertCronExecution :exec
|
||||||
|
INSERT INTO cron_executions (id, cron_job_id, started_at, status)
|
||||||
|
VALUES ($1, $2, $3, $4);
|
||||||
|
|
||||||
|
-- name: ListDueCronJobs :many
|
||||||
|
SELECT id, project_id, service_id, name, schedule, command, timezone, enabled, retention
|
||||||
|
FROM cron_jobs
|
||||||
|
WHERE enabled = TRUE AND next_run_at IS NOT NULL AND next_run_at <= NOW();
|
||||||
|
|
||||||
|
-- name: FinishCronExecution :exec
|
||||||
|
UPDATE cron_executions SET finished_at = $1, status = $2, output = $3, error = $4 WHERE id = $5;
|
||||||
|
|
||||||
|
-- name: UpdateCronJobRun :exec
|
||||||
|
UPDATE cron_jobs SET last_run_at = $1, last_status = $2, last_output = $3, next_run_at = $4 WHERE id = $5;
|
||||||
|
|
||||||
|
-- name: TrimCronExecutions :exec
|
||||||
|
DELETE FROM cron_executions WHERE cron_executions.cron_job_id = $1 AND cron_executions.id NOT IN (
|
||||||
|
SELECT ce.id FROM cron_executions ce WHERE ce.cron_job_id = $1
|
||||||
|
ORDER BY ce.started_at DESC LIMIT $2);
|
||||||
@@ -0,0 +1,105 @@
|
|||||||
|
-- name: CountDeploymentsForService :one
|
||||||
|
SELECT COUNT(*) FROM deployments WHERE service_id = $1;
|
||||||
|
|
||||||
|
-- name: ListDeploymentsForService :many
|
||||||
|
SELECT id, service_id, commit_hash, status, image_name, image_tag,
|
||||||
|
build_log, runtime_log, error, started_at, completed_at, created_at, updated_at
|
||||||
|
FROM deployments
|
||||||
|
WHERE service_id = $1
|
||||||
|
ORDER BY created_at DESC
|
||||||
|
LIMIT $2 OFFSET $3;
|
||||||
|
|
||||||
|
-- name: CountAccessibleDeployments :one
|
||||||
|
SELECT COUNT(*)
|
||||||
|
FROM deployments d
|
||||||
|
JOIN services s ON s.id = d.service_id
|
||||||
|
JOIN projects p ON p.id = s.project_id
|
||||||
|
WHERE p.is_approved OR p.owner_id = $1 OR $2::bool
|
||||||
|
OR EXISTS (SELECT 1 FROM project_members pm WHERE pm.project_id = p.id AND pm.user_id = $1);
|
||||||
|
|
||||||
|
-- name: ListRecentAccessibleDeployments :many
|
||||||
|
SELECT d.id, d.service_id, s.name AS service_name, p.name AS project_name, d.status,
|
||||||
|
COALESCE(d.image_name, '') AS image_name, d.started_at, d.completed_at, d.created_at
|
||||||
|
FROM deployments d
|
||||||
|
JOIN services s ON s.id = d.service_id
|
||||||
|
JOIN projects p ON p.id = s.project_id
|
||||||
|
WHERE p.is_approved OR p.owner_id = $1 OR $2::bool
|
||||||
|
OR EXISTS (SELECT 1 FROM project_members pm WHERE pm.project_id = p.id AND pm.user_id = $1)
|
||||||
|
ORDER BY d.created_at DESC
|
||||||
|
LIMIT $3 OFFSET $4;
|
||||||
|
|
||||||
|
-- name: GetServiceForDeployWithOwner :one
|
||||||
|
SELECT s.id, s.project_id, s.name, s.type, s.status, s.image, s.command,
|
||||||
|
s.environment, s.git_repo, s.git_branch, s.build_path, s.cpu, s.memory,
|
||||||
|
COALESCE(s.replicas, 1) AS replicas, COALESCE(s.port, 0) AS port,
|
||||||
|
COALESCE(s.domain, '') AS domain, COALESCE(s.healthcheck_path, '') AS healthcheck_path,
|
||||||
|
COALESCE(s.restart_policy, 'unless-stopped') AS restart_policy,
|
||||||
|
COALESCE(s.builder, 'auto') AS builder, COALESCE(s.cpu_reserve, '') AS cpu_reserve,
|
||||||
|
COALESCE(s.memory_reserve, '') AS memory_reserve, COALESCE(s.static_build_cmd, '') AS static_build_cmd,
|
||||||
|
COALESCE(s.static_dir, '') AS static_dir,
|
||||||
|
s.created_at, s.updated_at, p.owner_id
|
||||||
|
FROM services s
|
||||||
|
JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE s.id = $1;
|
||||||
|
|
||||||
|
-- name: InsertDeployment :exec
|
||||||
|
INSERT INTO deployments
|
||||||
|
(id, service_id, version, commit_hash, status, image_name, image_tag, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9);
|
||||||
|
|
||||||
|
-- name: FailDeployment :exec
|
||||||
|
UPDATE deployments
|
||||||
|
SET status = 'failed', error = $1, completed_at = $2, updated_at = $2
|
||||||
|
WHERE id = $3;
|
||||||
|
|
||||||
|
-- name: CompleteDeployment :exec
|
||||||
|
UPDATE deployments
|
||||||
|
SET status = $1, error = $2, completed_at = $3, updated_at = $3
|
||||||
|
WHERE id = $4;
|
||||||
|
|
||||||
|
-- name: SetDeploymentStatus :exec
|
||||||
|
UPDATE deployments SET status = $1, updated_at = $2 WHERE id = $3;
|
||||||
|
|
||||||
|
-- name: SetServiceStatus :exec
|
||||||
|
UPDATE services SET status = $1, updated_at = $2 WHERE id = $3;
|
||||||
|
|
||||||
|
-- name: SyncDeploymentProgress :exec
|
||||||
|
UPDATE deployments
|
||||||
|
SET status = $1,
|
||||||
|
image_name = $2,
|
||||||
|
image_tag = $3,
|
||||||
|
build_log = $4,
|
||||||
|
runtime_log = $5,
|
||||||
|
error = $6,
|
||||||
|
started_at = $7,
|
||||||
|
completed_at = $8,
|
||||||
|
updated_at = $9
|
||||||
|
WHERE id = $10;
|
||||||
|
|
||||||
|
-- name: GetDeploymentAccess :one
|
||||||
|
SELECT d.status, d.service_id, p.owner_id
|
||||||
|
FROM deployments d
|
||||||
|
JOIN services s ON d.service_id = s.id
|
||||||
|
JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE d.id = $1;
|
||||||
|
|
||||||
|
-- name: GetDeploymentWithProject :one
|
||||||
|
SELECT d.id, d.service_id, d.commit_hash, d.status, d.image_name, d.image_tag,
|
||||||
|
d.build_log, d.runtime_log, d.error, d.started_at, d.completed_at,
|
||||||
|
d.created_at, d.updated_at, p.id AS project_id
|
||||||
|
FROM deployments d
|
||||||
|
JOIN services s ON d.service_id = s.id
|
||||||
|
JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE d.id = $1;
|
||||||
|
|
||||||
|
-- name: GetDeploymentForRollback :one
|
||||||
|
SELECT d.id, d.service_id, d.commit_hash, d.status, d.image_name, d.image_tag,
|
||||||
|
d.build_log, d.runtime_log, d.error, d.started_at, d.completed_at,
|
||||||
|
d.created_at, d.updated_at, p.owner_id
|
||||||
|
FROM deployments d
|
||||||
|
JOIN services s ON d.service_id = s.id
|
||||||
|
JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE d.id = $1;
|
||||||
|
|
||||||
|
-- name: GetServicePublishedPort :one
|
||||||
|
SELECT published_port FROM services WHERE id = $1;
|
||||||
@@ -0,0 +1,46 @@
|
|||||||
|
-- name: ListServiceDomains :many
|
||||||
|
SELECT id, service_id, domain, is_default, cert_type, cert_status, last_checked_at, created_at
|
||||||
|
FROM service_domains
|
||||||
|
WHERE service_id = $1
|
||||||
|
ORDER BY is_default DESC, created_at ASC;
|
||||||
|
|
||||||
|
-- name: CountServiceDomains :one
|
||||||
|
SELECT COUNT(*) FROM service_domains WHERE service_id = $1;
|
||||||
|
|
||||||
|
-- name: ClearServiceDomainDefault :exec
|
||||||
|
UPDATE service_domains SET is_default = false WHERE service_id = $1;
|
||||||
|
|
||||||
|
-- name: CreateServiceDomain :one
|
||||||
|
INSERT INTO service_domains (service_id, domain, is_default)
|
||||||
|
VALUES ($1, $2, $3)
|
||||||
|
RETURNING id, service_id, domain, is_default, cert_type, cert_status, created_at;
|
||||||
|
|
||||||
|
-- name: DeleteServiceDomain :execrows
|
||||||
|
DELETE FROM service_domains WHERE id = $1 AND service_id = $2;
|
||||||
|
|
||||||
|
-- name: SetServiceDomainDefault :execrows
|
||||||
|
UPDATE service_domains SET is_default = (id = $1) WHERE service_id = $2;
|
||||||
|
|
||||||
|
-- name: TouchServiceDomainChecked :exec
|
||||||
|
UPDATE service_domains SET last_checked_at = NOW() WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetServiceDomainDefault :one
|
||||||
|
SELECT domain FROM service_domains WHERE service_id = $1 AND is_default;
|
||||||
|
|
||||||
|
-- name: GetServiceDomainFirst :one
|
||||||
|
SELECT domain FROM service_domains WHERE service_id = $1 ORDER BY created_at ASC LIMIT 1;
|
||||||
|
|
||||||
|
-- name: SetServiceLegacyDomain :exec
|
||||||
|
UPDATE services SET domain = $1 WHERE id = $2;
|
||||||
|
|
||||||
|
-- name: GetServicePlacementBrief :one
|
||||||
|
SELECT name, node_id, COALESCE(spread, false) AS spread FROM services WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetServiceAccess :one
|
||||||
|
SELECT maintenance_mode, basic_auth_users FROM services WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: ServiceWriteAccess :one
|
||||||
|
SELECT EXISTS(
|
||||||
|
SELECT 1 FROM services s JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE s.id = $1 AND (p.owner_id = $2 OR $3::bool)
|
||||||
|
) AS allowed;
|
||||||
@@ -0,0 +1,129 @@
|
|||||||
|
-- name: ListGitProviders :many
|
||||||
|
SELECT id, name, display_name, api_url, webhook_url, user_id, created_at, updated_at
|
||||||
|
FROM git_providers
|
||||||
|
WHERE user_id = $1
|
||||||
|
ORDER BY created_at DESC;
|
||||||
|
|
||||||
|
-- name: UpsertGitHubAppProvider :one
|
||||||
|
INSERT INTO git_providers (id, name, display_name, api_url, webhook_url, access_token, user_id, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, NOW(), NOW())
|
||||||
|
ON CONFLICT (name, user_id)
|
||||||
|
DO UPDATE SET
|
||||||
|
display_name = EXCLUDED.display_name,
|
||||||
|
api_url = EXCLUDED.api_url,
|
||||||
|
webhook_url = EXCLUDED.webhook_url,
|
||||||
|
access_token = EXCLUDED.access_token,
|
||||||
|
updated_at = NOW()
|
||||||
|
RETURNING id, created_at, updated_at;
|
||||||
|
|
||||||
|
-- name: InsertGitProvider :exec
|
||||||
|
INSERT INTO git_providers (id, name, display_name, api_url, webhook_url, access_token, user_id, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, NOW(), NOW());
|
||||||
|
|
||||||
|
-- name: GetGitProviderTimestamps :one
|
||||||
|
SELECT created_at, updated_at FROM git_providers WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetGitProviderForRepos :one
|
||||||
|
SELECT name, display_name, access_token, api_url
|
||||||
|
FROM git_providers
|
||||||
|
WHERE id = $1 AND user_id = $2;
|
||||||
|
|
||||||
|
-- name: GetGitProviderName :one
|
||||||
|
SELECT name FROM git_providers WHERE id = $1 AND user_id = $2;
|
||||||
|
|
||||||
|
-- name: GetGitProviderForFetch :one
|
||||||
|
SELECT name, api_url, access_token
|
||||||
|
FROM git_providers
|
||||||
|
WHERE id = $1 AND user_id = $2;
|
||||||
|
|
||||||
|
-- name: DeleteGitProvider :execrows
|
||||||
|
DELETE FROM git_providers WHERE id = $1 AND user_id = $2;
|
||||||
|
|
||||||
|
-- name: ListGitRepositories :many
|
||||||
|
SELECT id, provider_id, name, full_name, COALESCE(description, '') AS description,
|
||||||
|
clone_url, COALESCE(webhook_url, '') AS webhook_url,
|
||||||
|
default_branch, is_private, user_id, created_at, updated_at
|
||||||
|
FROM git_repositories
|
||||||
|
WHERE provider_id = $1 AND user_id = $2
|
||||||
|
AND ($3::text IS NULL OR full_name ILIKE ('%' || $3::text || '%') OR name ILIKE ('%' || $3::text || '%'))
|
||||||
|
ORDER BY updated_at DESC
|
||||||
|
LIMIT $4 OFFSET $5;
|
||||||
|
|
||||||
|
-- name: CountGitRepositories :one
|
||||||
|
SELECT COUNT(*)
|
||||||
|
FROM git_repositories
|
||||||
|
WHERE provider_id = $1 AND user_id = $2
|
||||||
|
AND ($3::text IS NULL OR full_name ILIKE ('%' || $3::text || '%') OR name ILIKE ('%' || $3::text || '%'));
|
||||||
|
|
||||||
|
-- name: GetGitRepositoryByFullName :one
|
||||||
|
SELECT id, provider_id, name, full_name, COALESCE(description, '') AS description,
|
||||||
|
clone_url, COALESCE(webhook_url, '') AS webhook_url,
|
||||||
|
default_branch, is_private, user_id, created_at, updated_at
|
||||||
|
FROM git_repositories
|
||||||
|
WHERE provider_id = $1 AND full_name = $2 AND user_id = $3;
|
||||||
|
|
||||||
|
-- name: InsertGitRepository :one
|
||||||
|
INSERT INTO git_repositories (id, provider_id, name, full_name, description, clone_url, default_branch, is_private, user_id, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, NOW(), NOW())
|
||||||
|
RETURNING created_at, updated_at;
|
||||||
|
|
||||||
|
-- name: UpsertGitRepository :exec
|
||||||
|
INSERT INTO git_repositories (id, provider_id, name, full_name, description, clone_url, default_branch, is_private, user_id, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, NOW(), NOW())
|
||||||
|
ON CONFLICT (provider_id, full_name)
|
||||||
|
DO UPDATE SET
|
||||||
|
name = EXCLUDED.name,
|
||||||
|
description = EXCLUDED.description,
|
||||||
|
clone_url = EXCLUDED.clone_url,
|
||||||
|
default_branch = EXCLUDED.default_branch,
|
||||||
|
is_private = EXCLUDED.is_private,
|
||||||
|
updated_at = NOW();
|
||||||
|
|
||||||
|
-- name: GetGitRepositoryProviderID :one
|
||||||
|
SELECT provider_id FROM git_repositories WHERE id = $1 AND user_id = $2;
|
||||||
|
|
||||||
|
-- name: UpsertGitWebhook :one
|
||||||
|
INSERT INTO git_webhooks (id, repo_id, provider_id, events, webhook_secret, active, branch_filter, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, TRUE, $6, NOW(), NOW())
|
||||||
|
ON CONFLICT (repo_id, provider_id)
|
||||||
|
DO UPDATE SET
|
||||||
|
events = EXCLUDED.events,
|
||||||
|
webhook_secret = EXCLUDED.webhook_secret,
|
||||||
|
active = TRUE,
|
||||||
|
branch_filter = EXCLUDED.branch_filter,
|
||||||
|
updated_at = NOW()
|
||||||
|
RETURNING id, active, created_at, updated_at;
|
||||||
|
|
||||||
|
-- name: ListConnectedRepositories :many
|
||||||
|
SELECT r.id, r.provider_id, r.name, r.full_name, COALESCE(r.description, '') AS description, r.clone_url,
|
||||||
|
r.default_branch, r.is_private, r.user_id, r.created_at, r.updated_at,
|
||||||
|
p.name AS provider_name, p.display_name
|
||||||
|
FROM git_repositories r
|
||||||
|
JOIN git_providers p ON r.provider_id = p.id
|
||||||
|
WHERE r.user_id = $1
|
||||||
|
ORDER BY r.updated_at DESC
|
||||||
|
LIMIT $2 OFFSET $3;
|
||||||
|
|
||||||
|
-- name: CountUserRepositories :one
|
||||||
|
SELECT COUNT(*) FROM git_repositories WHERE user_id = $1;
|
||||||
|
|
||||||
|
-- name: GetGitWebhookForPush :one
|
||||||
|
SELECT webhook_secret, repo_id, provider_id, COALESCE(branch_filter, '') AS branch_filter, active
|
||||||
|
FROM git_webhooks WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetGitProviderNameByID :one
|
||||||
|
SELECT name FROM git_providers WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetGitRepoForPush :one
|
||||||
|
SELECT clone_url, full_name, user_id FROM git_repositories WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: ListServicesForPush :many
|
||||||
|
SELECT s.id, s.project_id, s.name,
|
||||||
|
COALESCE(s.type, s.service_type, '') AS type, COALESCE(s.status, '') AS status,
|
||||||
|
COALESCE(s.image, s.image_name, '') AS image, COALESCE(s.command, s.start_command, '') AS command,
|
||||||
|
COALESCE(s.environment, '') AS environment, COALESCE(s.git_repo, s.source_url, '') AS git_repo,
|
||||||
|
COALESCE(s.git_branch, '') AS git_branch, COALESCE(s.build_path, '') AS build_path,
|
||||||
|
COALESCE(s.cpu, '') AS cpu, COALESCE(s.memory, '') AS memory,
|
||||||
|
s.created_at, s.updated_at
|
||||||
|
FROM services s
|
||||||
|
WHERE (s.git_branch = $1 OR $1 = '') AND (s.git_repo = $2 OR s.git_repo = $3);
|
||||||
@@ -0,0 +1,147 @@
|
|||||||
|
-- name: UpsertNodeMetrics :exec
|
||||||
|
INSERT INTO node_metrics (
|
||||||
|
node_id, timestamp, cpu_usage, cpu_cores, load_avg_1, load_avg_5, load_avg_15,
|
||||||
|
memory_total, memory_used, memory_available, memory_usage_percent,
|
||||||
|
storage_total, storage_used, storage_available, storage_usage_percent,
|
||||||
|
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
||||||
|
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
||||||
|
uptime, processes, os, kernel, architecture
|
||||||
|
) VALUES (
|
||||||
|
$1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15,
|
||||||
|
$16, $17, $18, $19, $20, $21, $22, $23, $24, $25, $26, $27, $28
|
||||||
|
)
|
||||||
|
ON CONFLICT (node_id, timestamp) DO UPDATE SET
|
||||||
|
cpu_usage = EXCLUDED.cpu_usage,
|
||||||
|
cpu_cores = EXCLUDED.cpu_cores,
|
||||||
|
load_avg_1 = EXCLUDED.load_avg_1,
|
||||||
|
load_avg_5 = EXCLUDED.load_avg_5,
|
||||||
|
load_avg_15 = EXCLUDED.load_avg_15,
|
||||||
|
memory_total = EXCLUDED.memory_total,
|
||||||
|
memory_used = EXCLUDED.memory_used,
|
||||||
|
memory_available = EXCLUDED.memory_available,
|
||||||
|
memory_usage_percent = EXCLUDED.memory_usage_percent,
|
||||||
|
storage_total = EXCLUDED.storage_total,
|
||||||
|
storage_used = EXCLUDED.storage_used,
|
||||||
|
storage_available = EXCLUDED.storage_available,
|
||||||
|
storage_usage_percent = EXCLUDED.storage_usage_percent,
|
||||||
|
network_bytes_in = EXCLUDED.network_bytes_in,
|
||||||
|
network_bytes_out = EXCLUDED.network_bytes_out,
|
||||||
|
network_packets_in = EXCLUDED.network_packets_in,
|
||||||
|
network_packets_out = EXCLUDED.network_packets_out,
|
||||||
|
network_connections_in = EXCLUDED.network_connections_in,
|
||||||
|
network_connections_out = EXCLUDED.network_connections_out,
|
||||||
|
network_errors_in = EXCLUDED.network_errors_in,
|
||||||
|
network_errors_out = EXCLUDED.network_errors_out,
|
||||||
|
uptime = EXCLUDED.uptime,
|
||||||
|
processes = EXCLUDED.processes,
|
||||||
|
os = EXCLUDED.os,
|
||||||
|
kernel = EXCLUDED.kernel,
|
||||||
|
architecture = EXCLUDED.architecture;
|
||||||
|
|
||||||
|
-- name: UpsertServiceMetrics :exec
|
||||||
|
INSERT INTO service_metrics (
|
||||||
|
service_id, service_name, project_id, timestamp,
|
||||||
|
requests_total, requests_success, requests_errors, requests_avg_latency,
|
||||||
|
requests_p95_latency, requests_p99_latency, requests_throughput,
|
||||||
|
errors_total, errors_rate, performance_response_time, performance_throughput,
|
||||||
|
performance_concurrency, performance_saturation, performance_utilization,
|
||||||
|
resource_cpu_usage, resource_memory_usage, resource_storage_usage,
|
||||||
|
resource_network_usage, resource_score
|
||||||
|
) VALUES (
|
||||||
|
$1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15,
|
||||||
|
$16, $17, $18, $19, $20, $21, $22, $23
|
||||||
|
)
|
||||||
|
ON CONFLICT (service_id, timestamp) DO UPDATE SET
|
||||||
|
requests_total = EXCLUDED.requests_total,
|
||||||
|
requests_success = EXCLUDED.requests_success,
|
||||||
|
requests_errors = EXCLUDED.requests_errors,
|
||||||
|
requests_avg_latency = EXCLUDED.requests_avg_latency,
|
||||||
|
requests_p95_latency = EXCLUDED.requests_p95_latency,
|
||||||
|
requests_p99_latency = EXCLUDED.requests_p99_latency,
|
||||||
|
requests_throughput = EXCLUDED.requests_throughput,
|
||||||
|
errors_total = EXCLUDED.errors_total,
|
||||||
|
errors_rate = EXCLUDED.errors_rate,
|
||||||
|
performance_response_time = EXCLUDED.performance_response_time,
|
||||||
|
performance_throughput = EXCLUDED.performance_throughput,
|
||||||
|
performance_concurrency = EXCLUDED.performance_concurrency,
|
||||||
|
performance_saturation = EXCLUDED.performance_saturation,
|
||||||
|
performance_utilization = EXCLUDED.performance_utilization,
|
||||||
|
resource_cpu_usage = EXCLUDED.resource_cpu_usage,
|
||||||
|
resource_memory_usage = EXCLUDED.resource_memory_usage,
|
||||||
|
resource_storage_usage = EXCLUDED.resource_storage_usage,
|
||||||
|
resource_network_usage = EXCLUDED.resource_network_usage,
|
||||||
|
resource_score = EXCLUDED.resource_score;
|
||||||
|
|
||||||
|
-- name: UpsertInstanceMetrics :exec
|
||||||
|
INSERT INTO instance_metrics (
|
||||||
|
service_id, timestamp, instance_id, node_id, status, cpu, memory,
|
||||||
|
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
||||||
|
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
||||||
|
start_time, last_seen, health_status, health_last_check, health_check_count, health_failure_count
|
||||||
|
) VALUES (
|
||||||
|
$1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15,
|
||||||
|
$16, $17, $18, $19, $20, $21
|
||||||
|
)
|
||||||
|
ON CONFLICT (service_id, timestamp, instance_id) DO UPDATE SET
|
||||||
|
node_id = EXCLUDED.node_id,
|
||||||
|
status = EXCLUDED.status,
|
||||||
|
cpu = EXCLUDED.cpu,
|
||||||
|
memory = EXCLUDED.memory,
|
||||||
|
network_bytes_in = EXCLUDED.network_bytes_in,
|
||||||
|
network_bytes_out = EXCLUDED.network_bytes_out,
|
||||||
|
network_packets_in = EXCLUDED.network_packets_in,
|
||||||
|
network_packets_out = EXCLUDED.network_packets_out,
|
||||||
|
network_connections_in = EXCLUDED.network_connections_in,
|
||||||
|
network_connections_out = EXCLUDED.network_connections_out,
|
||||||
|
network_errors_in = EXCLUDED.network_errors_in,
|
||||||
|
network_errors_out = EXCLUDED.network_errors_out,
|
||||||
|
start_time = EXCLUDED.start_time,
|
||||||
|
last_seen = EXCLUDED.last_seen,
|
||||||
|
health_status = EXCLUDED.health_status,
|
||||||
|
health_last_check = EXCLUDED.health_last_check,
|
||||||
|
health_check_count = EXCLUDED.health_check_count,
|
||||||
|
health_failure_count = EXCLUDED.health_failure_count;
|
||||||
|
|
||||||
|
-- name: ListNodeMetrics :many
|
||||||
|
SELECT node_id, timestamp, cpu_usage, cpu_cores, load_avg_1, load_avg_5, load_avg_15,
|
||||||
|
memory_total, memory_used, memory_available, memory_usage_percent,
|
||||||
|
storage_total, storage_used, storage_available, storage_usage_percent,
|
||||||
|
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
||||||
|
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
||||||
|
coalesce(extract(epoch from uptime), 0)::bigint AS uptime_seconds,
|
||||||
|
processes, os, kernel, architecture
|
||||||
|
FROM node_metrics
|
||||||
|
WHERE node_id = $1 AND timestamp BETWEEN $2 AND $3
|
||||||
|
ORDER BY timestamp ASC;
|
||||||
|
|
||||||
|
-- name: ListServiceMetrics :many
|
||||||
|
SELECT service_id, service_name, project_id, timestamp,
|
||||||
|
requests_total, requests_success, requests_errors, requests_avg_latency,
|
||||||
|
requests_p95_latency, requests_p99_latency, requests_throughput,
|
||||||
|
errors_total, errors_rate, performance_response_time, performance_throughput,
|
||||||
|
performance_concurrency, performance_saturation, performance_utilization,
|
||||||
|
resource_cpu_usage, resource_memory_usage, resource_storage_usage,
|
||||||
|
resource_network_usage, resource_score
|
||||||
|
FROM service_metrics
|
||||||
|
WHERE service_id = $1 AND timestamp BETWEEN $2 AND $3
|
||||||
|
ORDER BY timestamp ASC;
|
||||||
|
|
||||||
|
-- name: ListInstanceMetrics :many
|
||||||
|
SELECT instance_id, node_id, status, cpu, memory,
|
||||||
|
network_bytes_in, network_bytes_out, network_packets_in, network_packets_out,
|
||||||
|
network_connections_in, network_connections_out, network_errors_in, network_errors_out,
|
||||||
|
start_time, last_seen, health_status, health_last_check, health_check_count, health_failure_count
|
||||||
|
FROM instance_metrics
|
||||||
|
WHERE service_id = $1 AND timestamp = $2;
|
||||||
|
|
||||||
|
-- name: AggregateNodeMetrics :many
|
||||||
|
-- Portable bucket bucketing (no TimescaleDB): floor epoch seconds into
|
||||||
|
-- bucket-sized buckets, then back to timestamptz.
|
||||||
|
SELECT date_bin($1::interval, timestamp, '2000-01-01'::timestamptz) AS bucket,
|
||||||
|
AVG(cpu_usage)::float8 AS avg_cpu,
|
||||||
|
AVG(memory_usage_percent)::float8 AS avg_memory,
|
||||||
|
AVG(storage_usage_percent)::float8 AS avg_storage
|
||||||
|
FROM node_metrics
|
||||||
|
WHERE node_id = $2 AND timestamp BETWEEN $3 AND $4
|
||||||
|
GROUP BY bucket
|
||||||
|
ORDER BY bucket ASC;
|
||||||
@@ -0,0 +1,121 @@
|
|||||||
|
-- name: ListPreviewEnvironmentsForProject :many
|
||||||
|
SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.pr_number,
|
||||||
|
pe.environment, pe.status, pe.url, pe.expires_at, pe.created_at, pe.updated_at,
|
||||||
|
s.id AS svc_id, s.name AS service_name, s.type AS service_type
|
||||||
|
FROM preview_environments pe
|
||||||
|
LEFT JOIN services s ON pe.service_id = s.id
|
||||||
|
WHERE pe.project_id = $1
|
||||||
|
ORDER BY pe.created_at DESC;
|
||||||
|
|
||||||
|
-- name: GetPreviewEnvironment :one
|
||||||
|
SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.pr_number,
|
||||||
|
pe.environment, pe.status, pe.url, pe.expires_at, pe.created_at, pe.updated_at,
|
||||||
|
s.id AS svc_id, s.name AS service_name, s.type AS service_type
|
||||||
|
FROM preview_environments pe
|
||||||
|
LEFT JOIN services s ON pe.service_id = s.id
|
||||||
|
JOIN projects p ON pe.project_id = p.id
|
||||||
|
WHERE pe.id = $1 AND (p.is_approved OR p.owner_id = $2 OR $3::bool
|
||||||
|
OR EXISTS (SELECT 1 FROM project_members pm WHERE pm.project_id = p.id AND pm.user_id = $2));
|
||||||
|
|
||||||
|
-- name: GetPreviewEnvironmentForWrite :one
|
||||||
|
SELECT pe.id, pe.project_id, pe.service_id, pe.branch_name, pe.pr_number,
|
||||||
|
pe.environment, pe.status, pe.url, pe.expires_at, pe.created_at, pe.updated_at
|
||||||
|
FROM preview_environments pe
|
||||||
|
JOIN projects p ON pe.project_id = p.id
|
||||||
|
WHERE pe.id = $1 AND (p.owner_id = $2 OR $3::bool);
|
||||||
|
|
||||||
|
-- name: GetPreviewEnvironmentForPromote :one
|
||||||
|
SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.environment, pe.status
|
||||||
|
FROM preview_environments pe
|
||||||
|
JOIN projects p ON pe.project_id = p.id
|
||||||
|
WHERE pe.id = $1 AND (p.owner_id = $2 OR $3::bool);
|
||||||
|
|
||||||
|
-- name: GetPreviewEnvironmentOwner :one
|
||||||
|
SELECT p.owner_id
|
||||||
|
FROM preview_environments pe
|
||||||
|
JOIN projects p ON pe.project_id = p.id
|
||||||
|
WHERE pe.id = $1;
|
||||||
|
|
||||||
|
-- name: GetPreviewServiceID :one
|
||||||
|
SELECT preview_service_id FROM preview_environments WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: UpdatePreviewEnvironment :exec
|
||||||
|
UPDATE preview_environments
|
||||||
|
SET status = $1, url = $2, expires_at = $3, updated_at = $4
|
||||||
|
WHERE id = $5;
|
||||||
|
|
||||||
|
-- name: MarkPreviewEnvironmentStatus :exec
|
||||||
|
UPDATE preview_environments SET status = $1, updated_at = NOW() WHERE id = $2;
|
||||||
|
|
||||||
|
-- name: DeletePreviewEnvironment :exec
|
||||||
|
DELETE FROM preview_environments WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: CountActivePreviewsForBranch :one
|
||||||
|
SELECT COUNT(*) FROM preview_environments
|
||||||
|
WHERE service_id = $1 AND branch_name = $2 AND status NOT IN ('expired', 'stopped');
|
||||||
|
|
||||||
|
-- name: CreatePreviewEnvironment :exec
|
||||||
|
INSERT INTO preview_environments
|
||||||
|
(id, project_id, service_id, preview_service_id, branch_name, pr_number, environment,
|
||||||
|
status, url, expires_at, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12);
|
||||||
|
|
||||||
|
-- name: UpdateServicePreviewBranch :exec
|
||||||
|
UPDATE services SET git_branch = $1, domain = '', updated_at = NOW() WHERE id = $2;
|
||||||
|
|
||||||
|
-- name: DeleteAllServiceDomains :exec
|
||||||
|
DELETE FROM service_domains WHERE service_id = $1;
|
||||||
|
|
||||||
|
-- name: CreatePreviewDomain :exec
|
||||||
|
INSERT INTO service_domains (service_id, domain, is_default, cert_type) VALUES ($1, $2, true, 'tls');
|
||||||
|
|
||||||
|
-- name: InsertPendingDeployment :exec
|
||||||
|
INSERT INTO deployments (id, service_id, version, commit_hash, status, created_at, updated_at)
|
||||||
|
VALUES ($1, $2, $3, $4, 'pending', NOW(), NOW());
|
||||||
|
|
||||||
|
-- name: ListExpiredPreviewsForUser :many
|
||||||
|
SELECT pe.id, pe.project_id, pe.service_id, pe.preview_service_id, pe.branch_name, pe.environment
|
||||||
|
FROM preview_environments pe
|
||||||
|
JOIN projects p ON pe.project_id = p.id
|
||||||
|
WHERE (p.owner_id = $1 OR $2::bool) AND pe.expires_at < NOW() AND pe.status != 'expired';
|
||||||
|
|
||||||
|
-- name: ListSweepablePreviews :many
|
||||||
|
SELECT id, preview_service_id FROM preview_environments
|
||||||
|
WHERE expires_at < NOW() AND status NOT IN ('expired', 'stopped');
|
||||||
|
|
||||||
|
-- name: SyncPreviewStatuses :exec
|
||||||
|
UPDATE preview_environments pe
|
||||||
|
SET status = derived.st, updated_at = NOW()
|
||||||
|
FROM (
|
||||||
|
SELECT pe2.id,
|
||||||
|
CASE
|
||||||
|
WHEN s.status IN ('running', 'deployed') THEN 'running'
|
||||||
|
WHEN s.status IN ('failed', 'error') THEN 'failed'
|
||||||
|
WHEN s.status IN ('building', 'deploying', 'queued', 'pending', 'cloning') THEN 'building'
|
||||||
|
ELSE pe2.status
|
||||||
|
END AS st
|
||||||
|
FROM preview_environments pe2
|
||||||
|
JOIN services s ON pe2.preview_service_id = s.id
|
||||||
|
) derived
|
||||||
|
WHERE pe.id = derived.id
|
||||||
|
AND pe.status IN ('building', 'running', 'failed')
|
||||||
|
AND pe.status <> derived.st;
|
||||||
|
|
||||||
|
-- name: GetProjectBrief :one
|
||||||
|
SELECT id, name, owner_id FROM projects WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetServiceTypeBrief :one
|
||||||
|
SELECT id, name, COALESCE(type, service_type, '') AS service_type FROM services WHERE id = $1 AND project_id = $2;
|
||||||
|
|
||||||
|
-- name: GetServiceForDeploy :one
|
||||||
|
SELECT s.id, s.project_id, s.name, s.type, s.status, s.image, s.command,
|
||||||
|
s.environment, s.git_repo, s.git_branch, s.build_path, s.cpu, s.memory,
|
||||||
|
COALESCE(s.replicas, 1) AS replicas, COALESCE(s.port, 0) AS port,
|
||||||
|
COALESCE(s.domain, '') AS domain, COALESCE(s.healthcheck_path, '') AS healthcheck_path,
|
||||||
|
COALESCE(s.restart_policy, 'unless-stopped') AS restart_policy,
|
||||||
|
COALESCE(s.builder, 'auto') AS builder, COALESCE(s.cpu_reserve, '') AS cpu_reserve,
|
||||||
|
COALESCE(s.memory_reserve, '') AS memory_reserve, COALESCE(s.static_build_cmd, '') AS static_build_cmd,
|
||||||
|
COALESCE(s.static_dir, '') AS static_dir,
|
||||||
|
s.created_at, s.updated_at
|
||||||
|
FROM services s
|
||||||
|
WHERE s.id = $1;
|
||||||
@@ -0,0 +1,59 @@
|
|||||||
|
-- name: ListNodeAgentsForService :many
|
||||||
|
SELECT DISTINCT node_agent_id FROM container_instances WHERE service_id = $1;
|
||||||
|
|
||||||
|
-- name: ListServiceVariableValues :many
|
||||||
|
SELECT key, value FROM environment_variables WHERE service_id = $1;
|
||||||
|
|
||||||
|
-- name: ListServiceVariablesWithSecret :many
|
||||||
|
SELECT key, value, COALESCE(is_secret, false) AS is_secret
|
||||||
|
FROM environment_variables WHERE service_id = $1;
|
||||||
|
|
||||||
|
-- name: ListProjectServiceRefs :many
|
||||||
|
SELECT id, name FROM services WHERE project_id = $1;
|
||||||
|
|
||||||
|
-- name: GetServicePort :one
|
||||||
|
SELECT COALESCE(port, 0) AS port FROM services WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetServiceVariableValue :one
|
||||||
|
SELECT value FROM environment_variables WHERE service_id = $1 AND key = $2;
|
||||||
|
|
||||||
|
-- name: GetServiceRuntimeWithOwner :one
|
||||||
|
SELECT s.id, s.project_id, s.name,
|
||||||
|
COALESCE(s.type, s.service_type, '') AS type, COALESCE(s.status, '') AS status,
|
||||||
|
COALESCE(s.image, s.image_name, '') AS image, COALESCE(s.command, s.start_command, '') AS command,
|
||||||
|
COALESCE(s.environment, '') AS environment, COALESCE(s.git_repo, s.source_url, '') AS git_repo,
|
||||||
|
COALESCE(s.git_branch, '') AS git_branch, COALESCE(s.build_path, '') AS build_path,
|
||||||
|
COALESCE(s.cpu, '') AS cpu, COALESCE(s.memory, '') AS memory,
|
||||||
|
COALESCE(s.replicas, 1) AS replicas, COALESCE(s.port, 0) AS port,
|
||||||
|
COALESCE(s.domain, '') AS domain, COALESCE(s.healthcheck_path, '') AS healthcheck_path,
|
||||||
|
COALESCE(s.restart_policy, 'unless-stopped') AS restart_policy,
|
||||||
|
COALESCE(s.builder, 'auto') AS builder, COALESCE(s.cpu_reserve, '') AS cpu_reserve,
|
||||||
|
COALESCE(s.memory_reserve, '') AS memory_reserve, COALESCE(s.static_build_cmd, '') AS static_build_cmd,
|
||||||
|
COALESCE(s.static_dir, '') AS static_dir,
|
||||||
|
s.created_at, s.updated_at, p.owner_id
|
||||||
|
FROM services s
|
||||||
|
JOIN projects p ON s.project_id = p.id
|
||||||
|
WHERE s.id = $1;
|
||||||
|
|
||||||
|
-- name: GetLastDeployedImage :one
|
||||||
|
SELECT (image_name || ':' || image_tag)::text AS image FROM deployments
|
||||||
|
WHERE service_id = $1 AND status = 'deployed' AND image_name <> ''
|
||||||
|
ORDER BY created_at DESC LIMIT 1;
|
||||||
|
|
||||||
|
-- name: SetServicePublishedPort :exec
|
||||||
|
UPDATE services SET published_port = $1 WHERE id = $2 AND published_port <> $1;
|
||||||
|
|
||||||
|
-- name: SetServiceReplicas :exec
|
||||||
|
UPDATE services SET replicas = $1, updated_at = $2 WHERE id = $3;
|
||||||
|
|
||||||
|
-- name: ListSleepCandidates :many
|
||||||
|
SELECT id, project_id, name, COALESCE(status,'') AS status,
|
||||||
|
COALESCE(sleep_idle_minutes, 15) AS sleep_idle_minutes
|
||||||
|
FROM services
|
||||||
|
WHERE sleep_enabled AND COALESCE(status,'') IN ('running','deployed','degraded','starting');
|
||||||
|
|
||||||
|
-- name: GetServiceDomainText :one
|
||||||
|
SELECT COALESCE(domain,'') AS domain FROM services WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetServiceStatusText :one
|
||||||
|
SELECT COALESCE(status,'') AS status FROM services WHERE id = $1;
|
||||||
@@ -0,0 +1,78 @@
|
|||||||
|
-- name: GetProjectOwnerText :one
|
||||||
|
SELECT owner_id::text FROM projects WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: ServiceExistsInProject :one
|
||||||
|
SELECT EXISTS(SELECT 1 FROM services WHERE id = $1 AND project_id = $2);
|
||||||
|
|
||||||
|
-- name: ListVulnerabilities :many
|
||||||
|
SELECT id, type, severity, title, description, service_id, status, found_at, resolved_at
|
||||||
|
FROM vulnerabilities
|
||||||
|
WHERE project_id = $1
|
||||||
|
ORDER BY
|
||||||
|
CASE severity
|
||||||
|
WHEN 'critical' THEN 1
|
||||||
|
WHEN 'high' THEN 2
|
||||||
|
WHEN 'medium' THEN 3
|
||||||
|
WHEN 'low' THEN 4
|
||||||
|
END,
|
||||||
|
found_at DESC;
|
||||||
|
|
||||||
|
-- name: UpdateVulnerabilityStatus :exec
|
||||||
|
UPDATE vulnerabilities SET status = $1, resolved_at = $2 WHERE id = $3;
|
||||||
|
|
||||||
|
-- name: ComplianceFrameworkExists :one
|
||||||
|
SELECT EXISTS(SELECT 1 FROM compliance_frameworks WHERE id = $1);
|
||||||
|
|
||||||
|
-- name: ListComplianceFrameworks :many
|
||||||
|
SELECT id, name, description, version, enabled, created_at
|
||||||
|
FROM compliance_frameworks
|
||||||
|
WHERE enabled = true
|
||||||
|
ORDER BY name;
|
||||||
|
|
||||||
|
-- name: GetVulnerabilityMetrics :one
|
||||||
|
SELECT
|
||||||
|
COUNT(*) AS total,
|
||||||
|
COUNT(*) FILTER (WHERE severity = 'critical') AS critical,
|
||||||
|
COUNT(*) FILTER (WHERE severity = 'high') AS high,
|
||||||
|
COUNT(*) FILTER (WHERE severity = 'medium') AS medium,
|
||||||
|
COUNT(*) FILTER (WHERE severity = 'low') AS low,
|
||||||
|
COUNT(*) FILTER (WHERE status = 'open') AS open,
|
||||||
|
COUNT(*) FILTER (WHERE status = 'resolved') AS resolved
|
||||||
|
FROM vulnerabilities
|
||||||
|
WHERE project_id = $1;
|
||||||
|
|
||||||
|
-- name: GetLatestSecurityScan :one
|
||||||
|
SELECT id, COALESCE((summary->>'score')::int, 0)::int AS score, started_at AS scanned_at, status
|
||||||
|
FROM security_scans
|
||||||
|
WHERE project_id = $1
|
||||||
|
ORDER BY started_at DESC
|
||||||
|
LIMIT 1;
|
||||||
|
|
||||||
|
-- name: GetLatestComplianceReport :one
|
||||||
|
SELECT overall_status, score, assessment_date
|
||||||
|
FROM compliance_reports
|
||||||
|
WHERE project_id = $1
|
||||||
|
ORDER BY assessment_date DESC
|
||||||
|
LIMIT 1;
|
||||||
|
|
||||||
|
-- name: CheckSecurityProjectAccess :one
|
||||||
|
SELECT EXISTS (
|
||||||
|
SELECT 1
|
||||||
|
FROM projects p
|
||||||
|
WHERE p.id = $1
|
||||||
|
AND (p.is_approved
|
||||||
|
OR $3::bool
|
||||||
|
OR p.owner_id = $2 OR EXISTS (
|
||||||
|
SELECT 1 FROM project_members pm
|
||||||
|
WHERE pm.project_id = p.id AND pm.user_id = $2
|
||||||
|
))
|
||||||
|
);
|
||||||
|
|
||||||
|
-- name: GetSecurityScanProjectID :one
|
||||||
|
SELECT project_id FROM security_scans WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetComplianceReportProjectID :one
|
||||||
|
SELECT project_id FROM compliance_reports WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: GetVulnerabilityProjectID :one
|
||||||
|
SELECT project_id FROM vulnerabilities WHERE id = $1;
|
||||||
@@ -0,0 +1,38 @@
|
|||||||
|
-- name: CountServicesByName :one
|
||||||
|
SELECT COUNT(*) FROM services WHERE project_id = $1 AND name = $2;
|
||||||
|
|
||||||
|
-- name: GetServiceEnvironment :one
|
||||||
|
SELECT COALESCE(environment, '') FROM services WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: CloneServiceRow :exec
|
||||||
|
INSERT INTO services
|
||||||
|
(id, project_id, name, environment_id, service_type, source_type, source_url, image_name,
|
||||||
|
build_command, start_command, type, status, image, command, environment,
|
||||||
|
git_repo, git_branch, build_path, cpu, memory, replicas, port, domain,
|
||||||
|
healthcheck_path, restart_policy, volumes, maintenance_mode, basic_auth_users,
|
||||||
|
builder, cpu_reserve, memory_reserve, static_build_cmd, static_dir,
|
||||||
|
traefik_labels, created_at, updated_at)
|
||||||
|
SELECT $1, $2, $3, $4, service_type, source_type, source_url, image_name,
|
||||||
|
build_command, start_command, type, 'stopped', image, command, $5,
|
||||||
|
git_repo, git_branch, build_path, cpu, memory, replicas, port, domain,
|
||||||
|
healthcheck_path, restart_policy, volumes, $6, $7,
|
||||||
|
builder, cpu_reserve, memory_reserve, static_build_cmd, static_dir,
|
||||||
|
traefik_labels, $8, $8
|
||||||
|
FROM services s WHERE s.id = $9;
|
||||||
|
|
||||||
|
-- name: CloneServiceDomains :exec
|
||||||
|
INSERT INTO service_domains (service_id, domain, is_default, cert_type)
|
||||||
|
SELECT $1, domain, is_default, cert_type FROM service_domains sd WHERE sd.service_id = $2;
|
||||||
|
|
||||||
|
-- name: CloneServiceVariables :exec
|
||||||
|
INSERT INTO environment_variables (id, service_id, key, value, is_secret)
|
||||||
|
SELECT gen_random_uuid(), $1, key, value, is_secret FROM environment_variables ev WHERE ev.service_id = $2;
|
||||||
|
|
||||||
|
-- name: DeleteServiceByID :exec
|
||||||
|
DELETE FROM services WHERE id = $1;
|
||||||
|
|
||||||
|
-- name: MoveServiceProject :execrows
|
||||||
|
UPDATE services SET project_id = $1, environment_id = $2, updated_at = $3 WHERE id = $4;
|
||||||
|
|
||||||
|
-- name: GetProjectOwner :one
|
||||||
|
SELECT owner_id FROM projects WHERE id = $1;
|
||||||
+247
-3
@@ -69,6 +69,8 @@ CREATE TABLE services (
|
|||||||
static_dir VARCHAR(255) NOT NULL DEFAULT '',
|
static_dir VARCHAR(255) NOT NULL DEFAULT '',
|
||||||
node_id VARCHAR(255),
|
node_id VARCHAR(255),
|
||||||
spread BOOLEAN NOT NULL DEFAULT false,
|
spread BOOLEAN NOT NULL DEFAULT false,
|
||||||
|
sleep_enabled BOOLEAN NOT NULL DEFAULT false,
|
||||||
|
sleep_idle_minutes INTEGER NOT NULL DEFAULT 15,
|
||||||
placement_tags JSONB NOT NULL DEFAULT '[]',
|
placement_tags JSONB NOT NULL DEFAULT '[]',
|
||||||
traefik_labels JSONB NOT NULL DEFAULT '{}'
|
traefik_labels JSONB NOT NULL DEFAULT '{}'
|
||||||
);
|
);
|
||||||
@@ -98,11 +100,37 @@ CREATE TABLE service_domains (
|
|||||||
);
|
);
|
||||||
|
|
||||||
CREATE TABLE deployments (
|
CREATE TABLE deployments (
|
||||||
id UUID PRIMARY KEY,
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
service_id UUID NOT NULL REFERENCES services(id) ON DELETE CASCADE,
|
service_id UUID NOT NULL REFERENCES services(id) ON DELETE CASCADE,
|
||||||
status VARCHAR(50),
|
version VARCHAR(255) NOT NULL,
|
||||||
|
commit_hash VARCHAR(255),
|
||||||
|
image_digest VARCHAR(255),
|
||||||
|
status VARCHAR(50) DEFAULT 'created',
|
||||||
|
build_log TEXT,
|
||||||
|
deployment_log TEXT,
|
||||||
|
started_at TIMESTAMPTZ,
|
||||||
|
completed_at TIMESTAMPTZ,
|
||||||
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW()
|
updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
|
image_name VARCHAR(255),
|
||||||
|
image_tag VARCHAR(255),
|
||||||
|
runtime_log TEXT,
|
||||||
|
error TEXT
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE preview_environments (
|
||||||
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
project_id UUID NOT NULL,
|
||||||
|
service_id UUID NOT NULL,
|
||||||
|
branch_name VARCHAR(255) NOT NULL,
|
||||||
|
pr_number INTEGER,
|
||||||
|
environment VARCHAR(255) NOT NULL,
|
||||||
|
status VARCHAR(50) NOT NULL DEFAULT 'building',
|
||||||
|
url TEXT,
|
||||||
|
expires_at TIMESTAMPTZ NOT NULL,
|
||||||
|
created_at TIMESTAMPTZ DEFAULT now(),
|
||||||
|
updated_at TIMESTAMPTZ DEFAULT now(),
|
||||||
|
preview_service_id UUID
|
||||||
);
|
);
|
||||||
|
|
||||||
CREATE TABLE environment_variables (
|
CREATE TABLE environment_variables (
|
||||||
@@ -346,3 +374,219 @@ CREATE TABLE notification_channels (
|
|||||||
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW()
|
updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW()
|
||||||
);
|
);
|
||||||
|
|
||||||
|
CREATE TABLE node_metrics (
|
||||||
|
node_id VARCHAR(255) NOT NULL,
|
||||||
|
timestamp TIMESTAMP WITH TIME ZONE NOT NULL,
|
||||||
|
cpu_usage NUMERIC(5,2),
|
||||||
|
cpu_cores NUMERIC(10,2),
|
||||||
|
load_avg_1 NUMERIC(5,2),
|
||||||
|
load_avg_5 NUMERIC(5,2),
|
||||||
|
load_avg_15 NUMERIC(5,2),
|
||||||
|
memory_total BIGINT,
|
||||||
|
memory_used BIGINT,
|
||||||
|
memory_available BIGINT,
|
||||||
|
memory_usage_percent NUMERIC(5,2),
|
||||||
|
storage_total BIGINT,
|
||||||
|
storage_used BIGINT,
|
||||||
|
storage_available BIGINT,
|
||||||
|
storage_usage_percent NUMERIC(5,2),
|
||||||
|
network_bytes_in BIGINT,
|
||||||
|
network_bytes_out BIGINT,
|
||||||
|
network_packets_in BIGINT,
|
||||||
|
network_packets_out BIGINT,
|
||||||
|
network_connections_in INTEGER,
|
||||||
|
network_connections_out INTEGER,
|
||||||
|
network_errors_in BIGINT,
|
||||||
|
network_errors_out BIGINT,
|
||||||
|
uptime INTERVAL,
|
||||||
|
processes INTEGER,
|
||||||
|
os VARCHAR(50),
|
||||||
|
kernel VARCHAR(50),
|
||||||
|
architecture VARCHAR(20),
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
|
PRIMARY KEY (node_id, timestamp)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE service_metrics (
|
||||||
|
service_id VARCHAR(255) NOT NULL,
|
||||||
|
service_name VARCHAR(255) NOT NULL,
|
||||||
|
project_id VARCHAR(255) NOT NULL,
|
||||||
|
timestamp TIMESTAMP WITH TIME ZONE NOT NULL,
|
||||||
|
requests_total BIGINT DEFAULT 0,
|
||||||
|
requests_success BIGINT DEFAULT 0,
|
||||||
|
requests_errors BIGINT DEFAULT 0,
|
||||||
|
requests_avg_latency NUMERIC(10,3),
|
||||||
|
requests_p95_latency NUMERIC(10,3),
|
||||||
|
requests_p99_latency NUMERIC(10,3),
|
||||||
|
requests_throughput NUMERIC(10,3),
|
||||||
|
errors_total BIGINT DEFAULT 0,
|
||||||
|
errors_rate NUMERIC(5,4),
|
||||||
|
performance_response_time NUMERIC(10,3),
|
||||||
|
performance_throughput NUMERIC(10,3),
|
||||||
|
performance_concurrency BIGINT,
|
||||||
|
performance_saturation NUMERIC(5,2),
|
||||||
|
performance_utilization NUMERIC(5,2),
|
||||||
|
resource_cpu_usage NUMERIC(5,2),
|
||||||
|
resource_memory_usage BIGINT,
|
||||||
|
resource_storage_usage BIGINT,
|
||||||
|
resource_network_usage BIGINT,
|
||||||
|
resource_score NUMERIC(5,2),
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
|
PRIMARY KEY (service_id, timestamp)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE instance_metrics (
|
||||||
|
service_id VARCHAR(255) NOT NULL,
|
||||||
|
timestamp TIMESTAMP WITH TIME ZONE NOT NULL,
|
||||||
|
instance_id VARCHAR(255) NOT NULL,
|
||||||
|
node_id VARCHAR(255),
|
||||||
|
status VARCHAR(50),
|
||||||
|
cpu NUMERIC(5,2),
|
||||||
|
memory BIGINT,
|
||||||
|
network_bytes_in BIGINT,
|
||||||
|
network_bytes_out BIGINT,
|
||||||
|
network_packets_in BIGINT,
|
||||||
|
network_packets_out BIGINT,
|
||||||
|
network_connections_in INTEGER,
|
||||||
|
network_connections_out INTEGER,
|
||||||
|
network_errors_in BIGINT,
|
||||||
|
network_errors_out BIGINT,
|
||||||
|
start_time TIMESTAMP WITH TIME ZONE,
|
||||||
|
last_seen TIMESTAMP WITH TIME ZONE,
|
||||||
|
health_status VARCHAR(20),
|
||||||
|
health_last_check TIMESTAMP WITH TIME ZONE,
|
||||||
|
health_check_count INTEGER DEFAULT 0,
|
||||||
|
health_failure_count INTEGER DEFAULT 0,
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
|
PRIMARY KEY (service_id, timestamp, instance_id)
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE cron_jobs (
|
||||||
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
project_id UUID NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
|
||||||
|
service_id UUID NOT NULL REFERENCES services(id) ON DELETE CASCADE,
|
||||||
|
name VARCHAR(255) NOT NULL,
|
||||||
|
schedule VARCHAR(100) NOT NULL,
|
||||||
|
command TEXT NOT NULL,
|
||||||
|
timezone VARCHAR(50) DEFAULT 'UTC',
|
||||||
|
enabled BOOLEAN DEFAULT true,
|
||||||
|
last_run_at TIMESTAMP WITH TIME ZONE,
|
||||||
|
next_run_at TIMESTAMP WITH TIME ZONE,
|
||||||
|
last_status VARCHAR(50),
|
||||||
|
last_output TEXT,
|
||||||
|
retention INTEGER DEFAULT 30,
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW()
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE cron_executions (
|
||||||
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
cron_job_id UUID NOT NULL REFERENCES cron_jobs(id) ON DELETE CASCADE,
|
||||||
|
started_at TIMESTAMP WITH TIME ZONE NOT NULL,
|
||||||
|
finished_at TIMESTAMP WITH TIME ZONE,
|
||||||
|
status VARCHAR(50) DEFAULT 'pending',
|
||||||
|
output TEXT,
|
||||||
|
error TEXT
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE git_providers (
|
||||||
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
name VARCHAR(255) NOT NULL,
|
||||||
|
display_name VARCHAR(255) NOT NULL,
|
||||||
|
api_url VARCHAR(255) NOT NULL,
|
||||||
|
webhook_url VARCHAR(255) NOT NULL,
|
||||||
|
access_token TEXT NOT NULL,
|
||||||
|
user_id UUID NOT NULL,
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW()
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE git_repositories (
|
||||||
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
provider_id UUID NOT NULL,
|
||||||
|
name VARCHAR(255) NOT NULL,
|
||||||
|
full_name VARCHAR(255) NOT NULL,
|
||||||
|
description TEXT,
|
||||||
|
clone_url VARCHAR(255) NOT NULL,
|
||||||
|
webhook_url VARCHAR(255),
|
||||||
|
default_branch VARCHAR(255) DEFAULT 'main',
|
||||||
|
is_private BOOLEAN DEFAULT false,
|
||||||
|
user_id UUID NOT NULL,
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW()
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE git_webhooks (
|
||||||
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
repo_id UUID NOT NULL,
|
||||||
|
provider_id UUID NOT NULL,
|
||||||
|
events TEXT NOT NULL,
|
||||||
|
webhook_secret TEXT NOT NULL,
|
||||||
|
remote_webhook_id VARCHAR(255),
|
||||||
|
active BOOLEAN DEFAULT true,
|
||||||
|
branch_filter VARCHAR(255),
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW()
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE users (
|
||||||
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
email VARCHAR(255) NOT NULL UNIQUE,
|
||||||
|
password_hash VARCHAR(255) NOT NULL,
|
||||||
|
name VARCHAR(255) NOT NULL,
|
||||||
|
avatar_url VARCHAR(500),
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
||||||
|
is_admin BOOLEAN NOT NULL DEFAULT false
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE compliance_frameworks (
|
||||||
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
name VARCHAR(100) NOT NULL UNIQUE,
|
||||||
|
description TEXT,
|
||||||
|
version VARCHAR(20) NOT NULL,
|
||||||
|
enabled BOOLEAN NOT NULL DEFAULT true,
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW()
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE security_scans (
|
||||||
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
project_id UUID NOT NULL,
|
||||||
|
service_id UUID,
|
||||||
|
scan_type VARCHAR(50) NOT NULL,
|
||||||
|
status VARCHAR(50) NOT NULL DEFAULT 'running',
|
||||||
|
started_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW(),
|
||||||
|
completed_at TIMESTAMP WITH TIME ZONE,
|
||||||
|
summary JSONB DEFAULT '{}',
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW()
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE compliance_reports (
|
||||||
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
project_id UUID NOT NULL,
|
||||||
|
framework_id UUID NOT NULL,
|
||||||
|
assessment_date TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW(),
|
||||||
|
assessor VARCHAR(255),
|
||||||
|
overall_status VARCHAR(50) NOT NULL,
|
||||||
|
score INTEGER NOT NULL DEFAULT 0,
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW()
|
||||||
|
);
|
||||||
|
|
||||||
|
CREATE TABLE vulnerabilities (
|
||||||
|
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||||
|
type VARCHAR(50) NOT NULL,
|
||||||
|
severity VARCHAR(20) NOT NULL,
|
||||||
|
title VARCHAR(255) NOT NULL,
|
||||||
|
description TEXT,
|
||||||
|
service_id UUID,
|
||||||
|
project_id UUID NOT NULL,
|
||||||
|
status VARCHAR(50) NOT NULL DEFAULT 'open',
|
||||||
|
found_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW(),
|
||||||
|
resolved_at TIMESTAMP WITH TIME ZONE,
|
||||||
|
created_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW(),
|
||||||
|
updated_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW()
|
||||||
|
);
|
||||||
@@ -0,0 +1,467 @@
|
|||||||
|
import { useCallback, useMemo, useState } from 'react';
|
||||||
|
import {
|
||||||
|
Background,
|
||||||
|
BackgroundVariant,
|
||||||
|
Handle,
|
||||||
|
MarkerType,
|
||||||
|
Position,
|
||||||
|
ReactFlow,
|
||||||
|
ReactFlowProvider,
|
||||||
|
useReactFlow,
|
||||||
|
type Connection,
|
||||||
|
type Edge,
|
||||||
|
type EdgeChange,
|
||||||
|
type Node,
|
||||||
|
type NodeChange,
|
||||||
|
type NodeProps,
|
||||||
|
} from '@xyflow/react';
|
||||||
|
import '@xyflow/react/dist/style.css';
|
||||||
|
import { Database, GitBranch, Package, Timer, Trash2, X } from 'lucide-react';
|
||||||
|
import {
|
||||||
|
canvasFromConfig,
|
||||||
|
configFromCanvas,
|
||||||
|
newServiceKey,
|
||||||
|
validateGraph,
|
||||||
|
type BuilderNode,
|
||||||
|
type BuilderService,
|
||||||
|
type BuilderServiceType,
|
||||||
|
} from './builder-model';
|
||||||
|
|
||||||
|
type ServiceBuilderNode = Node<{ svc: BuilderService }, 'tplService'>;
|
||||||
|
|
||||||
|
const TYPE_META: Record<BuilderServiceType, { label: string; icon: typeof Package; cls: string }> = {
|
||||||
|
web: { label: 'Web', icon: Package, cls: 'text-[var(--accent-primary)]' },
|
||||||
|
worker: { label: 'Worker', icon: GitBranch, cls: 'text-[var(--info,#5eb3f6)]' },
|
||||||
|
cron: { label: 'Cron', icon: Timer, cls: 'text-[var(--warning)]' },
|
||||||
|
database: { label: 'Database', icon: Database, cls: 'text-[var(--success,#7bd88a)]' },
|
||||||
|
};
|
||||||
|
|
||||||
|
const DATABASE_ENGINES = ['postgresql', 'mysql', 'mariadb', 'mongodb', 'redis', 'dragonfly', 'clickhouse'];
|
||||||
|
|
||||||
|
function ServiceBuilderNodeView({ data, selected }: NodeProps<ServiceBuilderNode>) {
|
||||||
|
const meta = TYPE_META[data.svc.type] ?? TYPE_META.web;
|
||||||
|
const Icon = meta.icon;
|
||||||
|
return (
|
||||||
|
<div
|
||||||
|
className={`relative w-[210px] rounded-[var(--radius-md)] border bg-[var(--bg-elevated)] px-3 py-2.5 shadow-sm transition-colors ${
|
||||||
|
selected ? 'border-[var(--accent-primary)]' : 'border-[var(--border-subtle)]'
|
||||||
|
}`}
|
||||||
|
>
|
||||||
|
<Handle type="target" position={Position.Left} className="!h-2.5 !w-2.5 !border-[var(--border-default)] !bg-[var(--bg-base)]" />
|
||||||
|
<Handle type="source" position={Position.Right} className="!h-2.5 !w-2.5 !border-[var(--border-default)] !bg-[var(--bg-base)]" />
|
||||||
|
<div className="flex items-center gap-2">
|
||||||
|
<Icon size={14} className={meta.cls} />
|
||||||
|
<span className="truncate text-[12.5px] font-semibold text-[var(--text-primary)]">
|
||||||
|
{data.svc.name || data.svc.key}
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
<div className="mt-1 flex items-center justify-between gap-2">
|
||||||
|
<span className="text-[10.5px] uppercase tracking-wide text-[var(--text-tertiary)]">{meta.label}</span>
|
||||||
|
<span className="mono max-w-[120px] truncate text-[10.5px] text-[var(--text-secondary)]">
|
||||||
|
{data.svc.runtime || data.svc.repo || '—'}
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
const nodeTypes = { tplService: ServiceBuilderNodeView };
|
||||||
|
|
||||||
|
// Edges run dependency → dependent: connecting A → B means "B depends_on A".
|
||||||
|
export function TemplateBuilder(props: {
|
||||||
|
config: Record<string, unknown>;
|
||||||
|
onApply: (config: Record<string, unknown>) => void;
|
||||||
|
onCancel: () => void;
|
||||||
|
}) {
|
||||||
|
return (
|
||||||
|
<ReactFlowProvider>
|
||||||
|
<BuilderInner {...props} />
|
||||||
|
</ReactFlowProvider>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
function BuilderInner({
|
||||||
|
config,
|
||||||
|
onApply,
|
||||||
|
onCancel,
|
||||||
|
}: {
|
||||||
|
config: Record<string, unknown>;
|
||||||
|
onApply: (config: Record<string, unknown>) => void;
|
||||||
|
onCancel: () => void;
|
||||||
|
}) {
|
||||||
|
const { screenToFlowPosition } = useReactFlow();
|
||||||
|
const initial = useMemo(() => canvasFromConfig(config), [config]);
|
||||||
|
const [services, setServices] = useState<Map<string, BuilderService>>(
|
||||||
|
() => new Map(initial.map((n) => [n.key, { ...n.service }])),
|
||||||
|
);
|
||||||
|
const [positions, setPositions] = useState<Record<string, { x: number; y: number }>>(
|
||||||
|
() => Object.fromEntries(initial.map((n) => [n.key, n.position])),
|
||||||
|
);
|
||||||
|
const [selectedKey, setSelectedKey] = useState<string | null>(null);
|
||||||
|
|
||||||
|
const flowNodes: ServiceBuilderNode[] = useMemo(
|
||||||
|
() =>
|
||||||
|
[...services.values()].map((svc) => ({
|
||||||
|
id: svc.key,
|
||||||
|
type: 'tplService' as const,
|
||||||
|
position: positions[svc.key] ?? { x: 0, y: 0 },
|
||||||
|
data: { svc },
|
||||||
|
selected: svc.key === selectedKey,
|
||||||
|
})),
|
||||||
|
[services, positions, selectedKey],
|
||||||
|
);
|
||||||
|
|
||||||
|
const flowEdges: Edge[] = useMemo(
|
||||||
|
() =>
|
||||||
|
[...services.values()].flatMap((svc) =>
|
||||||
|
(svc.depends_on ?? [])
|
||||||
|
.filter((dep) => services.has(dep))
|
||||||
|
.map((dep) => ({
|
||||||
|
id: `${dep}->${svc.key}`,
|
||||||
|
source: dep,
|
||||||
|
target: svc.key,
|
||||||
|
markerEnd: { type: MarkerType.ArrowClosed, width: 14, height: 14, color: 'rgba(180,227,74,0.55)' },
|
||||||
|
style: { stroke: 'rgba(180,227,74,0.42)', strokeWidth: 1.6 },
|
||||||
|
className: 'edge-premium',
|
||||||
|
})),
|
||||||
|
),
|
||||||
|
[services],
|
||||||
|
);
|
||||||
|
|
||||||
|
const builderNodes: BuilderNode[] = useMemo(
|
||||||
|
() =>
|
||||||
|
[...services.values()].map((svc) => ({
|
||||||
|
key: svc.key,
|
||||||
|
service: svc,
|
||||||
|
position: positions[svc.key] ?? { x: 0, y: 0 },
|
||||||
|
})),
|
||||||
|
[services, positions],
|
||||||
|
);
|
||||||
|
const errors = useMemo(() => validateGraph(builderNodes), [builderNodes]);
|
||||||
|
|
||||||
|
const patchService = useCallback((key: string, patch: Partial<BuilderService>) => {
|
||||||
|
setServices((prev) => {
|
||||||
|
const svc = prev.get(key);
|
||||||
|
if (!svc) return prev;
|
||||||
|
const next = new Map(prev);
|
||||||
|
next.set(key, { ...svc, ...patch });
|
||||||
|
return next;
|
||||||
|
});
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
const addService = useCallback((type: BuilderServiceType, position?: { x: number; y: number }) => {
|
||||||
|
setServices((prev) => {
|
||||||
|
const key = newServiceKey(type, new Set(prev.keys()));
|
||||||
|
const next = new Map(prev);
|
||||||
|
next.set(key, {
|
||||||
|
key,
|
||||||
|
type,
|
||||||
|
runtime: type === 'database' ? 'postgresql' : undefined,
|
||||||
|
});
|
||||||
|
setPositions((pos) => ({ ...pos, [key]: position ?? { x: 80, y: 40 + (next.size - 1) * 150 } }));
|
||||||
|
setSelectedKey(key);
|
||||||
|
return next;
|
||||||
|
});
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
const removeService = useCallback((key: string) => {
|
||||||
|
setServices((prev) => {
|
||||||
|
const next = new Map(prev);
|
||||||
|
next.delete(key);
|
||||||
|
for (const svc of next.values()) {
|
||||||
|
svc.depends_on = svc.depends_on?.filter((d) => d !== key);
|
||||||
|
}
|
||||||
|
return next;
|
||||||
|
});
|
||||||
|
setPositions((prev) => {
|
||||||
|
const next = { ...prev };
|
||||||
|
delete next[key];
|
||||||
|
return next;
|
||||||
|
});
|
||||||
|
setSelectedKey((cur) => (cur === key ? null : cur));
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
const onConnect = useCallback(
|
||||||
|
(conn: Connection) => {
|
||||||
|
if (!conn.source || !conn.target || conn.source === conn.target) return;
|
||||||
|
const target = services.get(conn.target);
|
||||||
|
if (!target) return;
|
||||||
|
patchService(conn.target, {
|
||||||
|
depends_on: [...new Set([...(target.depends_on ?? []), conn.source])],
|
||||||
|
});
|
||||||
|
},
|
||||||
|
[patchService, services],
|
||||||
|
);
|
||||||
|
|
||||||
|
const onNodesChange = useCallback(
|
||||||
|
(changes: NodeChange<ServiceBuilderNode>[]) => {
|
||||||
|
for (const ch of changes) {
|
||||||
|
if (ch.type === 'position' && ch.position) {
|
||||||
|
setPositions((prev) => ({ ...prev, [ch.id]: ch.position! }));
|
||||||
|
} else if (ch.type === 'remove') {
|
||||||
|
removeService(ch.id);
|
||||||
|
} else if (ch.type === 'select') {
|
||||||
|
if (ch.selected) setSelectedKey(ch.id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[removeService],
|
||||||
|
);
|
||||||
|
|
||||||
|
const onEdgesChange = useCallback(
|
||||||
|
(changes: EdgeChange<Edge>[]) => {
|
||||||
|
for (const ch of changes) {
|
||||||
|
if (ch.type !== 'remove') continue;
|
||||||
|
const edge = flowEdges.find((e) => e.id === ch.id);
|
||||||
|
if (!edge) continue;
|
||||||
|
const svc = services.get(edge.target);
|
||||||
|
if (svc) {
|
||||||
|
patchService(edge.target, {
|
||||||
|
depends_on: (svc.depends_on ?? []).filter((d) => d !== edge.source),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[flowEdges, patchService, services],
|
||||||
|
);
|
||||||
|
|
||||||
|
const selected = selectedKey ? services.get(selectedKey) : null;
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="flex min-h-0 flex-1">
|
||||||
|
<div className="relative flex-1">
|
||||||
|
<div className="absolute left-3 top-3 z-10 flex gap-1.5">
|
||||||
|
{(['web', 'database', 'cron', 'worker'] as const).map((t) => {
|
||||||
|
const meta = TYPE_META[t];
|
||||||
|
const Icon = meta.icon;
|
||||||
|
return (
|
||||||
|
<button
|
||||||
|
key={t}
|
||||||
|
type="button"
|
||||||
|
onClick={() => addService(t)}
|
||||||
|
className="canvas-pill inline-flex items-center gap-1.5 px-2.5 py-1.5 text-[11px] font-medium text-[var(--text-secondary)] hover:text-[var(--text-primary)]"
|
||||||
|
>
|
||||||
|
<Icon size={12} className={meta.cls} />
|
||||||
|
{meta.label}
|
||||||
|
</button>
|
||||||
|
);
|
||||||
|
})}
|
||||||
|
</div>
|
||||||
|
<ReactFlow
|
||||||
|
nodes={flowNodes}
|
||||||
|
edges={flowEdges}
|
||||||
|
nodeTypes={nodeTypes}
|
||||||
|
onNodesChange={onNodesChange}
|
||||||
|
onEdgesChange={onEdgesChange}
|
||||||
|
onConnect={onConnect}
|
||||||
|
onPaneClick={() => setSelectedKey(null)}
|
||||||
|
onDoubleClick={(e) => {
|
||||||
|
if ((e.target as HTMLElement).closest('.react-flow__node')) return;
|
||||||
|
addService('web', screenToFlowPosition({ x: e.clientX, y: e.clientY }));
|
||||||
|
}}
|
||||||
|
fitView
|
||||||
|
deleteKeyCode={['Backspace', 'Delete']}
|
||||||
|
proOptions={{ hideAttribution: true }}
|
||||||
|
>
|
||||||
|
<Background variant={BackgroundVariant.Dots} gap={22} size={1.2} color="rgba(255,255,255,0.07)" />
|
||||||
|
</ReactFlow>
|
||||||
|
<div className="canvas-pill absolute bottom-3 left-3 z-10 px-3 py-1.5 text-[10.5px] text-[var(--text-tertiary)]">
|
||||||
|
drag = move · connect handles = depends_on · double-click = add web · Del = remove
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="flex w-[290px] shrink-0 flex-col border-l border-[var(--border-subtle)]">
|
||||||
|
{selected ? (
|
||||||
|
<Inspector
|
||||||
|
key={selected.key}
|
||||||
|
svc={selected}
|
||||||
|
onPatch={(patch) => patchService(selected.key, patch)}
|
||||||
|
onRemove={() => removeService(selected.key)}
|
||||||
|
/>
|
||||||
|
) : (
|
||||||
|
<div className="flex flex-1 flex-col items-center justify-center gap-2 px-6 text-center">
|
||||||
|
<Package size={22} className="text-[var(--text-tertiary)]" />
|
||||||
|
<p className="text-xs text-[var(--text-tertiary)]">
|
||||||
|
Select a node to edit it, or add one from the toolbar.
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
<div className="border-t border-[var(--border-subtle)] p-3">
|
||||||
|
{errors.length > 0 ? (
|
||||||
|
<div className="mb-2 max-h-[90px] overflow-y-auto rounded-[var(--radius-sm)] bg-[var(--error-soft)] px-3 py-2">
|
||||||
|
{errors.map((e) => (
|
||||||
|
<p key={e} className="text-[11px] text-[var(--error)]">• {e}</p>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
) : null}
|
||||||
|
<div className="flex gap-2">
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
onClick={onCancel}
|
||||||
|
className="flex-1 rounded-[var(--radius-md)] px-3 py-2 text-xs font-medium text-[var(--text-secondary)] hover:bg-[var(--surface-muted)]"
|
||||||
|
>
|
||||||
|
Back to JSON
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
disabled={errors.length > 0 || services.size === 0}
|
||||||
|
onClick={() => onApply(configFromCanvas(builderNodes, config))}
|
||||||
|
className="flex-1 rounded-[var(--radius-md)] bg-[var(--accent-primary)] px-3 py-2 text-xs font-semibold text-[var(--accent-on)] disabled:opacity-50"
|
||||||
|
>
|
||||||
|
Apply to JSON
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
const FIELD_CLS =
|
||||||
|
'w-full rounded-[var(--radius-sm)] border border-[var(--border-default)] bg-[var(--bg-void)] px-2.5 py-1.5 text-[12px] text-[var(--text-primary)] outline-none focus:border-[var(--accent-primary)]';
|
||||||
|
const LABEL_CLS = 'mb-1 block text-[10.5px] font-medium uppercase tracking-wide text-[var(--text-tertiary)]';
|
||||||
|
|
||||||
|
function Inspector({
|
||||||
|
svc,
|
||||||
|
onPatch,
|
||||||
|
onRemove,
|
||||||
|
}: {
|
||||||
|
svc: BuilderService;
|
||||||
|
onPatch: (patch: Partial<BuilderService>) => void;
|
||||||
|
onRemove: () => void;
|
||||||
|
}) {
|
||||||
|
const envText = Object.entries(svc.environment ?? {})
|
||||||
|
.map(([k, v]) => `${k}=${v}`)
|
||||||
|
.join('\n');
|
||||||
|
const volText = (svc.volumes ?? []).map((v) => `${v.source}:${v.target}${v.read_only ? ':ro' : ''}`).join('\n');
|
||||||
|
return (
|
||||||
|
<div className="flex-1 overflow-y-auto p-4">
|
||||||
|
<div className="mb-3 flex items-center justify-between">
|
||||||
|
<span className="text-[12px] font-semibold text-[var(--text-primary)]">{svc.key}</span>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
onClick={onRemove}
|
||||||
|
className="flex items-center gap-1 rounded-[var(--radius-sm)] px-2 py-1 text-[11px] text-[var(--error)] hover:bg-[var(--error-soft)]"
|
||||||
|
>
|
||||||
|
<Trash2 size={12} /> Remove
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<label className={LABEL_CLS}>Key</label>
|
||||||
|
<input className={FIELD_CLS} value={svc.key} disabled title="Key is the node id — cannot be renamed" />
|
||||||
|
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Display name</label>
|
||||||
|
<input className={FIELD_CLS} value={svc.name ?? ''} onChange={(e) => onPatch({ name: e.target.value || undefined })} placeholder="optional" />
|
||||||
|
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Type</label>
|
||||||
|
<select
|
||||||
|
className={FIELD_CLS}
|
||||||
|
value={svc.type}
|
||||||
|
onChange={(e) => onPatch({ type: e.target.value as BuilderServiceType })}
|
||||||
|
>
|
||||||
|
<option value="web">Web</option>
|
||||||
|
<option value="worker">Worker</option>
|
||||||
|
<option value="cron">Cron</option>
|
||||||
|
<option value="database">Database</option>
|
||||||
|
</select>
|
||||||
|
|
||||||
|
{svc.type === 'database' ? (
|
||||||
|
<>
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Engine</label>
|
||||||
|
<select className={FIELD_CLS} value={svc.runtime ?? 'postgresql'} onChange={(e) => onPatch({ runtime: e.target.value })}>
|
||||||
|
{DATABASE_ENGINES.map((eng) => (
|
||||||
|
<option key={eng} value={eng}>{eng}</option>
|
||||||
|
))}
|
||||||
|
</select>
|
||||||
|
</>
|
||||||
|
) : (
|
||||||
|
<>
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Image (runtime)</label>
|
||||||
|
<input className={FIELD_CLS} value={svc.runtime ?? ''} onChange={(e) => onPatch({ runtime: e.target.value || undefined })} placeholder="nginx:latest" />
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Git repo</label>
|
||||||
|
<input className={FIELD_CLS} value={svc.repo ?? ''} onChange={(e) => onPatch({ repo: e.target.value || undefined })} placeholder="owner/repo or clone URL" />
|
||||||
|
{svc.repo ? (
|
||||||
|
<>
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Branch</label>
|
||||||
|
<input className={FIELD_CLS} value={svc.branch ?? ''} onChange={(e) => onPatch({ branch: e.target.value || undefined })} placeholder="main" />
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Build command</label>
|
||||||
|
<input className={FIELD_CLS} value={svc.build_command ?? ''} onChange={(e) => onPatch({ build_command: e.target.value || undefined })} />
|
||||||
|
</>
|
||||||
|
) : null}
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Start command</label>
|
||||||
|
<input className={FIELD_CLS} value={svc.start_command ?? ''} onChange={(e) => onPatch({ start_command: e.target.value || undefined })} />
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{svc.type === 'web' ? (
|
||||||
|
<>
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Port</label>
|
||||||
|
<input
|
||||||
|
className={FIELD_CLS}
|
||||||
|
type="number"
|
||||||
|
value={svc.port ?? ''}
|
||||||
|
onChange={(e) => onPatch({ port: e.target.value ? Number(e.target.value) : undefined })}
|
||||||
|
/>
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Health check path</label>
|
||||||
|
<input className={FIELD_CLS} value={svc.health_check ?? ''} onChange={(e) => onPatch({ health_check: e.target.value || undefined })} placeholder="/health" />
|
||||||
|
</>
|
||||||
|
) : null}
|
||||||
|
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Environment (KEY=value, one per line)</label>
|
||||||
|
<textarea
|
||||||
|
className={`${FIELD_CLS} mono min-h-[70px] resize-y`}
|
||||||
|
defaultValue={envText}
|
||||||
|
spellCheck={false}
|
||||||
|
onBlur={(e) => {
|
||||||
|
const env: Record<string, string> = {};
|
||||||
|
for (const line of e.target.value.split('\n')) {
|
||||||
|
const idx = line.indexOf('=');
|
||||||
|
if (idx > 0) env[line.slice(0, idx).trim()] = line.slice(idx + 1).trim();
|
||||||
|
}
|
||||||
|
onPatch({ environment: Object.keys(env).length ? env : undefined });
|
||||||
|
}}
|
||||||
|
placeholder={'DATABASE_URL={{service.db.url}}\nSECRET={{secret}}'}
|
||||||
|
/>
|
||||||
|
|
||||||
|
<label className={`${LABEL_CLS} mt-3`}>Volumes (source:target[:ro], one per line)</label>
|
||||||
|
<textarea
|
||||||
|
className={`${FIELD_CLS} mono min-h-[50px] resize-y`}
|
||||||
|
defaultValue={volText}
|
||||||
|
spellCheck={false}
|
||||||
|
onBlur={(e) => {
|
||||||
|
const vols = e.target.value
|
||||||
|
.split('\n')
|
||||||
|
.map((l) => l.trim())
|
||||||
|
.filter(Boolean)
|
||||||
|
.map((l) => {
|
||||||
|
const [source, target, ro] = l.split(':');
|
||||||
|
return { source, target, read_only: ro === 'ro' || undefined };
|
||||||
|
})
|
||||||
|
.filter((v) => v.source && v.target);
|
||||||
|
onPatch({ volumes: vols.length ? vols : undefined });
|
||||||
|
}}
|
||||||
|
placeholder={'data:/var/lib/app\n./config:/etc/app:ro'}
|
||||||
|
/>
|
||||||
|
|
||||||
|
{(svc.depends_on ?? []).length > 0 ? (
|
||||||
|
<div className="mt-3">
|
||||||
|
<span className={LABEL_CLS}>Depends on</span>
|
||||||
|
<div className="flex flex-wrap gap-1">
|
||||||
|
{svc.depends_on!.map((dep) => (
|
||||||
|
<span key={dep} className="inline-flex items-center gap-1 rounded-full bg-[var(--surface-muted)] px-2 py-0.5 text-[11px] text-[var(--text-secondary)]">
|
||||||
|
{dep}
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
onClick={() => onPatch({ depends_on: (svc.depends_on ?? []).filter((d) => d !== dep) })}
|
||||||
|
className="text-[var(--text-tertiary)] hover:text-[var(--error)]"
|
||||||
|
>
|
||||||
|
<X size={10} />
|
||||||
|
</button>
|
||||||
|
</span>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
) : null}
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
@@ -0,0 +1,160 @@
|
|||||||
|
import { describe, expect, it } from 'vitest';
|
||||||
|
import {
|
||||||
|
canvasFromConfig,
|
||||||
|
configFromCanvas,
|
||||||
|
layoutPositions,
|
||||||
|
newServiceKey,
|
||||||
|
validateGraph,
|
||||||
|
type BuilderNode,
|
||||||
|
} from './builder-model';
|
||||||
|
|
||||||
|
const plausibleConfig = {
|
||||||
|
version: 2,
|
||||||
|
services: [
|
||||||
|
{ key: 'db', type: 'database', runtime: 'postgresql', environment: { POSTGRES_DB: 'plausible' } },
|
||||||
|
{ key: 'events', type: 'database', runtime: 'clickhouse' },
|
||||||
|
{
|
||||||
|
key: 'web',
|
||||||
|
name: 'plausible',
|
||||||
|
type: 'web',
|
||||||
|
runtime: 'ghcr.io/plausible/community-edition:v2',
|
||||||
|
port: 8000,
|
||||||
|
health_check: '/api/health',
|
||||||
|
depends_on: ['db', 'events'],
|
||||||
|
environment: { DATABASE_URL: '{{service.db.url}}' },
|
||||||
|
},
|
||||||
|
],
|
||||||
|
};
|
||||||
|
|
||||||
|
describe('canvasFromConfig', () => {
|
||||||
|
it('parses graph services into nodes with auto-layout', () => {
|
||||||
|
const nodes = canvasFromConfig(plausibleConfig);
|
||||||
|
expect(nodes.map((n) => n.key)).toEqual(['db', 'events', 'web']);
|
||||||
|
// databases are roots (depth 0), web depends on both (depth 1)
|
||||||
|
expect(nodes[0].position.x).toBeLessThan(nodes[2].position.x);
|
||||||
|
expect(nodes[1].position.y).toBeGreaterThan(nodes[0].position.y);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('restores saved canvas positions', () => {
|
||||||
|
const nodes = canvasFromConfig({
|
||||||
|
...plausibleConfig,
|
||||||
|
canvas: { positions: { db: { x: 500, y: 500 } } },
|
||||||
|
});
|
||||||
|
expect(nodes[0].position).toEqual({ x: 500, y: 500 });
|
||||||
|
});
|
||||||
|
|
||||||
|
it('migrates flat legacy configs to a single node', () => {
|
||||||
|
const nodes = canvasFromConfig({
|
||||||
|
type: 'web',
|
||||||
|
runtime: 'nginx:latest',
|
||||||
|
port: 80,
|
||||||
|
environment: { FOO: 'bar' },
|
||||||
|
});
|
||||||
|
expect(nodes).toHaveLength(1);
|
||||||
|
expect(nodes[0].service).toMatchObject({ key: 'app', type: 'web', runtime: 'nginx:latest', port: 80 });
|
||||||
|
});
|
||||||
|
|
||||||
|
it('drops entries without a key', () => {
|
||||||
|
const nodes = canvasFromConfig({ version: 2, services: [{ type: 'web', runtime: 'x' }, { key: 'ok', type: 'web', runtime: 'y' }] });
|
||||||
|
expect(nodes.map((n) => n.key)).toEqual(['ok']);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('configFromCanvas', () => {
|
||||||
|
it('round-trips a graph config', () => {
|
||||||
|
const nodes = canvasFromConfig(plausibleConfig);
|
||||||
|
const config = configFromCanvas(nodes, plausibleConfig);
|
||||||
|
const services = config.services as Record<string, unknown>[];
|
||||||
|
expect(config.version).toBe(2);
|
||||||
|
expect(services).toHaveLength(3);
|
||||||
|
const web = services.find((s) => s.key === 'web')!;
|
||||||
|
expect(web.depends_on).toEqual(['db', 'events']);
|
||||||
|
expect(web.port).toBe(8000);
|
||||||
|
expect((web.environment as Record<string, string>).DATABASE_URL).toBe('{{service.db.url}}');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('persists node positions under config.canvas', () => {
|
||||||
|
const nodes = canvasFromConfig(plausibleConfig);
|
||||||
|
nodes[0].position = { x: 321, y: 123 };
|
||||||
|
const config = configFromCanvas(nodes, {});
|
||||||
|
expect((config.canvas as { positions: Record<string, { x: number; y: number }> }).positions.db).toEqual({ x: 321, y: 123 });
|
||||||
|
// and parses back to the same spot
|
||||||
|
expect(canvasFromConfig(config)[0].position).toEqual({ x: 321, y: 123 });
|
||||||
|
});
|
||||||
|
|
||||||
|
it('strips flat fields when converting a legacy config', () => {
|
||||||
|
const nodes = canvasFromConfig({ type: 'web', runtime: 'nginx:latest', port: 80 });
|
||||||
|
const config = configFromCanvas(nodes, { type: 'web', runtime: 'nginx:latest', port: 80 });
|
||||||
|
expect(config.type).toBeUndefined();
|
||||||
|
expect(config.runtime).toBeUndefined();
|
||||||
|
expect((config.services as unknown[]).length).toBe(1);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('layoutPositions', () => {
|
||||||
|
it('depths follow the depends_on chain', () => {
|
||||||
|
const pos = layoutPositions([
|
||||||
|
{ key: 'a', type: 'database', runtime: 'postgresql' },
|
||||||
|
{ key: 'b', type: 'web', runtime: 'img', depends_on: ['a'] },
|
||||||
|
{ key: 'c', type: 'web', runtime: 'img', depends_on: ['b'] },
|
||||||
|
]);
|
||||||
|
expect(pos.a.x).toBeLessThan(pos.b.x);
|
||||||
|
expect(pos.b.x).toBeLessThan(pos.c.x);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('a dependency cycle does not hang', () => {
|
||||||
|
const pos = layoutPositions([
|
||||||
|
{ key: 'a', type: 'web', runtime: 'x', depends_on: ['b'] },
|
||||||
|
{ key: 'b', type: 'web', runtime: 'x', depends_on: ['a'] },
|
||||||
|
]);
|
||||||
|
expect(pos.a).toBeDefined();
|
||||||
|
expect(pos.b).toBeDefined();
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('validateGraph', () => {
|
||||||
|
const node = (key: string, extra: Partial<BuilderNode['service']> = {}): BuilderNode => ({
|
||||||
|
key,
|
||||||
|
position: { x: 0, y: 0 },
|
||||||
|
service: { key, type: 'web', runtime: 'img:1', ...extra },
|
||||||
|
});
|
||||||
|
|
||||||
|
it('accepts a valid graph', () => {
|
||||||
|
expect(validateGraph([node('db', { type: 'database', runtime: 'postgresql' }), node('web', { depends_on: ['db'] })])).toEqual([]);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('rejects duplicate keys', () => {
|
||||||
|
expect(validateGraph([node('a'), node('a')])).toContain('duplicate key "a"');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('rejects bad keys and self-deps', () => {
|
||||||
|
const errs = validateGraph([node('BAD KEY'), node('x', { depends_on: ['x'] })]);
|
||||||
|
expect(errs.some((e) => e.includes('not a valid key'))).toBe(true);
|
||||||
|
expect(errs).toContain('"x" depends on itself');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('rejects deps on missing services', () => {
|
||||||
|
expect(validateGraph([node('web', { depends_on: ['ghost'] })])).toContain(
|
||||||
|
'"web" depends on unknown service "ghost"',
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('rejects database without engine and web without source', () => {
|
||||||
|
const errs = validateGraph([node('db', { type: 'database', runtime: undefined }), node('web', { runtime: undefined })]);
|
||||||
|
expect(errs).toContain('"db" is a database with no engine selected');
|
||||||
|
expect(errs).toContain('"web" needs an image (runtime) or a git repo');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('reports dependency cycles', () => {
|
||||||
|
const errs = validateGraph([node('a', { depends_on: ['b'] }), node('b', { depends_on: ['a'] })]);
|
||||||
|
expect(errs.some((e) => e.startsWith('dependency cycle'))).toBe(true);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('newServiceKey', () => {
|
||||||
|
it('picks unused keys', () => {
|
||||||
|
expect(newServiceKey('database', new Set())).toBe('db');
|
||||||
|
expect(newServiceKey('database', new Set(['db']))).toBe('db2');
|
||||||
|
expect(newServiceKey('web', new Set(['web', 'web2']))).toBe('web3');
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -0,0 +1,241 @@
|
|||||||
|
// Pure model for the visual template builder: converts between the
|
||||||
|
// template `config` JSON (version 2 service graph, or a legacy flat
|
||||||
|
// single-service config) and canvas state (nodes + positions), and
|
||||||
|
// validates the graph before save. No React, no DOM — directly testable.
|
||||||
|
|
||||||
|
export type BuilderServiceType = 'web' | 'worker' | 'cron' | 'database';
|
||||||
|
|
||||||
|
export type BuilderService = {
|
||||||
|
key: string;
|
||||||
|
name?: string;
|
||||||
|
type: BuilderServiceType;
|
||||||
|
runtime?: string;
|
||||||
|
repo?: string;
|
||||||
|
branch?: string;
|
||||||
|
build_command?: string;
|
||||||
|
start_command?: string;
|
||||||
|
port?: number;
|
||||||
|
health_check?: string;
|
||||||
|
builder?: string;
|
||||||
|
environment?: Record<string, string>;
|
||||||
|
volumes?: { type?: string; source: string; target: string; read_only?: boolean }[];
|
||||||
|
depends_on?: string[];
|
||||||
|
};
|
||||||
|
|
||||||
|
export type BuilderCanvas = {
|
||||||
|
positions: Record<string, { x: number; y: number }>;
|
||||||
|
};
|
||||||
|
|
||||||
|
export type BuilderNode = {
|
||||||
|
key: string;
|
||||||
|
service: BuilderService;
|
||||||
|
position: { x: number; y: number };
|
||||||
|
};
|
||||||
|
|
||||||
|
const KEY_RE = /^[a-z][a-z0-9_-]*$/;
|
||||||
|
const COL_X = 60;
|
||||||
|
const ROW_Y = 200;
|
||||||
|
|
||||||
|
function asService(raw: unknown): BuilderService | null {
|
||||||
|
if (!raw || typeof raw !== 'object') return null;
|
||||||
|
const s = raw as Record<string, unknown>;
|
||||||
|
const key = typeof s.key === 'string' ? s.key : '';
|
||||||
|
if (!key) return null;
|
||||||
|
const type = typeof s.type === 'string' ? s.type : 'web';
|
||||||
|
return {
|
||||||
|
key,
|
||||||
|
name: typeof s.name === 'string' ? s.name : undefined,
|
||||||
|
type: (['web', 'worker', 'cron', 'database'] as const).includes(type as BuilderServiceType)
|
||||||
|
? (type as BuilderServiceType)
|
||||||
|
: 'web',
|
||||||
|
runtime: typeof s.runtime === 'string' ? s.runtime : undefined,
|
||||||
|
repo: typeof s.repo === 'string' ? s.repo : undefined,
|
||||||
|
branch: typeof s.branch === 'string' ? s.branch : undefined,
|
||||||
|
build_command: typeof s.build_command === 'string' ? s.build_command : undefined,
|
||||||
|
start_command: typeof s.start_command === 'string' ? s.start_command : undefined,
|
||||||
|
port: typeof s.port === 'number' ? s.port : undefined,
|
||||||
|
health_check: typeof s.health_check === 'string' ? s.health_check : undefined,
|
||||||
|
builder: typeof s.builder === 'string' ? s.builder : undefined,
|
||||||
|
environment:
|
||||||
|
s.environment && typeof s.environment === 'object' && !Array.isArray(s.environment)
|
||||||
|
? Object.fromEntries(
|
||||||
|
Object.entries(s.environment as Record<string, unknown>).map(([k, v]) => [k, String(v)]),
|
||||||
|
)
|
||||||
|
: undefined,
|
||||||
|
volumes: Array.isArray(s.volumes)
|
||||||
|
? (s.volumes
|
||||||
|
.map((v) => {
|
||||||
|
if (!v || typeof v !== 'object') return null;
|
||||||
|
const vol = v as Record<string, unknown>;
|
||||||
|
if (typeof vol.target !== 'string' || !vol.target) return null;
|
||||||
|
return {
|
||||||
|
type: typeof vol.type === 'string' ? vol.type : undefined,
|
||||||
|
source: typeof vol.source === 'string' ? vol.source : '',
|
||||||
|
target: vol.target,
|
||||||
|
read_only: typeof vol.read_only === 'boolean' ? vol.read_only : undefined,
|
||||||
|
};
|
||||||
|
})
|
||||||
|
.filter((v): v is NonNullable<typeof v> => v !== null)
|
||||||
|
)
|
||||||
|
: undefined,
|
||||||
|
depends_on: Array.isArray(s.depends_on)
|
||||||
|
? s.depends_on.filter((d): d is string => typeof d === 'string')
|
||||||
|
: undefined,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// Flat legacy configs ({type, runtime, port, …} with no services array)
|
||||||
|
// map to a single service node so the builder can edit them without
|
||||||
|
// losing fields the round-trip would drop.
|
||||||
|
function flatToService(config: Record<string, unknown>): BuilderService | null {
|
||||||
|
const type = typeof config.type === 'string' ? config.type : '';
|
||||||
|
const runtime = typeof config.runtime === 'string' ? config.runtime : '';
|
||||||
|
if (!type && !runtime) return null;
|
||||||
|
const svc = asService({
|
||||||
|
key: 'app',
|
||||||
|
type: type || 'web',
|
||||||
|
runtime,
|
||||||
|
port: config.port,
|
||||||
|
health_check: config.health_check,
|
||||||
|
build_command: config.build_command,
|
||||||
|
start_command: config.start_command,
|
||||||
|
environment: config.environment,
|
||||||
|
});
|
||||||
|
return svc;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Layered auto-layout: depth = longest depends_on chain; column per
|
||||||
|
// depth, row per member. Deterministic so tests can assert positions.
|
||||||
|
export function layoutPositions(services: BuilderService[]): Record<string, { x: number; y: number }> {
|
||||||
|
const depth = new Map<string, number>();
|
||||||
|
const visiting = new Set<string>();
|
||||||
|
const byKey = new Map(services.map((s) => [s.key, s]));
|
||||||
|
const depthOf = (key: string): number => {
|
||||||
|
const memo = depth.get(key);
|
||||||
|
if (memo !== undefined) return memo;
|
||||||
|
if (visiting.has(key)) return 0; // cycle guard
|
||||||
|
visiting.add(key);
|
||||||
|
const deps = byKey.get(key)?.depends_on ?? [];
|
||||||
|
let d = 0;
|
||||||
|
for (const dep of deps) {
|
||||||
|
if (byKey.has(dep)) d = Math.max(d, depthOf(dep) + 1);
|
||||||
|
}
|
||||||
|
visiting.delete(key);
|
||||||
|
depth.set(key, d);
|
||||||
|
return d;
|
||||||
|
};
|
||||||
|
const rows = new Map<number, number>();
|
||||||
|
const positions: Record<string, { x: number; y: number }> = {};
|
||||||
|
for (const svc of services) {
|
||||||
|
const d = depthOf(svc.key);
|
||||||
|
const row = rows.get(d) ?? 0;
|
||||||
|
rows.set(d, row + 1);
|
||||||
|
positions[svc.key] = { x: COL_X + d * 300, y: 40 + row * ROW_Y };
|
||||||
|
}
|
||||||
|
return positions;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function canvasFromConfig(config: Record<string, unknown>): BuilderNode[] {
|
||||||
|
const canvas = (config.canvas ?? {}) as Partial<BuilderCanvas>;
|
||||||
|
const saved = canvas.positions ?? {};
|
||||||
|
const rawServices = Array.isArray(config.services) ? config.services : null;
|
||||||
|
const services = rawServices
|
||||||
|
? rawServices.map(asService).filter((s): s is BuilderService => s !== null)
|
||||||
|
: [];
|
||||||
|
if (services.length === 0) {
|
||||||
|
const flat = flatToService(config);
|
||||||
|
if (flat) services.push(flat);
|
||||||
|
}
|
||||||
|
const auto = layoutPositions(services);
|
||||||
|
return services.map((service) => ({
|
||||||
|
key: service.key,
|
||||||
|
service,
|
||||||
|
position: saved[service.key] ?? auto[service.key] ?? { x: COL_X, y: 40 },
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
export function configFromCanvas(
|
||||||
|
nodes: BuilderNode[],
|
||||||
|
rest: Record<string, unknown>,
|
||||||
|
): Record<string, unknown> {
|
||||||
|
const services = nodes.map((node) => {
|
||||||
|
const svc = node.service;
|
||||||
|
const out: Record<string, unknown> = { key: svc.key, type: svc.type };
|
||||||
|
if (svc.name) out.name = svc.name;
|
||||||
|
if (svc.runtime) out.runtime = svc.runtime;
|
||||||
|
if (svc.repo) out.repo = svc.repo;
|
||||||
|
if (svc.branch) out.branch = svc.branch;
|
||||||
|
if (svc.build_command) out.build_command = svc.build_command;
|
||||||
|
if (svc.builder) out.builder = svc.builder;
|
||||||
|
if (svc.start_command) out.start_command = svc.start_command;
|
||||||
|
if (svc.port) out.port = svc.port;
|
||||||
|
if (svc.health_check) out.health_check = svc.health_check;
|
||||||
|
if (svc.environment && Object.keys(svc.environment).length > 0) out.environment = svc.environment;
|
||||||
|
if (svc.volumes && svc.volumes.length > 0) out.volumes = svc.volumes;
|
||||||
|
if (svc.depends_on && svc.depends_on.length > 0) out.depends_on = svc.depends_on;
|
||||||
|
return out;
|
||||||
|
});
|
||||||
|
const positions: Record<string, { x: number; y: number }> = {};
|
||||||
|
for (const node of nodes) positions[node.key] = node.position;
|
||||||
|
const restConfig = { ...rest };
|
||||||
|
// Flat (v1) fields and stale graph keys are dropped — the graph wins.
|
||||||
|
for (const k of ['type', 'runtime', 'build_command', 'start_command', 'port', 'health_check', 'environment', 'dockerfile', 'nixpacks_config', 'services', 'canvas']) {
|
||||||
|
delete restConfig[k];
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
version: 2,
|
||||||
|
...restConfig,
|
||||||
|
services,
|
||||||
|
canvas: { positions },
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
export function validateGraph(nodes: BuilderNode[]): string[] {
|
||||||
|
const errors: string[] = [];
|
||||||
|
const keys = new Set<string>();
|
||||||
|
for (const node of nodes) {
|
||||||
|
if (!KEY_RE.test(node.service.key)) {
|
||||||
|
errors.push(`"${node.service.key || '(empty)'}" is not a valid key (lowercase, digits, - or _)`);
|
||||||
|
}
|
||||||
|
if (keys.has(node.service.key)) errors.push(`duplicate key "${node.service.key}"`);
|
||||||
|
keys.add(node.service.key);
|
||||||
|
if (node.service.type === 'database' && !node.service.runtime) {
|
||||||
|
errors.push(`"${node.service.key}" is a database with no engine selected`);
|
||||||
|
}
|
||||||
|
if (node.service.type !== 'database' && !node.service.runtime && !node.service.repo) {
|
||||||
|
errors.push(`"${node.service.key}" needs an image (runtime) or a git repo`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for (const node of nodes) {
|
||||||
|
for (const dep of node.service.depends_on ?? []) {
|
||||||
|
if (dep === node.service.key) errors.push(`"${dep}" depends on itself`);
|
||||||
|
else if (!keys.has(dep)) errors.push(`"${node.service.key}" depends on unknown service "${dep}"`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Cycle detection — a cycle makes topo sort fail at deploy time.
|
||||||
|
const state = new Map<string, 0 | 1 | 2>();
|
||||||
|
const byKey = new Map(nodes.map((n) => [n.service.key, n]));
|
||||||
|
const visit = (key: string, trail: string[]): void => {
|
||||||
|
const s = state.get(key) ?? 0;
|
||||||
|
if (s === 2) return;
|
||||||
|
if (s === 1) {
|
||||||
|
errors.push(`dependency cycle: ${[...trail, key].join(' → ')}`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
state.set(key, 1);
|
||||||
|
for (const dep of byKey.get(key)?.service.depends_on ?? []) {
|
||||||
|
if (byKey.has(dep)) visit(dep, [...trail, key]);
|
||||||
|
}
|
||||||
|
state.set(key, 2);
|
||||||
|
};
|
||||||
|
for (const node of nodes) visit(node.service.key, []);
|
||||||
|
return errors;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function newServiceKey(type: BuilderServiceType, existing: Set<string>): string {
|
||||||
|
const base = type === 'database' ? 'db' : type;
|
||||||
|
if (!existing.has(base)) return base;
|
||||||
|
let i = 2;
|
||||||
|
while (existing.has(`${base}${i}`)) i += 1;
|
||||||
|
return `${base}${i}`;
|
||||||
|
}
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
import { useMemo, useState } from 'react';
|
import { lazy, Suspense, useMemo, useState } from 'react';
|
||||||
import { Link, useSearchParams } from 'react-router-dom';
|
import { Link, useSearchParams } from 'react-router-dom';
|
||||||
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
|
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
|
||||||
import {
|
import {
|
||||||
@@ -42,6 +42,10 @@ import {
|
|||||||
Network,
|
Network,
|
||||||
} from 'lucide-react';
|
} from 'lucide-react';
|
||||||
|
|
||||||
|
const TemplateBuilder = lazy(() =>
|
||||||
|
import('@/features/templates/builder/TemplateBuilder').then((m) => ({ default: m.TemplateBuilder })),
|
||||||
|
);
|
||||||
|
|
||||||
function toServiceName(value: string): string {
|
function toServiceName(value: string): string {
|
||||||
const normalized = value
|
const normalized = value
|
||||||
.toLowerCase()
|
.toLowerCase()
|
||||||
@@ -127,6 +131,7 @@ export function TemplatesPage() {
|
|||||||
const isAdmin = Boolean(profileQuery.data?.isAdmin);
|
const isAdmin = Boolean(profileQuery.data?.isAdmin);
|
||||||
|
|
||||||
const [editorOpen, setEditorOpen] = useState(false);
|
const [editorOpen, setEditorOpen] = useState(false);
|
||||||
|
const [editorMode, setEditorMode] = useState<'json' | 'visual'>('json');
|
||||||
const [editorJson, setEditorJson] = useState(EMPTY_TEMPLATE_JSON);
|
const [editorJson, setEditorJson] = useState(EMPTY_TEMPLATE_JSON);
|
||||||
const [editingTemplate, setEditingTemplate] = useState<TemplateEntity | null>(null);
|
const [editingTemplate, setEditingTemplate] = useState<TemplateEntity | null>(null);
|
||||||
const [editorIsPublic, setEditorIsPublic] = useState(false);
|
const [editorIsPublic, setEditorIsPublic] = useState(false);
|
||||||
@@ -1022,17 +1027,50 @@ export function TemplatesPage() {
|
|||||||
{/* Template editor — JSON only */}
|
{/* Template editor — JSON only */}
|
||||||
{editorOpen ? (
|
{editorOpen ? (
|
||||||
<div className="fixed inset-0 z-50 flex items-center justify-center bg-black/60 p-4 backdrop-blur-sm">
|
<div className="fixed inset-0 z-50 flex items-center justify-center bg-black/60 p-4 backdrop-blur-sm">
|
||||||
<div className="flex max-h-[90vh] w-full max-w-2xl flex-col overflow-hidden rounded-[var(--radius-lg)] border border-[var(--border-subtle)] bg-[var(--bg-base)] shadow-2xl">
|
<div className={`flex max-h-[92vh] w-full flex-col overflow-hidden rounded-[var(--radius-lg)] border border-[var(--border-subtle)] bg-[var(--bg-base)] shadow-2xl ${editorMode === 'visual' ? 'max-w-6xl' : 'max-w-2xl'}`}>
|
||||||
<div className="flex items-center justify-between border-b border-[var(--border-subtle)] px-5 py-4">
|
<div className="flex items-center justify-between border-b border-[var(--border-subtle)] px-5 py-4">
|
||||||
<div>
|
<div>
|
||||||
<h2 className="text-[15px] font-semibold text-[var(--text-primary)]">
|
<h2 className="text-[15px] font-semibold text-[var(--text-primary)]">
|
||||||
{editingTemplate ? `Edit ${editingTemplate.name}` : 'New template'}
|
{editingTemplate ? `Edit ${editingTemplate.name}` : 'New template'}
|
||||||
</h2>
|
</h2>
|
||||||
<p className="text-xs text-[var(--text-tertiary)]">
|
<p className="text-xs text-[var(--text-tertiary)]">
|
||||||
JSON definition — name, category, config and variables.
|
{editorMode === 'visual'
|
||||||
|
? 'Visual builder — drop service nodes, wire depends_on edges, apply back to JSON.'
|
||||||
|
: 'JSON definition — name, category, config and variables.'}
|
||||||
</p>
|
</p>
|
||||||
</div>
|
</div>
|
||||||
<div className="flex items-center gap-2">
|
<div className="flex items-center gap-2">
|
||||||
|
<div className="mr-1 inline-flex overflow-hidden rounded-[var(--radius-md)] border border-[var(--border-default)]">
|
||||||
|
{(['json', 'visual'] as const).map((mode) => (
|
||||||
|
<button
|
||||||
|
key={mode}
|
||||||
|
type="button"
|
||||||
|
onClick={() => {
|
||||||
|
if (mode === 'visual' && editorMode === 'json') {
|
||||||
|
try {
|
||||||
|
const parsed = JSON.parse(editorJson);
|
||||||
|
if (!parsed || typeof parsed.config !== 'object') {
|
||||||
|
setEditorError('Visual mode needs a valid JSON body with a "config" object first.');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
setEditorError('Fix the JSON before switching to visual mode.');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
setEditorError(null);
|
||||||
|
}
|
||||||
|
setEditorMode(mode);
|
||||||
|
}}
|
||||||
|
className={`px-3 py-1.5 text-[11px] font-medium capitalize ${
|
||||||
|
editorMode === mode
|
||||||
|
? 'bg-[var(--surface-muted)] text-[var(--text-primary)]'
|
||||||
|
: 'text-[var(--text-tertiary)] hover:text-[var(--text-primary)]'
|
||||||
|
}`}
|
||||||
|
>
|
||||||
|
{mode}
|
||||||
|
</button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
<label className="inline-flex cursor-pointer items-center gap-1.5 rounded-[var(--radius-md)] border border-[var(--border-default)] px-3 py-1.5 text-xs font-medium text-[var(--text-secondary)] hover:text-[var(--text-primary)]">
|
<label className="inline-flex cursor-pointer items-center gap-1.5 rounded-[var(--radius-md)] border border-[var(--border-default)] px-3 py-1.5 text-xs font-medium text-[var(--text-secondary)] hover:text-[var(--text-primary)]">
|
||||||
<Upload size={12} /> Upload .json
|
<Upload size={12} /> Upload .json
|
||||||
<input
|
<input
|
||||||
@@ -1055,12 +1093,49 @@ export function TemplatesPage() {
|
|||||||
</button>
|
</button>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<textarea
|
{editorMode === 'visual' ? (
|
||||||
value={editorJson}
|
<div className="flex min-h-[520px] flex-1 flex-col">
|
||||||
onChange={(e) => setEditorJson(e.target.value)}
|
<Suspense
|
||||||
spellCheck={false}
|
fallback={
|
||||||
className="mono min-h-[380px] flex-1 resize-none bg-[var(--bg-void)] p-4 text-[12.5px] leading-relaxed text-[var(--text-secondary)] outline-none"
|
<div className="flex flex-1 items-center justify-center text-[var(--text-tertiary)]">
|
||||||
/>
|
<Loader2 size={20} className="animate-spin" />
|
||||||
|
</div>
|
||||||
|
}
|
||||||
|
>
|
||||||
|
{(() => {
|
||||||
|
let cfg: Record<string, unknown> = {};
|
||||||
|
try {
|
||||||
|
const parsed = JSON.parse(editorJson) as { config?: Record<string, unknown> };
|
||||||
|
cfg = parsed?.config ?? {};
|
||||||
|
} catch {
|
||||||
|
cfg = {};
|
||||||
|
}
|
||||||
|
return (
|
||||||
|
<TemplateBuilder
|
||||||
|
config={cfg}
|
||||||
|
onCancel={() => setEditorMode('json')}
|
||||||
|
onApply={(newConfig) => {
|
||||||
|
try {
|
||||||
|
const parsed = JSON.parse(editorJson) as Record<string, unknown>;
|
||||||
|
setEditorJson(JSON.stringify({ ...parsed, config: newConfig }, null, 2));
|
||||||
|
} catch {
|
||||||
|
setEditorJson(JSON.stringify({ config: newConfig }, null, 2));
|
||||||
|
}
|
||||||
|
setEditorMode('json');
|
||||||
|
}}
|
||||||
|
/>
|
||||||
|
);
|
||||||
|
})()}
|
||||||
|
</Suspense>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<textarea
|
||||||
|
value={editorJson}
|
||||||
|
onChange={(e) => setEditorJson(e.target.value)}
|
||||||
|
spellCheck={false}
|
||||||
|
className="mono min-h-[380px] flex-1 resize-none bg-[var(--bg-void)] p-4 text-[12.5px] leading-relaxed text-[var(--text-secondary)] outline-none"
|
||||||
|
/>
|
||||||
|
)}
|
||||||
{composeWarnings.length > 0 ? (
|
{composeWarnings.length > 0 ? (
|
||||||
<div className="border-t border-[var(--border-subtle)] bg-[var(--warning-soft)]/60 px-5 py-2.5 text-xs text-[var(--warning)]">
|
<div className="border-t border-[var(--border-subtle)] bg-[var(--warning-soft)]/60 px-5 py-2.5 text-xs text-[var(--warning)]">
|
||||||
<p className="font-medium mb-1">Compose import notes</p>
|
<p className="font-medium mb-1">Compose import notes</p>
|
||||||
@@ -1074,7 +1149,7 @@ export function TemplatesPage() {
|
|||||||
{editorError}
|
{editorError}
|
||||||
</div>
|
</div>
|
||||||
) : null}
|
) : null}
|
||||||
<div className="flex items-center justify-between border-t border-[var(--border-subtle)] px-5 py-3.5">
|
<div className={`flex items-center justify-between border-t border-[var(--border-subtle)] px-5 py-3.5 ${editorMode === 'visual' ? 'hidden' : ''}`}>
|
||||||
<label className="inline-flex cursor-pointer items-center gap-2 text-[11px] text-[var(--text-secondary)]">
|
<label className="inline-flex cursor-pointer items-center gap-2 text-[11px] text-[var(--text-secondary)]">
|
||||||
<input
|
<input
|
||||||
type="checkbox"
|
type="checkbox"
|
||||||
|
|||||||
@@ -25,14 +25,22 @@ import {
|
|||||||
} from '@/lib/api-client';
|
} from '@/lib/api-client';
|
||||||
import { useToast } from '@/shared/hooks/use-toast';
|
import { useToast } from '@/shared/hooks/use-toast';
|
||||||
import { inferAutoConnections, type ServiceVariable } from '../auto-connections';
|
import { inferAutoConnections, type ServiceVariable } from '../auto-connections';
|
||||||
import {
|
import { DEFAULT_VIEWPORT, type ProjectCanvasMetadata } from '../model';
|
||||||
DEFAULT_VIEWPORT,
|
|
||||||
type CanvasGroup,
|
|
||||||
type CanvasNodeLayout,
|
|
||||||
type ProjectCanvasMetadata,
|
|
||||||
} from '../model';
|
|
||||||
import { hasCanvasMetadata, loadCanvasMetadata, saveCanvasMetadata } from '../storage';
|
import { hasCanvasMetadata, loadCanvasMetadata, saveCanvasMetadata } from '../storage';
|
||||||
import { computeLayeredLayout, layoutFromVariables, NODE_HEIGHT, NODE_WIDTH } from './auto-layout';
|
import { computeLayeredLayout, layoutFromVariables, NODE_HEIGHT, NODE_WIDTH } from './auto-layout';
|
||||||
|
import {
|
||||||
|
buildMetadataFromFlow,
|
||||||
|
GROUP_DEFAULT_HEIGHT,
|
||||||
|
GROUP_DEFAULT_WIDTH,
|
||||||
|
GROUP_PAD_BOTTOM,
|
||||||
|
GROUP_PAD_TOP,
|
||||||
|
GROUP_PAD_X,
|
||||||
|
GRID_STEP,
|
||||||
|
groupSlot,
|
||||||
|
resolveFreeSpot,
|
||||||
|
type CanvasNode,
|
||||||
|
type Rect,
|
||||||
|
} from './layout';
|
||||||
import { GroupNode, ServiceNode, type GroupNodeData, type ServiceNodeData } from './nodes';
|
import { GroupNode, ServiceNode, type GroupNodeData, type ServiceNodeData } from './nodes';
|
||||||
import { ServiceInspector, type InspectorGroup } from './inspector';
|
import { ServiceInspector, type InspectorGroup } from './inspector';
|
||||||
import { serviceAccent } from './service-visuals';
|
import { serviceAccent } from './service-visuals';
|
||||||
@@ -65,11 +73,7 @@ type CanvasProps = {
|
|||||||
|
|
||||||
type ServiceCanvasNode = Node<ServiceNodeData, 'serviceNode'>;
|
type ServiceCanvasNode = Node<ServiceNodeData, 'serviceNode'>;
|
||||||
type GroupCanvasNode = Node<GroupNodeData, 'groupNode'>;
|
type GroupCanvasNode = Node<GroupNodeData, 'groupNode'>;
|
||||||
type CanvasNode = ServiceCanvasNode | GroupCanvasNode;
|
|
||||||
type CanvasEdge = Edge<{ reasons: string[] }>;
|
type CanvasEdge = Edge<{ reasons: string[] }>;
|
||||||
|
|
||||||
const GROUP_DEFAULT_WIDTH = 340;
|
|
||||||
const GROUP_DEFAULT_HEIGHT = 230;
|
|
||||||
const TRANSIENT_STATUSES = new Set(['building', 'deploying', 'pending', 'rolling_back']);
|
const TRANSIENT_STATUSES = new Set(['building', 'deploying', 'pending', 'rolling_back']);
|
||||||
|
|
||||||
const nodeTypes: NodeTypes = {
|
const nodeTypes: NodeTypes = {
|
||||||
@@ -176,155 +180,6 @@ function toFlowEdges(
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
const GROUP_PAD_X = 28;
|
|
||||||
const GROUP_PAD_TOP = 56;
|
|
||||||
const GROUP_PAD_BOTTOM = 24;
|
|
||||||
const GROUP_ROW_GAP = 16;
|
|
||||||
const GRID_STEP = 26;
|
|
||||||
const NODE_GAP = 14;
|
|
||||||
|
|
||||||
type Rect = { x: number; y: number; w: number; h: number };
|
|
||||||
|
|
||||||
function rectsOverlap(a: Rect, b: Rect): boolean {
|
|
||||||
return (
|
|
||||||
a.x < b.x + b.w + NODE_GAP &&
|
|
||||||
a.x + a.w + NODE_GAP > b.x &&
|
|
||||||
a.y < b.y + b.h + NODE_GAP &&
|
|
||||||
a.y + a.h + NODE_GAP > b.y
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Nearest grid-aligned position where a node fits without touching any
|
|
||||||
// occupied rect. Spiral-searches outward on the canvas lattice; falls back to
|
|
||||||
// the given position when the neighbourhood is fully boxed in.
|
|
||||||
function resolveFreeSpot(
|
|
||||||
desired: { x: number; y: number },
|
|
||||||
occupied: Rect[],
|
|
||||||
fallback: { x: number; y: number },
|
|
||||||
bounds?: Rect,
|
|
||||||
): { x: number; y: number } {
|
|
||||||
const snap = (v: number) => Math.round(v / GRID_STEP) * GRID_STEP;
|
|
||||||
const base = { x: snap(desired.x), y: snap(desired.y) };
|
|
||||||
const fits = (x: number, y: number) => {
|
|
||||||
if (bounds && (x < bounds.x || y < bounds.y || x + NODE_WIDTH > bounds.x + bounds.w || y + NODE_HEIGHT > bounds.y + bounds.h)) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
return !occupied.some((rect) => rectsOverlap({ x, y, w: NODE_WIDTH, h: NODE_HEIGHT }, rect));
|
|
||||||
};
|
|
||||||
|
|
||||||
if (fits(base.x, base.y)) {
|
|
||||||
return base;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (let ring = 1; ring <= 12; ring++) {
|
|
||||||
for (let dx = -ring; dx <= ring; dx++) {
|
|
||||||
for (let dy = -ring; dy <= ring; dy++) {
|
|
||||||
if (Math.max(Math.abs(dx), Math.abs(dy)) !== ring) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
const x = base.x + dx * GRID_STEP;
|
|
||||||
const y = base.y + dy * GRID_STEP;
|
|
||||||
if (fits(x, y)) {
|
|
||||||
return { x, y };
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return fallback;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Where an incoming service lands inside a group frame. If the point already
|
|
||||||
// fits inside the padded area, honour it. Otherwise append beneath the lowest
|
|
||||||
// member and grow the frame just enough — keeps groups compact instead of
|
|
||||||
// ballooning the frame to wherever the node used to sit.
|
|
||||||
function groupSlot(
|
|
||||||
nodes: CanvasNode[],
|
|
||||||
groupId: string,
|
|
||||||
group: CanvasNode,
|
|
||||||
desiredX: number,
|
|
||||||
desiredY: number,
|
|
||||||
ignoreServiceId?: string,
|
|
||||||
): { x: number; y: number; width: number; height: number } {
|
|
||||||
const width = typeof group.style?.width === 'number' ? group.style.width : GROUP_DEFAULT_WIDTH;
|
|
||||||
const height = typeof group.style?.height === 'number' ? group.style.height : GROUP_DEFAULT_HEIGHT;
|
|
||||||
|
|
||||||
const members = nodes.filter(
|
|
||||||
(node) => node.type === 'serviceNode' && node.parentId === groupId && node.id !== ignoreServiceId,
|
|
||||||
);
|
|
||||||
const memberRects: Rect[] = members.map((node) => ({
|
|
||||||
x: node.position.x,
|
|
||||||
y: node.position.y,
|
|
||||||
w: NODE_WIDTH,
|
|
||||||
h: NODE_HEIGHT,
|
|
||||||
}));
|
|
||||||
const snap = (v: number) => Math.round(v / GRID_STEP) * GRID_STEP;
|
|
||||||
const snappedX = snap(desiredX);
|
|
||||||
const snappedY = snap(desiredY);
|
|
||||||
const snappedFits =
|
|
||||||
snappedX >= GROUP_PAD_X &&
|
|
||||||
snappedY >= GROUP_PAD_TOP &&
|
|
||||||
snappedX + NODE_WIDTH <= width - GROUP_PAD_X &&
|
|
||||||
snappedY + NODE_HEIGHT <= height - GROUP_PAD_BOTTOM;
|
|
||||||
const collides = memberRects.some((rect) =>
|
|
||||||
rectsOverlap({ x: snappedX, y: snappedY, w: NODE_WIDTH, h: NODE_HEIGHT }, rect),
|
|
||||||
);
|
|
||||||
|
|
||||||
if (snappedFits && !collides) {
|
|
||||||
return { x: snappedX, y: snappedY, width, height };
|
|
||||||
}
|
|
||||||
|
|
||||||
let bottom = GROUP_PAD_TOP - GROUP_ROW_GAP;
|
|
||||||
for (const node of nodes) {
|
|
||||||
if (node.type === 'serviceNode' && node.parentId === groupId && node.id !== ignoreServiceId) {
|
|
||||||
bottom = Math.max(bottom, node.position.y + NODE_HEIGHT);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const x = GROUP_PAD_X;
|
|
||||||
const y = bottom + GROUP_ROW_GAP;
|
|
||||||
return {
|
|
||||||
x,
|
|
||||||
y,
|
|
||||||
width: Math.max(width, x + NODE_WIDTH + GROUP_PAD_X),
|
|
||||||
height: Math.max(height, y + NODE_HEIGHT + GROUP_PAD_BOTTOM),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
function buildMetadataFromFlow(nodes: CanvasNode[], viewport: { x: number; y: number; zoom: number }): ProjectCanvasMetadata {
|
|
||||||
const groups: CanvasGroup[] = [];
|
|
||||||
const layouts: CanvasNodeLayout[] = [];
|
|
||||||
|
|
||||||
for (const node of nodes) {
|
|
||||||
if (node.type === 'groupNode') {
|
|
||||||
const width = typeof node.style?.width === 'number' ? node.style.width : GROUP_DEFAULT_WIDTH;
|
|
||||||
const height = typeof node.style?.height === 'number' ? node.style.height : GROUP_DEFAULT_HEIGHT;
|
|
||||||
groups.push({
|
|
||||||
id: node.id,
|
|
||||||
title: node.data.title,
|
|
||||||
position: node.position,
|
|
||||||
width,
|
|
||||||
height,
|
|
||||||
});
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (node.type === 'serviceNode') {
|
|
||||||
layouts.push({
|
|
||||||
serviceId: node.id,
|
|
||||||
position: node.position,
|
|
||||||
groupId: node.parentId,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return {
|
|
||||||
groups,
|
|
||||||
nodes: layouts,
|
|
||||||
edges: [],
|
|
||||||
viewport,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
function ZoomControls({ onTidy }: { onTidy: () => void }) {
|
function ZoomControls({ onTidy }: { onTidy: () => void }) {
|
||||||
const { zoom } = useViewport();
|
const { zoom } = useViewport();
|
||||||
|
|||||||
@@ -0,0 +1,142 @@
|
|||||||
|
import { describe, expect, it } from 'vitest';
|
||||||
|
import {
|
||||||
|
buildMetadataFromFlow,
|
||||||
|
groupSlot,
|
||||||
|
rectsOverlap,
|
||||||
|
resolveFreeSpot,
|
||||||
|
} from './layout';
|
||||||
|
import { NODE_HEIGHT, NODE_WIDTH } from './auto-layout';
|
||||||
|
import type { GroupNodeType, ServiceNodeType } from './nodes';
|
||||||
|
|
||||||
|
const GRID_STEP = 26;
|
||||||
|
const NODE_GAP = 14;
|
||||||
|
|
||||||
|
function serviceNode(
|
||||||
|
id: string,
|
||||||
|
position: { x: number; y: number },
|
||||||
|
parentId?: string,
|
||||||
|
): ServiceNodeType {
|
||||||
|
return { id, type: 'serviceNode', position, parentId } as ServiceNodeType;
|
||||||
|
}
|
||||||
|
|
||||||
|
function groupNode(
|
||||||
|
id: string,
|
||||||
|
width = 340,
|
||||||
|
height = 230,
|
||||||
|
): GroupNodeType {
|
||||||
|
return {
|
||||||
|
id,
|
||||||
|
type: 'groupNode',
|
||||||
|
position: { x: 0, y: 0 },
|
||||||
|
style: { width, height },
|
||||||
|
data: { title: `Group ${id}` },
|
||||||
|
} as GroupNodeType;
|
||||||
|
}
|
||||||
|
|
||||||
|
describe('rectsOverlap', () => {
|
||||||
|
const a = { x: 0, y: 0, w: 100, h: 100 };
|
||||||
|
|
||||||
|
it('detects intersection', () => {
|
||||||
|
expect(rectsOverlap(a, { x: 50, y: 50, w: 100, h: 100 })).toBe(true);
|
||||||
|
expect(rectsOverlap(a, { x: -50, y: -50, w: 100, h: 100 })).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('returns false for fully disjoint rects', () => {
|
||||||
|
expect(rectsOverlap(a, { x: 500, y: 500, w: 100, h: 100 })).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('treats the node gap as a collision zone', () => {
|
||||||
|
// b sits just inside NODE_GAP to the right of a.
|
||||||
|
expect(rectsOverlap(a, { x: 100 + NODE_GAP - 1, y: 0, w: 50, h: 50 })).toBe(true);
|
||||||
|
// Beyond the gap there is no overlap.
|
||||||
|
expect(rectsOverlap(a, { x: 100 + NODE_GAP + 1, y: 0, w: 50, h: 50 })).toBe(false);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('resolveFreeSpot', () => {
|
||||||
|
it('snaps the desired position to the grid when free', () => {
|
||||||
|
const spot = resolveFreeSpot({ x: 40, y: 70 }, [], { x: 0, y: 0 });
|
||||||
|
expect(spot).toEqual({ x: 52, y: 78 });
|
||||||
|
expect(spot.x % GRID_STEP).toBe(0);
|
||||||
|
expect(spot.y % GRID_STEP).toBe(0);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('finds a neighbouring free grid point when occupied', () => {
|
||||||
|
const occupied = [{ x: 0, y: 0, w: NODE_WIDTH, h: NODE_HEIGHT }];
|
||||||
|
const spot = resolveFreeSpot({ x: 0, y: 0 }, occupied, { x: -1, y: -1 });
|
||||||
|
expect(spot).not.toEqual({ x: -1, y: -1 });
|
||||||
|
expect(Number.isInteger(spot.x / GRID_STEP)).toBe(true);
|
||||||
|
expect(Number.isInteger(spot.y / GRID_STEP)).toBe(true);
|
||||||
|
expect(
|
||||||
|
rectsOverlap({ x: spot.x, y: spot.y, w: NODE_WIDTH, h: NODE_HEIGHT }, occupied[0]),
|
||||||
|
).toBe(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('falls back when bounds reject every candidate', () => {
|
||||||
|
// Node is wider than the bounds, so nothing can fit.
|
||||||
|
const bounds = { x: 0, y: 0, w: 10, h: 10 };
|
||||||
|
const fallback = { x: 999, y: 888 };
|
||||||
|
expect(resolveFreeSpot({ x: 0, y: 0 }, [], fallback, bounds)).toEqual(fallback);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('groupSlot', () => {
|
||||||
|
it('honours a point that fits inside the padded frame', () => {
|
||||||
|
const group = groupNode('g1');
|
||||||
|
const slot = groupSlot([group], 'g1', group, 40, 70);
|
||||||
|
expect(slot).toEqual({ x: 52, y: 78, width: 340, height: 230 });
|
||||||
|
});
|
||||||
|
|
||||||
|
it('appends below the lowest member and grows the frame on collision', () => {
|
||||||
|
const group = groupNode('g1');
|
||||||
|
const member = serviceNode('s1', { x: 52, y: 78 }, 'g1');
|
||||||
|
const slot = groupSlot([group, member], 'g1', group, 40, 70);
|
||||||
|
// bottom = 78 + 104 = 182; y = 182 + 16 = 198; grows height to 326.
|
||||||
|
expect(slot.x).toBe(28);
|
||||||
|
expect(slot.y).toBe(198);
|
||||||
|
expect(slot.height).toBe(326);
|
||||||
|
expect(slot.width).toBe(340);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('appends under the padding when the desired point is outside', () => {
|
||||||
|
const group = groupNode('g1');
|
||||||
|
const slot = groupSlot([group], 'g1', group, 5000, 5000);
|
||||||
|
expect(slot).toEqual({ x: 28, y: 56, width: 340, height: 230 });
|
||||||
|
});
|
||||||
|
|
||||||
|
it('ignores the dragged node when checking collisions', () => {
|
||||||
|
const group = groupNode('g1');
|
||||||
|
const dragged = serviceNode('s1', { x: 52, y: 78 }, 'g1');
|
||||||
|
const slot = groupSlot([group, dragged], 'g1', group, 40, 70, 's1');
|
||||||
|
expect(slot).toEqual({ x: 52, y: 78, width: 340, height: 230 });
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('buildMetadataFromFlow', () => {
|
||||||
|
it('serializes groups and service layouts', () => {
|
||||||
|
const nodes = [
|
||||||
|
{ ...groupNode('g1', 400, 300), position: { x: 10, y: 20 } },
|
||||||
|
serviceNode('s1', { x: 5, y: 6 }, 'g1'),
|
||||||
|
serviceNode('s2', { x: 100, y: 200 }),
|
||||||
|
];
|
||||||
|
const viewport = { x: 1, y: 2, zoom: 0.5 };
|
||||||
|
const meta = buildMetadataFromFlow(nodes, viewport);
|
||||||
|
|
||||||
|
expect(meta.groups).toEqual([
|
||||||
|
{ id: 'g1', title: 'Group g1', position: { x: 10, y: 20 }, width: 400, height: 300 },
|
||||||
|
]);
|
||||||
|
expect(meta.nodes).toEqual([
|
||||||
|
{ serviceId: 's1', position: { x: 5, y: 6 }, groupId: 'g1' },
|
||||||
|
{ serviceId: 's2', position: { x: 100, y: 200 }, groupId: undefined },
|
||||||
|
]);
|
||||||
|
expect(meta.edges).toEqual([]);
|
||||||
|
expect(meta.viewport).toEqual(viewport);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('uses default group dimensions when style is missing', () => {
|
||||||
|
const bare = { id: 'g2', type: 'groupNode', position: { x: 0, y: 0 }, data: { title: 'bare' } } as GroupNodeType;
|
||||||
|
const meta = buildMetadataFromFlow([bare], { x: 0, y: 0, zoom: 1 });
|
||||||
|
expect(meta.groups[0].width).toBe(340);
|
||||||
|
expect(meta.groups[0].height).toBe(230);
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -0,0 +1,156 @@
|
|||||||
|
import type { CanvasGroup, CanvasNodeLayout, ProjectCanvasMetadata } from '../model';
|
||||||
|
import { NODE_HEIGHT, NODE_WIDTH } from './auto-layout';
|
||||||
|
import type { GroupNodeType, ServiceNodeType } from './nodes';
|
||||||
|
|
||||||
|
export type CanvasNode = ServiceNodeType | GroupNodeType;
|
||||||
|
export type Rect = { x: number; y: number; w: number; h: number };
|
||||||
|
|
||||||
|
export const GROUP_DEFAULT_WIDTH = 340;
|
||||||
|
export const GROUP_DEFAULT_HEIGHT = 230;
|
||||||
|
export const GROUP_PAD_X = 28;
|
||||||
|
export const GROUP_PAD_TOP = 56;
|
||||||
|
export const GROUP_PAD_BOTTOM = 24;
|
||||||
|
export const GROUP_ROW_GAP = 16;
|
||||||
|
export const GRID_STEP = 26;
|
||||||
|
export const NODE_GAP = 14;
|
||||||
|
|
||||||
|
export function rectsOverlap(a: Rect, b: Rect): boolean {
|
||||||
|
return (
|
||||||
|
a.x < b.x + b.w + NODE_GAP &&
|
||||||
|
a.x + a.w + NODE_GAP > b.x &&
|
||||||
|
a.y < b.y + b.h + NODE_GAP &&
|
||||||
|
a.y + a.h + NODE_GAP > b.y
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Nearest grid-aligned position where a node fits without touching any
|
||||||
|
// occupied rect. Spiral-searches outward on the canvas lattice; falls back to
|
||||||
|
// the given position when the neighbourhood is fully boxed in.
|
||||||
|
export function resolveFreeSpot(
|
||||||
|
desired: { x: number; y: number },
|
||||||
|
occupied: Rect[],
|
||||||
|
fallback: { x: number; y: number },
|
||||||
|
bounds?: Rect,
|
||||||
|
): { x: number; y: number } {
|
||||||
|
const snap = (v: number) => Math.round(v / GRID_STEP) * GRID_STEP;
|
||||||
|
const base = { x: snap(desired.x), y: snap(desired.y) };
|
||||||
|
const fits = (x: number, y: number) => {
|
||||||
|
if (bounds && (x < bounds.x || y < bounds.y || x + NODE_WIDTH > bounds.x + bounds.w || y + NODE_HEIGHT > bounds.y + bounds.h)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return !occupied.some((rect) => rectsOverlap({ x, y, w: NODE_WIDTH, h: NODE_HEIGHT }, rect));
|
||||||
|
};
|
||||||
|
|
||||||
|
if (fits(base.x, base.y)) {
|
||||||
|
return base;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (let ring = 1; ring <= 12; ring++) {
|
||||||
|
for (let dx = -ring; dx <= ring; dx++) {
|
||||||
|
for (let dy = -ring; dy <= ring; dy++) {
|
||||||
|
if (Math.max(Math.abs(dx), Math.abs(dy)) !== ring) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
const x = base.x + dx * GRID_STEP;
|
||||||
|
const y = base.y + dy * GRID_STEP;
|
||||||
|
if (fits(x, y)) {
|
||||||
|
return { x, y };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return fallback;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Where an incoming service lands inside a group frame. If the point already
|
||||||
|
// fits inside the padded area, honour it. Otherwise append beneath the lowest
|
||||||
|
// member and grow the frame just enough — keeps groups compact instead of
|
||||||
|
// ballooning the frame to wherever the node used to sit.
|
||||||
|
export function groupSlot(
|
||||||
|
nodes: CanvasNode[],
|
||||||
|
groupId: string,
|
||||||
|
group: CanvasNode,
|
||||||
|
desiredX: number,
|
||||||
|
desiredY: number,
|
||||||
|
ignoreServiceId?: string,
|
||||||
|
): { x: number; y: number; width: number; height: number } {
|
||||||
|
const width = typeof group.style?.width === 'number' ? group.style.width : GROUP_DEFAULT_WIDTH;
|
||||||
|
const height = typeof group.style?.height === 'number' ? group.style.height : GROUP_DEFAULT_HEIGHT;
|
||||||
|
|
||||||
|
const members = nodes.filter(
|
||||||
|
(node) => node.type === 'serviceNode' && node.parentId === groupId && node.id !== ignoreServiceId,
|
||||||
|
);
|
||||||
|
const memberRects: Rect[] = members.map((node) => ({
|
||||||
|
x: node.position.x,
|
||||||
|
y: node.position.y,
|
||||||
|
w: NODE_WIDTH,
|
||||||
|
h: NODE_HEIGHT,
|
||||||
|
}));
|
||||||
|
const snap = (v: number) => Math.round(v / GRID_STEP) * GRID_STEP;
|
||||||
|
const snappedX = snap(desiredX);
|
||||||
|
const snappedY = snap(desiredY);
|
||||||
|
const snappedFits =
|
||||||
|
snappedX >= GROUP_PAD_X &&
|
||||||
|
snappedY >= GROUP_PAD_TOP &&
|
||||||
|
snappedX + NODE_WIDTH <= width - GROUP_PAD_X &&
|
||||||
|
snappedY + NODE_HEIGHT <= height - GROUP_PAD_BOTTOM;
|
||||||
|
const collides = memberRects.some((rect) =>
|
||||||
|
rectsOverlap({ x: snappedX, y: snappedY, w: NODE_WIDTH, h: NODE_HEIGHT }, rect),
|
||||||
|
);
|
||||||
|
|
||||||
|
if (snappedFits && !collides) {
|
||||||
|
return { x: snappedX, y: snappedY, width, height };
|
||||||
|
}
|
||||||
|
|
||||||
|
let bottom = GROUP_PAD_TOP - GROUP_ROW_GAP;
|
||||||
|
for (const node of nodes) {
|
||||||
|
if (node.type === 'serviceNode' && node.parentId === groupId && node.id !== ignoreServiceId) {
|
||||||
|
bottom = Math.max(bottom, node.position.y + NODE_HEIGHT);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const x = GROUP_PAD_X;
|
||||||
|
const y = bottom + GROUP_ROW_GAP;
|
||||||
|
return {
|
||||||
|
x,
|
||||||
|
y,
|
||||||
|
width: Math.max(width, x + NODE_WIDTH + GROUP_PAD_X),
|
||||||
|
height: Math.max(height, y + NODE_HEIGHT + GROUP_PAD_BOTTOM),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
export function buildMetadataFromFlow(nodes: CanvasNode[], viewport: { x: number; y: number; zoom: number }): ProjectCanvasMetadata {
|
||||||
|
const groups: CanvasGroup[] = [];
|
||||||
|
const layouts: CanvasNodeLayout[] = [];
|
||||||
|
|
||||||
|
for (const node of nodes) {
|
||||||
|
if (node.type === 'groupNode') {
|
||||||
|
const width = typeof node.style?.width === 'number' ? node.style.width : GROUP_DEFAULT_WIDTH;
|
||||||
|
const height = typeof node.style?.height === 'number' ? node.style.height : GROUP_DEFAULT_HEIGHT;
|
||||||
|
groups.push({
|
||||||
|
id: node.id,
|
||||||
|
title: node.data.title,
|
||||||
|
position: node.position,
|
||||||
|
width,
|
||||||
|
height,
|
||||||
|
});
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (node.type === 'serviceNode') {
|
||||||
|
layouts.push({
|
||||||
|
serviceId: node.id,
|
||||||
|
position: node.position,
|
||||||
|
groupId: node.parentId,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
groups,
|
||||||
|
nodes: layouts,
|
||||||
|
edges: [],
|
||||||
|
viewport,
|
||||||
|
};
|
||||||
|
}
|
||||||
Reference in new issue
Block a user