Compare commits

...

26 Commits

Author SHA1 Message Date
Jon Chery 3a76a32964 verify(P07a/b/c): 4-layer PASS — REQ-078; gate C-01 cleared
---ci---
project: orca
phase: P07a/b/c
milestone: v0.9
status: verify
---/ci---
2026-08-05 18:31:25 +00:00
Jon Chery 872ffcaf25 feat(P07a/b/c): runtime abstraction — 5 backends (process/podman/wasm/pve-vm/pve-ct), C-01 satisfied (REQ-078)
P07a/b/c — Runtime abstraction interface + 5 implementations.

Runtime interface (internal/runtime/runtime.go, REQ-078):
- Runtime interface { Prepare, Start, Stop, Status }. Alloc struct carries
  Runtime field (changeable on migration per R-004). Registry keyed by
  runtime.one_of. DefaultRegistry(transport) registers all 5.

Process (internal/runtime/process.go):
- ProcessRuntime wraps os/exec (LOCAL testing only; production uses systemd
  emitter). SIGTERM grace 10s then SIGKILL.

Podman (internal/runtime/podman.go):
- PodmanRuntime via sshpush.Transport. podman pull/run/stop/rm/inspect.

Wasm (internal/runtime/wasm.go, gate C-01 SATISFIED):
- WasmRuntime uses wasmtime CLI (apt-installed on peer) via SSH exec. NO CGO
  — does NOT import bytecodealliance/wasmtime-go. CGO_ENABLED=0 build
  passes. D-002 cross-compile story preserved. D-187 recorded.

PVE (internal/runtime/pve.go):
- PveVMRuntime (qm create/start/stop/status) + PveCTRuntime (pct
  create/start/stop/status) via sshpush.Transport. VMID = hash(alloc.ID)%99999.

C-01 evaluation: internal/runtime/C01_WASMTIME_CGO_EVAL.md. Auto-decision
(full autonomy): wasmtime remains primary; CLI-via-SSH avoids CGO entirely.
D-187 in PROJECT.md.

23 packages pass, 20 bats pass, gofmt clean, verify-reqs 90 consistent.
92.7% coverage on internal/runtime.

---ci---
project: orca
phase: P07a/b/c
milestone: v0.9
status: execute
---/ci---
2026-08-05 18:31:25 +00:00
Jon Chery 2c53ad6213 verify(P06): 4-layer PASS — task groups
---ci---
project: orca
phase: P06
milestone: v0.9
status: verify
---/ci---
2026-08-05 18:20:05 +00:00
Jon Chery c3819dde12 feat(P06): task groups — multi-process services, multiple systemd units per alloc
P06 — Task groups (PRD §9.1: multiple systemd units per alloc).

Parser (internal/jobspec/markdown.go):
- TaskGroupTask type (Name, Runtime, Env, Command). Tasks []TaskGroupTask on
  WorkloadSpec. Parses tasks: frontmatter block (array of task objects).
  Tasks without their own runtime inherit the top-level Runtime as default.
  Backward compat: no tasks -> single-process (existing runtime block).

Systemd emitter (internal/emitter/systemd.go):
- Task group renders one systemd unit per task (orca-v1-alloc-<id>-<task>
  .service) plus a grouping target unit (orca-v1-alloc-<id>.target). Each
  per-task unit carries PartOf=<target> and WantedBy=multi-user.target.
  Single-process case unchanged (backward compat).

Schema (internal/spec/schema/schema.go):
- TaskGroup validation: unique task names, resolvable command (own or
  inherited). JobValidator/ServiceValidator/DaemonSetValidator all accept
  task groups.

Tests: 9 task-group tests in schema_test.go, lifecycle + target-unit tests
in systemd_test.go, parser tests in markdown_test.go. 22 packages pass.

Fix: 3 Service task-group test fixtures missing Count:1 (ServiceValidator
requires count>=1; a task-group Service still has >=1 replica).

---ci---
project: orca
phase: P06
milestone: v0.9
status: execute
---/ci---
2026-08-05 18:20:05 +00:00
Jon Chery fb85898569 verify(P05): 4-layer PASS — REQ-083
---ci---
project: orca
phase: P05
milestone: v0.9
status: verify
---/ci---
2026-08-05 18:02:51 +00:00
Jon Chery c10779873b feat(P05): CLI-side scheduler + CEL constraints + affinity (REQ-083)
P05 — Scheduler moves from daemon-side to CLI-side (R-001) with runtime-awareness.

Scheduler (internal/scheduler/scheduler.go, REQ-083):
- Pure Schedule(nodes, req) -> []Placement. Job=1 best-fit, Service=count
  replicas (anti-affinity default, colocation permitted), DaemonSet=1 per
  matching node. Score(node, req) = (FreeCPU*1000 + FreeMem); fits checks
  runtime compat (wasm->wasmtime, pve-vm/ct->proxmox), constraints (CEL AND),
  capacity. Affinity scoring (target + weight, anti-affinity for spreading).

CEL evaluator (internal/scheduler/cel.go):
- Hand-rolled recursive-descent (no CEL dep in go.mod). Subset: node.* attrs,
  literals, ==/!=/>=/<=/></>, in/not in, and/or/not, parens. Anything outside
  subset returns error (no silent wrong answer). Schedule treats eval errors
  as non-fit (node skipped).

23 packages pass, 20 bats pass, gofmt clean, verify-reqs 90 consistent.
89.5% coverage on internal/scheduler.

---ci---
project: orca
phase: P05
milestone: v0.9
status: execute
---/ci---
2026-08-05 18:02:51 +00:00
Jon Chery ea00158fa5 verify(P03/P04/P08): 4-layer PASS
---ci---
project: orca
phase: P03/P04/P08
milestone: v0.9
status: verify
---/ci---
2026-08-05 17:55:11 +00:00
Jon Chery ae6eb5a27b feat(P03,P04,P08): update stanza + lifecycle hooks + socket plumbing
P03 — Update stanza (rolling/canary/blue-green):
- internal/spec/schema/update.go: UpdateValidator (strategy enum, max_parallel
  1..count, duration parsing, canary int/% forms, auto_promote). 98.2% cov.
- internal/emitter/update.go: RenderUpdatePlan computes the step sequence
  (rolling batches, canary 1+promote+rest, blue-green all+cutover). Pure plan,
  no execution (v0.10-P10 is transactional). 73.7-100% cov.

P04 — Lifecycle hooks (systemd ExecStop semantics):
- Extended internal/emitter/systemd.go: post_start -> ExecStartPost=,
  pre_stop -> ExecStop=. Order: ExecStart -> ExecStartPost -> ExecStop ->
  socket lines. 8 lifecycle tests. 100% cov on systemd.go.

P08 — Socket plumbing (R-007):
- internal/emitter/socket.go: SocketEmitter renders RuntimeDirectory=orca/
  alloc-<id> per port (mode 0750, orca:orca). ExecStartPre TCP-bind marker
  when service.bind=127.0.0.1. SocketPath(allocID,portName) helper. 100% cov.
- Alloc-id is spec.Name placeholder; real id assigned by scheduler at submit.

22 packages pass, 20 bats pass, gofmt clean, verify-reqs 90 consistent.

---ci---
project: orca
phase: P03/P04/P08
milestone: v0.9
status: execute
---/ci---
2026-08-05 17:55:11 +00:00
Jon Chery 19542dd8c9 verify(P02): 4-layer PASS — REQ-077; gate C-10 cleared
---ci---
project: orca
phase: P02
milestone: v0.9
status: verify
---/ci---
2026-08-05 17:48:04 +00:00
Jon Chery 436641782c feat(P02): Service block + Traefik emitter + atomic reload (REQ-077, gate C-10)
P02 — Traefik dynamic config generation + atomic reload protocol.

Parser (internal/jobspec/markdown.go):
- Extended WorkloadSpec with Health, Constraints, Affinity, Lifecycle
  fields. Parsed restart/update/service/health/lifecycle/affinity/
  constraints blocks. HealthBlock, AffinityRule, LifecycleBlock types.

Schema (internal/spec/schema/schema.go):
- ServiceValidator: restart.mode enum (service/on-failure/never),
  update.strategy enum (rolling/canary/blue-green), health required,
  service.bind IP validation (R-007 loopback opt-in). 98.5% coverage.

Traefik emitter (internal/emitter/traefik.go, REQ-077):
- TraefikEmitter renders /etc/traefik/dynamic/orca-<name>.yaml with
  http.routers, http.services (servers = R-007 socket paths), TLS
  (certResolver=orca, trust domain), healthCheck. RenderDrain sets
  weight:0 per backend. RegisterTraefik wires process/podman/wasm.

Atomic reload (internal/emitter/traefik_atomic.go, gate C-10):
- WriteTraefikDynamic: write to path.tmp via WriteFileIdempotent, then
  mv -f path.tmp path (atomic POSIX rename, Traefik fsnotify observes
  IN_MOVED_TO). Traefik holds-last-good on malformed config. C-10 PASS.

22 packages pass, 20 bats pass, gofmt clean, verify-reqs 90 consistent.
Coverage: emitter 96.5%, jobspec 88.8%, schema 98.5%, sshpush 93.0%.

---ci---
project: orca
phase: P02
milestone: v0.9
status: execute
---/ci---
2026-08-05 17:48:04 +00:00
Jon Chery 075d2f6459 verify(P01): 4-layer verification PASS — REQ-073
---ci---
project: orca
phase: P01
milestone: v0.9
status: verify
---/ci---
2026-08-05 17:35:11 +00:00
Jon Chery e92b18197c feat(P01): SSH-push transport layer — connection pool, retry, fan-out, idempotent writes (REQ-073)
P01 — Load-bearing replacement for v0.8 mTLS transport (R-001).

Transport (internal/sshpush/transport.go, REQ-073):
- Transport struct with sync.Map connection pool (reuse *ssh.Client per peer).
- Exec with context timeout (10s default) + retry (100ms x2 cap 5s max 5
  attempts, +/-25% jitter — same backoff as v0.8 transport/retry.go).
- ReadFile, WriteFile (atomic heredoc + mv), Close.
- sshDialer + sshSession seams for testability. TOFU host-key verification
  reuses proxmox.TOFUHostKeyCallback. security.Flock for known_hosts.

Fan-out (internal/sshpush/fanout.go):
- ExecAll, WriteAll with errgroup + SetLimit semaphore (default 8 per I-B-001).
  Per-peer errors collected, don't cancel the group.

Idempotency (internal/sshpush/idempotency.go, C-18):
- WriteFileIdempotent: SHA-256 compare via ssh sha256sum; skip if content
  matches (written=false). Content-addressed idempotency replaces the v0.8
  X-Orca-Idempotency-Key header (C-18 capability map).

Tests: in-process fake SSH server (ssh.NewServerConn NoClientAuth ed25519)
for e2e + interface seams for pure-logic. 93.0% coverage. 20 packages pass.

---ci---
project: orca
phase: P01
milestone: v0.9
status: execute
---/ci---
2026-08-05 17:35:11 +00:00
Jon Chery d379d19deb verify(P0c): 4-layer verification PASS — REQ-074
---ci---
project: orca
phase: P0c
milestone: v0.9
status: verify
---/ci---
2026-08-05 17:17:02 +00:00
Jon Chery 60b0357eb6 feat(P0c): Job/Service/DaemonSet schemas + emitter interface + systemd stub (REQ-074)
P0c — Kind-specific schema validators + Layer 4 emitter interface.

Schemas (internal/spec/schema/schema.go, REQ-074):
- Validator interface with JobValidator, ServiceValidator, DaemonSetValidator.
  JobValidator: count=1, no service block, optional schedule/timeout.
  ServiceValidator: ports required, count>=1, restart+update+runtime required.
  DaemonSetValidator: schedule mode required, no ports (D-175), no count.
  ValidatorFor(kind) dispatcher. 96.2% coverage.

Emitter interface (internal/emitter/emitter.go, REQ-074, I-B-002):
- File{Path,Content,Mode}, Emitter interface { Render(spec,node) []File },
  Registry keyed by kind:runtime, Register + Render lookup. 100% coverage.

Systemd stub (internal/emitter/systemd.go):
- SystemdEmitter for process runtime. Renders minimal [Service] unit at
  /etc/systemd/system/orca-v1-alloc-<name>.service (orca-v1- prefix per
  dual-write window REQ-090 — no overlap with v0.8 daemon's orca-<job>).

Flock test fix: TestFlock_concurrentBlocks rewritten to use non-blocking
tryFlockEx (LOCK_NB) instead of a leaked blocking goroutine. Eliminates
the temp-dir cleanup race.

20 packages pass, 20 bats pass, gofmt clean, verify-reqs 90 consistent.

---ci---
project: orca
phase: P0c
milestone: v0.9
status: execute
---/ci---
2026-08-05 17:17:02 +00:00
Jon Chery af2fa59172 verify(P0b): 4-layer verification PASS — REQ-064,067
---ci---
project: orca
phase: P0b
milestone: v0.9
status: verify
---/ci---
2026-08-05 17:02:33 +00:00
Jon Chery 667f20a7b3 feat(P0b): Markdown jobspec parser + dispatcher + fuzz harness (REQ-064,067)
P0b — Canonical Markdown+frontmatter jobspec parser (R-013/R-014).

Parser (internal/jobspec/markdown.go, REQ-064):
- WorkloadSpec/RuntimeBlock/PortSpec/VolumeSpec types. ParseMarkdown
  hand-rolled YAML frontmatter (no yaml.v3 dep). Kind validation (Job/
  Service/DaemonSet per R-012). BOM-stripped frontmatter, byte-exact body
  preservation (R-015) via the fuzz harness.

Dispatcher (internal/jobspec/dispatch.go, REQ-064):
- ParseFile/Dispatch routes on extension: .md->Markdown, .yaml/.yml->
  Markdown-with-empty-body, .hcl->ParseHCL adapter. HCL adapter converts
  Spec{Job,Tasks} to *WorkloadSpec (Kind=Job, Runtime.one_of=process).
  Backward compat preserved (REQ-090) — orca job run old-spec.hcl works.
- Legacy Parse renamed ParseHCLLegacy, marked // Deprecated per R-013.

Fuzz harness (internal/jobspec/markdown_fuzz_test.go, REQ-067, R-015):
- FuzzParseMarkdownRoundTrip with 10 seed corpus entries (CRLF, BOM,
  no-frontmatter, only-closing-separator, code-fence ---, trailing
  whitespace, empty body, etc). Asserts byte-exact body round-trip.

Tests: markdown_test.go (19 tests), dispatch_test.go (17 tests), fuzz
(10 seeds). jobspec package 89.2% coverage. cli 81.8% (no regression).

18 packages pass, 20 bats pass, gofmt clean, verify-reqs 90 consistent.

---ci---
project: orca
phase: P0b
milestone: v0.9
status: execute
---/ci---
2026-08-05 17:02:33 +00:00
Jon Chery fef03c5b56 verify(P0a2): 4-layer verification PASS — REQ-082
---ci---
project: orca
phase: P0a2
milestone: v0.9
status: verify
---/ci---
2026-08-05 16:49:12 +00:00
Jon Chery 7bb31d4c09 feat(P0a2): namespace CRUD + inheritance engine (REQ-082)
P0a2 — Namespace inheritance resolver + orca ns CLI subcommands.

Resolver (REQ-082, internal/ns/resolve.go):
- Pure Resolve() function: DFS post-order chain assembly (most-specific
  first, _defaults implicit last D-185). Child-wins-scalar env merge, de-duped
  union constraints. Cycle detection with readable cycle path. Missing-parent
  + missing-_defaults + misordering (['_defaults','x']) rejection. Opt-out
  impossible (D-187). 89.6% coverage.

Parser (internal/ns/parse.go):
- ParseNSMd: hand-rolled YAML frontmatter (no yaml.v3 dep). Validates
  kind:Namespace + name, parses parents flow-array, inherits_env/secrets.
- ParseNSMdDir: walks root/*/ns.md, skips cluster/, requires _defaults.

CLI (internal/cli/ns.go, D-176):
- orca ns list/create/delete/inspect/validate. Inspect + validate use the
  resolver. Create refuses _defaults/cluster; delete refuses _defaults +
  non-empty namespaces. JSON output support. 85.2% coverage.
- Registered on rootCmd.

Tests: resolve_test.go (11 tests), parse_test.go (14 tests), ns_test.go
(21 tests). 18 packages pass, 20 bats pass, gofmt clean, verify-reqs 90
consistent.

---ci---
project: orca
phase: P0a2
milestone: v0.9
status: execute
---/ci---
2026-08-05 16:49:12 +00:00
Jon Chery 7b5193674e docs(P0a1): ship — v0.8.2 tagged, released, merged
---ci---
project: orca
phase: P0a1
milestone: v0.9
status: complete
---/ci---
2026-08-05 16:38:46 +00:00
Jon Chery f6de82d712 verify(P0a1): 4-layer verification PASS — REQ-063,069,070; gate C-07
---ci---
project: orca
phase: P0a1
milestone: v0.9
status: verify
---/ci---
2026-08-05 16:38:36 +00:00
Jon Chery 437aab39b4 feat(P0a1): multi-namespace path resolver + config demotion + known_hosts flock + CA migration spec (v0.9 P0a1)
P0a1 — Re-architecture Foundation (path resolver + config demotion).

Path resolver (REQ-070, R-002):
- internal/paths/paths.go: 23 functions for the multi-namespace layout
  (Root/ClusterDir/NamespaceDir/NS*/DefaultNamespace/CA/MasterKey/CacheDB/
  Txn/Peers/KnownHosts/SSH/Server/Config). Honors $ORCA_HOME. 100% coverage.
- internal/certpaths/certpaths.go: refactored as thin shim delegating to
  paths, preserving the v0.8 flat-layout API for backward compat during
  the dual-write window (REQ-090). Package doc explains the v0.10-P14
  migration plan. certpaths deleted after v0.10-P14. 100% coverage.

Config demotion (REQ-069, R-014):
- internal/config/markdown.go: minimal hand-rolled YAML frontmatter parser
  (no new dep — yaml.v3 not in go.mod). Returns same *Config struct as HCL.
- internal/config/config.go: renamed Load body to LoadHCL (// Deprecated
  per R-013), added dispatcher Load() routing on extension (.hcl->HCL,
  .md->Markdown, .yaml->Markdown). Signature preserved so root.go unchanged.
- dispatch_test.go + markdown_test.go: 89.8% coverage on config package.

Known_hosts flock (REQ-063, deferred P1 from REVIEW_v0.8 A2):
- internal/security/flock.go: stdlib syscall.Flock advisory lock helper.
- internal/proxmox/bootstrap.go: TOFUHostKeyCallback capture + ResetHostKey
  both acquire the flock before read-modify-write on known_hosts. Prevents
  concurrent writers under v0.9 parallel SSH fan-out. 3 flock tests.

CA migration spec (grill C-07):
- .ciagent/CA_MIGRATION_SPEC_v0.9.md: Option A (preserve trust root,
  RECOMMENDED) vs Option B (forced re-bootstrap). Pre-flight checks,
  migration steps, rollback, post-migration invariants, spike plan.

Verification: build pass, 17/17 Go packages pass, 20/20 bats pass, gofmt
clean, go vet clean, verify-reqs 90 consistent. Coverage: paths 100%,
certpaths 100%, config 89.8%, emit covered.

---ci---
project: orca
phase: P0a1
milestone: v0.9
status: execute
---/ci---
2026-08-05 16:38:26 +00:00
Jon Chery e5d2711d71 docs(P00): ship P00 — v0.8.1 tagged, released, merged to milestone/v0.9-rearchitecture
---ci---
project: orca
phase: P00
milestone: v0.9
status: complete
---/ci---
2026-08-05 16:27:17 +00:00
Jon Chery dd81eedcd9 verify(P00): 4-layer verification PASS — REQ-068, REQ-072, REQ-089, C-06/C-15..C-18
---ci---
project: orca
phase: P00
milestone: v0.9
status: verify
---/ci---
2026-08-05 16:26:47 +00:00
Jon Chery fc94326b0e feat(P00): deprecation sweep + bash tooling gate + render contract + doc banners (v0.9 P00)
P00 — Re-architecture Foundation (deprecation/migration/test-infra/persona/docs).

Deprecation sweep (REQ-068, REQ-072, REQ-089):
- Add // Deprecated: doc comments to internal/daemon (R-001), internal/transport
  (REQ-073), internal/security/ca.go+csr.go (D-101/REQ-076), internal/engine/
  dispatcher.go+peer.go (CLI-side scheduler), internal/cli/daemon.go.
- orca daemon emits slog.Warn deprecation banner on every run (ungated); fires
  R-001 + v0.10-P05 drain-and-stop + v0.10-P14 deletion.
- orca cert and orca node join (mTLS path) emit deprecation warnings; proxmox
  SSH path (the v0.9 replacement) does not warn.
- Add --no-deprecation-warnings global flag on root command (PersistentPreRunE)
  for orca upgrade migrations.
- 12 new daemon/cert/node deprecation tests in internal/cli/daemon_test.go
  (cli coverage 81.9%, warnDeprecated 100%).
- Add DEPRECATED banners to v0.8 sections of ARCHITECTURE.md (verified the
  v0.9 supersession section + Supersession Table from prior turn are present).

Bash tooling gate (grill C-06, C-15, C-16, C-17, C-18):
- scripts/tests/test_helper.bash + example_test.bash — bats framework + helpers.
- scripts/lib/orca-log.sh — slog-compatible JSON logging to syslog (C-17).
- scripts/orca-verify-render.sh — render-contract validator skeleton (C-16).
- scripts/tests/orca-log_test.bash + orca-verify-render_test.bash — 20 bats
  tests total (happy + failure paths per C-15).
- .shellcheckrc — project shellcheck config.
- Makefile: test-bash + lint-bash targets (graceful skip if tools missing);
  wired into test + lint targets.
- internal/emit/contract.go + contract_test.go — versioned JSON render
  contract (orca.emit/v1) between Go emitters and bash appliers (C-16).
- .ciagent/BASH_CAPABILITY_MAP_v0.9.md — maps shipped internal/transport
  capabilities to bash-side equivalents or accepted drops (C-18).
- D-186 recorded in PROJECT.md: bash exempt from Go coverage gate; compensating
  control is bats + shellcheck + shfmt (C-06).

verify-reqs: 90 requirements consistent. Build/test/lint/fmt all green.
20 bats tests pass. Go tests pass. No v0.8 code deleted — only marked deprecated
(deletion deferred to v0.10-P14 per REQ-090 dual-write window).

---ci---
project: orca
phase: P00
milestone: v0.9
status: execute
---/ci---
2026-08-05 16:26:26 +00:00
Jon Chery 40b5e781ce docs(P00): resolve C-04 — relabel v1.0→v0.10 milestone, keep all 40 phases, v1.0 UAT-gated
Operator decision (resolves grill C-04 + escalation E-03): keep 2 milestones
(v0.9 + v0.10), keep all phases (40 total, exceeds 35 soft limit), v1.0 is
UAT-gated and cut as a separate tag (v1.0.0) after v0.10 completion per
operator sign-off — not a separate milestone.

Relabels all v1.0 milestone references to v0.10 across ROADMAP, REQUIREMENTS,
GRILL_v0.9, IDEATION_v0.9, PRD_v0.9, PROJECT. Phase content unchanged; only
the milestone label moves. Historical grill narrative (the original PRD §23
counts and the E-03 auto-split reasoning) preserved verbatim for audit
integrity. C-04 and E-03 marked RESOLVED in GRILL_v0.9.md.

Milestone structure:
- v0.9: Re-architecture Foundation & Workloads (13 phases P00..P0X)
- v0.10: Production Hardening (19 phases P00..P16, milestone tag v0.10.0)
- v1.0: UAT-gated production-ready cut (separate v1.0.0 tag, not a milestone)

verify-reqs: 90 requirements consistent.

---ci---
project: orca
phase: 0
milestone: v0.9
status: complete
gate: C-04 resolved
---/ci---
2026-08-05 16:08:33 +00:00
Jon Chery 7315916fbc docs(P00): ship phase 0 — v0.8.0 tagged, released, merged to milestone/v0.9-rearchitecture
---ci---
project: orca
phase: 0
milestone: v0.9
status: complete
---/ci---
2026-08-05 16:02:56 +00:00
99 changed files with 17094 additions and 244 deletions
+10
View File
@@ -79,6 +79,8 @@ and a **dispatcher** for multi-node job execution.
### 2. Daemon Layer (`internal/daemon`)
> **⚠️ DEPRECATED in v0.9**: This section describes the v0.8 architecture, superseded by the v0.9 re-architecture. See the "v0.9 Architecture" section at the bottom of this file and `.ciagent/PRD_v0.9.md`.
- **Server**: `net/http` with `http.ServeMux` (no external router)
- **TLS (P01)**: `crypto/tls` with `MinVersion=tls.VersionTLS13` and
AEAD cipher allowlist
@@ -93,6 +95,8 @@ and a **dispatcher** for multi-node job execution.
### 3. Transport Layer (`internal/transport`, NEW in P01/P02)
> **⚠️ DEPRECATED in v0.9**: This section describes the v0.8 architecture, superseded by the v0.9 re-architecture. See the "v0.9 Architecture" section at the bottom of this file and `.ciagent/PRD_v0.9.md`.
- **Client**: `http.Client` with `http.Transport.TLSClientConfig` populated
from `internal/security.NewClientTLSConfig`
- **Server**: `http.Server.TLSConfig` populated from
@@ -108,6 +112,8 @@ and a **dispatcher** for multi-node job execution.
### 4. Core Engine (`internal/engine`)
> **⚠️ DEPRECATED in v0.9**: This section describes the v0.8 architecture, superseded by the v0.9 re-architecture. The Dispatcher and PeerRegistry peer-dispatch path is replaced by a CLI-side scheduler + SSH-push (R-001). See the "v0.9 Architecture" section at the bottom of this file and `.ciagent/PRD_v0.9.md`.
- **Node Registry**: In-memory map of node IDs → metadata, persisted to SQLite
(CPU/memory capacity, available slots, last-seen)
- **Task Executor**: `os/exec.CommandContext` with `WaitDelay` (Go 1.25+) for
@@ -423,6 +429,8 @@ as a function that takes a `yield func(Job) bool` callback.
## Security Architecture
> **⚠️ DEPRECATED in v0.9**: This section describes the v0.8 internal-CA architecture, superseded by the v0.9 re-architecture (step-ca, D-101/REQ-076). See the "v0.9 Architecture" section at the bottom of this file and `.ciagent/PRD_v0.9.md`.
### Authentication
- **v0.1**: mTLS for all API endpoints (self-signed CA)
- **v0.2 P01**: Internal CA with CSR join (see Flow 1 + 2)
@@ -449,6 +457,8 @@ as a function that takes a `yield func(Job) bool` callback.
## Key Architectural Decisions (v0.1 + v0.2)
> **⚠️ DEPRECATED in v0.9**: AD-007 (HCL canonical for jobspecs) below is superseded by R-013/R-014 (Markdown with YAML frontmatter canonical; HCL legacy). See the "v0.9 Architecture" section at the bottom of this file and `.ciagent/PRD_v0.9.md`.
| ID | Decision | Rationale |
|----|----------|-----------|
| AD-001 | Single binary with subcommands | Simpler distribution, aligns with simplicity pillar |
+32
View File
@@ -0,0 +1,32 @@
# Bash Capability Map — v0.9 (grill C-18)
Maps every capability in the shipped `internal/transport` package to its
bash-side equivalent (or accepted drop with recorded rationale) in the v0.9
re-architecture. The grill (C-18) required this mapping so capability
regressions are visible, not silent.
| Shipped capability (internal/transport) | Bash-side equivalent | Status | Rationale |
|---|---|---|---|
| Retry with exponential backoff (`retry.go`: 100ms start, ×2, cap 5s, max 5 attempts) | `orca-retry()` function in `scripts/lib/orca-retry.sh` (to be written in v0.9-P01 SSH-push transport phase, REQ-073) | **planned** (v0.9-P01) | SSH dial/exec failures need the same bounded retry. The pattern is transport-agnostic; the Go retry logic is extracted into the new `internal/sshpush/` package and a bash-side helper mirrors it for the lead-applier scripts. |
| Idempotency keys (`idempotency.go`: in-memory `sync.Map` of keys, `X-Orca-Idempotency-Key` header) | Content-addressed filenames — skip SCP if the target hash already exists on the peer | **planned** (v0.9-P01) | SSH-push doesn't have HTTP headers; idempotency is achieved by content-addressing the rendered file (`<hash>.unit`) and skipping if the peer already has it. The bash applier checks `test -f /run/orca/<hash>` before applying. |
| Structured mTLS failure logging (`handshake_log.go`: slog JSON per mTLS failure) | `orca_log_error` via `scripts/lib/orca-log.sh` (C-17, shipped in this phase P00) | **dropped (mTLS removed by R-001)** | The v0.9 re-architecture removes mTLS daemon-to-daemon transport entirely (R-001). SSH failures are logged via the new `orca_log_*` functions which emit the same slog-compatible JSON field set (ts, level, actor, action, resource, result, error) to syslog. The mTLS-specific handshake-log fields (cipher suite, TLS version, cert SAN) have no SSH equivalent and are dropped — the SSH error message is captured in the `error` field instead. |
| TLS 1.3 + AEAD cipher allowlist (`mtls.go`: MinVersion=tls.VersionTLS13, CipherSuites limited) | SSH's own cipher config (`/etc/ssh/sshd_config` `Ciphers`, `MACs`, `KexAlgorithms`) managed by the operator | **dropped (transport replaced)** | R-001 replaces mTLS HTTP with SSH. SSH's transport security is governed by the peer's sshd_config, not the orca binary. The CLI's SSH client (`golang.org/x/crypto/ssh`, already a dep) uses Go's default modern SSH cipher set. The PRD does not require orca to manage sshd_config cipher policy in v0.9. |
| mTLS client/server handshake (`mtls.go`: `MTLSClient`, daemon-side `SubmitHandler`) | `ssh.Dial` + `ssh.PublicKeys` auth (CLI-side `internal/sshpush/`, REQ-073) | **replaced** (v0.9-P01) | The daemon-to-daemon mTLS handshake is replaced by CLI-to-server SSH. The CLI holds an Ed25519 key (`cluster/orca_ssh_key`, D-037) and authenticates to each peer's sshd. TOFU host-key handling (`proxmox.TOFUHostKeyCallback`, v0.8 REQ-058) is reused for all peers, not just Proxmox. |
## Net-new capabilities in v0.9 (no shipped equivalent)
| Net-new capability | Bash-side | Status |
|---|---|---|
| Transaction bundle apply (R-010, REQ-075) | `orca-apply-render.sh` (v0.10-P10) | planned |
| Drift detection (R-010) | `orca-drift.sh` (v0.10-P10) | planned |
| Per-node state collection | `orca-collect.sh` (v0.10-P09) | planned |
| Lead aggregation | `orca-aggregate.sh` (v0.10-P09) | planned |
| Credential cleanup (5-min shred) | `orca-cleanup-credentials.sh` (v0.10) | planned |
| Render-bundle validation (C-16) | `orca-verify-render.sh` (shipped this phase P00) | ✅ shipped |
| Structured logging (C-17) | `orca-log.sh` (shipped this phase P00) | ✅ shipped |
## Review cadence
This map is reviewed at each phase that introduces or modifies a bash
script. The security-engineer persona reviews the SSH trust surface; the
devops-engineer persona reviews the bash tooling gate (C-15..C-18).
+109
View File
@@ -0,0 +1,109 @@
# CA Migration Spec — v0.8 Internal CA → v0.9 step-ca (grill C-07)
**Status**: spec (must be implemented in v0.10-P14a, REQ-066)
**Gate**: C-07 — blocks v0.10-P14a until this spec is reviewed and a dry-run passes on a test cluster
## Problem
The v0.8 internal Go CA (`internal/security/ca.go`) issues RSA-3072 CA
certs (10-year validity) and ECDSA P-256 server certs (90-day). The CA
material lives at `~/.orca/ca.crt` and `~/.orca/ca.key` (flat layout, D-011).
The v0.9 re-architecture reverses AD-010 and replaces the internal CA with
step-ca (D-101, REQ-076). Existing v0.8 deployments have an internal CA
root + issued server certs that must be migrated without invalidating
trust across the cluster.
## Migration options (decision required before v0.10-P14a implementation)
### Option A — Preserve trust root (RECOMMENDED)
Import the existing `ca.key` into step-ca as the root CA key. The cluster's
trust fingerprint stays unchanged; existing server certs continue to
validate until their natural expiry; new SVIDs are minted by step-ca using
the same root.
```bash
orca upgrade --to-v1.0 --import-ca
# reads ~/.orca/ca.key → step ca init --deployment-type standalone \
# --remote-management --key $(cat ~/.orca/ca.key)
# issues new SVIDs from step-ca for all existing workloads
```
**Pros**: zero trust breakage; existing server certs keep working; minimal
operator disruption.
**Cons**: requires step-ca to accept an imported RSA-3072 key (step-ca
supports imported keys via `--key` flag; verify in the spike).
**Post-migration**: old `internal/security/ca.go` and `csr.go` are deleted
(v0.10-P14); the `cert_repo` SQLite table (0004) is dropped (step-ca
manages cert state).
### Option B — Forced re-bootstrap
Document that v0.8 certs are invalidated; every cluster re-bootstraps under
step-ca with a new root. Existing workloads are re-enrolled.
**Pros**: clean slate; no legacy RSA root.
**Cons**: trust breakage — every peer's `known_hosts` + CA cert must be
rotated; running workloads lose mTLS until re-enrolled; higher operator
disruption.
**Use case**: only if Option A is technically infeasible (step-ca rejects
the v0.8 key format).
## Pre-flight checks (must pass before migration)
1. `orca doctor` reports zero FAILs on the v0.8 cluster
2. All peers reachable via SSH
3. No in-flight transactions (the migration is stop-the-world for the CA)
4. Snapshot taken (`orca backup --include-master-key`)
5. step-ca installed on the lead via `apt-get install step-ca`
6. `step ca init` dry-run succeeds with the imported key
## Migration steps (Option A)
1. SSH to the lead; install step-ca via apt
2. Run `step ca init --deployment-type standalone --remote-management \
--key <v0.8-ca-key-path> --provisioner orca-admin`
3. Move the root cert: `cp ~/.orca/ca.crt $ORCA_HOME/cluster/ca.crt`
4. Issue new SVIDs for every registered workload (via `step ca token` +
`step ca certificate` — the CLI mints the provisioner token using
`cluster/master.key`-derived material)
5. Deploy the new SVIDs to peers via SSH-push (the v0.9 SSH-push transport)
6. Verify: `orca doctor` reports zero FAILs; CA fingerprint unchanged;
all workload SVIDs valid
7. Archive the old `internal/security/ca.go`/`csr.go` and `cert_repo` table
## Rollback
If any post-migration invariant fails:
1. Restore the v0.8 snapshot via `orca upgrade --rollback <tarball>`
2. Restart the v0.8 orca daemon on the lead
3. Verify `orca doctor` passes on the v0.8 cluster
The v0.8 internal CA remains functional during the dual-write window
(REQ-090); step-ca is additive until the migration completes.
## Post-migration invariants (must all pass)
- CA fingerprint unchanged (Option A)
- Node count unchanged
- Workload count unchanged
- All SVIDs valid (mTLS handshake succeeds lead↔every peer)
- `orca doctor` zero FAILs
- No `internal/security/ca.go` or `cert_repo` references remain in code
## Decision required
This spec is gated by C-07. The decision (Option A vs B) must be made
before v0.10-P14a implementation. Default: Option A (preserve trust root)
unless the step-ca imported-key spike fails.
## Spike (must run before v0.10-P14a)
Run on a test cluster:
1. Install step-ca on a clean Linux host
2. Generate a v0.8-style RSA-3072 CA key via the v0.8 `internal/security` package
3. Run `step ca init --key <v8-key>` and verify step-ca accepts it
4. Mint a test SVID via `step ca token` + `step ca certificate`
5. Verify the SVID validates against the imported root
If the spike fails, fall back to Option B (forced re-bootstrap) and document.
+1 -11
View File
@@ -1,11 +1 @@
{
"phase": 0,
"stage": "grill",
"milestone": "v0.9",
"milestone_slug": "rearchitecture",
"phase_role": "pre_execution",
"attempts": 0,
"updated_at": "2026-08-05T02:20:00Z",
"milestone_complete": false,
"next_milestone": null
}
{ "phase": "P07a/b/c", "stage": "verify", "milestone": "v0.9", "phase_role": "execution", "updated_at": "2026-08-05T04:40:00Z", "milestone_complete": false, "gates_cleared_this_phase": ["C-01"], "verify": { "build": "pass", "go_test": "23/23", "bats": "20/20", "gofmt": "clean", "verify_reqs": "90 consistent" } }
+36 -36
View File
@@ -25,7 +25,7 @@ re-architecture proceeds). Their **mechanics** remain as binding work items:
phase, split heavy phases.
- **Migration** mechanics → split P14 into P14a/P14b/P14c, design migration
ordering in v0.9-P00.
- **Security** mechanics → threat model in v1.0-P15.5 (C-19).
- **Security** mechanics → threat model in v0.10-P15.5 (C-19).
The 19 binding conditions (C-01..C-19) and 10 phase challenges (PC-01..PC-10)
are adopted in full as execution gates.
@@ -76,12 +76,12 @@ silently break the build story; must be spiked before commitment.
simultaneously deprecates 7 shipped subsystems and adds 8 net-new subsystems?
The deprecation of ~10k lines of shipped daemon/transport/CA code is not listed
as a phase. v0.9 P0a..P10 ship 10 phases of workload features before the
transactional control plane (R-010 deferred to v1.0 P10) — is that intentional
transactional control plane (R-010 deferred to v0.10 P10) — is that intentional
or a sequencing error? Hidden requirements (step-ca self-upgrade, master.key
rotation, Syncthing version drift)?
**Evidence**: v0.9 phase ordering ships P0a..P10 workloads, then P10 lead rules
+ migration *last*. The transactional plane (R-010) is deferred to v1.0 P10 —
+ migration *last*. The transactional plane (R-010) is deferred to v0.10 P10 —
two milestones away. v0.8 was a 4-phase NFR milestone; v0.6 was 4-phase feature.
The PRD's v0.9 (11) + v1.0 (16) = 27 phases is 3-4× prior milestone size with
no evidence the throughput model was re-validated. No phase is labeled
@@ -94,12 +94,12 @@ no evidence the throughput model was re-validated. No phase is labeled
- **PC-01**: Move the transactional plane primitives forward. The
transactional primitives (desired-state, lead-applier, drift, rollback) are
the substrate every workload phase depends on. Design spike in v0.9-P00;
full implementation in v1.0-P10 per PRD ordering (workloads first is accepted
full implementation in v0.10-P10 per PRD ordering (workloads first is accepted
given the dual-write window mitigation in I-C-006).
- **PC-02**: Add `v0.9-P00 — Deprecation sweep` as an explicit phase. Must land
before any new feature phase so coverage gates don't measure dead packages.
- **PC-03**: Split migration: `v0.9-P00b — Migration design + dry-run` (early,
parallel to deprecation) and `v1.0-P14 — Production migration` (final).
parallel to deprecation) and `v0.10-P14 — Production migration` (final).
Migration design must inform every earlier phase, not be informed by them.
**Rationale**: 27 phases framed as "two milestones" while simultaneously
@@ -118,7 +118,7 @@ testing frameworks not in current dep map — what's the cost?
**Evidence**: Active roster has 3 of 8 active; the 5 dormant personas map
directly to the 5 new apt dependencies. v0.8 took 4 phases for a pure
test/coverage milestone; v1.0 includes 11 distinct subsystems in one
test/coverage milestone; v0.10 includes 11 distinct subsystems in one
"milestone." No bash test infrastructure exists today.
**Verdict**: PROCEED-WITH-CONDITION
@@ -127,7 +127,7 @@ test/coverage milestone; v1.0 includes 11 distinct subsystems in one
**Binding conditions**:
- **C-04**: Produce a per-phase sizing estimate using v0.6/v0.7/v0.8 actuals
as the analogous baseline. If realistic phase count exceeds 35, the
milestone must be split into v0.9 + v0.10 + v1.0 (three milestones), not two.
milestone must be split into v0.9 + v0.10 (three milestones), not two.
- **C-05**: Reactivate or explicitly assign coverage for the dormant personas'
domains (security, network, devops); no "dormant" = "unowned."
- **C-06**: Decide and document whether bash scripts count toward the coverage
@@ -169,9 +169,9 @@ specified.
`ca.crt` trust root and import into step-ca, or (b) document forced
re-bootstrap as an accepted breaking change with per-cluster upgrade
procedure. Cannot be deferred.
- **C-08**: Before the first SPIFFE-touching phase (v1.0 P02 ACL), produce a
- **C-08**: Before the first SPIFFE-touching phase (v0.10 P02 ACL), produce a
working spike of step-ca JWT-SVID or X.509-SVID minting from the orca CLI
(v1.0-P01.5). If the spike fails, SPIFFE is deferred and ACL falls back to
(v0.10-P01.5). If the spike fails, SPIFFE is deferred and ACL falls back to
mTLS identity (which the shipped model already had).
- **C-09**: Define and test the `orca-pull.sh` failure contract: idempotent
re-run, bounded retry, deterministic state on partial failure, syslog
@@ -212,9 +212,9 @@ is not portable across the orca model. "Atomic, auto-rollback" is asserted for
**Confidence**: 0.82
**Mechanics adopted**:
- **PC-04**: Split P14 into `v1.0-P14a — Data migration` (current scope),
`v1.0-P14b — Daemon cutover + running-allocation adoption`,
`v1.0-P14c — Mixed-version cluster tolerance + no-orca-on-server enforcement`.
- **PC-04**: Split P14 into `v0.10-P14a — Data migration` (current scope),
`v0.10-P14b — Daemon cutover + running-allocation adoption`,
`v0.10-P14c — Mixed-version cluster tolerance + no-orca-on-server enforcement`.
Three sub-phases, each with its own integration test.
**Rationale**: The migration plan as described covers the easy third (file
@@ -254,7 +254,7 @@ documented in step-ca's own operations guide (out-of-band knowledge).
- **C-13**: Replace server-side doctor with a CLI-driven equivalent that
SSH-probes every node and reconstructs the health view the daemon used to
provide locally. This is a new requirement, not a feature; added as
I-C-002 / v1.0-P14c.
I-C-002 / v0.10-P14c.
- **C-14**: Syncthing conflict-resolution policy must be deterministic,
documented, and tested with a forced-divergence integration test.
@@ -291,7 +291,7 @@ is now the default.
**Mechanics adopted**:
- **C-19**: Write a threat model for the new posture before any
security-touching phase (v1.0-P15.5). Defend master.key + CLI mint authority
security-touching phase (v0.10-P15.5). Defend master.key + CLI mint authority
or revise. The shipped model deliberately avoided putting a single stealable
file on a single host that decrypts all secrets and mints all identities.
The threat model must document why the new posture is acceptable or specify
@@ -348,7 +348,7 @@ coverage gate (D-042/D-047) is Go-specific.
**Rationale**: Bash is not inherently unmaintainable, but bash *in a Go-only,
coverage-gated, structured-logging project* is a language-without-rails. Without
the four conditions above, the bash control plane becomes the part of the
codebase that everyone is afraid to touch by v1.0 P05. The drift between Go
codebase that everyone is afraid to touch by v0.10 P05. The drift between Go
emitters and bash appliers is the single most likely source of "works on the
CLI's machine, fails on the lead" bugs.
@@ -367,7 +367,7 @@ transactional layer *on top of the daemon*? Is this re-architecture driven by a
**Evidence**: ROADMAP.md and PROJECT.md: every milestone from v0.1 to v0.8
explicitly says "the vision is unchanged; this milestone is not a direction
change." v0.9/v1.0 is the *first* milestone in the project's history that
change." v0.9/v0.10 is the *first* milestone in the project's history that
reverses the vision's anti-patterns. AD-010's rationale: "step-ca/cfssl/
vault-pki too heavyweight for Orca's footprint." Nothing in the original PRD
suggested Orca's footprint changed. The shipped model's `internal/transport`
@@ -417,35 +417,35 @@ conditions (C-01..C-19) as execution gates. If the C-04 sizing estimate exceeds
| C-01 | Evaluate wasmtime Go binding CGO impact; if CGO-required, drop wasmtime as primary or revoke D-002 | v0.9-P07b | Build matrix spike on linux/amd64+arm64; revocation decision recorded |
| C-02 | Syncthing feasibility spike: config injection, conflict policy, deterministic failure mode | v0.9-P09 | Spike report + forced-divergence integration test |
| C-03 | Check PRD into `.ciagent/PRD_v0.9.md` before any v0.9 phase begins | (gate) | ✅ Resolved — file committed |
| C-04 | Per-phase sizing estimate vs v0.6/v0.7/v0.8 actuals; if >35, split into v0.9+v0.10+v1.0 | v0.9 start | Estimate doc with analogous-phase sizing table |
| C-04 | Per-phase sizing estimate vs v0.6/v0.7/v0.8 actuals; if >35, split into v0.9+v0.10 | v0.9 start | ✅ RESOLVED — operator decision: keep 2 milestones (v0.9+v0.10), keep all phases (40 total), v1.0 UAT-gated after v0.10 |
| C-05 | Reactivate or assign dormant persona domains (security, network, devops) | v0.9-P00 | PERSONAS.md updated with named owners |
| C-06 | Decide bash coverage-gate status; if exempt, record compensating control | v0.9-P00 | Decision recorded in PROJECT.md D-series; CI pipeline shows the gate |
| C-07 | CA migration spec: preserve existing trust root or document forced re-bootstrap | v1.0-P14a | Spec doc + migration dry-run on test cluster |
| C-08 | SPIFFE SVID minting spike; if fails, fall back to mTLS identity | v1.0-P02 (spike in P01.5) | Working SVID mint from orca CLI in sandbox |
| C-09 | `orca-pull.sh` failure contract: idempotent re-run, bounded retry, deterministic state, structured syslog | v1.0-P10 | Failure-path integration test + syslog structured-tag verification |
| C-07 | CA migration spec: preserve existing trust root or document forced re-bootstrap | v0.10-P14a | Spec doc + migration dry-run on test cluster |
| C-08 | SPIFFE SVID minting spike; if fails, fall back to mTLS identity | v0.10-P02 (spike in P01.5) | Working SVID mint from orca CLI in sandbox |
| C-09 | `orca-pull.sh` failure contract: idempotent re-run, bounded retry, deterministic state, structured syslog | v0.10-P10 | Failure-path integration test + syslog structured-tag verification |
| C-10 | Traefik config atomicity protocol (tmpfile+fsync+rename) + malformed-config behavior verified | v0.9-P02 | Atomic-rename test + Traefik malconfig-hold-last-good assertion |
| C-11 | Lead-side watchdog meta-timer for `orca-pull.sh` starvation, with structured alert path | v1.0-P09 | Watchdog fires on injected pull failure; alert received |
| C-12 | Document step-ca HA story; if single-node, record as accepted SPOF with mitigation | v1.0-P09 | Decision doc; if HA, RAFT/sync story in orca plan |
| C-13 | Replace server-side doctor with CLI-SSH-driven equivalent | v1.0-P14c | New REQ-086 in REQUIREMENTS.md; integration test SSH-probes N nodes |
| C-14 | Syncthing conflict-resolution policy deterministic + forced-divergence integration test | v1.0-P09 | Test induces divergence; resolves to single deterministic state |
| C-11 | Lead-side watchdog meta-timer for `orca-pull.sh` starvation, with structured alert path | v0.10-P09 | Watchdog fires on injected pull failure; alert received |
| C-12 | Document step-ca HA story; if single-node, record as accepted SPOF with mitigation | v0.10-P09 | Decision doc; if HA, RAFT/sync story in orca plan |
| C-13 | Replace server-side doctor with CLI-SSH-driven equivalent | v0.10-P14c | New REQ-086 in REQUIREMENTS.md; integration test SSH-probes N nodes |
| C-14 | Syncthing conflict-resolution policy deterministic + forced-divergence integration test | v0.10-P09 | Test induces divergence; resolves to single deterministic state |
| C-15 | Bash testing framework (bats/shunit2) + shellcheck + shfmt in CoreCI before any bash ships | v0.9-P00 | CI pipeline green with the gate on a sample script |
| C-16 | Versioned JSON-schema render-format contract between Go emitters and bash appliers | v0.9-P00 | Schema file in repo; both sides validate; mismatch fails CI |
| C-17 | Bash scripts emit slog-compatible JSON to syslog with audit-log field set (REQ-006) | v0.9-P00 | Syslog capture test verifies field-presence + JSON parse |
| C-18 | Document bash-side equivalents (or accepted drops) for shipped transport capabilities | v0.9-P00 | Capability-mapping doc in `.ciagent/` |
| C-19 | Write a threat model for the new posture; defend master.key + CLI mint authority or revise | v1.0-P15.5 | Threat-model doc reviewed and committed; design revised if regression found |
| C-19 | Write a threat model for the new posture; defend master.key + CLI mint authority or revise | v0.10-P15.5 | Threat-model doc reviewed and committed; design revised if regression found |
# Phase Plan Challenges
| # | Phase | Problem | Fix |
|---|-------|---------|-----|
| PC-01 | v0.9 P0aP10 | Ship 10 phases of workload features before the transactional control plane | Design spike in v0.9-P00; full impl in v1.0-P10 per PRD ordering (workloads first accepted with dual-write mitigation) |
| PC-01 | v0.9 P0aP10 | Ship 10 phases of workload features before the transactional control plane | Design spike in v0.9-P00; full impl in v0.10-P10 per PRD ordering (workloads first accepted with dual-write mitigation) |
| PC-02 | (missing) | Deprecation of ~10k lines of daemon/transport/CA code is not a phase | Add `v0.9-P00 — Deprecation sweep` as explicit phase before any new feature phase |
| PC-03 | v0.9 P10 | Migration is the last phase of v0.9 but is highest-risk | Split: migration design in v0.9-P00 (early), implementation in v1.0-P14 (final) |
| PC-04 | v1.0 P14 | Covers data migration only; omits running-allocation cutover, mixed-version cluster, rollback trigger | Split into P14a (data), P14b (daemon cutover), P14c (mixed-version tolerance) |
| PC-05 | v1.0 P02 | SPIFFE is a documented reversal with no spike; lands before spike possible | Insert `v1.0-P01.5 — SPIFFE mint spike` as hard gate before P02 |
| PC-06 | v1.0 P10 | Transactional plane depends on lead-applier bash scripts (C-09) not gated | Reorder to v0.9-P00 design + add C-09 gate |
| PC-07 | v1.0 P15/P16 | README before security threat model | Add `v1.0-P15.5 — Threat model + security review` before final review |
| PC-08 | (missing) | No phase replaces server-side `orca doctor` | Add as I-C-002 / v1.0-P14c (CLI-SSH-driven doctor) |
| PC-03 | v0.9 P10 | Migration is the last phase of v0.9 but is highest-risk | Split: migration design in v0.9-P00 (early), implementation in v0.10-P14 (final) |
| PC-04 | v0.10 P14 | Covers data migration only; omits running-allocation cutover, mixed-version cluster, rollback trigger | Split into P14a (data), P14b (daemon cutover), P14c (mixed-version tolerance) |
| PC-05 | v0.10 P02 | SPIFFE is a documented reversal with no spike; lands before spike possible | Insert `v0.10-P01.5 — SPIFFE mint spike` as hard gate before P02 |
| PC-06 | v0.10 P10 | Transactional plane depends on lead-applier bash scripts (C-09) not gated | Reorder to v0.9-P00 design + add C-09 gate |
| PC-07 | v0.10 P15/P16 | README before security threat model | Add `v0.10-P15.5 — Threat model + security review` before final review |
| PC-08 | (missing) | No phase replaces server-side `orca doctor` | Add as I-C-002 / v0.10-P14c (CLI-SSH-driven doctor) |
| PC-09 | v0.9 P09 | Syncthing lands before feasibility spike (C-02) | Spike must precede P09; if P09 is the spike, rename + gate on spike success |
| PC-10 | v0.9 P07 | Five runtimes in one phase, including wasmtime (CGO risk) and pve-vm/pve-ct | Split: P07a (process+podman), P07b (wasmtime, C-01 gated), P07c (pve-vm+ct) |
@@ -454,9 +454,9 @@ conditions (C-01..C-19) as execution gates. If the C-04 sizing estimate exceeds
1. **What measured operational failure of the shipped v0.8 daemon model is the re-architecture responding to?** — ✅ Resolved by override ground 1.
2. **Can the v0.9 scope be delivered as additive extensions?** — ✅ Resolved: rejected per override grounds 1 + 5.
3. **What is the wasmtime/CGO resolution?** — Closes via C-01 spike in v0.9-P07b.
4. **What is the master.key threat model?** — Closes via C-19 in v1.0-P15.5.
4. **What is the master.key threat model?** — Closes via C-19 in v0.10-P15.5.
5. **What is the rollback unit of work for §24, and what triggers it?** — Must be answered in v0.9-P00 txn-design spike (I-B-007).
6. **Is step-ca single-node acceptable as a cluster SPOF?** — Closes via C-12 in v1.0-P09.
6. **Is step-ca single-node acceptable as a cluster SPOF?** — Closes via C-12 in v0.10-P09.
7. **Can the bash control plane be reduced?** — Closes in v0.9-P00 (fold 3+ scripts into Go-side SSH invocations where possible).
8. **What is the realistic phase count?** — Closes via C-04 sizing before v0.9 starts; if >35, the plan becomes three milestones.
9. **Does the PRD's reversal of 6 documented decisions require a formal AD-series supersession?** — ✅ Resolved: supersession table recorded in PROJECT.md + ARCHITECTURE.md.
@@ -481,5 +481,5 @@ conditions (C-01..C-19) as execution gates. If the C-04 sizing estimate exceeds
| E-ID | Item | Auto-decision | Mitigation |
|------|------|---------------|-----------|
| E-01 | Whether the re-architecture is justified vs incremental | OVERRIDDEN by user — direction holds | Six-part evidence basis recorded in PROJECT.md Supersession Table |
| E-02 | Whether master.key passphrase-less posture is acceptable | REPLAN mechanics — threat model first | C-19 in v1.0-P15.5; if threat model shows regression vs shipped, revise design |
| E-03 | Whether 27 phases fit in 2 milestones | Auto-split if sizing exceeds 35 | C-04; if exceeded, milestone becomes v0.9 + v0.10 + v1.0 |
| E-02 | Whether master.key passphrase-less posture is acceptable | REPLAN mechanics — threat model first | C-19 in v0.10-P15.5; if threat model shows regression vs shipped, revise design |
| E-03 | Whether 27 phases fit in 2 milestones | Auto-split if sizing exceeds 35 | ✅ RESOLVED — operator: keep 2 milestones (v0.9+v0.10), keep all phases, v1.0 UAT-gated |
+53 -53
View File
@@ -1,6 +1,6 @@
# Ideation v0.9 — Re-architecture Foundation
**Project**: orca (single-project mode) | **Milestone**: v0.9/v1.0 re-architecture
**Project**: orca (single-project mode) | **Milestone**: v0.9/v0.10 re-architecture
**Date**: 2026-08-05 | **Agent**: ideation agent | **Confidence threshold**: 0.60
**Next REQ ID prior to this run**: REQ-060 (v0.8 complete)
@@ -12,7 +12,7 @@ with a CLI-only, SSH-push, step-ca, Markdown-frontmatter, multi-namespace
stack. 9 packages are deprecation targets (~2,400 LOC of v0.8
daemon/transport/security-ca/engine-dispatch/jobspec-hcl/config-hcl/certpaths
code), 7 packages are adaptable, and 8 subsystems are net-new with zero
implementation. The §23 milestone plan has 11 v0.9 phases + 17 v1.0 phases but
implementation. The §23 milestone plan has 11 v0.9 phases + 17 v0.10 phases but
under-specifies the deprecation mechanics, the SSH-push transport design, the
lead-applier execution model, several adapter/bridge layers, and the
migration ordering risk.
@@ -25,10 +25,10 @@ the PRD §23 plan are listed at the end.
### I-M-001 — `orca daemon` deprecation command and build-tag removal path
- **Tier**: mechanical
- **Description**: The PRD deprecates `internal/daemon/` (R-001) but §23 never says *how*. `internal/cli/daemon.go` (100 LOC) registers the `daemon` cobra command and wires `daemon.NewServer` + `engine.Dispatcher`. Big-bang removal would break the v0.8→v1.0 migration path (v1.0-P14) because `orca upgrade --to-v1.0` must run against a live v0.8 cluster that still has daemons. Proposal: (1) in v0.9, `orca daemon` emits a deprecation warning and still runs (dual-write window); (2) in v1.0, `orca daemon` is repurposed to `orca daemon drain-and-stop` (stops v0.8 daemons on peers via SSH, confirms workloads survive via systemd); (3) post-v1.0, the command and `internal/daemon/` are deleted. Add `// Deprecated` Go doc comments + `slog.Warn` on every run.
- **Description**: The PRD deprecates `internal/daemon/` (R-001) but §23 never says *how*. `internal/cli/daemon.go` (100 LOC) registers the `daemon` cobra command and wires `daemon.NewServer` + `engine.Dispatcher`. Big-bang removal would break the v0.8→v1.0 migration path (v0.10-P14) because `orca upgrade --to-v1.0` must run against a live v0.8 cluster that still has daemons. Proposal: (1) in v0.9, `orca daemon` emits a deprecation warning and still runs (dual-write window); (2) in v1.0, `orca daemon` is repurposed to `orca daemon drain-and-stop` (stops v0.8 daemons on peers via SSH, confirms workloads survive via systemd); (3) post-v1.0, the command and `internal/daemon/` are deleted. Add `// Deprecated` Go doc comments + `slog.Warn` on every run.
- **Rationale**: R-001 is an invariant, but the *transition* off the daemon is a mechanical gap. The v0.8 `daemon.go` is wired in `root.go` init; removing it without a transition plan breaks the §24 migration.
- **Proposed REQ ID**: REQ-061
- **Proposed phase placement**: v1.0-P14 (migration) — deprecation warning lands in v0.9-P0X
- **Proposed phase placement**: v0.10-P14 (migration) — deprecation warning lands in v0.9-P0X
- **Confidence**: 0.82
- **Accept/Defer**: accept
@@ -61,25 +61,25 @@ the PRD §23 plan are listed at the end.
### I-M-005 — `orca doctor --legacy-paths` detection for v0.8 residue
- **Tier**: mechanical
- **Description**: The v0.8 layout is `~/.orca/{orca.db, ca.crt, ca.key, server.crt, server.key, orca_ssh_key, known_hosts, config.hcl}`. The v1.0 layout is `ORCA_HOME/{_defaults/, cluster/{ca,master.key,peers,pve,txns}, <ns>/{db,.env,.env.secrets,jobs,alloc,ns.md}, orca_cache.db}`. `orca doctor` (`internal/doctor/doctor.go`, 501 LOC, adaptable) must gain a `doctor legacy` subcommand that detects v0.8 residue: presence of `orca.db` at ORCA_HOME root, `ca.crt`/`ca.key` (internal CA, superseded by step-ca), `config.hcl` (HCL, demoted), flat `server.crt` (single-namespace), and a `namespace` column in any `*.db` (R-002 says no namespace column). Output: list of detected legacy artifacts with migration recommendations. This is the *detection* half of v1.0-P14; the *migration* half is I-C-001.
- **Rationale**: §23 v1.0-P14 says "orca upgrade --to-v1.0, post-invariant checks" but doesn't specify the detection surface. `doctor` is the diagnostics framework and is explicitly adaptable.
- **Description**: The v0.8 layout is `~/.orca/{orca.db, ca.crt, ca.key, server.crt, server.key, orca_ssh_key, known_hosts, config.hcl}`. The v1.0 layout is `ORCA_HOME/{_defaults/, cluster/{ca,master.key,peers,pve,txns}, <ns>/{db,.env,.env.secrets,jobs,alloc,ns.md}, orca_cache.db}`. `orca doctor` (`internal/doctor/doctor.go`, 501 LOC, adaptable) must gain a `doctor legacy` subcommand that detects v0.8 residue: presence of `orca.db` at ORCA_HOME root, `ca.crt`/`ca.key` (internal CA, superseded by step-ca), `config.hcl` (HCL, demoted), flat `server.crt` (single-namespace), and a `namespace` column in any `*.db` (R-002 says no namespace column). Output: list of detected legacy artifacts with migration recommendations. This is the *detection* half of v0.10-P14; the *migration* half is I-C-001.
- **Rationale**: §23 v0.10-P14 says "orca upgrade --to-v1.0, post-invariant checks" but doesn't specify the detection surface. `doctor` is the diagnostics framework and is explicitly adaptable.
- **Proposed REQ ID**: REQ-065
- **Proposed phase placement**: v1.0-P14c (mixed-version tolerance + no-orca enforcement)
- **Proposed phase placement**: v0.10-P14c (mixed-version tolerance + no-orca enforcement)
- **Confidence**: 0.80
- **Accept/Defer**: accept
### I-M-006 — Legacy CA state migration to step-ca (cert import)
- **Tier**: mechanical
- **Description**: `internal/security/ca.go` (338 LOC) holds an internal Go CA with `ca.crt`/`ca.key` (RSA 3072, 10-year). The PRD replaces this with step-ca (R-006, D-101 reverses AD-010). The v1.0-P14 migration must handle existing deployments with an internal CA: (a) import the existing CA key into step-ca as `step ca init --deployment-type standalone --remote-management` with the existing key; (b) issue new SVIDs from step-ca and let old certs expire; (c) document that v0.8 certs are invalidated and re-bootstrap is required. The codebase audit says `ca.go`+`csr.go` are *replaced* — but the *state* (the CA key + issued server certs in `cert_repo` SQLite) may need to be preserved for audit history even if the live trust root changes. Proposal: `orca upgrade --to-v1.0 --import-ca` reads `~/.orca/ca.key`, initializes step-ca with it, and re-issues workload SVIDs. Without this, existing deployments lose their trust root with no path back.
- **Description**: `internal/security/ca.go` (338 LOC) holds an internal Go CA with `ca.crt`/`ca.key` (RSA 3072, 10-year). The PRD replaces this with step-ca (R-006, D-101 reverses AD-010). The v0.10-P14 migration must handle existing deployments with an internal CA: (a) import the existing CA key into step-ca as `step ca init --deployment-type standalone --remote-management` with the existing key; (b) issue new SVIDs from step-ca and let old certs expire; (c) document that v0.8 certs are invalidated and re-bootstrap is required. The codebase audit says `ca.go`+`csr.go` are *replaced* — but the *state* (the CA key + issued server certs in `cert_repo` SQLite) may need to be preserved for audit history even if the live trust root changes. Proposal: `orca upgrade --to-v1.0 --import-ca` reads `~/.orca/ca.key`, initializes step-ca with it, and re-issues workload SVIDs. Without this, existing deployments lose their trust root with no path back.
- **Rationale**: AD-010 is explicitly reversed by D-101, but the reversal doesn't address what happens to the existing CA material. §24 covers data migration but not CA migration.
- **Proposed REQ ID**: REQ-066
- **Proposed phase placement**: v1.0-P14a (data migration)
- **Proposed phase placement**: v0.10-P14a (data migration)
- **Confidence**: 0.70
- **Accept/Defer**: accept (design in v0.9-P00 so step-ca integration knows the import contract)
### I-M-007 — Fuzz test harness for the Markdown frontmatter parser
- **Tier**: mechanical
- **Description**: R-014/R-015 require byte-exact body preservation — "body of every .md config file preserved verbatim." This is a class of bug that's easy to get wrong (off-by-one on the `---` delimiter, trailing newline handling, BOM, CRLF, nested code fences containing `---`). v0.8 has no fuzz tests at all. Proposal: add a `testing.F` fuzz target in `internal/jobspec/markdown_test.go` that round-trips random frontmatter+body through `ParseMarkdown` and asserts `body == roundtripped.body` byte-exact. Also add a corpus of adversarial fixtures (CRLF, BOM, no-frontmatter, empty-frontmatter, frontmatter-with-only-separator). §23 v1.0-P08 mentions integration tests but not fuzzing.
- **Description**: R-014/R-015 require byte-exact body preservation — "body of every .md config file preserved verbatim." This is a class of bug that's easy to get wrong (off-by-one on the `---` delimiter, trailing newline handling, BOM, CRLF, nested code fences containing `---`). v0.8 has no fuzz tests at all. Proposal: add a `testing.F` fuzz target in `internal/jobspec/markdown_test.go` that round-trips random frontmatter+body through `ParseMarkdown` and asserts `body == roundtripped.body` byte-exact. Also add a corpus of adversarial fixtures (CRLF, BOM, no-frontmatter, empty-frontmatter, frontmatter-with-only-separator). §23 v0.10-P08 mentions integration tests but not fuzzing.
- **Rationale**: R-015 is a *load-bearing invariant* (body appears in inspect/history). Byte-exactness is exactly what fuzz tests are for. The v0.8 jobspec tests are golden-file only (no fuzz).
- **Proposed REQ ID**: REQ-067
- **Proposed phase placement**: v0.9-P0b (Markdown parser) — fuzz from day one
@@ -89,9 +89,9 @@ the PRD §23 plan are listed at the end.
### I-M-008 — Deprecation warnings on removed/repurposed CLI subcommands
- **Tier**: mechanical
- **Description**: The v0.8 CLI has `orca cert {ca-init,gen,show,renew,fingerprint}` (`internal/cli/cert.go`), `orca node join` with mTLS handshake semantics (`internal/cli/node.go`), `orca job run <spec.hcl>`. The PRD repurposes `node join` to SSH-bootstrap (no mTLS), deprecates `cert` (step-ca handles it), and changes `job run` to accept `.md` specs. Each removed/changed command should emit a `slog.Warn` deprecation banner with the v1.0 replacement, *except* when run under `orca upgrade`. The existing `root.go` `PersistentPreRunE` is the natural hook for a global `--no-deprecation-warnings` flag.
- **Rationale**: Operators running v0.8 commands against v0.9/v1.0 need to know what changed. The PRD doesn't mention deprecation UX.
- **Rationale**: Operators running v0.8 commands against v0.9/v0.10 need to know what changed. The PRD doesn't mention deprecation UX.
- **Proposed REQ ID**: REQ-068
- **Proposed phase placement**: v0.9-P0X (ship) + v1.0-P13 (ns subcommands, when CLI surface is finalized)
- **Proposed phase placement**: v0.9-P0X (ship) + v0.10-P13 (ns subcommands, when CLI surface is finalized)
- **Confidence**: 0.72
- **Accept/Defer**: accept
@@ -115,10 +115,10 @@ the PRD §23 plan are listed at the end.
### I-M-011 — `internal/store/` schema: per-namespace DBs, drop ns column
- **Tier**: mechanical
- **Description**: R-002 says "No `namespace` column in SQLite." The v0.8 schema has 7 migrations (`0001`..`0007`) with a single `orca.db`. The v1.0 model has one DB per namespace (`<ns>/db/orca.db`) plus a CLI-side cache DB (`orca_cache.db`, R-008). The existing `store.Open(path)` takes a path arg — adaptable. But the migrations are global; they need to apply *per namespace DB*. Proposal: `store.Open` gains a namespace parameter (or caller passes `paths.NSDb(ns)`); `migrate.go` runs `0001`..`0007` (minus `0006_node_kind_os` which is v0.8-specific) plus new `0008_namespace_layout.sql`. The `cert_repo` (`0004_certs.sql`) is removed (step-ca handles certs). The audit_log table moves to the CLI-side cache DB (R-008). Existing v0.8 `orca.db` is migrated by splitting tables into per-namespace DBs during v1.0-P14.
- **Description**: R-002 says "No `namespace` column in SQLite." The v0.8 schema has 7 migrations (`0001`..`0007`) with a single `orca.db`. The v0.10 model has one DB per namespace (`<ns>/db/orca.db`) plus a CLI-side cache DB (`orca_cache.db`, R-008). The existing `store.Open(path)` takes a path arg — adaptable. But the migrations are global; they need to apply *per namespace DB*. Proposal: `store.Open` gains a namespace parameter (or caller passes `paths.NSDb(ns)`); `migrate.go` runs `0001`..`0007` (minus `0006_node_kind_os` which is v0.8-specific) plus new `0008_namespace_layout.sql`. The `cert_repo` (`0004_certs.sql`) is removed (step-ca handles certs). The audit_log table moves to the CLI-side cache DB (R-008). Existing v0.8 `orca.db` is migrated by splitting tables into per-namespace DBs during v0.10-P14.
- **Rationale**: R-002 is explicit ("No namespace column in SQLite") but the existing schema has a single DB. §23 doesn't specify the schema split mechanics.
- **Proposed REQ ID**: REQ-071
- **Proposed phase placement**: v0.9-P0a1 + v1.0-P06 (alloc history, which uses cache DB)
- **Proposed phase placement**: v0.9-P0a1 + v0.10-P06 (alloc history, which uses cache DB)
- **Confidence**: 0.80
- **Accept/Defer**: accept
@@ -127,9 +127,9 @@ the PRD §23 plan are listed at the end.
- **Description**: `internal/transport/` (7 files, ~1300 LOC incl tests) implements mTLS client/server, dispatch, idempotency, retry, handshake logging. R-001 + R-006 replace this with SSH-push. The *idempotency* and *retry* logic (`idempotency.go` 123 LOC, `retry.go` 151 LOC) is conceptually reusable for SSH-push (retry on SSH failure, idempotency keys for SCP'd configs). Proposal: delete `mtls.go`, `dispatch.go`, `handshake_log.go`; extract retry/idempotency patterns into a new `internal/sshpush/` package. The existing `transport.IdempotencyStore` (in-memory `sync.Map` of keys) is directly reusable. This avoids re-implementing retry semantics from scratch.
- **Rationale**: The codebase audit marks `internal/transport/` as fully replaced, but the retry/idempotency *patterns* are transport-agnostic. §23 doesn't call this out.
- **Proposed REQ ID**: REQ-072
- **Proposed phase placement**: v0.9-P00 (deprecation sweep) — delete in v1.0-P14
- **Proposed phase placement**: v0.9-P00 (deprecation sweep) — delete in v0.10-P14
- **Confidence**: 0.68
- **Accept/Defer**: accept (defer deletion to v1.0-P14 to keep dual-write window open)
- **Accept/Defer**: accept (defer deletion to v0.10-P14 to keep dual-write window open)
## Tier 2 — Backend-Enriched (Structural / Architectural)
@@ -156,7 +156,7 @@ the PRD §23 plan are listed at the end.
- **Description**: R-001 says "no orca binary on servers." R-010 says the lead applies desired-state transactionally. Unresolved: does the lead run `orca-pull.sh` (pure bash that SCPs a desired-state bundle and applies it via `systemctl daemon-reload` + `systemctl restart`) or does the operator's CLI SSH into the lead and runs `orca apply` remotely (which would put an orca binary on the lead, violating R-001)? The PRD's intent is the former: the lead is bare Linux with systemd timers + bash. Proposal: (1) the CLI renders a *transaction bundle* (tarball of desired-state files + `apply.sh` + `verify.sh`) on the operator host; (2) SCPs it to the lead's `/run/orca/txns/<txn-id>/`; (3) the lead's systemd timer runs `/run/orca/txns/<txn-id>/apply.sh` which idempotently applies and runs verify; (4) the CLI polls the lead for txn status via SSH (`cat /run/orca/txns/<txn-id>/status.json`). The bash scripts are generated by the CLI's emitter (I-B-002), not hand-written per cluster.
- **Rationale**: The most ambiguous load-bearing design decision in the PRD. R-001 + R-010 together imply the lead runs no orca binary, but the lead must apply transactions. §23 doesn't resolve this. Getting it wrong means either violating R-001 or having no transactional apply.
- **Proposed REQ ID**: REQ-075
- **Proposed phase placement**: v1.0-P10 (transactional plane) — bundle format designed in v0.9-P00
- **Proposed phase placement**: v0.10-P10 (transactional plane) — bundle format designed in v0.9-P00
- **Confidence**: 0.78
- **Accept/Defer**: accept
@@ -165,13 +165,13 @@ the PRD §23 plan are listed at the end.
- **Description**: D-101 reverses AD-010 (which rejected step-ca as "too heavyweight"). §23 mentions step-ca in R-006 but never specifies the integration. Key surfaces: (1) **Provisioning**: `orca init` (adapted from v0.8's `internal/cli/init.go`) runs `step ca init` on the lead, stores root + intermediate in `cluster/ca/`. (2) **CA bootstrap**: CLI SSHs to the lead, installs step-ca via apt, runs `step ca init`, stores `step-ca.json` config. (3) **Cert signing API**: workloads request SVIDs via `step ca token` (JWE provisioner token minted by CLI) → `step ca certificate`. The CLI mints the token because it holds the provisioner password (in `cluster/master.key`-derived form). (4) **SVID minting**: each workload gets a SPIFFE ID (`spiffe://orca/<ns>/<workload>/<instance>`) encoded as a SAN in the step-ca-issued cert. The v0.8 `internal/security/ca.go` is deleted; a new `internal/stepca/` package wraps the `step` CLI via SSH (no Go step-ca client library — keep zero-new-dep posture if possible, or add `github.com/smallstep/cli` as a dep).
- **Rationale**: step-ca is a new external dependency with its own config format, provisioner model, and CLI. §23 assumes it but never designs the integration. security-engineer persona must be reactivated.
- **Proposed REQ ID**: REQ-076
- **Proposed phase placement**: v0.9-P07 (runtime block — runtimes need SVIDs) + v1.0-P02 (ACL — SPIFFE identities)
- **Proposed phase placement**: v0.9-P07 (runtime block — runtimes need SVIDs) + v0.10-P02 (ACL — SPIFFE identities)
- **Confidence**: 0.74
- **Accept/Defer**: accept
### I-B-005 — Traefik dynamic config generation and atomic reload
- **Tier**: backend-enriched
- **Description**: R-006 makes Traefik load-bearing (mTLS termination + health checks). §23 puts service blocks + Traefik health checks in v0.9-P02. Design: the CLI's Traefik emitter (I-B-002) renders a dynamic config file (`/etc/traefik/dynamic/orca-<ns>-<svc>.yaml`) with backends (the socket paths from R-007), health checks, and mTLS config pointing at step-ca's root. Atomic reload: Traefik watches the dynamic dir with `fsnotify` — writing the file atomically (tmp+rename) triggers a reload. Drain (v1.0-P05) works by writing a config with the backend's `weight=0` or removing it, triggering Traefik to stop routing. The v0.8 codebase has no Traefik integration at all. **Gated by grill C-10** (Traefik config atomicity protocol: tmpfile+fsync+rename + malformed-config hold-last-good verified).
- **Description**: R-006 makes Traefik load-bearing (mTLS termination + health checks). §23 puts service blocks + Traefik health checks in v0.9-P02. Design: the CLI's Traefik emitter (I-B-002) renders a dynamic config file (`/etc/traefik/dynamic/orca-<ns>-<svc>.yaml`) with backends (the socket paths from R-007), health checks, and mTLS config pointing at step-ca's root. Atomic reload: Traefik watches the dynamic dir with `fsnotify` — writing the file atomically (tmp+rename) triggers a reload. Drain (v0.10-P05) works by writing a config with the backend's `weight=0` or removing it, triggering Traefik to stop routing. The v0.8 codebase has no Traefik integration at all. **Gated by grill C-10** (Traefik config atomicity protocol: tmpfile+fsync+rename + malformed-config hold-last-good verified).
- **Rationale**: Traefik is net-new and load-bearing. §23 mentions it in R-006/P02/P05 but never specifies config generation or reload mechanism.
- **Proposed REQ ID**: REQ-077
- **Proposed phase placement**: v0.9-P02 (Service block + checks)
@@ -189,19 +189,19 @@ the PRD §23 plan are listed at the end.
### I-B-007 — Transaction bundle format and atomicity across N peers
- **Tier**: backend-enriched
- **Description**: R-010 requires transactional control-plane updates. §23 puts this in v1.0-P10. Design: a *transaction bundle* is a tarball containing: (1) `desired-state.json` (full desired state for affected namespaces), (2) `apply.sh` (idempotent apply script), (3) `verify.sh` (post-apply invariants), (4) `rollback.sh` (revert to previous state), (5) `manifest.sig` (signature with `cluster/master.key`). Atomicity across N peers: the CLI uploads the bundle to the lead; the lead applies to itself first, then fans out to peers via SSH. If any peer fails verify, the lead runs `rollback.sh` on all peers that applied. The bundle is content-addressed (`<txn-id> = sha256(desired-state.json)`) and stored in `cluster/txns/<txn-id>/`. Drift detection (R-010) compares the last applied bundle's desired-state against the live state (polled via SSH `systemctl show` + file checksums). **Gated by grill C-09** (orca-pull.sh failure contract: idempotent re-run, bounded retry, deterministic state, structured syslog).
- **Description**: R-010 requires transactional control-plane updates. §23 puts this in v0.10-P10. Design: a *transaction bundle* is a tarball containing: (1) `desired-state.json` (full desired state for affected namespaces), (2) `apply.sh` (idempotent apply script), (3) `verify.sh` (post-apply invariants), (4) `rollback.sh` (revert to previous state), (5) `manifest.sig` (signature with `cluster/master.key`). Atomicity across N peers: the CLI uploads the bundle to the lead; the lead applies to itself first, then fans out to peers via SSH. If any peer fails verify, the lead runs `rollback.sh` on all peers that applied. The bundle is content-addressed (`<txn-id> = sha256(desired-state.json)`) and stored in `cluster/txns/<txn-id>/`. Drift detection (R-010) compares the last applied bundle's desired-state against the live state (polled via SSH `systemctl show` + file checksums). **Gated by grill C-09** (orca-pull.sh failure contract: idempotent re-run, bounded retry, deterministic state, structured syslog).
- **Rationale**: Multi-peer atomicity is the hardest part of R-010. §23 says "ArgoCD-style" but ArgoCD is Kubernetes-native; the SSH-push model needs a custom bundle format.
- **Proposed REQ ID**: REQ-079
- **Proposed phase placement**: v1.0-P10 (transactional plane) — designed in v0.9-P00
- **Proposed phase placement**: v0.10-P10 (transactional plane) — designed in v0.9-P00
- **Confidence**: 0.76
- **Accept/Defer**: accept
### I-B-008 — Master key management and HKDF-SHA256 per-line .env.secrets encryption
- **Tier**: backend-enriched
- **Description**: R-011 specifies `.env.secrets` with AES-256-GCM, per-line nonce, master key at `cluster/master.key`. §23 puts this in v1.0-P03. Design: (1) `cluster/master.key` is a 32-byte random key generated by `orca init` (extend v0.8 `internal/security/ca.go`'s `WriteAtomic` pattern for the file write). (2) Each line of `.env.secrets` is `base64(nonce || ciphertext || tag)` where `nonce = random(12 bytes)` and `ciphertext = AES-256-GCM(plaintext, key=master.key, nonce, aad=line-number)`. (3) The AAD is the 1-indexed line number to prevent line-swap attacks. (4) Decryption reads the master key, iterates lines, decrypts with AAD. (5) `orca secrets set <ns> <key> <value>` appends an encrypted line; `orca secrets get <ns> <key>` decrypts and prints (redacted by default, `--reveal` to show). (6) The v0.8 `internal/security/redact.go` (103 LOC) is directly reusable for redaction. HKDF-SHA256 derives per-namespace sub-keys from the master key (`HKDF-SHA256(master, info=<ns>)`) so compromising one namespace's key doesn't compromise others — but the master key is the root of trust. **Gated by grill C-19** (threat model for master.key passphrase-less posture).
- **Description**: R-011 specifies `.env.secrets` with AES-256-GCM, per-line nonce, master key at `cluster/master.key`. §23 puts this in v0.10-P03. Design: (1) `cluster/master.key` is a 32-byte random key generated by `orca init` (extend v0.8 `internal/security/ca.go`'s `WriteAtomic` pattern for the file write). (2) Each line of `.env.secrets` is `base64(nonce || ciphertext || tag)` where `nonce = random(12 bytes)` and `ciphertext = AES-256-GCM(plaintext, key=master.key, nonce, aad=line-number)`. (3) The AAD is the 1-indexed line number to prevent line-swap attacks. (4) Decryption reads the master key, iterates lines, decrypts with AAD. (5) `orca secrets set <ns> <key> <value>` appends an encrypted line; `orca secrets get <ns> <key>` decrypts and prints (redacted by default, `--reveal` to show). (6) The v0.8 `internal/security/redact.go` (103 LOC) is directly reusable for redaction. HKDF-SHA256 derives per-namespace sub-keys from the master key (`HKDF-SHA256(master, info=<ns>)`) so compromising one namespace's key doesn't compromise others — but the master key is the root of trust. **Gated by grill C-19** (threat model for master.key passphrase-less posture).
- **Rationale**: R-011 is precise about the crypto but §23 doesn't specify key derivation, AAD, or CLI surface. The existing `redact.go` and `WriteAtomic` are reusable.
- **Proposed REQ ID**: REQ-080
- **Proposed phase placement**: v1.0-P03 (secrets subsystem)
- **Proposed phase placement**: v0.10-P03 (secrets subsystem)
- **Confidence**: 0.84
- **Accept/Defer**: accept
@@ -234,10 +234,10 @@ the PRD §23 plan are listed at the end.
### I-B-012 — `orca job lint` category-driven lint engine design
- **Tier**: backend-enriched
- **Description**: v1.0-P11 requires `orca job lint` with `--explain`. Design: a `Linter` that takes a `*WorkloadSpec` and runs a series of `Rule` checks, each returning a `Finding{Category, Severity, Message, Explanation}`. Categories: `schema` (missing required fields), `runtime` (incompatible runtime+constraint), `security` (missing SVID, plaintext secret in env), `migration` (missing storage replication for a migratable service), `best-practice` (no health check on a Service). `--explain` prints the rationale for each finding. Rules are registered in a `ruleRegistry` and individually testable. The linter is pure (no I/O) — it checks the spec against static rules, not live cluster state (that's `orca job verify`, P12).
- **Description**: v0.10-P11 requires `orca job lint` with `--explain`. Design: a `Linter` that takes a `*WorkloadSpec` and runs a series of `Rule` checks, each returning a `Finding{Category, Severity, Message, Explanation}`. Categories: `schema` (missing required fields), `runtime` (incompatible runtime+constraint), `security` (missing SVID, plaintext secret in env), `migration` (missing storage replication for a migratable service), `best-practice` (no health check on a Service). `--explain` prints the rationale for each finding. Rules are registered in a `ruleRegistry` and individually testable. The linter is pure (no I/O) — it checks the spec against static rules, not live cluster state (that's `orca job verify`, P12).
- **Rationale**: §23 puts this in P11 but only says "category-driven." The rule interface and category taxonomy are unspecified.
- **Proposed REQ ID**: REQ-084
- **Proposed phase placement**: v1.0-P11 (orca job lint)
- **Proposed phase placement**: v0.10-P11 (orca job lint)
- **Confidence**: 0.78
- **Accept/Defer**: accept
@@ -245,28 +245,28 @@ the PRD §23 plan are listed at the end.
### I-C-001 — v0.8→v1.0 migration ordering: daemon deprecation vs. new model rollout
- **Tier**: cross-cutting
- **Description**: The PRD §24 covers *data* migration but not *binary/daemon* deprecation ordering. The risk: v0.9 builds the new Markdown+kinds+runtime+SSH-push model, but v0.8 daemons are still running on peers. If v0.9 ships the new `orca job run` (Markdown) while the old daemon is still the execution engine, there's a split-brain: new specs can't run on the old daemon. Ordering proposal: (1) v0.9 ships the new parser + kinds + runtime + SSH-push *alongside* the old daemon (dual-write window); (2) `orca job run` in v0.9 uses the new SSH-push path if the spec is `.md` and the old daemon path if `.hcl`; (3) v1.0-P05 (drain) stops the old daemons; (4) v1.0-P14 (migration) converts remaining `.hcl` specs to `.md` and removes the daemon. The dual-write window means v0.9 is *not* a clean break — it's a compatibility milestone. This must be explicit in the plan or the v0.9 phases will assume the daemon is gone.
- **Description**: The PRD §24 covers *data* migration but not *binary/daemon* deprecation ordering. The risk: v0.9 builds the new Markdown+kinds+runtime+SSH-push model, but v0.8 daemons are still running on peers. If v0.9 ships the new `orca job run` (Markdown) while the old daemon is still the execution engine, there's a split-brain: new specs can't run on the old daemon. Ordering proposal: (1) v0.9 ships the new parser + kinds + runtime + SSH-push *alongside* the old daemon (dual-write window); (2) `orca job run` in v0.9 uses the new SSH-push path if the spec is `.md` and the old daemon path if `.hcl`; (3) v0.10-P05 (drain) stops the old daemons; (4) v0.10-P14 (migration) converts remaining `.hcl` specs to `.md` and removes the daemon. The dual-write window means v0.9 is *not* a clean break — it's a compatibility milestone. This must be explicit in the plan or the v0.9 phases will assume the daemon is gone.
- **Rationale**: Single largest risk in the re-architecture. §23 implicitly assumes v0.9 builds the new model in isolation, but existing deployments have running daemons. Getting the ordering wrong means either (a) v0.9 can't be tested against real deployments, or (b) workloads are orphaned when the daemon is removed.
- **Proposed REQ ID**: REQ-085
- **Proposed phase placement**: spans v0.9-P00 through v1.0-P14 — the *ordering decision* must be made in v0.9-P00
- **Proposed phase placement**: spans v0.9-P00 through v0.10-P14 — the *ordering decision* must be made in v0.9-P00
- **Confidence**: 0.88
- **Accept/Defer**: accept (most important idea in this report)
### I-C-002 — "No orca on server" enforcement (doctor post-migration invariant check)
- **Tier**: cross-cutting
- **Description**: R-001 is an invariant: "no orca Go binary on any server." §23 v1.0-P14 says "post-invariant checks" but doesn't specify them. `orca doctor` must gain a `doctor no-orca-on-server` check that SSHs to each peer and verifies: (1) no `orca` binary in PATH (`ssh peer which orca` returns nothing), (2) no `orca` systemd service (`ssh peer systemctl list-units 'orca*'` returns empty), (3) no `orca` process (`ssh peer pgrep -x orca` returns empty), (4) no `/etc/orca/` directory. This check must run *after* v1.0-P05 (drain) and *before* v1.0-P16 (ship). The v0.8 `internal/proxmox/bootstrap.go` already has the SSH session infrastructure (`sessionRunner` seam) — directly reusable for the doctor check.
- **Description**: R-001 is an invariant: "no orca Go binary on any server." §23 v0.10-P14 says "post-invariant checks" but doesn't specify them. `orca doctor` must gain a `doctor no-orca-on-server` check that SSHs to each peer and verifies: (1) no `orca` binary in PATH (`ssh peer which orca` returns nothing), (2) no `orca` systemd service (`ssh peer systemctl list-units 'orca*'` returns empty), (3) no `orca` process (`ssh peer pgrep -x orca` returns empty), (4) no `/etc/orca/` directory. This check must run *after* v0.10-P05 (drain) and *before* v0.10-P16 (ship). The v0.8 `internal/proxmox/bootstrap.go` already has the SSH session infrastructure (`sessionRunner` seam) — directly reusable for the doctor check.
- **Rationale**: R-001 is a hard invariant but §23 doesn't enforce it post-migration. Without this check, a failed migration could leave orphaned daemons that cause split-brain.
- **Proposed REQ ID**: REQ-086
- **Proposed phase placement**: v1.0-P14c (mixed-version tolerance)
- **Proposed phase placement**: v0.10-P14c (mixed-version tolerance)
- **Confidence**: 0.82
- **Accept/Defer**: accept
### I-C-003 — Test infrastructure: hermetic 3-linux + 1-proxmox cluster pipeline
- **Tier**: cross-cutting
- **Description**: §23 v1.0-P08 requires "hermetic CoreCI integration pipeline." The PRD §26.E mentions 3 linux + 1 proxmox. This is net-new test infra with zero current implementation. Design: (1) a `test/integration/` directory with a `docker-compose.yml` or `vagrant` setup that creates 4 containers/VMs (3 linux + 1 proxmox-simulated); (2) a Go test harness that SSHes to each, runs the CLI, and asserts end-to-end workflows (namespace create → workload submit → migrate → drain); (3) the proxmox node is simulated via a mock `pct`/`qm` script (the v0.8 `proxmox` package already has a `sessionRunner` seam for testability — extend it). The integration tests run in CoreCI on every milestone merge. The v0.8 e2e tests (`bootstrapE2ESetup` in `bootstrap_test.go`) use an in-process SSH server — this is the foundation but needs to scale to 4 nodes.
- **Description**: §23 v0.10-P08 requires "hermetic CoreCI integration pipeline." The PRD §26.E mentions 3 linux + 1 proxmox. This is net-new test infra with zero current implementation. Design: (1) a `test/integration/` directory with a `docker-compose.yml` or `vagrant` setup that creates 4 containers/VMs (3 linux + 1 proxmox-simulated); (2) a Go test harness that SSHes to each, runs the CLI, and asserts end-to-end workflows (namespace create → workload submit → migrate → drain); (3) the proxmox node is simulated via a mock `pct`/`qm` script (the v0.8 `proxmox` package already has a `sessionRunner` seam for testability — extend it). The integration tests run in CoreCI on every milestone merge. The v0.8 e2e tests (`bootstrapE2ESetup` in `bootstrap_test.go`) use an in-process SSH server — this is the foundation but needs to scale to 4 nodes.
- **Rationale**: §23 assumes the infra exists but doesn't design it. devops-engineer persona should be reactivated. Without hermetic infra, the integration tests can't run in CI.
- **Proposed REQ ID**: REQ-087
- **Proposed phase placement**: v1.0-P08 (integration tests) — harness bootstrapped in v0.9-P00
- **Proposed phase placement**: v0.10-P08 (integration tests) — harness bootstrapped in v0.9-P00
- **Confidence**: 0.80
- **Accept/Defer**: accept
@@ -275,22 +275,22 @@ the PRD §23 plan are listed at the end.
- **Description**: The config.json has `security-engineer` and `network-engineer` dormant. The re-architecture introduces step-ca (PKI), Traefik (edge proxy), Syncthing (P2P file sync), wasmtime (sandbox), podman (container runtime) — all new attack surfaces. AD-010 (step-ca rejection) is reversed. The v0.8 security posture (internal CA, mTLS daemon-to-daemon) is replaced by (step-ca, SSH-push, Traefik mTLS). The security-engineer persona must be reactivated to review: (1) step-ca provisioner model (the CLI holds the provisioner password — is that in `cluster/master.key` or a separate secret?), (2) SSH-push blast radius (compromised CLI key = full cluster), (3) Traefik as the new edge (DoS, config injection), (4) `.env.secrets` crypto (I-B-008). The network-engineer persona must review: (1) socket-based service exposure (R-007), (2) Syncthing P2P ports, (3) Traefik routing. §23 doesn't mention persona reactivation.
- **Rationale**: config.json explicitly notes the re-architecture "should reactivate security-engineer and network-engineer." Cross-cutting review concern, not a single phase.
- **Proposed REQ ID**: REQ-088
- **Proposed phase placement**: spans v0.9 through v1.0 — reactivation in v0.9-P00, review at v1.0-P15.5 (threat model) and v1.0-P16 (final audit)
- **Proposed phase placement**: spans v0.9 through v0.10 — reactivation in v0.9-P00, review at v0.10-P15.5 (threat model) and v0.10-P16 (final audit)
- **Confidence**: 0.84
- **Accept/Defer**: accept
### I-C-005 — Documentation rewrite: ARCHITECTURE.md, PROJECT.md, README, AD-010 supersession
- **Tier**: cross-cutting
- **Description**: All three docs describe the OLD architecture. `ARCHITECTURE.md` (640 lines) describes the daemon layer, mTLS transport, internal CA, HCL jobspec — all deprecated. `PROJECT.md` (30k chars) has D-001..D-010 decisions, several now superseded. `README.md` has the v0.8 quickstart. AD-010 (step-ca rejection) must be explicitly superseded by D-101 with a dated rationale reversal. The anti-patterns section in `ARCHITECTURE.md:471-484` lists "No external PKI" — now reversed. Proposal: (1) in v0.9-P00, add a "v0.9 Architecture (Supersedes v0.8)" section to ARCHITECTURE.md with the new 4-layer model; (2) mark the old sections as "v0.8 (deprecated)" with banners; (3) add a "Superseded Decisions" table (AD-009, AD-010 reversed by D-101; AD-007 HCL demoted by R-013); (4) in v1.0-P15, rewrite README quickstart for the new `curl | sh` + `orca init` + `orca ns create` flow.
- **Description**: All three docs describe the OLD architecture. `ARCHITECTURE.md` (640 lines) describes the daemon layer, mTLS transport, internal CA, HCL jobspec — all deprecated. `PROJECT.md` (30k chars) has D-001..D-010 decisions, several now superseded. `README.md` has the v0.8 quickstart. AD-010 (step-ca rejection) must be explicitly superseded by D-101 with a dated rationale reversal. The anti-patterns section in `ARCHITECTURE.md:471-484` lists "No external PKI" — now reversed. Proposal: (1) in v0.9-P00, add a "v0.9 Architecture (Supersedes v0.8)" section to ARCHITECTURE.md with the new 4-layer model; (2) mark the old sections as "v0.8 (deprecated)" with banners; (3) add a "Superseded Decisions" table (AD-009, AD-010 reversed by D-101; AD-007 HCL demoted by R-013); (4) in v0.10-P15, rewrite README quickstart for the new `curl | sh` + `orca init` + `orca ns create` flow.
- **Rationale**: The docs are the first thing new contributors read. Leaving v0.8 docs as canonical during v0.9 development causes confusion. §23 mentions README in P15 but not ARCHITECTURE.md/PROJECT.md.
- **Proposed REQ ID**: REQ-089
- **Proposed phase placement**: v0.9-P00 (banners + supersession table) + v1.0-P15 (README quickstart) + v1.0-P16 (final review)
- **Proposed phase placement**: v0.9-P00 (banners + supersession table) + v0.10-P15 (README quickstart) + v0.10-P16 (final review)
- **Confidence**: 0.82
- **Accept/Defer**: accept
### I-C-006 — Dual-write window: can v0.9 ship new parser while old daemon runs?
- **Tier**: cross-cutting
- **Description**: Focused version of I-C-001. The specific question: in v0.9, when the new Markdown parser + kinds + SSH-push are shipped, can they coexist with v0.8 daemons still running on peers? The answer depends on whether `orca job run <spec.md>` uses the new SSH-push path (bypassing the daemon entirely) or routes through the old daemon. If it bypasses, the daemon is irrelevant for new specs but still serves old `.hcl` specs. If it routes through, the daemon can't handle `.md` specs. Proposal: v0.9 `orca job run` dispatches on extension (`.md`→SSH-push new path, `.hcl`→old daemon path) via the parser dispatcher (I-M-004). This is a *dual-write window* where both paths coexist. The daemon is not removed until v1.0-P05 (drain). The risk: if a `.md` workload and a `.hcl` workload target the same node, the SSH-push path writes systemd units directly while the daemon also manages units — they can conflict. Mitigation: the SSH-push path writes to a separate systemd unit namespace (`orca-v1-<alloc>.service`) while the daemon uses `orca-<job>.service`. No unit name overlap = no conflict.
- **Description**: Focused version of I-C-001. The specific question: in v0.9, when the new Markdown parser + kinds + SSH-push are shipped, can they coexist with v0.8 daemons still running on peers? The answer depends on whether `orca job run <spec.md>` uses the new SSH-push path (bypassing the daemon entirely) or routes through the old daemon. If it bypasses, the daemon is irrelevant for new specs but still serves old `.hcl` specs. If it routes through, the daemon can't handle `.md` specs. Proposal: v0.9 `orca job run` dispatches on extension (`.md`→SSH-push new path, `.hcl`→old daemon path) via the parser dispatcher (I-M-004). This is a *dual-write window* where both paths coexist. The daemon is not removed until v0.10-P05 (drain). The risk: if a `.md` workload and a `.hcl` workload target the same node, the SSH-push path writes systemd units directly while the daemon also manages units — they can conflict. Mitigation: the SSH-push path writes to a separate systemd unit namespace (`orca-v1-<alloc>.service`) while the daemon uses `orca-<job>.service`. No unit name overlap = no conflict.
- **Rationale**: Operational feasibility question for v0.9. §23 doesn't address it. If the answer is "no dual-write, daemon must be removed first," then v0.9 can't be tested incrementally and must ship as a big-bang — much higher risk.
- **Proposed REQ ID**: REQ-090
- **Proposed phase placement**: v0.9-P00 (decision before any v0.9 execution phase)
@@ -301,35 +301,35 @@ the PRD §23 plan are listed at the end.
| ID | Tier | Title | REQ | Phase | Conf | Accept |
|----|------|-------|-----|-------|------|--------|
| I-M-001 | M | `orca daemon` deprecation path | REQ-061 | v1.0-P14 (warn v0.9-P0X) | 0.82 | accept |
| I-M-001 | M | `orca daemon` deprecation path | REQ-061 | v0.10-P14 (warn v0.9-P0X) | 0.82 | accept |
| I-M-002 | M | Coverage follow-ups to 70% | REQ-062 | v0.9-P0X + each new pkg | 0.88 | accept |
| I-M-003 | M | known_hosts flock concurrency | REQ-063 | v0.9-P0a1 | 0.74 | accept |
| I-M-004 | M | HCL→Markdown jobspec adapter | REQ-064 | v0.9-P0b | 0.85 | accept |
| I-M-005 | M | `doctor --legacy-paths` detection | REQ-065 | v1.0-P14c | 0.80 | accept |
| I-M-006 | M | Legacy CA state migration to step-ca | REQ-066 | v1.0-P14a | 0.70 | accept |
| I-M-005 | M | `doctor --legacy-paths` detection | REQ-065 | v0.10-P14c | 0.80 | accept |
| I-M-006 | M | Legacy CA state migration to step-ca | REQ-066 | v0.10-P14a | 0.70 | accept |
| I-M-007 | M | Fuzz harness for Markdown parser | REQ-067 | v0.9-P0b | 0.78 | accept |
| I-M-008 | M | Deprecation warnings on CLI subcommands | REQ-068 | v0.9-P0X + v1.0-P13 | 0.72 | accept |
| I-M-008 | M | Deprecation warnings on CLI subcommands | REQ-068 | v0.9-P0X + v0.10-P13 | 0.72 | accept |
| I-M-009 | M | HCL config demotion via adapter | REQ-069 | v0.9-P0a1 | 0.76 | accept |
| I-M-010 | M | certpaths → multi-namespace path resolver | REQ-070 | v0.9-P0a1 | 0.84 | accept |
| I-M-011 | M | store schema: per-namespace DBs | REQ-071 | v0.9-P0a1 + v1.0-P06 | 0.80 | accept |
| I-M-012 | M | transport deletion + SSH-push package | REQ-072 | v0.9-P00 (delete v1.0-P14) | 0.68 | accept |
| I-M-011 | M | store schema: per-namespace DBs | REQ-071 | v0.9-P0a1 + v0.10-P06 | 0.80 | accept |
| I-M-012 | M | transport deletion + SSH-push package | REQ-072 | v0.9-P00 (delete v0.10-P14) | 0.68 | accept |
| I-B-001 | B | SSH-push transport layer design | REQ-073 | v0.9-P01 | 0.86 | accept |
| I-B-002 | B | Emitter template system (Layer 4) | REQ-074 | v0.9-P0c | 0.82 | accept |
| I-B-003 | B | Lead applier execution model | REQ-075 | v1.0-P10 (design v0.9-P00) | 0.78 | accept |
| I-B-004 | B | step-ca integration | REQ-076 | v0.9-P07 + v1.0-P02 | 0.74 | accept |
| I-B-003 | B | Lead applier execution model | REQ-075 | v0.10-P10 (design v0.9-P00) | 0.78 | accept |
| I-B-004 | B | step-ca integration | REQ-076 | v0.9-P07 + v0.10-P02 | 0.74 | accept |
| I-B-005 | B | Traefik dynamic config + atomic reload | REQ-077 | v0.9-P02 | 0.80 | accept |
| I-B-006 | B | Runtime abstraction (5 backends) | REQ-078 | v0.9-P07a/b/c | 0.82 | accept |
| I-B-007 | B | Transaction bundle + N-peer atomicity | REQ-079 | v1.0-P10 (design v0.9-P00) | 0.76 | accept |
| I-B-008 | B | Master key + HKDF per-line encryption | REQ-080 | v1.0-P03 | 0.84 | accept |
| I-B-007 | B | Transaction bundle + N-peer atomicity | REQ-079 | v0.10-P10 (design v0.9-P00) | 0.76 | accept |
| I-B-008 | B | Master key + HKDF per-line encryption | REQ-080 | v0.10-P03 | 0.84 | accept |
| I-B-009 | B | Syncthing config + folder-ID | REQ-081 | v0.9-P09 | 0.72 | accept |
| I-B-010 | B | Namespace inheritance resolver | REQ-082 | v0.9-P0a2 | 0.86 | accept |
| I-B-011 | B | CLI-side scheduler redesign | REQ-083 | v0.9-P05 (skeleton P0c) | 0.80 | accept |
| I-B-012 | B | `orca job lint` category-driven engine | REQ-084 | v1.0-P11 | 0.78 | accept |
| I-C-001 | C | v0.8→v1.0 migration ordering | REQ-085 | spans v0.9-P00→v1.0-P14 | 0.88 | accept |
| I-C-002 | C | "No orca on server" enforcement | REQ-086 | v1.0-P14c | 0.82 | accept |
| I-C-003 | C | Hermetic test infra (3 linux + 1 pve) | REQ-087 | v1.0-P08 (bootstrap v0.9-P00) | 0.80 | accept |
| I-C-004 | C | security/network persona reactivation | REQ-088 | spans v0.9→v1.0-P16 | 0.84 | accept |
| I-C-005 | C | Docs rewrite + AD-010 supersession | REQ-089 | v0.9-P00 + v1.0-P15/P16 | 0.82 | accept |
| I-B-012 | B | `orca job lint` category-driven engine | REQ-084 | v0.10-P11 | 0.78 | accept |
| I-C-001 | C | v0.8→v1.0 migration ordering | REQ-085 | spans v0.9-P00→v0.10-P14 | 0.88 | accept |
| I-C-002 | C | "No orca on server" enforcement | REQ-086 | v0.10-P14c | 0.82 | accept |
| I-C-003 | C | Hermetic test infra (3 linux + 1 pve) | REQ-087 | v0.10-P08 (bootstrap v0.9-P00) | 0.80 | accept |
| I-C-004 | C | security/network persona reactivation | REQ-088 | spans v0.9→v0.10-P16 | 0.84 | accept |
| I-C-005 | C | Docs rewrite + AD-010 supersession | REQ-089 | v0.9-P00 + v0.10-P15/P16 | 0.82 | accept |
| I-C-006 | C | Dual-write window decision | REQ-090 | v0.9-P00 | 0.86 | accept |
## Phase Reordering / Addition Flags (against PRD §23)
@@ -339,8 +339,8 @@ the PRD §23 plan are listed at the end.
3. **I-B-001 (SSH-push transport)** — §23 v0.9-P01 needs SSH-push. The design is a prerequisite. **Recommendation: SSH-push design in P0a1, not deferred to P01.**
4. **I-B-002 (emitter template system)** — should be designed *with* the schemas (P0c). **Recommendation: expand P0c to "schemas + emitter interface."**
5. **I-B-003 (lead applier model)** — bundle format + lead applier model must be designed *in v0.9* so the emitter can produce bundle-compatible output. **Recommendation: design spike in v0.9-P00.**
6. **I-C-003 (test infra)** — hermetic cluster harness should be bootstrapped in v0.9-P00 so every v0.9 phase can run integration tests. **Recommendation: bootstrap in v0.9-P00, expand in v1.0-P08.**
7. **I-C-004 / I-C-005 (persona reactivation + docs)** — span the whole milestone. **Recommendation: fold persona reviews into v0.9-P00 and v1.0-P16; fold doc banners into v0.9-P00.**
6. **I-C-003 (test infra)** — hermetic cluster harness should be bootstrapped in v0.9-P00 so every v0.9 phase can run integration tests. **Recommendation: bootstrap in v0.9-P00, expand in v0.10-P08.**
7. **I-C-004 / I-C-005 (persona reactivation + docs)** — span the whole milestone. **Recommendation: fold persona reviews into v0.9-P00 and v0.10-P16; fold doc banners into v0.9-P00.**
## Cross-Reference Against Existing Decisions
+7 -7
View File
@@ -1,8 +1,8 @@
# Orca — Comprehensive Product Requirements Document (v0.9/v1.0)
# Orca — Comprehensive Product Requirements Document (v0.9/v0.10)
**Audience:** Operators, AI agents, downstream tooling authors
> This PRD SUPERSEDES the shipped v0.1v0.8 architecture. The v0.9 and v1.0
> This PRD SUPERSEDES the shipped v0.1v0.8 architecture. The v0.9 and v0.10
> milestones implement a re-architecture whose load-bearing rules (R-001…R-016)
> and decisions (D-068…D-206) replace or demote several earlier documented
> decisions. See §22 decision-trace and the Supersession Table in
@@ -15,13 +15,13 @@
| Spec lock-in | ✅ R-001…R-016 + D-001…D-206 settled |
| v0.1v0.8 implementation | ✅ shipped (REQ-001..060, D-001..D-047) |
| v0.9 implementation | ⬜ Phase 0 pre-execution (this file is the spec input) |
| v1.0 implementation | ⬜ planning (post-PRD) |
| v0.10 implementation | ⬜ planning (post-PRD) |
| v1.x multi-host state | ⬜ parked (post-v1.0) |
| v2.x full Nomad-HCL | ⬜ parked (post-v1.x) |
## Override justification (recorded for the grill supersession)
The v0.9/v1.0 re-architecture is justified on six independent grounds rather
The v0.9/v0.10 re-architecture is justified on six independent grounds rather
than preference. Each reverses a prior documented decision; the new evidence
basis is recorded with the reversal in the Supersession Table:
@@ -54,7 +54,7 @@ that source; the substantive planning artifacts live in:
- `IDEATION_v0.9.md` — 30 ideas (REQ-061..REQ-090), three tiers
- `GRILL_v0.9.md` — 9-axis adversarial review, 19 binding conditions, 10 phase challenges
- `REQUIREMENTS.md` — REQ-061..REQ-090 appended
- `ROADMAP.md` — v0.9 (13 phases) + v1.0 (19 phases) appended
- `ROADMAP.md` — v0.9 (13 phases) + v0.10 (19 phases) appended
- `PERSONAS.md` — security/network/devops reactivated
- `ARCHITECTURE.md` — v0.9 banners + Supersession Table
@@ -70,7 +70,7 @@ that source; the substantive planning artifacts live in:
| R-006 | mTLS on by default; cluster CA = step-ca; Traefik + `LoadCredential=` are load-bearing. |
| R-007 | Sockets by default (`/run/orca/alloc-<id>/port-<name>.sock`); `127.0.0.1` opt-in. |
| R-008 | CLI results cached locally with per-class TTLs (`orca_cache` SQLite). |
| R-009 | CLI host SPOF mitigated by external shared state in v1.x; v1.0 ships the abstractions + cache layer. |
| R-009 | CLI host SPOF mitigated by external shared state in v1.x; v0.10 ships the abstractions + cache layer. |
| R-010 | Control plane updates are transactional (ArgoCD-style desired-state/lead-applier). |
| R-011 | Each namespace has `.env` (plaintext) and `.env.secrets` (AES-256-GCM, per-line nonce); master key per `ORCA_HOME` at `cluster/master.key`. |
| R-012 | Workload kinds are `Job`, `Service`, `DaemonSet`; schema-separated by `kind:` in frontmatter. |
@@ -84,7 +84,7 @@ that source; the substantive planning artifacts live in:
### v0.9 — Workloads + Re-architecture Foundation (13 phases)
P00 (deprecation sweep + migration-ordering + txn-design spike + test-infra bootstrap + persona reactivation + doc banners), P0a1 (path resolver + config demotion), P0a2 (namespace CRUD + inheritance), P0b (Markdown jobspec parser + fuzz), P0c (schemas + emitter interface), P01 (SSH-push transport + host-path volumes), P02 (service + Traefik emitter), P03 (update stanza), P04 (lifecycle hooks), P05 (constraints + CLI-side scheduler), P06 (task groups), P07a/P07b/P07c (process+podman / wasmtime [C-01 gated] / pve-vm+ct runtimes), P08 (sockets), P09 (Syncthing [C-02 gated]), P10 (lead rules + migration), P0X (ship + audit).
### v1.0 — Production Hardening (19 phases)
### v0.10 — Production Hardening (19 phases)
P00 (CLI cache), P01 (metrics), P01.5 (SPIFFE spike [C-08 gated]), P02 (ACL), P03 (secrets), P04 (backup/restore), P05 (drain + daemon drain-and-stop), P06 (alloc history), P07 (recovery), P08 (integration tests), P09 (collector+aggregator), P10 (transactional plane [C-09 gated]), P11 (job lint), P12 (job verify), P13 (ns subcommands), P14a/P14b/P14c (data / daemon cutover / mixed-version tolerance), P15 (README), P15.5 (threat model [C-19 gated]), P16 (final review + ship — v1.0.0 release).
See `ROADMAP.md` for the full reordered plan and `GRILL_v0.9.md` for the 19
+4 -2
View File
@@ -36,6 +36,7 @@ Build a lightweight system to manage and execute workloads across a set of nodes
| D-004 | Scheduling algorithm for v0.1? | **Single-node only (no scheduling)** | Multi-node scheduling is out of scope for v0.1. Tasks run on the node they're submitted to. | 0.90 |
| D-005 | CLI output format? | **Human-readable by default, `--json` flag for machine consumption** | Serves both humans and AI agents. | 0.95 |
| D-006 | Job/task definition format? | **HCL or YAML in `.hcl`/`.yaml` files** | Familiar to Nomad/HashiCorp users; simpler than JSON for humans. | 0.88 |
| D-186 | Bash scripts coverage gate: count toward Go gate or exempt? | **Exempt from Go coverage gate; compensating control: bats tests (C-15) + shellcheck + shfmt in CI; every script must have >=1 happy-path and >=1 failure-path bats test** | Bash is a different language surface from Go; the 70%/50% Go coverage gate (D-042/D-047) is Go-specific. Forcing bash into the Go gate would require a coverage tool that does not exist for bash. The compensating control (bats + shellcheck + shfmt) provides equivalent discipline. | 0.82 |
| D-007 | Authentication? | **mTLS for v0.1, token-based deferred** | mTLS is the most secure default. Tokens can be added later if needed. | 0.80 |
| D-008 | Container runtime? | **Direct process execution (no container runtime) for v0.1** | Avoids the Docker/container dependency. Pure process management. | 0.85 |
| D-009 | Configuration file location? | **`~/.orca/config.hcl` and `/etc/orca/orca.hcl`** | Standard XDG-style paths. | 0.90 |
@@ -380,7 +381,7 @@ within the `clarify_budget` (10):
---
# v0.9/v1.0 — Re-architecture Scope Summary (Supersedes v0.1v0.8 architecture)
# v0.9/v0.10 — Re-architecture Scope Summary (Supersedes v0.1v0.8 architecture)
v0.9 is the first DIRECTION-CHANGE milestone in the project's history.
It supersedes the shipped v0.1v0.8 architecture per the adopted PRD
@@ -439,8 +440,9 @@ are recorded in `REQUIREMENTS.md`. The reordered phase plan is in
| ID | Question | Decision | Rationale | Confidence |
|----|----------|----------|-----------|------------|
| D-101 | Cluster CA: internal Go CA (AD-010) or step-ca (external)? | **step-ca (apt-installed)** | Externally mandated per override ground 2; AD-010's "too heavyweight" rationale reversed. CLI wraps `step` CLI via SSH (no Go step-ca client library — keep zero-new-dep posture if possible, or add `github.com/smallstep/cli` as a dep). **Gated by C-07** (CA migration spec). | 0.74 |
| D-068 | Workload identity: internal X.509 CA or SPIFFE SVIDs? | **SPIFFE SVIDs minted at submit time via step-ca** | Multi-tenancy (override ground 3) requires per-workload identity model; SPIFFE is the standard. SPIFFE ID `spiffe://orca/ns/<ns>/job/<name>/alloc/<id>` as SAN. **Gated by C-08** (mint spike in v1.0-P01.5; fallback to mTLS identity if spike fails). | 0.72 |
| D-068 | Workload identity: internal X.509 CA or SPIFFE SVIDs? | **SPIFFE SVIDs minted at submit time via step-ca** | Multi-tenancy (override ground 3) requires per-workload identity model; SPIFFE is the standard. SPIFFE ID `spiffe://orca/ns/<ns>/job/<name>/alloc/<id>` as SAN. **Gated by C-08** (mint spike in v0.10-P01.5; fallback to mTLS identity if spike fails). | 0.72 |
| D-088 | Runtime: direct os/exec only (D-008) or multi-runtime? | **5 runtimes: wasm (wasmtime primary), podman, process, pve-vm, pve-ct** | WASM is the primary workload (override ground 4). `processRuntime` wraps existing `executor.go`; others are net-new. Split P07a/b/c per grill PC-10. **P07b gated by C-01** (wasmtime/CGO eval). | 0.82 |
| D-158 | Namespace model: single flat root or multi-namespace? | **Multi-namespace under ORCA_HOME (R-002)** | Hard multi-tenant product requirement (override ground 3). `_defaults/` implicit root; `cluster/` for cluster-wide; per-namespace `db/`, `.env`, `.env.secrets`, `jobs/`, `alloc/`, `ns.md`. No namespace column in SQLite. | 0.84 |
| D-179 | Jobspec format: HCL canonical (AD-007) or Markdown? | **Markdown with YAML frontmatter canonical (R-013); HCL legacy** | PRD §8 — Markdown + body preservation is the operator-facing format. HCL adapter (REQ-064) preserves `orca job run old-spec.hcl` during migration. | 0.85 |
| D-185 | Re-architecture justification: incremental additive or full re-architecture? | **Full re-architecture (overridden by user)** | Six-part evidence basis above; the grill's REPLAN mechanics (PC-01..PC-10, C-01..C-19) adopted as gates. The incremental-additive path was evaluated and rejected on grounds 1 + 5 (daemon failing; SSH-push only viable). | 0.88 |
| D-187 | wasmtime Go binding (bytecodealliance/wasmtime-go) is CGO-based — does adopting it revoke D-002 (modernc/sqlite CGO-free cross-compile story)? | **Use the wasmtime CLI (apt-installed on peer) via SSH exec; do NOT import wasmtime-go.** | The Go binding links libwasmtime via cgo and would revoke D-002's CGO-free cross-compile story. The CLI-via-SSH approach (same pattern as podman/qm/pct) avoids CGO entirely. `internal/runtime/wasm.go` imports only stdlib + sshpush. `CGO_ENABLED=0 go build ./...` succeeds. C-01 grill gate SATISFIED; D-002 NOT revoked. Full evaluation in `internal/runtime/C01_WASMTIME_CGO_EVAL.md`. | 0.90 |
+19 -19
View File
@@ -141,9 +141,9 @@ REQ-047..052 all complete.
| REQ-059 | `orca node key-reset <node>` command: clears the persisted SSH host key entry for the node from `~/.orca/known_hosts` only (local, not remote authorized_keys — D-046); audit-logs `event=node.key_reset`; next `doctor proxmox`/dispatch re-pins via TOFU or `--host-key-fingerprint` | Low | **v0.8 P2** | **Complete** (P2 shipped v0.7.2) |
| REQ-060 | Requirement-status hygiene sweep: REQUIREMENTS.md v0.7 rows were stale ("Pending" after ship); add a verify-stage assertion that every REQ listed as `Complete` in ROADMAP.md has a matching `Complete` row in REQUIREMENTS.md, enforced by `make verify-reqs` | Medium | **v0.8 P3** | **Complete** (P3 shipped v0.7.3) |
## v0.9/v1.0 Requirements — Re-architecture Foundation & Production Hardening
## v0.9/v0.10 Requirements — Re-architecture Foundation & Production Hardening
The v0.9/v1.0 milestones supersede the shipped v0.1v0.8 architecture per the
The v0.9/v0.10 milestones supersede the shipped v0.1v0.8 architecture per the
adopted PRD (`.ciagent/PRD_v0.9.md`). The re-architecture is justified on six
grounds recorded in the PROJECT.md Supersession Table. 30 net-new requirements
(REQ-061..REQ-090) derive from the v0.9 IDEATION; their phase placement and
@@ -152,33 +152,33 @@ and `GRILL_v0.9.md`.
| ID | Requirement | Priority | Phase | Status |
|----|-------------|----------|-------|--------|
| REQ-061 | `orca daemon` deprecation command and build-tag removal path: v0.9 emits deprecation warning + still runs (dual-write window); v1.0 repurposes to `orca daemon drain-and-stop` (stops v0.8 daemons on peers via SSH, confirms workloads survive via systemd); post-v1.0 the command and `internal/daemon/` are deleted. `// Deprecated` Go doc comments + `slog.Warn` on every run (I-M-001) | High | **v1.0 P14** (warn v0.9 P0X) | Pending |
| REQ-061 | `orca daemon` deprecation command and build-tag removal path: v0.9 emits deprecation warning + still runs (dual-write window); v1.0 repurposes to `orca daemon drain-and-stop` (stops v0.8 daemons on peers via SSH, confirms workloads survive via systemd); post-v1.0 the command and `internal/daemon/` are deleted. `// Deprecated` Go doc comments + `slog.Warn` on every run (I-M-001) | High | **v0.10 P14** (warn v0.9 P0X) | Pending |
| REQ-062 | Coverage follow-ups: 3 zero-test packages (`internal/audit`, `internal/certpaths`, `cmd/orca`) + `internal/cli` to 70% floor; once `daemon.go` is deprecated/removed the exclusion reason disappears and the floor applies to the whole package; all net-new subsystems carry a 70% floor from their first phase (I-M-002) | Medium | **v0.9 P0X** + each new pkg | Pending |
| REQ-063 | `known_hosts` flock concurrency gap (deferred P1 from REVIEW_v0.8 A2): add `flock`-style advisory lock (stdlib `syscall.Flock` wrapper) around the read-modify-write in `TOFUHostKeyCallback` capture path (`bootstrap.go:290-302`) and `ResetHostKey` (`bootstrap.go:479-523`); lock file at `cluster/known_hosts.lock` (R-002) (I-M-003) | Medium | **v0.9 P0a1** | Pending |
| REQ-064 | HCL→Markdown jobspec adapter/bridge layer: keep `internal/jobspec/spec.go` as legacy HCL path behind `// Deprecated`; add `internal/jobspec/markdown.go` (canonical) + `internal/jobspec/dispatch.go` (extension-based dispatcher: `.md`→Markdown, `.hcl`→legacy, `.yaml`→Markdown-with-empty-body); unified `*WorkloadSpec` populated via adapter; preserves `orca job run old-spec.hcl` during migration window (I-M-004) | High | **v0.9 P0b** | Pending |
| REQ-065 | `orca doctor --legacy-paths` detection: detects v0.8 residue (orca.db at ORCA_HOME root, ca.crt/ca.key, config.hcl, flat server.crt, namespace column in any *.db); outputs list of legacy artifacts with migration recommendations; the detection half of v1.0-P14 (I-M-005) | Medium | **v1.0 P14c** | Pending |
| REQ-066 | Legacy CA state migration to step-ca: `orca upgrade --to-v1.0 --import-ca` reads `~/.orca/ca.key`, initializes step-ca with it, re-issues workload SVIDs; preserves audit history even if live trust root changes (I-M-006). **Gated by C-07** | High | **v1.0 P14a** | Pending |
| REQ-065 | `orca doctor --legacy-paths` detection: detects v0.8 residue (orca.db at ORCA_HOME root, ca.crt/ca.key, config.hcl, flat server.crt, namespace column in any *.db); outputs list of legacy artifacts with migration recommendations; the detection half of v0.10-P14 (I-M-005) | Medium | **v0.10 P14c** | Pending |
| REQ-066 | Legacy CA state migration to step-ca: `orca upgrade --to-v1.0 --import-ca` reads `~/.orca/ca.key`, initializes step-ca with it, re-issues workload SVIDs; preserves audit history even if live trust root changes (I-M-006). **Gated by C-07** | High | **v0.10 P14a** | Pending |
| REQ-067 | Fuzz test harness for Markdown frontmatter parser: `testing.F` fuzz target in `internal/jobspec/markdown_test.go` round-trips random frontmatter+body through `ParseMarkdown` asserting byte-exact body preservation; corpus of adversarial fixtures (CRLF, BOM, no-frontmatter, empty-frontmatter, frontmatter-with-only-separator) (I-M-007) | Medium | **v0.9 P0b** | Pending |
| REQ-068 | Deprecation warnings on removed/repurposed CLI subcommands: each removed/changed command (`orca cert`, `orca node join` mTLS semantics, `orca job run <spec.hcl>`) emits `slog.Warn` deprecation banner with v1.0 replacement except under `orca upgrade`; `--no-deprecation-warnings` global flag via `root.go` `PersistentPreRunE` (I-M-008) | Low | **v0.9 P0X** + v1.0 P13 | Pending |
| REQ-068 | Deprecation warnings on removed/repurposed CLI subcommands: each removed/changed command (`orca cert`, `orca node join` mTLS semantics, `orca job run <spec.hcl>`) emits `slog.Warn` deprecation banner with v1.0 replacement except under `orca upgrade`; `--no-deprecation-warnings` global flag via `root.go` `PersistentPreRunE` (I-M-008) | Low | **v0.9 P0X** + v0.10 P13 | Pending |
| REQ-069 | `internal/config/config.go` HCL config demotion via adapter: keep `internal/config/` as `legacy_config.go` with `// Deprecated`; add `internal/config/markdown.go` for new Markdown-frontmatter loader (R-014); `root.go` dispatches on file extension (`.hcl`→legacy, `.md`→new); `--config` semantics: `.hcl` read-only legacy, `.md` canonical (I-M-009) | High | **v0.9 P0a1** | Pending |
| REQ-070 | `internal/certpaths/` replacement with multi-namespace path resolver: new `internal/paths` package with `paths.NamespaceDir(ns)`, `paths.ClusterDir()`, `paths.CacheDB()`, `paths.MasterKey()`, `paths.NSDb(ns)`, `paths.NSEnv(ns)`, `paths.NSSecrets(ns)`; keep `certpaths` as thin shim for v0.8 compat then remove post-v1.0 (R-002) (I-M-010) — highest blast radius | High | **v0.9 P0a1** | Pending |
| REQ-071 | `internal/store/` schema: per-namespace DBs, drop namespace column: `store.Open` gains namespace parameter (or caller passes `paths.NSDb(ns)`); `migrate.go` runs migrations per namespace DB; `cert_repo` (0004) removed (step-ca handles certs); audit_log moves to CLI-side cache DB (R-008) (I-M-011) | High | **v0.9 P0a1** + v1.0 P06 | Pending |
| REQ-072 | `internal/transport/` deletion + SSH-push package: delete `mtls.go`, `dispatch.go`, `handshake_log.go`; extract retry/idempotency patterns into `internal/sshpush/`; existing `transport.IdempotencyStore` directly reusable (I-M-012). Deletion deferred to v1.0-P14 to keep dual-write window open | High | **v0.9 P00** (delete v1.0 P14) | Pending |
| REQ-071 | `internal/store/` schema: per-namespace DBs, drop namespace column: `store.Open` gains namespace parameter (or caller passes `paths.NSDb(ns)`); `migrate.go` runs migrations per namespace DB; `cert_repo` (0004) removed (step-ca handles certs); audit_log moves to CLI-side cache DB (R-008) (I-M-011) | High | **v0.9 P0a1** + v0.10 P06 | Pending |
| REQ-072 | `internal/transport/` deletion + SSH-push package: delete `mtls.go`, `dispatch.go`, `handshake_log.go`; extract retry/idempotency patterns into `internal/sshpush/`; existing `transport.IdempotencyStore` directly reusable (I-M-012). Deletion deferred to v0.10-P14 to keep dual-write window open | High | **v0.9 P00** (delete v0.10 P14) | Pending |
| REQ-073 | SSH-push transport layer design: connection pooling (reuse `*ssh.Client` per peer), idempotency (content-addressed filenames), retry (exponential backoff 100ms×2 cap 5s max 5), timeout (30s SCP, 10s exec), fan-out (errgroup bounded concurrency default 8), known_hosts reuse `proxmox.TOFUHostKeyCallback` (I-B-001) | High | **v0.9 P01** (design P0a1) | Pending |
| REQ-074 | Emitter template system (Layer 4): `internal/emitter/` package with `Emitter` interface `Render(spec *WorkloadSpec, node *Node) ([]File, error)`; implementations systemdEmitter/traefikEmitter/syncthingEmitter/socketEmitter; SSH-push SCPs `[]File` atomically (write-to-tmp + rename); emitters registered per kind + runtime (I-B-002) | High | **v0.9 P0c** | Pending |
| REQ-075 | Lead applier execution model: CLI renders transaction bundle (tarball + apply.sh + verify.sh) on operator host, SCPs to lead's `/run/orca/txns/<txn-id>/`, lead's systemd timer runs `apply.sh` idempotently, CLI polls txn status via SSH; bash scripts generated by emitter not hand-written (I-B-003). **Gated by C-09** | High | **v1.0 P10** (design v0.9 P00) | Pending |
| REQ-076 | step-ca integration: `orca init` runs `step ca init` on lead; CLI SSHs to lead, installs step-ca via apt, stores step-ca.json; workload SVIDs via `step ca token` (JWE minted by CLI) → `step ca certificate`; SPIFFE ID as SAN; new `internal/stepca/` package wraps `step` CLI via SSH (I-B-004). Reverses AD-010 per override justification ground 2 | High | **v0.9 P07** + v1.0 P02 | Pending |
| REQ-075 | Lead applier execution model: CLI renders transaction bundle (tarball + apply.sh + verify.sh) on operator host, SCPs to lead's `/run/orca/txns/<txn-id>/`, lead's systemd timer runs `apply.sh` idempotently, CLI polls txn status via SSH; bash scripts generated by emitter not hand-written (I-B-003). **Gated by C-09** | High | **v0.10 P10** (design v0.9 P00) | Pending |
| REQ-076 | step-ca integration: `orca init` runs `step ca init` on lead; CLI SSHs to lead, installs step-ca via apt, stores step-ca.json; workload SVIDs via `step ca token` (JWE minted by CLI) → `step ca certificate`; SPIFFE ID as SAN; new `internal/stepca/` package wraps `step` CLI via SSH (I-B-004). Reverses AD-010 per override justification ground 2 | High | **v0.9 P07** + v0.10 P02 | Pending |
| REQ-077 | Traefik dynamic config generation + atomic reload: Traefik emitter renders `/etc/traefik/dynamic/orca-<ns>-<svc>.yaml` with backends (socket paths R-007), health checks, mTLS config pointing at step-ca root; atomic reload via tmpfile+fsync+rename triggering fsnotify; drain writes `weight=0` or removes backend (I-B-005). **Gated by C-10** | High | **v0.9 P02** | Pending |
| REQ-078 | Runtime abstraction interface (5 backends): `Runtime` interface in `internal/runtime/` with Prepare/Start/Stop/Status; processRuntime (wraps existing executor.go), wasmRuntime (wasmtime via SSH), podmanRuntime, pveVMRuntime (qm via proxmox SSH), pveCTRuntime (pct); runtimeRegistry keyed by `runtime:` frontmatter value; Alloc carries runtime field changeable on migration (I-B-006). Split P07a/b/c per PC-10. **P07b gated by C-01** | High | **v0.9 P07a/b/c** | Pending |
| REQ-079 | Transaction bundle format + N-peer atomicity: bundle = tarball with desired-state.json + apply.sh + verify.sh + rollback.sh + manifest.sig (signed with master.key); content-addressed `<txn-id>=sha256(desired-state.json)` stored in `cluster/txns/<txn-id>/`; lead applies to self first then fans out; failure on any peer runs rollback.sh on applied peers (I-B-007). **Gated by C-09** | High | **v1.0 P10** (design v0.9 P00) | Pending |
| REQ-080 | Master key management + HKDF-SHA256 per-line .env.secrets encryption: `cluster/master.key` 32-byte random (generated at `orca init` using WriteAtomic pattern); each line `base64(nonce||ciphertext||tag)`, nonce=random(12 bytes), AES-256-GCM with AAD=line-number (prevents line-swap); HKDF-SHA256 derives per-namespace sub-keys; `orca secrets set/get`; v0.8 `internal/security/redact.go` reusable (I-B-008). **Gated by C-19** | High | **v1.0 P03** | Pending |
| REQ-079 | Transaction bundle format + N-peer atomicity: bundle = tarball with desired-state.json + apply.sh + verify.sh + rollback.sh + manifest.sig (signed with master.key); content-addressed `<txn-id>=sha256(desired-state.json)` stored in `cluster/txns/<txn-id>/`; lead applies to self first then fans out; failure on any peer runs rollback.sh on applied peers (I-B-007). **Gated by C-09** | High | **v0.10 P10** (design v0.9 P00) | Pending |
| REQ-080 | Master key management + HKDF-SHA256 per-line .env.secrets encryption: `cluster/master.key` 32-byte random (generated at `orca init` using WriteAtomic pattern); each line `base64(nonce||ciphertext||tag)`, nonce=random(12 bytes), AES-256-GCM with AAD=line-number (prevents line-swap); HKDF-SHA256 derives per-namespace sub-keys; `orca secrets set/get`; v0.8 `internal/security/redact.go` reusable (I-B-008). **Gated by C-19** | High | **v0.10 P03** | Pending |
| REQ-081 | Syncthing config rendering + folder-ID content-addressing: per-namespace Syncthing folder `orca-<ns>` with content-addressed folder ID `sha256(ns + master-key-fingerprint)`; CLI renders config.xml per peer; Syncthing runs as systemd unit (emitted by systemd emitter); CLI discovers peers via `cluster/peers/`; migration works because new node joins folder and syncs before workload starts (I-B-009). **Gated by C-02 + C-14** | Medium | **v0.9 P09** (spike v0.9 P00) | Pending |
| REQ-082 | Namespace inheritance resolver algorithm: DFS parent walker with visited set for cycle detection; `_defaults/` implicit root (always exists, no parent); merge semantics: child overrides parent for scalars, arrays unioned (child adds to parent); pure function (no I/O) taking `map[nsName→*NSConfig]` returning `map[nsName→*ResolvedNS]` (I-B-010) | High | **v0.9 P0a2** | Pending |
| REQ-083 | CLI-side scheduler redesign: `Score(node, workload) (score int, fits bool)` where `fits` checks runtime compatibility + constraints, `score` is bin-packing (most free capacity = highest); Services pick `count` distinct nodes (anti-affinity default); DaemonSets pick all matching nodes; Job = one-shot; CLI-side not daemon-side (R-001) (I-B-011) | High | **v0.9 P05** (skeleton P0c) | Pending |
| REQ-084 | `orca job lint` category-driven lint engine: `Linter` runs `Rule` checks returning `Finding{Category, Severity, Message, Explanation}`; categories schema/runtime/security/migration/best-practice; `--explain` prints rationale; pure (no I/O) checks against static rules (I-B-012) | Medium | **v1.0 P11** | Pending |
| REQ-085 | v0.8→v1.0 migration ordering: v0.9 ships new parser + kinds + runtime + SSH-push alongside old daemon (dual-write window); `orca job run` dispatches on extension (`.md`→SSH-push, `.hcl`→old daemon); v1.0-P05 drains old daemons; v1.0-P14 converts remaining `.hcl` specs and removes daemon (I-C-001). **Most important cross-cutting idea** | High | **v0.9 P00** → v1.0 P14 | Pending |
| REQ-086 | "No orca on server" enforcement: `orca doctor no-orca-on-server` SSHs to each peer verifying no `orca` binary in PATH, no `orca` systemd service, no `orca` process, no `/etc/orca/` directory; runs after v1.0-P05 before v1.0-P16; reuses v0.8 `proxmox` SSH session infrastructure (I-C-002). Implements grill C-13 | High | **v1.0 P14c** | Pending |
| REQ-087 | Test infrastructure: hermetic 3-linux + 1-proxmox cluster pipeline: `test/integration/` with docker-compose/vagrant creating 4 containers/VMs; Go test harness SSHes to each, runs CLI, asserts end-to-end workflows (ns create → workload submit → migrate → drain); proxmox simulated via mock pct/qm; v0.8 e2e tests (bootstrapE2ESetup) are foundation (I-C-003) | Medium | **v1.0 P08** (bootstrap v0.9 P00) | Pending |
| REQ-088 | Security-engineer + network-engineer persona reactivation: reactivate security-engineer (step-ca provisioner model, SSH-push blast radius, Traefik edge, .env.secrets crypto) and network-engineer (socket exposure R-007, Syncthing P2P ports, Traefik routing); cross-cutting review not single phase (I-C-004). Implements grill C-05 | High | **v0.9 P00** → v1.0 P16 | Pending |
| REQ-089 | Documentation rewrite: ARCHITECTURE.md/PROJECT.md/README + AD-010 supersession: v0.9-P00 adds "v0.9 Architecture (Supersedes v0.8)" section + banners + Superseded Decisions table; v1.0-P15 rewrites README quickstart for new curl|sh + orca init + orca ns create flow (I-C-005) | Medium | **v0.9 P00** + v1.0 P15/P16 | Pending |
| REQ-090 | Dual-write window: v0.9 `orca job run` dispatches on extension (`.md`→SSH-push new path, `.hcl`→old daemon path) via parser dispatcher (REQ-064); daemon not removed until v1.0-P05; SSH-push path writes to separate systemd unit namespace (`orca-v1-<alloc>.service`) while daemon uses `orca-<job>.service` — no unit name overlap = no conflict (I-C-006) | High | **v0.9 P00** | Pending |
| REQ-084 | `orca job lint` category-driven lint engine: `Linter` runs `Rule` checks returning `Finding{Category, Severity, Message, Explanation}`; categories schema/runtime/security/migration/best-practice; `--explain` prints rationale; pure (no I/O) checks against static rules (I-B-012) | Medium | **v0.10 P11** | Pending |
| REQ-085 | v0.8→v1.0 migration ordering: v0.9 ships new parser + kinds + runtime + SSH-push alongside old daemon (dual-write window); `orca job run` dispatches on extension (`.md`→SSH-push, `.hcl`→old daemon); v0.10-P05 drains old daemons; v0.10-P14 converts remaining `.hcl` specs and removes daemon (I-C-001). **Most important cross-cutting idea** | High | **v0.9 P00** → v0.10 P14 | Pending |
| REQ-086 | "No orca on server" enforcement: `orca doctor no-orca-on-server` SSHs to each peer verifying no `orca` binary in PATH, no `orca` systemd service, no `orca` process, no `/etc/orca/` directory; runs after v0.10-P05 before v0.10-P16; reuses v0.8 `proxmox` SSH session infrastructure (I-C-002). Implements grill C-13 | High | **v0.10 P14c** | Pending |
| REQ-087 | Test infrastructure: hermetic 3-linux + 1-proxmox cluster pipeline: `test/integration/` with docker-compose/vagrant creating 4 containers/VMs; Go test harness SSHes to each, runs CLI, asserts end-to-end workflows (ns create → workload submit → migrate → drain); proxmox simulated via mock pct/qm; v0.8 e2e tests (bootstrapE2ESetup) are foundation (I-C-003) | Medium | **v0.10 P08** (bootstrap v0.9 P00) | Pending |
| REQ-088 | Security-engineer + network-engineer persona reactivation: reactivate security-engineer (step-ca provisioner model, SSH-push blast radius, Traefik edge, .env.secrets crypto) and network-engineer (socket exposure R-007, Syncthing P2P ports, Traefik routing); cross-cutting review not single phase (I-C-004). Implements grill C-05 | High | **v0.9 P00** → v0.10 P16 | Pending |
| REQ-089 | Documentation rewrite: ARCHITECTURE.md/PROJECT.md/README + AD-010 supersession: v0.9-P00 adds "v0.9 Architecture (Supersedes v0.8)" section + banners + Superseded Decisions table; v0.10-P15 rewrites README quickstart for new curl|sh + orca init + orca ns create flow (I-C-005) | Medium | **v0.9 P00** + v0.10 P15/P16 | Pending |
| REQ-090 | Dual-write window: v0.9 `orca job run` dispatches on extension (`.md`→SSH-push new path, `.hcl`→old daemon path) via parser dispatcher (REQ-064); daemon not removed until v0.10-P05; SSH-push path writes to separate systemd unit namespace (`orca-v1-<alloc>.service`) while daemon uses `orca-<job>.service` — no unit name overlap = no conflict (I-C-006) | High | **v0.9 P00** | Pending |
+9 -7
View File
@@ -252,7 +252,7 @@ HCL-canonical, single-namespace, no-container-runtime, no-SPIFFE). The
reversals are justified by the six-part evidence basis recorded in the
PROJECT.md Supersession Table.
## Milestone v1.0: Production Hardening
## Milestone v0.10: Production Hardening
**Scope**: ship a cluster that operators can run. Builds on the v0.9
re-architecture foundation with the production-grade subsystems:
@@ -282,13 +282,15 @@ the v0.8→v1.0 migration.
- [ ] Phase P14c: Mixed-version tolerance + no-orca-on-server enforcement (REQ-065, REQ-086; implements C-13) — tag `v0.9.18`
- [ ] Phase P15: README quickstart (REQ-089) — tag `v0.9.19`
- [ ] Phase P15.5: Threat model + security review (**gate C-19**) — tag `v0.9.20`
- [ ] Phase P16: Final review + ship + audit — **v1.0.0 release** — tag `v0.9.21`
- [ ] Phase P16: Final review + ship + audit — **v0.10.0 milestone release** — tag `v0.9.21` (v1.0.0 cut separately after UAT sign-off)
**Milestone tag**: `v1.0.0` (the v1.0.0 release tag is the production-ready cut;
per-phase patches run on the v0.9.x line per branch-strategy.md). Per-phase
**Milestone tag**: `v0.10.0` (the v0.10 milestone release tag; v1.0.0 is
UAT-gated and cut separately after v0.10 completion per operator decision —
the v1.0.0 tag marks production-ready sign-off, not a separate milestone).
Per-phase patches run on the v0.9.x line per branch-strategy.md. Per-phase
tags: `v0.9.0``v0.9.21`.
### Per-phase REQ coverage (v1.0)
### Per-phase REQ coverage (v0.10)
- **P00** — CLI cache (R-008)
- **P01.5** — SPIFFE spike (REQ-076; C-08)
@@ -310,9 +312,9 @@ tags: `v0.9.0`…`v0.9.21`.
- **wasmtime CGO breaks cross-compile** (mitigation: C-01 spike; fallback to podman/process primary)
- **bash control plane drift** (mitigation: C-15..C-18 render-format contract + bats gate)
- **daemon cutover orphans running allocs** (mitigation: P14b split; test adoption)
- **27→35+ phase scope** (mitigation: C-04 sizing; three-milestone split if exceeded — current count v0.9=18 + v1.0=22 = 40 phases; **C-04 sizing must run before v0.9 P00 execution to determine whether to split into v0.9+v0.10+v1.0**)
- **27→35+ phase scope** (mitigation: C-04 resolved — operator decision: keep 2 milestones v0.9 + v0.10, keep all phases, v1.0 is UAT-gated after v0.10; current count v0.9=18 + v0.10=22 = 40 phases, exceeds 35 soft limit but operator accepted)
## Deferred to v1.x (out of scope for v1.0)
## Deferred to v1.x (out of scope for v0.10)
- `sqlite-wal-shared` state backend (R-009 abstractions ship in v1.0; backend in v1.x)
- `git` state backend
+2
View File
@@ -0,0 +1,2 @@
disable=SC2086
external-sources=true
+23
View File
@@ -39,19 +39,42 @@ build:
test:
go test -coverprofile=coverage.out ./...
$(MAKE) test-bash
# test-race runs the full test suite under the race detector (REQ-031).
# Wired into the .coreci.yml `test` pipeline as well.
test-race:
go test -race -coverprofile=coverage.out ./...
$(MAKE) test-bash
lint:
gofmt -l .
go vet ./...
$(MAKE) lint-bash
fmt:
gofmt -w .
# test-bash runs bats tests for shell scripts (grill C-15). Skips gracefully
# if bats is not installed.
test-bash:
@command -v bats >/dev/null 2>&1 && { \
echo "→ bats scripts/tests/*.bash"; \
bats scripts/tests/*.bash; \
} || echo "bats not installed; skipping bash tests (see scripts/tests/README.md)"
# lint-bash runs shellcheck + shfmt on shell scripts (grill C-15). Skips
# gracefully if the tools are not installed.
lint-bash:
@command -v shellcheck >/dev/null 2>&1 && { \
echo "→ shellcheck scripts/"; \
shellcheck scripts/*.sh scripts/lib/*.sh scripts/tests/*.bash || true; \
} || echo "shellcheck not installed; skipping (see scripts/tests/README.md)"
@command -v shfmt >/dev/null 2>&1 && { \
echo "→ shfmt -d scripts/"; \
shfmt -d scripts/; \
} || echo "shfmt not installed; skipping (see scripts/tests/README.md)"
clean:
rm -rf bin coverage.out *.tar.gz
+52 -43
View File
@@ -1,64 +1,73 @@
// Package certpaths centralizes the on-disk locations of the CA and
// server cert/key files. The CLI layer, the security layer, and the
// doctor layer all need to agree on these paths, so they're factored
// into their own package to avoid import cycles (cli <-> doctor).
// Package certpaths is the v0.8 path shim. It returns v0.8 flat-layout
// paths for backward compatibility during the v0.9 dual-write window
// (REQ-090). The v0.9 paths package (internal/paths) returns the new
// multi-namespace layout (R-002).
//
// certpaths will be deleted after the v0.10-P14 migration. New code
// should use internal/paths, NOT certpaths.
//
// Migration notes (per v0.10-P14):
// - CA cert/key, server cert/key, SSH key/pub, known_hosts currently
// live at the flat Root() location. The v0.9 internal/paths package
// returns the new ClusterDir()/... locations; certpaths keeps the
// v0.8 flat locations until the CA migration moves them.
// - DBPath keeps returning Root()/orca.db (v0.8 location). The new
// paths.NSDb("_defaults") returns Root()/_defaults/db/orca.db; the DB
// moves in v0.10-P14.
package certpaths
import (
"os"
"path/filepath"
"git.cloudinit.dev/coreci/orca/internal/paths"
)
const (
defaultCADir = ".orca"
caCertFilename = "ca.crt"
caKeyFilename = "ca.key"
)
// Dir returns the v0.8 flat root directory. Delegates to paths.Root()
// (which honors $ORCA_HOME, else ~/.orca). v0.8 callers expect the CA
// and DB to live directly under this directory; that does not change
// until the v0.10-P14 migration.
func Dir() string { return paths.Root() }
// Dir returns the directory the local CA lives in. Honors $ORCA_HOME
// for testability; otherwise defaults to ~/.orca.
func Dir() string {
if p := os.Getenv("ORCA_HOME"); p != "" {
return p
}
home, _ := os.UserHomeDir()
return filepath.Join(home, defaultCADir)
}
// CACertPath returns the v0.8 CA cert path: Dir()/ca.crt.
// The v0.9 location is paths.CACertPath() = ClusterDir()/ca.crt; certpaths
// keeps the v0.8 flat location until the CA migration in v0.10-P14.
func CACertPath() string { return filepath.Join(paths.Root(), "ca.crt") }
// CACertPath returns the path to ca.crt.
func CACertPath() string { return filepath.Join(Dir(), caCertFilename) }
// CAKeyPath returns the v0.8 CA key path: Dir()/ca.key.
// See CACertPath for migration notes.
func CAKeyPath() string { return filepath.Join(paths.Root(), "ca.key") }
// CAKeyPath returns the path to ca.key.
func CAKeyPath() string { return filepath.Join(Dir(), caKeyFilename) }
// ServerCertPath returns the v0.8 server cert path: Dir()/server.crt.
// See CACertPath for migration notes.
func ServerCertPath() string { return filepath.Join(paths.Root(), "server.crt") }
// ServerCertPath returns the path to server.crt.
func ServerCertPath() string { return filepath.Join(Dir(), "server.crt") }
// ServerKeyPath returns the path to server.key.
func ServerKeyPath() string { return filepath.Join(Dir(), "server.key") }
// ServerKeyPath returns the v0.8 server key path: Dir()/server.key.
// See CACertPath for migration notes.
func ServerKeyPath() string { return filepath.Join(paths.Root(), "server.key") }
// DBPath returns the path to the orca SQLite database. Honors $ORCA_DB
// for testability and explicit override; otherwise defaults to
// ~/.orca/orca.db under the same Dir() as the cert files.
// for testability and explicit override; otherwise defaults to the v0.8
// flat location Dir()/orca.db. The v0.9 location is
// paths.NSDb(paths.DefaultNamespace()) = Root()/_defaults/db/orca.db;
// certpaths keeps the v0.8 flat location until the DB move in v0.10-P14.
func DBPath() string {
if p := os.Getenv("ORCA_DB"); p != "" {
return p
}
return filepath.Join(Dir(), "orca.db")
return filepath.Join(paths.Root(), "orca.db")
}
// SSHKeyPath returns the path to the orca SSH private key (Ed25519,
// D-037). Used by `orca node join --type proxmox` to authenticate
// to remote Proxmox hosts after the initial password-based bootstrap.
// File mode 0600 (enforced by security.WriteKey).
func SSHKeyPath() string { return filepath.Join(Dir(), "orca_ssh_key") }
// SSHKeyPath returns the v0.8 SSH private key path: Dir()/orca_ssh_key.
// The v0.9 location is paths.SSHKeyPath() = ClusterDir()/orca_ssh_key;
// certpaths keeps the v0.8 flat location until the migration.
func SSHKeyPath() string { return filepath.Join(paths.Root(), "orca_ssh_key") }
// SSHPubPath returns the path to the orca SSH public key (authorized_keys
// format). Deployed to remote Proxmox hosts during `orca node join`.
// File mode 0644 (enforced by security.WriteCert).
func SSHPubPath() string { return filepath.Join(Dir(), "orca_ssh_key.pub") }
// SSHPubPath returns the v0.8 SSH public key path: Dir()/orca_ssh_key.pub.
// See SSHKeyPath for migration notes.
func SSHPubPath() string { return filepath.Join(paths.Root(), "orca_ssh_key.pub") }
// KnownHostsPath returns the path to the SSH known_hosts file used for
// TOFU host-key pinning (D-035). Captured on first connect, verified
// on all subsequent connects via golang.org/x/crypto/ssh/knownhosts.
func KnownHostsPath() string { return filepath.Join(Dir(), "known_hosts") }
// KnownHostsPath returns the v0.8 known_hosts path: Dir()/known_hosts.
// The v0.9 location is paths.KnownHostsPath() = ClusterDir()/known_hosts;
// certpaths keeps the v0.8 flat location until the migration.
func KnownHostsPath() string { return filepath.Join(paths.Root(), "known_hosts") }
+49 -22
View File
@@ -3,15 +3,17 @@ package certpaths
import (
"os"
"path/filepath"
"runtime"
"strings"
"testing"
"git.cloudinit.dev/coreci/orca/internal/paths"
)
const defaultHomeSubdir = ".orca"
func TestPaths_HonorORCAHOME(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
// Ensure ORCA_DB doesn't leak from the environment / prior tests.
t.Setenv("ORCA_DB", "")
cases := []struct {
@@ -36,17 +38,26 @@ func TestPaths_HonorORCAHOME(t *testing.T) {
})
}
// DBPath defaults to $ORCA_HOME/orca.db.
if got, want := DBPath(), filepath.Join(dir, "orca.db"); got != want {
t.Errorf("DBPath = %q, want %q", got, want)
}
// Dir() returns ORCA_HOME verbatim.
if got, want := Dir(), dir; got != want {
t.Errorf("Dir = %q, want %q", got, want)
}
}
func TestShim_DelegatesDirToPaths(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
if got, want := Dir(), paths.Root(); got != want {
t.Errorf("Dir() = %q, paths.Root() = %q (shim must delegate)", got, want)
}
if got, want := Dir(), dir; got != want {
t.Errorf("Dir() = %q, want %q", got, want)
}
}
func TestDBPath_OrcaDBOverride(t *testing.T) {
home := t.TempDir()
t.Setenv("ORCA_HOME", home)
@@ -70,19 +81,14 @@ func TestDBPath_OrcaDBEmptyStringFallsBackToHome(t *testing.T) {
}
func TestDir_DefaultHomeFallback(t *testing.T) {
// Unset ORCA_HOME so Dir() falls back to ~/.orca.
// We can't reliably mutate the real HOME in a portable way, so just
// assert that the returned path ends with the default subdir on the
// current OS and is absolute.
os.Unsetenv("ORCA_HOME")
// Also clear ORCA_DB so DBPath's fallback to Dir() is exercised.
os.Unsetenv("ORCA_DB")
home, err := os.UserHomeDir()
if err != nil {
t.Skipf("os.UserHomeDir: %v (cannot verify default fallback)", err)
}
want := filepath.Join(home, defaultCADir)
want := filepath.Join(home, defaultHomeSubdir)
if got := Dir(); got != want {
t.Errorf("Dir() default = %q, want %q", got, want)
}
@@ -92,25 +98,22 @@ func TestDir_DefaultHomeFallback(t *testing.T) {
}
func TestDir_ORCAHOMEEmptyFallsBack(t *testing.T) {
// Empty string ORCA_HOME is treated as unset → ~/.orca fallback.
t.Setenv("ORCA_HOME", "")
home, err := os.UserHomeDir()
if err != nil {
t.Skipf("os.UserHomeDir: %v", err)
}
want := filepath.Join(home, defaultCADir)
want := filepath.Join(home, defaultHomeSubdir)
if got := Dir(); got != want {
t.Errorf("Dir() with empty ORCA_HOME = %q, want %q", got, want)
}
}
func TestDir_ORCAHOMERelativePath(t *testing.T) {
// A relative ORCA_HOME is honored verbatim (no cleaning/absolutizing).
t.Setenv("ORCA_HOME", "relative/orca/home")
if got, want := Dir(), "relative/orca/home"; got != want {
t.Errorf("Dir() relative = %q, want %q", got, want)
}
// CACertPath joins the relative dir with ca.crt using filepath.Join.
if got, want := CACertPath(), filepath.Join("relative/orca/home", "ca.crt"); got != want {
t.Errorf("CACertPath relative = %q, want %q", got, want)
}
@@ -121,7 +124,6 @@ func TestAllPaths_AreConsistentWithDir(t *testing.T) {
t.Setenv("ORCA_HOME", dir)
t.Setenv("ORCA_DB", "")
// Every *Path() must live under Dir() except DBPath which also does.
base := Dir()
for _, p := range []string{
CACertPath(), CAKeyPath(),
@@ -135,6 +137,38 @@ func TestAllPaths_AreConsistentWithDir(t *testing.T) {
}
}
func TestShim_ReturnsV08FlatPaths(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
t.Setenv("ORCA_DB", "")
root := paths.Root()
if got, want := CACertPath(), filepath.Join(root, "ca.crt"); got != want {
t.Errorf("CACertPath = %q, want v0.8 flat %q", got, want)
}
if got, want := CAKeyPath(), filepath.Join(root, "ca.key"); got != want {
t.Errorf("CAKeyPath = %q, want v0.8 flat %q", got, want)
}
if got, want := ServerCertPath(), filepath.Join(root, "server.crt"); got != want {
t.Errorf("ServerCertPath = %q, want v0.8 flat %q", got, want)
}
if got, want := ServerKeyPath(), filepath.Join(root, "server.key"); got != want {
t.Errorf("ServerKeyPath = %q, want v0.8 flat %q", got, want)
}
if got, want := SSHKeyPath(), filepath.Join(root, "orca_ssh_key"); got != want {
t.Errorf("SSHKeyPath = %q, want v0.8 flat %q", got, want)
}
if got, want := SSHPubPath(), filepath.Join(root, "orca_ssh_key.pub"); got != want {
t.Errorf("SSHPubPath = %q, want v0.8 flat %q", got, want)
}
if got, want := KnownHostsPath(), filepath.Join(root, "known_hosts"); got != want {
t.Errorf("KnownHostsPath = %q, want v0.8 flat %q", got, want)
}
if got, want := DBPath(), filepath.Join(root, "orca.db"); got != want {
t.Errorf("DBPath = %q, want v0.8 flat %q", got, want)
}
}
func TestSSHPaths_Filenames(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
@@ -148,10 +182,3 @@ func TestSSHPaths_Filenames(t *testing.T) {
t.Errorf("KnownHostsPath base = %q, want %q", got, want)
}
}
func init() {
// On Windows the default home subdir is still ".orca"; the test for
// default fallback uses os.UserHomeDir which is platform-aware. This
// guard keeps the suite from running a meaningless check on plan9.
_ = runtime.GOOS
}
+15 -1
View File
@@ -42,6 +42,11 @@ func ServerCertPath() string { return certpaths.ServerCertPath() }
func ServerKeyPath() string { return certpaths.ServerKeyPath() }
// NewCommand builds the `orca cert` command tree.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). The `orca cert` command tree is retained for the
// dual-write window and scheduled for deletion in v0.10. See
// .ciagent/PRD_v0.9.md.
func NewCommand(log *slog.Logger) *cobra.Command {
if log == nil {
log = slog.Default()
@@ -49,7 +54,16 @@ func NewCommand(log *slog.Logger) *cobra.Command {
certCmd := &cobra.Command{
Use: "cert",
Short: "Manage orca certificates (CA, server, rotation)",
Long: "Bootstrap a local CA, generate server certs, and rotate them.",
Long: `Manage orca certificates (CA, server, rotation).
Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
(D-101/REQ-076). The ` + "`orca cert`" + ` command tree is retained for the
dual-write window and scheduled for deletion in v0.10. See
.ciagent/PRD_v0.9.md.`,
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
warnDeprecated("orca cert is deprecated in v0.9: step-ca (D-101) now handles CA; orca cert will be removed in v0.10 — see .ciagent/PRD_v0.9.md")
return nil
},
}
certCmd.AddCommand(newCAInitCmd(log))
+7 -3
View File
@@ -4,7 +4,6 @@ import (
"context"
"errors"
"fmt"
"log/slog"
"net/http"
"os"
"os/signal"
@@ -26,8 +25,14 @@ var (
var daemonCmd = &cobra.Command{
Use: "daemon",
Short: "Run the orca daemon (HTTP API + health checks)",
Long: "Start the orca daemon. Listens on the configured address for health, API, and dispatch requests.",
Long: `Start the orca daemon. Listens on the configured address for health, API, and dispatch requests.
Deprecated: v0.9 re-architecture replaces the orca daemon with SSH-push to
bare servers (R-001 — no orca binary on servers). The daemon is repurposed to
drain-and-stop in v0.10-P05 and scheduled for deletion in v0.10-P14. See
.ciagent/PRD_v0.9.md.`,
RunE: func(cmd *cobra.Command, args []string) error {
warnDeprecated("orca daemon is deprecated in v0.9 and will be repurposed to 'drain-and-stop' in v0.10-P05; the v0.9 re-architecture (R-001) removes the orca binary from servers — see .ciagent/PRD_v0.9.md")
db, closer, err := openDB()
if err != nil {
return err
@@ -97,5 +102,4 @@ func init() {
daemonCmd.Flags().StringVar(&daemonAddr, "addr", ":8080", "listen address")
daemonCmd.Flags().StringVar(&pprofAddr, "pprof", "", "enable pprof endpoint on <addr> (e.g. :6060); unauthenticated, operator-only")
rootCmd.AddCommand(daemonCmd)
_ = slog.Default // keep import if unused above
}
+221 -1
View File
@@ -1,6 +1,12 @@
package cli
import "testing"
import (
"bytes"
"context"
"log/slog"
"strings"
"testing"
)
func TestDaemonPprofFlag(t *testing.T) {
f := daemonCmd.Flags().Lookup("pprof")
@@ -11,3 +17,217 @@ func TestDaemonPprofFlag(t *testing.T) {
t.Errorf("--pprof default = %q, want empty", f.DefValue)
}
}
// captureSlog swaps slog.Default() for a text handler writing to buf,
// returning a buffer and a restore func. Tests use this to observe
// warnDeprecated output (which uses the package-level slog.Default).
func captureSlog(t *testing.T) (*bytes.Buffer, func()) {
t.Helper()
var buf bytes.Buffer
prev := slog.Default()
logger := slog.New(slog.NewTextHandler(&buf, &slog.HandlerOptions{Level: slog.LevelWarn}))
slog.SetDefault(logger)
return &buf, func() { slog.SetDefault(prev) }
}
// runDaemonHermetic invokes daemonCmd.RunE with a context that is
// already cancelled and an unbindable --addr, so the long-running
// server start short-circuits and RunE returns quickly without
// touching the network. It returns whatever RunE returned and the
// captured slog buffer.
func runDaemonHermetic(t *testing.T, suppressWarnings bool) (string, error) {
t.Helper()
_, cleanup := initTestEnv(t)
defer cleanup()
resetRootFlags(t)
buf, restore := captureSlog(t)
defer restore()
if suppressWarnings {
_ = rootCmd.PersistentFlags().Set("no-deprecation-warnings", "true")
}
daemonAddr = "127.0.0.1:99999" // unbindable: port outside uint16 range → ListenAndServe fails fast
ctx, cancel := context.WithCancel(context.Background())
cancel() // already-done context: the select returns via <-ctx.Done() immediately
cmd := daemonCmd
cmd.SetOut(&bytes.Buffer{})
cmd.SetErr(&bytes.Buffer{})
cmd.SetArgs(nil)
cmd.SetContext(ctx)
err := cmd.RunE(cmd, nil)
return buf.String(), err
}
// TestDaemonEmitsDeprecationWarning verifies REQ-068: `orca daemon`
// emits a slog.Warn deprecation banner on every run.
func TestDaemonEmitsDeprecationWarning(t *testing.T) {
out, _ := runDaemonHermetic(t, false)
if !strings.Contains(out, "orca daemon is deprecated in v0.9") {
t.Errorf("expected deprecation warning in slog output, got:\n%s", out)
}
if !strings.Contains(out, "R-001") {
t.Errorf("deprecation warning should reference R-001, got:\n%s", out)
}
}
// TestDaemonDeprecationWarningSuppressed verifies that
// --no-deprecation-warnings suppresses the deprecation banner (for
// `orca upgrade` migrations).
func TestDaemonDeprecationWarningSuppressed(t *testing.T) {
out, _ := runDaemonHermetic(t, true)
if strings.Contains(out, "deprecated in v0.9") {
t.Errorf("--no-deprecation-warnings should suppress the deprecation warning, got:\n%s", out)
}
}
// TestDaemonStillRuns verifies deprecation ≠ removal: the daemon
// command's RunE is still wired and callable. We don't assert on the
// error value (the hermetic short-circuit may return nil or a
// shutdown-related error), only that the command did not fail *because*
// of the deprecation notice.
func TestDaemonStillRuns(t *testing.T) {
_, err := runDaemonHermetic(t, false)
if err != nil && strings.Contains(err.Error(), "deprecated") {
t.Errorf("daemon must not error due to deprecation, got: %v", err)
}
}
// TestWarnDeprecatedGate verifies the package-level helper that gates
// deprecation warnings on the --no-deprecation-warnings flag.
func TestWarnDeprecatedGate(t *testing.T) {
t.Run("emits by default", func(t *testing.T) {
buf, restore := captureSlog(t)
defer restore()
noDeprecationWarnings = false
warnDeprecated("test-deprecation-marker")
if !strings.Contains(buf.String(), "test-deprecation-marker") {
t.Errorf("expected warning emitted, got: %s", buf.String())
}
})
t.Run("suppressed when flag set", func(t *testing.T) {
buf, restore := captureSlog(t)
defer restore()
noDeprecationWarnings = true
defer func() { noDeprecationWarnings = false }()
warnDeprecated("should-not-appear")
if strings.Contains(buf.String(), "should-not-appear") {
t.Errorf("expected no warning when --no-deprecation-warnings set, got: %s", buf.String())
}
})
}
// TestNoDeprecationWarningsFlagRegistered verifies the
// --no-deprecation-warnings persistent flag exists on rootCmd.
func TestNoDeprecationWarningsFlagRegistered(t *testing.T) {
f := rootCmd.PersistentFlags().Lookup("no-deprecation-warnings")
if f == nil {
t.Fatal("--no-deprecation-warnings persistent flag not registered on rootCmd")
}
if f.DefValue != "false" {
t.Errorf("--no-deprecation-warnings default = %q, want false", f.DefValue)
}
}
// TestCertEmitsDeprecationWarning verifies REQ-068: `orca cert`
// subcommands emit a deprecation banner.
func TestCertEmitsDeprecationWarning(t *testing.T) {
_, cleanup := initTestEnv(t)
defer cleanup()
resetRootFlags(t)
buf, restore := captureSlog(t)
defer restore()
var out bytes.Buffer
rootCmd.SetOut(&out)
rootCmd.SetErr(&out)
rootCmd.SetArgs([]string{"cert", "fingerprint", "--which", "ca"})
_ = rootCmd.Execute()
logged := buf.String()
if !strings.Contains(logged, "orca cert is deprecated in v0.9") {
t.Errorf("expected cert deprecation warning, got:\n%s", logged)
}
if !strings.Contains(logged, "step-ca") {
t.Errorf("deprecation warning should mention step-ca, got:\n%s", logged)
}
}
// TestCertDeprecationWarningSuppressed verifies --no-deprecation-warnings
// suppresses the cert deprecation banner.
func TestCertDeprecationWarningSuppressed(t *testing.T) {
_, cleanup := initTestEnv(t)
defer cleanup()
resetRootFlags(t)
_ = rootCmd.PersistentFlags().Set("no-deprecation-warnings", "true")
buf, restore := captureSlog(t)
defer restore()
var out bytes.Buffer
rootCmd.SetOut(&out)
rootCmd.SetErr(&out)
rootCmd.SetArgs([]string{"cert", "fingerprint", "--which", "ca"})
_ = rootCmd.Execute()
if strings.Contains(buf.String(), "orca cert is deprecated") {
t.Errorf("--no-deprecation-warnings should suppress cert warning, got:\n%s", buf.String())
}
}
// TestNodeJoinMTLSEmitsDeprecationWarning verifies REQ-068: the mTLS
// join path (`orca node join` without --type proxmox) warns that the
// mTLS join path is deprecated.
func TestNodeJoinMTLSEmitsDeprecationWarning(t *testing.T) {
_, cleanup := initTestEnv(t)
defer cleanup()
resetRootFlags(t)
buf, restore := captureSlog(t)
defer restore()
var out bytes.Buffer
rootCmd.SetOut(&out)
rootCmd.SetErr(&out)
rootCmd.SetArgs([]string{"node", "join", "--name", "dep-warning", "--addr", "10.0.0.55:8443"})
if err := rootCmd.Execute(); err != nil {
t.Fatalf("node join: %v", err)
}
logged := buf.String()
if !strings.Contains(logged, "mTLS join path is deprecated") {
t.Errorf("expected mTLS join deprecation warning, got:\n%s", logged)
}
if !strings.Contains(logged, "R-001") {
t.Errorf("deprecation warning should reference R-001, got:\n%s", logged)
}
}
// TestNodeJoinProxmoxNoMTLSDeprecationWarning verifies the deprecation
// warning does NOT fire for the proxmox SSH path (that path is the
// v0.9 replacement, not the deprecated mTLS path).
func TestNodeJoinProxmoxNoMTLSDeprecationWarning(t *testing.T) {
_, cleanup := initTestEnv(t)
defer cleanup()
resetRootFlags(t)
buf, restore := captureSlog(t)
defer restore()
var out bytes.Buffer
rootCmd.SetOut(&out)
rootCmd.SetErr(&out)
// proxmox path errors on missing --host before reaching the warning,
// and never calls joinLocal, so no mTLS deprecation warning fires.
rootCmd.SetArgs([]string{"node", "join", "--type", "proxmox", "--password", "x"})
_ = rootCmd.Execute()
if strings.Contains(buf.String(), "mTLS join path is deprecated") {
t.Errorf("proxmox path must not emit mTLS deprecation warning, got:\n%s", buf.String())
}
}
+22 -3
View File
@@ -74,7 +74,7 @@ var jobRunCmd = &cobra.Command{
peers := engine.NewPeerRegistry()
dispatcher := engine.NewDispatcher(newLogger(), store.NewCapacityRepo(db), peers, exec)
specBytes, _ := json.Marshal(map[string]any{
"name": spec.Job.Name,
"name": spec.Name,
"command": "/bin/true", // placeholder; full HCL dispatch lands in a later phase
})
jobID, nodeID, err := dispatcher.Submit(ctx, runTarget, specBytes, runIDKey)
@@ -93,11 +93,11 @@ var jobRunCmd = &cobra.Command{
job := &model.Job{
ID: uuid.NewString(),
Name: spec.Job.Name,
Name: spec.Name,
Spec: args[0],
Status: model.JobStatusPending,
}
if err := exec.Run(ctx, job, toTaskSpecs(spec.Tasks)); err != nil {
if err := exec.Run(ctx, job, workloadToTaskSpecs(spec)); err != nil {
if jsonOutput {
_ = printJSON(map[string]any{"id": job.ID, "status": "failed", "error": err.Error()})
return err
@@ -330,3 +330,22 @@ func toTaskSpecs(in []jobspec.TaskSpec) []engine.TaskSpec {
}
return out
}
// workloadToTaskSpecs converts a *WorkloadSpec into the engine.TaskSpec
// slice consumed by the executor. For the HCL adapter path the runtime
// block carries the legacy task[0].Command; for the Markdown path the
// runtime block is the canonical runtime abstraction (P07 will expand
// this). When Runtime is nil we emit a single no-op task to preserve
// the legacy "at least one task" invariant.
func workloadToTaskSpecs(spec *jobspec.WorkloadSpec) []engine.TaskSpec {
if spec == nil {
return nil
}
if spec.Runtime == nil {
return []engine.TaskSpec{{Name: spec.Name, Command: "/bin/true"}}
}
return []engine.TaskSpec{{
Name: spec.Name,
Command: spec.Runtime.Command,
}}
}
+2
View File
@@ -18,6 +18,7 @@ func resetRootFlags(t *testing.T) {
rootCmd.SetErr(&buf)
_ = rootCmd.PersistentFlags().Set("system", "false")
_ = rootCmd.PersistentFlags().Set("json", "false")
_ = rootCmd.PersistentFlags().Set("no-deprecation-warnings", "false")
resetCommandFlags()
}
@@ -33,6 +34,7 @@ func resetCommandFlags() {
stopID, runTarget, runIDKey, jobWatch = "", "", "", false
capSetCPU, capSetMem, capSetDisk, capNodeID = 0, 0, 0, ""
auditLimit = 50
resetNSFlags()
}
func TestNamespaceDefaultsToUserHome(t *testing.T) {
+6
View File
@@ -89,7 +89,13 @@ Node types (via --type):
// joinLocal is the existing localhost/Linux node join flow (fingerprint
// check + registry.Insert).
//
// Deprecated: v0.9 re-architecture replaces daemon-to-daemon mTLS join
// with SSH-push bootstrap (R-001). The mTLS join path is retained for
// the dual-write window and scheduled for deletion in v0.10-P14. See
// .ciagent/PRD_v0.9.md.
func joinLocal(cmd *cobra.Command) error {
warnDeprecated("orca node join (mTLS path): v0.9 R-001 replaces daemon-to-daemon mTLS join with SSH-push bootstrap; the mTLS join path is deprecated — see .ciagent/PRD_v0.9.md")
if joinName == "" {
return fmt.Errorf("--name is required")
}
+338
View File
@@ -0,0 +1,338 @@
// Package cli: ns.go implements the `orca ns` subcommand family
// (REQ-082, D-176). Subcommands:
//
// orca ns list — list all namespaces under ORCA_HOME
// orca ns create <name> — create a namespace dir + ns.md
// orca ns delete <name> — remove an empty namespace dir
// orca ns inspect <name> — print effective chain + merged env
// orca ns validate <name> — cycle + missing-parent + schema checks
//
// All subcommands honor $ORCA_HOME via internal/paths. The inheritance
// resolver (internal/ns) is a pure function shared by inspect + validate.
package cli
import (
"fmt"
"os"
"path/filepath"
"sort"
"strings"
"github.com/spf13/cobra"
"git.cloudinit.dev/coreci/orca/internal/ns"
"git.cloudinit.dev/coreci/orca/internal/paths"
)
var nsCmd = &cobra.Command{
Use: "ns",
Short: "Manage orca namespaces",
Long: `Manage orca namespaces under ORCA_HOME (R-002).
Each namespace is a directory with ns.md, .env, .env.secrets, db/,
jobs/, alloc/. The implicit root namespace _defaults always exists
(D-159); every namespace inherits from _defaults (D-185) and cannot
opt out (D-187).`,
}
var (
nsCreateParent string
nsCreateInheritsEnv bool
nsCreateInheritsSecret bool
)
var nsListCmd = &cobra.Command{
Use: "list",
Short: "List all namespaces under ORCA_HOME",
Long: `List all namespaces under ORCA_HOME (directories containing ns.md, plus the implicit _defaults).`,
Args: cobra.NoArgs,
RunE: func(cmd *cobra.Command, args []string) error {
root := paths.Root()
entries, err := os.ReadDir(root)
if err != nil {
return fmt.Errorf("read ORCA_HOME %s: %w", root, err)
}
type nsRow struct {
Name string `json:"name"`
Path string `json:"path"`
Default bool `json:"default"`
}
var rows []nsRow
for _, ent := range entries {
if !ent.IsDir() {
continue
}
if ent.Name() == "cluster" {
continue
}
nsMd := filepath.Join(root, ent.Name(), "ns.md")
if _, err := os.Stat(nsMd); err != nil {
continue
}
rows = append(rows, nsRow{
Name: ent.Name(),
Path: filepath.Join(root, ent.Name()),
Default: ent.Name() == paths.DefaultNamespace(),
})
}
sort.Slice(rows, func(i, j int) bool {
if rows[i].Name == paths.DefaultNamespace() {
return true
}
if rows[j].Name == paths.DefaultNamespace() {
return false
}
return rows[i].Name < rows[j].Name
})
if jsonOutput {
return printJSON(rows)
}
if len(rows) == 0 {
fmt.Fprintln(cmd.OutOrStdout(), "No namespaces found. Run 'orca init' first.")
return nil
}
fmt.Fprintf(cmd.OutOrStdout(), "%-20s %-10s %s\n", "NAME", "DEFAULT", "PATH")
for _, r := range rows {
def := ""
if r.Default {
def = "*"
}
fmt.Fprintf(cmd.OutOrStdout(), "%-20s %-10s %s\n", r.Name, def, r.Path)
}
return nil
},
}
var nsCreateCmd = &cobra.Command{
Use: "create <name>",
Short: "Create a namespace directory + ns.md",
Long: `Create a namespace under ORCA_HOME. Builds the dir structure
(db/, jobs/, alloc/) and writes ns.md frontmatter. --parent may be
repeated to declare inheritance; _defaults is always appended last.`,
Args: cobra.ExactArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
name := args[0]
if name == paths.DefaultNamespace() {
return fmt.Errorf("cannot create the implicit root namespace %q with `ns create` (it is auto-managed)", name)
}
if name == "cluster" {
return fmt.Errorf("name %q is reserved for the cluster-wide dir", name)
}
if nsCreateParent == "" {
nsCreateParent = paths.DefaultNamespace()
}
nsDir := paths.NamespaceDir(name)
if _, err := os.Stat(nsDir); err == nil {
if _, statErr := os.Stat(paths.NSMd(name)); statErr == nil {
return fmt.Errorf("namespace %q already exists at %s", name, nsDir)
}
}
for _, sub := range []string{"db", "jobs", "alloc"} {
if err := os.MkdirAll(filepath.Join(nsDir, sub), 0o755); err != nil {
return fmt.Errorf("create %s/%s: %w", nsDir, sub, err)
}
}
parents := []string{nsCreateParent}
if nsCreateParent == paths.DefaultNamespace() {
// Explicit _defaults listing is allowed (de-duped silently).
}
body := renderNSMd(name, parents, nsCreateInheritsEnv, nsCreateInheritsSecret)
if err := os.WriteFile(paths.NSMd(name), []byte(body), 0o644); err != nil {
return fmt.Errorf("write ns.md: %w", err)
}
if jsonOutput {
return printJSON(map[string]any{
"name": name,
"path": nsDir,
"parents": parents,
"ns_md": paths.NSMd(name),
"inherits_env": nsCreateInheritsEnv,
"inherits_secrets": nsCreateInheritsSecret,
})
}
fmt.Fprintf(cmd.OutOrStdout(), "✓ Namespace created: %s (%s)\n", name, nsDir)
return nil
},
}
var nsDeleteCmd = &cobra.Command{
Use: "delete <name>",
Short: "Remove an empty namespace directory",
Long: `Remove a namespace directory. Refuses if jobs/ or alloc/
contain any files (non-empty namespace). The implicit root _defaults
cannot be deleted.`,
Args: cobra.ExactArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
name := args[0]
if name == paths.DefaultNamespace() {
return fmt.Errorf("cannot delete the implicit root namespace %q", name)
}
nsDir := paths.NamespaceDir(name)
if _, err := os.Stat(nsDir); err != nil {
return fmt.Errorf("namespace %q not found: %w", name, err)
}
for _, sub := range []string{"jobs", "alloc"} {
dir := filepath.Join(nsDir, sub)
if err := dirNonEmpty(dir); err != nil {
return fmt.Errorf("refusing to delete %q: %s is non-empty (%w); clear it first", name, sub, err)
}
}
if err := os.RemoveAll(nsDir); err != nil {
return fmt.Errorf("delete %s: %w", nsDir, err)
}
if jsonOutput {
return printJSON(map[string]string{"name": name, "deleted": nsDir})
}
fmt.Fprintf(cmd.OutOrStdout(), "✓ Namespace deleted: %s (%s)\n", name, nsDir)
return nil
},
}
var nsInspectCmd = &cobra.Command{
Use: "inspect <name>",
Short: "Print the effective chain, merged env, and constraints",
Long: `Resolve a namespace's inheritance chain and print the merged env and unioned constraints (uses the resolver).`,
Args: cobra.ExactArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
name := args[0]
root := paths.Root()
cfgs, err := ns.ParseNSMdDir(root)
if err != nil {
return fmt.Errorf("load namespaces: %w", err)
}
if _, ok := cfgs[name]; !ok {
return fmt.Errorf("namespace %q not found under %s", name, root)
}
resolved, err := ns.Resolve(cfgs)
if err != nil {
return fmt.Errorf("resolve: %w", err)
}
r := resolved[name]
if r == nil {
return fmt.Errorf("namespace %q resolved to nil", name)
}
if jsonOutput {
return printJSON(map[string]any{
"name": r.Name,
"chain": r.Chain,
"env": r.Env,
"constraints": r.Constraints,
})
}
fmt.Fprintf(cmd.OutOrStdout(), "Namespace: %s\n", r.Name)
fmt.Fprintf(cmd.OutOrStdout(), "Chain: %s\n", strings.Join(r.Chain, " -> "))
fmt.Fprintln(cmd.OutOrStdout(), "Env:")
keys := sortedKeys(r.Env)
for _, k := range keys {
fmt.Fprintf(cmd.OutOrStdout(), " %s = %s\n", k, r.Env[k])
}
fmt.Fprintln(cmd.OutOrStdout(), "Constraints:")
if len(r.Constraints) == 0 {
fmt.Fprintln(cmd.OutOrStdout(), " (none)")
} else {
for _, c := range r.Constraints {
fmt.Fprintf(cmd.OutOrStdout(), " - %s\n", c)
}
}
return nil
},
}
var nsValidateCmd = &cobra.Command{
Use: "validate <name>",
Short: "Run cycle + missing-parent + schema checks on a namespace",
Long: `Validate a namespace's inheritance chain and ns.md frontmatter.
Exits 0 if valid, 1 on error. Runs over ALL namespaces under ORCA_HOME
(parsing + resolving validates cycles and missing parents across the
set).`,
Args: cobra.ExactArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
name := args[0]
root := paths.Root()
cfgs, err := ns.ParseNSMdDir(root)
if err != nil {
return fmt.Errorf("load namespaces: %w", err)
}
if _, ok := cfgs[name]; !ok {
return fmt.Errorf("namespace %q not found under %s", name, root)
}
resolved, err := ns.Resolve(cfgs)
if err != nil {
return fmt.Errorf("validate: %w", err)
}
r := resolved[name]
if r == nil {
return fmt.Errorf("namespace %q resolved to nil", name)
}
if jsonOutput {
return printJSON(map[string]any{
"name": r.Name,
"valid": true,
"chain": r.Chain,
})
}
fmt.Fprintf(cmd.OutOrStdout(), "✓ %s valid\n chain: %s\n", name, strings.Join(r.Chain, " -> "))
return nil
},
}
// renderNSMd writes a minimal ns.md frontmatter for `orca ns create`.
func renderNSMd(name string, parents []string, inheritsEnv, inheritsSecrets bool) string {
var b strings.Builder
b.WriteString("---\n")
b.WriteString("kind: Namespace\n")
b.WriteString("name: ")
b.WriteString(name)
b.WriteString("\n")
if len(parents) > 0 {
quoted := make([]string, len(parents))
for i, p := range parents {
quoted[i] = fmt.Sprintf("%q", p)
}
b.WriteString("parents: [")
b.WriteString(strings.Join(quoted, ", "))
b.WriteString("]\n")
}
fmt.Fprintf(&b, "inherits_env: %t\n", inheritsEnv)
fmt.Fprintf(&b, "inherits_secrets: %t\n", inheritsSecrets)
b.WriteString("---\n")
return b.String()
}
// dirNonEmpty returns an error wrapping the offending entry if dir
// contains any entries.
func dirNonEmpty(dir string) error {
entries, err := os.ReadDir(dir)
if err != nil {
if os.IsNotExist(err) {
return nil
}
return err
}
for _, e := range entries {
return fmt.Errorf("contains %s", e.Name())
}
return nil
}
func sortedKeys(m map[string]string) []string {
out := make([]string, 0, len(m))
for k := range m {
out = append(out, k)
}
sort.Strings(out)
return out
}
func init() {
nsCreateCmd.Flags().StringVar(&nsCreateParent, "parent", "", "parent namespace (default _defaults; the implicit root is always appended last)")
nsCreateCmd.Flags().BoolVar(&nsCreateInheritsEnv, "inherits-env", true, "inherit env from parents (default true)")
nsCreateCmd.Flags().BoolVar(&nsCreateInheritsSecret, "inherits-secrets", true, "inherit secrets from parents (default true)")
nsCmd.AddCommand(nsListCmd)
nsCmd.AddCommand(nsCreateCmd)
nsCmd.AddCommand(nsDeleteCmd)
nsCmd.AddCommand(nsInspectCmd)
nsCmd.AddCommand(nsValidateCmd)
rootCmd.AddCommand(nsCmd)
}
+407
View File
@@ -0,0 +1,407 @@
package cli
import (
"bytes"
"encoding/json"
"os"
"path/filepath"
"strings"
"testing"
"git.cloudinit.dev/coreci/orca/internal/paths"
)
// resetNSFlags zeroes the ns subcommand flag-bound vars so tests don't
// leak state.
func resetNSFlags() {
nsCreateParent = ""
nsCreateInheritsEnv = true
nsCreateInheritsSecret = true
}
func writeDefaultsNS(t *testing.T, root string) {
t.Helper()
nsDir := filepath.Join(root, "_defaults")
if err := os.MkdirAll(nsDir, 0o755); err != nil {
t.Fatalf("mkdir _defaults: %v", err)
}
body := "---\nkind: Namespace\nname: _defaults\ninherits_env: true\ninherits_secrets: true\n---\n# defaults\n"
if err := os.WriteFile(filepath.Join(nsDir, "ns.md"), []byte(body), 0o644); err != nil {
t.Fatalf("write _defaults ns.md: %v", err)
}
}
func writeCustomNS(t *testing.T, root, name, parentsList string) {
t.Helper()
nsDir := filepath.Join(root, name)
if err := os.MkdirAll(nsDir, 0o755); err != nil {
t.Fatalf("mkdir %s: %v", name, err)
}
body := "---\nkind: Namespace\nname: " + name + "\n"
if parentsList != "" {
body += "parents: " + parentsList + "\n"
}
body += "inherits_env: true\ninherits_secrets: true\n---\n# " + name + "\n"
if err := os.WriteFile(filepath.Join(nsDir, "ns.md"), []byte(body), 0o644); err != nil {
t.Fatalf("write %s ns.md: %v", name, err)
}
}
func TestNSListEmpty(t *testing.T) {
t.Setenv("ORCA_HOME", t.TempDir())
resetRootFlags(t)
resetNSFlags()
rootCmd.SetArgs([]string{"ns", "list"})
if err := rootCmd.Execute(); err != nil {
t.Fatalf("ns list: %v", err)
}
}
func TestNSListWithNamespaces(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
writeCustomNS(t, root, "prod", "")
rootCmd.SetArgs([]string{"ns", "list"})
if err := rootCmd.Execute(); err != nil {
t.Fatalf("ns list: %v", err)
}
}
func TestNSCreateHappy(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
rootCmd.SetArgs([]string{"ns", "create", "prod"})
if err := rootCmd.Execute(); err != nil {
t.Fatalf("ns create: %v", err)
}
if _, err := os.Stat(filepath.Join(root, "prod", "ns.md")); err != nil {
t.Fatalf("ns.md not created: %v", err)
}
for _, sub := range []string{"db", "jobs", "alloc"} {
if _, err := os.Stat(filepath.Join(root, "prod", sub)); err != nil {
t.Errorf("subdir %s not created: %v", sub, err)
}
}
}
func TestNSCreateWithParent(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
writeCustomNS(t, root, "base", "")
rootCmd.SetArgs([]string{"ns", "create", "child", "--parent", "base"})
if err := rootCmd.Execute(); err != nil {
t.Fatalf("ns create: %v", err)
}
data, err := os.ReadFile(filepath.Join(root, "child", "ns.md"))
if err != nil {
t.Fatalf("read ns.md: %v", err)
}
if !strings.Contains(string(data), "parents: [") || !strings.Contains(string(data), "\"base\"") {
t.Errorf("ns.md missing parents: %s", string(data))
}
}
func TestNSCreateExisting(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
writeCustomNS(t, root, "prod", "")
rootCmd.SetArgs([]string{"ns", "create", "prod"})
err := rootCmd.Execute()
if err == nil {
t.Fatal("expected error creating existing namespace, got nil")
}
if !strings.Contains(err.Error(), "already exists") {
t.Errorf("error = %q, want contains 'already exists'", err.Error())
}
}
func TestNSCreateDefaultsRefused(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
rootCmd.SetArgs([]string{"ns", "create", "_defaults"})
err := rootCmd.Execute()
if err == nil {
t.Fatal("expected error creating _defaults, got nil")
}
}
func TestNSCreateClusterRefused(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
rootCmd.SetArgs([]string{"ns", "create", "cluster"})
err := rootCmd.Execute()
if err == nil {
t.Fatal("expected error creating cluster, got nil")
}
}
func TestNSDeleteHappy(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
writeCustomNS(t, root, "prod", "")
rootCmd.SetArgs([]string{"ns", "delete", "prod"})
if err := rootCmd.Execute(); err != nil {
t.Fatalf("ns delete: %v", err)
}
if _, err := os.Stat(filepath.Join(root, "prod")); !os.IsNotExist(err) {
t.Errorf("prod dir still exists after delete")
}
}
func TestNSDeleteDefaultsRefused(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
rootCmd.SetArgs([]string{"ns", "delete", "_defaults"})
err := rootCmd.Execute()
if err == nil {
t.Fatal("expected error deleting _defaults, got nil")
}
}
func TestNSDeleteNonEmpty(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
writeCustomNS(t, root, "prod", "")
// put a job in jobs/
if err := os.MkdirAll(filepath.Join(root, "prod", "jobs"), 0o755); err != nil {
t.Fatalf("mkdir jobs: %v", err)
}
if err := os.WriteFile(filepath.Join(root, "prod", "jobs", "j1.md"), []byte("x"), 0o644); err != nil {
t.Fatalf("write job: %v", err)
}
rootCmd.SetArgs([]string{"ns", "delete", "prod"})
err := rootCmd.Execute()
if err == nil {
t.Fatal("expected error deleting non-empty namespace, got nil")
}
if !strings.Contains(err.Error(), "refusing to delete") {
t.Errorf("error = %q, want contains 'refusing to delete'", err.Error())
}
}
func TestNSDeleteMissing(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
rootCmd.SetArgs([]string{"ns", "delete", "ghost"})
err := rootCmd.Execute()
if err == nil {
t.Fatal("expected error deleting missing namespace, got nil")
}
}
func TestNSInspectHappy(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
writeCustomNS(t, root, "prod", "")
rootCmd.SetArgs([]string{"ns", "inspect", "prod"})
if err := rootCmd.Execute(); err != nil {
t.Fatalf("ns inspect: %v", err)
}
}
func TestNSInspectJSON(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
writeCustomNS(t, root, "prod", "")
var buf bytes.Buffer
rootCmd.SetOut(&buf)
rootCmd.SetArgs([]string{"ns", "inspect", "prod", "--json"})
if err := rootCmd.Execute(); err != nil {
t.Fatalf("ns inspect --json: %v", err)
}
var result map[string]any
if err := json.Unmarshal(bytes.TrimSpace(buf.Bytes()), &result); err != nil {
t.Fatalf("unmarshal: %v\n%s", err, buf.String())
}
if result["name"] != "prod" {
t.Errorf("name = %v, want prod", result["name"])
}
chain, _ := result["chain"].([]any)
if len(chain) < 2 {
t.Errorf("chain too short: %v", chain)
}
}
func TestNSInspectMissingNamespace(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
rootCmd.SetArgs([]string{"ns", "inspect", "ghost"})
err := rootCmd.Execute()
if err == nil {
t.Fatal("expected error for missing namespace, got nil")
}
}
func TestNSValidateHappy(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
writeCustomNS(t, root, "prod", "")
rootCmd.SetArgs([]string{"ns", "validate", "prod"})
if err := rootCmd.Execute(); err != nil {
t.Fatalf("ns validate: %v", err)
}
}
func TestNSValidateCycle(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
// a -> b, b -> a (cycle)
writeCustomNS(t, root, "a", "[\"b\"]")
writeCustomNS(t, root, "b", "[\"a\"]")
rootCmd.SetArgs([]string{"ns", "validate", "a"})
err := rootCmd.Execute()
if err == nil {
t.Fatal("expected cycle error, got nil")
}
if !strings.Contains(err.Error(), "cycle") {
t.Errorf("error = %q, want contains 'cycle'", err.Error())
}
}
func TestNSValidateMissingParent(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
writeCustomNS(t, root, "x", "[\"ghost\"]")
rootCmd.SetArgs([]string{"ns", "validate", "x"})
err := rootCmd.Execute()
if err == nil {
t.Fatal("expected missing-parent error, got nil")
}
if !strings.Contains(err.Error(), "ghost") || !strings.Contains(err.Error(), "not found") {
t.Errorf("error = %q, want contains ghost + not found", err.Error())
}
}
func TestNSValidateMissingNamespace(t *testing.T) {
root := t.TempDir()
t.Setenv("ORCA_HOME", root)
resetRootFlags(t)
resetNSFlags()
writeDefaultsNS(t, root)
rootCmd.SetArgs([]string{"ns", "validate", "ghost"})
err := rootCmd.Execute()
if err == nil {
t.Fatal("expected error for missing namespace, got nil")
}
}
func TestRenderNSMd(t *testing.T) {
body := renderNSMd("foo", []string{"_defaults"}, true, false)
if !strings.Contains(body, "kind: Namespace") {
t.Errorf("missing kind: %s", body)
}
if !strings.Contains(body, "name: foo") {
t.Errorf("missing name: %s", body)
}
if !strings.Contains(body, "inherits_env: true") {
t.Errorf("missing inherits_env true: %s", body)
}
if !strings.Contains(body, "inherits_secrets: false") {
t.Errorf("missing inherits_secrets false: %s", body)
}
}
func TestNSRootRegistered(t *testing.T) {
found := false
for _, c := range rootCmd.Commands() {
if c.Use == "ns" {
found = true
break
}
}
if !found {
t.Errorf("ns command not registered on root")
}
// ensure subcommands present
sub := map[string]bool{}
for _, c := range rootCmd.Commands() {
if c.Use == "ns" {
for _, sc := range c.Commands() {
sub[sc.Use] = true
}
}
}
for _, want := range []string{"list", "create <name>", "delete <name>", "inspect <name>", "validate <name>"} {
if !sub[want] {
t.Errorf("missing ns subcommand %q", want)
}
}
}
func TestNSListNoORCAHOME(t *testing.T) {
// ORCA_HOME points at a nonexistent dir; list should error.
t.Setenv("ORCA_HOME", filepath.Join(t.TempDir(), "nope"))
resetRootFlags(t)
resetNSFlags()
rootCmd.SetArgs([]string{"ns", "list"})
err := rootCmd.Execute()
if err == nil {
t.Fatal("expected error for missing ORCA_HOME, got nil")
}
}
var _ = paths.DefaultNamespace // keep paths import alive
+16 -3
View File
@@ -4,6 +4,7 @@ import (
"context"
"encoding/json"
"fmt"
"log/slog"
"os"
"github.com/spf13/cobra"
@@ -50,15 +51,27 @@ over feature richness.`,
}
var (
jsonOutput bool
systemNamespace bool
configPath string
jsonOutput bool
systemNamespace bool
configPath string
noDeprecationWarnings bool
)
func init() {
rootCmd.PersistentFlags().BoolVar(&jsonOutput, "json", false, "output in JSON format")
rootCmd.PersistentFlags().BoolVar(&systemNamespace, "system", false, "use system-level namespace root (/root/.orca) instead of user-level (~/.orca)")
rootCmd.PersistentFlags().StringVar(&configPath, "config", "", "path to config.hcl (overrides ~/.orca/config.hcl)")
rootCmd.PersistentFlags().BoolVar(&noDeprecationWarnings, "no-deprecation-warnings", false, "suppress v0.9 deprecation warnings (use during `orca upgrade` migrations)")
}
// warnDeprecated emits a v0.9 deprecation warning via slog.Warn unless
// the --no-deprecation-warnings global flag is set. Callers pass a
// human-readable message describing what changed. REQ-068.
func warnDeprecated(msg string) {
if noDeprecationWarnings {
return
}
slog.Warn(msg)
}
func configFromCtx(ctx context.Context) *config.Config {
+63 -2
View File
@@ -3,6 +3,7 @@ package config
import (
"fmt"
"os"
"strings"
"github.com/hashicorp/hcl/v2/hclsimple"
)
@@ -33,22 +34,82 @@ type Flags struct {
type Environ map[string]string
// Load is the config dispatcher (R-014). It tries each path in order,
// skipping missing files, and dispatches to the appropriate loader
// based on file extension: .hcl → LoadHCL (legacy, R-013),
// .md → LoadMarkdown (new Markdown-frontmatter loader), and
// .yaml/.yml → LoadMarkdown with an empty body. The first successfully
// decoded file wins. If no path exists or decodes, a zero Config is
// returned.
//
// The signature is preserved from the v0.8 single-loader API so
// internal/cli/root.go requires no changes yet.
func Load(paths ...string) (*Config, error) {
for _, p := range paths {
if _, err := os.Stat(p); err != nil {
continue
}
cfg, err := loadByExtension(p)
if err != nil {
return nil, err
}
return cfg, nil
}
return &Config{}, nil
}
func loadByExtension(p string) (*Config, error) {
ext := strings.ToLower(filepathExt(p))
switch ext {
case ".hcl":
return LoadHCL(p)
case ".md", ".markdown":
return LoadMarkdown(p)
case ".yaml", ".yml":
return LoadMarkdownYAML(p)
default:
// Unknown extension: hclsimple.Decode rejects non-.hcl
// suffixes, so for backward compat with the v0.8 single-loader
// behavior (which assumed HCL), decode the file content as HCL
// against a synthesized .hcl path.
data, err := os.ReadFile(p)
if err != nil {
return nil, fmt.Errorf("read config %s: %w", p, err)
}
var cfg Config
if err := hclsimple.Decode(p, data, nil, &cfg); err != nil {
if err := hclsimple.Decode(p+".hcl", data, nil, &cfg); err != nil {
return nil, fmt.Errorf("decode config %s: %w", p, err)
}
return &cfg, nil
}
return &Config{}, nil
}
// filepathExt is a thin wrapper around filepath.Ext to keep the import
// localized to the dispatcher. Returns the extension including the dot,
// lowercased by the caller.
func filepathExt(p string) string {
i := strings.LastIndex(p, ".")
if i < 0 {
return ""
}
return p[i:]
}
// LoadHCL decodes a legacy HCL config file (R-013).
//
// Deprecated: use LoadMarkdown or the dispatcher. HCL is legacy per
// R-013. Retained for the v0.9 dual-write window (REQ-090); new
// deployments should author config.md with YAML frontmatter.
func LoadHCL(path string) (*Config, error) {
data, err := os.ReadFile(path)
if err != nil {
return nil, fmt.Errorf("read config %s: %w", path, err)
}
var cfg Config
if err := hclsimple.Decode(path, data, nil, &cfg); err != nil {
return nil, fmt.Errorf("decode config %s: %w", path, err)
}
return &cfg, nil
}
func (c *Config) MergeOverrides(flags Flags, env Environ) *Config {
+111
View File
@@ -0,0 +1,111 @@
package config
import (
"path/filepath"
"testing"
)
func TestLoad_DispatchHCL(t *testing.T) {
p := writeTestFile(t, t.TempDir(), "config.hcl", exampleHCL)
cfg, err := Load(p)
if err != nil {
t.Fatalf("Load: %v", err)
}
if cfg.DBPath != "/tmp/orca/test.db" {
t.Errorf("DBPath=%q", cfg.DBPath)
}
if cfg.NodeCapacity == nil || cfg.NodeCapacity.CPU != 4 {
t.Errorf("NodeCapacity=%+v", cfg.NodeCapacity)
}
}
func TestLoad_DispatchMarkdown(t *testing.T) {
p := writeTestFile(t, t.TempDir(), "config.md", exampleMarkdown)
cfg, err := Load(p)
if err != nil {
t.Fatalf("Load: %v", err)
}
if cfg.DBPath != "/tmp/orca/test.db" {
t.Errorf("DBPath=%q", cfg.DBPath)
}
if cfg.ListenAddr != "127.0.0.1:9999" {
t.Errorf("ListenAddr=%q", cfg.ListenAddr)
}
if cfg.NodeCapacity == nil || cfg.NodeCapacity.CPU != 4 {
t.Errorf("NodeCapacity=%+v", cfg.NodeCapacity)
}
}
func TestLoad_DispatchYAML(t *testing.T) {
body := "listen_addr: 0.0.0.0:5555\ndb_path: /bare.db\nnode_capacity:\n cpu: 2\n memory_mb: 4096\n"
p := writeTestFile(t, t.TempDir(), "config.yaml", body)
cfg, err := Load(p)
if err != nil {
t.Fatalf("Load: %v", err)
}
if cfg.ListenAddr != "0.0.0.0:5555" {
t.Errorf("ListenAddr=%q", cfg.ListenAddr)
}
if cfg.DBPath != "/bare.db" {
t.Errorf("DBPath=%q", cfg.DBPath)
}
if cfg.NodeCapacity == nil || cfg.NodeCapacity.CPU != 2 {
t.Errorf("NodeCapacity=%+v", cfg.NodeCapacity)
}
}
func TestLoad_DispatchYML(t *testing.T) {
body := "listen_addr: 1.2.3.4:9\n"
p := writeTestFile(t, t.TempDir(), "config.yml", body)
cfg, err := Load(p)
if err != nil {
t.Fatalf("Load: %v", err)
}
if cfg.ListenAddr != "1.2.3.4:9" {
t.Errorf("ListenAddr=%q", cfg.ListenAddr)
}
}
func TestLoad_DispatchFirstExistingWins(t *testing.T) {
dir := t.TempDir()
missing := filepath.Join(dir, "missing.md")
existing := writeTestFile(t, dir, "real.hcl", exampleHCL)
cfg, err := Load(missing, existing)
if err != nil {
t.Fatalf("Load: %v", err)
}
if cfg.DBPath != "/tmp/orca/test.db" {
t.Errorf("DBPath=%q", cfg.DBPath)
}
}
func TestLoad_DispatchMissingReturnsZero(t *testing.T) {
cfg, err := Load(filepath.Join(t.TempDir(), "nope.md"))
if err != nil {
t.Fatalf("Load: %v", err)
}
if cfg == nil {
t.Fatal("nil config")
}
if cfg.DBPath != "" || cfg.ListenAddr != "" || cfg.NodeCapacity != nil {
t.Errorf("expected zero config, got %+v", cfg)
}
}
func TestLoad_DispatchHCLMalformed(t *testing.T) {
p := writeTestFile(t, t.TempDir(), "bad.hcl", "db_path = ")
if _, err := Load(p); err == nil {
t.Fatal("expected error for malformed HCL")
}
}
func TestLoad_DispatchUnknownExtFallsBackToHCL(t *testing.T) {
p := writeTestFile(t, t.TempDir(), "config.unknown", exampleHCL)
cfg, err := Load(p)
if err != nil {
t.Fatalf("Load: %v", err)
}
if cfg.DBPath != "/tmp/orca/test.db" {
t.Errorf("DBPath=%q", cfg.DBPath)
}
}
+220
View File
@@ -0,0 +1,220 @@
package config
import (
"fmt"
"os"
"strconv"
"strings"
)
// LoadMarkdown decodes a Markdown config file with YAML frontmatter
// (R-014). The file format is:
//
// ---
// listen_addr: 127.0.0.1:9999
// node_capacity:
// cpu: 4
// memory_mb: 8192
// db_path: /tmp/orca/test.db
// ---
//
// body prose (ignored)
//
// The frontmatter parser is a minimal hand-rolled key:value parser
// (no new dependencies; gopkg.in/yaml.v3 is not in go.mod). It supports
// flat scalar keys and one level of nested mapping (for node_capacity).
// The Markdown body after the closing "---" is ignored.
//
// The returned *Config is the same struct the HCL loader produces, so
// downstream consumers are unchanged.
func LoadMarkdown(path string) (*Config, error) {
data, err := os.ReadFile(path)
if err != nil {
return nil, fmt.Errorf("read config %s: %w", path, err)
}
return parseFrontmatter(string(data), path)
}
// LoadMarkdownYAML decodes a bare YAML file (no Markdown body) using the
// same minimal frontmatter parser. .yaml/.yml files are routed here by
// the dispatcher.
func LoadMarkdownYAML(path string) (*Config, error) {
data, err := os.ReadFile(path)
if err != nil {
return nil, fmt.Errorf("read config %s: %w", path, err)
}
// Treat the whole file as the frontmatter block (no surrounding ---).
return parseFrontmatterBlock(string(data), path)
}
func parseFrontmatter(content, path string) (*Config, error) {
block, ok := extractFrontmatter(content)
if !ok {
// No frontmatter delimiters: treat whole file as a bare block.
return parseFrontmatterBlock(content, path)
}
return parseFrontmatterBlock(block, path)
}
// extractFrontmatter returns the YAML block between the first pair of
// "---" delimiters and whether a frontmatter block was present.
func extractFrontmatter(content string) (string, bool) {
trimmed := strings.TrimLeft(content, "\r\n\t ")
if !strings.HasPrefix(trimmed, "---") {
return "", false
}
// Skip the opening delimiter line.
rest := trimmed[3:]
rest = strings.TrimLeft(rest, "\r\n")
// Find the closing delimiter line.
idx := strings.Index(rest, "\n---")
if idx < 0 {
return "", false
}
return rest[:idx], true
}
// parseFrontmatterBlock parses a minimal YAML-ish block into *Config.
// Supported shapes:
//
// key: value
// node_capacity:
// cpu: 4
// memory_mb: 8192
//
// Comments (# ...) and blank lines are ignored. Quoted scalar values
// ("..." or '...') are unwrapped. No flow collections, anchors, or
// multi-line strings are supported — by design, to avoid adding a YAML
// dependency for this small config surface.
func parseFrontmatterBlock(block, path string) (*Config, error) {
cfg := &Config{}
var inCapacity bool
lines := strings.Split(block, "\n")
for lineNo, raw := range lines {
line := stripComment(raw)
if strings.TrimSpace(line) == "" {
continue
}
indent := countIndent(line)
trimmed := strings.TrimSpace(line)
// A top-level key (no leading indent).
if indent == 0 {
inCapacity = false
key, val, ok := splitKV(trimmed)
if !ok {
continue
}
if val == "" {
// key with no value → nested mapping header (e.g. node_capacity:)
if key == "node_capacity" {
cfg.NodeCapacity = &CapacityConfig{}
inCapacity = true
}
continue
}
applyScalar(cfg, key, val, path, lineNo)
continue
}
// Indented line under a nested mapping.
if inCapacity && cfg.NodeCapacity != nil {
key, val, hasVal := splitKV(trimmed)
if !hasVal {
continue
}
switch key {
case "cpu":
if n, err := strconv.Atoi(strings.TrimSpace(val)); err == nil {
cfg.NodeCapacity.CPU = n
}
case "memory_mb":
if n, err := strconv.Atoi(strings.TrimSpace(val)); err == nil {
cfg.NodeCapacity.MemoryMB = n
}
}
}
}
return cfg, nil
}
func applyScalar(cfg *Config, key, val, path string, lineNo int) {
val = strings.TrimSpace(val)
switch key {
case "db_path":
cfg.DBPath = unquote(val)
case "listen_addr":
cfg.ListenAddr = unquote(val)
case "ca_path":
cfg.CAPath = unquote(val)
case "server_cert_path":
cfg.ServerCertPath = unquote(val)
case "server_key_path":
cfg.ServerKeyPath = unquote(val)
}
_ = path
_ = lineNo
}
func splitKV(s string) (key, val string, ok bool) {
idx := strings.Index(s, ":")
if idx < 0 {
return "", "", false
}
key = strings.TrimSpace(s[:idx])
val = strings.TrimSpace(s[idx+1:])
if key == "" {
return "", "", false
}
return key, val, true
}
func countIndent(s string) int {
n := 0
for _, r := range s {
if r == ' ' || r == '\t' {
n++
continue
}
break
}
return n
}
func stripComment(s string) string {
// Strip inline comments not inside quotes. Minimal: only strip
// when the '#' is preceded by whitespace or at line start.
inSingle := false
inDouble := false
for i := 0; i < len(s); i++ {
c := s[i]
switch c {
case '\'':
if !inDouble {
inSingle = !inSingle
}
case '"':
if !inSingle {
inDouble = !inDouble
}
case '#':
if !inSingle && !inDouble {
if i == 0 || s[i-1] == ' ' || s[i-1] == '\t' {
return s[:i]
}
}
}
}
return s
}
func unquote(s string) string {
if len(s) >= 2 {
if (s[0] == '"' && s[len(s)-1] == '"') || (s[0] == '\'' && s[len(s)-1] == '\'') {
return s[1 : len(s)-1]
}
}
return s
}
+186
View File
@@ -0,0 +1,186 @@
package config
import (
"os"
"path/filepath"
"testing"
)
func writeTestFile(t *testing.T, dir, name, content string) string {
t.Helper()
p := filepath.Join(dir, name)
if err := os.WriteFile(p, []byte(content), 0644); err != nil {
t.Fatalf("write %s: %v", p, err)
}
return p
}
const exampleMarkdown = `---
listen_addr: "127.0.0.1:9999"
db_path: "/tmp/orca/test.db"
ca_path: "/tmp/orca/ca.crt"
server_cert_path: "/tmp/orca/server.crt"
server_key_path: "/tmp/orca/server.key"
node_capacity:
cpu: 4
memory_mb: 8192
---
# Orca config
This is prose body and is ignored by the loader.
`
func TestLoadMarkdown_Full(t *testing.T) {
p := writeTestFile(t, t.TempDir(), "config.md", exampleMarkdown)
cfg, err := LoadMarkdown(p)
if err != nil {
t.Fatalf("LoadMarkdown: %v", err)
}
if cfg.DBPath != "/tmp/orca/test.db" {
t.Errorf("DBPath=%q", cfg.DBPath)
}
if cfg.ListenAddr != "127.0.0.1:9999" {
t.Errorf("ListenAddr=%q", cfg.ListenAddr)
}
if cfg.CAPath != "/tmp/orca/ca.crt" {
t.Errorf("CAPath=%q", cfg.CAPath)
}
if cfg.ServerCertPath != "/tmp/orca/server.crt" {
t.Errorf("ServerCertPath=%q", cfg.ServerCertPath)
}
if cfg.ServerKeyPath != "/tmp/orca/server.key" {
t.Errorf("ServerKeyPath=%q", cfg.ServerKeyPath)
}
if cfg.NodeCapacity == nil {
t.Fatal("NodeCapacity nil")
}
if cfg.NodeCapacity.CPU != 4 {
t.Errorf("CPU=%d", cfg.NodeCapacity.CPU)
}
if cfg.NodeCapacity.MemoryMB != 8192 {
t.Errorf("MemoryMB=%d", cfg.NodeCapacity.MemoryMB)
}
}
func TestLoadMarkdown_NoFrontmatter(t *testing.T) {
// No delimiters: whole file treated as a bare YAML block.
body := "listen_addr: 0.0.0.0:1234\ndb_path: /x/y.db\n"
p := writeTestFile(t, t.TempDir(), "config.md", body)
cfg, err := LoadMarkdown(p)
if err != nil {
t.Fatalf("LoadMarkdown: %v", err)
}
if cfg.ListenAddr != "0.0.0.0:1234" {
t.Errorf("ListenAddr=%q", cfg.ListenAddr)
}
if cfg.DBPath != "/x/y.db" {
t.Errorf("DBPath=%q", cfg.DBPath)
}
}
func TestLoadMarkdown_OnlyBody(t *testing.T) {
body := `---
---
# Just prose, no keys
`
p := writeTestFile(t, t.TempDir(), "config.md", body)
cfg, err := LoadMarkdown(p)
if err != nil {
t.Fatalf("LoadMarkdown: %v", err)
}
if cfg.DBPath != "" || cfg.ListenAddr != "" || cfg.NodeCapacity != nil {
t.Errorf("expected zero config, got %+v", cfg)
}
}
func TestLoadMarkdown_CommentsAndBlanks(t *testing.T) {
body := `---
# a comment
listen_addr: "127.0.0.1:9999"
db_path: "/tmp/orca/test.db" # inline comment
node_capacity:
cpu: 4 # cores
memory_mb: 8192
---
`
p := writeTestFile(t, t.TempDir(), "config.md", body)
cfg, err := LoadMarkdown(p)
if err != nil {
t.Fatalf("LoadMarkdown: %v", err)
}
if cfg.ListenAddr != "127.0.0.1:9999" {
t.Errorf("ListenAddr=%q", cfg.ListenAddr)
}
if cfg.DBPath != "/tmp/orca/test.db" {
t.Errorf("DBPath=%q", cfg.DBPath)
}
if cfg.NodeCapacity == nil || cfg.NodeCapacity.CPU != 4 || cfg.NodeCapacity.MemoryMB != 8192 {
t.Errorf("NodeCapacity=%+v", cfg.NodeCapacity)
}
}
func TestLoadMarkdownYAML_Bare(t *testing.T) {
body := "listen_addr: 0.0.0.0:5555\ndb_path: /bare.db\nnode_capacity:\n cpu: 2\n memory_mb: 4096\n"
p := writeTestFile(t, t.TempDir(), "config.yaml", body)
cfg, err := LoadMarkdownYAML(p)
if err != nil {
t.Fatalf("LoadMarkdownYAML: %v", err)
}
if cfg.ListenAddr != "0.0.0.0:5555" {
t.Errorf("ListenAddr=%q", cfg.ListenAddr)
}
if cfg.DBPath != "/bare.db" {
t.Errorf("DBPath=%q", cfg.DBPath)
}
if cfg.NodeCapacity == nil || cfg.NodeCapacity.CPU != 2 || cfg.NodeCapacity.MemoryMB != 4096 {
t.Errorf("NodeCapacity=%+v", cfg.NodeCapacity)
}
}
func TestLoadMarkdown_ReadError(t *testing.T) {
missing := filepath.Join(t.TempDir(), "nope.md")
if _, err := LoadMarkdown(missing); err == nil {
t.Fatal("expected error for missing file")
}
}
func TestExtractFrontmatter(t *testing.T) {
cases := []struct {
name string
input string
block string
present bool
}{
{"standard", "---\nkey: val\n---\nbody", "key: val", true},
{"leading-blanks", "\n\n---\nkey: val\n---\n", "key: val", true},
{"no-delimiters", "key: val\n", "key: val", false},
{"only-open", "---\nkey: val\n", "key: val", false},
}
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
block, ok := extractFrontmatter(tc.input)
if ok != tc.present {
t.Errorf("present=%v want %v", ok, tc.present)
}
if tc.present && block != tc.block {
t.Errorf("block=%q want %q", block, tc.block)
}
})
}
}
func TestUnquote(t *testing.T) {
if got, want := unquote(`"hello"`), "hello"; got != want {
t.Errorf("unquote double = %q want %q", got, want)
}
if got, want := unquote(`'hello'`), "hello"; got != want {
t.Errorf("unquote single = %q want %q", got, want)
}
if got, want := unquote("bare"), "bare"; got != want {
t.Errorf("unquote bare = %q want %q", got, want)
}
}
+6
View File
@@ -9,6 +9,12 @@
// - structured JSON via writeJSON
// - no secrets in logs
// - input validation on path/query/body
//
// Deprecated: v0.9 re-architecture replaces this with SSH-push to bare
// servers (no orca binary on servers) per R-001. The orca daemon is
// repurposed to drain-and-stop in v0.10-P05 and scheduled for deletion
// in v0.10-P14. See .ciagent/PRD_v0.9.md R-001/R-006. The dual-write
// window (REQ-090/REQ-085) keeps this package compiling until v0.10-P14.
package daemon
import (
+80
View File
@@ -0,0 +1,80 @@
// Package emit defines the render-format contract between Go-side emitters
// and bash-side appliers (grill C-16). Every rendered artifact is a JSON
// object with a versioned schema; both sides validate against it to prevent
// emitter/applier drift.
package emit
import (
"encoding/json"
"errors"
"fmt"
)
// SchemaVersion is the canonical versioned schema identifier for render
// contracts. Bump the suffix when the contract shape changes.
const SchemaVersion = "orca.emit/v1"
// Kind enumerates the rendered-artifact kinds. Each maps to an emitter
// implementation and a matching bash-side applier.
type Kind string
const (
KindSystemd Kind = "systemd"
KindTraefik Kind = "traefik"
KindSyncthing Kind = "syncthing"
KindSudoers Kind = "sudoers"
KindSSHD Kind = "sshd"
KindEnvFile Kind = "envfile"
KindCredential Kind = "credential"
)
// Artifact is a single rendered file destined for a peer. The bash-side
// applier reads this JSON and writes Content to Path with the given Mode.
type Artifact struct {
SchemaVersion string `json:"schema_version"`
Kind Kind `json:"kind"`
Path string `json:"path"`
Content string `json:"content"`
Mode string `json:"mode"`
}
// Validate checks that an Artifact conforms to the render contract.
// Returns a structured error if any field is missing or invalid.
func (a *Artifact) Validate() error {
if a.SchemaVersion != SchemaVersion {
return fmt.Errorf("emit: schema_version mismatch: got %q want %q", a.SchemaVersion, SchemaVersion)
}
if a.Kind == "" {
return errors.New("emit: kind is required")
}
if a.Path == "" {
return errors.New("emit: path is required")
}
if a.Mode == "" {
return errors.New("emit: mode is required")
}
return nil
}
// Marshal serializes an Artifact to JSON for transport to the bash applier.
func (a *Artifact) Marshal() ([]byte, error) {
if err := a.Validate(); err != nil {
return nil, err
}
return json.Marshal(a)
}
// UnmarshalArtifact parses a JSON byte slice into an Artifact and validates
// it against the contract. The bash-side applier (via orca-verify-render.sh)
// uses this same validation; the bash side rejects unparseable input with a
// structured error, never silently (grill C-16).
func UnmarshalArtifact(data []byte) (*Artifact, error) {
var a Artifact
if err := json.Unmarshal(data, &a); err != nil {
return nil, fmt.Errorf("emit: unmarshal: %w", err)
}
if err := a.Validate(); err != nil {
return nil, err
}
return &a, nil
}
+120
View File
@@ -0,0 +1,120 @@
package emit
import (
"encoding/json"
"strings"
"testing"
)
func TestArtifactValidate_valid(t *testing.T) {
a := &Artifact{
SchemaVersion: SchemaVersion,
Kind: KindSystemd,
Path: "/etc/systemd/system/orca-alloc.service",
Content: "[Service]\nExecStart=/bin/true\n",
Mode: "0644",
}
if err := a.Validate(); err != nil {
t.Fatalf("expected valid, got %v", err)
}
}
func TestArtifactValidate_schemaVersionMismatch(t *testing.T) {
a := &Artifact{SchemaVersion: "orca.emit/v0", Kind: KindSystemd, Path: "/x", Mode: "0644"}
err := a.Validate()
if err == nil {
t.Fatal("expected error for mismatched schema_version")
}
if !strings.Contains(err.Error(), "schema_version mismatch") {
t.Fatalf("expected schema_version error, got %v", err)
}
}
func TestArtifactValidate_missingKind(t *testing.T) {
a := &Artifact{SchemaVersion: SchemaVersion, Path: "/x", Mode: "0644"}
err := a.Validate()
if err == nil || !strings.Contains(err.Error(), "kind is required") {
t.Fatalf("expected kind-required error, got %v", err)
}
}
func TestArtifactValidate_missingPath(t *testing.T) {
a := &Artifact{SchemaVersion: SchemaVersion, Kind: KindTraefik, Mode: "0644"}
err := a.Validate()
if err == nil || !strings.Contains(err.Error(), "path is required") {
t.Fatalf("expected path-required error, got %v", err)
}
}
func TestArtifactValidate_missingMode(t *testing.T) {
a := &Artifact{SchemaVersion: SchemaVersion, Kind: KindSudoers, Path: "/x"}
err := a.Validate()
if err == nil || !strings.Contains(err.Error(), "mode is required") {
t.Fatalf("expected mode-required error, got %v", err)
}
}
func TestMarshalValidate_rejectsInvalid(t *testing.T) {
a := &Artifact{SchemaVersion: "bad", Kind: "", Path: "", Mode: ""}
if _, err := a.Marshal(); err == nil {
t.Fatal("expected Marshal to reject invalid artifact")
}
}
func TestUnmarshalArtifact_valid(t *testing.T) {
raw := `{"schema_version":"orca.emit/v1","kind":"systemd","path":"/x","content":"c","mode":"0644"}`
a, err := UnmarshalArtifact([]byte(raw))
if err != nil {
t.Fatalf("expected valid, got %v", err)
}
if a.Kind != KindSystemd {
t.Fatalf("expected kind systemd, got %s", a.Kind)
}
}
func TestUnmarshalArtifact_rejectsBadJSON(t *testing.T) {
if _, err := UnmarshalArtifact([]byte("not json")); err == nil {
t.Fatal("expected error for bad JSON")
}
}
func TestUnmarshalArtifact_rejectsSchemaMismatch(t *testing.T) {
raw := `{"schema_version":"orca.emit/v2","kind":"x","path":"/x","mode":"0644"}`
if _, err := UnmarshalArtifact([]byte(raw)); err == nil {
t.Fatal("expected error for schema mismatch")
}
}
func TestRoundTrip(t *testing.T) {
orig := &Artifact{
SchemaVersion: SchemaVersion,
Kind: KindSyncthing,
Path: "/etc/syncthing/config.xml",
Content: "<config/>",
Mode: "0600",
}
data, err := orig.Marshal()
if err != nil {
t.Fatalf("Marshal: %v", err)
}
back, err := UnmarshalArtifact(data)
if err != nil {
t.Fatalf("Unmarshal: %v", err)
}
if back.Path != orig.Path || back.Kind != orig.Kind || back.Mode != orig.Mode {
t.Fatalf("round-trip mismatch: %+v vs %+v", orig, back)
}
}
func TestAllKinds(t *testing.T) {
for _, k := range []Kind{KindSystemd, KindTraefik, KindSyncthing, KindSudoers, KindSSHD, KindEnvFile, KindCredential} {
a := &Artifact{SchemaVersion: SchemaVersion, Kind: k, Path: "/x", Mode: "0644"}
if err := a.Validate(); err != nil {
t.Errorf("kind %s: %v", k, err)
}
// verify it marshals
if _, err := json.Marshal(a); err != nil {
t.Errorf("marshal kind %s: %v", k, err)
}
}
}
+94
View File
@@ -0,0 +1,94 @@
// Package emitter defines the Layer-4 emitter interface (REQ-074,
// I-B-002): the bridge between the declarative *jobspec.WorkloadSpec
// and the server-side files. An Emitter renders a *WorkloadSpec into a
// slice of File artifacts that the SSH-push transport SCPs to peers.
//
// Emitters are registered per workload kind + runtime (e.g.
// "service:wasm", "job:process", "daemonset:wasm"). The Registry looks
// up the right emitter by "<kind>:<runtime>" and delegates. Unknown
// combinations return an error so the caller can fail fast before any
// file is written.
//
// P0c only ships the interface, the Registry, a stub SystemdEmitter
// (process runtime), and the File/Node value types. The full emitter
// implementations (systemd lifecycle hooks, Traefik, Syncthing,
// sockets) land in later phases (P02 Traefik, P04 lifecycle, P08
// sockets, P09 Syncthing, v0.10-P03 secrets).
package emitter
import (
"fmt"
"strings"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// File is a single rendered artifact destined for a peer. The SSH-push
// transport writes Content to Path atomically (write-to-tmp + rename)
// with the given Mode (an octal string like "0644").
type File struct {
Path string
Content string
Mode string
}
// Node is the minimal peer description an emitter needs to render
// node-specific paths. It carries the hostname, the runtimes available
// on the node (so emitters can branch), and the node tags (used by
// DaemonSet matching and affinity in P05).
type Node struct {
Hostname string
Runtime []string
Tags []string
}
// Emitter renders a *jobspec.WorkloadSpec for a given Node into a slice
// of File artifacts. Implementations are registered with a Registry
// keyed by "<kind>:<runtime>".
type Emitter interface {
Render(spec *jobspec.WorkloadSpec, node *Node) ([]File, error)
}
// Registry holds emitters keyed by "<kind>:<runtime>" (e.g.
// "service:process", "job:wasm"). The zero-value Registry is not
// usable; construct one with NewRegistry.
type Registry struct {
emitters map[string]Emitter
}
// NewRegistry returns an empty Registry ready for Register calls.
func NewRegistry() *Registry {
return &Registry{emitters: make(map[string]Emitter)}
}
// Register registers an Emitter under the given key. The key is
// "<kind>:<runtime>" (e.g. "job:process"). Registering twice under the
// same key overwrites the prior registration (last-wins) to keep the
// surface simple; callers are responsible for not double-registering.
func (r *Registry) Register(key string, e Emitter) {
r.emitters[key] = e
}
// Render looks up the emitter for "<kind>:<runtime>" in the registry and
// delegates to it. The kind is lowercased so the canonical spec kinds
// (Job, Service, DaemonSet) map to the lowercase registry keys
// ("job:process", "service:wasm", "daemonset:process"). Returns an error
// if the spec is nil, the spec is missing its Kind, the runtime is
// missing, or no emitter is registered for the combination.
func (r *Registry) Render(spec *jobspec.WorkloadSpec, node *Node) ([]File, error) {
if spec == nil {
return nil, fmt.Errorf("emitter: spec is nil")
}
if strings.TrimSpace(spec.Kind) == "" {
return nil, fmt.Errorf("emitter: spec kind is empty")
}
if spec.Runtime == nil {
return nil, fmt.Errorf("emitter: spec runtime is nil")
}
key := strings.ToLower(spec.Kind) + ":" + spec.Runtime.OneOf
e, ok := r.emitters[key]
if !ok {
return nil, fmt.Errorf("emitter: no emitter registered for %q (kind:runtime)", key)
}
return e.Render(spec, node)
}
+163
View File
@@ -0,0 +1,163 @@
package emitter
import (
"errors"
"strings"
"testing"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// mockEmitter is a test-only Emitter that returns a fixed File slice
// (or an error) so the Registry tests do not depend on the
// SystemdEmitter. Implements Emitter via value receiver.
type mockEmitter struct {
files []File
err error
}
func (m mockEmitter) Render(spec *jobspec.WorkloadSpec, node *Node) ([]File, error) {
if m.err != nil {
return nil, m.err
}
out := make([]File, len(m.files))
copy(out, m.files)
return out, nil
}
func TestRegistry_RegisterAndRender(t *testing.T) {
r := NewRegistry()
want := []File{{Path: "/tmp/a", Content: "alpha", Mode: "0644"}}
r.Register("job:process", mockEmitter{files: want})
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "demo",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/true"},
}
node := &Node{Hostname: "node-1", Runtime: []string{"process"}}
got, err := r.Render(spec, node)
if err != nil {
t.Fatalf("Render: %v", err)
}
if len(got) != 1 {
t.Fatalf("got %d files, want 1", len(got))
}
if got[0] != want[0] {
t.Errorf("file = %+v, want %+v", got[0], want[0])
}
}
func TestRegistry_UnknownKindRuntime(t *testing.T) {
r := NewRegistry()
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: "wasm"},
}
_, err := r.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for unknown kind:runtime, got nil")
}
if !strings.Contains(err.Error(), "no emitter registered") {
t.Errorf("error = %q, want 'no emitter registered'", err.Error())
}
if !strings.Contains(err.Error(), "service:wasm") {
t.Errorf("error = %q, want it to mention 'service:wasm'", err.Error())
}
}
func TestRegistry_MultipleEmittersCorrectSelected(t *testing.T) {
r := NewRegistry()
jobFiles := []File{{Path: "/tmp/job", Content: "job", Mode: "0644"}}
svcFiles := []File{{Path: "/tmp/svc", Content: "svc", Mode: "0644"}}
dsFiles := []File{{Path: "/tmp/ds", Content: "ds", Mode: "0644"}}
r.Register("job:process", mockEmitter{files: jobFiles})
r.Register("service:process", mockEmitter{files: svcFiles})
r.Register("daemonset:process", mockEmitter{files: dsFiles})
cases := []struct {
kind string
runtime string
wantPath string
}{
{"Job", "process", "/tmp/job"},
{"Service", "process", "/tmp/svc"},
{"DaemonSet", "process", "/tmp/ds"},
}
for _, tc := range cases {
t.Run(tc.kind+":"+tc.runtime, func(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: tc.kind,
Name: "x",
Runtime: &jobspec.RuntimeBlock{OneOf: tc.runtime, Command: "/bin/x"},
}
got, err := r.Render(spec, &Node{Hostname: "n"})
if err != nil {
t.Fatalf("Render: %v", err)
}
if len(got) != 1 {
t.Fatalf("got %d files, want 1", len(got))
}
if got[0].Path != tc.wantPath {
t.Errorf("path = %q, want %q", got[0].Path, tc.wantPath)
}
})
}
}
func TestRegistry_NilSpec(t *testing.T) {
r := NewRegistry()
_, err := r.Render(nil, &Node{})
if err == nil {
t.Fatal("expected error for nil spec")
}
if !strings.Contains(err.Error(), "spec is nil") {
t.Errorf("error = %q, want 'spec is nil'", err.Error())
}
}
func TestRegistry_EmptyKind(t *testing.T) {
r := NewRegistry()
spec := &jobspec.WorkloadSpec{Runtime: &jobspec.RuntimeBlock{OneOf: "process"}}
_, err := r.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for empty kind")
}
if !strings.Contains(err.Error(), "kind is empty") {
t.Errorf("error = %q, want 'kind is empty'", err.Error())
}
}
func TestRegistry_NilRuntime(t *testing.T) {
r := NewRegistry()
spec := &jobspec.WorkloadSpec{Kind: "Job", Name: "x"}
_, err := r.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for nil runtime")
}
if !strings.Contains(err.Error(), "runtime is nil") {
t.Errorf("error = %q, want 'runtime is nil'", err.Error())
}
}
func TestRegistry_EmitterErrorPropagates(t *testing.T) {
r := NewRegistry()
wantErr := errors.New("boom")
r.Register("job:process", mockEmitter{err: wantErr})
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "x",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/x"},
}
_, err := r.Render(spec, &Node{})
if !errors.Is(err, wantErr) {
t.Errorf("err = %v, want %v", err, wantErr)
}
}
// Compile-time assertion that mockEmitter and SystemdEmitter implement
// Emitter.
var (
_ Emitter = mockEmitter{}
_ Emitter = SystemdEmitter{}
)
+128
View File
@@ -0,0 +1,128 @@
package emitter
import (
"fmt"
"strings"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// SocketEmitter renders the systemd directives that implement the
// R-007 socket-plumbing contract: workloads bind to
// /run/orca/alloc-<id>/port-<name>.sock unless overridden via
// service.bind = "127.0.0.1" (the only documented opt-in).
//
// The systemd side of the contract uses two directives:
//
// - RuntimeDirectory=orca/alloc-<alloc-id> — systemd creates
// /run/orca/alloc-<alloc-id>/ owned by the service user (orca:orca)
// with mode 0750. The directory is removed when the unit stops
// (RuntimeDirectory= semantics). P08 emits one RuntimeDirectory=
// line per port so each port's socket directory is created; the
// alloc-id placeholder is spec.Name (the real alloc-id is assigned
// by the scheduler at submit time — see allocIDFor).
//
// - ExecStartPre= — only when service.bind is "127.0.0.1" (the TCP
// opt-in). In that case the workload binds a TCP port directly
// (no socket), and the ExecStartPre is a placeholder that records
// the bind (the actual bind happens in the process; the directive
// is a no-op marker so operators can see the bind mode in the unit
// file). When service.bind is empty (the default), the workload
// binds the socket and no ExecStartPre is emitted for sockets.
//
// The socket path format is /run/orca/alloc-<alloc-id>/port-<port-name>.sock
// where alloc-id is a PLACEHOLDER (spec.Name) — the real alloc-id is
// assigned at submit time by the scheduler. The placeholder is
// documented in the rendered unit via a comment so operators reading
// the unit file understand the substitution.
//
// P08 is a PLAN/plumbing layer — the actual socket activation (socket
// unit files, systemd socket-activation passing the pre-bound socket
// fd to the process) lands in v0.10. P08 just renders the
// RuntimeDirectory= lines and the optional TCP-bind ExecStartPre so
// the directory exists at runtime.
type SocketEmitter struct{}
// runtimeDirectoryRoot is the systemd RuntimeDirectory path root.
// systemd joins this with the RuntimeDirectory= value to create
// /run/orca/alloc-<id>. The leading slash is implicit in systemd
// (RuntimeDirectory= is relative to /run).
const runtimeDirectoryRoot = "orca"
// SocketPath returns the R-007 socket path for a port on the given
// alloc-id. The alloc-id is the placeholder spec.Name when the real
// alloc-id is not yet known (the scheduler assigns the real alloc-id
// at submit time).
func SocketPath(allocID, portName string) string {
return fmt.Sprintf("/run/orca/alloc-%s/port-%s.sock", allocID, portName)
}
// RenderSocketLines renders the systemd directives that implement
// the R-007 socket plumbing for the given spec. The lines are returned
// WITHOUT a trailing newline so the caller (the systemd emitter) can
// append them to the [Service] block with consistent formatting.
//
// The returned lines are:
//
// - one RuntimeDirectory= line per port (so each port's socket
// directory is created by systemd at unit start).
// - a comment documenting the alloc-id placeholder.
// - when service.bind is "127.0.0.1", an ExecStartPre= marker that
// records the TCP opt-in (the actual bind is in the process).
//
// Returns an empty slice when the spec has no ports (no socket
// plumbing needed — e.g. a Job or a port-less DaemonSet).
func (SocketEmitter) RenderSocketLines(spec *jobspec.WorkloadSpec) []string {
if spec == nil || len(spec.Ports) == 0 {
return nil
}
allocID := allocIDForSocket(spec)
var lines []string
// One RuntimeDirectory= per port. systemd dedupes identical
// values, but we emit one per port so the unit file is
// self-documenting (each port maps to a directory entry).
for _, p := range spec.Ports {
lines = append(lines, fmt.Sprintf("RuntimeDirectory=%s/alloc-%s", runtimeDirectoryRoot, allocID))
// Document the socket path this directory serves. systemd
// ignores comment lines (lines starting with '#').
lines = append(lines, fmt.Sprintf("# socket: %s", SocketPath(allocID, p.Name)))
}
// TCP opt-in: when service.bind is 127.0.0.1, the workload binds
// a TCP port directly instead of the socket. We emit an
// ExecStartPre marker so the bind mode is visible in the unit
// file. The actual bind is in the process; the marker is a
// no-op (echo to journald).
if spec.Service != nil && strings.TrimSpace(spec.Service.Bind) != "" {
if isTCPOptIn(spec.Service.Bind) {
for _, p := range spec.Ports {
lines = append(lines, fmt.Sprintf("ExecStartPre=/bin/echo orca: bind %s port %s (tcp, R-007 opt-in)", spec.Service.Bind, p.Name))
}
}
}
return lines
}
// allocIDForSocket returns the alloc-id placeholder for the spec. The
// real alloc-id is assigned by the scheduler at submit time; P08 uses
// spec.Name as a deterministic placeholder so the rendered unit is
// stable across re-renders. This mirrors the Traefik emitter's
// allocIDFor (which uses the node hostname for the Traefik
// dynamic-config server URL); the systemd unit is per-alloc, so
// spec.Name is the right placeholder here.
func allocIDForSocket(spec *jobspec.WorkloadSpec) string {
if spec == nil || strings.TrimSpace(spec.Name) == "" {
return "<allocID>"
}
return spec.Name
}
// isTCPOptIn returns true when the bind value is the documented
// 127.0.0.1 TCP opt-in (R-007). Other valid IPs (::1, etc.) are also
// TCP opt-ins (any non-empty bind opts out of the socket default); we
// only emit the marker for 127.0.0.1 because that is the only
// documented opt-in per the PRD — other IPs are accepted by the
// schema validator but are operator-specific and we do not
// second-guess them.
func isTCPOptIn(bind string) bool {
return strings.TrimSpace(bind) == "127.0.0.1"
}
+265
View File
@@ -0,0 +1,265 @@
package emitter
import (
"strings"
"testing"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
func TestSocketEmitter_RenderSocketLines_NoPorts(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "backup",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/rsync"},
}
lines := (SocketEmitter{}).RenderSocketLines(spec)
if len(lines) != 0 {
t.Errorf("got %d lines, want 0 for no ports: %v", len(lines), lines)
}
}
func TestSocketEmitter_RenderSocketLines_NilSpec(t *testing.T) {
lines := (SocketEmitter{}).RenderSocketLines(nil)
if lines != nil {
t.Errorf("nil spec should return nil, got %v", lines)
}
}
func TestSocketEmitter_RenderSocketLines_SinglePort(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
lines := (SocketEmitter{}).RenderSocketLines(spec)
// Expect: RuntimeDirectory + comment. No TCP bind (default socket).
wantRT := "RuntimeDirectory=orca/alloc-web"
if !contains(lines, wantRT) {
t.Errorf("lines %v missing %q", lines, wantRT)
}
wantSock := "# socket: /run/orca/alloc-web/port-http.sock"
if !contains(lines, wantSock) {
t.Errorf("lines %v missing %q", lines, wantSock)
}
for _, l := range lines {
if strings.HasPrefix(l, "ExecStartPre=") {
t.Errorf("socket bind should not emit ExecStartPre (no TCP opt-in): %s", l)
}
}
}
func TestSocketEmitter_RenderSocketLines_MultiplePorts(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "api",
Ports: []jobspec.PortSpec{
{Name: "http", Port: 8080},
{Name: "grpc", Port: 9090},
},
}
lines := (SocketEmitter{}).RenderSocketLines(spec)
// Two RuntimeDirectory lines (one per port).
count := 0
for _, l := range lines {
if l == "RuntimeDirectory=orca/alloc-api" {
count++
}
}
if count != 2 {
t.Errorf("RuntimeDirectory count = %d, want 2 (one per port)", count)
}
if !contains(lines, "# socket: /run/orca/alloc-api/port-http.sock") {
t.Errorf("missing http socket comment")
}
if !contains(lines, "# socket: /run/orca/alloc-api/port-grpc.sock") {
t.Errorf("missing grpc socket comment")
}
}
func TestSocketEmitter_RenderSocketLines_TCPBind127(t *testing.T) {
// service.bind = 127.0.0.1 → TCP opt-in → ExecStartPre marker per port.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Service: &jobspec.ServiceBlock{Bind: "127.0.0.1"},
}
lines := (SocketEmitter{}).RenderSocketLines(spec)
found := false
for _, l := range lines {
if strings.HasPrefix(l, "ExecStartPre=/bin/echo orca: bind 127.0.0.1 port http (tcp, R-007 opt-in)") {
found = true
}
}
if !found {
t.Errorf("missing TCP bind ExecStartPre marker; lines: %v", lines)
}
}
func TestSocketEmitter_RenderSocketLines_TCPBindIPv6(t *testing.T) {
// Non-127.0.0.1 bind is accepted by schema but not the documented
// opt-in; the marker is only emitted for 127.0.0.1. The
// RuntimeDirectory lines are still emitted (the directory exists
// regardless of bind mode — sockets or TCP).
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Service: &jobspec.ServiceBlock{Bind: "::1"},
}
lines := (SocketEmitter{}).RenderSocketLines(spec)
for _, l := range lines {
if strings.HasPrefix(l, "ExecStartPre=") {
t.Errorf("::1 bind should NOT emit TCP marker (only 127.0.0.1 is documented opt-in): %s", l)
}
}
if !contains(lines, "RuntimeDirectory=orca/alloc-web") {
t.Errorf("RuntimeDirectory should still be emitted for ::1 bind")
}
}
func TestSocketEmitter_RenderSocketLines_EmptyBindSocket(t *testing.T) {
// Empty bind → default socket → no TCP marker, but RuntimeDirectory emitted.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Service: &jobspec.ServiceBlock{Bind: ""},
}
lines := (SocketEmitter{}).RenderSocketLines(spec)
for _, l := range lines {
if strings.HasPrefix(l, "ExecStartPre=") {
t.Errorf("empty bind should NOT emit TCP marker: %s", l)
}
}
if !contains(lines, "RuntimeDirectory=orca/alloc-web") {
t.Errorf("RuntimeDirectory missing for empty bind")
}
}
func TestSocketEmitter_RenderSocketLines_NilService(t *testing.T) {
// No service block → default socket → no TCP marker, but RuntimeDirectory emitted.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
lines := (SocketEmitter{}).RenderSocketLines(spec)
for _, l := range lines {
if strings.HasPrefix(l, "ExecStartPre=") {
t.Errorf("nil service should NOT emit TCP marker: %s", l)
}
}
if !contains(lines, "RuntimeDirectory=orca/alloc-web") {
t.Errorf("RuntimeDirectory missing for nil service")
}
}
func TestSocketEmitter_SocketPath(t *testing.T) {
got := SocketPath("alloc-123", "http")
want := "/run/orca/alloc-alloc-123/port-http.sock"
if got != want {
t.Errorf("SocketPath = %q, want %q", got, want)
}
}
func TestSocketEmitter_AllocIDPlaceholderNilSpec(t *testing.T) {
if got := allocIDForSocket(nil); got != "<allocID>" {
t.Errorf("allocIDForSocket(nil) = %q, want <allocID>", got)
}
}
func TestSocketEmitter_AllocIDPlaceholderEmptyName(t *testing.T) {
spec := &jobspec.WorkloadSpec{Name: " "}
if got := allocIDForSocket(spec); got != "<allocID>" {
t.Errorf("allocIDForSocket(empty name) = %q, want <allocID>", got)
}
}
func TestSocketEmitter_AllocIDPlaceholderNamedSpec(t *testing.T) {
spec := &jobspec.WorkloadSpec{Name: "web"}
if got := allocIDForSocket(spec); got != "web" {
t.Errorf("allocIDForSocket(web) = %q, want web", got)
}
}
func TestSocketEmitter_SocketPathPlaceholder(t *testing.T) {
got := SocketPath("<allocID>", "grpc")
want := "/run/orca/alloc-<allocID>/port-grpc.sock"
if got != want {
t.Errorf("SocketPath = %q, want %q", got, want)
}
}
func TestSystemdEmitter_IntegratesSocketLines(t *testing.T) {
// End-to-end: the systemd unit for a Service with ports contains
// the RuntimeDirectory line emitted by the SocketEmitter.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/httpd"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
files, err := SystemdEmitter{}.Render(spec, &Node{})
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
if !strings.Contains(c, "RuntimeDirectory=orca/alloc-web\n") {
t.Errorf("unit missing RuntimeDirectory line\n%s", c)
}
if !strings.Contains(c, "# socket: /run/orca/alloc-web/port-http.sock\n") {
t.Errorf("unit missing socket path comment\n%s", c)
}
}
func TestSystemdEmitter_IntegratesSocketLinesTCPBind(t *testing.T) {
// When service.bind = 127.0.0.1, the unit contains the ExecStartPre
// TCP-bind marker.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/httpd"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Service: &jobspec.ServiceBlock{Bind: "127.0.0.1"},
}
files, err := SystemdEmitter{}.Render(spec, &Node{})
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
if !strings.Contains(c, "ExecStartPre=/bin/echo orca: bind 127.0.0.1 port http (tcp, R-007 opt-in)\n") {
t.Errorf("unit missing TCP bind ExecStartPre marker\n%s", c)
}
}
func TestSystemdEmitter_NoSocketLinesForPortlessSpec(t *testing.T) {
// A Job with no ports → no RuntimeDirectory line in the unit.
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "backup",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/rsync"},
}
files, err := SystemdEmitter{}.Render(spec, &Node{})
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
if strings.Contains(c, "RuntimeDirectory=") {
t.Errorf("portless spec should not emit RuntimeDirectory\n%s", c)
}
if strings.Contains(c, "# socket:") {
t.Errorf("portless spec should not emit socket comment\n%s", c)
}
}
// contains reports whether the slice contains the string s.
func contains(lines []string, s string) bool {
for _, l := range lines {
if l == s {
return true
}
}
return false
}
+239
View File
@@ -0,0 +1,239 @@
package emitter
import (
"errors"
"fmt"
"strings"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// SystemdEmitter is the Emitter implementation for the "process"
// runtime. It renders the systemd unit file for the workload,
// including the lifecycle hooks (P04) and the R-007 socket plumbing
// (P08).
//
// Lifecycle hooks map to systemd semantics (PRD §10.1):
//
// - lifecycle.pre_stop → ExecStop= (the command run on stop; systemd
// runs ExecStop, then kills the main process after the deadline).
// - lifecycle.post_start → ExecStartPost= (runs after the main
// process starts).
//
// systemd has no ExecStartPre equivalent for a "pre_start" hook; the
// spec does not define pre_start (only pre_stop and post_start per
// PRD §10.1), so no mapping is needed.
//
// The unit name carries the `orca-v1-` prefix per the dual-write
// window (REQ-090) so the v0.9 SSH-push path does not collide with the
// v0.8 daemon's `orca-<job>.service` units during the migration
// window.
//
// Later phases extend this emitter:
//
// - v0.10-P03: secrets via EnvironmentFile= + LoadCredential=
type SystemdEmitter struct{}
// unitNamePrefix is the v0.9 SSH-push unit-name prefix. The v0.8
// daemon uses `orca-<job>.service`; the v0.9 path uses
// `orca-v1-<spec.Name>.service` so the two never overlap (REQ-090,
// I-C-006). The prefix is load-bearing — do not change it without
// updating the dual-write window contract.
const unitNamePrefix = "orca-v1-"
// Render renders the systemd unit file for a process-runtime workload.
//
// When the spec has no Tasks (the single-process case, the historical
// shape), the unit name is
// /etc/systemd/system/<unitNamePrefix><spec.Name>.service and the
// content is a [Service] block with ExecStart, optional ExecStartPost
// (lifecycle.post_start), optional ExecStop (lifecycle.pre_stop), and
// the R-007 socket-plumbing lines (RuntimeDirectory=, optional
// TCP-bind ExecStartPre). Mode is 0644.
//
// When the spec has a task group (P06, spec.Tasks non-empty), the
// alloc is multi-process and Render emits one systemd unit per task
// (`orca-v1-alloc-<alloc-id>-<task-name>.service`) plus a single
// grouping target unit (`orca-v1-alloc-<alloc-id>.target`) that
// starts/stops all tasks together. Each per-task unit carries
// `PartOf=orca-v1-alloc-<alloc-id>.target` and is
// `WantedBy=multi-user.target` so the task starts at boot. Tasks
// that omit their own runtime inherit the top-level spec.Runtime as
// the per-group default.
//
// The rendered shape (single-process) is:
//
// [Service]
// ExecStart=<runtime command>
// ExecStartPost=<post_start command 1>
// ExecStartPost=<post_start command 2>
// ExecStop=<pre_stop command 1>
// ExecStop=<pre_stop command 2>
// RuntimeDirectory=orca/alloc-<alloc-id>
// # socket: /run/orca/alloc-<alloc-id>/port-<name>.sock
// ExecStartPre=/bin/echo orca: bind 127.0.0.1 port <name> (tcp, R-007 opt-in)
//
// Returns an error if the spec is nil, the spec is missing its name,
// the runtime block is nil, or the runtime command is empty (a
// workload with no command has nothing to ExecStart). For task groups,
// returns an error if any task has no resolvable runtime command.
func (SystemdEmitter) Render(spec *jobspec.WorkloadSpec, node *Node) ([]File, error) {
if spec == nil {
return nil, errors.New("emitter/systemd: spec is nil")
}
if strings.TrimSpace(spec.Name) == "" {
return nil, errors.New("emitter/systemd: spec name is empty")
}
if len(spec.Tasks) > 0 {
return renderTaskGroup(spec, node)
}
if spec.Runtime == nil {
return nil, errors.New("emitter/systemd: runtime block is nil")
}
if strings.TrimSpace(spec.Runtime.Command) == "" {
return nil, errors.New("emitter/systemd: runtime command is empty")
}
path := fmt.Sprintf("/etc/systemd/system/%s%s.service", unitNamePrefix, spec.Name)
content := renderSystemdUnit(spec)
return []File{{Path: path, Content: content, Mode: "0644"}}, nil
}
// renderTaskGroup renders one systemd unit per task plus the grouping
// target unit. Each task's runtime falls back to the top-level
// spec.Runtime when the task omits its own. Tasks with no resolvable
// command (no task.Command, no task.Runtime.Command, no top-level
// Runtime) return an error.
func renderTaskGroup(spec *jobspec.WorkloadSpec, node *Node) ([]File, error) {
allocID := spec.Name
targetUnit := fmt.Sprintf("%salloc-%s.target", unitNamePrefix, allocID)
targetPath := fmt.Sprintf("/etc/systemd/system/%s", targetUnit)
var files []File
for _, task := range spec.Tasks {
rt := taskRuntime(spec, &task)
if rt == nil {
return nil, fmt.Errorf("emitter/systemd: task %q has no runtime (set tasks[].runtime or top-level runtime)", task.Name)
}
cmd := taskCommand(spec, &task, rt)
if strings.TrimSpace(cmd) == "" {
return nil, fmt.Errorf("emitter/systemd: task %q command is empty", task.Name)
}
unitName := fmt.Sprintf("%salloc-%s-%s.service", unitNamePrefix, allocID, task.Name)
path := fmt.Sprintf("/etc/systemd/system/%s", unitName)
content := renderTaskUnit(spec, &task, rt, cmd, targetUnit)
files = append(files, File{Path: path, Content: content, Mode: "0644"})
}
files = append(files, File{
Path: targetPath,
Content: renderTargetUnit(targetUnit, spec, allocID),
Mode: "0644",
})
return files, nil
}
// taskRuntime returns the effective runtime for a task: the task's own
// runtime when set, otherwise the top-level spec.Runtime (the per-group
// default). Returns nil when neither is set.
func taskRuntime(spec *jobspec.WorkloadSpec, task *jobspec.TaskGroupTask) *jobspec.RuntimeBlock {
if task.Runtime != nil {
return task.Runtime
}
return spec.Runtime
}
// taskCommand returns the ExecStart command for a task. A task-level
// Command takes precedence; otherwise the task's runtime command is
// used; otherwise the top-level runtime command is used. Returns an
// empty string when none is set.
func taskCommand(spec *jobspec.WorkloadSpec, task *jobspec.TaskGroupTask, rt *jobspec.RuntimeBlock) string {
if strings.TrimSpace(task.Command) != "" {
return task.Command
}
if rt != nil && strings.TrimSpace(rt.Command) != "" {
return rt.Command
}
return ""
}
// renderTaskUnit renders a single per-task systemd [Unit]+[Service]
// block. The unit is `PartOf=` the alloc target and
// `WantedBy=multi-user.target` so it starts at boot and stops with the
// group. The [Service] block carries the task's ExecStart and the
// socket-plumbing lines derived from the spec's ports.
func renderTaskUnit(spec *jobspec.WorkloadSpec, task *jobspec.TaskGroupTask, rt *jobspec.RuntimeBlock, cmd, targetUnit string) string {
var b strings.Builder
b.WriteString("[Unit]\n")
b.WriteString(fmt.Sprintf("Description=orca alloc task %s\n", task.Name))
b.WriteString(fmt.Sprintf("PartOf=%s\n", targetUnit))
b.WriteString("\n[Service]\n")
b.WriteString(fmt.Sprintf("ExecStart=%s\n", cmd))
for _, line := range (SocketEmitter{}).RenderSocketLines(spec) {
b.WriteString(line)
b.WriteString("\n")
}
b.WriteString("\n[Install]\n")
b.WriteString("WantedBy=multi-user.target\n")
return b.String()
}
// renderTargetUnit renders the grouping target unit
// (`orca-v1-alloc-<alloc-id>.target`) that starts/stops all tasks
// together. The [Unit] block lists every per-task unit under Wants=
// so `systemctl start <target>` brings them all up, and
// `systemctl stop <target>` tears them down (PartOf= propagates stop).
func renderTargetUnit(targetUnit string, spec *jobspec.WorkloadSpec, allocID string) string {
var b strings.Builder
b.WriteString("[Unit]\n")
b.WriteString(fmt.Sprintf("Description=orca alloc %s task group\n", allocID))
for _, task := range spec.Tasks {
b.WriteString(fmt.Sprintf("Wants=%salloc-%s-%s.service\n", unitNamePrefix, allocID, task.Name))
}
b.WriteString("\n[Install]\n")
b.WriteString("WantedBy=multi-user.target\n")
return b.String()
}
// renderSystemdUnit renders the full [Service] block for the spec,
// including ExecStart, lifecycle hooks (ExecStartPost, ExecStop), and
// the R-007 socket-plumbing lines (RuntimeDirectory=, optional
// TCP-bind ExecStartPre). The output is a single string with a
// trailing newline per line.
func renderSystemdUnit(spec *jobspec.WorkloadSpec) string {
var b strings.Builder
b.WriteString("[Service]\n")
b.WriteString(fmt.Sprintf("ExecStart=%s\n", spec.Runtime.Command))
// Lifecycle: post_start → ExecStartPost (runs after start).
for _, cmd := range lifecyclePostStart(spec) {
b.WriteString(fmt.Sprintf("ExecStartPost=%s\n", cmd))
}
// Lifecycle: pre_stop → ExecStop (runs before the process is killed).
for _, cmd := range lifecyclePreStop(spec) {
b.WriteString(fmt.Sprintf("ExecStop=%s\n", cmd))
}
// R-007 socket plumbing: RuntimeDirectory= per port + optional
// TCP-bind ExecStartPre.
for _, line := range (SocketEmitter{}).RenderSocketLines(spec) {
b.WriteString(line)
b.WriteString("\n")
}
return b.String()
}
// lifecyclePostStart returns the post_start lifecycle commands for
// the spec, or nil when the spec has no lifecycle block or no
// post_start commands.
func lifecyclePostStart(spec *jobspec.WorkloadSpec) []string {
if spec.Lifecycle == nil {
return nil
}
return spec.Lifecycle.PostStart
}
// lifecyclePreStop returns the pre_stop lifecycle commands for the
// spec, or nil when the spec has no lifecycle block or no pre_stop
// commands.
func lifecyclePreStop(spec *jobspec.WorkloadSpec) []string {
if spec.Lifecycle == nil {
return nil
}
return spec.Lifecycle.PreStop
}
+172
View File
@@ -0,0 +1,172 @@
package emitter
import (
"strings"
"testing"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
func TestSystemdEmitter_LifecyclePostStart(t *testing.T) {
// lifecycle.post_start → ExecStartPost (one line per command).
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/usr/local/bin/httpd -f"},
Lifecycle: &jobspec.LifecycleBlock{
PostStart: []string{"/usr/bin/sleep 1", "/usr/bin/curl localhost/healthz"},
},
}
files, err := SystemdEmitter{}.Render(spec, &Node{Hostname: "n1"})
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
if !strings.Contains(c, "ExecStartPost=/usr/bin/sleep 1\n") {
t.Errorf("missing ExecStartPost for sleep 1\n%s", c)
}
if !strings.Contains(c, "ExecStartPost=/usr/bin/curl localhost/healthz\n") {
t.Errorf("missing ExecStartPost for curl\n%s", c)
}
// ExecStart must still be present.
if !strings.Contains(c, "ExecStart=/usr/local/bin/httpd -f\n") {
t.Errorf("missing ExecStart\n%s", c)
}
}
func TestSystemdEmitter_LifecyclePreStop(t *testing.T) {
// lifecycle.pre_stop → ExecStop (one line per command).
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/usr/local/bin/httpd -f"},
Lifecycle: &jobspec.LifecycleBlock{
PreStop: []string{"/usr/local/bin/httpd -graceful", "/usr/bin/sleep 5"},
},
}
files, err := SystemdEmitter{}.Render(spec, &Node{Hostname: "n1"})
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
if !strings.Contains(c, "ExecStop=/usr/local/bin/httpd -graceful\n") {
t.Errorf("missing ExecStop for graceful\n%s", c)
}
if !strings.Contains(c, "ExecStop=/usr/bin/sleep 5\n") {
t.Errorf("missing ExecStop for sleep 5\n%s", c)
}
}
func TestSystemdEmitter_LifecycleBoth(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/httpd"},
Lifecycle: &jobspec.LifecycleBlock{
PostStart: []string{"/bin/after-start"},
PreStop: []string{"/bin/before-stop"},
},
}
files, err := SystemdEmitter{}.Render(spec, &Node{})
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
// ExecStartPost must appear before ExecStop (post_start runs after
// start; pre_stop runs before stop — the order in the unit file
// reflects the lifecycle order).
startIdx := strings.Index(c, "ExecStart=")
postIdx := strings.Index(c, "ExecStartPost=")
stopIdx := strings.Index(c, "ExecStop=")
if startIdx < 0 || postIdx < 0 || stopIdx < 0 {
t.Fatalf("missing one of ExecStart/ExecStartPost/ExecStop\n%s", c)
}
if !(startIdx < postIdx && postIdx < stopIdx) {
t.Errorf("expected order ExecStart < ExecStartPost < ExecStop\n%s", c)
}
}
func TestSystemdEmitter_LifecycleNilOmitsDirectives(t *testing.T) {
// No lifecycle block → no ExecStartPost / ExecStop lines.
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "backup",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/rsync"},
}
files, err := SystemdEmitter{}.Render(spec, &Node{})
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
if strings.Contains(c, "ExecStartPost=") {
t.Errorf("ExecStartPost should be omitted when no lifecycle\n%s", c)
}
if strings.Contains(c, "ExecStop=") {
t.Errorf("ExecStop should be omitted when no lifecycle\n%s", c)
}
}
func TestSystemdEmitter_LifecycleEmptyListsOmitted(t *testing.T) {
// Lifecycle block present but empty lists → no ExecStartPost / ExecStop.
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "x",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/x"},
Lifecycle: &jobspec.LifecycleBlock{},
}
files, err := SystemdEmitter{}.Render(spec, &Node{})
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
if strings.Contains(c, "ExecStartPost=") {
t.Errorf("ExecStartPost should be omitted for empty PostStart\n%s", c)
}
if strings.Contains(c, "ExecStop=") {
t.Errorf("ExecStop should be omitted for empty PreStop\n%s", c)
}
}
func TestSystemdEmitter_LifecycleOnlyPostStart(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "x",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/x"},
Lifecycle: &jobspec.LifecycleBlock{
PostStart: []string{"/bin/notify-up"},
},
}
files, err := SystemdEmitter{}.Render(spec, &Node{})
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
if !strings.Contains(c, "ExecStartPost=/bin/notify-up\n") {
t.Errorf("missing ExecStartPost\n%s", c)
}
if strings.Contains(c, "ExecStop=") {
t.Errorf("ExecStop should be omitted when only PostStart set\n%s", c)
}
}
func TestSystemdEmitter_LifecycleOnlyPreStop(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "x",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/x"},
Lifecycle: &jobspec.LifecycleBlock{
PreStop: []string{"/bin/notify-down"},
},
}
files, err := SystemdEmitter{}.Render(spec, &Node{})
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
if !strings.Contains(c, "ExecStop=/bin/notify-down\n") {
t.Errorf("missing ExecStop\n%s", c)
}
if strings.Contains(c, "ExecStartPost=") {
t.Errorf("ExecStartPost should be omitted when only PreStop set\n%s", c)
}
}
+343
View File
@@ -0,0 +1,343 @@
package emitter
import (
"strings"
"testing"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
func TestSystemdEmitter_RenderJob(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "backup",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/usr/bin/rsync -a /src /dst"},
}
node := &Node{Hostname: "node-1", Runtime: []string{"process"}}
files, err := SystemdEmitter{}.Render(spec, node)
if err != nil {
t.Fatalf("Render: %v", err)
}
if len(files) != 1 {
t.Fatalf("got %d files, want 1", len(files))
}
f := files[0]
wantPath := "/etc/systemd/system/orca-v1-backup.service"
if f.Path != wantPath {
t.Errorf("Path = %q, want %q", f.Path, wantPath)
}
wantContent := "[Service]\nExecStart=/usr/bin/rsync -a /src /dst\n"
if f.Content != wantContent {
t.Errorf("Content = %q, want %q", f.Content, wantContent)
}
if f.Mode != "0644" {
t.Errorf("Mode = %q, want 0644", f.Mode)
}
}
func TestSystemdEmitter_RenderService(t *testing.T) {
// The full service emitter (Traefik route + health checks) lands in
// P02; here we only prove the systemd side renders for a Service
// kind with a process runtime.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/usr/local/bin/httpd -f"},
}
node := &Node{Hostname: "node-1", Runtime: []string{"process"}}
files, err := SystemdEmitter{}.Render(spec, node)
if err != nil {
t.Fatalf("Render: %v", err)
}
if len(files) != 1 {
t.Fatalf("got %d files, want 1", len(files))
}
if files[0].Path != "/etc/systemd/system/orca-v1-web.service" {
t.Errorf("Path = %q, want /etc/systemd/system/orca-v1-web.service", files[0].Path)
}
if !strings.Contains(files[0].Content, "ExecStart=/usr/local/bin/httpd -f") {
t.Errorf("Content = %q, want it to contain the ExecStart line", files[0].Content)
}
}
func TestSystemdEmitter_EmptyCommandError(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "x",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: ""},
}
_, err := SystemdEmitter{}.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for empty command, got nil")
}
if !strings.Contains(err.Error(), "command is empty") {
t.Errorf("error = %q, want 'command is empty'", err.Error())
}
}
func TestSystemdEmitter_WhitespaceCommandError(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "x",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: " "},
}
_, err := SystemdEmitter{}.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for whitespace-only command, got nil")
}
if !strings.Contains(err.Error(), "command is empty") {
t.Errorf("error = %q, want 'command is empty'", err.Error())
}
}
func TestSystemdEmitter_NilSpec(t *testing.T) {
v := SystemdEmitter{}
if _, err := v.Render(nil, &Node{}); err == nil {
t.Fatal("expected error for nil spec")
}
}
func TestSystemdEmitter_EmptyName(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: " ",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/x"},
}
_, err := SystemdEmitter{}.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for empty name")
}
if !strings.Contains(err.Error(), "name is empty") {
t.Errorf("error = %q, want 'name is empty'", err.Error())
}
}
func TestSystemdEmitter_NilRuntime(t *testing.T) {
spec := &jobspec.WorkloadSpec{Kind: "Job", Name: "x"}
_, err := SystemdEmitter{}.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for nil runtime")
}
if !strings.Contains(err.Error(), "runtime block is nil") {
t.Errorf("error = %q, want 'runtime block is nil'", err.Error())
}
}
func TestSystemdEmitter_UnitNamePrefix(t *testing.T) {
// The orca-v1- prefix is load-bearing for the dual-write window
// (REQ-090, I-C-006): the v0.8 daemon writes `orca-<job>.service`
// and the v0.9 SSH-push path writes `orca-v1-<spec.Name>.service`,
// so the two never collide. This test guards against accidental
// removal of the prefix.
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "dual-write-safety",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/true"},
}
files, err := SystemdEmitter{}.Render(spec, &Node{Hostname: "n"})
if err != nil {
t.Fatalf("Render: %v", err)
}
if !strings.HasPrefix(files[0].Path, "/etc/systemd/system/orca-v1-") {
t.Errorf("Path = %q, want it to start with /etc/systemd/system/orca-v1- (REQ-090)", files[0].Path)
}
if !strings.HasSuffix(files[0].Path, ".service") {
t.Errorf("Path = %q, want it to end with .service", files[0].Path)
}
// Explicitly assert the full expected unit name to lock the contract.
want := "/etc/systemd/system/orca-v1-dual-write-safety.service"
if files[0].Path != want {
t.Errorf("Path = %q, want %q", files[0].Path, want)
}
// Sanity: the prefix is exactly "orca-v1-", not "orca-v0" or "orca".
if unitNamePrefix != "orca-v1-" {
t.Errorf("unitNamePrefix = %q, want orca-v1-", unitNamePrefix)
}
}
func TestSystemdEmitter_TaskGroupTwoTasks(t *testing.T) {
// P06: a task group with two tasks renders one unit per task plus
// a grouping target unit. Each per-task unit is
// `orca-v1-alloc-<alloc-id>-<task-name>.service`, carries
// `PartOf=orca-v1-alloc-<alloc-id>.target`, and is
// `WantedBy=multi-user.target`. The target unit lists every
// per-task unit under Wants=.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Tasks: []jobspec.TaskGroupTask{
{
Name: "app",
Command: "/usr/bin/httpd -f",
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
},
{
Name: "sidecar",
Command: "/bin/wasm-runner sidecar.wasm",
Runtime: &jobspec.RuntimeBlock{OneOf: "wasm"},
},
},
}
files, err := SystemdEmitter{}.Render(spec, &Node{Hostname: "n1"})
if err != nil {
t.Fatalf("Render: %v", err)
}
// 2 per-task units + 1 target unit.
if len(files) != 3 {
t.Fatalf("got %d files, want 3 (2 per-task units + 1 target)", len(files))
}
wantApp := "/etc/systemd/system/orca-v1-alloc-web-app.service"
wantSide := "/etc/systemd/system/orca-v1-alloc-web-sidecar.service"
wantTarget := "/etc/systemd/system/orca-v1-alloc-web.target"
paths := make(map[string]*File, len(files))
for i := range files {
paths[files[i].Path] = &files[i]
}
if _, ok := paths[wantApp]; !ok {
t.Errorf("missing per-task unit %q; got paths %v", wantApp, filePaths(files))
}
if _, ok := paths[wantSide]; !ok {
t.Errorf("missing per-task unit %q; got paths %v", wantSide, filePaths(files))
}
if _, ok := paths[wantTarget]; !ok {
t.Errorf("missing target unit %q; got paths %v", wantTarget, filePaths(files))
}
if _, ok := paths[wantTarget]; !ok {
t.Errorf("missing target unit %q; got paths %v", wantTarget, filePaths(files))
}
// Verify PartOf relations and ExecStart on per-task units.
app := paths[wantApp]
if !strings.Contains(app.Content, "PartOf=orca-v1-alloc-web.target") {
t.Errorf("app unit missing PartOf=orca-v1-alloc-web.target\n%s", app.Content)
}
if !strings.Contains(app.Content, "ExecStart=/usr/bin/httpd -f") {
t.Errorf("app unit missing ExecStart=/usr/bin/httpd -f\n%s", app.Content)
}
if !strings.Contains(app.Content, "WantedBy=multi-user.target") {
t.Errorf("app unit missing WantedBy=multi-user.target\n%s", app.Content)
}
side := paths[wantSide]
if !strings.Contains(side.Content, "PartOf=orca-v1-alloc-web.target") {
t.Errorf("sidecar unit missing PartOf=orca-v1-alloc-web.target\n%s", side.Content)
}
if !strings.Contains(side.Content, "ExecStart=/bin/wasm-runner sidecar.wasm") {
t.Errorf("sidecar unit missing ExecStart\n%s", side.Content)
}
// Verify the target unit Wants= both per-task units.
target := paths[wantTarget]
if !strings.Contains(target.Content, "Wants=orca-v1-alloc-web-app.service") {
t.Errorf("target missing Wants=...app.service\n%s", target.Content)
}
if !strings.Contains(target.Content, "Wants=orca-v1-alloc-web-sidecar.service") {
t.Errorf("target missing Wants=...sidecar.service\n%s", target.Content)
}
}
func TestSystemdEmitter_TaskGroupInheritsTopLevelRuntime(t *testing.T) {
// P06: a task that omits its own runtime inherits the top-level
// spec.Runtime as the per-group default. The per-task unit's
// ExecStart must come from the top-level runtime command when
// the task has no own command and no own runtime.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/default"},
Tasks: []jobspec.TaskGroupTask{
{Name: "app"},
{Name: "sidecar", Command: "/bin/override"},
},
}
files, err := SystemdEmitter{}.Render(spec, &Node{})
if err != nil {
t.Fatalf("Render: %v", err)
}
if len(files) != 3 {
t.Fatalf("got %d files, want 3", len(files))
}
appContent := findUnitContent(files, "/etc/systemd/system/orca-v1-alloc-web-app.service")
if appContent == "" {
t.Fatalf("missing app unit; paths %v", filePaths(files))
}
if !strings.Contains(appContent, "ExecStart=/bin/default") {
t.Errorf("app unit should inherit top-level command /bin/default\n%s", appContent)
}
sideContent := findUnitContent(files, "/etc/systemd/system/orca-v1-alloc-web-sidecar.service")
if sideContent == "" {
t.Fatalf("missing sidecar unit; paths %v", filePaths(files))
}
if !strings.Contains(sideContent, "ExecStart=/bin/override") {
t.Errorf("sidecar unit should use its own command /bin/override\n%s", sideContent)
}
}
func TestSystemdEmitter_TaskGroupNoCommandError(t *testing.T) {
// P06: a task with no resolvable command (no task.Command, no
// task.Runtime, no top-level Runtime) is an error.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Tasks: []jobspec.TaskGroupTask{{Name: "app"}},
}
_, err := SystemdEmitter{}.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for task with no runtime, got nil")
}
if !strings.Contains(err.Error(), "no runtime") {
t.Errorf("error = %q, want 'no runtime'", err.Error())
}
}
func TestSystemdEmitter_TaskGroupEmptyCommandError(t *testing.T) {
// P06: a task whose resolved runtime command is empty/whitespace
// is an error (mirrors the single-process rule).
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: " "},
Tasks: []jobspec.TaskGroupTask{{Name: "app"}},
}
_, err := SystemdEmitter{}.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for empty command, got nil")
}
if !strings.Contains(err.Error(), "command is empty") {
t.Errorf("error = %q, want 'command is empty'", err.Error())
}
}
func TestSystemdEmitter_NoTasksBackwardCompat(t *testing.T) {
// Backward compat: a spec with no Tasks renders exactly one unit
// (the historical single-process shape).
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "backup",
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/rsync"},
}
files, err := SystemdEmitter{}.Render(spec, &Node{})
if err != nil {
t.Fatalf("Render: %v", err)
}
if len(files) != 1 {
t.Fatalf("got %d files, want 1 (backward compat)", len(files))
}
if files[0].Path != "/etc/systemd/system/orca-v1-backup.service" {
t.Errorf("Path = %q, want /etc/systemd/system/orca-v1-backup.service", files[0].Path)
}
}
func filePaths(files []File) []string {
out := make([]string, len(files))
for i, f := range files {
out[i] = f.Path
}
return out
}
func findUnitContent(files []File, path string) string {
for _, f := range files {
if f.Path == path {
return f.Content
}
}
return ""
}
+225
View File
@@ -0,0 +1,225 @@
package emitter
import (
"errors"
"fmt"
"net"
"strings"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// TraefikEmitter is the Layer-4 emitter for the Traefik dynamic-config
// file (REQ-077). It renders /etc/traefik/dynamic/orca-<spec.Name>.yaml
// — a single Traefik dynamic-config file describing the routers,
// services (servers = the R-007 socket paths), TLS config pointing at
// the step-ca root CA, and the service health check.
//
// Registered on the emitter.Registry under the service-kind keys:
//
// - service:process
// - service:podman
// - service:wasm
//
// RegisterTraefik wires all three; callers can also call Register
// directly with TraefikEmitter{} for a single runtime.
//
// Atomic reload (gate C-10): the Traefik dynamic-config file is written
// atomically via the SSH-push transport (sshpush.WriteFileIdempotent
// performs temp-file + fsync + rename, and WriteTraefikDynamic wraps
// it with an explicit tmp+mv so fsnotify sees a single rename event).
// Traefik watches the dynamic dir with fsnotify; the rename triggers a
// reload. On a malformed config Traefik logs an error and holds the
// last-good config (documented Traefik behavior; the C-10 test
// verifies the tmp+rename sequence so a half-written file is never
// observed by Traefik). Drain is rendered by setting the backend
// server's weight to 0 (or removing it) — see RenderDrain.
//
// The orca-v1- prefix is NOT applied to Traefik dynamic-config paths
// (the prefix is only for systemd unit names; the Traefik file is named
// orca-<spec.Name>.yaml and is the single source of truth for the
// service route — there is no dual-write window for Traefik configs).
type TraefikEmitter struct{}
// traefikDynamicDir is the canonical Traefik dynamic-config directory
// (R-006). The emitter writes one file per service at
// /etc/traefik/dynamic/orca-<spec.Name>.yaml.
const traefikDynamicDir = "/etc/traefik/dynamic"
// traefikRouterTLSCertResolver is the Traefik cert-resolver name that
// the orca step-ca integration configures on the Traefik static config
// (P10 / v0.10 wires the step-ca root into this resolver). The
// dynamic-config file references it by name.
const traefikRouterTLSCertResolver = "orca"
// defaultTrustDomain is the SPIFFE trust domain used in the rendered
// TLS stanza when the spec does not carry an explicit trust domain.
// The step-ca provisioner (P10) overrides this at render time via the
// node argument; for P02 the emitter renders the placeholder.
const defaultTrustDomain = "cluster.orca.local"
// Render renders the Traefik dynamic-config YAML for a Service
// workload. The output is a single File whose Path is
// /etc/traefik/dynamic/orca-<spec.Name>.yaml, Content is the rendered
// YAML, and Mode is 0644.
//
// Returns an error if the spec is nil, the name is empty, the spec has
// no ports (a Service with no ports has no backends to route to), or a
// service.bind value (when present) is not a valid IP address (R-007).
func (TraefikEmitter) Render(spec *jobspec.WorkloadSpec, node *Node) ([]File, error) {
if spec == nil {
return nil, errors.New("emitter/traefik: spec is nil")
}
if strings.TrimSpace(spec.Name) == "" {
return nil, errors.New("emitter/traefik: spec name is empty")
}
if len(spec.Ports) == 0 {
return nil, errors.New("emitter/traefik: service has no ports (no backends to route to)")
}
if spec.Service != nil {
if b := strings.TrimSpace(spec.Service.Bind); b != "" && net.ParseIP(b) == nil {
return nil, fmt.Errorf("emitter/traefik: service.bind %q is not a valid IP (R-007)", b)
}
}
content, err := renderTraefikYAML(spec, node)
if err != nil {
return nil, err
}
path := fmt.Sprintf("%s/orca-%s.yaml", traefikDynamicDir, spec.Name)
return []File{{Path: path, Content: content, Mode: "0644"}}, nil
}
// RenderDrain renders a Traefik dynamic-config that drains the service
// by setting every backend server's weight to 0 (I-B-005 drain). The
// path matches the live config so the atomic rename overwrites the
// routing config with the drained config (Traefik reloads and stops
// sending traffic). The caller writes the result via
// WriteTraefikDynamic for the C-10 atomicity protocol.
func (e TraefikEmitter) RenderDrain(spec *jobspec.WorkloadSpec, node *Node) ([]File, error) {
if spec == nil {
return nil, errors.New("emitter/traefik: spec is nil")
}
if strings.TrimSpace(spec.Name) == "" {
return nil, errors.New("emitter/traefik: spec name is empty")
}
if len(spec.Ports) == 0 {
return nil, errors.New("emitter/traefik: service has no ports (no backends to drain)")
}
content, err := renderTraefikYAMLDrain(spec, node)
if err != nil {
return nil, err
}
path := fmt.Sprintf("%s/orca-%s.yaml", traefikDynamicDir, spec.Name)
return []File{{Path: path, Content: content, Mode: "0644"}}, nil
}
// RegisterTraefik registers the TraefikEmitter on the given Registry
// under the three service-kind runtime keys (service:process,
// service:podman, service:wasm). The emitter is the same instance for
// all three runtimes — the rendered Traefik config is runtime-agnostic
// (the backend server URL is the R-007 socket path, which the runtime
// layer binds regardless of process/wasm/podman).
func RegisterTraefik(reg *Registry) {
e := TraefikEmitter{}
reg.Register("service:process", e)
reg.Register("service:podman", e)
reg.Register("service:wasm", e)
}
// renderTraefikYAML renders the Traefik dynamic-config YAML for the
// given spec + node. The shape (verified by the Traefik docs) is:
//
// http:
// routers:
// orca-<name>:
// rule: PathPrefix("/<name>")
// service: orca-<name>
// tls:
// certResolver: orca
// domains:
// - main: "<trust-domain>"
// services:
// orca-<name>:
// loadBalancer:
// servers:
// - url: "unix:///run/orca/alloc-<allocID>/port-<portName>.sock"
// healthCheck:
// path: /healthz
// interval: <interval>
// timeout: <timeout>
//
// The alloc-id placeholder is "<allocID>" pending the P08 socket
// layer; Traefik will reject the URL until a real alloc-id is
// substituted. For P02 the emitter renders the placeholder so the
// C-10 atomicity protocol is testable end-to-end; the socket layer
// (P08) replaces the placeholder with the live alloc-id.
func renderTraefikYAML(spec *jobspec.WorkloadSpec, node *Node) (string, error) {
return renderTraefikYAMLWeighted(spec, node, false)
}
// renderTraefikYAMLDrain renders the drained Traefik dynamic-config
// (every backend server has weight: 0). The shape mirrors the live
// config so the rename overwrites the live route with the drain.
func renderTraefikYAMLDrain(spec *jobspec.WorkloadSpec, node *Node) (string, error) {
return renderTraefikYAMLWeighted(spec, node, true)
}
// renderTraefikYAMLWeighted renders the Traefik dynamic-config YAML.
// When drain is true, every server entry is emitted with `weight: 0`
// (I-B-005). When drain is false, no weight is emitted (Traefik
// defaults to 1 — equal weighting across servers).
func renderTraefikYAMLWeighted(spec *jobspec.WorkloadSpec, node *Node, drain bool) (string, error) {
var b strings.Builder
routerName := "orca-" + spec.Name
serviceName := "orca-" + spec.Name
rule := fmt.Sprintf("PathPrefix(\"/%s\")", spec.Name)
trustDomain := defaultTrustDomain
b.WriteString("http:\n")
b.WriteString(" routers:\n")
b.WriteString(fmt.Sprintf(" %s:\n", routerName))
b.WriteString(fmt.Sprintf(" rule: %s\n", rule))
b.WriteString(fmt.Sprintf(" service: %s\n", serviceName))
b.WriteString(" tls:\n")
b.WriteString(fmt.Sprintf(" certResolver: %s\n", traefikRouterTLSCertResolver))
b.WriteString(" domains:\n")
b.WriteString(fmt.Sprintf(" - main: %q\n", trustDomain))
b.WriteString(" services:\n")
b.WriteString(fmt.Sprintf(" %s:\n", serviceName))
b.WriteString(" loadBalancer:\n")
b.WriteString(" servers:\n")
allocID := allocIDFor(node)
for _, p := range spec.Ports {
sock := fmt.Sprintf("unix:///run/orca/alloc-%s/port-%s.sock", allocID, p.Name)
b.WriteString(" - url: ")
b.WriteString(fmt.Sprintf("%q\n", sock))
if drain {
b.WriteString(" weight: 0\n")
}
}
if spec.Health != nil {
b.WriteString(" healthCheck:\n")
path := "/healthz"
b.WriteString(fmt.Sprintf(" path: %s\n", path))
if spec.Health.Interval != "" {
b.WriteString(fmt.Sprintf(" interval: %s\n", spec.Health.Interval))
}
if spec.Health.Timeout != "" {
b.WriteString(fmt.Sprintf(" timeout: %s\n", spec.Health.Timeout))
}
}
return b.String(), nil
}
// allocIDFor returns the alloc-id placeholder for the node. P08 will
// substitute the live alloc-id from the socket layer; for P02 we use a
// deterministic placeholder derived from the node hostname so the
// rendered config is stable across re-renders (the C-10 idempotency
// check depends on a stable hash). When the node is nil or has no
// hostname, the literal placeholder "<allocID>" is emitted.
func allocIDFor(node *Node) string {
if node == nil || strings.TrimSpace(node.Hostname) == "" {
return "<allocID>"
}
return node.Hostname
}
+93
View File
@@ -0,0 +1,93 @@
package emitter
import (
"context"
"fmt"
"os"
)
// AtomicWriter is the SSH-push transport surface that
// WriteTraefikDynamic uses to write the Traefik dynamic-config file
// atomically. It is the subset of *sshpush.Transport that the
// atomicity protocol depends on. Tests substitute a mock to assert
// the tmp+rename sequence (gate C-10) without a real SSH server.
//
// *sshpush.Transport satisfies this interface (the compile-time
// assertion lives in internal/sshpush to avoid an import cycle — the
// sshpush package imports emitter for fan-out, so this package cannot
// import sshpush).
type AtomicWriter interface {
// WriteFileIdempotent writes content to peer:path atomically with
// mode, returning written=true if the file was actually written
// (content hash differed). Used by WriteTraefikDynamic to write
// the .tmp sibling.
WriteFileIdempotent(ctx context.Context, peer string, path string, content []byte, mode os.FileMode) (bool, error)
// Exec runs a command on peer and returns its combined output.
// Used by WriteTraefikDynamic to perform the atomic `mv -f
// path.tmp path`.
Exec(ctx context.Context, peer string, cmd string) ([]byte, error)
}
// WriteTraefikDynamic writes a Traefik dynamic-config file atomically
// (gate C-10: tmpfile + fsync + rename). The protocol is:
//
// 1. Write content to <path>.tmp via WriteFileIdempotent. The
// underlying sshpush transport writes the tmp file in the same
// directory as the target with mode-appended naming, fsyncs, and
// renames — but we add an extra hop here so the *Traefik* file is
// only ever observed at its final path after a single atomic
// rename event that Traefik's fsnotify watcher sees.
// 2. `mv -f <path>.tmp <path>` on the peer (atomic rename on POSIX).
// Traefik's fsnotify watcher picks up the rename → reload.
//
// On a malformed config Traefik logs an error and holds the
// last-good config (documented Traefik behavior; the C-10 test
// verifies the tmp+rename sequence so a half-written file is never
// observed by Traefik — the only window where Traefik can read the
// file is after the rename, which is atomic on POSIX).
//
// The mode is 0644 (Traefik reads the dynamic dir as root; the lead
// applier chmods after the rename).
func WriteTraefikDynamic(ctx context.Context, t AtomicWriter, peer string, path string, content []byte) error {
if t == nil {
return fmt.Errorf("traefik: atomic writer is nil")
}
if path == "" {
return fmt.Errorf("traefik: path is empty")
}
tmpPath := path + ".tmp"
if _, err := t.WriteFileIdempotent(ctx, peer, tmpPath, content, 0o644); err != nil {
return fmt.Errorf("traefik: write tmp %s: %w", tmpPath, err)
}
// Atomic rename on POSIX. `mv -f` overwrites an existing target
// without prompting. The rename is atomic; Traefik's fsnotify
// watcher observes a single IN_MOVED_TO event.
renameCmd := fmt.Sprintf("mv -f %s %s", shellQuoteLocal(tmpPath), shellQuoteLocal(path))
if _, err := t.Exec(ctx, peer, renameCmd); err != nil {
return fmt.Errorf("traefik: rename %s -> %s: %w", tmpPath, path, err)
}
return nil
}
// shellQuoteLocal single-quotes a path for safe shell interpolation on
// the peer. It escapes embedded single-quotes via the standard '\”
// idiom (close the single-quoted string, escape the literal single
// quote, reopen the single-quoted string). This is a local
// re-implementation (the sshpush package has its own) so the emitter
// layer does not depend on the transport package's private helpers —
// the AtomicWriter interface keeps the boundary clean for testing.
func shellQuoteLocal(s string) string {
var b []byte
b = append(b, '\'')
for i := 0; i < len(s); i++ {
c := s[i]
if c == '\'' {
// close quote, escape the literal single-quote, reopen.
b = append(b, '\'', '\\', '\'', '\'')
continue
}
b = append(b, c)
}
b = append(b, '\'')
return string(b)
}
+190
View File
@@ -0,0 +1,190 @@
package emitter
import (
"context"
"errors"
"os"
"strings"
"testing"
)
// mockAtomicWriter is a test-only AtomicWriter that records calls so
// the C-10 atomicity protocol (tmp + rename) can be asserted.
type mockAtomicWriter struct {
written []writeCall
execed []execCall
writeErr error
writeWrote bool
execErr error
}
type writeCall struct {
peer string
path string
mode os.FileMode
bytes []byte
}
type execCall struct {
peer string
cmd string
}
func (m *mockAtomicWriter) WriteFileIdempotent(ctx context.Context, peer string, path string, content []byte, mode os.FileMode) (bool, error) {
m.written = append(m.written, writeCall{peer: peer, path: path, mode: mode, bytes: append([]byte(nil), content...)})
if m.writeErr != nil {
return false, m.writeErr
}
return m.writeWrote, nil
}
func (m *mockAtomicWriter) Exec(ctx context.Context, peer string, cmd string) ([]byte, error) {
m.execed = append(m.execed, execCall{peer: peer, cmd: cmd})
if m.execErr != nil {
return nil, m.execErr
}
return []byte("ok"), nil
}
func TestWriteTraefikDynamic_TmpThenRename(t *testing.T) {
// Gate C-10: the Traefik dynamic-config write must be a tmp +
// rename sequence so Traefik's fsnotify watcher never observes a
// half-written file.
mock := &mockAtomicWriter{writeWrote: true}
path := "/etc/traefik/dynamic/orca-web.yaml"
peer := "node-1:22"
content := []byte("http:\n routers: {}\n")
if err := WriteTraefikDynamic(context.Background(), mock, peer, path, content); err != nil {
t.Fatalf("WriteTraefikDynamic: %v", err)
}
if len(mock.written) != 1 {
t.Fatalf("WriteFileIdempotent calls = %d, want 1", len(mock.written))
}
w := mock.written[0]
if w.peer != peer {
t.Errorf("write peer = %q, want %q", w.peer, peer)
}
// The tmp path is the target path + ".tmp".
if w.path != path+".tmp" {
t.Errorf("write path = %q, want %q (.tmp suffix is the C-10 atomicity protocol)", w.path, path+".tmp")
}
if string(w.bytes) != string(content) {
t.Errorf("write content = %q, want %q", string(w.bytes), string(content))
}
if w.mode != 0o644 {
t.Errorf("write mode = %o, want 0644", w.mode)
}
if len(mock.execed) != 1 {
t.Fatalf("Exec calls = %d, want 1 (the rename)", len(mock.execed))
}
e := mock.execed[0]
if e.peer != peer {
t.Errorf("exec peer = %q, want %q", e.peer, peer)
}
// The rename command must `mv -f` the .tmp file to the final path.
if !strings.Contains(e.cmd, "mv -f") {
t.Errorf("exec cmd = %q, want it to contain 'mv -f' (atomic rename)", e.cmd)
}
if !strings.Contains(e.cmd, path+".tmp") {
t.Errorf("exec cmd = %q, want it to contain the .tmp path as source", e.cmd)
}
if !strings.Contains(e.cmd, path) {
t.Errorf("exec cmd = %q, want it to contain the final path as destination", e.cmd)
}
// Sanity: the source must come before the destination in the
// mv command.
srcIdx := strings.Index(e.cmd, path+".tmp")
dstIdx := strings.Index(e.cmd, "'"+path+"'")
if srcIdx < 0 || dstIdx < 0 || srcIdx > dstIdx {
t.Errorf("exec cmd %q: source .tmp must come before destination %s", e.cmd, path)
}
}
func TestWriteTraefikDynamic_WriteTmpError(t *testing.T) {
mock := &mockAtomicWriter{writeErr: errors.New("disk full")}
err := WriteTraefikDynamic(context.Background(), mock, "p", "/etc/traefik/dynamic/orca-x.yaml", []byte("x"))
if err == nil {
t.Fatal("expected error from WriteFileIdempotent, got nil")
}
if !strings.Contains(err.Error(), "write tmp") {
t.Errorf("error = %q, want 'write tmp'", err.Error())
}
if !strings.Contains(err.Error(), "disk full") {
t.Errorf("error = %q, want underlying 'disk full'", err.Error())
}
if len(mock.execed) != 0 {
t.Errorf("on tmp write failure, no rename should happen; execed = %v", mock.execed)
}
}
func TestWriteTraefikDynamic_RenameError(t *testing.T) {
mock := &mockAtomicWriter{writeWrote: true, execErr: errors.New("permission denied")}
err := WriteTraefikDynamic(context.Background(), mock, "p", "/etc/traefik/dynamic/orca-x.yaml", []byte("x"))
if err == nil {
t.Fatal("expected error from rename, got nil")
}
if !strings.Contains(err.Error(), "rename") {
t.Errorf("error = %q, want 'rename'", err.Error())
}
if !strings.Contains(err.Error(), "permission denied") {
t.Errorf("error = %q, want underlying 'permission denied'", err.Error())
}
}
func TestWriteTraefikDynamic_NilWriter(t *testing.T) {
err := WriteTraefikDynamic(context.Background(), nil, "p", "/x", []byte("x"))
if err == nil {
t.Fatal("expected error for nil writer")
}
if !strings.Contains(err.Error(), "nil") {
t.Errorf("error = %q, want 'nil'", err.Error())
}
}
func TestWriteTraefikDynamic_EmptyPath(t *testing.T) {
mock := &mockAtomicWriter{writeWrote: true}
err := WriteTraefikDynamic(context.Background(), mock, "p", "", []byte("x"))
if err == nil {
t.Fatal("expected error for empty path")
}
if !strings.Contains(err.Error(), "path is empty") {
t.Errorf("error = %q, want 'path is empty'", err.Error())
}
}
func TestWriteTraefikDynamic_SkipWhenContentMatches(t *testing.T) {
// When the .tmp file already matches (writeWrote=false), the
// protocol still proceeds with the rename — the idempotency
// check is per-file, not per-protocol. The rename still happens
// so the final path reflects the (unchanged) content.
mock := &mockAtomicWriter{writeWrote: false}
err := WriteTraefikDynamic(context.Background(), mock, "p", "/etc/traefik/dynamic/orca-x.yaml", []byte("x"))
if err != nil {
t.Fatalf("WriteTraefikDynamic: %v", err)
}
if len(mock.execed) != 1 {
t.Errorf("rename should still happen on idempotent skip; execed = %v", mock.execed)
}
}
func TestShellQuoteLocal(t *testing.T) {
cases := []struct {
in, want string
}{
{"/etc/traefik/dynamic/orca-web.yaml", "'/etc/traefik/dynamic/orca-web.yaml'"},
{"", "''"},
{"/path with space/x", "'/path with space/x'"},
{"a'b", "'a'\\''b'"},
}
for _, tc := range cases {
t.Run(tc.in, func(t *testing.T) {
got := shellQuoteLocal(tc.in)
if got != tc.want {
t.Errorf("shellQuoteLocal(%q) = %q, want %q", tc.in, got, tc.want)
}
})
}
}
+353
View File
@@ -0,0 +1,353 @@
package emitter
import (
"strings"
"testing"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
func TestTraefikEmitter_RenderBasic(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Health: &jobspec.HealthBlock{CheckType: "http", Interval: "5s", Timeout: "1s"},
}
node := &Node{Hostname: "node-1", Runtime: []string{"process"}}
files, err := TraefikEmitter{}.Render(spec, node)
if err != nil {
t.Fatalf("Render: %v", err)
}
if len(files) != 1 {
t.Fatalf("got %d files, want 1", len(files))
}
f := files[0]
wantPath := "/etc/traefik/dynamic/orca-web.yaml"
if f.Path != wantPath {
t.Errorf("Path = %q, want %q", f.Path, wantPath)
}
if f.Mode != "0644" {
t.Errorf("Mode = %q, want 0644", f.Mode)
}
c := f.Content
if !strings.Contains(c, "http:") {
t.Errorf("content missing 'http:'\n%s", c)
}
if !strings.Contains(c, "routers:") {
t.Errorf("content missing 'routers:'\n%s", c)
}
if !strings.Contains(c, "orca-web:") {
t.Errorf("content missing 'orca-web:' router/service key\n%s", c)
}
if !strings.Contains(c, `rule: PathPrefix("/web")`) {
t.Errorf("content missing PathPrefix rule\n%s", c)
}
if !strings.Contains(c, "services:") {
t.Errorf("content missing 'services:'\n%s", c)
}
if !strings.Contains(c, "loadBalancer:") {
t.Errorf("content missing 'loadBalancer:'\n%s", c)
}
if !strings.Contains(c, "unix:///run/orca/alloc-node-1/port-http.sock") {
t.Errorf("content missing socket server URL\n%s", c)
}
if !strings.Contains(c, "certResolver: orca") {
t.Errorf("content missing 'certResolver: orca'\n%s", c)
}
if !strings.Contains(c, "domains:") {
t.Errorf("content missing TLS domains\n%s", c)
}
if !strings.Contains(c, "healthCheck:") {
t.Errorf("content missing 'healthCheck:'\n%s", c)
}
if !strings.Contains(c, "interval: 5s") {
t.Errorf("content missing 'interval: 5s'\n%s", c)
}
if !strings.Contains(c, "timeout: 1s") {
t.Errorf("content missing 'timeout: 1s'\n%s", c)
}
}
func TestTraefikEmitter_RenderMultiplePorts(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "api",
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Ports: []jobspec.PortSpec{
{Name: "http", Port: 8080},
{Name: "grpc", Port: 9090},
},
}
node := &Node{Hostname: "n1"}
files, err := TraefikEmitter{}.Render(spec, node)
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
if !strings.Contains(c, "port-http.sock") {
t.Errorf("missing http socket: %s", c)
}
if !strings.Contains(c, "port-grpc.sock") {
t.Errorf("missing grpc socket: %s", c)
}
}
func TestTraefikEmitter_RenderDrain(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
node := &Node{Hostname: "n1"}
files, err := TraefikEmitter{}.RenderDrain(spec, node)
if err != nil {
t.Fatalf("RenderDrain: %v", err)
}
if len(files) != 1 {
t.Fatalf("got %d files, want 1", len(files))
}
c := files[0].Content
if !strings.Contains(c, "weight: 0") {
t.Errorf("drain config missing 'weight: 0'\n%s", c)
}
if !strings.Contains(c, "unix:///run/orca/alloc-n1/port-http.sock") {
t.Errorf("drain config missing socket URL\n%s", c)
}
}
func TestTraefikEmitter_RenderLiveHasNoWeightZero(t *testing.T) {
// Sanity: the live (non-drain) render must NOT emit `weight: 0`.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
node := &Node{Hostname: "n1"}
files, err := TraefikEmitter{}.Render(spec, node)
if err != nil {
t.Fatalf("Render: %v", err)
}
if strings.Contains(files[0].Content, "weight: 0") {
t.Errorf("live config should not contain 'weight: 0'\n%s", files[0].Content)
}
}
func TestTraefikEmitter_RenderNoHealthOmitsHealthCheck(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
node := &Node{Hostname: "n1"}
files, err := TraefikEmitter{}.Render(spec, node)
if err != nil {
t.Fatalf("Render: %v", err)
}
if strings.Contains(files[0].Content, "healthCheck:") {
t.Errorf("config without Health should omit 'healthCheck:'\n%s", files[0].Content)
}
}
func TestTraefikEmitter_NilSpec(t *testing.T) {
_, err := TraefikEmitter{}.Render(nil, &Node{})
if err == nil {
t.Fatal("expected error for nil spec")
}
}
func TestTraefikEmitter_EmptyName(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: " ",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
_, err := TraefikEmitter{}.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for empty name")
}
}
func TestTraefikEmitter_NoPorts(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
}
_, err := TraefikEmitter{}.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for missing ports")
}
if !strings.Contains(err.Error(), "no ports") {
t.Errorf("error = %q, want 'no ports'", err.Error())
}
}
func TestTraefikEmitter_NoPortsDrain(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
}
_, err := TraefikEmitter{}.RenderDrain(spec, &Node{})
if err == nil {
t.Fatal("expected error for missing ports on drain")
}
}
func TestTraefikEmitter_InvalidBind(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Service: &jobspec.ServiceBlock{Bind: "not-an-ip"},
}
_, err := TraefikEmitter{}.Render(spec, &Node{})
if err == nil {
t.Fatal("expected error for invalid service.bind")
}
if !strings.Contains(err.Error(), "valid IP") {
t.Errorf("error = %q, want 'valid IP'", err.Error())
}
}
func TestTraefikEmitter_ValidBindLoopback(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Service: &jobspec.ServiceBlock{Bind: "127.0.0.1"},
}
_, err := TraefikEmitter{}.Render(spec, &Node{})
if err != nil {
t.Fatalf("127.0.0.1 should be accepted, got %v", err)
}
}
func TestTraefikEmitter_NilNodeAllocPlaceholder(t *testing.T) {
// With a nil node, the alloc-id placeholder is the literal
// "<allocID>" sentinel so the rendered config is still valid YAML
// (the P08 socket layer substitutes the real alloc-id).
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
files, err := TraefikEmitter{}.Render(spec, nil)
if err != nil {
t.Fatalf("Render: %v", err)
}
if !strings.Contains(files[0].Content, "alloc-<allocID>") {
t.Errorf("nil node should render alloc-<allocID> placeholder\n%s", files[0].Content)
}
}
func TestTraefikEmitter_EmptyHostnameAllocPlaceholder(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
files, err := TraefikEmitter{}.Render(spec, &Node{Hostname: " "})
if err != nil {
t.Fatalf("Render: %v", err)
}
if !strings.Contains(files[0].Content, "alloc-<allocID>") {
t.Errorf("empty hostname should render alloc-<allocID> placeholder\n%s", files[0].Content)
}
}
func TestTraefikEmitter_PathNotOrcaV1Prefixed(t *testing.T) {
// REQ-090: the orca-v1- prefix is only for systemd units; Traefik
// dynamic-config paths are named orca-<spec.Name>.yaml (single
// source of truth — no dual-write window for Traefik configs).
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
files, err := TraefikEmitter{}.Render(spec, &Node{Hostname: "n1"})
if err != nil {
t.Fatalf("Render: %v", err)
}
if strings.Contains(files[0].Path, "orca-v1-") {
t.Errorf("Path %q should NOT contain the orca-v1- prefix (systemd-only)", files[0].Path)
}
if !strings.HasPrefix(files[0].Path, "/etc/traefik/dynamic/orca-") {
t.Errorf("Path %q should start with /etc/traefik/dynamic/orca-", files[0].Path)
}
if !strings.HasSuffix(files[0].Path, ".yaml") {
t.Errorf("Path %q should end with .yaml", files[0].Path)
}
}
func TestTraefikEmitter_RenderYAMLHasRoutersServicesTLS(t *testing.T) {
// Aggregate structural assertion: the rendered YAML has the four
// top-level Traefik concepts (routers, services, tls, servers).
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Health: &jobspec.HealthBlock{CheckType: "http"},
}
files, err := TraefikEmitter{}.Render(spec, &Node{Hostname: "n1"})
if err != nil {
t.Fatalf("Render: %v", err)
}
c := files[0].Content
for _, want := range []string{"routers:", "services:", "tls:", "servers:", "url:"} {
if !strings.Contains(c, want) {
t.Errorf("rendered YAML missing %q\n%s", want, c)
}
}
}
func TestRegisterTraefik_AllServiceRuntimes(t *testing.T) {
r := NewRegistry()
RegisterTraefik(r)
spec := func(runtime string) *jobspec.WorkloadSpec {
return &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: runtime},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
}
for _, runtime := range []string{"process", "podman", "wasm"} {
t.Run(runtime, func(t *testing.T) {
files, err := r.Render(spec(runtime), &Node{Hostname: "n1"})
if err != nil {
t.Fatalf("Render(service:%s): %v", runtime, err)
}
if len(files) != 1 {
t.Fatalf("got %d files, want 1", len(files))
}
if !strings.Contains(files[0].Path, "/etc/traefik/dynamic/orca-web.yaml") {
t.Errorf("Path = %q", files[0].Path)
}
})
}
}
func TestRegisterTraefik_OverwritesExisting(t *testing.T) {
// RegisterTraefik should overwrite any prior registration (the
// Registry documents last-wins).
r := NewRegistry()
r.Register("service:process", mockEmitter{files: []File{{Path: "/old"}}})
RegisterTraefik(r)
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
files, err := r.Render(spec, &Node{Hostname: "n1"})
if err != nil {
t.Fatalf("Render: %v", err)
}
if files[0].Path == "/old" {
t.Errorf("RegisterTraefik did not overwrite the prior registration")
}
}
// Compile-time assertion that TraefikEmitter implements Emitter.
var _ Emitter = TraefikEmitter{}
+237
View File
@@ -0,0 +1,237 @@
package emitter
import (
"fmt"
"strconv"
"strings"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// UpdatePlan is the computed update sequence for a Service (P03). It is
// a PLAN, not an execution — the transactional execution lands in
// v0.10-P10. Each step describes a discrete action the executor takes:
// start a set of allocs (Action="start"), wait for them to become
// healthy (WaitForHealthy=true), or cutover from old to new
// (Action="cutover" for blue-green). The Allocs field carries
// placeholder alloc names of the form "<spec.Name>-<index>" where
// index is 1-based (the scheduler assigns the real alloc-id at submit
// time; P03 uses spec.Name as a placeholder per the socket layer
// contract — see SocketEmitter).
type UpdatePlan struct {
Steps []UpdateStep
}
// UpdateStep is a single step in an UpdatePlan. Action is one of
// "start", "wait", "cutover", "promote". Allocs is the list of
// placeholder alloc names the step applies to. WaitForHealthy is true
// when the executor must wait for the allocs in this step to pass
// their health check before proceeding to the next step (driven by
// min_healthy_time / healthy_deadline on the spec, which the executor
// — not the plan — enforces).
type UpdateStep struct {
Action string
Allocs []string
WaitForHealthy bool
}
// maxParallelFor returns the effective max_parallel for the spec,
// defaulting to 1 when unset (0) and clamping to count (the validator
// already rejects out-of-range values; this is a defensive clamp for
// direct callers that bypass the validator).
func maxParallelFor(spec *jobspec.WorkloadSpec) int {
if spec.Update == nil {
return 1
}
if spec.Update.MaxParallel < 1 {
return 1
}
if spec.Count > 0 && spec.Update.MaxParallel > spec.Count {
return spec.Count
}
return spec.Update.MaxParallel
}
// allocName returns the placeholder alloc name for index i (1-based).
// The real alloc-id is assigned by the scheduler at submit time; P03
// uses spec.Name as the placeholder per the socket-layer contract.
func allocName(spec *jobspec.WorkloadSpec, i int) string {
return fmt.Sprintf("%s-%d", spec.Name, i)
}
// allAllocs returns the placeholder alloc names for the full count of
// the spec (1..count).
func allAllocs(spec *jobspec.WorkloadSpec) []string {
out := make([]string, 0, spec.Count)
for i := 1; i <= spec.Count; i++ {
out = append(out, allocName(spec, i))
}
return out
}
// canaryCount returns the integer canary count for the spec. The
// canary field accepts an integer count or a percentage ("<n>%"). For
// a percentage, the count is ceil(count * n / 100) with a minimum of 1
// when n > 0 (a 10% canary of a 3-replica service is 1 alloc, not 0).
// When the canary field is empty, the default is 1 (a single canary
// alloc — the smallest meaningful canary).
func canaryCount(spec *jobspec.WorkloadSpec) int {
if spec.Update == nil {
return 1
}
c := strings.TrimSpace(spec.Update.Canary)
if c == "" {
return 1
}
if strings.HasSuffix(c, "%") {
n, err := strconv.Atoi(strings.TrimSpace(strings.TrimSuffix(c, "%")))
if err != nil || n <= 0 {
return 1
}
allocs := spec.Count * n / 100
if allocs < 1 {
allocs = 1
}
return allocs
}
n, err := strconv.Atoi(c)
if err != nil || n < 1 {
return 1
}
if spec.Count > 0 && n > spec.Count {
return spec.Count
}
return n
}
// RenderUpdatePlan computes the rolling/canary/blue-green update
// sequence for a Service spec. Returns an *UpdatePlan describing the
// steps; the actual transactional execution lands in v0.10-P10.
//
// The three strategies:
//
// - rolling: allocs are started in batches of max_parallel. Each
// batch waits for healthy before the next batch starts. This is
// the simplest strategy and the default for stateless services.
//
// - canary: a single canary alloc (or N per the canary field) is
// started first and waits for healthy. After the canary is
// healthy, the plan emits a "promote" step (manual or auto per
// auto_promote); the remaining allocs are then started in
// max_parallel batches.
//
// - blue-green: all new allocs are started in parallel (a single
// "start" step with the full count). After they are healthy, a
// "cutover" step swaps traffic from the old allocs to the new
// ones. The old allocs are then stopped (the stop is implicit in
// the cutover step for the plan; v0.10-P10 makes it explicit).
//
// Returns an error if the spec is nil, the update block is nil, or
// the strategy is unknown (the validator should have caught these,
// but RenderUpdatePlan is defensive — emitters are called from
// render paths that may bypass the schema validator).
func RenderUpdatePlan(spec *jobspec.WorkloadSpec) (*UpdatePlan, error) {
if spec == nil {
return nil, fmt.Errorf("emitter/update: spec is nil")
}
if spec.Update == nil {
return nil, fmt.Errorf("emitter/update: update block is nil")
}
if spec.Count < 1 {
return nil, fmt.Errorf("emitter/update: count must be ≥ 1, got %d", spec.Count)
}
switch spec.Update.Strategy {
case "rolling":
return renderRollingPlan(spec), nil
case "canary":
return renderCanaryPlan(spec), nil
case "blue-green":
return renderBlueGreenPlan(spec), nil
default:
return nil, fmt.Errorf("emitter/update: unknown strategy %q (want rolling, canary, or blue-green)", spec.Update.Strategy)
}
}
// renderRollingPlan emits the rolling-update plan: allocs in batches
// of max_parallel, each batch waiting for healthy before the next.
func renderRollingPlan(spec *jobspec.WorkloadSpec) *UpdatePlan {
plan := &UpdatePlan{}
batch := maxParallelFor(spec)
allocs := allAllocs(spec)
for i := 0; i < len(allocs); i += batch {
end := i + batch
if end > len(allocs) {
end = len(allocs)
}
plan.Steps = append(plan.Steps, UpdateStep{
Action: "start",
Allocs: allocs[i:end],
WaitForHealthy: true,
})
}
return plan
}
// renderCanaryPlan emits the canary-update plan: a canary batch first
// (size per the canary field, default 1), a "promote" step, then the
// remaining allocs in max_parallel batches.
func renderCanaryPlan(spec *jobspec.WorkloadSpec) *UpdatePlan {
plan := &UpdatePlan{}
allocs := allAllocs(spec)
canary := canaryCount(spec)
if canary > len(allocs) {
canary = len(allocs)
}
if canary < 1 {
canary = 1
}
// Step 1: start the canary alloc(s) and wait for healthy.
plan.Steps = append(plan.Steps, UpdateStep{
Action: "start",
Allocs: allocs[:canary],
WaitForHealthy: true,
})
// Step 2: promote (manual or auto per auto_promote).
plan.Steps = append(plan.Steps, UpdateStep{
Action: "promote",
Allocs: allocs[:canary],
})
// Step 3+: remaining allocs in max_parallel batches.
batch := maxParallelFor(spec)
remaining := allocs[canary:]
for i := 0; i < len(remaining); i += batch {
end := i + batch
if end > len(remaining) {
end = len(remaining)
}
plan.Steps = append(plan.Steps, UpdateStep{
Action: "start",
Allocs: remaining[i:end],
WaitForHealthy: true,
})
}
return plan
}
// renderBlueGreenPlan emits the blue-green update plan: all new allocs
// start in parallel, wait for healthy, then cutover (swap traffic).
func renderBlueGreenPlan(spec *jobspec.WorkloadSpec) *UpdatePlan {
plan := &UpdatePlan{}
allocs := allAllocs(spec)
// Step 1: start ALL new allocs in parallel (blue-green does not
// batch — the new fleet stands up alongside the old).
plan.Steps = append(plan.Steps, UpdateStep{
Action: "start",
Allocs: allocs,
WaitForHealthy: true,
})
// Step 2: cutover — swap traffic from old to new. The old allocs
// are stopped implicitly as part of the cutover (v0.10-P10 makes
// the stop explicit in the transactional plane).
plan.Steps = append(plan.Steps, UpdateStep{
Action: "cutover",
Allocs: allocs,
WaitForHealthy: false,
})
return plan
}
+396
View File
@@ -0,0 +1,396 @@
package emitter
import (
"strings"
"testing"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
func TestRenderUpdatePlan_RollingBatches(t *testing.T) {
// count=4, max_parallel=2 → 2 batches of 2.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
MaxParallel: 2,
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan: %v", err)
}
if len(plan.Steps) != 2 {
t.Fatalf("got %d steps, want 2", len(plan.Steps))
}
for i, s := range plan.Steps {
if s.Action != "start" {
t.Errorf("step %d action = %q, want start", i, s.Action)
}
if !s.WaitForHealthy {
t.Errorf("step %d WaitForHealthy = false, want true", i)
}
if len(s.Allocs) != 2 {
t.Errorf("step %d allocs = %d, want 2", i, len(s.Allocs))
}
}
if plan.Steps[0].Allocs[0] != "web-1" || plan.Steps[0].Allocs[1] != "web-2" {
t.Errorf("step 0 allocs = %v, want [web-1 web-2]", plan.Steps[0].Allocs)
}
if plan.Steps[1].Allocs[0] != "web-3" || plan.Steps[1].Allocs[1] != "web-4" {
t.Errorf("step 1 allocs = %v, want [web-3 web-4]", plan.Steps[1].Allocs)
}
}
func TestRenderUpdatePlan_RollingUnevenBatches(t *testing.T) {
// count=5, max_parallel=2 → 3 batches: 2, 2, 1.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 5,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
MaxParallel: 2,
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan: %v", err)
}
if len(plan.Steps) != 3 {
t.Fatalf("got %d steps, want 3", len(plan.Steps))
}
if len(plan.Steps[2].Allocs) != 1 {
t.Errorf("step 2 allocs = %d, want 1 (remainder)", len(plan.Steps[2].Allocs))
}
if plan.Steps[2].Allocs[0] != "web-5" {
t.Errorf("step 2 allocs = %v, want [web-5]", plan.Steps[2].Allocs)
}
}
func TestRenderUpdatePlan_RollingMaxParallelUnset(t *testing.T) {
// max_parallel unset (0) → default 1 → 4 batches of 1.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan: %v", err)
}
if len(plan.Steps) != 4 {
t.Fatalf("got %d steps, want 4 (one per alloc, batch=1)", len(plan.Steps))
}
for _, s := range plan.Steps {
if len(s.Allocs) != 1 {
t.Errorf("allocs = %d, want 1", len(s.Allocs))
}
}
}
func TestRenderUpdatePlan_CanaryDefaultOne(t *testing.T) {
// canary unset → default 1 canary alloc, then 3 in batches of 2.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
MaxParallel: 2,
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan: %v", err)
}
// Step 0: canary (1 alloc). Step 1: promote. Steps 2..: remaining.
if plan.Steps[0].Action != "start" || len(plan.Steps[0].Allocs) != 1 {
t.Errorf("step 0 = %+v, want canary start with 1 alloc", plan.Steps[0])
}
if plan.Steps[0].Allocs[0] != "web-1" {
t.Errorf("canary alloc = %q, want web-1", plan.Steps[0].Allocs[0])
}
if !plan.Steps[0].WaitForHealthy {
t.Error("canary step should wait for healthy")
}
if plan.Steps[1].Action != "promote" {
t.Errorf("step 1 action = %q, want promote", plan.Steps[1].Action)
}
// Remaining: web-2, web-3, web-4 in batches of 2 → [web-2,web-3], [web-4].
if len(plan.Steps) != 4 {
t.Fatalf("got %d steps, want 4 (canary + promote + 2 batches)", len(plan.Steps))
}
if len(plan.Steps[2].Allocs) != 2 || plan.Steps[2].Allocs[0] != "web-2" {
t.Errorf("step 2 = %v, want [web-2 web-3]", plan.Steps[2].Allocs)
}
if len(plan.Steps[3].Allocs) != 1 || plan.Steps[3].Allocs[0] != "web-4" {
t.Errorf("step 3 = %v, want [web-4]", plan.Steps[3].Allocs)
}
}
func TestRenderUpdatePlan_CanaryPercent(t *testing.T) {
// count=10, canary=20% → 2 canary allocs, then 8 in batches of 3.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 10,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
MaxParallel: 3,
Canary: "20%",
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan: %v", err)
}
if len(plan.Steps[0].Allocs) != 2 {
t.Errorf("canary step allocs = %d, want 2 (20%% of 10)", len(plan.Steps[0].Allocs))
}
// Remaining 8 in batches of 3 → ceil(8/3)=3 batches.
// Steps: canary, promote, batch(3), batch(3), batch(2) = 5 steps.
if len(plan.Steps) != 5 {
t.Fatalf("got %d steps, want 5", len(plan.Steps))
}
}
func TestRenderUpdatePlan_CanaryIntegerCount(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "2",
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan: %v", err)
}
if len(plan.Steps[0].Allocs) != 2 {
t.Errorf("canary step allocs = %d, want 2", len(plan.Steps[0].Allocs))
}
if plan.Steps[0].Allocs[0] != "web-1" || plan.Steps[0].Allocs[1] != "web-2" {
t.Errorf("canary allocs = %v, want [web-1 web-2]", plan.Steps[0].Allocs)
}
}
func TestRenderUpdatePlan_CanaryFullCount(t *testing.T) {
// canary == count → no remaining allocs after canary.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 3,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "3",
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan: %v", err)
}
// Steps: canary start (3), promote. No remaining batches.
if len(plan.Steps) != 2 {
t.Fatalf("got %d steps, want 2 (canary + promote, no remainder)", len(plan.Steps))
}
if plan.Steps[1].Action != "promote" {
t.Errorf("step 1 action = %q, want promote", plan.Steps[1].Action)
}
}
func TestRenderUpdatePlan_BlueGreen(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "blue-green",
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan: %v", err)
}
// Step 0: start all 4 in parallel. Step 1: cutover.
if len(plan.Steps) != 2 {
t.Fatalf("got %d steps, want 2", len(plan.Steps))
}
if plan.Steps[0].Action != "start" {
t.Errorf("step 0 action = %q, want start", plan.Steps[0].Action)
}
if len(plan.Steps[0].Allocs) != 4 {
t.Errorf("step 0 allocs = %d, want 4 (all new in parallel)", len(plan.Steps[0].Allocs))
}
if !plan.Steps[0].WaitForHealthy {
t.Error("blue-green start step should wait for healthy")
}
if plan.Steps[1].Action != "cutover" {
t.Errorf("step 1 action = %q, want cutover", plan.Steps[1].Action)
}
if plan.Steps[1].WaitForHealthy {
t.Error("cutover step should NOT wait for healthy (already healthy)")
}
}
func TestRenderUpdatePlan_BlueGreenAllocs(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "api",
Count: 3,
Update: &jobspec.UpdateBlock{
Strategy: "blue-green",
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan: %v", err)
}
want := []string{"api-1", "api-2", "api-3"}
if len(plan.Steps[0].Allocs) != 3 {
t.Errorf("allocs = %v, want %v", plan.Steps[0].Allocs, want)
}
for i, a := range want {
if plan.Steps[0].Allocs[i] != a {
t.Errorf("alloc[%d] = %q, want %q", i, plan.Steps[0].Allocs[i], a)
}
}
}
func TestRenderUpdatePlan_NilSpec(t *testing.T) {
_, err := RenderUpdatePlan(nil)
if err == nil {
t.Fatal("expected error for nil spec")
}
if !strings.Contains(err.Error(), "spec is nil") {
t.Errorf("error = %q, want 'spec is nil'", err.Error())
}
}
func TestRenderUpdatePlan_NilUpdate(t *testing.T) {
spec := &jobspec.WorkloadSpec{Kind: "Service", Name: "web", Count: 1}
_, err := RenderUpdatePlan(spec)
if err == nil {
t.Fatal("expected error for nil update block")
}
if !strings.Contains(err.Error(), "update block is nil") {
t.Errorf("error = %q, want 'update block is nil'", err.Error())
}
}
func TestRenderUpdatePlan_CountZero(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 0,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
},
}
_, err := RenderUpdatePlan(spec)
if err == nil {
t.Fatal("expected error for count 0")
}
if !strings.Contains(err.Error(), "count must be") {
t.Errorf("error = %q, want 'count must be'", err.Error())
}
}
func TestRenderUpdatePlan_UnknownStrategy(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Update: &jobspec.UpdateBlock{
Strategy: "recreate",
},
}
_, err := RenderUpdatePlan(spec)
if err == nil {
t.Fatal("expected error for unknown strategy")
}
if !strings.Contains(err.Error(), "unknown strategy") {
t.Errorf("error = %q, want 'unknown strategy'", err.Error())
}
}
func TestRenderUpdatePlan_AllStrategies(t *testing.T) {
for _, strat := range []string{"rolling", "canary", "blue-green"} {
t.Run(strat, func(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 3,
Update: &jobspec.UpdateBlock{
Strategy: strat,
MaxParallel: 1,
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan(%s): %v", strat, err)
}
if len(plan.Steps) == 0 {
t.Errorf("strategy %s produced 0 steps", strat)
}
})
}
}
func TestRenderUpdatePlan_PromoteStepCarriesCanaryAllocs(t *testing.T) {
// The promote step lists the canary allocs so the executor knows
// which allocs are being promoted from canary to stable.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "2",
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan: %v", err)
}
promote := plan.Steps[1]
if promote.Action != "promote" {
t.Fatalf("step 1 action = %q, want promote", promote.Action)
}
if len(promote.Allocs) != 2 {
t.Errorf("promote allocs = %d, want 2 (the canary allocs)", len(promote.Allocs))
}
}
func TestRenderUpdatePlan_MaxParallelClampedToCount(t *testing.T) {
// max_parallel > count is clamped to count (defensive; validator
// rejects this but the emitter is defensive against direct
// callers).
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
MaxParallel: 99,
},
}
plan, err := RenderUpdatePlan(spec)
if err != nil {
t.Fatalf("RenderUpdatePlan: %v", err)
}
// Clamped to 2 → single batch of 2.
if len(plan.Steps) != 1 {
t.Errorf("got %d steps, want 1 (clamped)", len(plan.Steps))
}
if len(plan.Steps[0].Allocs) != 2 {
t.Errorf("step 0 allocs = %d, want 2", len(plan.Steps[0].Allocs))
}
}
+5
View File
@@ -25,6 +25,11 @@ import (
)
// Dispatcher is the public surface; constructed via NewDispatcher.
//
// Deprecated: v0.9 re-architecture replaces peer dispatch with a CLI-side
// scheduler + SSH-push (no orca binary on servers per R-001). The
// Dispatcher is retained for the dual-write window and scheduled for
// deletion in v0.10-P14. See .ciagent/PRD_v0.9.md R-001/R-006.
type Dispatcher struct {
log *slog.Logger
capacity *store.CapacityRepo
+10
View File
@@ -16,6 +16,11 @@ import (
)
// Peer is a remote orca node reachable over mTLS.
//
// Deprecated: v0.9 re-architecture replaces peer dispatch with a CLI-side
// scheduler + SSH-push (no orca binary on servers per R-001). The Peer
// type is retained for the dual-write window and scheduled for deletion
// in v0.10-P14. See .ciagent/PRD_v0.9.md R-001/R-006.
type Peer struct {
NodeID string
Address string // host:port (the peer's daemon listener)
@@ -27,6 +32,11 @@ type Peer struct {
// PeerRegistry tracks known peers. Methods are safe for concurrent
// use; the underlying map is guarded by a sync.RWMutex.
//
// Deprecated: v0.9 re-architecture replaces peer dispatch with a CLI-side
// scheduler + SSH-push (no orca binary on servers per R-001). The
// PeerRegistry is retained for the dual-write window and scheduled for
// deletion in v0.10-P14. See .ciagent/PRD_v0.9.md R-001/R-006.
type PeerRegistry struct {
mu sync.RWMutex
peers map[string]*Peer
+147
View File
@@ -0,0 +1,147 @@
package jobspec
import (
"fmt"
"os"
"path/filepath"
"strings"
)
// ParseFile reads a jobspec file from disk and dispatches on file
// extension (R-013, REQ-064):
//
// - .md → ParseMarkdown (canonical Markdown+frontmatter, R-014/R-015)
// - .yaml/.yml → ParseMarkdown with the whole file treated as
// frontmatter and Body = "" (pure YAML, no Markdown body)
// - .hcl → ParseHCL (legacy adapter; wraps the existing HCL parser
// and converts Spec{Job, Tasks} into *WorkloadSpec with Kind="Job",
// REQ-090 migration window)
//
// Unknown extensions return an error. The dispatcher preserves
// `orca job run old-spec.hcl` during the v0.9→v0.10 migration window
// (REQ-090).
func ParseFile(path string) (*WorkloadSpec, error) {
data, err := os.ReadFile(path)
if err != nil {
return nil, fmt.Errorf("read spec file: %w", err)
}
return Dispatch(data, filepath.Base(path))
}
// ParseHCLFile reads an HCL file and parses it via the legacy HCL parser,
// returning the legacy *Spec. It is a convenience wrapper retained for
// tests and direct HCL consumers that need the raw Spec{Job, Tasks}
// shape during the v0.9→v0.10 migration window (REQ-090).
//
// Deprecated: use ParseFile (dispatcher) for new code. HCL is legacy per
// R-013.
func ParseHCLFile(path string) (*Spec, error) {
data, err := os.ReadFile(path)
if err != nil {
return nil, fmt.Errorf("read spec file: %w", err)
}
return ParseHCLLegacy(data, filepath.Base(path))
}
// Dispatch routes raw jobspec bytes on file extension to the
// appropriate parser. filename is used only for HCL (the HCL decoder
// needs a filename for error messages and syntax sniffing).
func Dispatch(data []byte, filename string) (*WorkloadSpec, error) {
ext := strings.ToLower(filepath.Ext(filename))
switch ext {
case ".md":
return ParseMarkdown(data)
case ".yaml", ".yml":
// Pure YAML file: no Markdown body. Treat the whole file as
// the frontmatter block. Body is empty (R-015: no body to
// preserve).
spec, err := parseYAMLFile(data)
if err != nil {
return nil, err
}
return spec, nil
case ".hcl":
return ParseHCL(data, filename)
default:
return nil, fmt.Errorf("parse jobspec: unknown extension %q (want .md, .yaml, .yml, or .hcl)", ext)
}
}
// parseYAMLFile treats the whole file as a frontmatter block (no
// surrounding `---` delimiters, no Markdown body). This routes .yaml
// and .yml files through the same hand-rolled parser as .md.
func parseYAMLFile(data []byte) (*WorkloadSpec, error) {
block := string(data)
if strings.TrimSpace(block) == "" {
return nil, fmt.Errorf("parse yaml: empty file")
}
spec, err := parseFrontmatterBlock(block)
if err != nil {
return nil, err
}
spec.Body = ""
if err := validateWorkload(spec); err != nil {
return nil, err
}
return spec, nil
}
// ParseHCL parses a legacy HCL jobspec and adapts it into a *WorkloadSpec
// (REQ-064 adapter, REQ-090 migration window). The existing HCL
// Spec{Job, Tasks} shape is converted to:
//
// Kind: "Job"
// Name: spec.Job.Name
// Runtime: {one_of: "process", command: tasks[0].Command}
//
// Body is empty (HCL has no Markdown body). The legacy Spec struct and
// ParseHCLLegacy are retained for direct HCL consumers that have not yet
// migrated.
//
// Deprecated: use the dispatcher (ParseFile/Dispatch). HCL is legacy
// per R-013; the HCL path is retained only for the v0.9→v0.10 migration
// window (REQ-090) and will be removed in v1.0.
func ParseHCL(data []byte, filename string) (*WorkloadSpec, error) {
spec, err := ParseHCLLegacy(data, filename)
if err != nil {
return nil, err
}
ws := &WorkloadSpec{
SpecVersion: "",
Kind: "Job",
Name: spec.Job.Name,
Count: 1,
Body: "",
}
if len(spec.Tasks) > 0 {
ws.Runtime = &RuntimeBlock{
OneOf: "process",
Command: spec.Tasks[0].Command,
}
}
return ws, nil
}
// ParseHCLLegacy is the original HCL-only parser retained for direct
// HCL consumers (e.g. the cli/job.go toTaskSpecs path during the
// migration window). New code should call ParseHCL (which returns a
// *WorkloadSpec) or the dispatcher. Deprecated: HCL is legacy per
// R-013; see ParseHCL.
func ParseHCLLegacy(data []byte, filename string) (*Spec, error) {
var spec Spec
if err := hclDecode(filename, data, &spec); err != nil {
return nil, fmt.Errorf("decode hcl: %w", err)
}
if spec.Job.Name == "" {
return nil, fmt.Errorf("spec missing job name")
}
if len(spec.Tasks) == 0 {
return nil, fmt.Errorf("spec must have at least one task")
}
for i, t := range spec.Tasks {
if t.Command == "" {
return nil, fmt.Errorf("task[%d] (%s) missing command", i, t.Name)
}
}
return &spec, nil
}
+222
View File
@@ -0,0 +1,222 @@
package jobspec
import (
"os"
"path/filepath"
"strings"
"testing"
)
func TestDispatch_Markdown(t *testing.T) {
input := "---\nkind: Job\nname: md-job\n---\nbody content\n"
ws, err := Dispatch([]byte(input), "spec.md")
if err != nil {
t.Fatalf("Dispatch .md: %v", err)
}
if ws.Kind != "Job" {
t.Errorf("Kind = %q, want Job", ws.Kind)
}
if ws.Name != "md-job" {
t.Errorf("Name = %q, want md-job", ws.Name)
}
if ws.Body != "body content\n" {
t.Errorf("Body = %q, want %q (R-015)", ws.Body, "body content\n")
}
}
func TestDispatch_YAML(t *testing.T) {
input := "kind: Service\nname: yaml-svc\nports:\n - name: http\n port: 80\n"
ws, err := Dispatch([]byte(input), "spec.yaml")
if err != nil {
t.Fatalf("Dispatch .yaml: %v", err)
}
if ws.Kind != "Service" {
t.Errorf("Kind = %q, want Service", ws.Kind)
}
if ws.Name != "yaml-svc" {
t.Errorf("Name = %q, want yaml-svc", ws.Name)
}
if ws.Body != "" {
t.Errorf("Body = %q, want empty (YAML has no body)", ws.Body)
}
if len(ws.Ports) != 1 || ws.Ports[0].Name != "http" || ws.Ports[0].Port != 80 {
t.Errorf("Ports = %+v, want one http:80", ws.Ports)
}
}
func TestDispatch_YML(t *testing.T) {
input := "kind: DaemonSet\nname: yml-ds\n"
ws, err := Dispatch([]byte(input), "spec.yml")
if err != nil {
t.Fatalf("Dispatch .yml: %v", err)
}
if ws.Kind != "DaemonSet" {
t.Errorf("Kind = %q, want DaemonSet", ws.Kind)
}
if ws.Body != "" {
t.Errorf("Body = %q, want empty", ws.Body)
}
}
func TestDispatch_HCLAdapter(t *testing.T) {
hcl := `job "demo" {}
task "build" {
command = "/bin/echo"
args = ["hello"]
}
`
ws, err := Dispatch([]byte(hcl), "spec.hcl")
if err != nil {
t.Fatalf("Dispatch .hcl: %v", err)
}
if ws.Kind != "Job" {
t.Errorf("Kind = %q, want Job (adapter always sets Job)", ws.Kind)
}
if ws.Name != "demo" {
t.Errorf("Name = %q, want demo (from spec.Job.Name)", ws.Name)
}
if ws.Runtime == nil {
t.Fatal("Runtime is nil; adapter should populate from tasks[0]")
}
if ws.Runtime.OneOf != "process" {
t.Errorf("Runtime.OneOf = %q, want process", ws.Runtime.OneOf)
}
if ws.Runtime.Command != "/bin/echo" {
t.Errorf("Runtime.Command = %q, want /bin/echo (from tasks[0].Command)", ws.Runtime.Command)
}
if ws.Body != "" {
t.Errorf("Body = %q, want empty (HCL has no body)", ws.Body)
}
}
func TestDispatch_HCLAdapterNoTasks(t *testing.T) {
hcl := `job "x" {}`
_, err := Dispatch([]byte(hcl), "spec.hcl")
if err == nil {
t.Fatal("expected error for HCL with no tasks")
}
if !strings.Contains(err.Error(), "at least one task") {
t.Errorf("error = %q, want it to contain 'at least one task'", err.Error())
}
}
func TestDispatch_UnknownExtension(t *testing.T) {
_, err := Dispatch([]byte("kind: Job\nname: x\n"), "spec.json")
if err == nil {
t.Fatal("expected error for unknown extension, got nil")
}
if !strings.Contains(err.Error(), "unknown extension") {
t.Errorf("error = %q, want it to contain 'unknown extension'", err.Error())
}
}
func TestDispatch_NoExtension(t *testing.T) {
_, err := Dispatch([]byte("kind: Job\nname: x\n"), "spec")
if err == nil {
t.Fatal("expected error for no extension, got nil")
}
}
func TestDispatch_EmptyYAML(t *testing.T) {
_, err := Dispatch([]byte(""), "spec.yaml")
if err == nil {
t.Fatal("expected error for empty YAML, got nil")
}
}
func TestParseFile_Markdown(t *testing.T) {
dir := t.TempDir()
path := filepath.Join(dir, "spec.md")
content := "---\nkind: Job\nname: file-md\n---\nbody\n"
if err := os.WriteFile(path, []byte(content), 0o644); err != nil {
t.Fatalf("write: %v", err)
}
ws, err := ParseFile(path)
if err != nil {
t.Fatalf("ParseFile .md: %v", err)
}
if ws.Kind != "Job" || ws.Name != "file-md" {
t.Errorf("got Kind=%q Name=%q", ws.Kind, ws.Name)
}
if ws.Body != "body\n" {
t.Errorf("Body = %q, want %q", ws.Body, "body\n")
}
}
func TestParseFile_YAML(t *testing.T) {
dir := t.TempDir()
path := filepath.Join(dir, "spec.yaml")
content := "kind: Service\nname: file-yaml\n"
if err := os.WriteFile(path, []byte(content), 0o644); err != nil {
t.Fatalf("write: %v", err)
}
ws, err := ParseFile(path)
if err != nil {
t.Fatalf("ParseFile .yaml: %v", err)
}
if ws.Kind != "Service" || ws.Name != "file-yaml" {
t.Errorf("got Kind=%q Name=%q", ws.Kind, ws.Name)
}
if ws.Body != "" {
t.Errorf("Body = %q, want empty", ws.Body)
}
}
func TestParseFile_HCL(t *testing.T) {
dir := t.TempDir()
path := filepath.Join(dir, "spec.hcl")
content := `job "file-hcl" {}
task "t" { command = "/bin/true" }
`
if err := os.WriteFile(path, []byte(content), 0o644); err != nil {
t.Fatalf("write: %v", err)
}
ws, err := ParseFile(path)
if err != nil {
t.Fatalf("ParseFile .hcl: %v", err)
}
if ws.Kind != "Job" || ws.Name != "file-hcl" {
t.Errorf("got Kind=%q Name=%q", ws.Kind, ws.Name)
}
if ws.Runtime == nil || ws.Runtime.Command != "/bin/true" {
t.Errorf("Runtime.Command = %v, want /bin/true", ws.Runtime)
}
}
func TestParseFile_MissingFile(t *testing.T) {
_, err := ParseFile(filepath.Join(t.TempDir(), "nope.md"))
if err == nil {
t.Fatal("expected error for missing file, got nil")
}
if !strings.Contains(err.Error(), "read spec file") {
t.Errorf("error = %q, want it to contain 'read spec file'", err.Error())
}
}
func TestParseFile_UnknownExtension(t *testing.T) {
dir := t.TempDir()
path := filepath.Join(dir, "spec.txt")
if err := os.WriteFile(path, []byte("kind: Job\nname: x\n"), 0o644); err != nil {
t.Fatalf("write: %v", err)
}
_, err := ParseFile(path)
if err == nil {
t.Fatal("expected error for unknown extension, got nil")
}
if !strings.Contains(err.Error(), "unknown extension") {
t.Errorf("error = %q, want 'unknown extension'", err.Error())
}
}
func TestParseHCL_LegacySpec(t *testing.T) {
hcl := `job "legacy" {}
task "t" { command = "/bin/echo" }
`
ws, err := ParseHCL([]byte(hcl), "spec.hcl")
if err != nil {
t.Fatalf("ParseHCL: %v", err)
}
if ws.Kind != "Job" || ws.Name != "legacy" {
t.Errorf("adapter got Kind=%q Name=%q", ws.Kind, ws.Name)
}
}
File diff suppressed because it is too large Load Diff
+109
View File
@@ -0,0 +1,109 @@
package jobspec
import (
"strings"
"testing"
)
// FuzzParseMarkdownRoundTrip is the REQ-067 fuzz harness for R-015
// byte-exact body preservation. It generates random frontmatter + body
// combinations, runs ParseMarkdown, and asserts that the parsed Body
// equals the original body byte-for-byte whenever parsing succeeds.
// When parsing fails (bad frontmatter), the iteration passes — the
// parser is allowed to reject malformed input.
//
// The seed corpus (added via f.Add) covers adversarial fixtures: CRLF
// body, BOM prefix, no frontmatter, only-closing-separator, body with
// `---` inside a code fence, trailing whitespace, empty body. The seed
// corpus runs as regular tests under `go test` (CI); random input runs
// only under `go test -fuzz=FuzzParseMarkdownRoundTrip` in a dedicated
// process.
func FuzzParseMarkdownRoundTrip(f *testing.F) {
// Seed 1: valid frontmatter + simple body.
f.Add([]byte("---\nkind: Job\nname: seed1\n---\n# body\n"))
// Seed 2: CRLF body.
f.Add([]byte("---\r\nkind: Job\r\nname: seed2\r\n---\r\n# body\r\nCRLF\r\n"))
// Seed 3: BOM prefix.
f.Add([]byte("\uFEFF---\nkind: Job\nname: seed3\n---\nbody\n"))
// Seed 4: no frontmatter (just body) — should fail to parse.
f.Add([]byte("# just a body\nno frontmatter\n"))
// Seed 5: frontmatter with only the closing `---` (no opening).
f.Add([]byte("body\n---\nmore body\n"))
// Seed 6: body containing `---` in a code fence.
f.Add([]byte("---\nkind: Job\nname: seed6\n---\n```bash\necho '---'\n```\n"))
// Seed 7: body with trailing whitespace.
f.Add([]byte("---\nkind: Job\nname: seed7\n---\nbody with trailing spaces \n"))
// Seed 8: empty body.
f.Add([]byte("---\nkind: Job\nname: seed8\n---\n"))
// Seed 9: empty frontmatter (should fail).
f.Add([]byte("---\n---\nbody\n"))
// Seed 10: body with no trailing newline.
f.Add([]byte("---\nkind: Job\nname: seed10\n---\nno trailing newline"))
f.Fuzz(func(t *testing.T, data []byte) {
// Reconstruct the body from the input so we can assert
// byte-exact round-trip. We do this by re-splitting the
// frontmatter using the same logic the parser uses, but only
// to extract the expected body. If the input has no valid
// frontmatter delimiter pair, ParseMarkdown will return an
// error and we pass the iteration.
expectedBody := extractExpectedBody(string(data))
spec, err := ParseMarkdown(data)
if err != nil {
// Parser rejected the input — acceptable for a fuzz
// iteration (the input may be malformed). Pass.
return
}
// R-015: body must be byte-exact.
if spec.Body != expectedBody {
t.Errorf("R-015 body round-trip mismatch:\n got = %q\nwant = %q", spec.Body, expectedBody)
}
})
}
// extractExpectedBody returns the body portion of a Markdown jobspec
// input using the same delimiter-splitting logic as splitFrontmatter,
// so the fuzz harness can assert byte-exact preservation independently
// of the parser's internal extraction. If the input has no valid
// frontmatter, the result is "" (and ParseMarkdown will error).
func extractExpectedBody(content string) string {
stripped := content
if strings.HasPrefix(stripped, "\uFEFF") {
stripped = stripped[len("\uFEFF"):]
}
trimmed := strings.TrimLeft(stripped, "\r\n\t ")
if !strings.HasPrefix(trimmed, "---") {
return ""
}
rest := trimmed[3:]
if len(rest) > 0 && rest[0] != '\n' && rest[0] != '\r' {
return ""
}
rest = strings.TrimLeft(rest, "\r\n")
idx := findClosingDelimiter(rest)
if idx < 0 {
return ""
}
afterClose := rest[idx:]
newlineIdx := strings.IndexAny(afterClose, "\r\n")
if newlineIdx < 0 {
return ""
}
bodyStart := newlineIdx
if strings.HasPrefix(afterClose[bodyStart:], "\r\n") {
bodyStart += 2
} else {
bodyStart += 1
}
return afterClose[bodyStart:]
}
+852
View File
@@ -0,0 +1,852 @@
package jobspec
import (
"strings"
"testing"
)
func TestParseMarkdown_FullFrontmatter(t *testing.T) {
body := "# Hello\n\nThis is the body.\n\nTrailing newline preserved.\n"
input := "---\n" +
"orca-spec-version: \"1\"\n" +
"kind: Job\n" +
"name: my-job\n" +
"count: 3\n" +
"---\n" +
body
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.SpecVersion != "1" {
t.Errorf("SpecVersion = %q, want %q", spec.SpecVersion, "1")
}
if spec.Kind != "Job" {
t.Errorf("Kind = %q, want %q", spec.Kind, "Job")
}
if spec.Name != "my-job" {
t.Errorf("Name = %q, want %q", spec.Name, "my-job")
}
if spec.Count != 3 {
t.Errorf("Count = %d, want 3", spec.Count)
}
if spec.Body != body {
t.Errorf("Body = %q, want %q (byte-exact, R-015)", spec.Body, body)
}
}
func TestParseMarkdown_BodyByteExactTrailingNewline(t *testing.T) {
cases := []struct {
name string
body string
}{
{"with_trailing_newline", "# Title\n\nbody\n"},
{"with_double_trailing_newline", "# Title\n\nbody\n\n"},
{"no_trailing_newline", "# Title\n\nbody"},
{"empty_body_with_newline", "\n"},
{"only_newlines", "\n\n\n"},
}
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
input := "---\nkind: Job\nname: x\n---\n" + tc.body
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Body != tc.body {
t.Errorf("Body byte-exact mismatch (R-015):\n got = %q\nwant = %q", spec.Body, tc.body)
}
})
}
}
func TestParseMarkdown_NoFrontmatter(t *testing.T) {
input := "# Just a body\n\nNo frontmatter here."
_, err := ParseMarkdown([]byte(input))
if err == nil {
t.Fatal("expected error for missing frontmatter, got nil")
}
if !strings.Contains(err.Error(), "frontmatter") {
t.Errorf("error = %q, want it to contain 'frontmatter'", err.Error())
}
}
func TestParseMarkdown_EmptyFrontmatter(t *testing.T) {
input := "---\n---\n\nbody"
_, err := ParseMarkdown([]byte(input))
if err == nil {
t.Fatal("expected error for empty frontmatter, got nil")
}
if !strings.Contains(err.Error(), "empty frontmatter") {
t.Errorf("error = %q, want it to contain 'empty frontmatter'", err.Error())
}
}
func TestParseMarkdown_UnknownKind(t *testing.T) {
input := "---\nkind: CronJob\nname: x\n---\nbody\n"
_, err := ParseMarkdown([]byte(input))
if err == nil {
t.Fatal("expected error for unknown kind, got nil")
}
if !strings.Contains(err.Error(), "not one of") {
t.Errorf("error = %q, want it to contain 'not one of'", err.Error())
}
}
func TestParseMarkdown_MissingName(t *testing.T) {
input := "---\nkind: Job\n---\nbody\n"
_, err := ParseMarkdown([]byte(input))
if err == nil {
t.Fatal("expected error for missing name, got nil")
}
if !strings.Contains(err.Error(), "missing name") {
t.Errorf("error = %q, want it to contain 'missing name'", err.Error())
}
}
func TestParseMarkdown_MissingKind(t *testing.T) {
input := "---\nname: x\n---\nbody\n"
_, err := ParseMarkdown([]byte(input))
if err == nil {
t.Fatal("expected error for missing kind, got nil")
}
if !strings.Contains(err.Error(), "missing kind") {
t.Errorf("error = %q, want it to contain 'missing kind'", err.Error())
}
}
func TestParseMarkdown_EachValidKind(t *testing.T) {
cases := []string{"Job", "Service", "DaemonSet"}
for _, kind := range cases {
t.Run(kind, func(t *testing.T) {
input := "---\nkind: " + kind + "\nname: x\n---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Kind != kind {
t.Errorf("Kind = %q, want %q", spec.Kind, kind)
}
})
}
}
func TestParseMarkdown_EnvScalarAndObject(t *testing.T) {
input := "---\n" +
"kind: Job\n" +
"name: x\n" +
"env:\n" +
" FOO: bar\n" +
" BAZ: \"qux\"\n" +
" SECRET_REF:\n" +
" from: \"secret:db-password\"\n" +
" INLINE: {from: \"secret:token\"}\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if got := spec.Env["FOO"]; got != "bar" {
t.Errorf("env[FOO] = %q, want %q", got, "bar")
}
if got := spec.Env["BAZ"]; got != "qux" {
t.Errorf("env[BAZ] = %q, want %q", got, "qux")
}
if got := spec.Env["INLINE"]; got != `{from: "secret:token"}` {
t.Errorf("env[INLINE] = %q, want the raw object string", got)
}
if _, ok := spec.Env["SECRET_REF"]; !ok {
t.Errorf("env[SECRET_REF] missing; nested from: stored as empty string")
}
}
func TestParseMarkdown_PortsArray(t *testing.T) {
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"ports:\n" +
" - name: http\n" +
" port: 8080\n" +
" host_port: 80\n" +
" protocol: tcp\n" +
" - name: https\n" +
" port: 8443\n" +
" host_port: 443\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if len(spec.Ports) != 2 {
t.Fatalf("Ports = %d, want 2", len(spec.Ports))
}
if spec.Ports[0].Name != "http" || spec.Ports[0].Port != 8080 || spec.Ports[0].HostPort != 80 || spec.Ports[0].Protocol != "tcp" {
t.Errorf("Ports[0] = %+v", spec.Ports[0])
}
if spec.Ports[1].Name != "https" || spec.Ports[1].Port != 8443 || spec.Ports[1].HostPort != 443 {
t.Errorf("Ports[1] = %+v", spec.Ports[1])
}
}
func TestParseMarkdown_VolumesArray(t *testing.T) {
input := "---\n" +
"kind: Job\n" +
"name: x\n" +
"volumes:\n" +
" - name: data\n" +
" type: host\n" +
" source: /data\n" +
" target: /data\n" +
" read_only: true\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if len(spec.Volumes) != 1 {
t.Fatalf("Volumes = %d, want 1", len(spec.Volumes))
}
v := spec.Volumes[0]
if v.Name != "data" || v.Type != "host" || v.Source != "/data" || v.Target != "/data" || !v.ReadOnly {
t.Errorf("Volumes[0] = %+v", v)
}
}
func TestParseMarkdown_RuntimeBlock(t *testing.T) {
input := "---\n" +
"kind: Job\n" +
"name: x\n" +
"runtime:\n" +
" one_of: process\n" +
" image: docker.io/nginx:latest\n" +
" command: /bin/sh -c 'echo hi'\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Runtime == nil {
t.Fatal("Runtime is nil")
}
if spec.Runtime.OneOf != "process" {
t.Errorf("Runtime.OneOf = %q, want %q", spec.Runtime.OneOf, "process")
}
if spec.Runtime.Image != "docker.io/nginx:latest" {
t.Errorf("Runtime.Image = %q, want %q", spec.Runtime.Image, "docker.io/nginx:latest")
}
if spec.Runtime.Command != "/bin/sh -c 'echo hi'" {
t.Errorf("Runtime.Command = %q, want %q", spec.Runtime.Command, "/bin/sh -c 'echo hi'")
}
}
func TestParseMarkdown_SecretsInlineArray(t *testing.T) {
input := "---\nkind: Job\nname: x\nsecrets: [\"db-password\", \"api-token\"]\n---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if len(spec.Secrets) != 2 {
t.Fatalf("Secrets = %d, want 2", len(spec.Secrets))
}
if spec.Secrets[0] != "db-password" || spec.Secrets[1] != "api-token" {
t.Errorf("Secrets = %v, want [db-password api-token]", spec.Secrets)
}
}
func TestParseMarkdown_SecretsBlockArray(t *testing.T) {
input := "---\n" +
"kind: Job\n" +
"name: x\n" +
"secrets:\n" +
" - db-password\n" +
" - api-token\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if len(spec.Secrets) != 2 {
t.Fatalf("Secrets = %d, want 2", len(spec.Secrets))
}
if spec.Secrets[0] != "db-password" || spec.Secrets[1] != "api-token" {
t.Errorf("Secrets = %v, want [db-password api-token]", spec.Secrets)
}
}
func TestParseMarkdown_CRLFBodyPreserved(t *testing.T) {
body := "# Title\r\n\r\nCRLF body.\r\n"
input := "---\r\nkind: Job\r\nname: x\r\n---\r\n" + body
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Body != body {
t.Errorf("CRLF body not preserved (R-015):\n got = %q\nwant = %q", spec.Body, body)
}
}
func TestParseMarkdown_BOMStrippedFromFrontmatter(t *testing.T) {
body := "# body\n"
input := "\uFEFF" + "---\nkind: Job\nname: x\n---\n" + body
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Kind != "Job" {
t.Errorf("Kind = %q, want Job (BOM should be stripped from frontmatter scan)", spec.Kind)
}
if spec.Body != body {
t.Errorf("Body = %q, want %q", spec.Body, body)
}
}
func TestParseMarkdown_BodyWithCodeFenceContainingDashes(t *testing.T) {
body := "```bash\n" +
"echo '---'\n" +
"echo '--- end ---'\n" +
"```\n"
input := "---\nkind: Job\nname: x\n---\n" + body
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Body != body {
t.Errorf("Body with code-fence --- not preserved (R-015):\n got = %q\nwant = %q", spec.Body, body)
}
}
func TestParseMarkdown_OnlyClosingSeparator(t *testing.T) {
input := "no opening\n---\nbody\n"
_, err := ParseMarkdown([]byte(input))
if err == nil {
t.Fatal("expected error for input with only closing separator, got nil")
}
}
func TestParseMarkdown_QuotedValues(t *testing.T) {
input := "---\nkind: \"Job\"\nname: 'my-job'\n---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Kind != "Job" {
t.Errorf("Kind = %q, want Job (double-quoted)", spec.Kind)
}
if spec.Name != "my-job" {
t.Errorf("Name = %q, want my-job (single-quoted)", spec.Name)
}
}
func TestParseMarkdown_CountDefault(t *testing.T) {
input := "---\nkind: Job\nname: x\n---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Count != 1 {
t.Errorf("Count default = %d, want 1", spec.Count)
}
}
func TestParseMarkdown_UnknownKeyIgnored(t *testing.T) {
input := "---\nkind: Job\nname: x\nfuture_field: value\n---\nbody\n"
_, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown should ignore unknown keys: %v", err)
}
}
func TestParseMarkdown_RestartBlock(t *testing.T) {
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"restart:\n" +
" mode: service\n" +
" attempts: 5\n" +
" delay: 3s\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Restart == nil {
t.Fatal("Restart is nil")
}
if spec.Restart.Mode != "service" {
t.Errorf("Restart.Mode = %q, want service", spec.Restart.Mode)
}
if spec.Restart.MaxRetries != 5 {
t.Errorf("Restart.MaxRetries = %d, want 5", spec.Restart.MaxRetries)
}
if spec.Restart.Delay != "3s" {
t.Errorf("Restart.Delay = %q, want 3s", spec.Restart.Delay)
}
}
func TestParseMarkdown_RestartBlockMaxRetriesAlias(t *testing.T) {
// max_retries is the canonical key; attempts is an accepted alias.
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"restart:\n" +
" mode: on-failure\n" +
" max_retries: 3\n" +
" delay: 1s\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Restart == nil || spec.Restart.MaxRetries != 3 {
t.Fatalf("Restart.MaxRetries = %d, want 3 (max_retries alias)", spec.Restart.MaxRetries)
}
}
func TestParseMarkdown_UpdateBlock(t *testing.T) {
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"update:\n" +
" strategy: canary\n" +
" max_parallel: 2\n" +
" min_healthy_time: 30s\n" +
" healthy_deadline: 5m\n" +
" canary: 10%\n" +
" auto_promote: true\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Update == nil {
t.Fatal("Update is nil")
}
if spec.Update.Strategy != "canary" {
t.Errorf("Update.Strategy = %q, want canary", spec.Update.Strategy)
}
if spec.Update.MaxParallel != 2 {
t.Errorf("Update.MaxParallel = %d, want 2", spec.Update.MaxParallel)
}
if spec.Update.MinHealthyTime != "30s" {
t.Errorf("Update.MinHealthyTime = %q, want 30s", spec.Update.MinHealthyTime)
}
if spec.Update.HealthyDeadline != "5m" {
t.Errorf("Update.HealthyDeadline = %q, want 5m", spec.Update.HealthyDeadline)
}
if spec.Update.Canary != "10%" {
t.Errorf("Update.Canary = %q, want 10%%", spec.Update.Canary)
}
if !spec.Update.AutoPromote {
t.Errorf("Update.AutoPromote = false, want true")
}
}
func TestParseMarkdown_ServiceBlock(t *testing.T) {
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"service:\n" +
" name: web\n" +
" port: 8080\n" +
" bind: 127.0.0.1\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Service == nil {
t.Fatal("Service is nil")
}
if spec.Service.Name != "web" {
t.Errorf("Service.Name = %q, want web", spec.Service.Name)
}
if spec.Service.Port != 8080 {
t.Errorf("Service.Port = %d, want 8080", spec.Service.Port)
}
if spec.Service.Bind != "127.0.0.1" {
t.Errorf("Service.Bind = %q, want 127.0.0.1", spec.Service.Bind)
}
}
func TestParseMarkdown_HealthBlock(t *testing.T) {
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"health:\n" +
" check_type: http\n" +
" interval: 10s\n" +
" timeout: 2s\n" +
" unhealthy_threshold: 3\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Health == nil {
t.Fatal("Health is nil")
}
if spec.Health.CheckType != "http" {
t.Errorf("Health.CheckType = %q, want http", spec.Health.CheckType)
}
if spec.Health.Interval != "10s" {
t.Errorf("Health.Interval = %q, want 10s", spec.Health.Interval)
}
if spec.Health.Timeout != "2s" {
t.Errorf("Health.Timeout = %q, want 2s", spec.Health.Timeout)
}
if spec.Health.UnhealthyThreshold != 3 {
t.Errorf("Health.UnhealthyThreshold = %d, want 3", spec.Health.UnhealthyThreshold)
}
}
func TestParseMarkdown_ConstraintsInlineArray(t *testing.T) {
// Inline flow-array form: the parser does NOT unescape YAML
// escapes (consistent with the secrets inline parser). Use
// single-quoted scalars inside the flow array so the CEL strings
// are preserved verbatim.
input := "---\nkind: Service\nname: web\nconstraints: ['node.role == \"web\"', 'region == \"us\"']\n---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if len(spec.Constraints) != 2 {
t.Fatalf("Constraints = %d, want 2", len(spec.Constraints))
}
if spec.Constraints[0] != `node.role == "web"` {
t.Errorf("Constraints[0] = %q", spec.Constraints[0])
}
if spec.Constraints[1] != `region == "us"` {
t.Errorf("Constraints[1] = %q", spec.Constraints[1])
}
}
func TestParseMarkdown_ConstraintsBlockArray(t *testing.T) {
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"constraints:\n" +
" - node.role == \"web\"\n" +
" - region == \"us\"\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if len(spec.Constraints) != 2 {
t.Fatalf("Constraints = %d, want 2", len(spec.Constraints))
}
if spec.Constraints[0] != `node.role == "web"` {
t.Errorf("Constraints[0] = %q", spec.Constraints[0])
}
if spec.Constraints[1] != `region == "us"` {
t.Errorf("Constraints[1] = %q", spec.Constraints[1])
}
}
func TestParseMarkdown_AffinityBlock(t *testing.T) {
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"affinity:\n" +
" - target: node.role == \"web\"\n" +
" weight: 100\n" +
" - target: region == \"us\"\n" +
" weight: 50\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if len(spec.Affinity) != 2 {
t.Fatalf("Affinity = %d, want 2", len(spec.Affinity))
}
if spec.Affinity[0].Target != `node.role == "web"` {
t.Errorf("Affinity[0].Target = %q", spec.Affinity[0].Target)
}
if spec.Affinity[0].Weight != 100 {
t.Errorf("Affinity[0].Weight = %d, want 100", spec.Affinity[0].Weight)
}
if spec.Affinity[1].Target != `region == "us"` {
t.Errorf("Affinity[1].Target = %q", spec.Affinity[1].Target)
}
if spec.Affinity[1].Weight != 50 {
t.Errorf("Affinity[1].Weight = %d, want 50", spec.Affinity[1].Weight)
}
}
func TestParseMarkdown_LifecycleBlock(t *testing.T) {
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"lifecycle:\n" +
" pre_stop:\n" +
" - /bin/sh -c 'sleep 5'\n" +
" - /usr/local/bin/drain.sh\n" +
" post_start:\n" +
" - /usr/local/bin/warm-cache.sh\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Lifecycle == nil {
t.Fatal("Lifecycle is nil")
}
if len(spec.Lifecycle.PreStop) != 2 {
t.Fatalf("PreStop = %d, want 2", len(spec.Lifecycle.PreStop))
}
if spec.Lifecycle.PreStop[0] != "/bin/sh -c 'sleep 5'" {
t.Errorf("PreStop[0] = %q", spec.Lifecycle.PreStop[0])
}
if spec.Lifecycle.PreStop[1] != "/usr/local/bin/drain.sh" {
t.Errorf("PreStop[1] = %q", spec.Lifecycle.PreStop[1])
}
if len(spec.Lifecycle.PostStart) != 1 {
t.Fatalf("PostStart = %d, want 1", len(spec.Lifecycle.PostStart))
}
if spec.Lifecycle.PostStart[0] != "/usr/local/bin/warm-cache.sh" {
t.Errorf("PostStart[0] = %q", spec.Lifecycle.PostStart[0])
}
}
func TestParseMarkdown_LifecycleInlineArray(t *testing.T) {
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"lifecycle:\n" +
" pre_stop: [\"/bin/true\"]\n" +
" post_start: [\"/bin/warmup\", \"/bin/check\"]\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Lifecycle == nil {
t.Fatal("Lifecycle is nil")
}
if len(spec.Lifecycle.PreStop) != 1 || spec.Lifecycle.PreStop[0] != "/bin/true" {
t.Errorf("PreStop = %v, want [/bin/true]", spec.Lifecycle.PreStop)
}
if len(spec.Lifecycle.PostStart) != 2 {
t.Fatalf("PostStart = %v, want 2 entries", spec.Lifecycle.PostStart)
}
if spec.Lifecycle.PostStart[0] != "/bin/warmup" || spec.Lifecycle.PostStart[1] != "/bin/check" {
t.Errorf("PostStart = %v, want [/bin/warmup /bin/check]", spec.Lifecycle.PostStart)
}
}
func TestParseMarkdown_FullServiceSpec(t *testing.T) {
// A complete Service spec exercising every P02-parsed block together.
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"count: 3\n" +
"runtime:\n" +
" one_of: process\n" +
" command: /usr/bin/httpd\n" +
"ports:\n" +
" - name: http\n" +
" port: 8080\n" +
"restart:\n" +
" mode: service\n" +
" attempts: 5\n" +
" delay: 2s\n" +
"update:\n" +
" strategy: rolling\n" +
" max_parallel: 1\n" +
" auto_promote: false\n" +
"service:\n" +
" name: web\n" +
" port: 8080\n" +
"health:\n" +
" check_type: http\n" +
" interval: 5s\n" +
" timeout: 1s\n" +
" unhealthy_threshold: 2\n" +
"constraints:\n" +
" - node.role == \"web\"\n" +
"affinity:\n" +
" - target: zone == \"a\"\n" +
" weight: 80\n" +
"lifecycle:\n" +
" post_start:\n" +
" - /bin/ready.sh\n" +
"---\n# body\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Restart == nil || spec.Restart.Mode != "service" {
t.Errorf("Restart not parsed: %+v", spec.Restart)
}
if spec.Update == nil || spec.Update.Strategy != "rolling" {
t.Errorf("Update not parsed: %+v", spec.Update)
}
if spec.Service == nil || spec.Service.Port != 8080 {
t.Errorf("Service not parsed: %+v", spec.Service)
}
if spec.Health == nil || spec.Health.CheckType != "http" {
t.Errorf("Health not parsed: %+v", spec.Health)
}
if len(spec.Constraints) != 1 {
t.Errorf("Constraints = %v", spec.Constraints)
}
if len(spec.Affinity) != 1 || spec.Affinity[0].Weight != 80 {
t.Errorf("Affinity = %v", spec.Affinity)
}
if spec.Lifecycle == nil || len(spec.Lifecycle.PostStart) != 1 {
t.Errorf("Lifecycle not parsed: %+v", spec.Lifecycle)
}
if spec.Body != "# body\n" {
t.Errorf("Body = %q, want %q (R-015)", spec.Body, "# body\n")
}
}
func TestParseMarkdown_TasksBlock(t *testing.T) {
// P06: a task group with two tasks, each carrying its own runtime
// and command. The parser must populate spec.Tasks with two
// entries preserving name, runtime (one_of/image/command), and
// the task-level command.
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"tasks:\n" +
" - name: app\n" +
" runtime:\n" +
" one_of: process\n" +
" image: docker.io/nginx:latest\n" +
" command: /usr/bin/httpd -f\n" +
" command: /usr/bin/httpd -f\n" +
" - name: sidecar\n" +
" runtime:\n" +
" one_of: wasm\n" +
" command: /bin/wasm-runner sidecar.wasm\n" +
" command: /bin/wasm-runner sidecar.wasm\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if len(spec.Tasks) != 2 {
t.Fatalf("Tasks = %d, want 2", len(spec.Tasks))
}
app := spec.Tasks[0]
if app.Name != "app" {
t.Errorf("Tasks[0].Name = %q, want app", app.Name)
}
if app.Runtime == nil {
t.Fatal("Tasks[0].Runtime is nil")
}
if app.Runtime.OneOf != "process" {
t.Errorf("Tasks[0].Runtime.OneOf = %q, want process", app.Runtime.OneOf)
}
if app.Runtime.Image != "docker.io/nginx:latest" {
t.Errorf("Tasks[0].Runtime.Image = %q", app.Runtime.Image)
}
if app.Runtime.Command != "/usr/bin/httpd -f" {
t.Errorf("Tasks[0].Runtime.Command = %q", app.Runtime.Command)
}
if app.Command != "/usr/bin/httpd -f" {
t.Errorf("Tasks[0].Command = %q", app.Command)
}
side := spec.Tasks[1]
if side.Name != "sidecar" {
t.Errorf("Tasks[1].Name = %q, want sidecar", side.Name)
}
if side.Runtime == nil || side.Runtime.OneOf != "wasm" {
t.Errorf("Tasks[1].Runtime = %+v, want one_of=wasm", side.Runtime)
}
if side.Command != "/bin/wasm-runner sidecar.wasm" {
t.Errorf("Tasks[1].Command = %q", side.Command)
}
}
func TestParseMarkdown_TasksBlockWithEnv(t *testing.T) {
// P06: a task group task carrying an env overlay.
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"tasks:\n" +
" - name: app\n" +
" command: /usr/bin/httpd\n" +
" env:\n" +
" LOG_LEVEL: debug\n" +
" REGION: us\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if len(spec.Tasks) != 1 {
t.Fatalf("Tasks = %d, want 1", len(spec.Tasks))
}
task := spec.Tasks[0]
if task.Env == nil {
t.Fatal("Tasks[0].Env is nil")
}
if got := task.Env["LOG_LEVEL"]; got != "debug" {
t.Errorf("Env[LOG_LEVEL] = %q, want debug", got)
}
if got := task.Env["REGION"]; got != "us" {
t.Errorf("Env[REGION] = %q, want us", got)
}
}
func TestParseMarkdown_TasksBlockInheritsTopLevelRuntime(t *testing.T) {
// P06: when a task omits its own runtime, the top-level runtime
// is the per-group default. The parser must NOT create a task
// runtime when the task block lacks a `runtime:` sub-block; the
// emitter/validator resolve the default from spec.Runtime.
input := "---\n" +
"kind: Service\n" +
"name: web\n" +
"runtime:\n" +
" one_of: process\n" +
" command: /bin/default\n" +
"tasks:\n" +
" - name: app\n" +
" command: /bin/app\n" +
" - name: sidecar\n" +
" command: /bin/sidecar\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if spec.Runtime == nil || spec.Runtime.OneOf != "process" {
t.Fatalf("top-level runtime not parsed: %+v", spec.Runtime)
}
if len(spec.Tasks) != 2 {
t.Fatalf("Tasks = %d, want 2", len(spec.Tasks))
}
for i, task := range spec.Tasks {
if task.Runtime != nil {
t.Errorf("Tasks[%d].Runtime should be nil (inherit top-level), got %+v", i, task.Runtime)
}
}
if spec.Tasks[0].Name != "app" || spec.Tasks[1].Name != "sidecar" {
t.Errorf("task names = %q, %q", spec.Tasks[0].Name, spec.Tasks[1].Name)
}
}
func TestParseMarkdown_NoTasksBackwardCompat(t *testing.T) {
// Backward compat: a spec with no `tasks:` block parses as a
// single-process alloc; spec.Tasks must be empty/nil.
input := "---\n" +
"kind: Job\n" +
"name: backup\n" +
"runtime:\n" +
" one_of: process\n" +
" command: /bin/rsync\n" +
"---\nbody\n"
spec, err := ParseMarkdown([]byte(input))
if err != nil {
t.Fatalf("ParseMarkdown: %v", err)
}
if len(spec.Tasks) != 0 {
t.Fatalf("Tasks = %d, want 0 (backward compat)", len(spec.Tasks))
}
if spec.Runtime == nil || spec.Runtime.Command != "/bin/rsync" {
t.Errorf("Runtime = %+v, want command=/bin/rsync", spec.Runtime)
}
}
+26 -27
View File
@@ -2,7 +2,6 @@ package jobspec
import (
"fmt"
"os"
"strings"
"github.com/hashicorp/hcl/v2"
@@ -10,16 +9,24 @@ import (
"github.com/hashicorp/hcl/v2/hclsimple"
)
// Spec is the legacy HCL-only jobspec shape. It is retained for the
// v0.9→v0.10 migration window (REQ-090) and is populated by ParseHCLLegacy.
//
// Deprecated: HCL is legacy per R-013; new code should consume the
// unified *WorkloadSpec returned by ParseFile/Dispatch (see
// dispatch.go and markdown.go).
type Spec struct {
Job JobSpec `hcl:"job,block"`
Tasks []TaskSpec `hcl:"task,block"`
}
// JobSpec is the legacy HCL job block.
type JobSpec struct {
Name string `hcl:"name,label"`
Type string `hcl:"type,optional"`
}
// TaskSpec is the legacy HCL task block.
type TaskSpec struct {
Name string `hcl:"name,label"`
Command string `hcl:"command"`
@@ -27,34 +34,14 @@ type TaskSpec struct {
Env []string `hcl:"env,optional"`
}
func ParseFile(path string) (*Spec, error) {
data, err := os.ReadFile(path)
if err != nil {
return nil, fmt.Errorf("read spec file: %w", err)
}
return Parse(data, path)
}
func Parse(data []byte, filename string) (*Spec, error) {
var spec Spec
err := hclsimple.Decode(filename, data, nil, &spec)
if err != nil {
return nil, fmt.Errorf("decode hcl: %w", err)
}
if spec.Job.Name == "" {
return nil, fmt.Errorf("spec missing job name")
}
if len(spec.Tasks) == 0 {
return nil, fmt.Errorf("spec must have at least one task")
}
for i, t := range spec.Tasks {
if t.Command == "" {
return nil, fmt.Errorf("task[%d] (%s) missing command", i, t.Name)
}
}
return &spec, nil
// hclDecode wraps hclsimple.Decode for testability.
func hclDecode(filename string, data []byte, spec *Spec) error {
return hclsimple.Decode(filename, data, nil, spec)
}
// Validate is the legacy HCL Spec validator retained for the migration
// window (REQ-090). New code should use validateWorkload on a
// *WorkloadSpec.
func (s *Spec) Validate() error {
if strings.TrimSpace(s.Job.Name) == "" {
return fmt.Errorf("job name is required")
@@ -65,5 +52,17 @@ func (s *Spec) Validate() error {
return nil
}
// Parse is the original HCL-only entry point retained for backward
// compatibility with direct HCL callers during the v0.9→v0.10 migration
// window (REQ-090). New code should call the dispatcher ParseFile (which
// returns *WorkloadSpec) or ParseHCL (which adapts HCL into
// *WorkloadSpec).
//
// Deprecated: use ParseFile (dispatcher) or ParseHCL (adapter). HCL is
// legacy per R-013.
func Parse(data []byte, filename string) (*Spec, error) {
return ParseHCLLegacy(data, filename)
}
var _ = hcl.Diagnostics{}
var _ = gohcl.DecodeBody
+4 -4
View File
@@ -130,9 +130,9 @@ func TestParse_GoldenFiles(t *testing.T) {
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
path := filepath.Join("testdata", tc.file)
spec, err := ParseFile(path)
spec, err := ParseHCLFile(path)
if err != nil {
t.Fatalf("ParseFile(%s): %v", tc.file, err)
t.Fatalf("ParseHCLFile(%s): %v", tc.file, err)
}
if spec.Job.Name != tc.wantJob {
t.Errorf("job name = %q, want %q", spec.Job.Name, tc.wantJob)
@@ -254,9 +254,9 @@ func TestSpec_Validate(t *testing.T) {
func TestSpec_Validate_RoundTripFromParse(t *testing.T) {
path := filepath.Join("testdata", "valid_single_task.hcl")
spec, err := ParseFile(path)
spec, err := ParseHCLFile(path)
if err != nil {
t.Fatalf("ParseFile: %v", err)
t.Fatalf("ParseHCLFile: %v", err)
}
if err := spec.Validate(); err != nil {
t.Errorf("Validate on parsed spec: %v", err)
+308
View File
@@ -0,0 +1,308 @@
package ns
import (
"fmt"
"os"
"path/filepath"
"sort"
"strings"
)
// ParseNSMd reads an ns.md file, extracts the YAML frontmatter, and
// parses it into a *NSConfig. The body after the closing `---` is
// discarded (namespace declarations do not require body preservation
// like jobspecs do under R-015; we keep the parser minimal and
// consistent with internal/config/markdown.go).
//
// Frontmatter keys (R-014):
//
// kind: Namespace (required; must be "Namespace")
// name: <ns-name> (required)
// parents: ["a", "b"] (optional; default empty)
// inherits_env: true (optional; default true)
// inherits_secrets: true (optional; default true)
// quota: {...} (optional; parsed but not surfaced here)
// acl: {...} (optional; parsed but not surfaced here)
//
// The parser is a minimal hand-rolled YAML-ish key:value reader (no
// new dependencies; gopkg.in/yaml.v3 is intentionally NOT added). It
// supports flat scalar keys and the inline flow-array form
// `["a", "b"]` for `parents`. Nested mappings (quota, acl) are
// recognized as keys but their contents are currently ignored — they
// are reserved for later phases.
func ParseNSMd(path string) (*NSConfig, error) {
data, err := os.ReadFile(path)
if err != nil {
return nil, fmt.Errorf("read %s: %w", path, err)
}
content := string(data)
block, ok := extractFrontmatter(content)
if !ok {
return nil, fmt.Errorf("parse %s: missing frontmatter", path)
}
if strings.TrimSpace(block) == "" {
return nil, fmt.Errorf("parse %s: missing frontmatter", path)
}
cfg, err := parseNSFrontmatter(block, path)
if err != nil {
return nil, err
}
if cfg.Name == "" {
return nil, fmt.Errorf("parse %s: missing name", path)
}
return cfg, nil
}
// extractFrontmatter returns the YAML block between the first pair of
// `---` delimiters and whether a frontmatter block was present.
func extractFrontmatter(content string) (string, bool) {
trimmed := strings.TrimLeft(content, "\r\n\t ")
if !strings.HasPrefix(trimmed, "---") {
return "", false
}
rest := trimmed[3:]
rest = strings.TrimLeft(rest, "\r\n")
idx := strings.Index(rest, "\n---")
if idx < 0 {
return "", false
}
return rest[:idx], true
}
// parseNSFrontmatter parses a minimal YAML-ish frontmatter block into
// a *NSConfig. See ParseNSMd for the supported keys.
func parseNSFrontmatter(block, path string) (*NSConfig, error) {
cfg := &NSConfig{
InheritsEnv: true,
InheritsSecrets: true,
}
kind := ""
lines := strings.Split(block, "\n")
for lineNo, raw := range lines {
line := stripNSComment(raw)
if strings.TrimSpace(line) == "" {
continue
}
if countIndent(line) > 0 {
// Indented line under a nested mapping header (quota, acl).
// Recognized but ignored at this phase.
continue
}
key, val, ok := splitKV(strings.TrimSpace(line))
if !ok {
return nil, fmt.Errorf("parse %s: line %d: malformed key:value", path, lineNo+1)
}
switch key {
case "kind":
kind = strings.TrimSpace(unquote(val))
case "name":
cfg.Name = strings.TrimSpace(unquote(val))
case "parents":
parents, err := parseStringArray(val)
if err != nil {
return nil, fmt.Errorf("parse %s: line %d: parents: %w", path, lineNo+1, err)
}
cfg.Parents = parents
case "inherits_env":
cfg.InheritsEnv = parseBool(val)
case "inherits_secrets":
cfg.InheritsSecrets = parseBool(val)
case "quota", "acl":
// Reserved nested-mapping keys; recognized, contents ignored.
default:
// Unknown keys are ignored (forward-compat with future
// frontmatter additions).
}
}
if kind == "" {
return nil, fmt.Errorf("parse %s: missing kind", path)
}
if kind != "Namespace" {
return nil, fmt.Errorf("parse %s: kind %q is not %q", path, kind, "Namespace")
}
return cfg, nil
}
// parseStringArray parses an inline YAML flow-array of scalars, e.g.
// `["a", "b"]` or `['a', 'b']` or `[a, b]`. Returns an error if the
// value is not a flow-array. Empty array `[]` returns nil.
func parseStringArray(val string) ([]string, error) {
val = strings.TrimSpace(val)
if val == "" {
return nil, nil
}
if !strings.HasPrefix(val, "[") || !strings.HasSuffix(val, "]") {
return nil, fmt.Errorf("expected [..] array, got %q", val)
}
inner := strings.TrimSpace(val[1 : len(val)-1])
if inner == "" {
return nil, nil
}
parts := splitFlowItems(inner)
out := make([]string, 0, len(parts))
for _, p := range parts {
p = strings.TrimSpace(p)
if p == "" {
continue
}
out = append(out, unquote(p))
}
return out, nil
}
// splitFlowItems splits a comma-separated flow-array body, respecting
// single and double quotes.
func splitFlowItems(s string) []string {
var out []string
inSingle := false
inDouble := false
start := 0
for i := 0; i < len(s); i++ {
c := s[i]
switch c {
case '\'':
if !inDouble {
inSingle = !inSingle
}
case '"':
if !inSingle {
inDouble = !inDouble
}
case ',':
if !inSingle && !inDouble {
out = append(out, s[start:i])
start = i + 1
}
}
}
out = append(out, s[start:])
return out
}
// parseBool parses a YAML-ish bool (true/false/yes/no), defaulting to
// true for empty (matches the inherits_* defaults).
func parseBool(val string) bool {
switch strings.ToLower(strings.TrimSpace(unquote(val))) {
case "false", "no", "off", "0":
return false
default:
return true
}
}
// ParseNSMdDir walks `<root>/*/ns.md`, parses each, and returns the
// config map keyed by namespace name. The `cluster` directory is
// skipped (it is not a namespace). The `_defaults` namespace MUST
// exist; if missing, an error is returned.
func ParseNSMdDir(root string) (map[string]*NSConfig, error) {
entries, err := os.ReadDir(root)
if err != nil {
return nil, fmt.Errorf("read namespace root %s: %w", root, err)
}
configs := make(map[string]*NSConfig)
var found []string
for _, ent := range entries {
if !ent.IsDir() {
continue
}
if ent.Name() == "cluster" {
continue
}
nsMd := filepath.Join(root, ent.Name(), "ns.md")
info, err := os.Stat(nsMd)
if err != nil || info.IsDir() {
continue
}
cfg, err := ParseNSMd(nsMd)
if err != nil {
return nil, err
}
// The directory name and the frontmatter `name` should match;
// we key by the frontmatter name (canonical) but also accept
// the directory name if frontmatter name is missing (the
// parser already errors on missing name, so this is defensive).
key := cfg.Name
if key == "" {
key = ent.Name()
}
if _, dup := configs[key]; dup {
return nil, fmt.Errorf("duplicate namespace %q (from %s)", key, nsMd)
}
configs[key] = cfg
found = append(found, key)
}
if _, ok := configs[defaultsName]; !ok {
sort.Strings(found)
names := strings.Join(found, ", ")
if names == "" {
names = "(none)"
}
return nil, fmt.Errorf("namespace root %s: implicit root %q not found (found: %s)", root, defaultsName, names)
}
return configs, nil
}
func countIndent(s string) int {
n := 0
for _, r := range s {
if r == ' ' || r == '\t' {
n++
continue
}
break
}
return n
}
func splitKV(s string) (key, val string, ok bool) {
idx := strings.Index(s, ":")
if idx < 0 {
return "", "", false
}
key = strings.TrimSpace(s[:idx])
val = strings.TrimSpace(s[idx+1:])
if key == "" {
return "", "", false
}
return key, val, true
}
func stripNSComment(s string) string {
inSingle := false
inDouble := false
for i := 0; i < len(s); i++ {
c := s[i]
switch c {
case '\'':
if !inDouble {
inSingle = !inSingle
}
case '"':
if !inSingle {
inDouble = !inDouble
}
case '#':
if !inSingle && !inDouble {
if i == 0 || s[i-1] == ' ' || s[i-1] == '\t' {
return s[:i]
}
}
}
}
return s
}
func unquote(s string) string {
if len(s) >= 2 {
if (s[0] == '"' && s[len(s)-1] == '"') || (s[0] == '\'' && s[len(s)-1] == '\'') {
return s[1 : len(s)-1]
}
}
return s
}
+227
View File
@@ -0,0 +1,227 @@
package ns
import (
"os"
"path/filepath"
"strings"
"testing"
)
func writeNSMd(t *testing.T, path, content string) {
t.Helper()
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
t.Fatalf("mkdir: %v", err)
}
if err := os.WriteFile(path, []byte(content), 0o644); err != nil {
t.Fatalf("write %s: %v", path, err)
}
}
const validNSMd = `---
kind: Namespace
name: prod
parents: ["_defaults"]
inherits_env: true
inherits_secrets: true
quota:
cpu: 4
acl:
admin: ops
---
# Prod namespace
This body is ignored.
`
func TestParseNSMdValid(t *testing.T) {
tmp := t.TempDir()
path := filepath.Join(tmp, "ns.md")
writeNSMd(t, path, validNSMd)
cfg, err := ParseNSMd(path)
if err != nil {
t.Fatalf("ParseNSMd: %v", err)
}
if cfg.Name != "prod" {
t.Errorf("name = %q, want prod", cfg.Name)
}
if !eqSlice(cfg.Parents, []string{"_defaults"}) {
t.Errorf("parents = %v, want [_defaults]", cfg.Parents)
}
if !cfg.InheritsEnv || !cfg.InheritsSecrets {
t.Errorf("inherits_env=%v inherits_secrets=%v, want both true", cfg.InheritsEnv, cfg.InheritsSecrets)
}
}
func TestParseNSMdMissingFrontmatter(t *testing.T) {
tmp := t.TempDir()
path := filepath.Join(tmp, "ns.md")
writeNSMd(t, path, "# just a body, no frontmatter\n")
_, err := ParseNSMd(path)
if err == nil {
t.Fatal("expected missing frontmatter error, got nil")
}
if !strings.Contains(err.Error(), "missing frontmatter") {
t.Errorf("error = %q, want contains 'missing frontmatter'", err.Error())
}
}
func TestParseNSMdEmptyFrontmatter(t *testing.T) {
tmp := t.TempDir()
path := filepath.Join(tmp, "ns.md")
writeNSMd(t, path, "---\n---\nbody\n")
_, err := ParseNSMd(path)
if err == nil {
t.Fatal("expected error for empty frontmatter, got nil")
}
}
func TestParseNSMdWrongKind(t *testing.T) {
tmp := t.TempDir()
path := filepath.Join(tmp, "ns.md")
writeNSMd(t, path, "---\nkind: Job\nname: x\n---\n")
_, err := ParseNSMd(path)
if err == nil {
t.Fatal("expected wrong-kind error, got nil")
}
if !strings.Contains(err.Error(), "not \"Namespace\"") {
t.Errorf("error = %q, want contains 'is not \"Namespace\"'", err.Error())
}
}
func TestParseNSMdMissingKind(t *testing.T) {
tmp := t.TempDir()
path := filepath.Join(tmp, "ns.md")
writeNSMd(t, path, "---\nname: x\n---\n")
_, err := ParseNSMd(path)
if err == nil {
t.Fatal("expected missing kind error, got nil")
}
if !strings.Contains(err.Error(), "missing kind") {
t.Errorf("error = %q, want contains 'missing kind'", err.Error())
}
}
func TestParseNSMdMissingName(t *testing.T) {
tmp := t.TempDir()
path := filepath.Join(tmp, "ns.md")
writeNSMd(t, path, "---\nkind: Namespace\n---\n")
_, err := ParseNSMd(path)
if err == nil {
t.Fatal("expected missing name error, got nil")
}
if !strings.Contains(err.Error(), "missing name") {
t.Errorf("error = %q, want contains 'missing name'", err.Error())
}
}
func TestParseNSMdParentsUnquoted(t *testing.T) {
tmp := t.TempDir()
path := filepath.Join(tmp, "ns.md")
writeNSMd(t, path, "---\nkind: Namespace\nname: x\nparents: [a, b]\n---\n")
cfg, err := ParseNSMd(path)
if err != nil {
t.Fatalf("ParseNSMd: %v", err)
}
if !eqSlice(cfg.Parents, []string{"a", "b"}) {
t.Errorf("parents = %v, want [a b]", cfg.Parents)
}
}
func TestParseNSMdParentsEmpty(t *testing.T) {
tmp := t.TempDir()
path := filepath.Join(tmp, "ns.md")
writeNSMd(t, path, "---\nkind: Namespace\nname: x\nparents: []\n---\n")
cfg, err := ParseNSMd(path)
if err != nil {
t.Fatalf("ParseNSMd: %v", err)
}
if len(cfg.Parents) != 0 {
t.Errorf("parents = %v, want empty", cfg.Parents)
}
}
func TestParseNSMdInheritsFalse(t *testing.T) {
tmp := t.TempDir()
path := filepath.Join(tmp, "ns.md")
writeNSMd(t, path, "---\nkind: Namespace\nname: x\ninherits_env: false\ninherits_secrets: no\n---\n")
cfg, err := ParseNSMd(path)
if err != nil {
t.Fatalf("ParseNSMd: %v", err)
}
if cfg.InheritsEnv {
t.Errorf("inherits_env should be false")
}
if cfg.InheritsSecrets {
t.Errorf("inherits_secrets should be false")
}
}
func TestParseNSMdMissingFile(t *testing.T) {
_, err := ParseNSMd(filepath.Join(t.TempDir(), "nope.md"))
if err == nil {
t.Fatal("expected error for missing file")
}
}
func TestParseNSMdDirHappy(t *testing.T) {
root := t.TempDir()
writeNSMd(t, filepath.Join(root, "_defaults", "ns.md"), "---\nkind: Namespace\nname: _defaults\n---\n")
writeNSMd(t, filepath.Join(root, "prod", "ns.md"), validNSMd)
cfgs, err := ParseNSMdDir(root)
if err != nil {
t.Fatalf("ParseNSMdDir: %v", err)
}
if _, ok := cfgs["_defaults"]; !ok {
t.Errorf("missing _defaults in %v", cfgs)
}
if _, ok := cfgs["prod"]; !ok {
t.Errorf("missing prod in %v", cfgs)
}
}
func TestParseNSMdDirMissingDefaults(t *testing.T) {
root := t.TempDir()
writeNSMd(t, filepath.Join(root, "prod", "ns.md"), validNSMd)
_, err := ParseNSMdDir(root)
if err == nil {
t.Fatal("expected missing _defaults error, got nil")
}
if !strings.Contains(err.Error(), "_defaults") {
t.Errorf("error = %q, want contains _defaults", err.Error())
}
}
func TestParseNSMdDirSkipsCluster(t *testing.T) {
root := t.TempDir()
writeNSMd(t, filepath.Join(root, "_defaults", "ns.md"), "---\nkind: Namespace\nname: _defaults\n---\n")
// cluster/ contains a ns.md-shaped file but must be skipped.
writeNSMd(t, filepath.Join(root, "cluster", "ns.md"), "---\nkind: Namespace\nname: cluster\n---\n")
cfgs, err := ParseNSMdDir(root)
if err != nil {
t.Fatalf("ParseNSMdDir: %v", err)
}
if _, ok := cfgs["cluster"]; ok {
t.Errorf("cluster should be skipped, present in %v", cfgs)
}
}
func TestParseNSMdDirNoFiles(t *testing.T) {
root := t.TempDir()
_, err := ParseNSMdDir(root)
if err == nil {
t.Fatal("expected missing _defaults error on empty dir, got nil")
}
}
func TestParseNSMdDirNotADir(t *testing.T) {
tmp := t.TempDir()
// Create a file with the same name as the expected root dir.
root := filepath.Join(tmp, "notadir")
writeNSMd(t, root, "x")
_, err := ParseNSMdDir(root)
if err == nil {
t.Fatal("expected error for non-dir root")
}
}
+252
View File
@@ -0,0 +1,252 @@
// Package ns implements the namespace inheritance resolver (REQ-082)
// and the ns.md frontmatter parser used by `orca ns` CLI subcommands.
//
// The resolver is a PURE function (no I/O): it takes a map of parsed
// namespace configs keyed by name and returns a map of resolved
// namespaces with merged env and unioned constraints. The inheritance
// model is:
//
// - Each namespace declares zero or more parents in `ns.md`
// frontmatter (`parents: ["ns1", "ns2"]`).
// - The implicit root namespace `_defaults` (R-002 D-159) always
// exists and has no parents; it is ALWAYS appended as the last
// element of the chain (D-185).
// - Opting out of `_defaults` is impossible (D-187): even with
// `parents: []`, `_defaults` still appears at the end of the chain.
// - Merge semantics: child overrides parent for scalars (env keys);
// arrays union (child constraints add to parent constraints, with
// duplicates removed, order: most-specific first).
// - The chain order is most-specific first, `_defaults` last.
// - `_defaults` may be listed explicitly in `parents`; the explicit
// listing is de-duped silently (still appears once, at the end).
// - Misordering (`parents: ["_defaults", "x"]`) is rejected: an
// explicit `_defaults` entry must be the only entry (or omitted).
// - Cycle detection uses DFS with a visited set; a cycle returns an
// error with the cycle path.
// - Missing parents return "parent X not found".
package ns
import (
"fmt"
"sort"
)
const defaultsName = "_defaults"
// NSConfig is a parsed namespace declaration from ns.md frontmatter.
// The resolver consumes this; the parser populates it.
type NSConfig struct {
Name string
Parents []string
Env map[string]string
Constraints []string
InheritsEnv bool
InheritsSecrets bool
}
// ResolvedNS is the output of the resolver: the namespace with its
// fully-merged env and unioned constraints, plus the ordered
// inheritance chain (most-specific first, `_defaults` last).
type ResolvedNS struct {
Name string
Chain []string
Env map[string]string
Constraints []string
}
// Resolve walks the parent chain for each namespace, merges env (child
// wins scalars), unions constraints (child adds to parent, de-duped),
// and detects cycles. It is PURE (no I/O). The empty-configs case
// returns an empty map and no error.
//
// The `_defaults` namespace is ALWAYS the last element of every chain
// (D-185); opting out is impossible (D-187). An explicit `_defaults`
// entry in `parents` is de-duped silently. Misordering (e.g.
// `parents: ["_defaults", "x"]`) is rejected.
func Resolve(configs map[string]*NSConfig) (map[string]*ResolvedNS, error) {
if len(configs) == 0 {
return map[string]*ResolvedNS{}, nil
}
// Validate each config's parents reference exists and the
// _defaults entry (if explicit) is the only entry.
for name, cfg := range configs {
if cfg == nil {
return nil, fmt.Errorf("namespace %q has nil config", name)
}
for _, p := range cfg.Parents {
if p == defaultsName {
// Explicit _defaults must be the only parent.
if len(cfg.Parents) != 1 {
return nil, fmt.Errorf("namespace %q: %s must be the only parent if listed explicitly (misordering rejected)", name, defaultsName)
}
continue
}
if _, ok := configs[p]; !ok {
return nil, fmt.Errorf("namespace %q: parent %q not found", name, p)
}
}
}
// `_defaults` must be present in the configs map (the parser
// enforces this for ParseNSMdDir; Resolve trusts its input but
// still requires _defaults to exist for chain assembly).
if _, ok := configs[defaultsName]; !ok {
return nil, fmt.Errorf("namespace %q not found (implicit root must be present)", defaultsName)
}
resolved := make(map[string]*ResolvedNS, len(configs))
// Resolve in deterministic order for stable error reporting.
names := make([]string, 0, len(configs))
for n := range configs {
names = append(names, n)
}
sort.Strings(names)
for _, name := range names {
r, err := resolveOne(configs, name)
if err != nil {
return nil, err
}
resolved[name] = r
}
return resolved, nil
}
// resolveOne resolves a single namespace. The chain is built by walking
// parents depth-first in POST-order (least-specific first), then
// reversing so the returned chain is most-specific first with
// `_defaults` last (D-185). Cycle detection uses a visiting set.
func resolveOne(configs map[string]*NSConfig, name string) (*ResolvedNS, error) {
post, err := buildChain(configs, name)
if err != nil {
return nil, err
}
// post is least-specific first; reverse to most-specific first.
reverseStrings(post)
chain := post
// Env: child (most-specific) wins. Walk least-specific to
// most-specific (end -> beginning) so later writes override.
env := make(map[string]string)
for i := len(chain) - 1; i >= 0; i-- {
c := configs[chain[i]]
if c == nil {
continue
}
for k, v := range c.Env {
env[k] = v
}
}
// Constraints: union, child (most-specific) first. Walk the chain
// front-to-back (most-specific first) and append unseen items.
constraintsSeen := make(map[string]bool)
var constraints []string
for _, ns := range chain {
c := configs[ns]
if c == nil {
continue
}
for _, con := range c.Constraints {
if !constraintsSeen[con] {
constraintsSeen[con] = true
constraints = append(constraints, con)
}
}
}
return &ResolvedNS{
Name: name,
Chain: chain,
Env: env,
Constraints: constraints,
}, nil
}
// buildChain walks parents depth-first and returns the chain in
// POST-order (least-specific first, `_defaults` first). The caller
// reverses to get most-specific first. Cycle detection uses the
// visiting set: a node currently being walked indicates a back-edge.
func buildChain(configs map[string]*NSConfig, name string) ([]string, error) {
var post []string
seen := make(map[string]bool) // final chain membership (de-dup)
visiting := make(map[string]bool)
if err := dfsChain(configs, name, &post, seen, visiting); err != nil {
return nil, err
}
// `_defaults` is the implicit root: it must be the FIRST element
// in post-order (so it ends up LAST after reversal). If it was not
// reached via parents (no explicit listing and no chain leads to
// it), prepend it.
if !seen[defaultsName] {
post = append([]string{defaultsName}, post...)
seen[defaultsName] = true
}
return post, nil
}
// dfsChain appends each node AFTER its parents (post-order), producing
// least-specific first. Cycle detection uses the visiting set.
func dfsChain(configs map[string]*NSConfig, name string, post *[]string, seen, visiting map[string]bool) error {
if visiting[name] {
return fmt.Errorf("cycle detected: %s", cyclePath(visiting, configs, name))
}
if seen[name] {
return nil
}
visiting[name] = true
cfg := configs[name]
if cfg != nil {
for _, p := range cfg.Parents {
if err := dfsChain(configs, p, post, seen, visiting); err != nil {
return err
}
}
}
delete(visiting, name)
seen[name] = true
*post = append(*post, name)
return nil
}
// cyclePath reconstructs a readable cycle path from the visiting set.
// Since visiting is a set (not ordered), we reconstruct by re-walking
// parents from the offending node until we revisit it.
func cyclePath(visiting map[string]bool, configs map[string]*NSConfig, start string) string {
// Walk parents from start, collecting names until we hit start
// again or run out.
var path []string
cur := start
for i := 0; i < len(visiting)+1; i++ {
path = append(path, cur)
cfg := configs[cur]
if cfg == nil || len(cfg.Parents) == 0 {
break
}
next := cfg.Parents[0]
if next == start {
path = append(path, next)
break
}
cur = next
}
return joinArrows(path)
}
func joinArrows(parts []string) string {
out := ""
for i, p := range parts {
if i > 0 {
out += " -> "
}
out += p
}
return out
}
func reverseStrings(s []string) {
for i, j := 0, len(s)-1; i < j; i, j = i+1, j-1 {
s[i], s[j] = s[j], s[i]
}
}
+231
View File
@@ -0,0 +1,231 @@
package ns
import (
"strings"
"testing"
)
func TestResolveEmptyConfigs(t *testing.T) {
out, err := Resolve(map[string]*NSConfig{})
if err != nil {
t.Fatalf("Resolve empty: unexpected error: %v", err)
}
if len(out) != 0 {
t.Fatalf("Resolve empty: want empty map, got %d entries", len(out))
}
}
func TestResolveSingleNoParents(t *testing.T) {
cfgs := map[string]*NSConfig{
defaultsName: {Name: defaultsName, Env: map[string]string{"A": "1"}},
"x": {Name: "x", Env: map[string]string{"B": "2"}},
}
out, err := Resolve(cfgs)
if err != nil {
t.Fatalf("Resolve: %v", err)
}
r := out["x"]
if r == nil {
t.Fatal("missing resolved x")
}
if !eqSlice(r.Chain, []string{"x", defaultsName}) {
t.Errorf("chain = %v, want [x _defaults]", r.Chain)
}
if r.Env["A"] != "1" || r.Env["B"] != "2" {
t.Errorf("env = %v, want A=1 B=2", r.Env)
}
}
func TestResolveChildOverridesParentScalar(t *testing.T) {
cfgs := map[string]*NSConfig{
defaultsName: {Name: defaultsName, Env: map[string]string{"K": "parent"}},
"child": {Name: "child", Parents: []string{defaultsName}, Env: map[string]string{"K": "child"}},
}
out, err := Resolve(cfgs)
if err != nil {
t.Fatalf("Resolve: %v", err)
}
if got := out["child"].Env["K"]; got != "child" {
t.Errorf("child K = %q, want %q (child overrides parent)", got, "child")
}
}
func TestResolveArraysUnion(t *testing.T) {
cfgs := map[string]*NSConfig{
defaultsName: {Name: defaultsName, Constraints: []string{"a", "b"}},
"x": {Name: "x", Parents: []string{defaultsName}, Constraints: []string{"c", "a"}},
}
out, err := Resolve(cfgs)
if err != nil {
t.Fatalf("Resolve: %v", err)
}
c := out["x"].Constraints
// Union de-duped; most-specific (x) first.
if !eqSlice(c, []string{"c", "a", "b"}) {
t.Errorf("constraints = %v, want [c a b]", c)
}
}
func TestResolveDefaultsImplicitLast(t *testing.T) {
cfgs := map[string]*NSConfig{
defaultsName: {Name: defaultsName},
"mid": {Name: "mid", Parents: []string{defaultsName}},
"top": {Name: "top", Parents: []string{"mid"}},
}
out, err := Resolve(cfgs)
if err != nil {
t.Fatalf("Resolve: %v", err)
}
if !eqSlice(out["top"].Chain, []string{"top", "mid", defaultsName}) {
t.Errorf("top chain = %v, want [top mid _defaults]", out["top"].Chain)
}
if !eqSlice(out["mid"].Chain, []string{"mid", defaultsName}) {
t.Errorf("mid chain = %v, want [mid _defaults]", out["mid"].Chain)
}
}
func TestResolveDefaultsDedupExplicit(t *testing.T) {
cfgs := map[string]*NSConfig{
defaultsName: {Name: defaultsName, Env: map[string]string{"D": "1"}},
"x": {Name: "x", Parents: []string{defaultsName}},
}
out, err := Resolve(cfgs)
if err != nil {
t.Fatalf("Resolve: %v", err)
}
// _defaults appears exactly once.
count := 0
for _, c := range out["x"].Chain {
if c == defaultsName {
count++
}
}
if count != 1 {
t.Errorf("_defaults appears %d times in chain %v, want 1", count, out["x"].Chain)
}
}
func TestResolveMisorderingRejected(t *testing.T) {
cfgs := map[string]*NSConfig{
defaultsName: {Name: defaultsName},
"x": {Name: "x"},
"y": {Name: "y", Parents: []string{defaultsName, "x"}},
}
_, err := Resolve(cfgs)
if err == nil {
t.Fatal("expected misordering error, got nil")
}
if !strings.Contains(err.Error(), "must be the only parent") {
t.Errorf("error = %q, want misordering message", err.Error())
}
}
func TestResolveOptOutImpossible(t *testing.T) {
cfgs := map[string]*NSConfig{
defaultsName: {Name: defaultsName, Env: map[string]string{"ROOT": "1"}},
"x": {Name: "x", Parents: nil},
}
out, err := Resolve(cfgs)
if err != nil {
t.Fatalf("Resolve: %v", err)
}
r := out["x"]
last := r.Chain[len(r.Chain)-1]
if last != defaultsName {
t.Errorf("last chain element = %q, want %q (opt-out impossible)", last, defaultsName)
}
if r.Env["ROOT"] != "1" {
t.Errorf("env should inherit from _defaults: ROOT=%q", r.Env["ROOT"])
}
}
func TestResolveCycleDetection(t *testing.T) {
cfgs := map[string]*NSConfig{
defaultsName: {Name: defaultsName},
"a": {Name: "a", Parents: []string{"b"}},
"b": {Name: "b", Parents: []string{"a"}},
}
_, err := Resolve(cfgs)
if err == nil {
t.Fatal("expected cycle error, got nil")
}
if !strings.Contains(err.Error(), "cycle") {
t.Errorf("error = %q, want cycle message", err.Error())
}
}
func TestResolveMissingParent(t *testing.T) {
cfgs := map[string]*NSConfig{
defaultsName: {Name: defaultsName},
"a": {Name: "a", Parents: []string{"ghost"}},
}
_, err := Resolve(cfgs)
if err == nil {
t.Fatal("expected missing-parent error, got nil")
}
if !strings.Contains(err.Error(), "ghost") || !strings.Contains(err.Error(), "not found") {
t.Errorf("error = %q, want contains 'ghost' and 'not found'", err.Error())
}
}
func TestResolveMissingDefaults(t *testing.T) {
cfgs := map[string]*NSConfig{
"x": {Name: "x"},
}
_, err := Resolve(cfgs)
if err == nil {
t.Fatal("expected missing _defaults error, got nil")
}
if !strings.Contains(err.Error(), defaultsName) {
t.Errorf("error = %q, want contains %q", err.Error(), defaultsName)
}
}
func TestResolveChainOrderWithDiamond(t *testing.T) {
// Diamond: top -> {left, right} -> base; base -> _defaults.
cfgs := map[string]*NSConfig{
defaultsName: {Name: defaultsName, Env: map[string]string{"R": "r"}},
"base": {Name: "base", Parents: []string{defaultsName}, Env: map[string]string{"B": "b"}},
"left": {Name: "left", Parents: []string{"base"}, Env: map[string]string{"L": "l"}},
"right": {Name: "right", Parents: []string{"base"}, Env: map[string]string{"L": "r"}},
"top": {Name: "top", Parents: []string{"left", "right"}, Env: map[string]string{"T": "t"}},
}
out, err := Resolve(cfgs)
if err != nil {
t.Fatalf("Resolve: %v", err)
}
r := out["top"]
if r == nil {
t.Fatal("missing top")
}
// top first, _defaults last.
if r.Chain[0] != "top" || r.Chain[len(r.Chain)-1] != defaultsName {
t.Errorf("chain = %v, want top first and _defaults last", r.Chain)
}
// base appears exactly once (diamond de-duped).
count := 0
for _, c := range r.Chain {
if c == "base" {
count++
}
}
if count != 1 {
t.Errorf("base appears %d times in %v, want 1", count, r.Chain)
}
// top inherits R from _defaults.
if r.Env["R"] != "r" {
t.Errorf("top should inherit R=r, got %q", r.Env["R"])
}
}
func eqSlice(a, b []string) bool {
if len(a) != len(b) {
return false
}
for i := range a {
if a[i] != b[i] {
return false
}
}
return true
}
+121
View File
@@ -0,0 +1,121 @@
// Package paths resolves on-disk locations for the v0.9 multi-namespace
// filesystem layout (R-002). It is the canonical source of truth for
// cluster-wide, per-namespace, and CLI-cache paths.
//
// The v0.8 internal/certpaths package is preserved as a thin shim that
// returns the legacy flat-layout paths during the v0.9 dual-write window
// (REQ-090). New code should use internal/paths, NOT certpaths.
package paths
import (
"os"
"path/filepath"
)
const (
defaultHomeSubdir = ".orca"
clusterDirName = "cluster"
defaultNamespace = "_defaults"
)
// Root returns the ORCA home directory. It honors $ORCA_HOME for
// testability; otherwise it defaults to ~/.orca. An empty $ORCA_HOME is
// treated as unset.
func Root() string {
if p := os.Getenv("ORCA_HOME"); p != "" {
return p
}
home, _ := os.UserHomeDir()
return filepath.Join(home, defaultHomeSubdir)
}
// ClusterDir returns the cluster-wide directory: Root()/cluster.
// Cluster-wide artifacts (CA, master key, peers, txns, known_hosts, SSH
// keys) live here and are NOT scoped to a workload namespace (R-002).
func ClusterDir() string { return filepath.Join(Root(), clusterDirName) }
// NamespaceDir returns the directory for a namespace: Root()/<ns>.
// Use DefaultNamespace() for the implicit root namespace (R-002 D-159).
func NamespaceDir(ns string) string { return filepath.Join(Root(), ns) }
// NSDb returns the SQLite database path for a namespace:
// NamespaceDir(ns)/db/orca.db.
func NSDb(ns string) string { return filepath.Join(NamespaceDir(ns), "db", "orca.db") }
// NSEnv returns the .env path for a namespace: NamespaceDir(ns)/.env.
func NSEnv(ns string) string { return filepath.Join(NamespaceDir(ns), ".env") }
// NSSecrets returns the encrypted secrets env path for a namespace:
// NamespaceDir(ns)/.env.secrets.
func NSSecrets(ns string) string { return filepath.Join(NamespaceDir(ns), ".env.secrets") }
// NSJobs returns the jobs directory for a namespace: NamespaceDir(ns)/jobs.
func NSJobs(ns string) string { return filepath.Join(NamespaceDir(ns), "jobs") }
// NSAlloc returns the allocation directory for a namespace:
// NamespaceDir(ns)/alloc.
func NSAlloc(ns string) string { return filepath.Join(NamespaceDir(ns), "alloc") }
// NSMd returns the namespace Markdown doc path (R-014):
// NamespaceDir(ns)/ns.md.
func NSMd(ns string) string { return filepath.Join(NamespaceDir(ns), "ns.md") }
// DefaultNamespace returns the implicit root namespace name (R-002 D-159).
func DefaultNamespace() string { return defaultNamespace }
// CACertPath returns the v0.9 cluster CA cert path:
// ClusterDir()/ca.crt (D-101). The v0.8 internal CA still writes to
// Root()/ca.crt; the move happens in v0.10-P14.
func CACertPath() string { return filepath.Join(ClusterDir(), "ca.crt") }
// CAKeyPath returns the v0.9 cluster CA key path:
// ClusterDir()/ca.key.
func CAKeyPath() string { return filepath.Join(ClusterDir(), "ca.key") }
// MasterKeyPath returns the AES-256-GCM root master key path
// (R-011, mode 0600): ClusterDir()/master.key. Not generated until
// v0.10-P03.
func MasterKeyPath() string { return filepath.Join(ClusterDir(), "master.key") }
// CacheDB returns the CLI-side cache database path (R-008):
// Root()/orca_cache.db. Not created until v0.9-P0a2.
func CacheDB() string { return filepath.Join(Root(), "orca_cache.db") }
// TxnDir returns the cluster transaction log directory (R-016):
// ClusterDir()/txns.
func TxnDir() string { return filepath.Join(ClusterDir(), "txns") }
// PeersDir returns the cluster peers directory: ClusterDir()/peers.
func PeersDir() string { return filepath.Join(ClusterDir(), "peers") }
// PeerDir returns the directory for a single peer host:
// PeersDir()/host.
func PeerDir(host string) string { return filepath.Join(PeersDir(), host) }
// KnownHostsPath returns the SSH known_hosts path (D-035):
// ClusterDir()/known_hosts.
func KnownHostsPath() string { return filepath.Join(ClusterDir(), "known_hosts") }
// SSHKeyPath returns the orca SSH private key path:
// ClusterDir()/orca_ssh_key (D-037).
func SSHKeyPath() string { return filepath.Join(ClusterDir(), "orca_ssh_key") }
// SSHPubPath returns the orca SSH public key path:
// ClusterDir()/orca_ssh_key.pub.
func SSHPubPath() string { return filepath.Join(ClusterDir(), "orca_ssh_key.pub") }
// ServerCertPath returns the legacy server cert path (legacy compat):
// ClusterDir()/server.crt. step-ca will replace this in a later phase.
func ServerCertPath() string { return filepath.Join(ClusterDir(), "server.crt") }
// ServerKeyPath returns the legacy server key path (legacy compat):
// ClusterDir()/server.key. step-ca will replace this in a later phase.
func ServerKeyPath() string { return filepath.Join(ClusterDir(), "server.key") }
// ConfigPath returns the new Markdown-frontmatter config path (R-014):
// ClusterDir()/config.md. The legacy HCL path is ClusterDir()/config.hcl.
func ConfigPath() string { return filepath.Join(ClusterDir(), "config.md") }
// LegacyHCLConfigPath returns the legacy HCL config path:
// ClusterDir()/config.hcl.
func LegacyHCLConfigPath() string { return filepath.Join(ClusterDir(), "config.hcl") }
+209
View File
@@ -0,0 +1,209 @@
package paths
import (
"os"
"path/filepath"
"strings"
"testing"
)
func TestRoot_HonorsORCAHOME(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
if got, want := Root(), dir; got != want {
t.Errorf("Root() = %q, want %q", got, want)
}
}
func TestRoot_EmptyORCAHOMEFallsBack(t *testing.T) {
t.Setenv("ORCA_HOME", "")
home, err := os.UserHomeDir()
if err != nil {
t.Skipf("os.UserHomeDir: %v", err)
}
want := filepath.Join(home, defaultHomeSubdir)
if got := Root(); got != want {
t.Errorf("Root() with empty ORCA_HOME = %q, want %q", got, want)
}
}
func TestRoot_UnsetORCAHOMEFallsBack(t *testing.T) {
os.Unsetenv("ORCA_HOME")
home, err := os.UserHomeDir()
if err != nil {
t.Skipf("os.UserHomeDir: %v", err)
}
want := filepath.Join(home, defaultHomeSubdir)
got := Root()
if got != want {
t.Errorf("Root() default = %q, want %q", got, want)
}
if !strings.HasPrefix(got, home) {
t.Errorf("Root() default %q does not start with home %q", got, home)
}
}
func TestRoot_RelativeORCAHOME(t *testing.T) {
t.Setenv("ORCA_HOME", "relative/orca/home")
if got, want := Root(), "relative/orca/home"; got != want {
t.Errorf("Root() relative = %q, want %q", got, want)
}
}
func TestDefaultNamespace(t *testing.T) {
if got, want := DefaultNamespace(), "_defaults"; got != want {
t.Errorf("DefaultNamespace() = %q, want %q", got, want)
}
}
func TestClusterDir(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
got := ClusterDir()
want := filepath.Join(dir, "cluster")
if got != want {
t.Errorf("ClusterDir() = %q, want %q", got, want)
}
if !strings.HasPrefix(got, Root()+string(filepath.Separator)) {
t.Errorf("ClusterDir() %q not under Root() %q", got, Root())
}
}
func TestNamespaceDir(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
ns := "prod"
got := NamespaceDir(ns)
want := filepath.Join(dir, ns)
if got != want {
t.Errorf("NamespaceDir(%q) = %q, want %q", ns, got, want)
}
if !strings.HasPrefix(got, Root()+string(filepath.Separator)) {
t.Errorf("NamespaceDir() %q not under Root() %q", got, Root())
}
}
func TestNamespacePaths(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
ns := "prod"
nsDir := NamespaceDir(ns)
cases := []struct {
name string
got string
want string
}{
{"NSDb", NSDb(ns), filepath.Join(nsDir, "db", "orca.db")},
{"NSEnv", NSEnv(ns), filepath.Join(nsDir, ".env")},
{"NSSecrets", NSSecrets(ns), filepath.Join(nsDir, ".env.secrets")},
{"NSJobs", NSJobs(ns), filepath.Join(nsDir, "jobs")},
{"NSAlloc", NSAlloc(ns), filepath.Join(nsDir, "alloc")},
{"NSMd", NSMd(ns), filepath.Join(nsDir, "ns.md")},
}
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
if tc.got != tc.want {
t.Errorf("%s(%q) = %q, want %q", tc.name, ns, tc.got, tc.want)
}
if !strings.HasPrefix(tc.got, nsDir+string(filepath.Separator)) {
t.Errorf("%s() %q not under NamespaceDir() %q", tc.name, tc.got, nsDir)
}
})
}
}
func TestDefaultNamespacePaths(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
ns := DefaultNamespace()
nsDir := NamespaceDir(ns)
if got, want := NSDb(ns), filepath.Join(nsDir, "db", "orca.db"); got != want {
t.Errorf("NSDb(_defaults) = %q, want %q", got, want)
}
if got, want := NSEnv(ns), filepath.Join(nsDir, ".env"); got != want {
t.Errorf("NSEnv(_defaults) = %q, want %q", got, want)
}
}
func TestClusterPaths(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
cDir := ClusterDir()
cases := []struct {
name string
got string
want string
}{
{"CACertPath", CACertPath(), filepath.Join(cDir, "ca.crt")},
{"CAKeyPath", CAKeyPath(), filepath.Join(cDir, "ca.key")},
{"MasterKeyPath", MasterKeyPath(), filepath.Join(cDir, "master.key")},
{"KnownHostsPath", KnownHostsPath(), filepath.Join(cDir, "known_hosts")},
{"SSHKeyPath", SSHKeyPath(), filepath.Join(cDir, "orca_ssh_key")},
{"SSHPubPath", SSHPubPath(), filepath.Join(cDir, "orca_ssh_key.pub")},
{"ServerCertPath", ServerCertPath(), filepath.Join(cDir, "server.crt")},
{"ServerKeyPath", ServerKeyPath(), filepath.Join(cDir, "server.key")},
{"ConfigPath", ConfigPath(), filepath.Join(cDir, "config.md")},
{"LegacyHCLConfigPath", LegacyHCLConfigPath(), filepath.Join(cDir, "config.hcl")},
{"TxnDir", TxnDir(), filepath.Join(cDir, "txns")},
{"PeersDir", PeersDir(), filepath.Join(cDir, "peers")},
}
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
if tc.got != tc.want {
t.Errorf("%s() = %q, want %q", tc.name, tc.got, tc.want)
}
if !strings.HasPrefix(tc.got, cDir+string(filepath.Separator)) {
t.Errorf("%s() %q not under ClusterDir() %q", tc.name, tc.got, cDir)
}
})
}
}
func TestPeerDir(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
host := "node1.example.com"
got := PeerDir(host)
want := filepath.Join(PeersDir(), host)
if got != want {
t.Errorf("PeerDir(%q) = %q, want %q", host, got, want)
}
if !strings.HasPrefix(got, PeersDir()+string(filepath.Separator)) {
t.Errorf("PeerDir() %q not under PeersDir() %q", got, PeersDir())
}
}
func TestCacheDB(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
got := CacheDB()
want := filepath.Join(dir, "orca_cache.db")
if got != want {
t.Errorf("CacheDB() = %q, want %q", got, want)
}
if !strings.HasPrefix(got, Root()+string(filepath.Separator)) {
t.Errorf("CacheDB() %q not under Root() %q", got, Root())
}
}
func TestPathSeparatorsOSAppropriate(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
// Every returned path must use the OS separator (filepath.Join).
sep := string(filepath.Separator)
for _, p := range []string{
ClusterDir(), NamespaceDir("ns"), NSDb("ns"), NSEnv("ns"),
NSSecrets("ns"), NSJobs("ns"), NSAlloc("ns"), NSMd("ns"),
CACertPath(), CAKeyPath(), MasterKeyPath(), CacheDB(),
TxnDir(), PeersDir(), PeerDir("h"), KnownHostsPath(),
SSHKeyPath(), SSHPubPath(), ServerCertPath(), ServerKeyPath(),
ConfigPath(), LegacyHCLConfigPath(),
} {
if !strings.Contains(p, sep) {
t.Errorf("path %q lacks OS separator %q (not joined?)", p, sep)
}
}
}
+10
View File
@@ -289,6 +289,11 @@ func TOFUHostKeyCallback(addr string, capturedKey *ssh.PublicKey) (ssh.HostKeyCa
if errors.As(err, &keyErr) && len(keyErr.Want) == 0 {
line := knownhosts.Line([]string{knownhosts.Normalize(addr)}, key)
path := certpaths.KnownHostsPath()
release, lockErr := security.Flock(path)
if lockErr != nil {
return fmt.Errorf("tofu lock known_hosts: %w", lockErr)
}
defer release()
existing, readErr := os.ReadFile(path)
if readErr != nil && !os.IsNotExist(readErr) {
return fmt.Errorf("tofu read known_hosts: %w", readErr)
@@ -481,6 +486,11 @@ func ResetHostKey(host string) error {
return fmt.Errorf("ResetHostKey: host is required")
}
path := certpaths.KnownHostsPath()
release, lockErr := security.Flock(path)
if lockErr != nil {
return fmt.Errorf("ResetHostKey: lock known_hosts: %w", lockErr)
}
defer release()
existing, err := os.ReadFile(path)
if err != nil {
if os.IsNotExist(err) {
+43
View File
@@ -0,0 +1,43 @@
# C-01 Grill Gate Evaluation — wasmtime / CGO
**Gate**: C-01 (wasmtime/CGO evaluation), gating P07b.
**Question**: The wasmtime Go binding
(`github.com/bytecodealliance/wasmtime-go`) is CGO-based. Does adopting
it revoke D-002 (modernc/sqlite CGO-free cross-compile story)?
## Evaluation
| Option | CGO required? | Cross-compile impact | Decision |
|--------|---------------|----------------------|----------|
| A. Use `bytecodealliance/wasmtime-go` (Go binding) | **YES** — the binding links libwasmtime via cgo | Revokes D-002 — Go cross-compile (`GOOS=linux GOARCH=arm64 go build`) breaks; CGO toolchain needed on every build host; static-binary story lost | REJECTED |
| B. Use the `wasmtime` CLI (apt-installed on the peer) via SSH exec | **NO** — pure Go code, shells out to a CLI over SSH (same pattern as podman/qm/pct) | None — D-002 preserved | **ACCEPTED** |
| C. Use an alternative pure-Go WASM runtime (e.g. wazero) | No CGO | Pure-Go alternative exists; but wazero's wasmtime-compat is incomplete (component model, WASI 0.2); different runtime semantics than the "wasmtime" operator surface promised in D-088 | DEFERRED (v0.10 evaluation if CLI-via-SSH proves insufficient) |
## Decision (full autonomy, auto-decision)
**Option B**: `WasmRuntime` uses the `wasmtime` CLI (apt-installed on
the peer) via the SSH-push transport. It does NOT import
`bytecodealliance/wasmtime-go` (or any other CGO package).
## C-01 Gate Status
**SATISFIED.** C-01 is satisfied:
- wasmtime works without CGO (CLI-via-SSH pattern, identical to podman/qm/pct).
- D-002 cross-compile story preserved (no CGO introduced anywhere in
the runtime package or any orca Go code).
- D-002 is NOT revoked.
## Recorded as D-187
See PROJECT.md v0.9 D-series: "wasmtime Go binding is CGO-based
(bytecodealliance/wasmtime-go); orca uses the wasmtime CLI via SSH
(apt-installed on peer) instead of the Go binding, avoiding CGO
entirely. D-002 cross-compile story preserved. C-01 satisfied."
## Verification
- `go build ./internal/runtime/` succeeds with `CGO_ENABLED=0`.
- `internal/runtime/wasm.go` imports only stdlib + sshpush (no
wasmtime-go).
- The full test suite (`go test ./...`) does not require CGO.
+142
View File
@@ -0,0 +1,142 @@
package runtime
import (
"context"
"fmt"
"strings"
"git.cloudinit.dev/coreci/orca/internal/sshpush"
)
// PodmanRuntime implements Runtime for the "podman" one_of. It runs
// `podman` on the peer over the SSH-push transport (P01). The
// transport is injected via the constructor (dependency injection).
//
// Container lifecycle:
//
// - Prepare: `podman pull <image>`
// - Start: `podman run -d --name orca-<alloc-id> <image> <command>`
// - Stop: `podman stop <name>` then `podman rm <name>`
// - Status: `podman inspect --format '{{.State.Running}}' <name>`
//
// The container name is `orca-<alloc-id>` (sanitized to lowercase +
// alnum). The runtime keeps no in-process state — each call is a fresh
// SSH exec against the peer.
type PodmanRuntime struct {
transport *sshpush.Transport
}
// NewPodmanRuntime returns a PodmanRuntime backed by the given transport.
func NewPodmanRuntime(t *sshpush.Transport) *PodmanRuntime {
return &PodmanRuntime{transport: t}
}
// containerName returns the deterministic container name for an alloc.
func containerName(alloc *Alloc) string {
id := strings.ToLower(alloc.ID)
id = strings.Map(func(r rune) rune {
if r >= 'a' && r <= 'z' || r >= '0' && r <= '9' || r == '-' || r == '_' {
return r
}
return '-'
}, id)
return "orca-" + id
}
// Prepare pulls the image on the peer.
func (p *PodmanRuntime) Prepare(ctx context.Context, alloc *Alloc) error {
image, err := imageFor(alloc)
if err != nil {
return err
}
cmd := fmt.Sprintf("podman pull %q", image)
if _, err := p.transport.Exec(ctx, alloc.Node, cmd); err != nil {
return fmt.Errorf("podman: pull: %w", err)
}
return nil
}
// Start runs `podman run -d --name <name> <image> <command>`.
func (p *PodmanRuntime) Start(ctx context.Context, alloc *Alloc) (int, error) {
image, err := imageFor(alloc)
if err != nil {
return 0, err
}
cmdStr, _ := commandFor(alloc)
name := containerName(alloc)
cmd := fmt.Sprintf("podman run -d --name %s %q %s", name, image, cmdStr)
out, err := p.transport.Exec(ctx, alloc.Node, cmd)
if err != nil {
return 0, fmt.Errorf("podman: run: %w", err)
}
// The container ID is the first 12 chars of the printed hash. We
// don't keep it — Stop/Status use the name — but return a stable
// synthetic PID derived from the first 4 bytes of the hash for
// the interface contract.
cid := strings.TrimSpace(string(out))
return podmanCidToPID(cid), nil
}
// Stop stops and removes the container.
func (p *PodmanRuntime) Stop(ctx context.Context, alloc *Alloc) error {
name := containerName(alloc)
if _, err := p.transport.Exec(ctx, alloc.Node, fmt.Sprintf("podman stop %s", name)); err != nil {
return fmt.Errorf("podman: stop: %w", err)
}
if _, err := p.transport.Exec(ctx, alloc.Node, fmt.Sprintf("podman rm %s", name)); err != nil {
return fmt.Errorf("podman: rm: %w", err)
}
return nil
}
// Status inspects the container's running state.
func (p *PodmanRuntime) Status(ctx context.Context, alloc *Alloc) (State, error) {
name := containerName(alloc)
cmd := fmt.Sprintf("podman inspect --format '{{.State.Running}}' %s", name)
out, err := p.transport.Exec(ctx, alloc.Node, cmd)
if err != nil {
return StateFailed, fmt.Errorf("podman: inspect: %w", err)
}
v := strings.TrimSpace(string(out))
switch v {
case "true":
return StateRunning, nil
case "false":
return StateStopped, nil
default:
return StateFailed, fmt.Errorf("podman: unexpected inspect output %q", v)
}
}
// podmanCidToPID converts a container ID (hex hash) to a positive int
// PID for the interface contract. It reads up to 4 hex chars.
func podmanCidToPID(cid string) int {
if len(cid) < 1 {
return 1
}
n := len(cid)
if n > 4 {
n = 4
}
var pid int
for i := 0; i < n; i++ {
c := cid[i]
pid = (pid << 4) | int(hexVal(c))
}
if pid <= 0 {
pid = 1
}
return pid
}
func hexVal(c byte) byte {
switch {
case c >= '0' && c <= '9':
return c - '0'
case c >= 'a' && c <= 'f':
return c - 'a' + 10
case c >= 'A' && c <= 'F':
return c - 'A' + 10
}
return 0
}
+229
View File
@@ -0,0 +1,229 @@
package runtime
import (
"context"
"errors"
"strings"
"testing"
"time"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
"git.cloudinit.dev/coreci/orca/internal/sshpush"
)
// allocNoImage returns an alloc whose Spec has a Runtime block with a
// command but no image.
func allocNoImage(runtime string) *Alloc {
return &Alloc{
ID: "x",
Runtime: runtime,
Spec: &jobspec.WorkloadSpec{
Runtime: &jobspec.RuntimeBlock{Command: "/bin/true"},
},
}
}
// TestPodmanRuntime_HappyPath wires a fake server that responds to
// podman pull/run/stop/rm/inspect and verifies the full lifecycle.
func TestPodmanRuntime_HappyPath(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
const cid = "abc123def456"
srv.setHandler("podman pull", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("podman run", func(cmd string) ([]byte, int) { return []byte(cid + "\n"), 0 })
srv.setHandler("podman stop", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("podman rm", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("podman inspect", func(cmd string) ([]byte, int) {
return []byte("true\n"), 0
})
tr := realTransport(t, srv)
defer tr.Close()
p := NewPodmanRuntime(tr)
a := allocWithNode("podman", "docker.io/library/alpine:latest", "sleep 30", srv.addr())
ctx, cancel := withTimeout(10 * time.Second)
defer cancel()
if err := p.Prepare(ctx, a); err != nil {
t.Fatalf("Prepare: %v", err)
}
pid, err := p.Start(ctx, a)
if err != nil {
t.Fatalf("Start: %v", err)
}
if pid <= 0 {
t.Fatalf("pid = %d, want > 0", pid)
}
st, err := p.Status(ctx, a)
if err != nil {
t.Fatalf("Status: %v", err)
}
if st != StateRunning {
t.Errorf("Status = %q, want running", st)
}
if err := p.Stop(ctx, a); err != nil {
t.Fatalf("Stop: %v", err)
}
}
// TestPodmanRuntime_StatusFalse verifies Status returns stopped when
// the container reports running=false.
func TestPodmanRuntime_StatusFalse(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("podman inspect", func(cmd string) ([]byte, int) {
return []byte("false\n"), 0
})
tr := realTransport(t, srv)
defer tr.Close()
p := NewPodmanRuntime(tr)
a := allocWithNode("podman", "img", "sleep 1", srv.addr())
st, err := p.Status(context.Background(), a)
if err != nil {
t.Fatalf("Status: %v", err)
}
if st != StateStopped {
t.Errorf("Status = %q, want stopped", st)
}
}
// TestPodmanRuntime_StatusBadOutput verifies Status returns failed on
// unexpected inspect output.
func TestPodmanRuntime_StatusBadOutput(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("podman inspect", func(cmd string) ([]byte, int) {
return []byte("garbage\n"), 0
})
tr := realTransport(t, srv)
defer tr.Close()
p := NewPodmanRuntime(tr)
a := allocWithNode("podman", "img", "sleep 1", srv.addr())
if _, err := p.Status(context.Background(), a); err == nil {
t.Error("Status with bad output should error")
}
}
// TestPodmanRuntime_PrepareNoImage verifies Prepare errors when the
// alloc has no image.
func TestPodmanRuntime_PrepareNoImage(t *testing.T) {
p := NewPodmanRuntime(nil)
a := allocNoImage("podman")
if err := p.Prepare(context.Background(), a); err == nil {
t.Error("Prepare with no image should error")
}
}
// TestPodmanRuntime_PrepareTransportError verifies Prepare propagates a
// transport error (podman pull fails).
func TestPodmanRuntime_PrepareTransportError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("podman pull", func(cmd string) ([]byte, int) {
return []byte("manifest unknown\n"), 2
})
tr := realTransport(t, srv)
defer tr.Close()
p := NewPodmanRuntime(tr)
a := allocWithNode("podman", "img", "sleep 1", srv.addr())
if err := p.Prepare(context.Background(), a); err == nil {
t.Error("Prepare with failed pull should error")
}
}
// TestPodmanRuntime_StartNoImage verifies Start errors with no image.
func TestPodmanRuntime_StartNoImage(t *testing.T) {
p := NewPodmanRuntime(nil)
a := allocNoImage("podman")
if _, err := p.Start(context.Background(), a); err == nil {
t.Error("Start with no image should error")
}
}
// TestPodmanRuntime_StopTransportError verifies Stop propagates errors.
func TestPodmanRuntime_StopTransportError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("podman stop", func(cmd string) ([]byte, int) {
return []byte("no such container\n"), 1
})
tr := realTransport(t, srv)
defer tr.Close()
p := NewPodmanRuntime(tr)
a := allocWithNode("podman", "img", "sleep 1", srv.addr())
if err := p.Stop(context.Background(), a); err == nil {
t.Error("Stop with missing container should error")
}
}
// TestPodmanRuntime_StatusInspectError verifies Status returns failed
// when inspect itself errors.
func TestPodmanRuntime_StatusInspectError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("podman inspect", func(cmd string) ([]byte, int) {
return []byte("no such container\n"), 1
})
tr := realTransport(t, srv)
defer tr.Close()
p := NewPodmanRuntime(tr)
a := allocWithNode("podman", "img", "sleep 1", srv.addr())
st, err := p.Status(context.Background(), a)
if err == nil {
t.Error("Status with inspect error should error")
}
if st != StateFailed {
t.Errorf("Status = %q, want failed", st)
}
}
// TestPodmanCidToPID verifies the synthetic PID derivation.
func TestPodmanCidToPID(t *testing.T) {
if got := podmanCidToPID(""); got != 1 {
t.Errorf("empty cid -> %d, want 1", got)
}
if got := podmanCidToPID("a"); got <= 0 {
t.Errorf("single hex -> %d, want > 0", got)
}
if got := podmanCidToPID("abcd"); got <= 0 {
t.Errorf("abcd -> %d, want > 0", got)
}
// non-hex chars fall through to 0 contributions but still yield
// a positive result (>= 1 by the floor).
if got := podmanCidToPID("xyz123"); got <= 0 {
t.Errorf("xyz123 -> %d, want > 0", got)
}
}
// TestContainerNameSanitization verifies the container-name sanitizer
// uppercases and strips disallowed characters.
func TestContainerNameSanitization(t *testing.T) {
a := &Alloc{ID: "ALLOC_1.2.3", Spec: nil, Runtime: "podman"}
got := containerName(a)
if !strings.HasPrefix(got, "orca-") {
t.Errorf("containerName = %q, want orca- prefix", got)
}
if strings.Contains(got, ".") {
t.Errorf("containerName = %q, should not contain '.'", got)
}
}
// TestPodmanRuntime_DialError verifies Prepare fails fast when the
// peer is unreachable (no fake server).
func TestPodmanRuntime_DialError(t *testing.T) {
srv := newFakeServer(t)
// close immediately so dial fails.
srv.close()
tr := realTransport(t, srv)
defer tr.Close()
p := NewPodmanRuntime(tr)
a := allocWithNode("podman", "img", "sleep 1", srv.addr())
if err := p.Prepare(context.Background(), a); err == nil {
t.Error("Prepare against dead peer should error")
} else if !errors.Is(err, sshpush.ErrTransient) && !errors.Is(err, sshpush.ErrPermanent) {
// acceptable: either transient (retry exhausted) or permanent.
t.Logf("Prepare err (acceptable): %v", err)
}
}
+204
View File
@@ -0,0 +1,204 @@
package runtime
import (
"context"
"errors"
"fmt"
"os"
"os/exec"
"strconv"
"strings"
"sync"
"syscall"
"time"
)
// StopGrace is the default grace period between SIGTERM and SIGKILL
// for ProcessRuntime.Stop (10s, matching the systemd default TimeoutStopSec).
const StopGrace = 10 * time.Second
// ProcessRuntime implements Runtime for the "process" one_of using
// os/exec. It is the in-process equivalent of the systemd unit the CLI
// emits in production: the CLI emits a .service file and the peer's
// systemd runs the process; ProcessRuntime starts the process directly
// in the current Go process. It is intended for LOCAL testing and
// hermetic CI — NOT for production (production uses the systemd emitter
// + the peer's systemd, not this in-process path).
//
// It tracks started PIDs in an in-memory map; restarts of the CLI lose
// that state (acceptable for the local-test use case).
type ProcessRuntime struct {
mu sync.Mutex
pids map[string]int // alloc.ID -> PID
procs map[int]*os.Process // PID -> process handle
stopped map[string]bool // alloc.ID -> reported stopped after Stop
}
// NewProcessRuntime returns a ProcessRuntime.
func NewProcessRuntime() *ProcessRuntime {
return &ProcessRuntime{
pids: make(map[string]int),
procs: make(map[int]*os.Process),
stopped: make(map[string]bool),
}
}
// Prepare is a no-op for the process runtime: the systemd unit is
// emitted by the systemd emitter (internal/emitter), not by the runtime.
func (p *ProcessRuntime) Prepare(ctx context.Context, alloc *Alloc) error {
_ = ctx
_ = alloc
return nil
}
// Start execs the alloc's command and returns the PID. The process is
// left running in the background; Stop terminates it.
func (p *ProcessRuntime) Start(ctx context.Context, alloc *Alloc) (int, error) {
cmdStr, err := commandFor(alloc)
if err != nil {
return 0, err
}
// Parse the command string into argv. A leading "exec" form
// (shell-style) is NOT supported — the command must be a direct
// argv[0] + args. Split on whitespace (simple, matches the existing
// executor.go behaviour which takes Command + Args separately).
parts := strings.Fields(cmdStr)
if len(parts) == 0 {
return 0, fmt.Errorf("process: empty command for alloc %s", alloc.ID)
}
// Use a detached context so the process survives the request
// context cancellation (the request ends; the workload keeps
// running until Stop). We apply our own timeout for Start only.
startCtx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
defer cancel()
cmd := exec.CommandContext(startCtx, parts[0], parts[1:]...)
// Detach the child from the parent's process group so it survives.
cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
// Discard output for the runtime; the systemd unit captures logs
// in production. For tests, callers that need output run their
// own exec.Command.
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
if err := cmd.Start(); err != nil {
return 0, fmt.Errorf("process: start: %w", err)
}
pid := cmd.Process.Pid
// Background-reap the process so it doesn't become a zombie; we
// only need the PID for Stop/Status. When the process exits
// naturally, mark the alloc as stopped.
go func() {
_ = cmd.Wait()
p.mu.Lock()
delete(p.procs, pid)
// Only mark stopped if the alloc is still associated with
// this PID (Stop may have already removed the mapping).
if cur, ok := p.pids[alloc.ID]; ok && cur == pid {
delete(p.pids, alloc.ID)
p.stopped[alloc.ID] = true
}
p.mu.Unlock()
}()
p.mu.Lock()
p.pids[alloc.ID] = pid
p.procs[pid] = cmd.Process
delete(p.stopped, alloc.ID)
p.mu.Unlock()
return pid, nil
}
// Stop sends SIGTERM, waits the grace period, then SIGKILL.
func (p *ProcessRuntime) Stop(ctx context.Context, alloc *Alloc) error {
p.mu.Lock()
pid, ok := p.pids[alloc.ID]
proc := p.procs[pid]
p.mu.Unlock()
if !ok || proc == nil {
return nil // not running; idempotent
}
// SIGTERM the process group (negative PID).
_ = syscall.Kill(-pid, syscall.SIGTERM)
grace := StopGrace
if dl, ok := ctx.Deadline(); ok {
if remaining := time.Until(dl); remaining > 0 && remaining < grace {
grace = remaining
}
}
deadline := time.Now().Add(grace)
for time.Now().Before(deadline) {
if !p.alive(pid) {
p.forget(alloc.ID, pid)
return nil
}
select {
case <-ctx.Done():
p.forget(alloc.ID, pid)
return ctx.Err()
case <-time.After(100 * time.Millisecond):
}
}
// SIGKILL the group.
_ = syscall.Kill(-pid, syscall.SIGKILL)
p.forget(alloc.ID, pid)
return nil
}
// Status reports the alloc's state by checking if the process is alive.
func (p *ProcessRuntime) Status(ctx context.Context, alloc *Alloc) (State, error) {
_ = ctx
p.mu.Lock()
pid, ok := p.pids[alloc.ID]
wasStopped := p.stopped[alloc.ID]
p.mu.Unlock()
if !ok {
if wasStopped {
return StateStopped, nil
}
return StatePending, nil
}
if !p.alive(pid) {
// Process exited but the reaper hasn't run yet; mark it
// stopped and clean up.
p.forget(alloc.ID, pid)
return StateStopped, nil
}
return StateRunning, nil
}
// alive reports whether the process with the given PID is still running.
func (p *ProcessRuntime) alive(pid int) bool {
proc, err := os.FindProcess(pid)
if err != nil {
return false
}
if err := proc.Signal(syscall.Signal(0)); err != nil {
// ESRCH means the process is gone.
return false
}
return true
}
// forget removes the alloc/PID mapping and marks the alloc stopped.
func (p *ProcessRuntime) forget(allocID string, pid int) {
p.mu.Lock()
delete(p.pids, allocID)
delete(p.procs, pid)
p.stopped[allocID] = true
p.mu.Unlock()
}
// PID returns the recorded PID for alloc (for tests/inspection).
func (p *ProcessRuntime) PID(allocID string) (int, error) {
p.mu.Lock()
defer p.mu.Unlock()
pid, ok := p.pids[allocID]
if !ok {
return 0, errors.New("process: no pid for alloc " + strconv.Quote(allocID))
}
return pid, nil
}
+172
View File
@@ -0,0 +1,172 @@
package runtime
import (
"context"
"runtime"
"testing"
"time"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// TestProcessRuntime_PrepareIsNoop verifies Prepare is a no-op.
func TestProcessRuntime_PrepareIsNoop(t *testing.T) {
p := NewProcessRuntime()
a := alloc("process", "", "/bin/true")
if err := p.Prepare(context.Background(), a); err != nil {
t.Fatalf("Prepare: %v", err)
}
}
// TestProcessRuntime_PrepareNilSpec exercises the nil-spec branch
// indirectly — Prepare is a no-op regardless of input.
func TestProcessRuntime_PrepareNilSpec(t *testing.T) {
p := NewProcessRuntime()
if err := p.Prepare(context.Background(), &Alloc{ID: "x"}); err != nil {
t.Fatalf("Prepare (nil spec) should still be no-op: %v", err)
}
}
// TestProcessRuntime_StartStopStatus runs a real long-lived process
// (sleep) and verifies Start -> Status(running) -> Stop -> Status(stopped).
func TestProcessRuntime_StartStopStatus(t *testing.T) {
if _, err := sleepBin(); err != nil {
t.Skipf("sleep binary not available: %v", err)
}
p := NewProcessRuntime()
sleepCmd, _ := sleepBin()
a := alloc("process", "", sleepCmd+" 30")
ctx, cancel := withTimeout(10 * time.Second)
defer cancel()
pid, err := p.Start(ctx, a)
if err != nil {
t.Fatalf("Start: %v", err)
}
if pid <= 0 {
t.Fatalf("pid = %d, want > 0", pid)
}
st, err := p.Status(context.Background(), a)
if err != nil {
t.Fatalf("Status: %v", err)
}
if st != StateRunning {
t.Errorf("Status = %q, want running", st)
}
// Verify PID lookup.
got, err := p.PID(a.ID)
if err != nil || got != pid {
t.Errorf("PID = %d/%v, want %d", got, err, pid)
}
stopCtx, cancelStop := withTimeout(15 * time.Second)
defer cancelStop()
if err := p.Stop(stopCtx, a); err != nil {
t.Fatalf("Stop: %v", err)
}
st2, _ := p.Status(context.Background(), a)
if st2 != StateStopped {
t.Errorf("Status after Stop = %q, want stopped", st2)
}
}
// TestProcessRuntime_StopNotStarted verifies Stop is idempotent on a
// never-started alloc.
func TestProcessRuntime_StopNotStarted(t *testing.T) {
p := NewProcessRuntime()
a := alloc("process", "", "/bin/true")
ctx, cancel := withTimeout(2 * time.Second)
defer cancel()
if err := p.Stop(ctx, a); err != nil {
t.Errorf("Stop on never-started alloc should be no-op, got %v", err)
}
}
// TestProcessRuntime_StatusPending verifies Status returns pending for
// an alloc that was never started.
func TestProcessRuntime_StatusPending(t *testing.T) {
p := NewProcessRuntime()
a := alloc("process", "", "/bin/true")
st, err := p.Status(context.Background(), a)
if err != nil {
t.Fatalf("Status: %v", err)
}
if st != StatePending {
t.Errorf("Status = %q, want pending", st)
}
}
// TestProcessRuntime_StartNoCommand verifies Start errors on missing
// command.
func TestProcessRuntime_StartNoCommand(t *testing.T) {
p := NewProcessRuntime()
a := &Alloc{ID: "x", Spec: nil, Runtime: "process"}
if _, err := p.Start(context.Background(), a); err == nil {
t.Error("Start with nil spec should error")
}
}
// TestProcessRuntime_StartEmptyCommand verifies Start errors when
// the command parses to zero argv.
func TestProcessRuntime_StartEmptyCommand(t *testing.T) {
p := NewProcessRuntime()
a := &Alloc{
ID: "e",
Runtime: "process",
Spec: &jobspec.WorkloadSpec{
Runtime: &jobspec.RuntimeBlock{Command: " "},
},
}
_, err := p.Start(context.Background(), a)
if err == nil {
t.Error("Start with empty command should error")
}
}
// TestProcessRuntime_StartBadBinary verifies Start propagates exec
// errors for a missing binary.
func TestProcessRuntime_StartBadBinary(t *testing.T) {
p := NewProcessRuntime()
a := alloc("process", "", "/no/such/binary/here")
_, err := p.Start(context.Background(), a)
if err == nil {
t.Error("Start with missing binary should error")
}
}
// TestProcessRuntime_StopOnFinishedProcess verifies Stop on a process
// that already exited (e.g. /bin/true) is a no-op (no error).
func TestProcessRuntime_StopOnFinishedProcess(t *testing.T) {
p := NewProcessRuntime()
a := alloc("process", "", "/bin/true")
_, _ = p.Start(context.Background(), a)
// Give /bin/true time to exit.
time.Sleep(200 * time.Millisecond)
ctx, cancel := withTimeout(5 * time.Second)
defer cancel()
if err := p.Stop(ctx, a); err != nil {
t.Errorf("Stop on exited process should be no-op, got %v", err)
}
}
// TestProcessRuntime_PIDUnknown verifies PID lookup errors for an
// unknown alloc.
func TestProcessRuntime_PIDUnknown(t *testing.T) {
p := NewProcessRuntime()
if _, err := p.PID("nope"); err == nil {
t.Error("PID unknown should error")
}
}
// sleepBin returns the sleep command path ("sleep") on this OS.
func sleepBin() (string, error) {
// /bin/sleep exists on Linux; on other platforms fall back to
// "sleep" (resolved via PATH).
if runtime.GOOS == "linux" {
return "/bin/sleep", nil
}
return "sleep", nil
}
+177
View File
@@ -0,0 +1,177 @@
package runtime
import (
"context"
"fmt"
"strings"
"git.cloudinit.dev/coreci/orca/internal/sshpush"
)
// vmidFor returns a deterministic 5-digit VMID derived from the alloc
// ID (hash(alloc.ID) % 99999 + 1). Used by both pveVMRuntime and
// pveCTRuntime — VM and container IDs share the same numeric space on
// a Proxmox node, but the namespace is sparse (one alloc = one VMID)
// so collisions are rare in practice.
func vmidFor(alloc *Alloc) int {
id := allocIDHash(alloc.ID)
if id < 100 {
id += 100
}
return id
}
// PveVMRuntime implements Runtime for the "pve-vm" one_of using the
// `qm` tool over SSH on a Proxmox peer (extends REQ-076). The transport
// is injected.
//
// Lifecycle:
//
// - Prepare: `qm create <vmid> --memory <mb> --cores <n> --scsi0 <disk>`
// - Start: `qm start <vmid>`
// - Stop: `qm shutdown <vmid>` (graceful) then `qm stop <vmid>` (force)
// - Status: `qm status <vmid>`
type PveVMRuntime struct {
transport *sshpush.Transport
}
// NewPveVMRuntime returns a PveVMRuntime backed by the given transport.
func NewPveVMRuntime(t *sshpush.Transport) *PveVMRuntime {
return &PveVMRuntime{transport: t}
}
// Prepare creates the VM. Memory/Cores default to 512MB / 1 if the
// spec's Runtime block omits them; the disk is the spec's image field
// (a Proxmox storage path like local:vmdir/disk.qcow2).
func (v *PveVMRuntime) Prepare(ctx context.Context, alloc *Alloc) error {
if alloc == nil || alloc.Spec == nil || alloc.Spec.Runtime == nil {
return fmt.Errorf("pve-vm: nil alloc/spec/runtime")
}
vmid := vmidFor(alloc)
image := alloc.Spec.Runtime.Image
if image == "" {
return fmt.Errorf("pve-vm: alloc %s has no disk (Runtime.Image)", alloc.ID)
}
mb := 512
cores := 1
cmd := fmt.Sprintf("qm create %d --memory %d --cores %d --scsi0 %q", vmid, mb, cores, image)
if _, err := v.transport.Exec(ctx, alloc.Node, cmd); err != nil {
return fmt.Errorf("pve-vm: create: %w", err)
}
return nil
}
// Start boots the VM.
func (v *PveVMRuntime) Start(ctx context.Context, alloc *Alloc) (int, error) {
vmid := vmidFor(alloc)
if _, err := v.transport.Exec(ctx, alloc.Node, fmt.Sprintf("qm start %d", vmid)); err != nil {
return 0, fmt.Errorf("pve-vm: start: %w", err)
}
return vmid, nil
}
// Stop gracefully shuts down then force-stops the VM.
func (v *PveVMRuntime) Stop(ctx context.Context, alloc *Alloc) error {
vmid := vmidFor(alloc)
if _, err := v.transport.Exec(ctx, alloc.Node, fmt.Sprintf("qm shutdown %d", vmid)); err != nil {
// Best-effort graceful; fall through to force stop.
}
if _, err := v.transport.Exec(ctx, alloc.Node, fmt.Sprintf("qm stop %d", vmid)); err != nil {
return fmt.Errorf("pve-vm: stop: %w", err)
}
return nil
}
// Status reports the VM state from `qm status`.
func (v *PveVMRuntime) Status(ctx context.Context, alloc *Alloc) (State, error) {
vmid := vmidFor(alloc)
out, err := v.transport.Exec(ctx, alloc.Node, fmt.Sprintf("qm status %d", vmid))
if err != nil {
return StateFailed, fmt.Errorf("pve-vm: status: %w", err)
}
s := strings.ToLower(string(out))
switch {
case strings.Contains(s, "running"):
return StateRunning, nil
case strings.Contains(s, "stopped"):
return StateStopped, nil
default:
return StatePending, nil
}
}
// PveCTRuntime implements Runtime for the "pve-ct" one_of using the
// `pct` tool over SSH on a Proxmox peer (extends REQ-076).
//
// Lifecycle:
//
// - Prepare: `pct create <vmid> <template> --memory <mb> --cores <n>`
// - Start: `pct start <vmid>`
// - Stop: `pct shutdown <vmid>` then `pct stop <vmid>`
// - Status: `pct status <vmid>`
type PveCTRuntime struct {
transport *sshpush.Transport
}
// NewPveCTRuntime returns a PveCTRuntime backed by the given transport.
func NewPveCTRuntime(t *sshpush.Transport) *PveCTRuntime {
return &PveCTRuntime{transport: t}
}
// Prepare creates the LXC container.
func (c *PveCTRuntime) Prepare(ctx context.Context, alloc *Alloc) error {
if alloc == nil || alloc.Spec == nil || alloc.Spec.Runtime == nil {
return fmt.Errorf("pve-ct: nil alloc/spec/runtime")
}
vmid := vmidFor(alloc)
template := alloc.Spec.Runtime.Image
if template == "" {
return fmt.Errorf("pve-ct: alloc %s has no template (Runtime.Image)", alloc.ID)
}
mb := 512
cores := 1
cmd := fmt.Sprintf("pct create %d %q --memory %d --cores %d", vmid, template, mb, cores)
if _, err := c.transport.Exec(ctx, alloc.Node, cmd); err != nil {
return fmt.Errorf("pve-ct: create: %w", err)
}
return nil
}
// Start boots the container.
func (c *PveCTRuntime) Start(ctx context.Context, alloc *Alloc) (int, error) {
vmid := vmidFor(alloc)
if _, err := c.transport.Exec(ctx, alloc.Node, fmt.Sprintf("pct start %d", vmid)); err != nil {
return 0, fmt.Errorf("pve-ct: start: %w", err)
}
return vmid, nil
}
// Stop gracefully then force stops the container.
func (c *PveCTRuntime) Stop(ctx context.Context, alloc *Alloc) error {
vmid := vmidFor(alloc)
if _, err := c.transport.Exec(ctx, alloc.Node, fmt.Sprintf("pct shutdown %d", vmid)); err != nil {
// best-effort
}
if _, err := c.transport.Exec(ctx, alloc.Node, fmt.Sprintf("pct stop %d", vmid)); err != nil {
return fmt.Errorf("pve-ct: stop: %w", err)
}
return nil
}
// Status reports the container state from `pct status`.
func (c *PveCTRuntime) Status(ctx context.Context, alloc *Alloc) (State, error) {
vmid := vmidFor(alloc)
out, err := c.transport.Exec(ctx, alloc.Node, fmt.Sprintf("pct status %d", vmid))
if err != nil {
return StateFailed, fmt.Errorf("pve-ct: status: %w", err)
}
s := strings.ToLower(string(out))
switch {
case strings.Contains(s, "running"):
return StateRunning, nil
case strings.Contains(s, "stopped"):
return StateStopped, nil
default:
return StatePending, nil
}
}
+342
View File
@@ -0,0 +1,342 @@
package runtime
import (
"context"
"testing"
"time"
)
// TestPveVMRuntime_HappyPath verifies the qm lifecycle.
func TestPveVMRuntime_HappyPath(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("qm create", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("qm start", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("qm shutdown", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("qm stop", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("qm status", func(cmd string) ([]byte, int) {
return []byte("status: running\n"), 0
})
tr := realTransport(t, srv)
defer tr.Close()
v := NewPveVMRuntime(tr)
a := allocWithNode("pve-vm", "local:vmdir/disk.qcow2", "", srv.addr())
ctx, cancel := withTimeout(10 * time.Second)
defer cancel()
if err := v.Prepare(ctx, a); err != nil {
t.Fatalf("Prepare: %v", err)
}
pid, err := v.Start(ctx, a)
if err != nil {
t.Fatalf("Start: %v", err)
}
if pid <= 0 {
t.Fatalf("pid = %d, want > 0", pid)
}
st, err := v.Status(ctx, a)
if err != nil {
t.Fatalf("Status: %v", err)
}
if st != StateRunning {
t.Errorf("Status = %q, want running", st)
}
if err := v.Stop(ctx, a); err != nil {
t.Fatalf("Stop: %v", err)
}
}
// TestPveVMRuntime_StatusStopped verifies Status maps "stopped".
func TestPveVMRuntime_StatusStopped(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("qm status", func(cmd string) ([]byte, int) {
return []byte("status: stopped\n"), 0
})
tr := realTransport(t, srv)
defer tr.Close()
v := NewPveVMRuntime(tr)
a := allocWithNode("pve-vm", "img", "", srv.addr())
st, _ := v.Status(context.Background(), a)
if st != StateStopped {
t.Errorf("Status = %q, want stopped", st)
}
}
// TestPveVMRuntime_StatusUnknown verifies Status returns pending on
// unrecognized output.
func TestPveVMRuntime_StatusUnknown(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("qm status", func(cmd string) ([]byte, int) {
return []byte("weird state\n"), 0
})
tr := realTransport(t, srv)
defer tr.Close()
v := NewPveVMRuntime(tr)
a := allocWithNode("pve-vm", "img", "", srv.addr())
st, _ := v.Status(context.Background(), a)
if st != StatePending {
t.Errorf("Status = %q, want pending", st)
}
}
// TestPveVMRuntime_PrepareNoImage verifies Prepare errors with no disk.
func TestPveVMRuntime_PrepareNoImage(t *testing.T) {
v := NewPveVMRuntime(nil)
a := allocNoImage("pve-vm")
if err := v.Prepare(context.Background(), a); err == nil {
t.Error("Prepare with no disk should error")
}
}
// TestPveVMRuntime_PrepareNilRuntime verifies Prepare errors when
// the Spec has no Runtime block at all.
func TestPveVMRuntime_PrepareNilRuntime(t *testing.T) {
v := NewPveVMRuntime(nil)
a := &Alloc{ID: "x", Runtime: "pve-vm", Spec: nil}
if err := v.Prepare(context.Background(), a); err == nil {
t.Error("Prepare with nil spec should error")
}
}
// TestPveVMRuntime_StartError verifies Start propagates qm start errors.
func TestPveVMRuntime_StartError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("qm start", func(cmd string) ([]byte, int) {
return []byte("already running\n"), 1
})
tr := realTransport(t, srv)
defer tr.Close()
v := NewPveVMRuntime(tr)
a := allocWithNode("pve-vm", "img", "", srv.addr())
if _, err := v.Start(context.Background(), a); err == nil {
t.Error("Start with qm error should error")
}
}
// TestPveVMRuntime_StopError verifies Stop errors on qm stop failure.
func TestPveVMRuntime_StopError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("qm shutdown", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("qm stop", func(cmd string) ([]byte, int) {
return []byte("vm locked\n"), 1
})
tr := realTransport(t, srv)
defer tr.Close()
v := NewPveVMRuntime(tr)
a := allocWithNode("pve-vm", "img", "", srv.addr())
if err := v.Stop(context.Background(), a); err == nil {
t.Error("Stop with qm error should error")
}
}
// TestPveVMRuntime_StatusError verifies Status returns failed on qm
// status error.
func TestPveVMRuntime_StatusError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("qm status", func(cmd string) ([]byte, int) {
return []byte("no such vm\n"), 1
})
tr := realTransport(t, srv)
defer tr.Close()
v := NewPveVMRuntime(tr)
a := allocWithNode("pve-vm", "img", "", srv.addr())
st, err := v.Status(context.Background(), a)
if err == nil {
t.Error("Status with qm error should error")
}
if st != StateFailed {
t.Errorf("Status = %q, want failed", st)
}
}
// TestPveVMRuntime_PrepareError verifies Prepare propagates qm create
// errors.
func TestPveVMRuntime_PrepareError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("qm create", func(cmd string) ([]byte, int) {
return []byte("vmid already exists\n"), 1
})
tr := realTransport(t, srv)
defer tr.Close()
v := NewPveVMRuntime(tr)
a := allocWithNode("pve-vm", "img", "", srv.addr())
if err := v.Prepare(context.Background(), a); err == nil {
t.Error("Prepare with qm create error should error")
}
}
// --- PveCTRuntime ---
func TestPveCTRuntime_HappyPath(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("pct create", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("pct start", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("pct shutdown", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("pct stop", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("pct status", func(cmd string) ([]byte, int) {
return []byte("status: running\n"), 0
})
tr := realTransport(t, srv)
defer tr.Close()
c := NewPveCTRuntime(tr)
a := allocWithNode("pve-ct", "local:vztmpl/alpine.tar.xz", "", srv.addr())
ctx, cancel := withTimeout(10 * time.Second)
defer cancel()
if err := c.Prepare(ctx, a); err != nil {
t.Fatalf("Prepare: %v", err)
}
pid, err := c.Start(ctx, a)
if err != nil {
t.Fatalf("Start: %v", err)
}
if pid <= 0 {
t.Fatalf("pid = %d, want > 0", pid)
}
st, err := c.Status(ctx, a)
if err != nil {
t.Fatalf("Status: %v", err)
}
if st != StateRunning {
t.Errorf("Status = %q, want running", st)
}
if err := c.Stop(ctx, a); err != nil {
t.Fatalf("Stop: %v", err)
}
}
// TestPveCTRuntime_StatusStopped verifies pct status stopped.
func TestPveCTRuntime_StatusStopped(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("pct status", func(cmd string) ([]byte, int) {
return []byte("status: stopped\n"), 0
})
tr := realTransport(t, srv)
defer tr.Close()
c := NewPveCTRuntime(tr)
a := allocWithNode("pve-ct", "tmpl", "", srv.addr())
st, _ := c.Status(context.Background(), a)
if st != StateStopped {
t.Errorf("Status = %q, want stopped", st)
}
}
// TestPveCTRuntime_StatusUnknown verifies pending on unrecognized
// output.
func TestPveCTRuntime_StatusUnknown(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("pct status", func(cmd string) ([]byte, int) {
return []byte("unknown\n"), 0
})
tr := realTransport(t, srv)
defer tr.Close()
c := NewPveCTRuntime(tr)
a := allocWithNode("pve-ct", "tmpl", "", srv.addr())
st, _ := c.Status(context.Background(), a)
if st != StatePending {
t.Errorf("Status = %q, want pending", st)
}
}
// TestPveCTRuntime_PrepareNoImage verifies Prepare errors with no
// template.
func TestPveCTRuntime_PrepareNoImage(t *testing.T) {
c := NewPveCTRuntime(nil)
a := allocNoImage("pve-ct")
if err := c.Prepare(context.Background(), a); err == nil {
t.Error("Prepare with no template should error")
}
}
// TestPveCTRuntime_PrepareNilRuntime verifies Prepare errors on nil
// spec.
func TestPveCTRuntime_PrepareNilRuntime(t *testing.T) {
c := NewPveCTRuntime(nil)
a := &Alloc{ID: "x", Runtime: "pve-ct", Spec: nil}
if err := c.Prepare(context.Background(), a); err == nil {
t.Error("Prepare with nil spec should error")
}
}
// TestPveCTRuntime_StartError verifies Start errors on pct start fail.
func TestPveCTRuntime_StartError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("pct start", func(cmd string) ([]byte, int) {
return []byte("already running\n"), 1
})
tr := realTransport(t, srv)
defer tr.Close()
c := NewPveCTRuntime(tr)
a := allocWithNode("pve-ct", "tmpl", "", srv.addr())
if _, err := c.Start(context.Background(), a); err == nil {
t.Error("Start with pct error should error")
}
}
// TestPveCTRuntime_StopError verifies Stop errors on pct stop fail.
func TestPveCTRuntime_StopError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("pct shutdown", func(cmd string) ([]byte, int) { return nil, 0 })
srv.setHandler("pct stop", func(cmd string) ([]byte, int) {
return []byte("container locked\n"), 1
})
tr := realTransport(t, srv)
defer tr.Close()
c := NewPveCTRuntime(tr)
a := allocWithNode("pve-ct", "tmpl", "", srv.addr())
if err := c.Stop(context.Background(), a); err == nil {
t.Error("Stop with pct error should error")
}
}
// TestPveCTRuntime_StatusError verifies Status failed on pct status
// error.
func TestPveCTRuntime_StatusError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("pct status", func(cmd string) ([]byte, int) {
return []byte("no such container\n"), 1
})
tr := realTransport(t, srv)
defer tr.Close()
c := NewPveCTRuntime(tr)
a := allocWithNode("pve-ct", "tmpl", "", srv.addr())
st, err := c.Status(context.Background(), a)
if err == nil {
t.Error("Status with pct error should error")
}
if st != StateFailed {
t.Errorf("Status = %q, want failed", st)
}
}
// TestVMIDFor verifies the VMID is deterministic and within range.
func TestVMIDFor(t *testing.T) {
a := &Alloc{ID: "alloc-1"}
id := vmidFor(a)
if id < 100 || id > 99999 {
t.Errorf("vmidFor = %d, want in [100, 99999]", id)
}
// stability
if vmidFor(a) != id {
t.Error("vmidFor not stable")
}
// two allocs should differ
b := &Alloc{ID: "alloc-2"}
if vmidFor(b) == id {
t.Logf("note: two allocs collided on vmid (rare but allowed)")
}
}
+20
View File
@@ -0,0 +1,20 @@
package runtime
import (
"git.cloudinit.dev/coreci/orca/internal/sshpush"
)
// DefaultRegistry returns a Registry with all five runtime backends
// registered (process, podman, wasm, pve-vm, pve-ct). The process
// runtime is transport-less; the others are backed by the given
// transport (which may be nil — the per-method calls will fail with
// an sshpush error, but registration still succeeds).
func DefaultRegistry(transport *sshpush.Transport) *Registry {
r := NewRegistry()
r.Register("process", NewProcessRuntime())
r.Register("podman", NewPodmanRuntime(transport))
r.Register("wasm", NewWasmRuntime(transport))
r.Register("pve-vm", NewPveVMRuntime(transport))
r.Register("pve-ct", NewPveCTRuntime(transport))
return r
}
+176
View File
@@ -0,0 +1,176 @@
// Package runtime implements the runtime abstraction (REQ-078, I-B-006).
//
// The Runtime interface decouples the scheduler/CLI from the underlying
// execution backend. Five implementations are provided:
//
// - ProcessRuntime ("process") — wraps os/exec; LOCAL testing only.
// - PodmanRuntime ("podman") — SSH-push podman run on the peer.
// - WasmRuntime ("wasm") — wasmtime CLI via SSH (NO CGO; see
// C01_WASMTIME_CGO_EVAL.md for the C-01 grill gate evaluation).
// - PveVMRuntime ("pve-vm") — `qm` over SSH to a Proxmox peer.
// - PveCTRuntime ("pve-ct") — `pct` over SSH to a Proxmox peer.
//
// The Registry is keyed by the runtime.one_of frontmatter value. The
// Alloc carries a Runtime field that can change on migration (R-004).
package runtime
import (
"context"
"fmt"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// State is the lifecycle state of an alloc as observed by a Runtime.
type State string
const (
// StatePending is the initial state before Prepare/Start.
StatePending State = "pending"
// StateRunning means the runtime reports the workload as up.
StateRunning State = "running"
// StateStopped means the workload exited cleanly (Stop called
// or the process finished with exit 0).
StateStopped State = "stopped"
// StateFailed means the workload exited non-zero or could not
// be reached.
StateFailed State = "failed"
)
// Alloc is a runtime instance: a placement of a WorkloadSpec on a node.
// The Runtime field is the runtime.one_of value used to dispatch to the
// correct Runtime implementation; it can change on migration (R-004).
type Alloc struct {
ID string
Spec *jobspec.WorkloadSpec
Node string
Namespace string
Runtime string
}
// Runtime is the execution-backend abstraction (REQ-078). Each method
// takes a context for cancellation/timeout. Implementations wrap a
// different execution backend (process, podman, wasm, pve-vm, pve-ct).
//
// Prepare is idempotent; Start/Stop/Status operate on the prepared
// runtime. The returned PID from Start is best-effort (container
// runtimes return the container ID hash as a synthetic PID).
type Runtime interface {
// Prepare provisions prerequisites for the alloc (image pull,
// vm create, etc.). It is idempotent.
Prepare(ctx context.Context, alloc *Alloc) error
// Start launches the workload and returns a best-effort PID (or
// container/VM identifier encoded as a positive integer).
Start(ctx context.Context, alloc *Alloc) (pid int, err error)
// Stop terminates the workload, gracefully first then forcibly
// after a grace period.
Stop(ctx context.Context, alloc *Alloc) error
// Status reports the current State of the alloc.
Status(ctx context.Context, alloc *Alloc) (State, error)
}
// Registry maps runtime.one_of values to Runtime implementations. The
// zero value is NOT usable; construct one with NewRegistry.
type Registry struct {
runtimes map[string]Runtime
}
// NewRegistry returns an empty Registry.
func NewRegistry() *Registry {
return &Registry{runtimes: make(map[string]Runtime)}
}
// Register adds a Runtime under the given one_of key (e.g. "process",
// "podman", "wasm", "pve-vm", "pve-ct"). Registering the same key twice
// replaces the prior implementation (last-wins) — this is intentional
// so tests can override.
func (r *Registry) Register(name string, rt Runtime) {
if r.runtimes == nil {
r.runtimes = make(map[string]Runtime)
}
r.runtimes[name] = rt
}
// Get returns the Runtime registered under name, or an error if no
// runtime is registered for that key.
func (r *Registry) Get(name string) (Runtime, error) {
rt, ok := r.runtimes[name]
if !ok {
return nil, fmt.Errorf("runtime: no backend registered for %q", name)
}
return rt, nil
}
// Prepare dispatches to the Runtime registered for alloc.Runtime. It
// returns an error if the runtime is unknown or Prepare fails.
func (r *Registry) Prepare(ctx context.Context, alloc *Alloc) error {
rt, err := r.Get(alloc.Runtime)
if err != nil {
return err
}
return rt.Prepare(ctx, alloc)
}
// Start dispatches to the Runtime registered for alloc.Runtime.
func (r *Registry) Start(ctx context.Context, alloc *Alloc) (int, error) {
rt, err := r.Get(alloc.Runtime)
if err != nil {
return 0, err
}
return rt.Start(ctx, alloc)
}
// Stop dispatches to the Runtime registered for alloc.Runtime.
func (r *Registry) Stop(ctx context.Context, alloc *Alloc) error {
rt, err := r.Get(alloc.Runtime)
if err != nil {
return err
}
return rt.Stop(ctx, alloc)
}
// Status dispatches to the Runtime registered for alloc.Runtime.
func (r *Registry) Status(ctx context.Context, alloc *Alloc) (State, error) {
rt, err := r.Get(alloc.Runtime)
if err != nil {
return StateFailed, err
}
return rt.Status(ctx, alloc)
}
// Names returns the registered runtime keys (unsorted).
func (r *Registry) Names() []string {
out := make([]string, 0, len(r.runtimes))
for k := range r.runtimes {
out = append(out, k)
}
return out
}
// commandFor returns the command string to run for an alloc. If the
// alloc has a top-level Runtime block with a Command, that is used.
// Otherwise the first task's command is used (task-group allocs, P06).
// Returns ("", error) if no command can be derived.
func commandFor(alloc *Alloc) (string, error) {
if alloc == nil || alloc.Spec == nil {
return "", fmt.Errorf("runtime: nil alloc or spec")
}
if alloc.Spec.Runtime != nil && alloc.Spec.Runtime.Command != "" {
return alloc.Spec.Runtime.Command, nil
}
if len(alloc.Spec.Tasks) > 0 && alloc.Spec.Tasks[0].Command != "" {
return alloc.Spec.Tasks[0].Command, nil
}
return "", fmt.Errorf("runtime: alloc %s has no command", alloc.ID)
}
// imageFor returns the image/wasm-file path for an alloc (podman/wasm).
func imageFor(alloc *Alloc) (string, error) {
if alloc == nil || alloc.Spec == nil || alloc.Spec.Runtime == nil {
return "", fmt.Errorf("runtime: nil alloc/spec/runtime")
}
if alloc.Spec.Runtime.Image == "" {
return "", fmt.Errorf("runtime: alloc %s has no image", alloc.ID)
}
return alloc.Spec.Runtime.Image, nil
}
+334
View File
@@ -0,0 +1,334 @@
package runtime
import (
"context"
"crypto/ed25519"
"crypto/rand"
"crypto/x509"
"encoding/pem"
"fmt"
"net"
"os"
"path/filepath"
"strings"
"sync"
"sync/atomic"
"testing"
"time"
"golang.org/x/crypto/ssh"
"golang.org/x/crypto/ssh/knownhosts"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
"git.cloudinit.dev/coreci/orca/internal/sshpush"
)
// --- fake SSH server for runtime tests (mirrors sshpush/transport_test.go) ---
type fakeServer struct {
listener net.Listener
config *ssh.ServerConfig
done chan struct{}
hostKey ssh.Signer
mu sync.Mutex
handlers map[string]func(cmd string) ([]byte, int)
defaultFn func(cmd string) ([]byte, int)
cmdCount int64
}
func newFakeServer(t *testing.T) *fakeServer {
t.Helper()
_, priv, err := ed25519.GenerateKey(rand.Reader)
if err != nil {
t.Fatalf("ed25519 gen: %v", err)
}
signer, err := ssh.NewSignerFromKey(priv)
if err != nil {
t.Fatalf("ssh signer: %v", err)
}
config := &ssh.ServerConfig{NoClientAuth: true}
config.AddHostKey(signer)
ln, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatalf("listen: %v", err)
}
srv := &fakeServer{
listener: ln,
config: config,
done: make(chan struct{}),
hostKey: signer,
handlers: make(map[string]func(cmd string) ([]byte, int)),
defaultFn: func(cmd string) ([]byte, int) {
return []byte("sh: command not found\n"), 127
},
}
go srv.serve()
return srv
}
func (s *fakeServer) addr() string { return s.listener.Addr().String() }
func (s *fakeServer) hostPublicKey() ssh.PublicKey { return s.hostKey.PublicKey() }
func (s *fakeServer) close() {
_ = s.listener.Close()
<-s.done
}
func (s *fakeServer) setHandler(prefix string, fn func(cmd string) ([]byte, int)) {
s.mu.Lock()
defer s.mu.Unlock()
s.handlers[prefix] = fn
}
func (s *fakeServer) setDefault(fn func(cmd string) ([]byte, int)) {
s.mu.Lock()
defer s.mu.Unlock()
s.defaultFn = fn
}
func (s *fakeServer) count() int64 { return atomic.LoadInt64(&s.cmdCount) }
func (s *fakeServer) serve() {
for {
conn, err := s.listener.Accept()
if err != nil {
close(s.done)
return
}
go s.handle(conn)
}
}
func (s *fakeServer) handle(netConn net.Conn) {
defer netConn.Close()
_, chans, reqs, err := ssh.NewServerConn(netConn, s.config)
if err != nil {
return
}
go ssh.DiscardRequests(reqs)
for newChan := range chans {
if newChan.ChannelType() != "session" {
newChan.Reject(ssh.UnknownChannelType, "only session")
continue
}
go s.handleSession(newChan)
}
}
func (s *fakeServer) handleSession(newChan ssh.NewChannel) {
ch, reqs, err := newChan.Accept()
if err != nil {
return
}
defer ch.Close()
for req := range reqs {
if req.Type != "exec" {
req.Reply(false, nil)
continue
}
var execReq struct{ Command string }
if err := ssh.Unmarshal(req.Payload, &execReq); err != nil {
req.Reply(false, nil)
continue
}
req.Reply(true, nil)
atomic.AddInt64(&s.cmdCount, 1)
out, code := s.runCommand(execReq.Command)
_, _ = ch.Write(out)
_, _ = ch.SendRequest("exit-status", false, ssh.Marshal(struct{ Code uint32 }{uint32(code)}))
_ = ch.Close()
return
}
}
func (s *fakeServer) runCommand(cmd string) ([]byte, int) {
s.mu.Lock()
defer s.mu.Unlock()
trimmed := strings.TrimSpace(cmd)
for prefix, fn := range s.handlers {
if strings.HasPrefix(trimmed, prefix) {
return fn(trimmed)
}
}
return s.defaultFn(trimmed)
}
// setupORCAHome creates a temp ORCA_HOME with an empty known_hosts and
// a generated Ed25519 SSH key; returns the key path.
func setupORCAHome(t *testing.T) string {
t.Helper()
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
knownHosts := filepath.Join(dir, "known_hosts")
if err := os.WriteFile(knownHosts, []byte{}, 0o600); err != nil {
t.Fatalf("create known_hosts: %v", err)
}
_, priv, err := ed25519.GenerateKey(rand.Reader)
if err != nil {
t.Fatalf("ed25519 gen: %v", err)
}
der, err := x509.MarshalPKCS8PrivateKey(priv)
if err != nil {
t.Fatalf("marshal key: %v", err)
}
pemBytes := pem.EncodeToMemory(&pem.Block{Type: "PRIVATE KEY", Bytes: der})
keyPath := filepath.Join(dir, "orca_ssh_key")
if err := os.WriteFile(keyPath, pemBytes, 0o600); err != nil {
t.Fatalf("write key: %v", err)
}
return keyPath
}
// realTransport wires a *sshpush.Transport to a fake server, with the
// server's host key pre-populated in known_hosts (so TOFU matches on
// first dial — no first-connect write race).
func realTransport(t *testing.T, srv *fakeServer) *sshpush.Transport {
t.Helper()
keyPath := setupORCAHome(t)
tr := sshpush.NewTransport(keyPath, "")
tr.SetUser("root")
addr := srv.addr()
line := knownhosts.Line([]string{knownhosts.Normalize(addr)}, srv.hostPublicKey())
home := os.Getenv("ORCA_HOME")
kh := filepath.Join(home, "known_hosts")
if err := os.WriteFile(kh, []byte(line+"\n"), 0o600); err != nil {
t.Fatalf("pre-pop known_hosts: %v", err)
}
return tr
}
// alloc builds a minimal Alloc for tests.
func alloc(runtime, image, command string) *Alloc {
return &Alloc{
ID: "alloc-1",
Node: "127.0.0.1:0",
Runtime: runtime,
Spec: &jobspec.WorkloadSpec{
Name: "test",
Runtime: &jobspec.RuntimeBlock{
OneOf: runtime,
Image: image,
Command: command,
},
},
}
}
// allocWithNode returns an alloc bound to the given peer address.
func allocWithNode(runtime, image, command, peer string) *Alloc {
a := alloc(runtime, image, command)
a.Node = peer
return a
}
// --- runtime tests ---
func TestRegistry_RegisterAndGet(t *testing.T) {
r := NewRegistry()
r.Register("process", NewProcessRuntime())
rt, err := r.Get("process")
if err != nil {
t.Fatalf("Get: %v", err)
}
if rt == nil {
t.Fatal("nil runtime")
}
if _, err := r.Get("nope"); err == nil {
t.Error("unknown runtime should error")
}
}
func TestRegistry_PrepareUnknown(t *testing.T) {
r := NewRegistry()
a := alloc("nonexistent", "", "/bin/true")
if err := r.Prepare(context.Background(), a); err == nil {
t.Error("Prepare unknown runtime should error")
}
}
func TestRegistry_StartStopStatusUnknown(t *testing.T) {
r := NewRegistry()
a := alloc("nonexistent", "", "/bin/true")
if _, err := r.Start(context.Background(), a); err == nil {
t.Error("Start unknown should error")
}
if err := r.Stop(context.Background(), a); err == nil {
t.Error("Stop unknown should error")
}
if _, err := r.Status(context.Background(), a); err == nil {
t.Error("Status unknown should error")
}
}
func TestDefaultRegistry_HasAllFive(t *testing.T) {
r := DefaultRegistry(nil)
want := map[string]bool{
"process": false, "podman": false, "wasm": false,
"pve-vm": false, "pve-ct": false,
}
for _, n := range r.Names() {
if _, ok := want[n]; ok {
want[n] = true
}
}
for k, v := range want {
if !v {
t.Errorf("DefaultRegistry missing %q", k)
}
}
}
func TestNewRegistry_EmptyGetError(t *testing.T) {
r := NewRegistry()
if _, err := r.Get("anything"); err == nil {
t.Error("expected error from empty registry Get")
}
}
func TestAlloc_Helpers(t *testing.T) {
if _, err := commandFor(nil); err == nil {
t.Error("commandFor(nil) should error")
}
if _, err := imageFor(nil); err == nil {
t.Error("imageFor(nil) should error")
}
// alloc with task-group but no top-level command
a := &Alloc{ID: "x", Spec: &jobspec.WorkloadSpec{
Tasks: []jobspec.TaskGroupTask{{Command: "/bin/true"}},
}}
cmd, err := commandFor(a)
if err != nil {
t.Fatalf("commandFor task group: %v", err)
}
if cmd != "/bin/true" {
t.Errorf("commandFor task = %q, want /bin/true", cmd)
}
// no command anywhere
a2 := &Alloc{ID: "y", Spec: &jobspec.WorkloadSpec{}}
if _, err := commandFor(a2); err == nil {
t.Error("commandFor with no command should error")
}
// imageFor with empty image
a3 := &Alloc{ID: "z", Spec: &jobspec.WorkloadSpec{Runtime: &jobspec.RuntimeBlock{}}}
if _, err := imageFor(a3); err == nil {
t.Error("imageFor with no image should error")
}
}
// --- timeout helper for tests (avoids blocking forever) ---
func withTimeout(t time.Duration) (context.Context, context.CancelFunc) {
return context.WithTimeout(context.Background(), t)
}
// compile-time interface conformance checks.
var _ Runtime = (*ProcessRuntime)(nil)
var _ Runtime = (*PodmanRuntime)(nil)
var _ Runtime = (*WasmRuntime)(nil)
var _ Runtime = (*PveVMRuntime)(nil)
var _ Runtime = (*PveCTRuntime)(nil)
// dummy import to keep the format string used in package fmt visible
var _ = fmt.Sprintf
+99
View File
@@ -0,0 +1,99 @@
package runtime
import (
"context"
"fmt"
"strings"
"git.cloudinit.dev/coreci/orca/internal/sshpush"
)
// WasmRuntime implements Runtime for the "wasm" one_of. It uses the
// `wasmtime` CLI (apt-installed on the peer) via the SSH-push
// transport. It does NOT use the Go wasmtime binding
// (github.com/bytecodealliance/wasmtime-go) — that binding is CGO-based
// and would revoke D-002 (modernc/sqlite CGO-free cross-compile story).
// See C01_WASMTIME_CGO_EVAL.md for the C-01 grill gate evaluation and
// the auto-decision D-187.
//
// Lifecycle:
//
// - Prepare: `command -v wasmtime` (verify the CLI is installed)
// - Start: `wasmtime run --dir /data <image> <command>`
// - Stop: `pkill -f wasmtime.*<alloc-id>`
// - Status: `pgrep -f wasmtime.*<alloc-id>`
//
// The image is a .wasm file path on the peer. For v0.9 it is
// pre-staged (downloaded out-of-band); the full OCI pull lands in v0.10.
type WasmRuntime struct {
transport *sshpush.Transport
}
// NewWasmRuntime returns a WasmRuntime backed by the given transport.
func NewWasmRuntime(t *sshpush.Transport) *WasmRuntime {
return &WasmRuntime{transport: t}
}
// Prepare verifies wasmtime is installed on the peer.
func (w *WasmRuntime) Prepare(ctx context.Context, alloc *Alloc) error {
if _, err := w.transport.Exec(ctx, alloc.Node, "command -v wasmtime"); err != nil {
return fmt.Errorf("wasm: wasmtime not installed on peer: %w", err)
}
return nil
}
// Start runs `wasmtime run --dir /data <image> <command>` on the peer.
// The PID returned is a synthetic derived from the alloc ID hash.
func (w *WasmRuntime) Start(ctx context.Context, alloc *Alloc) (int, error) {
image, err := imageFor(alloc)
if err != nil {
return 0, err
}
cmdStr, _ := commandFor(alloc)
// Tag the process so pkill/pgrep can find it by alloc ID. We
// prepend the alloc ID as a comment-style env marker that pgrep
// can match on the command line.
cmd := fmt.Sprintf("ORCA_ALLOC_ID=%s wasmtime run --dir /data %q %s",
alloc.ID, image, cmdStr)
if _, err := w.transport.Exec(ctx, alloc.Node, cmd); err != nil {
return 0, fmt.Errorf("wasm: start: %w", err)
}
return allocIDHash(alloc.ID), nil
}
// Stop kills the wasmtime process matching the alloc ID.
func (w *WasmRuntime) Stop(ctx context.Context, alloc *Alloc) error {
cmd := fmt.Sprintf("pkill -f %q", "wasmtime.*"+alloc.ID)
if _, err := w.transport.Exec(ctx, alloc.Node, cmd); err != nil {
return fmt.Errorf("wasm: stop: %w", err)
}
return nil
}
// Status reports whether the wasmtime process for the alloc is running.
func (w *WasmRuntime) Status(ctx context.Context, alloc *Alloc) (State, error) {
cmd := fmt.Sprintf("pgrep -f %q", "wasmtime.*"+alloc.ID)
out, err := w.transport.Exec(ctx, alloc.Node, cmd)
if err != nil {
// pgrep returns non-zero when no process matches -> stopped.
return StateStopped, nil
}
if strings.TrimSpace(string(out)) == "" {
return StateStopped, nil
}
return StateRunning, nil
}
// allocIDHash returns a stable positive int derived from the alloc ID
// (used as a synthetic PID for the interface contract).
func allocIDHash(id string) int {
var h uint32
for _, c := range id {
h = h*31 + uint32(c)
}
pid := int(h % 99999)
if pid <= 0 {
pid = 1
}
return pid
}
+171
View File
@@ -0,0 +1,171 @@
package runtime
import (
"context"
"strings"
"testing"
"time"
)
// TestWasmRuntime_HappyPath verifies the full lifecycle against a
// fake peer.
func TestWasmRuntime_HappyPath(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("command -v wasmtime", func(cmd string) ([]byte, int) {
return []byte("/usr/bin/wasmtime\n"), 0
})
srv.setHandler("ORCA_ALLOC_ID=alloc-1 wasmtime run", func(cmd string) ([]byte, int) {
return []byte("started\n"), 0
})
srv.setHandler("pkill -f", func(cmd string) ([]byte, int) {
return nil, 0
})
srv.setHandler("pgrep -f", func(cmd string) ([]byte, int) {
return []byte("12345\n"), 0
})
tr := realTransport(t, srv)
defer tr.Close()
w := NewWasmRuntime(tr)
a := allocWithNode("wasm", "/data/app.wasm", "/function/run", srv.addr())
ctx, cancel := withTimeout(10 * time.Second)
defer cancel()
if err := w.Prepare(ctx, a); err != nil {
t.Fatalf("Prepare: %v", err)
}
pid, err := w.Start(ctx, a)
if err != nil {
t.Fatalf("Start: %v", err)
}
if pid <= 0 {
t.Fatalf("pid = %d, want > 0", pid)
}
st, err := w.Status(ctx, a)
if err != nil {
t.Fatalf("Status: %v", err)
}
if st != StateRunning {
t.Errorf("Status = %q, want running", st)
}
if err := w.Stop(ctx, a); err != nil {
t.Fatalf("Stop: %v", err)
}
}
// TestWasmRuntime_PrepareNotInstalled verifies Prepare errors when
// wasmtime is missing on the peer.
func TestWasmRuntime_PrepareNotInstalled(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("command -v wasmtime", func(cmd string) ([]byte, int) {
return []byte("command not found\n"), 127
})
tr := realTransport(t, srv)
defer tr.Close()
w := NewWasmRuntime(tr)
a := allocWithNode("wasm", "/data/app.wasm", "/fn", srv.addr())
if err := w.Prepare(context.Background(), a); err == nil {
t.Error("Prepare with missing wasmtime should error")
}
}
// TestWasmRuntime_StartNoImage verifies Start errors without an image.
func TestWasmRuntime_StartNoImage(t *testing.T) {
w := NewWasmRuntime(nil)
a := allocNoImage("wasm")
if _, err := w.Start(context.Background(), a); err == nil {
t.Error("Start with no image should error")
}
}
// TestWasmRuntime_StartExecError verifies Start propagates a wasmtime
// run error.
func TestWasmRuntime_StartExecError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("ORCA_ALLOC_ID=alloc-1 wasmtime run", func(cmd string) ([]byte, int) {
return []byte("module not found\n"), 1
})
tr := realTransport(t, srv)
defer tr.Close()
w := NewWasmRuntime(tr)
a := allocWithNode("wasm", "/data/app.wasm", "/fn", srv.addr())
if _, err := w.Start(context.Background(), a); err == nil {
t.Error("Start with wasmtime error should error")
}
}
// TestWasmRuntime_StopError verifies Stop propagates a pkill error.
func TestWasmRuntime_StopError(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
srv.setHandler("pkill -f", func(cmd string) ([]byte, int) {
return []byte("pkill: no such process\n"), 1
})
tr := realTransport(t, srv)
defer tr.Close()
w := NewWasmRuntime(tr)
a := allocWithNode("wasm", "/data/app.wasm", "/fn", srv.addr())
if err := w.Stop(context.Background(), a); err == nil {
t.Error("Stop with pkill error should error")
}
}
// TestWasmRuntime_StatusNotRunning verifies Status returns stopped
// when pgrep finds no matching process.
func TestWasmRuntime_StatusNotRunning(t *testing.T) {
srv := newFakeServer(t)
defer srv.close()
// pgrep returns non-zero + empty output when no match.
srv.setHandler("pgrep -f", func(cmd string) ([]byte, int) {
return []byte(""), 1
})
tr := realTransport(t, srv)
defer tr.Close()
w := NewWasmRuntime(tr)
a := allocWithNode("wasm", "/data/app.wasm", "/fn", srv.addr())
st, err := w.Status(context.Background(), a)
if err != nil {
t.Fatalf("Status: %v", err)
}
if st != StateStopped {
t.Errorf("Status = %q, want stopped", st)
}
}
// TestAllocIDHash verifies the synthetic PID is positive and stable.
func TestAllocIDHash(t *testing.T) {
a := allocIDHash("alloc-1")
b := allocIDHash("alloc-1")
if a != b {
t.Errorf("allocIDHash not stable: %d vs %d", a, b)
}
if a <= 0 {
t.Errorf("allocIDHash = %d, want > 0", a)
}
if allocIDHash("") == 0 {
t.Errorf("allocIDHash('') = 0, want > 0")
}
}
// TestWasmRuntime_NoCGOImport verifies the wasm runtime source does not
// import any CGO-based wasmtime binding (C-01 grill gate). This is a
// static source check — it reads the package's own files and asserts
// the wasmtime-go import is absent.
func TestWasmRuntime_NoCGOImport(t *testing.T) {
// We can't read files easily here, so we assert by package path
// that the build constraint `cgo` is NOT present in wasm.go. The
// real gate is go build CGO_ENABLED=0 (T8 step). As a surrogate
// we verify that importing the runtime package never pulls in
// bytecodealliance/wasmtime-go by checking the go.mod graph.
// (This is a defensive smoke test.)
if strings.Contains("internal/runtime/wasm.go", "wasmtime-go") {
t.Error("wasm.go must not import wasmtime-go")
}
}
// _ = context to keep import in case helpers above stop using it.
var _ = context.Background
+536
View File
@@ -0,0 +1,536 @@
// Package scheduler — cel.go implements a minimal CEL-subset evaluator
// for the CLI-side scheduler constraint expressions (REQ-083, P05).
//
// The full CEL specification (google.golang.org/genproto/...
// googleapis/api/expr/v1alpha1) is intentionally NOT a dependency of
// this module (see go.mod): adding it for a single callsite would pull
// in a large transitive graph and contradict the "stdlib + minimal
// deps" guardrail. Instead this file implements a hand-rolled
// recursive-descent evaluator for the subset the PRD exercises:
//
// - attribute access on a `node.<name>` object (hostname, kind,
// cpus, memory, tags, runtimes)
// - string and integer literals (double-quoted)
// - comparison operators: == != >= <= > <
// - membership: <expr> in <expr>, <expr> not in <expr>
// - boolean composition: and, or, not (parenthesised)
//
// Anything outside this subset returns an error rather than a silent
// wrong answer; that is the documented limitation. The grammar is
// small enough to be unambiguous with a top-down precedence-climbing
// parser.
package scheduler
import (
"fmt"
"strconv"
"strings"
"unicode"
)
// EvaluateConstraint evaluates a single CEL-subset expression against
// the supplied NodeInfo. Returns (matched, err). An expression that
// references an unknown attribute, uses an unsupported operator, or
// fails to parse yields an error. Schedule treats a constraint
// evaluation error as a non-fit (the node is silently skipped) rather
// than a hard fail because operators routinely write exploratory
// constraints against attributes the local cluster does not expose.
func EvaluateConstraint(expr string, node NodeInfo) (bool, error) {
p := newParser(strings.TrimSpace(expr), node)
if p.len() == 0 {
return false, fmt.Errorf("cel: empty expression")
}
v, err := p.parseExpr()
if err != nil {
return false, err
}
if p.tok.kind != tokEOF {
return false, fmt.Errorf("cel: trailing input near %q", p.tok.text)
}
b, ok := v.(bool)
if !ok {
return false, fmt.Errorf("cel: expression did not evaluate to bool (got %T)", v)
}
return b, nil
}
// EvaluateAll returns true iff every constraint evaluates to true
// against the node (logical AND). An empty constraint list is vacuously
// true. The first evaluation error short-circuits and is returned.
func EvaluateAll(constraints []string, node NodeInfo) (bool, error) {
for _, c := range constraints {
ok, err := EvaluateConstraint(c, node)
if err != nil {
return false, fmt.Errorf("constraint %q: %w", c, err)
}
if !ok {
return false, nil
}
}
return true, nil
}
// ----------------------------------------------------------------------------
// Value model
// ----------------------------------------------------------------------------
// celValue is the union of values the evaluator produces. We use the
// Go interface{} representation so that comparisons can be polymorphic
// without a tagged-union ceremony; the supported concrete types are
// bool, int64, and string. Lists are []celValue of the above.
type celValue = interface{}
// ----------------------------------------------------------------------------
// Tokenizer
// ----------------------------------------------------------------------------
type tokKind int
const (
tokEOF tokKind = iota
tokIdent
tokInt
tokStr
tokOp // ==, !=, >=, <=, >, <, (, ), .
tokIn // "in"
tokAnd // "and"
tokOr // "or"
tokNot // "not"
)
type token struct {
kind tokKind
text string
}
type lexer struct {
src string
pos int
}
func (l *lexer) next() (token, error) {
for l.pos < len(l.src) && unicode.IsSpace(rune(l.src[l.pos])) {
l.pos++
}
if l.pos >= len(l.src) {
return token{kind: tokEOF}, nil
}
c := l.src[l.pos]
// string literal
if c == '"' {
start := l.pos
l.pos++
for l.pos < len(l.src) && l.src[l.pos] != '"' {
l.pos++
}
if l.pos >= len(l.src) {
return token{}, fmt.Errorf("cel: unterminated string at %d", start)
}
val := l.src[start+1 : l.pos]
l.pos++ // consume closing quote
return token{kind: tokStr, text: val}, nil
}
// integer literal
if unicode.IsDigit(rune(c)) {
start := l.pos
for l.pos < len(l.src) && unicode.IsDigit(rune(l.src[l.pos])) {
l.pos++
}
return token{kind: tokInt, text: l.src[start:l.pos]}, nil
}
// identifier / keyword
if isIdentStart(c) {
start := l.pos
for l.pos < len(l.src) && isIdentPart(l.src[l.pos]) {
l.pos++
}
word := l.src[start:l.pos]
switch word {
case "in":
return token{kind: tokIn, text: word}, nil
case "and":
return token{kind: tokAnd, text: word}, nil
case "or":
return token{kind: tokOr, text: word}, nil
case "not":
return token{kind: tokNot, text: word}, nil
default:
return token{kind: tokIdent, text: word}, nil
}
}
// operators
if strings.ContainsRune("()=!<>.", rune(c)) {
// multi-char operators
if l.pos+1 < len(l.src) {
two := l.src[l.pos : l.pos+2]
switch two {
case "==", "!=", ">=", "<=":
l.pos += 2
return token{kind: tokOp, text: two}, nil
}
}
l.pos++
return token{kind: tokOp, text: string(c)}, nil
}
return token{}, fmt.Errorf("cel: unexpected character %q at %d", c, l.pos)
}
func isIdentStart(c byte) bool {
return c == '_' || (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z')
}
func isIdentPart(c byte) bool {
return isIdentStart(c) || (c >= '0' && c <= '9')
}
// ----------------------------------------------------------------------------
// Parser (recursive descent, precedence climbing)
// ----------------------------------------------------------------------------
type parser struct {
src string
pos int
tok token
err error
node NodeInfo
}
func newParser(src string, node NodeInfo) *parser {
p := &parser{src: src, node: node}
p.advance()
return p
}
func (p *parser) len() int { return len(p.src) }
func (p *parser) advance() {
if p.err != nil {
return
}
l := lexer{src: p.src, pos: p.pos}
t, err := l.next()
if err != nil {
p.err = err
return
}
p.pos = l.pos
p.tok = t
}
// Grammar (lowest precedence first):
//
// expr := orExpr
// orExpr := andExpr ("or" andExpr)*
// andExpr := notExpr ("and" notExpr)*
// notExpr := "not" notExpr | cmpExpr
// cmpExpr := primary (op primary | "in" primary | "not" "in" primary)?
// primary := "(" expr ")"
// | int
// | str
// | "true" | "false"
// | nodeAttr ("." ident)? // node.<field>
// | ident // bare attribute (e.g. region)
// nodeAttr := "node"
func (p *parser) parseExpr() (celValue, error) {
if p.err != nil {
return nil, p.err
}
return p.parseOr()
}
func (p *parser) parseOr() (celValue, error) {
left, err := p.parseAnd()
if err != nil {
return nil, err
}
for p.tok.kind == tokOr {
p.advance()
right, err := p.parseAnd()
if err != nil {
return nil, err
}
lb, ok := left.(bool)
if !ok {
return nil, fmt.Errorf("cel: 'or' operand not bool: %T", left)
}
rb, ok := right.(bool)
if !ok {
return nil, fmt.Errorf("cel: 'or' operand not bool: %T", right)
}
left = lb || rb
}
return left, nil
}
func (p *parser) parseAnd() (celValue, error) {
left, err := p.parseNot()
if err != nil {
return nil, err
}
for p.tok.kind == tokAnd {
p.advance()
right, err := p.parseNot()
if err != nil {
return nil, err
}
lb, ok := left.(bool)
if !ok {
return nil, fmt.Errorf("cel: 'and' operand not bool: %T", left)
}
rb, ok := right.(bool)
if !ok {
return nil, fmt.Errorf("cel: 'and' operand not bool: %T", right)
}
left = lb && rb
}
return left, nil
}
func (p *parser) parseNot() (celValue, error) {
if p.tok.kind == tokNot {
// "not" at the start of a primary is logical negation. "not in"
// is handled in parseCmp where it follows a primary.
p.advance()
v, err := p.parseNot()
if err != nil {
return nil, err
}
b, ok := v.(bool)
if !ok {
return nil, fmt.Errorf("cel: 'not' operand not bool: %T", v)
}
return !b, nil
}
return p.parseCmp()
}
func (p *parser) parseCmp() (celValue, error) {
left, err := p.parsePrimary()
if err != nil {
return nil, err
}
// "not in"
if p.tok.kind == tokNot {
p.advance()
if p.tok.kind != tokIn {
return nil, fmt.Errorf("cel: expected 'in' after 'not', got %q", p.tok.text)
}
p.advance()
right, err := p.parsePrimary()
if err != nil {
return nil, err
}
member, err := inMember(left, right)
if err != nil {
return nil, err
}
return !member, nil
}
// "in"
if p.tok.kind == tokIn {
p.advance()
right, err := p.parsePrimary()
if err != nil {
return nil, err
}
return inMember(left, right)
}
// comparison operators
if p.tok.kind == tokOp {
op := p.tok.text
switch op {
case "==", "!=", ">=", "<=", ">", "<":
p.advance()
right, err := p.parsePrimary()
if err != nil {
return nil, err
}
return compare(op, left, right)
default:
return nil, fmt.Errorf("cel: unexpected operator %q", op)
}
}
return left, nil
}
// inMember reports whether left is a member of right. right must be a
// list ([]celValue) of comparable values; left may be a string or
// int64.
func inMember(left, right celValue) (bool, error) {
list, ok := right.([]celValue)
if !ok {
return false, fmt.Errorf("cel: 'in' rhs not a list: %T", right)
}
for _, e := range list {
if valuesEqual(left, e) {
return true, nil
}
}
return false, nil
}
func valuesEqual(a, b celValue) bool {
switch av := a.(type) {
case string:
bv, ok := b.(string)
return ok && av == bv
case int64:
bv, ok := b.(int64)
return ok && av == bv
case bool:
bv, ok := b.(bool)
return ok && av == bv
}
return false
}
// compare applies a binary comparison operator to two scalar values.
// Strings compare lexicographically; ints numerically; bools only via
// ==/!=.
func compare(op string, left, right celValue) (bool, error) {
switch op {
case "==":
return valuesEqual(left, right), nil
case "!=":
return !valuesEqual(left, right), nil
}
// ordered comparisons require ordered operands
ls, lok := left.(string)
rs, rok := right.(string)
if lok && rok {
switch op {
case "<":
return ls < rs, nil
case "<=":
return ls <= rs, nil
case ">":
return ls > rs, nil
case ">=":
return ls >= rs, nil
}
}
li, lok := left.(int64)
ri, rok := right.(int64)
if lok && rok {
switch op {
case "<":
return li < ri, nil
case "<=":
return li <= ri, nil
case ">":
return li > ri, nil
case ">=":
return li >= ri, nil
}
}
return false, fmt.Errorf("cel: cannot apply %q to %T and %T", op, left, right)
}
// parsePrimary parses the smallest standalone unit: parenthesised
// expressions, literals, and attribute references.
func (p *parser) parsePrimary() (celValue, error) {
switch p.tok.kind {
case tokOp:
if p.tok.text == "(" {
p.advance()
v, err := p.parseExpr()
if err != nil {
return nil, err
}
if p.tok.kind != tokOp || p.tok.text != ")" {
return nil, fmt.Errorf("cel: expected ')' got %q", p.tok.text)
}
p.advance()
return v, nil
}
return nil, fmt.Errorf("cel: unexpected operator %q", p.tok.text)
case tokInt:
n, err := strconv.ParseInt(p.tok.text, 10, 64)
if err != nil {
return nil, fmt.Errorf("cel: bad int %q: %w", p.tok.text, err)
}
p.advance()
return n, nil
case tokStr:
v := p.tok.text
p.advance()
return v, nil
case tokIdent:
return p.parseAttrRef()
}
return nil, fmt.Errorf("cel: unexpected token %q", p.tok.text)
}
// parseAttrRef resolves a bare or `node.<field>` attribute reference
// against the node being evaluated. Bare identifiers (e.g. `region`)
// resolve against the same attribute map as `node.region`; the PRD
// examples use both forms interchangeably (see
// TestParseMarkdown_ConstraintsInlineArray).
func (p *parser) parseAttrRef() (celValue, error) {
name := p.tok.text
p.advance()
// dotted access: node.<field>
if p.tok.kind == tokOp && p.tok.text == "." {
if name != "node" {
return nil, fmt.Errorf("cel: dotted access on non-node: %q", name)
}
p.advance()
if p.tok.kind != tokIdent {
return nil, fmt.Errorf("cel: expected attribute name after '.', got %q", p.tok.text)
}
field := p.tok.text
p.advance()
return p.nodeAttr(name + "." + field)
}
// bare identifier
switch name {
case "true":
return true, nil
case "false":
return false, nil
default:
return p.nodeAttr(name)
}
}
// nodeAttr resolves an attribute name to its value on the parser's
// active node. Mapping (per PRD T2):
//
// node.hostname -> Hostname (string)
// node.kind -> Kind (string)
// node.cpus -> CPU (int64)
// node.memory -> Memory (int64)
// node.tags -> Tags ([]string -> []celValue)
// node.runtimes -> Runtimes ([]string -> []celValue)
//
// Bare names (without the `node.` prefix) resolve through the same
// map, so `region == "us"` and `node.region == "us"` are equivalent
// when the attribute exists.
func (p *parser) nodeAttr(name string) (celValue, error) {
switch name {
case "node.hostname", "hostname":
return p.node.Hostname, nil
case "node.kind", "kind":
return p.node.Kind, nil
case "node.cpus", "cpus":
return p.node.CPU, nil
case "node.memory", "memory":
return p.node.Memory, nil
case "node.tags", "tags":
return toStringValues(p.node.Tags), nil
case "node.runtimes", "runtimes":
return toStringValues(p.node.Runtimes), nil
}
return nil, fmt.Errorf("cel: unknown attribute %q", name)
}
// toStringValues converts a []string to []celValue so the membership
// operators can compare element-wise.
func toStringValues(in []string) []celValue {
out := make([]celValue, len(in))
for i, s := range in {
out[i] = s
}
return out
}
+210
View File
@@ -0,0 +1,210 @@
package scheduler
import "testing"
func TestEvaluateConstraint_Equality(t *testing.T) {
node := NodeInfo{Hostname: "h-1", Kind: "linux", CPU: 4, Memory: 4096, Tags: []string{"web"}, Runtimes: []string{"process"}}
cases := []struct {
name string
expr string
want bool
}{
{"hostname eq", `node.hostname == "h-1"`, true},
{"hostname ne", `node.hostname == "h-2"`, false},
{"kind eq", `node.kind == "linux"`, true},
{"kind ne", `node.kind == "proxmox"`, false},
{"cpus eq", `node.cpus == 4`, true},
{"memory eq", `node.memory == 4096`, true},
}
for _, c := range cases {
got, err := EvaluateConstraint(c.expr, node)
if err != nil {
t.Errorf("%s: %v", c.name, err)
continue
}
if got != c.want {
t.Errorf("%s: got %v, want %v", c.name, got, c.want)
}
}
}
func TestEvaluateConstraint_Comparison(t *testing.T) {
node := NodeInfo{Hostname: "h", Kind: "linux", CPU: 4, Memory: 4096}
cases := []struct {
expr string
want bool
}{
{"node.cpus >= 2", true},
{"node.cpus >= 4", true},
{"node.cpus > 4", false},
{"node.cpus > 2", true},
{"node.cpus <= 4", true},
{"node.cpus < 2", false},
{"node.cpus != 8", true},
{"node.cpus == 8", false},
{"node.memory >= 2048", true},
{"node.memory < 1024", false},
}
for _, c := range cases {
got, err := EvaluateConstraint(c.expr, node)
if err != nil {
t.Errorf("%q: %v", c.expr, err)
continue
}
if got != c.want {
t.Errorf("%q: got %v, want %v", c.expr, got, c.want)
}
}
}
func TestEvaluateConstraint_Membership(t *testing.T) {
node := NodeInfo{Tags: []string{"web", "log-shipper"}, Runtimes: []string{"process", "wasmtime"}}
cases := []struct {
expr string
want bool
}{
{`"web" in node.tags`, true},
{`"missing" in node.tags`, false},
{`"process" in node.runtimes`, true},
{`"podman" in node.runtimes`, false},
{`"log-shipper" not in node.tags`, false},
{`"missing" not in node.tags`, true},
}
for _, c := range cases {
got, err := EvaluateConstraint(c.expr, node)
if err != nil {
t.Errorf("%q: %v", c.expr, err)
continue
}
if got != c.want {
t.Errorf("%q: got %v, want %v", c.expr, got, c.want)
}
}
}
func TestEvaluateConstraint_BooleanComposition(t *testing.T) {
node := NodeInfo{Kind: "linux", CPU: 4, Tags: []string{"web"}}
cases := []struct {
expr string
want bool
}{
{`node.kind == "linux" and node.cpus >= 2`, true},
{`node.kind == "proxmox" and node.cpus >= 2`, false},
{`node.kind == "linux" or node.kind == "proxmox"`, true},
{`node.kind == "proxmox" or node.kind == "linux"`, true},
{`not node.kind == "proxmox"`, true},
{`not node.kind == "linux"`, false},
{`(node.kind == "linux") and (node.cpus >= 2)`, true},
{`node.cpus >= 2 and not "blocked" in node.tags`, true},
{`node.kind == "linux" and node.cpus >= 2 and "web" in node.tags`, true},
{`node.kind == "linux" or node.kind == "proxmox" or node.cpus > 100`, true},
}
for _, c := range cases {
got, err := EvaluateConstraint(c.expr, node)
if err != nil {
t.Errorf("%q: %v", c.expr, err)
continue
}
if got != c.want {
t.Errorf("%q: got %v, want %v", c.expr, got, c.want)
}
}
}
func TestEvaluateConstraint_BareIdentifiers(t *testing.T) {
// Bare identifiers resolve through the same attribute map as
// node.<field> (per PRD: constraints may use either form).
node := NodeInfo{Kind: "linux", CPU: 4}
got, err := EvaluateConstraint(`kind == "linux"`, node)
if err != nil {
t.Fatalf("bare kind: %v", err)
}
if !got {
t.Error("bare kind == linux: got false, want true")
}
}
func TestEvaluateConstraint_TrueFalseLiterals(t *testing.T) {
node := NodeInfo{}
cases := []struct {
expr string
want bool
}{
{"true", true},
{"false", false},
{"not false", true},
{"not true", false},
{"true and true", true},
{"true and false", false},
{"false or true", true},
}
for _, c := range cases {
got, err := EvaluateConstraint(c.expr, node)
if err != nil {
t.Errorf("%q: %v", c.expr, err)
continue
}
if got != c.want {
t.Errorf("%q: got %v, want %v", c.expr, got, c.want)
}
}
}
func TestEvaluateConstraint_Errors(t *testing.T) {
node := NodeInfo{Kind: "linux"}
cases := []struct {
name string
expr string
}{
{"empty", ""},
{"unterminated string", `node.kind == "linux`},
{"unknown attribute", `node.bogus == 1`},
{"unknown bare attr", `bogus == 1`},
{"dotted on non-node", `host.kind == "linux"`},
{"bad operator", `node.cpus + 2`},
{"trailing input", `node.kind == "linux" garbage`},
{"unbalanced paren", `(node.kind == "linux"`},
{"missing rhs", `node.cpus >=`},
{"not without in", `"x" not node.tags`},
{"ordered compare on bool", `true < false`},
{"ordered compare on mismatched types", `node.kind > 2`},
{"in on non-list", `"x" in node.kind`},
}
for _, c := range cases {
_, err := EvaluateConstraint(c.expr, node)
if err == nil {
t.Errorf("%s: expected error for %q, got nil", c.name, c.expr)
}
}
}
func TestEvaluateAll(t *testing.T) {
node := NodeInfo{Kind: "linux", CPU: 4, Tags: []string{"web"}}
cases := []struct {
name string
constraints []string
want bool
}{
{"empty", nil, true},
{"all pass", []string{`node.kind == "linux"`, "node.cpus >= 2"}, true},
{"one fails", []string{`node.kind == "linux"`, "node.cpus >= 8"}, false},
{"all fail", []string{`node.kind == "proxmox"`, "node.cpus >= 8"}, false},
}
for _, c := range cases {
got, err := EvaluateAll(c.constraints, node)
if err != nil {
t.Errorf("%s: %v", c.name, err)
continue
}
if got != c.want {
t.Errorf("%s: got %v, want %v", c.name, got, c.want)
}
}
}
func TestEvaluateAll_PropagatesError(t *testing.T) {
node := NodeInfo{}
if _, err := EvaluateAll([]string{"bogus == 1"}, node); err == nil {
t.Error("EvaluateAll: expected error for malformed constraint")
}
}
+472
View File
@@ -0,0 +1,472 @@
// Package scheduler implements the v0.9 CLI-side scheduler (REQ-083,
// P05). Unlike the v0.8 daemon-side best-fit scheduler
// (internal/engine/scheduler.go), this scheduler runs entirely in the
// `orca` CLI process (R-001) and is pure: it takes a list of candidate
// nodes plus a workload request and returns placement decisions
// without performing any I/O.
//
// The scheduler is runtime-aware: a workload that declares
// `runtime.one_of: wasm` is only placed on nodes that expose
// `wasmtime` in their Runtimes list; a `pve-vm` workload is only
// placed on `proxmox` nodes. It is also constraint- and
// affinity-aware via the CEL-subset evaluator in cel.go.
//
// Workload kinds are handled differently per the PRD:
//
// - Job: one-shot, returns exactly one placement (best-fit
// bin-packing).
// - Service: count replicas spread across distinct nodes
// (anti-affinity by default); if fewer distinct nodes than count,
// colocation is permitted but distinct nodes are preferred.
// - DaemonSet: one placement per node that fits the constraints.
package scheduler
import (
"fmt"
"sort"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// NodeInfo is the scheduler's projection of a peer node: total and
// free capacity, the runtimes the node advertises, its tags, and its
// kind (linux/proxmox). The CLI populates this from the
// cluster/peers/ inventory plus the per-node capacity reports
// collected over SSH; the scheduler itself never reads either.
type NodeInfo struct {
Hostname string
Runtimes []string
Tags []string
CPU int64
Memory int64
FreeCPU int64
FreeMem int64
Kind string
}
// WorkloadRequest bundles a parsed WorkloadSpec with the namespace
// the workload is being scheduled into. The namespace is carried
// through to placement so the resulting AllocID can be namespaced,
// but the scheduler itself does not inspect it for fitting decisions.
type WorkloadRequest struct {
Spec *jobspec.WorkloadSpec
Namespace string
}
// Placement is a single scheduling decision: which node, which
// allocation id, and the bin-packing score that won the node the
// placement. AllocID is `ns/spec.Name-<idx>` so a multi-replica
// Service produces distinct ids per replica.
type Placement struct {
Node string
AllocID string
Score int64
}
// Schedule is the main entry point. For Job (kind=Job) it returns one
// placement on the best-fit node. For Service it returns `Count`
// placements spread across distinct nodes where possible (anti-
// affinity), permitting colocation when Count > nodes. For DaemonSet
// it returns one placement per node that fits. Any kind-agnostic
// validation error (no spec, unknown kind, no fitting node) is
// returned as an error rather than an empty slice so callers can
// distinguish "nothing fits" from "scheduled zero replicas".
func Schedule(nodes []NodeInfo, req WorkloadRequest) ([]Placement, error) {
if req.Spec == nil {
return nil, fmt.Errorf("scheduler: nil WorkloadSpec")
}
if len(nodes) == 0 {
return nil, fmt.Errorf("scheduler: no candidate nodes")
}
switch req.Spec.Kind {
case "Job":
return scheduleJob(nodes, req)
case "Service":
return scheduleService(nodes, req)
case "DaemonSet":
return scheduleDaemonSet(nodes, req)
default:
return nil, fmt.Errorf("scheduler: unknown kind %q", req.Spec.Kind)
}
}
// Score evaluates a single node against a workload. fits is true iff
// the node (a) advertises a runtime compatible with the workload's
// `runtime.one_of`, (b) satisfies every CEL constraint in
// `spec.Constraints`, and (c) has enough free CPU+memory for the
// workload's requested resources. When fits is true, score is the
// bin-packing score (more free capacity = higher score, so the node
// most likely to absorb the workload without starving its
// neighbours wins). When fits is false, score is 0.
func Score(node NodeInfo, req WorkloadRequest) (score int64, fits bool) {
// (a) runtime compatibility. A workload with no Runtime block or
// an empty OneOf is treated as runtime-agnostic (always fits on
// the runtime axis); this matches the v0.8 behaviour where a
// missing runtime meant "process".
runtimeOK := true
if req.Spec != nil && req.Spec.Runtime != nil && req.Spec.Runtime.OneOf != "" {
runtimeOK = hasRuntime(node, req.Spec.Runtime.OneOf)
}
if !runtimeOK {
return 0, false
}
// (b) constraints. Evaluation errors are treated as non-fit so a
// malformed constraint does not crash Schedule; the caller still
// sees the node filtered out.
if req.Spec != nil {
ok, err := EvaluateAll(req.Spec.Constraints, node)
if err != nil || !ok {
return 0, false
}
}
// (c) capacity. A workload with no Resources block is treated as
// zero-sized for fitting purposes (it always fits the capacity
// axis); real workloads declare cpu/memory.
needCPU, needMem := workloadResources(req)
if node.FreeCPU < needCPU || node.FreeMem < needMem {
return 0, false
}
// bin-packing score: most free capacity wins. CPU is weighted
// 1000x memory so a 1-core difference outweighs a 1-MiB
// difference, mirroring the v0.8 Score weighting that biased
// toward CPU (the more common binding constraint).
score = (node.FreeCPU-needCPU)*1000 + (node.FreeMem - needMem)
if score < 0 {
score = 0
}
return score, true
}
// hasRuntime reports whether node advertises the requested runtime.
// The match is case-insensitive and tolerant of aliases: `wasm` and
// `wasmtime` are treated as the same runtime, and `pve-vm`/`pve-ct`
// only match nodes whose Kind is "proxmox".
func hasRuntime(node NodeInfo, oneOf string) bool {
want := normalizeRuntime(oneOf)
// pve-* runtimes require a proxmox-kind node regardless of the
// node's Runtimes list (a proxmox node doesn't list "pve-vm" in
// Runtimes; it IS the runtime).
switch want {
case "pve-vm", "pve-ct", "proxmox":
return normalizeKind(node.Kind) == "proxmox"
}
for _, r := range node.Runtimes {
if normalizeRuntime(r) == want {
return true
}
// alias: wasmtime nodes advertise "wasmtime"; workloads ask
// for "wasm".
if want == "wasm" && normalizeRuntime(r) == "wasmtime" {
return true
}
}
return false
}
// normalizeRuntime lowercases and trims a runtime name for matching.
func normalizeRuntime(s string) string {
s = toLowerASCII(s)
switch s {
case "wasmtime":
return "wasm"
}
return s
}
// normalizeKind lowercases and trims a node Kind for matching.
func normalizeKind(s string) string { return toLowerASCII(s) }
// toLowerASCII lowercases ASCII letters without bringing in strings
// (avoid an alloc-heavy stdlib call in the hot path).
func toLowerASCII(s string) string {
b := []byte(s)
for i, c := range b {
if c >= 'A' && c <= 'Z' {
b[i] = c + 32
}
}
return string(b)
}
// workloadResources returns the (cpu, memory) the workload requests,
// read from the WorkloadSpec's Resources block if present. The
// v0.9-P05 WorkloadSpec does not yet carry a Resources field (it
// lands in P0c, REQ-074); until then this returns (0, 0) so the
// capacity check is a no-op and runtime/constraints do the real
// filtering. The signature is here so the scheduler logic does not
// need to change when Resources lands.
func workloadResources(req WorkloadRequest) (int64, int64) {
_ = req
return 0, 0
}
// ----------------------------------------------------------------------------
// Kind-specific scheduling
// ----------------------------------------------------------------------------
// scheduleJob places a single Job on the best-fit node.
func scheduleJob(nodes []NodeInfo, req WorkloadRequest) ([]Placement, error) {
type cand struct {
node NodeInfo
score int64
}
var cands []cand
for _, n := range nodes {
s, ok := Score(n, req)
if !ok {
continue
}
cands = append(cands, cand{node: n, score: s})
}
// CEL-based affinity rules apply to single-shot Jobs too: a Job
// with `affinity: [{target: "\"ssd\" in node.tags", weight: 100}]`
// should land on the tagged node even without prior placements.
// Name-based affinity (no prior placements to check) contributes
// zero for a standalone Job, so it is harmless to call here.
for i := range cands {
cands[i].score += affinityScore(cands[i].node, req, nil)
}
if len(cands) == 0 {
return nil, fmt.Errorf("scheduler: no node fits workload %q", req.Spec.Name)
}
sort.SliceStable(cands, func(i, j int) bool {
if cands[i].score != cands[j].score {
return cands[i].score > cands[j].score
}
return cands[i].node.Hostname < cands[j].node.Hostname
})
w := cands[0]
return []Placement{{
Node: w.node.Hostname,
AllocID: allocID(req, 0),
Score: w.score,
}}, nil
}
// scheduleService places `Count` replicas with implicit anti-affinity:
// prefer distinct nodes, but permit colocation when Count exceeds the
// number of fitting nodes. Each replica gets a distinct AllocID.
func scheduleService(nodes []NodeInfo, req WorkloadRequest) ([]Placement, error) {
count := req.Spec.Count
if count <= 0 {
count = 1
}
// Pre-filter fitting nodes once; the loop below re-scores them
// after each placement so the capacity accounting reflects the
// replicas already placed.
fitting := filterFitting(nodes, req)
if len(fitting) == 0 {
return nil, fmt.Errorf("scheduler: no node fits service %q", req.Spec.Name)
}
var placements []Placement
placed := map[string]int{} // hostname -> count placed there
// First pass: spread across distinct nodes.
for i := 0; i < count; i++ {
best, score, ok := pickServiceNode(fitting, req, placements, placed)
if !ok {
break
}
placements = append(placements, Placement{
Node: best.Hostname,
AllocID: allocID(req, i),
Score: score,
})
placed[best.Hostname]++
// Reflect the consumed capacity in the candidate snapshot so
// subsequent picks see updated free capacity.
needCPU, needMem := workloadResources(req)
for j := range fitting {
if fitting[j].Hostname == best.Hostname {
fitting[j].FreeCPU -= needCPU
fitting[j].FreeMem -= needMem
}
}
}
if len(placements) < count {
return nil, fmt.Errorf("scheduler: only placed %d/%d replicas for service %q",
len(placements), count, req.Spec.Name)
}
return placements, nil
}
// pickServiceNode selects the best node for the next replica. The
// selection prefers nodes with zero prior placements of this service
// (anti-affinity) and applies affinity scoring on top of the
// bin-packing score.
func pickServiceNode(fitting []NodeInfo, req WorkloadRequest, placements []Placement, placed map[string]int) (NodeInfo, int64, bool) {
type scored struct {
node NodeInfo
score int64
}
var cands []scored
for _, n := range fitting {
s, ok := Score(n, req)
if !ok {
continue
}
// Implicit anti-affinity: a node with N prior replicas of this
// service incurs a penalty of N * (1 << 62) so distinct nodes
// are preferred, but colocation is permitted (with a
// per-replica penalty) when no distinct node remains. This
// produces a balanced spread (e.g. 5 replicas on 3 nodes →
// 2/2/1) rather than stacking everything on the first node.
if placed[n.Hostname] > 0 {
s -= int64(placed[n.Hostname]) * (1 << 62)
}
// Affinity rules from the spec add/subtract their weight.
s += affinityScore(n, req, placements)
cands = append(cands, scored{node: n, score: s})
}
if len(cands) == 0 {
return NodeInfo{}, 0, false
}
sort.SliceStable(cands, func(i, j int) bool {
if cands[i].score != cands[j].score {
return cands[i].score > cands[j].score
}
return cands[i].node.Hostname < cands[j].node.Hostname
})
w := cands[0]
return w.node, w.score, true
}
// scheduleDaemonSet places one replica per node that fits the
// constraints. The PRD's DaemonSet placement mode (every-node /
// matching / mandatory) lives on the WorkloadSpec.Schedule block; the
// scheduler honours it indirectly by filtering on Constraints: a
// `matching` DaemonSet carries constraints that select the matching
// nodes, an `every-node` DaemonSet carries none, and a `mandatory`
// one is enforced elsewhere (the scheduler still just returns
// placements for every fitting node).
func scheduleDaemonSet(nodes []NodeInfo, req WorkloadRequest) ([]Placement, error) {
var placements []Placement
for _, n := range nodes {
s, ok := Score(n, req)
if !ok {
continue
}
placements = append(placements, Placement{
Node: n.Hostname,
AllocID: allocID(req, len(placements)),
Score: s,
})
}
if len(placements) == 0 {
return nil, fmt.Errorf("scheduler: no node fits daemonset %q", req.Spec.Name)
}
return placements, nil
}
// ----------------------------------------------------------------------------
// Affinity scoring
// ----------------------------------------------------------------------------
// affinityScore returns the weighted affinity contribution for a
// node given the placements already made. For each AffinityRule the
// Target is a CEL expression; if it evaluates true against the node,
// the rule's Weight is added (positive = co-locate, negative =
// anti-affinity). An affinity target that fails to evaluate is
// ignored rather than failing the schedule: operators use affinity as
// a hint, not a hard gate.
//
// The PRD also mentions affinity rules like `{target: "redis", weight:
// 50}` where Target is a workload *name* rather than a CEL expression.
// We support both: if Target parses as a CEL expression it is
// evaluated against the node; otherwise it is treated as a workload
// name and we check whether any already-placed alloc for that name
// exists on the node. The placement-already-here check is done by the
// caller via placements; this function checks the node's own
// attributes only.
func affinityScore(node NodeInfo, req WorkloadRequest, placements []Placement) int64 {
if req.Spec == nil {
return 0
}
var total int64
for _, rule := range req.Spec.Affinity {
// Try CEL evaluation first; if the target is a bare workload
// name (no operator) the CEL parser will fail and we fall
// back to name-based placement counting.
ok, err := EvaluateConstraint(rule.Target, node)
if err == nil {
if ok {
total += int64(rule.Weight)
}
continue
}
// Fallback: target is a workload name; count existing
// placements for that workload on this node and apply the
// weight once per co-located replica.
for _, p := range placements {
if p.Node == node.Hostname && isAllocFor(p.AllocID, rule.Target) {
total += int64(rule.Weight)
}
}
}
return total
}
// isAllocFor reports whether an AllocID encodes a placement for the
// named workload. AllocIDs are `ns/name-idx`, so we look for the
// workload name as the segment after the first slash and before the
// trailing `-idx`.
func isAllocFor(allocID, workloadName string) bool {
// strip namespace prefix
rest := allocID
if i := indexByte(rest, '/'); i >= 0 {
rest = rest[i+1:]
}
// strip trailing -idx
if i := lastIndexByte(rest, '-'); i >= 0 {
rest = rest[:i]
}
return rest == workloadName
}
// indexByte returns the index of the first occurrence of b in s, or
// -1. Avoids importing strings just for one helper.
func indexByte(s string, b byte) int {
for i := 0; i < len(s); i++ {
if s[i] == b {
return i
}
}
return -1
}
// lastIndexByte returns the index of the last occurrence of b in s, or
// -1.
func lastIndexByte(s string, b byte) int {
for i := len(s) - 1; i >= 0; i-- {
if s[i] == b {
return i
}
}
return -1
}
// ----------------------------------------------------------------------------
// Helpers
// ----------------------------------------------------------------------------
// filterFitting returns a copy of the nodes that pass Score for the
// request, preserving order. Capacity is not yet decremented; the
// caller adjusts FreeCPU/FreeMem as it places replicas.
func filterFitting(nodes []NodeInfo, req WorkloadRequest) []NodeInfo {
var out []NodeInfo
for _, n := range nodes {
if _, ok := Score(n, req); ok {
out = append(out, n)
}
}
return out
}
// allocID renders a stable, namespaced allocation id for a placement.
// Format: `ns/spec.Name-<idx>`.
func allocID(req WorkloadRequest, idx int) string {
ns := req.Namespace
if ns == "" {
ns = "default"
}
return fmt.Sprintf("%s/%s-%d", ns, req.Spec.Name, idx)
}
+451
View File
@@ -0,0 +1,451 @@
package scheduler
import (
"strings"
"testing"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// threeLinuxNodes returns a small cluster of three Linux nodes with
// distinct free capacities so best-fit ordering is unambiguous.
func threeLinuxNodes() []NodeInfo {
return []NodeInfo{
{Hostname: "node-a", Runtimes: []string{"process"}, Tags: nil, CPU: 4, Memory: 4096, FreeCPU: 4, FreeMem: 4096, Kind: "linux"},
{Hostname: "node-b", Runtimes: []string{"process"}, Tags: nil, CPU: 8, Memory: 8192, FreeCPU: 8, FreeMem: 8192, Kind: "linux"},
{Hostname: "node-c", Runtimes: []string{"process"}, Tags: nil, CPU: 2, Memory: 2048, FreeCPU: 2, FreeMem: 2048, Kind: "linux"},
}
}
func jobSpec(name, oneOf string, constraints []string) *jobspec.WorkloadSpec {
return &jobspec.WorkloadSpec{
Kind: "Job",
Name: name,
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: oneOf},
Constraints: constraints,
}
}
func serviceSpec(name, oneOf string, count int, constraints []string) *jobspec.WorkloadSpec {
return &jobspec.WorkloadSpec{
Kind: "Service",
Name: name,
Count: count,
Runtime: &jobspec.RuntimeBlock{OneOf: oneOf},
Constraints: constraints,
}
}
func daemonSetSpec(name, oneOf string, constraints []string) *jobspec.WorkloadSpec {
return &jobspec.WorkloadSpec{
Kind: "DaemonSet",
Name: name,
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: oneOf},
Constraints: constraints,
}
}
// ---------------------------------------------------------------------------
// Job
// ---------------------------------------------------------------------------
func TestScheduleJob_BestFit(t *testing.T) {
nodes := threeLinuxNodes()
req := WorkloadRequest{Spec: jobSpec("batch", "process", nil), Namespace: "ns"}
got, err := Schedule(nodes, req)
if err != nil {
t.Fatalf("Schedule: %v", err)
}
if len(got) != 1 {
t.Fatalf("placements = %d, want 1", len(got))
}
if got[0].Node != "node-b" {
t.Errorf("Node = %q, want node-b (most free capacity)", got[0].Node)
}
if !strings.HasPrefix(got[0].AllocID, "ns/batch-") {
t.Errorf("AllocID = %q, want ns/batch-*", got[0].AllocID)
}
if got[0].Score <= 0 {
t.Errorf("Score = %d, want > 0", got[0].Score)
}
}
func TestScheduleJob_NoFittingNode(t *testing.T) {
nodes := threeLinuxNodes()
// wasm runtime not advertised by any node.
req := WorkloadRequest{Spec: jobSpec("wasmjob", "wasm", nil), Namespace: "ns"}
if _, err := Schedule(nodes, req); err == nil {
t.Fatal("Schedule: expected error for no-fitting node, got nil")
}
}
// ---------------------------------------------------------------------------
// Service
// ---------------------------------------------------------------------------
func TestScheduleService_SpreadAcrossNodes(t *testing.T) {
nodes := threeLinuxNodes()
req := WorkloadRequest{Spec: serviceSpec("web", "process", 3, nil), Namespace: "ns"}
got, err := Schedule(nodes, req)
if err != nil {
t.Fatalf("Schedule: %v", err)
}
if len(got) != 3 {
t.Fatalf("placements = %d, want 3", len(got))
}
seen := map[string]int{}
for _, p := range got {
seen[p.Node]++
}
if len(seen) != 3 {
t.Errorf("anti-affinity spread: distinct nodes = %d, want 3; %v", len(seen), seen)
}
}
func TestScheduleService_ColocationWhenFewerNodes(t *testing.T) {
nodes := threeLinuxNodes()
req := WorkloadRequest{Spec: serviceSpec("web", "process", 5, nil), Namespace: "ns"}
got, err := Schedule(nodes, req)
if err != nil {
t.Fatalf("Schedule: %v", err)
}
if len(got) != 5 {
t.Fatalf("placements = %d, want 5", len(got))
}
seen := map[string]int{}
for _, p := range got {
seen[p.Node]++
}
if len(seen) != 3 {
t.Errorf("colocation: distinct nodes = %d, want 3 (all used)", len(seen))
}
// No node should host more than 2 (3 nodes, 5 replicas: 2+2+1).
for n, c := range seen {
if c > 2 {
t.Errorf("node %s has %d replicas, want <= 2", n, c)
}
}
}
func TestScheduleService_NoFittingNode(t *testing.T) {
nodes := threeLinuxNodes()
req := WorkloadRequest{Spec: serviceSpec("wasm-svc", "wasm", 3, nil), Namespace: "ns"}
if _, err := Schedule(nodes, req); err == nil {
t.Fatal("Schedule: expected error for service with no fitting node")
}
}
// ---------------------------------------------------------------------------
// DaemonSet
// ---------------------------------------------------------------------------
func TestScheduleDaemonSet_AllMatching(t *testing.T) {
nodes := threeLinuxNodes()
req := WorkloadRequest{Spec: daemonSetSpec("logrotate", "process", nil), Namespace: "ns"}
got, err := Schedule(nodes, req)
if err != nil {
t.Fatalf("Schedule: %v", err)
}
if len(got) != 3 {
t.Errorf("placements = %d, want 3 (one per node)", len(got))
}
seen := map[string]bool{}
for _, p := range got {
seen[p.Node] = true
}
if len(seen) != 3 {
t.Errorf("DaemonSet distinct nodes = %d, want 3", len(seen))
}
}
func TestScheduleDaemonSet_SomeExcludedByConstraint(t *testing.T) {
nodes := threeLinuxNodes()
// Only nodes with cpus >= 4 qualify: node-a (4) and node-b (8).
req := WorkloadRequest{Spec: daemonSetSpec("heavy", "process", []string{"node.cpus >= 4"}), Namespace: "ns"}
got, err := Schedule(nodes, req)
if err != nil {
t.Fatalf("Schedule: %v", err)
}
if len(got) != 2 {
t.Errorf("placements = %d, want 2 (cpus>=4)", len(got))
}
}
// ---------------------------------------------------------------------------
// Runtime compatibility
// ---------------------------------------------------------------------------
func TestSchedule_RuntimeCompatibilityWasm(t *testing.T) {
nodes := []NodeInfo{
{Hostname: "no-wasm", Runtimes: []string{"process"}, Kind: "linux", CPU: 8, Memory: 8192, FreeCPU: 8, FreeMem: 8192},
{Hostname: "has-wasm", Runtimes: []string{"process", "wasmtime"}, Kind: "linux", CPU: 4, Memory: 4096, FreeCPU: 4, FreeMem: 4096},
}
// Even though no-wasm has more free capacity, the wasm workload
// must land on has-wasm.
req := WorkloadRequest{Spec: jobSpec("wasmjob", "wasm", nil), Namespace: "ns"}
got, err := Schedule(nodes, req)
if err != nil {
t.Fatalf("Schedule: %v", err)
}
if got[0].Node != "has-wasm" {
t.Errorf("Node = %q, want has-wasm (runtime compatibility)", got[0].Node)
}
}
func TestSchedule_RuntimeCompatibilityPveVM(t *testing.T) {
nodes := []NodeInfo{
{Hostname: "linux-1", Runtimes: []string{"process"}, Kind: "linux", CPU: 8, Memory: 8192, FreeCPU: 8, FreeMem: 8192},
{Hostname: "pve-1", Runtimes: []string{"process"}, Kind: "proxmox", CPU: 8, Memory: 8192, FreeCPU: 8, FreeMem: 8192},
}
req := WorkloadRequest{Spec: jobSpec("vmjob", "pve-vm", nil), Namespace: "ns"}
got, err := Schedule(nodes, req)
if err != nil {
t.Fatalf("Schedule: %v", err)
}
if got[0].Node != "pve-1" {
t.Errorf("Node = %q, want pve-1 (pve-vm requires proxmox kind)", got[0].Node)
}
}
// ---------------------------------------------------------------------------
// Constraints
// ---------------------------------------------------------------------------
func TestSchedule_ConstraintKindExcludesProxmox(t *testing.T) {
nodes := []NodeInfo{
{Hostname: "linux-1", Runtimes: []string{"process"}, Kind: "linux", CPU: 8, Memory: 8192, FreeCPU: 8, FreeMem: 8192},
{Hostname: "pve-1", Runtimes: []string{"process"}, Kind: "proxmox", CPU: 8, Memory: 8192, FreeCPU: 8, FreeMem: 8192},
}
req := WorkloadRequest{Spec: jobSpec("linuxonly", "process", []string{`node.kind == "linux"`}), Namespace: "ns"}
got, err := Schedule(nodes, req)
if err != nil {
t.Fatalf("Schedule: %v", err)
}
if got[0].Node != "linux-1" {
t.Errorf("Node = %q, want linux-1 (kind==linux)", got[0].Node)
}
}
func TestSchedule_ConstraintCPUsExcludesSmall(t *testing.T) {
nodes := threeLinuxNodes() // node-c has cpus=2
req := WorkloadRequest{Spec: jobSpec("big", "process", []string{"node.cpus >= 4"}), Namespace: "ns"}
got, err := Schedule(nodes, req)
if err != nil {
t.Fatalf("Schedule: %v", err)
}
if got[0].Node == "node-c" {
t.Errorf("Node = node-c, want node-a or node-b (cpus>=4)")
}
}
func TestSchedule_ConstraintNotInTags(t *testing.T) {
nodes := []NodeInfo{
{Hostname: "tagged", Runtimes: []string{"process"}, Tags: []string{"log-shipper"}, Kind: "linux", CPU: 8, Memory: 8192, FreeCPU: 8, FreeMem: 8192},
{Hostname: "clean", Runtimes: []string{"process"}, Tags: nil, Kind: "linux", CPU: 4, Memory: 4096, FreeCPU: 4, FreeMem: 4096},
}
req := WorkloadRequest{Spec: jobSpec("worker", "process", []string{`"log-shipper" not in node.tags`}), Namespace: "ns"}
got, err := Schedule(nodes, req)
if err != nil {
t.Fatalf("Schedule: %v", err)
}
if got[0].Node != "clean" {
t.Errorf("Node = %q, want clean (log-shipper not in tags)", got[0].Node)
}
}
// ---------------------------------------------------------------------------
// Affinity
// ---------------------------------------------------------------------------
func TestSchedule_AffinityPrefersColocatedNode(t *testing.T) {
// Place a redis service first, then a worker with affinity for
// redis; the worker should prefer the node where redis already
// runs even if another node has more free capacity.
nodes := []NodeInfo{
{Hostname: "big", Runtimes: []string{"process"}, Kind: "linux", CPU: 8, Memory: 8192, FreeCPU: 8, FreeMem: 8192},
{Hostname: "small", Runtimes: []string{"process"}, Kind: "linux", CPU: 4, Memory: 4096, FreeCPU: 4, FreeMem: 4096},
}
redisReq := WorkloadRequest{Spec: serviceSpec("redis", "process", 1, nil), Namespace: "ns"}
redisPlacements, err := Schedule(nodes, redisReq)
if err != nil {
t.Fatalf("redis Schedule: %v", err)
}
// Redis lands on "big" (most free capacity). Now schedule the
// worker with affinity to redis; it should also land on "big".
workerReq := WorkloadRequest{Spec: &jobspec.WorkloadSpec{
Kind: "Job",
Name: "worker",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Affinity: []jobspec.AffinityRule{
{Target: "redis", Weight: 1000},
},
}, Namespace: "ns"}
// The affinity is name-based; we need to seed the worker schedule
// with the redis placement so affinityScore can see it. Schedule
// does not take prior placements, so test affinityScore directly.
got := affinityScore(nodes[0], workerReq, redisPlacements)
if got <= 0 {
t.Errorf("affinityScore(big) = %d, want > 0 (redis colocated)", got)
}
gotSmall := affinityScore(nodes[1], workerReq, redisPlacements)
if gotSmall != 0 {
t.Errorf("affinityScore(small) = %d, want 0 (redis not colocated)", gotSmall)
}
}
func TestSchedule_AffinityCELExpression(t *testing.T) {
// Affinity with a CEL target: prefer nodes tagged "ssd".
nodes := []NodeInfo{
{Hostname: "hdd", Runtimes: []string{"process"}, Tags: []string{"hdd"}, Kind: "linux", CPU: 8, Memory: 8192, FreeCPU: 8, FreeMem: 8192},
{Hostname: "ssd", Runtimes: []string{"process"}, Tags: []string{"ssd"}, Kind: "linux", CPU: 4, Memory: 4096, FreeCPU: 4, FreeMem: 4096},
}
req := WorkloadRequest{Spec: &jobspec.WorkloadSpec{
Kind: "Job",
Name: "db",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Affinity: []jobspec.AffinityRule{
{Target: `"ssd" in node.tags`, Weight: 10000},
},
}, Namespace: "ns"}
got, err := Schedule(nodes, req)
if err != nil {
t.Fatalf("Schedule: %v", err)
}
if got[0].Node != "ssd" {
t.Errorf("Node = %q, want ssd (affinity to ssd tag outweighs capacity)", got[0].Node)
}
}
// ---------------------------------------------------------------------------
// Error paths
// ---------------------------------------------------------------------------
func TestSchedule_EmptyNodes(t *testing.T) {
req := WorkloadRequest{Spec: jobSpec("x", "process", nil), Namespace: "ns"}
if _, err := Schedule(nil, req); err == nil {
t.Fatal("Schedule: expected error for empty nodes, got nil")
}
}
func TestSchedule_NilSpec(t *testing.T) {
if _, err := Schedule(threeLinuxNodes(), WorkloadRequest{}); err == nil {
t.Fatal("Schedule: expected error for nil spec, got nil")
}
}
func TestSchedule_UnknownKind(t *testing.T) {
req := WorkloadRequest{Spec: &jobspec.WorkloadSpec{Kind: "Cron", Name: "x", Count: 1}, Namespace: "ns"}
if _, err := Schedule(threeLinuxNodes(), req); err == nil {
t.Fatal("Schedule: expected error for unknown kind")
}
}
// ---------------------------------------------------------------------------
// Score unit tests
// ---------------------------------------------------------------------------
func TestScore_FitsAndDoesNotFit(t *testing.T) {
node := NodeInfo{Hostname: "n", Runtimes: []string{"process"}, Kind: "linux", CPU: 4, Memory: 4096, FreeCPU: 4, FreeMem: 4096}
req := WorkloadRequest{Spec: jobSpec("j", "process", nil), Namespace: "ns"}
score, fits := Score(node, req)
if !fits {
t.Error("fits = false, want true")
}
if score <= 0 {
t.Errorf("score = %d, want > 0", score)
}
}
func TestScore_RuntimeMismatchDoesNotFit(t *testing.T) {
node := NodeInfo{Hostname: "n", Runtimes: []string{"process"}, Kind: "linux", CPU: 4, Memory: 4096, FreeCPU: 4, FreeMem: 4096}
req := WorkloadRequest{Spec: jobSpec("j", "wasm", nil), Namespace: "ns"}
if _, fits := Score(node, req); fits {
t.Error("fits = true for wasm on process-only node, want false")
}
}
func TestScore_ConstraintFailsDoesNotFit(t *testing.T) {
node := NodeInfo{Hostname: "n", Runtimes: []string{"process"}, Kind: "linux", CPU: 4, Memory: 4096, FreeCPU: 4, FreeMem: 4096}
req := WorkloadRequest{Spec: jobSpec("j", "process", []string{`node.kind == "proxmox"`}), Namespace: "ns"}
if _, fits := Score(node, req); fits {
t.Error("fits = true for kind==proxmox on linux node, want false")
}
}
// ---------------------------------------------------------------------------
// allocID / isAllocFor helpers
// ---------------------------------------------------------------------------
func TestAllocID(t *testing.T) {
req := WorkloadRequest{Spec: &jobspec.WorkloadSpec{Name: "web"}, Namespace: "prod"}
if got := allocID(req, 2); got != "prod/web-2" {
t.Errorf("allocID = %q, want prod/web-2", got)
}
req.Namespace = ""
if got := allocID(req, 0); got != "default/web-0" {
t.Errorf("allocID = %q, want default/web-0", got)
}
}
func TestIsAllocFor(t *testing.T) {
cases := []struct {
allocID string
workload string
want bool
}{
{"ns/redis-0", "redis", true},
{"ns/redis-12", "redis", true},
{"ns/worker-0", "redis", false},
{"redis-0", "redis", true},
{"ns/web-canary-3", "web-canary", true},
}
for _, c := range cases {
if got := isAllocFor(c.allocID, c.workload); got != c.want {
t.Errorf("isAllocFor(%q,%q) = %v, want %v", c.allocID, c.workload, got, c.want)
}
}
}
// ---------------------------------------------------------------------------
// normalizeRuntime / hasRuntime
// ---------------------------------------------------------------------------
func TestHasRuntimeAliases(t *testing.T) {
cases := []struct {
name string
node NodeInfo
want bool
}{
{"wasm on wasmtime node", NodeInfo{Runtimes: []string{"wasmtime"}, Kind: "linux"}, true},
{"wasm on process node", NodeInfo{Runtimes: []string{"process"}, Kind: "linux"}, false},
{"pve-vm on linux node", NodeInfo{Runtimes: []string{"pve-vm"}, Kind: "linux"}, false},
{"pve-vm on proxmox node", NodeInfo{Runtimes: nil, Kind: "proxmox"}, true},
{"process on process node", NodeInfo{Runtimes: []string{"process"}, Kind: "linux"}, true},
{"empty runtime on any node", NodeInfo{Runtimes: []string{"process"}, Kind: "linux"}, true},
}
for _, c := range cases {
if c.name == "empty runtime on any node" {
// hasRuntime is only called when OneOf != "".
continue
}
if got := hasRuntime(c.node, "wasm"); c.name == "wasm on wasmtime node" || c.name == "wasm on process node" {
if got != c.want {
t.Errorf("%s: hasRuntime(wasm) = %v, want %v", c.name, got, c.want)
}
}
}
// Explicit pve-vm and process checks.
if !hasRuntime(NodeInfo{Runtimes: nil, Kind: "proxmox"}, "pve-vm") {
t.Error("pve-vm on proxmox node should fit")
}
if hasRuntime(NodeInfo{Runtimes: nil, Kind: "linux"}, "pve-vm") {
t.Error("pve-vm on linux node should not fit")
}
if !hasRuntime(NodeInfo{Runtimes: []string{"process"}, Kind: "linux"}, "process") {
t.Error("process on process node should fit")
}
}
+48
View File
@@ -16,27 +16,51 @@ import (
// CAValidity is how long a CA cert is valid. Per D-013, the CA is long-lived
// (10 years) because manual rotation is expensive.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). This constant is retained for the dual-write window and
// scheduled for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
const CAValidity = 10 * 365 * 24 * time.Hour
// ServerCertValidity is the default validity window for server certs. D-013
// says server certs are short-lived (90 days) to limit the compromise window.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). This constant is retained for the dual-write window and
// scheduled for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
const ServerCertValidity = 90 * 24 * time.Hour
// CAKeySize is the RSA key size used for both CA and server certs. 3072 is
// the minimum we accept for v0.2 — matches REQ-033 spirit and Go's stdlib
// defaults for new RSA keys are typically 2048 or 4096. 3072 is the
// sweet spot for balance of safety and key-gen latency.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). This constant is retained for the dual-write window and
// scheduled for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
const CAKeySize = 3072
// CAMode is the file mode used when persisting the CA private key. REQ-033
// requires 0600.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). This constant is retained for the dual-write window and
// scheduled for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
const CAMode os.FileMode = 0o600
// CACPEMMode is the file mode used when persisting the CA public cert.
// REQ-033 requires 0644 (public, but still mode-pinned).
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). This constant is retained for the dual-write window and
// scheduled for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
const CACPEMMode os.FileMode = 0o644
// File names used inside the CA directory.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). These constants are retained for the dual-write window and
// scheduled for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
const (
CACertFile = "ca.crt"
CAKeyFile = "ca.key"
@@ -44,6 +68,10 @@ const (
// CA wraps a loaded CA. Use CAInit to mint a new one, LoadCA to read an
// existing one from disk.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). The CA type is retained for the dual-write window and
// scheduled for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
type CA struct {
Cert *x509.Certificate
Key *rsa.PrivateKey
@@ -60,6 +88,10 @@ type CA struct {
// commonName is the CA's CommonName (typically an org/cluster identifier).
// Returns a *CA wrapping the loaded cert + key. The CA is valid for
// CAValidity from now.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). CAInit is retained for the dual-write window and scheduled
// for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
func CAInit(dir, commonName string) (*CA, error) {
if dir == "" {
return nil, errors.New("CAInit: dir is required")
@@ -133,6 +165,10 @@ func CAInit(dir, commonName string) (*CA, error) {
// LoadCA reads a previously-initialized CA from disk. Returns a *CA or an
// error. Verifies file modes (REQ-033).
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). LoadCA is retained for the dual-write window and scheduled
// for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
func LoadCA(dir string) (*CA, error) {
if dir == "" {
return nil, errors.New("LoadCA: dir is required")
@@ -187,6 +223,10 @@ func LoadCA(dir string) (*CA, error) {
// EnforceFileModes refuses to operate if ca.crt / ca.key do not have the
// required modes (REQ-033). Returns nil on success. Callers (daemon start,
// CA loaders) MUST call this and abort on error.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). EnforceFileModes is retained for the dual-write window
// and scheduled for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
func EnforceFileModes(dir string) error {
certPath := filepath.Join(dir, CACertFile)
keyPath := filepath.Join(dir, CAKeyFile)
@@ -217,6 +257,10 @@ func EnforceFileModes(dir string) error {
// in PEM form. The resulting cert is valid for ServerCertValidity and
// inherits the SANs from the CSR (DNS, IP). If the CSR has no SANs, the
// call fails — REQ-036 requires server certs to have identifying SANs.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). SignCSR is retained for the dual-write window and
// scheduled for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
func (c *CA) SignCSR(csrPEM []byte) ([]byte, error) {
if c == nil || c.Cert == nil || c.Key == nil {
return nil, errors.New("SignCSR: nil CA")
@@ -266,6 +310,10 @@ func (c *CA) SignCSR(csrPEM []byte) ([]byte, error) {
// Fingerprint returns the SHA-256 hex fingerprint of the CA cert. Useful
// for the operator to communicate to peers out-of-band; peers then pin
// this value at `orca node join --ca-fingerprint <sha>`.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). CA.Fingerprint is retained for the dual-write window and
// scheduled for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
func (c *CA) Fingerprint() string {
return FingerprintOf(c.Cert.Raw)
}
+4
View File
@@ -21,6 +21,10 @@ import (
// Validation: dns entries must be syntactically valid hostnames; ip entries
// must be parseable by net.ParseIP. Bad inputs are rejected up-front so
// the operator gets a clear error before signing.
//
// Deprecated: v0.9 re-architecture replaces the internal CA with step-ca
// (D-101/REQ-076). GenerateCSR is retained for the dual-write window and
// scheduled for deletion in v0.10-P14. See .ciagent/PRD_v0.9.md.
func GenerateCSR(commonName string, sans []string) (keyPEM, csrPEM []byte, err error) {
if commonName == "" {
return nil, nil, errors.New("GenerateCSR: commonName is required")
+35
View File
@@ -0,0 +1,35 @@
package security
import (
"os"
"syscall"
)
// Flock acquires an exclusive advisory lock on the file at path, creating it
// if missing. Returns a release function that MUST be called (deferred) to
// release the lock and close the file descriptor. Used by the known_hosts
// read-modify-write paths (TOFUHostKeyCallback capture + ResetHostKey) to
// prevent concurrent writers under v0.9's parallel SSH fan-out (REQ-063,
// deferred P1 from REVIEW_v0.8 A2).
func Flock(path string) (release func(), err error) {
f, err := os.OpenFile(path, os.O_CREATE|os.O_RDWR, 0o600)
if err != nil {
return nil, err
}
if err := syscall.Flock(int(f.Fd()), syscall.LOCK_EX); err != nil {
f.Close()
return nil, err
}
return func() {
_ = syscall.Flock(int(f.Fd()), syscall.LOCK_UN)
_ = f.Close()
}, nil
}
func tryFlockEx(fd int) error {
return syscall.Flock(fd, syscall.LOCK_EX|syscall.LOCK_NB)
}
func releaseFlock(fd int) error {
return syscall.Flock(fd, syscall.LOCK_UN)
}
+82
View File
@@ -0,0 +1,82 @@
package security
import (
"os"
"path/filepath"
"testing"
"time"
)
func TestFlock_acquireAndRelease(t *testing.T) {
dir := t.TempDir()
path := filepath.Join(dir, "test.lock")
release, err := Flock(path)
if err != nil {
t.Fatalf("Flock: %v", err)
}
if _, statErr := os.Stat(path); statErr != nil {
t.Fatalf("lock file not created: %v", statErr)
}
release()
}
func TestFlock_reentrantAfterRelease(t *testing.T) {
dir := t.TempDir()
path := filepath.Join(dir, "test.lock")
r1, err := Flock(path)
if err != nil {
t.Fatalf("first Flock: %v", err)
}
r1()
r2, err := Flock(path)
if err != nil {
t.Fatalf("second Flock after release: %v", err)
}
r2()
}
func TestFlock_concurrentBlocks(t *testing.T) {
dir := t.TempDir()
path := filepath.Join(dir, "test.lock")
r1, err := Flock(path)
if err != nil {
t.Fatalf("first Flock: %v", err)
}
// Give the blocking goroutine a chance to start and block.
time.Sleep(50 * time.Millisecond)
// Verify the second lock is blocked by checking it hasn't acquired after a short window.
// Use a non-blocking attempt: open the file and try LOCK_EX|LOCK_NB.
blocked := make(chan bool, 1)
go func() {
f, err := os.OpenFile(path, os.O_RDWR, 0o600)
if err != nil {
blocked <- false
return
}
defer f.Close()
// LOCK_NB = non-blocking; returns EWOULDBLOCK if locked.
if err := tryFlockEx(int(f.Fd())); err != nil {
blocked <- true // got EWOULDBLOCK = the lock is held by r1
return
}
releaseFlock(int(f.Fd()))
blocked <- false // acquired = r1 didn't hold the lock (bug)
}()
select {
case b := <-blocked:
if !b {
t.Fatal("second lock acquired while first holds it — lock not working")
}
case <-time.After(2 * time.Second):
t.Fatal("non-blocking try-lock timed out")
}
r1()
}
+254
View File
@@ -0,0 +1,254 @@
// Package schema provides kind-specific validators for the unified
// *jobspec.WorkloadSpec introduced in P0b (REQ-064). Each workload kind
// (Job, Service, DaemonSet per R-012) has different required fields;
// this package exposes a Validator interface and a ValidatorFor
// dispatcher so the emitter layer (REQ-074) and the lint engine
// (REQ-084) can reject invalid specs before rendering.
//
// The validators operate purely on the *WorkloadSpec shape; they do no
// I/O. Required-field violations return a structured error listing
// every problem found (missing required fields, invalid combinations).
package schema
import (
"errors"
"fmt"
"net"
"strings"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// Validator validates a *jobspec.WorkloadSpec against a kind-specific
// schema. Implementations are pure (no I/O) and return a clear error
// listing every violation found.
type Validator interface {
Validate(spec *jobspec.WorkloadSpec) error
}
// JobValidator validates the Job workload kind (R-012).
//
// Rules:
// - no service block required (Job has no Traefik route by default D-175)
// - restart optional (defaults to never/on-failure when omitted)
// - schedule optional (cron string)
// - timeout optional
// - ports optional
// - count must be 1 (or unset → 1); count > 1 is an error for Job
// (use a Service for replicas)
// - no Traefik route (a ServiceBlock is rejected)
// - task group (spec.Tasks) optional; when present, each task must
// have a unique name and a resolvable command (P06).
type JobValidator struct{}
// ServiceValidator validates the Service workload kind (R-012).
//
// Rules:
// - ports required (at least one)
// - count ≥ 1
// - restart required (mode must be service)
// - update required (strategy must be rolling/canary/blue-green)
// - runtime required (unless a task group is present; each task
// can carry its own runtime — P06)
// - health block required (Traefik routing depends on health checks)
// - service block, if present, must have a valid bind (127.0.0.1
// opt-in per R-007; default is socket — empty bind is OK)
// - service block implied (Traefik route YES)
// - task group (spec.Tasks) optional; when present, each task must
// have a unique name and a resolvable command (P06).
type ServiceValidator struct{}
// DaemonSetValidator validates the DaemonSet workload kind (R-012).
//
// Rules:
// - schedule block with mode (every-node/matching/mandatory) required
// - no ports (no Traefik route by default D-175)
// - no count (implicit = nodes matching condition)
// - restart required
// - task group (spec.Tasks) optional; when present, each task must
// have a unique name and a resolvable command (P06).
type DaemonSetValidator struct{}
// ValidatorFor returns the Validator for the given workload kind, or an
// error for an unknown kind. kind must be one of Job, Service,
// DaemonSet (R-012).
func ValidatorFor(kind string) (Validator, error) {
switch kind {
case "Job":
return JobValidator{}, nil
case "Service":
return ServiceValidator{}, nil
case "DaemonSet":
return DaemonSetValidator{}, nil
default:
return nil, fmt.Errorf("schema: unknown kind %q (want one of Job, Service, DaemonSet)", kind)
}
}
// Validate validates a Job spec. See JobValidator for the rules.
func (JobValidator) Validate(spec *jobspec.WorkloadSpec) error {
if spec == nil {
return errors.New("schema/Job: spec is nil")
}
var errs []string
if strings.TrimSpace(spec.Name) == "" {
errs = append(errs, "name is required")
}
if spec.Count != 0 && spec.Count != 1 {
errs = append(errs, fmt.Sprintf("count must be 1 (or unset) for Job, got %d (use Service for replicas)", spec.Count))
}
if spec.Service != nil {
errs = append(errs, "service block (Traefik route) is not allowed for Job (D-175)")
}
errs = append(errs, validateTaskGroup(spec)...)
return composeErrors("schema/Job", errs)
}
// Validate validates a Service spec. See ServiceValidator for the rules.
func (ServiceValidator) Validate(spec *jobspec.WorkloadSpec) error {
if spec == nil {
return errors.New("schema/Service: spec is nil")
}
var errs []string
if strings.TrimSpace(spec.Name) == "" {
errs = append(errs, "name is required")
}
if len(spec.Ports) == 0 {
errs = append(errs, "ports required (at least one)")
}
if spec.Count < 1 {
errs = append(errs, fmt.Sprintf("count must be ≥ 1 for Service, got %d", spec.Count))
}
if spec.Restart == nil {
errs = append(errs, "restart block required for Service")
} else {
switch spec.Restart.Mode {
case "service", "on-failure", "never":
// Valid per R-012 (default for Service is "service",
// but the validator accepts the full enum; the
// Service-specific "must be service" rule is enforced
// below for the default case where mode is empty).
case "":
errs = append(errs, "restart mode required for Service (one of service, on-failure, never; default is service)")
default:
errs = append(errs, fmt.Sprintf("restart mode %q invalid (want one of service, on-failure, never)", spec.Restart.Mode))
}
}
if spec.Update == nil {
errs = append(errs, "update block required for Service")
} else {
switch spec.Update.Strategy {
case "rolling", "canary", "blue-green":
case "":
errs = append(errs, "update strategy required for Service (one of rolling, canary, blue-green)")
default:
errs = append(errs, fmt.Sprintf("update strategy %q invalid (want one of rolling, canary, blue-green)", spec.Update.Strategy))
}
}
if spec.Runtime == nil && len(spec.Tasks) == 0 {
errs = append(errs, "runtime block required for Service (or a task group with per-task runtimes)")
}
if spec.Health == nil {
errs = append(errs, "health block required for Service (Traefik routing requires health checks)")
}
if spec.Service != nil {
if err := validateServiceBind(spec.Service.Bind); err != nil {
errs = append(errs, err.Error())
}
}
errs = append(errs, validateTaskGroup(spec)...)
return composeErrors("schema/Service", errs)
}
// validateServiceBind validates the service.bind field (R-007). Empty
// is OK (default = socket). When set, it must be a valid IPv4/IPv6
// address (the only opt-in to bind on a non-loopback address); the
// loopback 127.0.0.1 is the documented opt-in. Anything that is not
// parseable as an IP address is rejected.
func validateServiceBind(bind string) error {
if strings.TrimSpace(bind) == "" {
return nil
}
if net.ParseIP(bind) == nil {
return fmt.Errorf("service.bind %q is not a valid IP address (R-007: 127.0.0.1 opt-in; default is socket)", bind)
}
return nil
}
// Validate validates a DaemonSet spec. See DaemonSetValidator for the rules.
func (DaemonSetValidator) Validate(spec *jobspec.WorkloadSpec) error {
if spec == nil {
return errors.New("schema/DaemonSet: spec is nil")
}
var errs []string
if strings.TrimSpace(spec.Name) == "" {
errs = append(errs, "name is required")
}
if spec.Schedule == nil {
errs = append(errs, "schedule block required for DaemonSet")
} else {
switch spec.Schedule.Mode {
case "every-node", "matching", "mandatory":
case "":
errs = append(errs, "schedule mode required for DaemonSet (one of every-node, matching, mandatory)")
default:
errs = append(errs, fmt.Sprintf("schedule mode %q invalid (want one of every-node, matching, mandatory)", spec.Schedule.Mode))
}
}
if len(spec.Ports) > 0 {
errs = append(errs, "ports not allowed for DaemonSet (no Traefik route by default D-175)")
}
if spec.Count != 0 {
errs = append(errs, fmt.Sprintf("count not allowed for DaemonSet (implicit = nodes matching condition), got %d", spec.Count))
}
if spec.Restart == nil {
errs = append(errs, "restart block required for DaemonSet")
}
errs = append(errs, validateTaskGroup(spec)...)
return composeErrors("schema/DaemonSet", errs)
}
// composeErrors joins the per-field errors into a single error prefixed
// by the validator name. Returns nil when there are no errors so the
// caller can return the result directly.
func composeErrors(name string, errs []string) error {
if len(errs) == 0 {
return nil
}
return fmt.Errorf("%s: %s", name, strings.Join(errs, "; "))
}
// validateTaskGroup validates the task-group list shared by all kinds
// (P06, PRD §9.1). When the spec carries a task group (spec.Tasks
// non-empty), each task must have a unique name and a resolvable
// command (the task's own Command, the task's runtime command, or the
// top-level runtime command as the per-group default). The top-level
// runtime is optional when tasks is present (each task can carry its
// own runtime). Returns nil when the spec has no task group.
func validateTaskGroup(spec *jobspec.WorkloadSpec) []string {
if len(spec.Tasks) == 0 {
return nil
}
var errs []string
seen := make(map[string]bool, len(spec.Tasks))
for i, task := range spec.Tasks {
if strings.TrimSpace(task.Name) == "" {
errs = append(errs, fmt.Sprintf("tasks[%d]: name is required", i))
} else if seen[task.Name] {
errs = append(errs, fmt.Sprintf("tasks[%d]: duplicate task name %q (names must be unique within the group)", i, task.Name))
} else {
seen[task.Name] = true
}
cmd := task.Command
if strings.TrimSpace(cmd) == "" && task.Runtime != nil {
cmd = task.Runtime.Command
}
if strings.TrimSpace(cmd) == "" && spec.Runtime != nil {
cmd = spec.Runtime.Command
}
if strings.TrimSpace(cmd) == "" {
errs = append(errs, fmt.Sprintf("tasks[%d]: command is required (set tasks[].command, tasks[].runtime.command, or top-level runtime.command)", i))
}
}
return errs
}
+871
View File
@@ -0,0 +1,871 @@
package schema
import (
"strings"
"testing"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
func TestJobValidator_ValidMinimal(t *testing.T) {
spec := &jobspec.WorkloadSpec{Kind: "Job", Name: "backup", Count: 1}
v := JobValidator{}
if err := v.Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestJobValidator_ValidWithSchedule(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "backup",
Count: 1,
Schedule: &jobspec.ScheduleBlock{Cron: "0 2 * * *"},
Timeout: "1h",
}
v := JobValidator{}
if err := v.Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestJobValidator_ValidUnsetCount(t *testing.T) {
spec := &jobspec.WorkloadSpec{Kind: "Job", Name: "one-shot"}
v := JobValidator{}
if err := v.Validate(spec); err != nil {
t.Fatalf("unset count should default-accept, got %v", err)
}
}
func TestJobValidator_MissingName(t *testing.T) {
spec := &jobspec.WorkloadSpec{Kind: "Job", Count: 1}
err := JobValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for missing name, got nil")
}
if !strings.Contains(err.Error(), "name is required") {
t.Errorf("error = %q, want 'name is required'", err.Error())
}
}
func TestJobValidator_CountGreaterThanOne(t *testing.T) {
spec := &jobspec.WorkloadSpec{Kind: "Job", Name: "batch", Count: 3}
err := JobValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for count > 1, got nil")
}
if !strings.Contains(err.Error(), "count must be 1") {
t.Errorf("error = %q, want 'count must be 1'", err.Error())
}
}
func TestJobValidator_ServiceBlockRejected(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "x",
Count: 1,
Service: &jobspec.ServiceBlock{Host: "x.example"},
}
err := JobValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for service block on Job, got nil")
}
if !strings.Contains(err.Error(), "service block") {
t.Errorf("error = %q, want 'service block'", err.Error())
}
}
func TestJobValidator_NilSpec(t *testing.T) {
v := JobValidator{}
if err := v.Validate(nil); err == nil {
t.Fatal("expected error for nil spec")
}
}
func TestServiceValidator_ValidFull(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 3,
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/http"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling", MaxSurge: 1},
Health: &jobspec.HealthBlock{CheckType: "http", Interval: "5s"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
v := ServiceValidator{}
if err := v.Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestServiceValidator_MissingPorts(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for missing ports, got nil")
}
if !strings.Contains(err.Error(), "ports required") {
t.Errorf("error = %q, want 'ports required'", err.Error())
}
}
func TestServiceValidator_CountZero(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 0,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for count 0, got nil")
}
if !strings.Contains(err.Error(), "count must be") {
t.Errorf("error = %q, want 'count must be'", err.Error())
}
}
func TestServiceValidator_MissingRestart(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for missing restart, got nil")
}
if !strings.Contains(err.Error(), "restart block required") {
t.Errorf("error = %q, want 'restart block required'", err.Error())
}
}
func TestServiceValidator_WrongRestartMode(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "always"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for invalid restart mode, got nil")
}
if !strings.Contains(err.Error(), "restart mode") {
t.Errorf("error = %q, want 'restart mode'", err.Error())
}
}
func TestServiceValidator_AcceptedRestartModes(t *testing.T) {
// R-012: restart.mode accepts service / on-failure / never for
// Service; the default per R-012 is "service" but the validator
// accepts the full enum (a Service that wants on-failure is
// unusual but not invalid — only "always" and unknown modes are
// rejected).
for _, mode := range []string{"service", "on-failure", "never"} {
t.Run(mode, func(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: mode},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
v := ServiceValidator{}
if err := v.Validate(spec); err != nil {
t.Errorf("mode %q should be accepted, got: %v", mode, err)
}
})
}
}
func TestServiceValidator_MissingUpdate(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for missing update, got nil")
}
if !strings.Contains(err.Error(), "update block required") {
t.Errorf("error = %q, want 'update block required'", err.Error())
}
}
func TestServiceValidator_MissingRuntime(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for missing runtime, got nil")
}
if !strings.Contains(err.Error(), "runtime block required") {
t.Errorf("error = %q, want 'runtime block required'", err.Error())
}
}
func TestServiceValidator_NilSpec(t *testing.T) {
v := ServiceValidator{}
if err := v.Validate(nil); err == nil {
t.Fatal("expected error for nil spec")
}
}
func TestServiceValidator_MissingHealth(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for missing health block, got nil")
}
if !strings.Contains(err.Error(), "health block required") {
t.Errorf("error = %q, want 'health block required'", err.Error())
}
}
func TestServiceValidator_InvalidRestartMode(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "always"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for invalid restart mode, got nil")
}
if !strings.Contains(err.Error(), "restart mode") || !strings.Contains(err.Error(), "invalid") {
t.Errorf("error = %q, want 'restart mode ... invalid'", err.Error())
}
}
func TestServiceValidator_EmptyRestartMode(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: ""},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for empty restart mode, got nil")
}
if !strings.Contains(err.Error(), "restart mode required") {
t.Errorf("error = %q, want 'restart mode required'", err.Error())
}
}
func TestServiceValidator_InvalidUpdateStrategy(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "recreate"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for invalid update strategy, got nil")
}
if !strings.Contains(err.Error(), "update strategy") || !strings.Contains(err.Error(), "invalid") {
t.Errorf("error = %q, want 'update strategy ... invalid'", err.Error())
}
}
func TestServiceValidator_EmptyUpdateStrategy(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: ""},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for empty update strategy, got nil")
}
if !strings.Contains(err.Error(), "update strategy required") {
t.Errorf("error = %q, want 'update strategy required'", err.Error())
}
}
func TestServiceValidator_AcceptedUpdateStrategies(t *testing.T) {
for _, strat := range []string{"rolling", "canary", "blue-green"} {
t.Run(strat, func(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: strat},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
v := ServiceValidator{}
if err := v.Validate(spec); err != nil {
t.Errorf("strategy %q should be accepted, got: %v", strat, err)
}
})
}
}
func TestServiceValidator_InvalidServiceBind(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Service: &jobspec.ServiceBlock{Bind: "not-an-ip"},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for invalid service.bind, got nil")
}
if !strings.Contains(err.Error(), "service.bind") || !strings.Contains(err.Error(), "valid IP") {
t.Errorf("error = %q, want 'service.bind ... valid IP'", err.Error())
}
}
func TestServiceValidator_ValidServiceBindLoopback(t *testing.T) {
// R-007: 127.0.0.1 is the documented opt-in for a non-socket bind.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Service: &jobspec.ServiceBlock{Bind: "127.0.0.1"},
}
v := ServiceValidator{}
if err := v.Validate(spec); err != nil {
t.Fatalf("127.0.0.1 should be accepted, got: %v", err)
}
}
func TestServiceValidator_ValidServiceBindIPv6(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Service: &jobspec.ServiceBlock{Bind: "::1"},
}
v := ServiceValidator{}
if err := v.Validate(spec); err != nil {
t.Fatalf("::1 should be accepted, got: %v", err)
}
}
func TestServiceValidator_EmptyServiceBindOK(t *testing.T) {
// R-007: empty bind = default = socket; valid.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process"},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
Service: &jobspec.ServiceBlock{Bind: ""},
}
v := ServiceValidator{}
if err := v.Validate(spec); err != nil {
t.Fatalf("empty bind should default to socket (valid), got: %v", err)
}
}
func TestServiceValidator_MultipleErrors(t *testing.T) {
// Multiple violations should all surface in the composed error.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "",
Count: 0,
Restart: &jobspec.RestartBlock{Mode: "always"},
Update: &jobspec.UpdateBlock{Strategy: "recreate"},
Service: &jobspec.ServiceBlock{Bind: "not-an-ip"},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error, got nil")
}
for _, want := range []string{
"name is required",
"ports required",
"count must be",
"restart mode",
"update strategy",
"runtime block required",
"health block required",
"service.bind",
} {
if !strings.Contains(err.Error(), want) {
t.Errorf("error %q missing %q", err.Error(), want)
}
}
}
func TestDaemonSetValidator_Valid(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "DaemonSet",
Name: "log-shipper",
Schedule: &jobspec.ScheduleBlock{Mode: "every-node"},
Restart: &jobspec.RestartBlock{Mode: "on-failure"},
}
v := DaemonSetValidator{}
if err := v.Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestDaemonSetValidator_ValidMatchingMode(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "DaemonSet",
Name: "x",
Schedule: &jobspec.ScheduleBlock{Mode: "matching"},
Restart: &jobspec.RestartBlock{Mode: "on-failure"},
}
v := DaemonSetValidator{}
if err := v.Validate(spec); err != nil {
t.Fatalf("matching mode should be accepted, got %v", err)
}
}
func TestDaemonSetValidator_ValidMandatoryMode(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "DaemonSet",
Name: "x",
Schedule: &jobspec.ScheduleBlock{Mode: "mandatory"},
Restart: &jobspec.RestartBlock{Mode: "on-failure"},
}
v := DaemonSetValidator{}
if err := v.Validate(spec); err != nil {
t.Fatalf("mandatory mode should be accepted, got %v", err)
}
}
func TestDaemonSetValidator_MissingScheduleMode(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "DaemonSet",
Name: "x",
Schedule: &jobspec.ScheduleBlock{Mode: ""},
Restart: &jobspec.RestartBlock{Mode: "on-failure"},
}
err := DaemonSetValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for missing schedule mode, got nil")
}
if !strings.Contains(err.Error(), "schedule mode required") {
t.Errorf("error = %q, want 'schedule mode required'", err.Error())
}
}
func TestDaemonSetValidator_MissingScheduleBlock(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "DaemonSet",
Name: "x",
Restart: &jobspec.RestartBlock{Mode: "on-failure"},
}
err := DaemonSetValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for missing schedule block, got nil")
}
if !strings.Contains(err.Error(), "schedule block required") {
t.Errorf("error = %q, want 'schedule block required'", err.Error())
}
}
func TestDaemonSetValidator_InvalidScheduleMode(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "DaemonSet",
Name: "x",
Schedule: &jobspec.ScheduleBlock{Mode: "always"},
Restart: &jobspec.RestartBlock{Mode: "on-failure"},
}
err := DaemonSetValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for invalid schedule mode, got nil")
}
if !strings.Contains(err.Error(), "schedule mode") {
t.Errorf("error = %q, want 'schedule mode'", err.Error())
}
}
func TestDaemonSetValidator_HasPorts(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "DaemonSet",
Name: "x",
Schedule: &jobspec.ScheduleBlock{Mode: "every-node"},
Restart: &jobspec.RestartBlock{Mode: "on-failure"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 80}},
}
err := DaemonSetValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for ports on DaemonSet, got nil")
}
if !strings.Contains(err.Error(), "ports not allowed") {
t.Errorf("error = %q, want 'ports not allowed'", err.Error())
}
}
func TestDaemonSetValidator_HasCount(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "DaemonSet",
Name: "x",
Count: 3,
Schedule: &jobspec.ScheduleBlock{Mode: "every-node"},
Restart: &jobspec.RestartBlock{Mode: "on-failure"},
}
err := DaemonSetValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for count on DaemonSet, got nil")
}
if !strings.Contains(err.Error(), "count not allowed") {
t.Errorf("error = %q, want 'count not allowed'", err.Error())
}
}
func TestDaemonSetValidator_MissingRestart(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "DaemonSet",
Name: "x",
Schedule: &jobspec.ScheduleBlock{Mode: "every-node"},
}
err := DaemonSetValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for missing restart on DaemonSet, got nil")
}
if !strings.Contains(err.Error(), "restart block required") {
t.Errorf("error = %q, want 'restart block required'", err.Error())
}
}
func TestDaemonSetValidator_NilSpec(t *testing.T) {
v := DaemonSetValidator{}
if err := v.Validate(nil); err == nil {
t.Fatal("expected error for nil spec")
}
}
func TestValidatorFor_EachKind(t *testing.T) {
cases := []struct {
kind string
want string
}{
{"Job", "schema.JobValidator"},
{"Service", "schema.ServiceValidator"},
{"DaemonSet", "schema.DaemonSetValidator"},
}
for _, tc := range cases {
t.Run(tc.kind, func(t *testing.T) {
v, err := ValidatorFor(tc.kind)
if err != nil {
t.Fatalf("ValidatorFor(%q): %v", tc.kind, err)
}
got := fmtType(v)
if got != tc.want {
t.Errorf("ValidatorFor(%q) type = %q, want %q", tc.kind, got, tc.want)
}
})
}
}
func TestValidatorFor_UnknownKind(t *testing.T) {
_, err := ValidatorFor("CronJob")
if err == nil {
t.Fatal("expected error for unknown kind, got nil")
}
if !strings.Contains(err.Error(), "unknown kind") {
t.Errorf("error = %q, want 'unknown kind'", err.Error())
}
}
// fmtType returns a readable type name for a validator. Uses fmt.Sprintf
// with %T rather than reflection to keep the test surface minimal.
func fmtType(v Validator) string {
switch v.(type) {
case JobValidator:
return "schema.JobValidator"
case ServiceValidator:
return "schema.ServiceValidator"
case DaemonSetValidator:
return "schema.DaemonSetValidator"
default:
return "unknown"
}
}
// Ensure composeErrors returns nil for empty input (covers the
// short-circuit branch that the validators rely on).
func TestComposeErrors_Empty(t *testing.T) {
if err := composeErrors("schema/X", nil); err != nil {
t.Errorf("composeErrors(nil) = %v, want nil", err)
}
if err := composeErrors("schema/X", []string{}); err != nil {
t.Errorf("composeErrors([]) = %v, want nil", err)
}
}
// Ensure the error type returned by composeErrors is a non-nil error
// when violations are present (guards against accidental nil-return).
func TestComposeErrors_NonEmpty(t *testing.T) {
err := composeErrors("schema/X", []string{"a", "b"})
if err == nil {
t.Fatal("expected non-nil error")
}
if !strings.Contains(err.Error(), "a") || !strings.Contains(err.Error(), "b") {
t.Errorf("error = %q, want both 'a' and 'b'", err.Error())
}
}
// Compile-time assertion that the validators implement the interface.
var (
_ Validator = JobValidator{}
_ Validator = ServiceValidator{}
_ Validator = DaemonSetValidator{}
)
func TestTaskGroup_Valid(t *testing.T) {
// P06: a valid task group — two tasks, each with a unique name
// and a resolvable command (own command). The top-level runtime
// is optional when each task carries its own.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Tasks: []jobspec.TaskGroupTask{
{Name: "app", Command: "/usr/bin/httpd"},
{Name: "sidecar", Command: "/bin/wasm-runner sidecar.wasm"},
},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
if err := (ServiceValidator{}).Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestTaskGroup_ValidInheritsTopLevelRuntime(t *testing.T) {
// P06: tasks without their own runtime inherit the top-level
// runtime command. The validator accepts this as long as the
// resolved command is non-empty.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/default"},
Tasks: []jobspec.TaskGroupTask{
{Name: "app"},
{Name: "sidecar"},
},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
if err := (ServiceValidator{}).Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestTaskGroup_ValidTaskRuntimeCommand(t *testing.T) {
// P06: a task whose command is provided via the task's own
// runtime.command (no top-level runtime) is valid.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 1,
Tasks: []jobspec.TaskGroupTask{
{Name: "app", Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/usr/bin/httpd"}},
},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
if err := (ServiceValidator{}).Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestTaskGroup_MissingTaskName(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Tasks: []jobspec.TaskGroupTask{
{Command: "/usr/bin/httpd"},
{Name: "sidecar", Command: "/bin/wasm-runner"},
},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for missing task name, got nil")
}
if !strings.Contains(err.Error(), "name is required") {
t.Errorf("error = %q, want 'name is required'", err.Error())
}
}
func TestTaskGroup_DuplicateTaskNames(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Tasks: []jobspec.TaskGroupTask{
{Name: "app", Command: "/usr/bin/httpd"},
{Name: "app", Command: "/bin/other"},
},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for duplicate task names, got nil")
}
if !strings.Contains(err.Error(), "duplicate task name") {
t.Errorf("error = %q, want 'duplicate task name'", err.Error())
}
}
func TestTaskGroup_MissingCommand(t *testing.T) {
// P06: a task with no resolvable command (no task.Command, no
// task.Runtime, no top-level Runtime) is rejected.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Tasks: []jobspec.TaskGroupTask{
{Name: "app"},
},
Restart: &jobspec.RestartBlock{Mode: "service"},
Update: &jobspec.UpdateBlock{Strategy: "rolling"},
Health: &jobspec.HealthBlock{CheckType: "http"},
Ports: []jobspec.PortSpec{{Name: "http", Port: 8080}},
}
err := ServiceValidator{}.Validate(spec)
if err == nil {
t.Fatal("expected error for missing task command, got nil")
}
if !strings.Contains(err.Error(), "command is required") {
t.Errorf("error = %q, want 'command is required'", err.Error())
}
}
func TestTaskGroup_JobAcceptsTaskGroup(t *testing.T) {
// P06: task groups apply to all kinds, not just Service. Job
// accepts a task group with unique names + resolvable commands.
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "batch",
Count: 1,
Tasks: []jobspec.TaskGroupTask{
{Name: "step1", Command: "/bin/extract"},
{Name: "step2", Command: "/bin/transform"},
},
}
if err := (JobValidator{}).Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestTaskGroup_DaemonSetAcceptsTaskGroup(t *testing.T) {
// P06: DaemonSet accepts a task group.
spec := &jobspec.WorkloadSpec{
Kind: "DaemonSet",
Name: "log-shipper",
Schedule: &jobspec.ScheduleBlock{Mode: "every-node"},
Restart: &jobspec.RestartBlock{Mode: "on-failure"},
Tasks: []jobspec.TaskGroupTask{
{Name: "collector", Command: "/bin/collect"},
{Name: "forwarder", Command: "/bin/forward"},
},
}
if err := (DaemonSetValidator{}).Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestTaskGroup_NoTasksBackwardCompat(t *testing.T) {
// Backward compat: a spec with no Tasks is validated by the
// existing kind-specific rules (no task-group check fires).
spec := &jobspec.WorkloadSpec{
Kind: "Job",
Name: "backup",
Count: 1,
Runtime: &jobspec.RuntimeBlock{OneOf: "process", Command: "/bin/rsync"},
}
if err := (JobValidator{}).Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
+126
View File
@@ -0,0 +1,126 @@
package schema
import (
"fmt"
"strconv"
"strings"
"time"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
// UpdateValidator validates the rolling/canary/blue-green update stanza
// (P03). The schema validator (ServiceValidator) already enforces that
// the strategy is one of rolling/canary/blue-green and that the update
// block is present for a Service. UpdateValidator adds the
// field-level validation:
//
// - max_parallel: integer in [1, count] (defaults to 1 when unset)
// - min_healthy_time: a valid time.Duration when set (time.ParseDuration)
// - healthy_deadline: a valid time.Duration when set (time.ParseDuration)
// - canary: an integer count in [0, count] OR a percentage string of
// the form "<n>%" where n is in [0, 100] (the parser already accepts
// both shapes; the validator accepts them too). Only meaningful
// for the canary strategy; ignored (but still validated for shape)
// for rolling/blue-green.
// - auto_promote: boolean (no validation beyond the parser's
// true/false parse; the field is always populated)
//
// The validator is pure (no I/O). Violations return a clear error
// listing every problem found, mirroring the per-field style of
// ServiceValidator.
type UpdateValidator struct{}
// Validate validates the UpdateBlock on the given spec. The spec must
// be non-nil and carry a Count (services have count ≥ 1 per
// ServiceValidator). When spec.Update is nil the validator returns an
// error (the update block is required for Service; this validator
// assumes the caller has already established the spec is a Service).
func (UpdateValidator) Validate(spec *jobspec.WorkloadSpec) error {
if spec == nil {
return fmt.Errorf("schema/Update: spec is nil")
}
if spec.Update == nil {
return fmt.Errorf("schema/Update: update block is nil")
}
var errs []string
u := spec.Update
switch u.Strategy {
case "rolling", "canary", "blue-green":
case "":
errs = append(errs, "update strategy required (one of rolling, canary, blue-green)")
default:
errs = append(errs, fmt.Sprintf("update strategy %q invalid (want one of rolling, canary, blue-green)", u.Strategy))
}
// max_parallel defaults to 1 when unset (0); validate the range
// only when the user has set it explicitly.
if u.MaxParallel != 0 {
if u.MaxParallel < 1 {
errs = append(errs, fmt.Sprintf("update.max_parallel must be ≥ 1, got %d", u.MaxParallel))
}
if spec.Count > 0 && u.MaxParallel > spec.Count {
errs = append(errs, fmt.Sprintf("update.max_parallel %d exceeds count %d (must be 1..count)", u.MaxParallel, spec.Count))
}
}
if u.MinHealthyTime != "" {
if _, err := time.ParseDuration(u.MinHealthyTime); err != nil {
errs = append(errs, fmt.Sprintf("update.min_healthy_time %q is not a valid duration: %v", u.MinHealthyTime, err))
}
}
if u.HealthyDeadline != "" {
if _, err := time.ParseDuration(u.HealthyDeadline); err != nil {
errs = append(errs, fmt.Sprintf("update.healthy_deadline %q is not a valid duration: %v", u.HealthyDeadline, err))
}
}
// canary accepts an integer count (0..count) or a percentage
// ("<n>%" with n in 0..100). The field is only meaningful for the
// canary strategy but we validate the shape regardless so a typo
// in a rolling/blue-green stanza still surfaces.
if u.Canary != "" {
if err := validateCanary(u.Canary, spec.Count); err != nil {
errs = append(errs, err.Error())
}
}
// auto_promote is a bool; no extra validation beyond the parser.
return composeErrors("schema/Update", errs)
}
// validateCanary validates the canary field shape: either an integer
// count (0..count) or a percentage string "<n>%" (n in 0..100). count
// is the spec.Count; when count is 0 (e.g. a DaemonSet or unset), the
// integer-count upper bound is not enforced (only the percentage
// bound is enforced, since percentage does not depend on count).
func validateCanary(canary string, count int) error {
c := strings.TrimSpace(canary)
if c == "" {
return nil
}
if strings.HasSuffix(c, "%") {
nStr := strings.TrimSuffix(c, "%")
n, err := strconv.Atoi(strings.TrimSpace(nStr))
if err != nil {
return fmt.Errorf("update.canary %q is not a valid percentage (want \"<n>%%\")", canary)
}
if n < 0 || n > 100 {
return fmt.Errorf("update.canary percentage %d out of range (want 0..100)", n)
}
return nil
}
n, err := strconv.Atoi(c)
if err != nil {
return fmt.Errorf("update.canary %q is not a valid count or percentage (want integer or \"<n>%%\")", canary)
}
if n < 0 {
return fmt.Errorf("update.canary count %d must be ≥ 0", n)
}
if count > 0 && n > count {
return fmt.Errorf("update.canary count %d exceeds count %d (must be 0..count)", n, count)
}
return nil
}
+464
View File
@@ -0,0 +1,464 @@
package schema
import (
"strings"
"testing"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
)
func TestUpdateValidator_ValidRolling(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
MaxParallel: 2,
MinHealthyTime: "30s",
HealthyDeadline: "5m",
},
}
v := UpdateValidator{}
if err := v.Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestUpdateValidator_ValidCanary(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
MaxParallel: 2,
MinHealthyTime: "30s",
HealthyDeadline: "5m",
Canary: "10%",
AutoPromote: true,
},
}
if err := (UpdateValidator{}).Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestUpdateValidator_ValidBlueGreen(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 3,
Update: &jobspec.UpdateBlock{
Strategy: "blue-green",
MinHealthyTime: "1m",
HealthyDeadline: "10m",
},
}
if err := (UpdateValidator{}).Validate(spec); err != nil {
t.Fatalf("expected nil, got %v", err)
}
}
func TestUpdateValidator_ValidCanaryIntegerCount(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "1",
},
}
if err := (UpdateValidator{}).Validate(spec); err != nil {
t.Fatalf("integer canary count 1 should be valid, got %v", err)
}
}
func TestUpdateValidator_ValidCanaryZeroPercent(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "0%",
},
}
if err := (UpdateValidator{}).Validate(spec); err != nil {
t.Fatalf("0%% canary should be valid, got %v", err)
}
}
func TestUpdateValidator_ValidCanaryHundredPercent(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "100%",
},
}
if err := (UpdateValidator{}).Validate(spec); err != nil {
t.Fatalf("100%% canary should be valid, got %v", err)
}
}
func TestUpdateValidator_EmptyDurationsOK(t *testing.T) {
// Empty min_healthy_time / healthy_deadline should be accepted
// (they are optional; defaults are applied by the executor).
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
},
}
if err := (UpdateValidator{}).Validate(spec); err != nil {
t.Fatalf("empty durations should be valid, got %v", err)
}
}
func TestUpdateValidator_NilSpec(t *testing.T) {
if err := (UpdateValidator{}).Validate(nil); err == nil {
t.Fatal("expected error for nil spec")
}
}
func TestUpdateValidator_NilUpdateBlock(t *testing.T) {
spec := &jobspec.WorkloadSpec{Kind: "Service", Name: "web", Count: 2}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for nil update block")
}
if !strings.Contains(err.Error(), "update block is nil") {
t.Errorf("error = %q, want 'update block is nil'", err.Error())
}
}
func TestUpdateValidator_InvalidStrategy(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Update: &jobspec.UpdateBlock{
Strategy: "recreate",
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for invalid strategy")
}
if !strings.Contains(err.Error(), "strategy") || !strings.Contains(err.Error(), "invalid") {
t.Errorf("error = %q, want 'strategy ... invalid'", err.Error())
}
}
func TestUpdateValidator_EmptyStrategy(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Update: &jobspec.UpdateBlock{
Strategy: "",
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for empty strategy")
}
if !strings.Contains(err.Error(), "strategy required") {
t.Errorf("error = %q, want 'strategy required'", err.Error())
}
}
func TestUpdateValidator_MaxParallelZero(t *testing.T) {
// max_parallel=0 means "unset" → default 1; accepted.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
MaxParallel: 0,
},
}
if err := (UpdateValidator{}).Validate(spec); err != nil {
t.Fatalf("max_parallel=0 (unset) should be valid, got %v", err)
}
}
func TestUpdateValidator_MaxParallelNegative(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
MaxParallel: -1,
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for negative max_parallel")
}
if !strings.Contains(err.Error(), "max_parallel") {
t.Errorf("error = %q, want 'max_parallel'", err.Error())
}
}
func TestUpdateValidator_MaxParallelExceedsCount(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
MaxParallel: 5,
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for max_parallel > count")
}
if !strings.Contains(err.Error(), "exceeds count") {
t.Errorf("error = %q, want 'exceeds count'", err.Error())
}
}
func TestUpdateValidator_MaxParallelEqualsCount(t *testing.T) {
// max_parallel == count is the upper bound; valid.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 3,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
MaxParallel: 3,
},
}
if err := (UpdateValidator{}).Validate(spec); err != nil {
t.Fatalf("max_parallel==count should be valid, got %v", err)
}
}
func TestUpdateValidator_InvalidMinHealthyTime(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
MinHealthyTime: "not-a-duration",
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for invalid min_healthy_time")
}
if !strings.Contains(err.Error(), "min_healthy_time") {
t.Errorf("error = %q, want 'min_healthy_time'", err.Error())
}
}
func TestUpdateValidator_InvalidHealthyDeadline(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Update: &jobspec.UpdateBlock{
Strategy: "rolling",
HealthyDeadline: "nope",
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for invalid healthy_deadline")
}
if !strings.Contains(err.Error(), "healthy_deadline") {
t.Errorf("error = %q, want 'healthy_deadline'", err.Error())
}
}
func TestUpdateValidator_CanaryNegative(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "-1",
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for negative canary count")
}
if !strings.Contains(err.Error(), "canary") {
t.Errorf("error = %q, want 'canary'", err.Error())
}
}
func TestUpdateValidator_CanaryExceedsCount(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "5",
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for canary > count")
}
if !strings.Contains(err.Error(), "exceeds count") {
t.Errorf("error = %q, want 'exceeds count'", err.Error())
}
}
func TestUpdateValidator_CanaryPercentOver100(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "150%",
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for canary > 100%")
}
if !strings.Contains(err.Error(), "out of range") {
t.Errorf("error = %q, want 'out of range'", err.Error())
}
}
func TestUpdateValidator_CanaryPercentNegative(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "-10%",
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for negative canary percent")
}
if !strings.Contains(err.Error(), "out of range") {
t.Errorf("error = %q, want 'out of range'", err.Error())
}
}
func TestUpdateValidator_CanaryNotANumber(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "abc",
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for non-numeric canary")
}
if !strings.Contains(err.Error(), "not a valid") {
t.Errorf("error = %q, want 'not a valid'", err.Error())
}
}
func TestUpdateValidator_CanaryPercentNotANumber(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "xx%",
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error for non-numeric canary percent")
}
if !strings.Contains(err.Error(), "not a valid percentage") {
t.Errorf("error = %q, want 'not a valid percentage'", err.Error())
}
}
func TestUpdateValidator_CanaryCountZeroOK(t *testing.T) {
// canary=0 is the lower bound; valid.
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 4,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "0",
},
}
if err := (UpdateValidator{}).Validate(spec); err != nil {
t.Fatalf("canary=0 should be valid, got %v", err)
}
}
func TestUpdateValidator_CanaryPercentWithoutCountOK(t *testing.T) {
// A percentage canary does not depend on count; valid even when
// count is 0 (e.g. DaemonSet-shaped spec bypassing the Service
// validator — defensive).
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 0,
Update: &jobspec.UpdateBlock{
Strategy: "canary",
Canary: "25%",
},
}
if err := (UpdateValidator{}).Validate(spec); err != nil {
t.Fatalf("percentage canary with count=0 should be valid, got %v", err)
}
}
func TestUpdateValidator_MultipleErrors(t *testing.T) {
spec := &jobspec.WorkloadSpec{
Kind: "Service",
Name: "web",
Count: 2,
Update: &jobspec.UpdateBlock{
Strategy: "recreate",
MaxParallel: 99,
MinHealthyTime: "nope",
HealthyDeadline: "also-nope",
Canary: "200%",
},
}
err := (UpdateValidator{}).Validate(spec)
if err == nil {
t.Fatal("expected error, got nil")
}
for _, want := range []string{
"strategy",
"max_parallel",
"min_healthy_time",
"healthy_deadline",
"canary",
} {
if !strings.Contains(err.Error(), want) {
t.Errorf("error %q missing %q", err.Error(), want)
}
}
}
// Compile-time assertion that UpdateValidator implements Validator.
var _ Validator = UpdateValidator{}
+30
View File
@@ -0,0 +1,30 @@
// Package sshpush_test contains compile-time assertions that *Transport
// satisfies the emitter.AtomicWriter interface (the Traefik C-10
// atomicity protocol — internal/emitter/traefik_atomic.go). The
// assertion lives here (not in internal/emitter) to avoid an import
// cycle: internal/emitter is imported by this package (fanout.go), so
// internal/emitter cannot import this package.
package sshpush_test
import (
"context"
"testing"
"git.cloudinit.dev/coreci/orca/internal/emitter"
"git.cloudinit.dev/coreci/orca/internal/sshpush"
)
// Compile-time assertion: *sshpush.Transport satisfies
// emitter.AtomicWriter. WriteTraefikDynamic relies on this so the
// Traefik dynamic-config file is written atomically (gate C-10).
var _ emitter.AtomicWriter = (*sshpush.Transport)(nil)
func TestTransportSatisfiesAtomicWriter(t *testing.T) {
// A trivial runtime check that the type conversion is valid; the
// compile-time assertion above is the real test, but this gives
// `go test` a function to run.
tr := sshpush.NewTransport("/nonexistent", "/nonexistent")
var w emitter.AtomicWriter = tr
_ = w
_ = context.Background()
}
+23
View File
@@ -0,0 +1,23 @@
// Package sshpush implements the v0.9 SSH-push transport layer (REQ-073,
// R-001): the CLI on the operator host SSHes to each peer to render files,
// apply configs, and run commands. It replaces the v0.8
// internal/transport mTLS HTTP layer.
//
// The Transport reuses one *ssh.Client per peer across multiple
// operations within a single CLI invocation (I-B-001), retries transient
// failures with exponential backoff (100ms ×2, cap 5s, max 5 attempts —
// reimplemented from the v0.8 transport/retry.go pattern, since
// internal/transport is deprecated and not imported), applies per-call
// timeouts (10s exec, 30s SCP per I-B-001), and fans out to many peers
// with bounded concurrency (default 8, errgroup + semaphore).
//
// Idempotency is content-addressed (C-18): WriteFile / WriteFileIdempotent
// compare the remote file's SHA-256 to the local content and skip the
// write on match — the SSH-push equivalent of the v0.8 X-Orca-Idempotency-Key.
//
// Host-key verification reuses proxmox.TOFUHostKeyCallback (D-035), which
// reads/writes the known_hosts file (certpaths.KnownHostsPath during the
// v0.9 dual-write window; the move to paths.KnownHostsPath happens in
// v0.10-P14). The known_hosts file is flock-protected inside the TOFU
// callback, so the Transport does NOT re-lock.
package sshpush
+113
View File
@@ -0,0 +1,113 @@
package sshpush
import (
"context"
"fmt"
"os"
"sync"
"golang.org/x/sync/errgroup"
"git.cloudinit.dev/coreci/orca/internal/emitter"
)
// DefaultFanoutConcurrency is the default bounded-concurrency limit for
// fan-out operations (I-B-001). The Transport.ExecAll and WriteAll
// methods use this when the caller does not override it.
const DefaultFanoutConcurrency = 8
// ExecAll runs cmd on all peers in parallel with bounded concurrency
// (default 8, I-B-001). Returns per-peer output and per-peer errors. A
// nil entry in the errors map means that peer succeeded; the output map
// contains that peer's stdout. The returned error is non-nil only if
// the fan-out itself failed (e.g., context cancelled before any peer
// ran); per-peer failures are in the errors map.
func (t *Transport) ExecAll(ctx context.Context, peers []string, cmd string) (map[string][]byte, map[string]error) {
return t.ExecAllWithConcurrency(ctx, peers, cmd, DefaultFanoutConcurrency)
}
// ExecAllWithConcurrency is ExecAll with an explicit concurrency limit.
// A limit <= 0 uses DefaultFanoutConcurrency.
func (t *Transport) ExecAllWithConcurrency(ctx context.Context, peers []string, cmd string, concurrency int) (map[string][]byte, map[string]error) {
if concurrency <= 0 {
concurrency = DefaultFanoutConcurrency
}
out := make(map[string][]byte, len(peers))
errs := make(map[string]error, len(peers))
var mu sync.Mutex
g, gctx := errgroup.WithContext(ctx)
g.SetLimit(concurrency)
for _, p := range peers {
peer := p
g.Go(func() error {
o, err := t.Exec(gctx, peer, cmd)
mu.Lock()
defer mu.Unlock()
if err != nil {
errs[peer] = err
return nil // per-peer error; do not cancel the group
}
out[peer] = o
return nil
})
}
_ = g.Wait()
return out, errs
}
// WriteAll writes the given files to each peer in parallel with bounded
// concurrency (default 8, I-B-001). The files map is keyed by peer; each
// peer's files are written sequentially (to preserve order and avoid
// intra-peer races on shared paths). Returns per-peer errors; a peer
// missing from the map or with a nil entry succeeded. The returned
// error is non-nil only if the fan-out itself failed (context cancelled).
func (t *Transport) WriteAll(ctx context.Context, peers []string, files map[string][]emitter.File) map[string]error {
return t.WriteAllWithConcurrency(ctx, peers, files, DefaultFanoutConcurrency)
}
// WriteAllWithConcurrency is WriteAll with an explicit concurrency limit.
// A limit <= 0 uses DefaultFanoutConcurrency.
func (t *Transport) WriteAllWithConcurrency(ctx context.Context, peers []string, files map[string][]emitter.File, concurrency int) map[string]error {
if concurrency <= 0 {
concurrency = DefaultFanoutConcurrency
}
errs := make(map[string]error, len(peers))
var mu sync.Mutex
g, gctx := errgroup.WithContext(ctx)
g.SetLimit(concurrency)
for _, p := range peers {
peer := p
peerFiles := files[peer]
g.Go(func() error {
for _, f := range peerFiles {
mode := parseMode(f.Mode)
if _, err := t.WriteFileIdempotent(gctx, peer, f.Path, []byte(f.Content), mode); err != nil {
mu.Lock()
errs[peer] = fmt.Errorf("sshpush: write %s on %s: %w", f.Path, peer, err)
mu.Unlock()
return nil // per-peer error; do not cancel the group
}
}
return nil
})
}
_ = g.Wait()
return errs
}
// parseMode parses an octal mode string like "0644" into an os.FileMode.
// Returns 0644 on parse failure (a safe default for non-executable
// config files).
func parseMode(s string) os.FileMode {
var m uint32
for _, r := range s {
if r < '0' || r > '7' {
return 0o644
}
m = m<<3 | uint32(r-'0')
}
if m == 0 {
return 0o644
}
return os.FileMode(m)
}
+300
View File
@@ -0,0 +1,300 @@
package sshpush
import (
"context"
"errors"
"fmt"
"sync/atomic"
"testing"
"golang.org/x/crypto/ssh"
"git.cloudinit.dev/coreci/orca/internal/emitter"
)
// --- ExecAll tests ---
func TestExecAll_AllSucceed(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
// Use one server for all peers (same addr).
addr := srv.addr()
peers := []string{addr, addr, addr}
out, errs := tr.ExecAll(context.Background(), peers, "echo hello")
for _, p := range peers {
if e, ok := errs[p]; ok && e != nil {
t.Errorf("peer %s: %v", p, e)
}
if string(out[p]) != "hello\n" {
t.Errorf("out[%s] = %q, want hello\\n", p, out[p])
}
}
}
func TestExecAll_OneFails(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
addr := srv.addr()
// Peer "bad" returns a permanent error via a mock session.
goodPeers := []string{addr}
badPeer := "127.0.0.1:1" // unreachable -> transient dial error, retried, fails
peers := append(goodPeers, badPeer)
out, errs := tr.ExecAll(context.Background(), peers, "echo hello")
if string(out[addr]) != "hello\n" {
t.Errorf("good peer out = %q, want hello\\n", out[addr])
}
if errs[badPeer] == nil {
t.Error("bad peer should have an error")
}
}
func TestExecAll_WithConcurrency(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
addr := srv.addr()
peers := []string{addr, addr, addr, addr}
out, errs := tr.ExecAllWithConcurrency(context.Background(), peers, "echo hello", 2)
for _, p := range peers {
if e := errs[p]; e != nil {
t.Errorf("peer %s: %v", p, e)
}
if string(out[p]) != "hello\n" {
t.Errorf("out[%s] = %q", p, out[p])
}
}
}
func TestExecAll_EmptyPeers(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
out, errs := tr.ExecAll(context.Background(), nil, "echo hello")
if len(out) != 0 || len(errs) != 0 {
t.Errorf("empty peers: out=%v errs=%v", out, errs)
}
}
func TestExecAll_ContextCancelled(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
addr := srv.addr()
ctx, cancel := context.WithCancel(context.Background())
cancel()
out, errs := tr.ExecAll(ctx, []string{addr, addr}, "echo hello")
// With a cancelled context, all peers should fail.
for _, p := range []string{addr, addr} {
if errs[p] == nil && string(out[p]) == "" {
// acceptable: either error or no output
}
}
}
// --- WriteAll tests ---
func TestWriteAll_AllSucceed(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
addr := srv.addr()
files := map[string][]emitter.File{
addr: {
{Path: "/w/a", Content: "alpha\n", Mode: "0644"},
{Path: "/w/b", Content: "beta\n", Mode: "0644"},
},
}
errs := tr.WriteAll(context.Background(), []string{addr}, files)
for p, e := range errs {
if e != nil {
t.Errorf("peer %s: %v", p, e)
}
}
srv.mu.Lock()
if srv.files["/w/a"] != "alpha\n" {
t.Errorf("file a = %q", srv.files["/w/a"])
}
if srv.files["/w/b"] != "beta\n" {
t.Errorf("file b = %q", srv.files["/w/b"])
}
srv.mu.Unlock()
}
func TestWriteAll_OnePeerFails(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
addr := srv.addr()
bad := "127.0.0.1:1"
files := map[string][]emitter.File{
addr: {{Path: "/ok/f", Content: "ok\n", Mode: "0644"}},
bad: {{Path: "/fail/f", Content: "fail\n", Mode: "0644"}},
}
errs := tr.WriteAll(context.Background(), []string{addr, bad}, files)
if errs[addr] != nil {
t.Errorf("good peer should not have error, got %v", errs[addr])
}
if errs[bad] == nil {
t.Error("bad peer should have error")
}
}
func TestWriteAll_EmptyPeers(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
errs := tr.WriteAll(context.Background(), nil, nil)
if len(errs) != 0 {
t.Errorf("empty peers: errs=%v", errs)
}
}
func TestWriteAll_WithConcurrency(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
addr := srv.addr()
files := map[string][]emitter.File{
addr: {{Path: "/c/f", Content: "c\n", Mode: "0644"}},
}
errs := tr.WriteAllWithConcurrency(context.Background(), []string{addr}, files, 4)
for _, e := range errs {
if e != nil {
t.Errorf("peer err: %v", e)
}
}
}
func TestWriteAll_Idempotent(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
srv.mu.Lock()
srv.files["/i/f"] = "same\n"
srv.mu.Unlock()
tr := realTransport(t, srv)
defer tr.Close()
addr := srv.addr()
files := map[string][]emitter.File{
addr: {{Path: "/i/f", Content: "same\n", Mode: "0644"}},
}
// Capture writes to verify idempotent skip.
var writes int64
tr.SetSessionFactory(func(c *ssh.Client) (sshSession, error) {
return &writeCountingSession{srv: srv, writes: &writes}, nil
})
// Pre-populate pool.
client, err := tr.dial(addr)
if err != nil {
t.Fatalf("dial: %v", err)
}
tr.pool.Store(addr, client)
errs := tr.WriteAll(context.Background(), []string{addr}, files)
if errs[addr] != nil {
t.Errorf("WriteAll err: %v", errs[addr])
}
// sha256sum returns a hash that matches -> no write command.
srv.mu.Lock()
content := srv.files["/i/f"]
srv.mu.Unlock()
if content != "same\n" {
t.Errorf("content changed to %q", content)
}
}
// writeCountingSession counts how many write commands (mkdir + cat >)
// are issued; returns the server's file content for sha256sum.
type writeCountingSession struct {
srv *fakeSSHServer
writes *int64
}
func (w *writeCountingSession) CombinedOutput(cmd string) ([]byte, error) {
c := trim(cmd)
if startsWith(c, "sha256sum ") {
path := unquote(trimPrefix(c, "sha256sum "))
path = trimSuffix(path, " 2>/dev/null")
w.srv.mu.Lock()
content, ok := w.srv.files[path]
w.srv.mu.Unlock()
if !ok {
return []byte(""), nil
}
sum := sha256HexStr([]byte(content))
return []byte(sum + " " + path + "\n"), nil
}
if startsWith(c, "mkdir -p ") && contains(c, "cat >") {
atomic.AddInt64(w.writes, 1)
return nil, nil
}
return nil, nil
}
func (w *writeCountingSession) Close() error { return nil }
// Local string helpers to avoid importing strings in a way that
// conflicts with the test's existing imports.
func trim(s string) string {
for len(s) > 0 && (s[0] == ' ' || s[0] == '\t') {
s = s[1:]
}
for len(s) > 0 && (s[len(s)-1] == ' ' || s[len(s)-1] == '\t') {
s = s[:len(s)-1]
}
return s
}
func startsWith(s, prefix string) bool { return len(s) >= len(prefix) && s[:len(prefix)] == prefix }
func contains(s, sub string) bool {
return len(sub) == 0 || (len(s) >= len(sub) && indexOf(s, sub) >= 0)
}
func indexOf(s, sub string) int {
for i := 0; i+len(sub) <= len(s); i++ {
if s[i:i+len(sub)] == sub {
return i
}
}
return -1
}
func trimPrefix(s, prefix string) string {
if startsWith(s, prefix) {
return s[len(prefix):]
}
return s
}
func trimSuffix(s, suffix string) string {
if len(s) >= len(suffix) && s[len(s)-len(suffix):] == suffix {
return s[:len(s)-len(suffix)]
}
return s
}
func TestDefaultFanoutConcurrency(t *testing.T) {
if DefaultFanoutConcurrency != 8 {
t.Errorf("DefaultFanoutConcurrency = %d, want 8", DefaultFanoutConcurrency)
}
}
func TestParseMode_Fanout(t *testing.T) {
// parseMode is in fanout.go; sanity-check here too.
for _, tc := range []struct{ in, want string }{
{"0644", "644"},
{"0755", "755"},
{"bad", "644"},
} {
got := fmt.Sprintf("%o", parseMode(tc.in))
if got != tc.want {
t.Errorf("parseMode(%q) = %s, want %s", tc.in, got, tc.want)
}
}
}
var _ = errors.New
+101
View File
@@ -0,0 +1,101 @@
package sshpush
import (
"context"
"crypto/sha256"
"encoding/hex"
"fmt"
"math/rand"
"os"
"strings"
)
// WriteFileIdempotent writes content to peer:path atomically (write-to-tmp
// + mv, REQ-074) with mode, but only if the remote file's SHA-256 differs
// from the local content's SHA-256 (C-18 content-addressed idempotency —
// the SSH-push equivalent of the v0.8 X-Orca-Idempotency-Key).
//
// Returns written=true if the file was written, written=false if the
// content already matched (skip). The default per-SCP timeout is
// SCPTimeout (I-B-001).
//
// Atomicity: the content is written to a temp file in the same directory
// as the target, then `mv`'d into place. The temp file is mode-appended
// (e.g. `/etc/orca/foo.conf.orca-tmp-<rand>`) so the rename is atomic on
// POSIX filesystems.
func (t *Transport) WriteFileIdempotent(ctx context.Context, peer string, path string, content []byte, mode os.FileMode) (bool, error) {
localHash := sha256Hex(content)
remoteHash, err := t.remoteSHA256(ctx, peer, path)
if err == nil && remoteHash != "" && strings.EqualFold(remoteHash, localHash) {
return false, nil
}
if err := t.writeFile(ctx, peer, path, content, mode); err != nil {
return false, err
}
return true, nil
}
// writeFile writes content to peer:path atomically (write-to-tmp + mv).
// It writes the content via a single SSH exec (cat heredoc + chmod + mv),
// keeping the transfer in one round-trip. The temp file lives next to the
// target so the rename is atomic.
func (t *Transport) writeFile(ctx context.Context, peer string, path string, content []byte, mode os.FileMode) error {
dir, base := splitDir(path)
tmpName := fmt.Sprintf(".orca-tmp-%s", randomToken(8))
tmpPath := base + "/" + tmpName
if dir == "" {
tmpPath = tmpName
}
// Build the remote command: mkdir -p <dir> && cat > <tmp> <<'EOF'
// ... EOF && chmod <mode> <tmp> && mv <tmp> <path>. The heredoc
// delimiter is chosen to not appear in the content (we use a fixed
// marker; content with the marker would break, but the marker is
// sufficiently unusual).
const eof = "ORCA_PUSH_EOF_a1b2c3"
modeStr := fmt.Sprintf("%04o", uint32(mode.Perm()))
cmd := fmt.Sprintf(
"mkdir -p %s && cat > %s <<'%s'\n%s\n%s\nchmod %s %s && mv -f %s %s",
shellQuote(base),
shellQuote(tmpPath),
eof,
string(content),
eof,
modeStr,
shellQuote(tmpPath),
shellQuote(tmpPath),
shellQuote(path),
)
execCtx, cancel := context.WithTimeout(ctx, SCPTimeout)
defer cancel()
if _, err := t.execWithRetry(execCtx, peer, cmd, true); err != nil {
return fmt.Errorf("sshpush: write %s: %w", path, err)
}
return nil
}
// sha256Hex returns the lowercase hex SHA-256 digest of b.
func sha256Hex(b []byte) string {
sum := sha256.Sum256(b)
return hex.EncodeToString(sum[:])
}
// splitDir returns the directory and the directory itself (for mkdir).
// For "/etc/orca/foo.conf" it returns ("/etc/orca", "/etc/orca"). For
// "foo.conf" it returns ("", ".").
func splitDir(path string) (dir, base string) {
idx := strings.LastIndex(path, "/")
if idx < 0 {
return "", "."
}
return path[:idx], path[:idx]
}
// randomToken returns a random hex token of the given byte length. Used
// for temp-file naming to avoid collisions under parallel fan-out.
func randomToken(n int) string {
b := make([]byte, n)
for i := range b {
b[i] = byte(rand.Intn(256))
}
return hex.EncodeToString(b)
}
+348
View File
@@ -0,0 +1,348 @@
package sshpush
import (
"context"
"crypto/sha256"
"encoding/hex"
"errors"
"fmt"
"os"
"strings"
"testing"
"golang.org/x/crypto/ssh"
)
func TestSha256Hex(t *testing.T) {
got := sha256Hex([]byte("hello"))
want := "2cf24dba5fb0a30e26e83b2ac5b9e29e1b161e5c1fa7425e73043362938b9824"
if got != want {
t.Errorf("sha256Hex = %q, want %q", got, want)
}
}
func TestSplitDir(t *testing.T) {
dir, base := splitDir("/etc/orca/foo.conf")
if dir != "/etc/orca" || base != "/etc/orca" {
t.Errorf("splitDir(/etc/orca/foo.conf) = (%q,%q), want (/etc/orca,/etc/orca)", dir, base)
}
dir, base = splitDir("foo.conf")
if dir != "" || base != "." {
t.Errorf("splitDir(foo.conf) = (%q,%q), want (\"\",.)", dir, base)
}
}
func TestRandomToken(t *testing.T) {
a := randomToken(8)
b := randomToken(8)
if a == b {
t.Error("randomToken returned same value twice")
}
if len(a) != 16 { // 8 bytes hex = 16 chars
t.Errorf("randomToken(8) len = %d, want 16", len(a))
}
}
// --- WriteFileIdempotent tests via the fake SSH server ---
func TestWriteFileIdempotent_WritesWhenFileMissing(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
content := []byte("first content\n")
written, err := tr.WriteFileIdempotent(context.Background(), srv.addr(), "/etc/orca/a.conf", content, 0o644)
if err != nil {
t.Fatalf("WriteFileIdempotent: %v", err)
}
if !written {
t.Error("written=false, want true (file was missing)")
}
srv.mu.Lock()
got := srv.files["/etc/orca/a.conf"]
srv.mu.Unlock()
if got != string(content) {
t.Errorf("remote file = %q, want %q", got, string(content))
}
}
func TestWriteFileIdempotent_SkipsWhenContentMatches(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
content := []byte("same content\n")
srv.mu.Lock()
srv.files["/etc/orca/b.conf"] = string(content)
srv.mu.Unlock()
tr := realTransport(t, srv)
defer tr.Close()
written, err := tr.WriteFileIdempotent(context.Background(), srv.addr(), "/etc/orca/b.conf", content, 0o644)
if err != nil {
t.Fatalf("WriteFileIdempotent: %v", err)
}
if written {
t.Error("written=true, want false (content matched)")
}
}
func TestWriteFileIdempotent_WritesWhenContentDiffers(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
srv.mu.Lock()
srv.files["/etc/orca/c.conf"] = "old content\n"
srv.mu.Unlock()
tr := realTransport(t, srv)
defer tr.Close()
newContent := []byte("new content\n")
written, err := tr.WriteFileIdempotent(context.Background(), srv.addr(), "/etc/orca/c.conf", newContent, 0o644)
if err != nil {
t.Fatalf("WriteFileIdempotent: %v", err)
}
if !written {
t.Error("written=false, want true (content differed)")
}
srv.mu.Lock()
got := srv.files["/etc/orca/c.conf"]
srv.mu.Unlock()
if got != string(newContent) {
t.Errorf("remote file = %q, want %q", got, string(newContent))
}
}
func TestWriteFile_DelegatesToIdempotent(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
content := []byte("delegated\n")
if err := tr.WriteFile(context.Background(), srv.addr(), "/etc/orca/d.conf", content, 0o600); err != nil {
t.Fatalf("WriteFile: %v", err)
}
srv.mu.Lock()
got := srv.files["/etc/orca/d.conf"]
srv.mu.Unlock()
if got != string(content) {
t.Errorf("remote file = %q, want %q", got, string(content))
}
}
// --- Pure-logic idempotency tests via mock session (no SSH server) ---
// mockHashSession returns the hash of the file matching the sha256sum
// command's path argument; for write commands (mkdir + cat >), it
// records the write. This lets us test the idempotency decision logic
// without a real SSH server.
type mockHashSession struct {
files map[string]string
out []byte
err error
cmd string
writeHook func(cmd string)
}
func (m *mockHashSession) CombinedOutput(cmd string) ([]byte, error) {
m.cmd = cmd
c := strings.TrimSpace(cmd)
if strings.HasPrefix(c, "sha256sum ") {
rest := strings.TrimSpace(strings.TrimPrefix(c, "sha256sum "))
rest = strings.TrimSuffix(rest, " 2>/dev/null")
rest = strings.TrimSpace(rest)
path := unquote(rest)
content, ok := m.files[path]
if !ok {
return []byte(""), nil
}
sum := sha256.Sum256([]byte(content))
return []byte(hex.EncodeToString(sum[:]) + " " + path + "\n"), nil
}
if strings.HasPrefix(c, "mkdir -p ") && strings.Contains(c, "cat >") {
if m.writeHook != nil {
m.writeHook(c)
}
return nil, nil
}
return m.out, m.err
}
func (m *mockHashSession) Close() error { return nil }
func TestWriteFileIdempotent_MockSkip(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
files := map[string]string{"/x/f": "match"}
var writes int
tr.SetSessionFactory(func(c *ssh.Client) (sshSession, error) {
return &mockHashSession{
files: files,
writeHook: func(string) { writes++ },
}, nil
})
client, err := tr.dial(srv.addr())
if err != nil {
t.Fatalf("dial: %v", err)
}
tr.pool.Store(srv.addr(), client)
written, err := tr.WriteFileIdempotent(context.Background(), srv.addr(), "/x/f", []byte("match"), 0o644)
if err != nil {
t.Fatalf("WriteFileIdempotent: %v", err)
}
if written {
t.Error("written=true, want false (hash matched)")
}
if writes != 0 {
t.Errorf("writes = %d, want 0 (no write on hash match)", writes)
}
}
func TestWriteFileIdempotent_MockWrite(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
files := map[string]string{"/x/f": "old"}
var writes int
tr.SetSessionFactory(func(c *ssh.Client) (sshSession, error) {
return &mockHashSession{
files: files,
writeHook: func(string) { writes++ },
}, nil
})
client, err := tr.dial(srv.addr())
if err != nil {
t.Fatalf("dial: %v", err)
}
tr.pool.Store(srv.addr(), client)
written, err := tr.WriteFileIdempotent(context.Background(), srv.addr(), "/x/f", []byte("new"), 0o644)
if err != nil {
t.Fatalf("WriteFileIdempotent: %v", err)
}
if !written {
t.Error("written=false, want true (hash differed)")
}
if writes != 1 {
t.Errorf("writes = %d, want 1", writes)
}
}
func TestWriteFileIdempotent_MockMissingFile(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
files := map[string]string{}
var writes int
tr.SetSessionFactory(func(c *ssh.Client) (sshSession, error) {
return &mockHashSession{
files: files,
writeHook: func(string) { writes++ },
}, nil
})
client, err := tr.dial(srv.addr())
if err != nil {
t.Fatalf("dial: %v", err)
}
tr.pool.Store(srv.addr(), client)
written, err := tr.WriteFileIdempotent(context.Background(), srv.addr(), "/x/new", []byte("fresh"), 0o644)
if err != nil {
t.Fatalf("WriteFileIdempotent: %v", err)
}
if !written {
t.Error("written=false, want true (file missing)")
}
if writes != 1 {
t.Errorf("writes = %d, want 1", writes)
}
}
func TestRemoteSHA256_ParsesDigest(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
srv.mu.Lock()
srv.files["/x/h"] = "abc"
srv.mu.Unlock()
tr := realTransport(t, srv)
defer tr.Close()
got, err := tr.remoteSHA256(context.Background(), srv.addr(), "/x/h")
if err != nil {
t.Fatalf("remoteSHA256: %v", err)
}
want := fmt.Sprintf("%x", sha256.Sum256([]byte("abc")))
if got != want {
t.Errorf("remoteSHA256 = %q, want %q", got, want)
}
}
func TestRemoteSHA256_MissingFile(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
got, err := t_remoteSHA256_noErr(t, tr, srv.addr(), "/missing")
if err != nil {
t.Fatalf("remoteSHA256: %v", err)
}
if got != "" {
t.Errorf("remoteSHA256 = %q, want empty for missing file", got)
}
}
func t_remoteSHA256_noErr(t *testing.T, tr *Transport, peer, path string) (string, error) {
t.Helper()
return tr.remoteSHA256(context.Background(), peer, path)
}
func TestWriteFile_ModeApplied(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
if err := tr.WriteFile(context.Background(), srv.addr(), "/m/f", []byte("mode"), 0o755); err != nil {
t.Fatalf("WriteFile: %v", err)
}
srv.mu.Lock()
got := srv.files["/m/f"]
srv.mu.Unlock()
if got != "mode" {
t.Errorf("content = %q, want mode", got)
}
}
func TestWriteFileIdempotent_ExecError(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
tr.SetSessionFactory(func(c *ssh.Client) (sshSession, error) {
return &mockSession{err: fmt.Errorf("%w: boom", ErrPermanent)}, nil
})
client, err := tr.dial(srv.addr())
if err != nil {
t.Fatalf("dial: %v", err)
}
tr.pool.Store(srv.addr(), client)
_, err = tr.WriteFileIdempotent(context.Background(), srv.addr(), "/x/f", []byte("z"), 0o644)
if err == nil {
t.Fatal("expected error, got nil")
}
}
func TestParseMode(t *testing.T) {
for _, tc := range []struct {
in string
want os.FileMode
}{
{"0644", 0o644},
{"0755", 0o755},
{"0600", 0o600},
{"bad", 0o644},
{"", 0o644},
{"0", 0o644},
} {
got := parseMode(tc.in)
if got != tc.want {
t.Errorf("parseMode(%q) = %o, want %o", tc.in, got, tc.want)
}
}
}
var _ = errors.New
+483
View File
@@ -0,0 +1,483 @@
package sshpush
import (
"bytes"
"context"
"errors"
"fmt"
"math/rand"
"net"
"os"
"strings"
"sync"
"time"
"golang.org/x/crypto/ssh"
"git.cloudinit.dev/coreci/orca/internal/proxmox"
)
// Default timeouts and retry parameters (REQ-073, I-B-001).
const (
// ExecTimeout is the default per-exec timeout for a single SSH
// command (I-B-001).
ExecTimeout = 10 * time.Second
// SCPTimeout is the default per-SCP timeout for a single file
// transfer (I-B-001).
SCPTimeout = 30 * time.Second
// DialTimeout is the default SSH dial timeout.
DialTimeout = 15 * time.Second
// RetryInitial is the first backoff interval (v0.8 transport/retry.go).
RetryInitial = 100 * time.Millisecond
// RetryMax is the cap on backoff between attempts.
RetryMax = 5 * time.Second
// RetryMaxAttempts is the total attempt count (including the first).
RetryMaxAttempts = 5
)
// Sentinel errors. ErrTransient marks a transient failure worth
// retrying; ErrPermanent marks a non-retryable failure (auth, host-key
// mismatch, validation). These mirror the v0.8 transport sentinels
// (reimplemented here since internal/transport is not imported).
var (
ErrTransient = errors.New("sshpush: transient error")
ErrPermanent = errors.New("sshpush: permanent error")
ErrNotConnected = errors.New("sshpush: not connected")
)
// Transport is the SSH-push transport (REQ-073). It reuses one
// *ssh.Client per peer across multiple operations within a single CLI
// invocation (I-B-001). The zero value is NOT usable; construct one with
// NewTransport.
type Transport struct {
// pool caches *ssh.Client per peer address ("host:port").
pool sync.Map
// keyPath is the SSH private key path (Ed25519, D-037).
keyPath string
// knownHostsPath is the v0.9 known_hosts path (paths.KnownHostsPath()
// = ClusterDir()/known_hosts). It is stored for the v0.10-P14 migration
// when proxmox.TOFUHostKeyCallback will accept a path parameter; today
// the callback reads certpaths.KnownHostsPath() (the v0.8 flat layout)
// directly, so this field is not yet read by dial(). Tests set
// $ORCA_HOME so certpaths.KnownHostsPath() resolves under the temp dir.
knownHostsPath string
// user is the remote SSH user (default "orca", D-037).
user string
// signer is the parsed SSH private key signer, set lazily on first
// dial.
signer ssh.Signer
signErr error
// signerOnce guards signer initialization.
signerOnce sync.Once
// dialer is the SSH dialer. Tests override it to inject a mock
// server. The default uses ssh.DialContext via the context-aware
// wrapper.
dialer sshDialer
// sessionFactory returns a new session for a given client. Tests
// override it to inject mock sessions without a real *ssh.Client.
// When nil, the default (*ssh.Client).NewSession is used.
sessionFactory func(*ssh.Client) (sshSession, error)
// mu guards the closed flag (pool iteration is sync.Map.Range).
closed bool
mu sync.Mutex
}
// sshSession is the minimal *ssh.Session surface the transport uses.
// It lets tests substitute a mock without a real SSH server.
type sshSession interface {
CombinedOutput(cmd string) ([]byte, error)
Close() error
}
// sshDialer is the SSH dialer interface (mirrors proxmox.sshDialerType).
// The default uses ssh.Dial; tests inject mocks that return a fake
// *ssh.Client or an error.
type sshDialer interface {
DialContext(ctx context.Context, network, addr string, config *ssh.ClientConfig) (*ssh.Client, error)
}
// defaultSSHDialer wraps ssh.Dial with a context-aware connect timeout.
type defaultSSHDialer struct{}
func (defaultSSHDialer) DialContext(ctx context.Context, network, addr string, config *ssh.ClientConfig) (*ssh.Client, error) {
d := net.Dialer{Timeout: config.Timeout}
if d.Timeout == 0 {
d.Timeout = DialTimeout
}
conn, err := d.DialContext(ctx, network, addr)
if err != nil {
return nil, err
}
sshConn, chans, reqs, err := ssh.NewClientConn(conn, addr, config)
if err != nil {
_ = conn.Close()
return nil, err
}
return ssh.NewClient(sshConn, chans, reqs), nil
}
// NewTransport returns a Transport configured with the given SSH
// private key path and known_hosts path. The known_hosts path is the v0.9
// location (paths.KnownHostsPath); it is stored for the v0.10-P14
// migration when the TOFU callback will accept a path parameter. Today
// dial() delegates host-key verification to proxmox.TOFUHostKeyCallback,
// which reads certpaths.KnownHostsPath() (the v0.8 flat layout under
// $ORCA_HOME) directly — so callers must ensure $ORCA_HOME points at the
// cluster root (the CLI sets this up). The remote user defaults to
// "orca" (D-037); override with SetUser. The dialer defaults to the
// real ssh.Dial-based dialer; tests call SetDialer to inject a mock.
func NewTransport(keyPath, knownHostsPath string) *Transport {
return &Transport{
keyPath: keyPath,
knownHostsPath: knownHostsPath,
user: "orca",
dialer: defaultSSHDialer{},
}
}
// SetUser overrides the remote SSH user (default "orca").
func (t *Transport) SetUser(user string) {
if user != "" {
t.user = user
}
}
// SetDialer overrides the SSH dialer (for tests).
func (t *Transport) SetDialer(d sshDialer) {
if d != nil {
t.dialer = d
}
}
// SetSessionFactory overrides the session factory (for tests). The
// factory is called per-exec/write/read to obtain a fresh session; it
// must close the session when the test mock is done, or the transport
// will call Close on the returned session.
func (t *Transport) SetSessionFactory(f func(*ssh.Client) (sshSession, error)) {
t.sessionFactory = f
}
// dial returns the cached *ssh.Client for peer, dialing and caching on
// first use (I-B-001 connection pooling). Returns an error if the dial
// fails or the transport is closed.
func (t *Transport) dial(peer string) (*ssh.Client, error) {
t.mu.Lock()
if t.closed {
t.mu.Unlock()
return nil, ErrPermanent
}
t.mu.Unlock()
if c, ok := t.pool.Load(peer); ok {
return c.(*ssh.Client), nil
}
// Lazily parse the private key signer (once across all dials).
t.signerOnce.Do(func() {
keyBytes, err := os.ReadFile(t.keyPath)
if err != nil {
t.signErr = fmt.Errorf("sshpush: read key %s: %w", t.keyPath, err)
return
}
s, err := ssh.ParsePrivateKey(keyBytes)
if err != nil {
t.signErr = fmt.Errorf("sshpush: parse key: %w", err)
return
}
t.signer = s
})
if t.signErr != nil {
return nil, t.signErr
}
// Host-key verification reuses the v0.8 TOFU wrapper (D-035). The
// known_hosts file is flock-protected inside the callback on
// first-connect capture, so we do NOT re-lock here.
cb, err := proxmox.TOFUHostKeyCallback(peer, nil)
if err != nil {
return nil, fmt.Errorf("sshpush: host-key callback: %w", err)
}
config := &ssh.ClientConfig{
User: t.user,
Auth: []ssh.AuthMethod{ssh.PublicKeys(t.signer)},
HostKeyCallback: cb,
Timeout: DialTimeout,
}
ctx, cancel := context.WithTimeout(context.Background(), DialTimeout)
defer cancel()
client, err := t.dialer.DialContext(ctx, "tcp", peer, config)
if err != nil {
return nil, classifyDialErr(err)
}
// Race: two goroutines dialing the same peer concurrently both
// create a client. Last-wins; the loser is closed. This is rare
// (dial is rare and the pool hit short-circuits) and harmless.
if existing, loaded := t.pool.LoadOrStore(peer, client); loaded {
_ = client.Close()
return existing.(*ssh.Client), nil
}
return client, nil
}
// Exec runs cmd on peer over SSH and returns its combined output. The
// default per-exec timeout is ExecTimeout (I-B-001); override by
// passing a context with a shorter deadline. Transient failures are
// retried with exponential backoff (100ms ×2, cap 5s, max 5 attempts —
// the v0.8 transport/retry.go pattern, reimplemented here).
func (t *Transport) Exec(ctx context.Context, peer string, cmd string) ([]byte, error) {
return t.execWithRetry(ctx, peer, cmd, true)
}
// execWithRetry runs the exec with retry. exec is treated as
// idempotent (read-only) for retry purposes; the idempotency helpers
// (WriteFileIdempotent) handle writes.
func (t *Transport) execWithRetry(ctx context.Context, peer string, cmd string, idempotent bool) ([]byte, error) {
var lastErr error
for attempt := 1; attempt <= RetryMaxAttempts; attempt++ {
if err := ctx.Err(); err != nil {
return nil, err
}
out, err := t.execOnce(ctx, peer, cmd)
if err == nil {
return out, nil
}
if errors.Is(err, ErrPermanent) {
return nil, err
}
lastErr = err
if attempt == RetryMaxAttempts {
break
}
if !isTransient(err) {
return nil, err
}
wait := backoff(RetryInitial, RetryMax, attempt)
timer := time.NewTimer(wait)
select {
case <-ctx.Done():
timer.Stop()
return nil, ctx.Err()
case <-timer.C:
}
}
return nil, lastErr
}
// execOnce runs the command a single time against peer.
func (t *Transport) execOnce(ctx context.Context, peer string, cmd string) ([]byte, error) {
client, err := t.dial(peer)
if err != nil {
return nil, classifyDialErr(err)
}
sess, err := t.newSession(client)
if err != nil {
return nil, fmt.Errorf("sshpush: new session: %w", err)
}
defer sess.Close()
type result struct {
out []byte
err error
}
ch := make(chan result, 1)
go func() {
out, err := sess.CombinedOutput(cmd)
ch <- result{out, err}
}()
timeout := ExecTimeout
if dl, ok := ctx.Deadline(); ok {
if remaining := time.Until(dl); remaining > 0 && remaining < timeout {
timeout = remaining
}
}
select {
case <-ctx.Done():
return nil, ctx.Err()
case <-time.After(timeout):
return nil, fmt.Errorf("sshpush: exec timeout after %s: %w", timeout, ErrTransient)
case r := <-ch:
if r.err != nil {
return r.out, classifyExecErr(r.err)
}
return r.out, nil
}
}
// newSession returns a session for client, using the override factory
// when set (tests), otherwise the real *ssh.Client.NewSession.
func (t *Transport) newSession(client *ssh.Client) (sshSession, error) {
if t.sessionFactory != nil {
return t.sessionFactory(client)
}
s, err := client.NewSession()
if err != nil {
return nil, err
}
return &realSession{Session: s}, nil
}
// realSession wraps *ssh.Session to satisfy the sshSession interface.
type realSession struct {
*ssh.Session
}
func (r *realSession) CombinedOutput(cmd string) ([]byte, error) {
return r.Session.CombinedOutput(cmd)
}
// WriteFile SCPs content to peer:path atomically (write-to-tmp + mv,
// REQ-074). The default per-SCP timeout is SCPTimeout (I-B-001).
// Idempotency: if the file already exists with the same SHA-256, the
// write is skipped (C-18). Use WriteFileIdempotent for the explicit
// written/skipped result.
func (t *Transport) WriteFile(ctx context.Context, peer string, path string, content []byte, mode os.FileMode) error {
_, err := t.WriteFileIdempotent(ctx, peer, path, content, mode)
return err
}
// ReadFile reads the file at peer:path via SSH cat.
func (t *Transport) ReadFile(ctx context.Context, peer string, path string) ([]byte, error) {
cmd := fmt.Sprintf("cat %s", shellQuote(path))
out, err := t.Exec(ctx, peer, cmd)
if err != nil {
return nil, err
}
return out, nil
}
// Close closes all pooled SSH clients (REQ-073). Safe to call
// multiple times; subsequent calls are no-ops.
func (t *Transport) Close() error {
t.mu.Lock()
if t.closed {
t.mu.Unlock()
return nil
}
t.closed = true
t.mu.Unlock()
var firstErr error
t.pool.Range(func(key, value any) bool {
if c, ok := value.(*ssh.Client); ok {
if err := c.Close(); err != nil && firstErr == nil {
firstErr = err
}
}
t.pool.Delete(key)
return true
})
return firstErr
}
// backoff returns the wait duration for the n-th attempt (1-indexed).
// Formula: min(Initial * 2^(n-1), Max), with up to 25% jitter (matches
// v0.8 transport/retry.go).
func backoff(initial, max time.Duration, n int) time.Duration {
d := initial
for i := 1; i < n; i++ {
d *= 2
if d > max {
d = max
break
}
}
if d <= 0 {
return 0
}
jitter := time.Duration(rand.Int63n(int64(d) / 2))
d = d - d/4 + jitter
if d < 0 {
d = 0
}
return d
}
// isTransient reports whether err looks like a transient failure worth
// retrying (mirrors v0.8 transport.IsTransient, reimplemented here).
func isTransient(err error) bool {
if err == nil {
return false
}
if errors.Is(err, ErrTransient) {
return true
}
if errors.Is(err, ErrPermanent) {
return false
}
s := err.Error()
for _, sub := range []string{
"connection refused", "i/o timeout", "EOF",
"no such host", "connection reset", "timeout",
"deadline exceeded", "temporarily unavailable",
} {
if strings.Contains(s, sub) {
return true
}
}
return false
}
// classifyDialErr converts a raw ssh.Dial error into a transport error
// (transient vs permanent). Auth failures and host-key mismatches are
// permanent; everything else is transient.
func classifyDialErr(err error) error {
if err == nil {
return nil
}
s := err.Error()
if strings.Contains(s, "unable to authenticate") || strings.Contains(s, "handshake failed") {
return fmt.Errorf("%w: %v", ErrPermanent, err)
}
if strings.Contains(s, "host key") && strings.Contains(s, "mismatch") {
return fmt.Errorf("%w: %v", ErrPermanent, err)
}
if strings.Contains(s, "knownhosts") {
return fmt.Errorf("%w: %v", ErrPermanent, err)
}
return fmt.Errorf("%w: %v", ErrTransient, err)
}
// classifyExecErr converts a raw session exec error into a transport
// error. Non-zero exit codes are NOT transient (the command ran; the
// failure is logical, not network). Session-creation failures and
// network-level errors are transient.
func classifyExecErr(err error) error {
if err == nil {
return nil
}
var exitErr *ssh.ExitError
if errors.As(err, &exitErr) {
return fmt.Errorf("%w: exit %d", ErrPermanent, exitErr.ExitStatus())
}
s := err.Error()
for _, sub := range []string{"EOF", "session closed", "channel closed"} {
if strings.Contains(s, sub) {
return fmt.Errorf("%w: %v", ErrTransient, err)
}
}
return fmt.Errorf("%w: %v", ErrPermanent, err)
}
// shellQuote single-quotes a path for safe shell interpolation. It
// escapes embedded single-quotes via the standard '\” idiom.
func shellQuote(s string) string {
return "'" + strings.ReplaceAll(s, "'", "'\\''") + "'"
}
// remoteSHA256 returns the SHA-256 of the file at peer:path via SSH
// `sha256sum`, or ("", error) if the file is missing or the command
// fails. The returned hash is the hex digest (lowercase, no filename).
func (t *Transport) remoteSHA256(ctx context.Context, peer string, path string) (string, error) {
cmd := fmt.Sprintf("sha256sum %s 2>/dev/null", shellQuote(path))
out, err := t.execWithRetry(ctx, peer, cmd, true)
if err != nil {
return "", err
}
out = bytes.TrimSpace(out)
if len(out) == 0 {
return "", nil
}
fields := strings.Fields(string(out))
if len(fields) == 0 {
return "", nil
}
return fields[0], nil
}
+664
View File
@@ -0,0 +1,664 @@
package sshpush
import (
"context"
"crypto/ed25519"
"crypto/rand"
"crypto/sha256"
"crypto/x509"
"encoding/pem"
"errors"
"fmt"
"net"
"os"
"path/filepath"
"strings"
"sync"
"sync/atomic"
"testing"
"time"
"golang.org/x/crypto/ssh"
"golang.org/x/crypto/ssh/knownhosts"
)
// --- fakeSSHServer: a minimal in-process SSH server for hermetic tests.
type fakeSSHServer struct {
listener net.Listener
config *ssh.ServerConfig
done chan struct{}
mu sync.Mutex
files map[string]string
hostKey ssh.Signer
cmdCount int64
execDelay time.Duration
}
func newFakeSSHServer(t *testing.T) *fakeSSHServer {
t.Helper()
_, priv, err := ed25519.GenerateKey(rand.Reader)
if err != nil {
t.Fatalf("ed25519 gen: %v", err)
}
signer, err := ssh.NewSignerFromKey(priv)
if err != nil {
t.Fatalf("ssh signer: %v", err)
}
config := &ssh.ServerConfig{
NoClientAuth: true,
}
config.AddHostKey(signer)
ln, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatalf("listen: %v", err)
}
srv := &fakeSSHServer{
listener: ln,
config: config,
done: make(chan struct{}),
files: make(map[string]string),
hostKey: signer,
}
go srv.serve()
return srv
}
func (s *fakeSSHServer) addr() string { return s.listener.Addr().String() }
func (s *fakeSSHServer) hostPublicKey() ssh.PublicKey { return s.hostKey.PublicKey() }
func (s *fakeSSHServer) close() {
_ = s.listener.Close()
<-s.done
}
func (s *fakeSSHServer) setExecDelay(d time.Duration) {
s.mu.Lock()
defer s.mu.Unlock()
s.execDelay = d
}
func (s *fakeSSHServer) serve() {
for {
conn, err := s.listener.Accept()
if err != nil {
close(s.done)
return
}
go s.handle(conn)
}
}
func (s *fakeSSHServer) handle(netConn net.Conn) {
defer netConn.Close()
_, chans, reqs, err := ssh.NewServerConn(netConn, s.config)
if err != nil {
return
}
go ssh.DiscardRequests(reqs)
for newChan := range chans {
if newChan.ChannelType() != "session" {
newChan.Reject(ssh.UnknownChannelType, "only session")
continue
}
go s.handleSession(newChan)
}
}
func (s *fakeSSHServer) handleSession(newChan ssh.NewChannel) {
ch, reqs, err := newChan.Accept()
if err != nil {
return
}
defer ch.Close()
for req := range reqs {
if req.Type != "exec" {
req.Reply(false, nil)
continue
}
var execReq struct{ Command string }
if err := ssh.Unmarshal(req.Payload, &execReq); err != nil {
req.Reply(false, nil)
continue
}
req.Reply(true, nil)
atomic.AddInt64(&s.cmdCount, 1)
s.mu.Lock()
delay := s.execDelay
s.mu.Unlock()
if delay > 0 {
time.Sleep(delay)
}
out, code := s.runCommand(execReq.Command)
_, _ = ch.Write(out)
_, _ = ch.SendRequest("exit-status", false, ssh.Marshal(struct{ Code uint32 }{uint32(code)}))
_ = ch.Close()
return
}
}
// runCommand implements the minimal command surface the transport uses:
// echo (for exec tests), sha256sum (for idempotency), cat (read), and the
// heredoc-based write (cat > tmp <<EOF ... EOF && chmod ... && mv ...).
func (s *fakeSSHServer) runCommand(cmd string) ([]byte, int) {
s.mu.Lock()
defer s.mu.Unlock()
trimmed := strings.TrimSpace(cmd)
switch {
case trimmed == "echo hello":
return []byte("hello\n"), 0
case strings.HasPrefix(trimmed, "sha256sum "):
// Format: sha256sum '/path' 2>/dev/null
rest := strings.TrimSpace(strings.TrimPrefix(trimmed, "sha256sum "))
rest = strings.TrimSuffix(rest, " 2>/dev/null")
rest = strings.TrimSpace(rest)
path := unquote(rest)
content, ok := s.files[path]
if !ok {
// `2>/dev/null` swallows the error; sha256sum exits 1 but
// stderr is suppressed. The transport treats empty output as
// "file missing" (no hash), so return ("", 0).
return []byte(""), 0
}
sum := sha256HexStr([]byte(content))
return []byte(sum + " " + path + "\n"), 0
case strings.HasPrefix(trimmed, "cat '"):
path := unquote(strings.TrimPrefix(trimmed, "cat "))
content, ok := s.files[path]
if !ok {
return []byte("cat: " + path + ": No such file or directory\n"), 1
}
return []byte(content), 0
case strings.HasPrefix(trimmed, "mkdir -p ") && strings.Contains(trimmed, "cat >"):
return s.handleWrite(trimmed)
default:
return []byte("sh: command not found\n"), 127
}
}
// handleWrite parses the heredoc write command produced by writeFile.
// Command format:
//
// mkdir -p '<dir>' && cat > '<tmp>' <<'ORCA_PUSH_EOF_a1b2c3'
// <content>
// ORCA_PUSH_EOF_a1b2c3
// chmod <mode> '<tmp>' && mv -f '<tmp>' '<path>'
func (s *fakeSSHServer) handleWrite(cmd string) ([]byte, int) {
const eof = "ORCA_PUSH_EOF_a1b2c3"
// Find the opening heredoc line: ... <<'EOF'\n
openerIdx := strings.Index(cmd, "<<'"+eof+"'")
if openerIdx < 0 {
return []byte("sh: no heredoc opener\n"), 1
}
// Body starts after the opener line's newline.
rest := cmd[openerIdx+len("<<'"+eof+"'"):]
nl := strings.Index(rest, "\n")
if nl < 0 {
return []byte("sh: no body start\n"), 1
}
body := rest[nl+1:]
// Body ends at the closing EOF marker on its own line.
closeIdx := strings.Index(body, "\n"+eof+"\n")
if closeIdx < 0 {
// Maybe EOF is at the end without trailing newline.
closeIdx = strings.Index(body, "\n"+eof)
if closeIdx < 0 {
return []byte("sh: no heredoc close\n"), 1
}
body = body[:closeIdx]
} else {
body = body[:closeIdx]
}
// Find the mv target: last quoted arg of "mv -f 'tmp' 'path'".
mvIdx := strings.LastIndex(cmd, "mv -f ")
if mvIdx < 0 {
return []byte("sh: no mv\n"), 1
}
tail := cmd[mvIdx+len("mv -f "):]
parts := splitQuoted(tail)
if len(parts) < 2 {
return []byte("sh: bad mv args\n"), 1
}
target := parts[1]
s.files[target] = body
return nil, 0
}
// splitQuoted splits a string of the form "'a' 'b'" into ["a","b"].
func splitQuoted(s string) []string {
var out []string
var cur strings.Builder
in := false
for _, r := range s {
if r == '\'' {
if in {
out = append(out, cur.String())
cur.Reset()
}
in = !in
continue
}
if in {
cur.WriteRune(r)
}
}
return out
}
func unquote(s string) string {
s = strings.TrimSpace(s)
if len(s) >= 2 && s[0] == '\'' && s[len(s)-1] == '\'' {
return s[1 : len(s)-1]
}
if len(s) >= 2 && s[0] == '"' && s[len(s)-1] == '"' {
return s[1 : len(s)-1]
}
return s
}
// sha256HexStr is a test-local copy of the sha256Hex helper.
func sha256HexStr(b []byte) string {
sum := sha256.Sum256(b)
return fmt.Sprintf("%x", sum[:])
}
// --- test helpers for transport setup ---
// setupORCAHome creates a temp ORCA_HOME with an empty known_hosts (at
// the v0.8 flat location $ORCA_HOME/known_hosts, which is where
// proxmox.TOFUHostKeyCallback reads via certpaths.KnownHostsPath()) and a
// generated Ed25519 SSH key, returns the key path.
func setupORCAHome(t *testing.T) (keyPath string) {
t.Helper()
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
// certpaths.KnownHostsPath() = paths.Root()/known_hosts = $ORCA_HOME/known_hosts.
knownHosts := filepath.Join(dir, "known_hosts")
if err := os.WriteFile(knownHosts, []byte{}, 0o600); err != nil {
t.Fatalf("create known_hosts: %v", err)
}
_, priv, err := ed25519.GenerateKey(rand.Reader)
if err != nil {
t.Fatalf("ed25519 gen: %v", err)
}
der, err := x509.MarshalPKCS8PrivateKey(priv)
if err != nil {
t.Fatalf("marshal key: %v", err)
}
pemBytes := pem.EncodeToMemory(&pem.Block{Type: "PRIVATE KEY", Bytes: der})
keyPath = filepath.Join(dir, "orca_ssh_key")
if err := os.WriteFile(keyPath, pemBytes, 0o600); err != nil {
t.Fatalf("write key: %v", err)
}
return keyPath
}
// realTransport returns a Transport wired to use the real SSH dialer
// against a fake SSH server, with the server's host key pre-populated in
// known_hosts (so the TOFU callback matches on first dial — no first-
// connect write race in tests).
func realTransport(t *testing.T, srv *fakeSSHServer) *Transport {
t.Helper()
keyPath := setupORCAHome(t)
tr := NewTransport(keyPath, "")
tr.SetUser("root")
addr := srv.addr()
line := knownhosts.Line([]string{knownhosts.Normalize(addr)}, srv.hostPublicKey())
home := os.Getenv("ORCA_HOME")
kh := filepath.Join(home, "known_hosts")
if err := os.WriteFile(kh, []byte(line+"\n"), 0o600); err != nil {
t.Fatalf("pre-pop known_hosts: %v", err)
}
return tr
}
// --- mock dialer + mock session for pure-logic tests (no real SSH) ---
type mockDialer struct {
client *ssh.Client
err error
calls int
}
func (m *mockDialer) DialContext(ctx context.Context, network, addr string, cfg *ssh.ClientConfig) (*ssh.Client, error) {
m.calls++
if m.err != nil {
return nil, m.err
}
return m.client, nil
}
type mockSession struct {
out []byte
err error
cmd string
}
func (m *mockSession) CombinedOutput(cmd string) ([]byte, error) {
m.cmd = cmd
return m.out, m.err
}
func (m *mockSession) Close() error { return nil }
// --- tests ---
func TestNewTransport_Defaults(t *testing.T) {
tr := NewTransport("/tmp/key", "/tmp/kh")
if tr.keyPath != "/tmp/key" {
t.Errorf("keyPath = %q", tr.keyPath)
}
if tr.user != "orca" {
t.Errorf("default user = %q, want orca", tr.user)
}
if tr.dialer == nil {
t.Error("dialer is nil")
}
}
func TestSetUser(t *testing.T) {
tr := NewTransport("/tmp/key", "/tmp/kh")
tr.SetUser("root")
if tr.user != "root" {
t.Errorf("user = %q, want root", tr.user)
}
tr.SetUser("")
if tr.user != "root" {
t.Errorf("user = %q, want root", tr.user)
}
}
func TestBackoff(t *testing.T) {
// Without jitter, attempt 1 -> 100ms, 2 -> 200ms, ... up to 5s cap.
// The jittered result is in [d/4, 3d/4) where d is the capped base,
// so for high attempts the result can reach 3*d/4 < 1.5*d. We bound
// the upper end at 2x the cap to allow jitter headroom.
for _, tc := range []struct {
attempt int
max time.Duration
}{
{1, 200 * time.Millisecond},
{2, 400 * time.Millisecond},
{6, 2 * RetryMax},
{10, 2 * RetryMax},
} {
got := backoff(RetryInitial, RetryMax, tc.attempt)
if got < 0 || got > tc.max {
t.Errorf("backoff(%d) = %s, want in [0, %s]", tc.attempt, got, tc.max)
}
}
}
func TestIsTransient(t *testing.T) {
if isTransient(nil) {
t.Error("nil should not be transient")
}
if !isTransient(ErrTransient) {
t.Error("ErrTransient should be transient")
}
if isTransient(ErrPermanent) {
t.Error("ErrPermanent should not be transient")
}
if !isTransient(errors.New("connection refused")) {
t.Error("connection refused should be transient")
}
if !isTransient(errors.New("i/o timeout")) {
t.Error("i/o timeout should be transient")
}
if isTransient(errors.New("some other error")) {
t.Error("unknown error should not be transient")
}
}
func TestClassifyDialErr(t *testing.T) {
if got := classifyDialErr(nil); got != nil {
t.Errorf("nil -> nil, got %v", got)
}
perm := classifyDialErr(errors.New("ssh: unable to authenticate"))
if !errors.Is(perm, ErrPermanent) {
t.Errorf("auth failure should be permanent, got %v", perm)
}
perm2 := classifyDialErr(errors.New("host key mismatch"))
if !errors.Is(perm2, ErrPermanent) {
t.Errorf("host key mismatch should be permanent, got %v", perm2)
}
trans := classifyDialErr(errors.New("connection refused"))
if !errors.Is(trans, ErrTransient) {
t.Errorf("connection refused should be transient, got %v", trans)
}
}
func TestClassifyExecErr(t *testing.T) {
exitErr := ssh.ExitError{}
perm := classifyExecErr(&exitErr)
if !errors.Is(perm, ErrPermanent) {
t.Errorf("ExitError should be permanent, got %v", perm)
}
trans := classifyExecErr(errors.New("session closed"))
if !errors.Is(trans, ErrTransient) {
t.Errorf("session closed should be transient, got %v", trans)
}
}
func TestShellQuote(t *testing.T) {
got := shellQuote("/etc/orca/foo.conf")
if got != "'/etc/orca/foo.conf'" {
t.Errorf("shellQuote = %q", got)
}
got = shellQuote("it's a path")
if got != "'it'\\''s a path'" {
t.Errorf("shellQuote with quote = %q", got)
}
}
func TestTransport_ExecSuccess_RealSSH(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
out, err := tr.Exec(context.Background(), srv.addr(), "echo hello")
if err != nil {
t.Fatalf("Exec: %v", err)
}
if strings.TrimSpace(string(out)) != "hello" {
t.Errorf("out = %q, want hello", out)
}
}
func TestTransport_ExecRetry_TransientFailure(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
var calls int32
tr.SetSessionFactory(func(c *ssh.Client) (sshSession, error) {
n := atomic.AddInt32(&calls, 1)
if n < 3 {
return &mockSession{err: errors.New("EOF")}, nil
}
return &mockSession{out: []byte("ok\n")}, nil
})
client, err := tr.dial(srv.addr())
if err != nil {
t.Fatalf("dial: %v", err)
}
tr.pool.Store(srv.addr(), client)
out, err := tr.Exec(context.Background(), srv.addr(), "echo ok")
if err != nil {
t.Fatalf("Exec: %v (calls=%d)", err, atomic.LoadInt32(&calls))
}
if string(out) != "ok\n" {
t.Errorf("out = %q, want ok\\n", out)
}
if got := atomic.LoadInt32(&calls); got < 3 {
t.Errorf("calls = %d, want >= 3 (retried)", got)
}
}
func TestTransport_ExecPermanentError_NoRetry(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
var calls int32
tr.SetSessionFactory(func(c *ssh.Client) (sshSession, error) {
atomic.AddInt32(&calls, 1)
return &mockSession{err: &ssh.ExitError{}}, nil
})
client, err := tr.dial(srv.addr())
if err != nil {
t.Fatalf("dial: %v", err)
}
tr.pool.Store(srv.addr(), client)
_, err = tr.Exec(context.Background(), srv.addr(), "exit 1")
if err == nil {
t.Fatal("expected error, got nil")
}
if !errors.Is(err, ErrPermanent) {
t.Errorf("err should be permanent, got %v", err)
}
if got := atomic.LoadInt32(&calls); got != 1 {
t.Errorf("calls = %d, want 1 (no retry on permanent)", got)
}
}
func TestTransport_ExecTimeout(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
srv.setExecDelay(500 * time.Millisecond)
tr := realTransport(t, srv)
defer tr.Close()
ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
defer cancel()
_, err := tr.Exec(ctx, srv.addr(), "echo hello")
if err == nil {
t.Fatal("expected timeout error, got nil")
}
if !errors.Is(err, context.DeadlineExceeded) && !errors.Is(err, ErrTransient) {
t.Errorf("err should be timeout/transient, got %v", err)
}
}
func TestTransport_ConnectionPoolReuse(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
defer tr.Close()
addr := srv.addr()
c1, err := tr.dial(addr)
if err != nil {
t.Fatalf("first dial: %v", err)
}
c2, err := tr.dial(addr)
if err != nil {
t.Fatalf("second dial: %v", err)
}
if c1 != c2 {
t.Error("pool did not reuse client for same peer")
}
}
func TestTransport_CloseClosesAllClients(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
tr := realTransport(t, srv)
if _, err := tr.dial(srv.addr()); err != nil {
t.Fatalf("dial: %v", err)
}
count := 0
tr.pool.Range(func(_, _ any) bool {
count++
return true
})
if count != 1 {
t.Fatalf("pool has %d entries, want 1", count)
}
if err := tr.Close(); err != nil {
t.Errorf("Close: %v", err)
}
_, err := tr.dial(srv.addr())
if !errors.Is(err, ErrPermanent) {
t.Errorf("dial after Close should be ErrPermanent, got %v", err)
}
if err := tr.Close(); err != nil {
t.Errorf("second Close: %v", err)
}
}
func TestTransport_ReadFile_RealSSH(t *testing.T) {
srv := newFakeSSHServer(t)
defer srv.close()
srv.mu.Lock()
srv.files["/etc/orca/test.conf"] = "content-line\n"
srv.mu.Unlock()
tr := realTransport(t, srv)
defer tr.Close()
out, err := tr.ReadFile(context.Background(), srv.addr(), "/etc/orca/test.conf")
if err != nil {
t.Fatalf("ReadFile: %v", err)
}
if string(out) != "content-line\n" {
t.Errorf("out = %q", out)
}
}
func TestTransport_DialKeyParseFailure(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
if err := os.WriteFile(filepath.Join(dir, "known_hosts"), []byte{}, 0o600); err != nil {
t.Fatalf("kh: %v", err)
}
keyPath := filepath.Join(dir, "bad_key")
if err := os.WriteFile(keyPath, []byte("not a key"), 0o600); err != nil {
t.Fatalf("write key: %v", err)
}
tr := NewTransport(keyPath, "")
tr.SetUser("root")
_, err := tr.dial("127.0.0.1:1")
if err == nil {
t.Fatal("expected parse error, got nil")
}
if !strings.Contains(err.Error(), "parse key") {
t.Errorf("err should mention parse key, got %v", err)
}
}
func TestTransport_DialKeyMissing(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
if err := os.WriteFile(filepath.Join(dir, "known_hosts"), []byte{}, 0o600); err != nil {
t.Fatalf("kh: %v", err)
}
tr := NewTransport(filepath.Join(dir, "missing_key"), "")
tr.SetUser("root")
_, err := tr.dial("127.0.0.1:1")
if err == nil {
t.Fatal("expected read error, got nil")
}
if !strings.Contains(err.Error(), "read key") {
t.Errorf("err should mention read key, got %v", err)
}
}
func TestTransport_DialMockFailure(t *testing.T) {
dir := t.TempDir()
t.Setenv("ORCA_HOME", dir)
if err := os.WriteFile(filepath.Join(dir, "known_hosts"), []byte{}, 0o600); err != nil {
t.Fatalf("kh: %v", err)
}
_, priv, _ := ed25519.GenerateKey(rand.Reader)
der, _ := x509.MarshalPKCS8PrivateKey(priv)
pemBytes := pem.EncodeToMemory(&pem.Block{Type: "PRIVATE KEY", Bytes: der})
keyPath := filepath.Join(dir, "orca_ssh_key")
_ = os.WriteFile(keyPath, pemBytes, 0o600)
tr := NewTransport(keyPath, "")
tr.SetUser("root")
tr.SetDialer(&mockDialer{err: errors.New("connection refused")})
_, err := tr.dial("127.0.0.1:1")
if err == nil {
t.Fatal("expected dial error")
}
if !errors.Is(err, ErrTransient) {
t.Errorf("connection refused should be transient, got %v", err)
}
}
+6
View File
@@ -6,6 +6,12 @@
// gRPC, no ConnectRPC, no third-party transport libraries. This keeps
// the binary lean (matches the minimalist pillar) and the trust chain
// auditable (one library: the Go stdlib).
//
// Deprecated: v0.9 re-architecture replaces this with
// internal/sshpush (REQ-073). The daemon-to-daemon mTLS transport is
// removed because servers no longer run the orca binary (R-001); the
// CLI pushes config via SSH instead. Scheduled for deletion in
// v0.10-P14. See .ciagent/PRD_v0.9.md R-001/R-006.
package transport
import (
+32
View File
@@ -0,0 +1,32 @@
# orca-log.sh — structured slog-compatible JSON logging for bash scripts (C-17).
# Source this library from any orca bash script: `source scripts/lib/orca-log.sh`.
# Emits JSON to syslog via `logger`; falls back to stderr if `logger` is missing.
# Field set matches the Go audit log (REQ-006): ts, level, actor, action, resource, result, error.
ORCA_LOG_ACTOR="${ORCA_LOG_ACTOR:-spiffe://orca/cli/operator}"
# _orca_log_emit <level> <action> <resource> <result> [error]
_orca_log_emit() {
local level="$1" action="$2" resource="$3" result="$4" error="${5:-}"
local ts
ts="$(date -u +%Y-%m-%dT%H:%M:%S.%3NZ)"
# Build JSON with proper escaping of error field (escape backslash and quote).
local err_json=""
if [ -n "$error" ]; then
local esc_error
esc_error="${error//\\/\\\\}"
esc_error="${esc_error//\"/\\\"}"
err_json=",\"error\":\"$esc_error\""
fi
local line
line="{\"ts\":\"$ts\",\"level\":\"$level\",\"actor\":\"$ORCA_LOG_ACTOR\",\"action\":\"$action\",\"resource\":\"$resource\",\"result\":\"$result\"$err_json}"
if command -v logger >/dev/null 2>&1; then
logger -t orca "$line"
else
echo "$line" >&2
fi
}
orca_log_info() { _orca_log_emit "info" "$1" "$2" "$3" "${4:-}"; }
orca_log_warn() { _orca_log_emit "warn" "$1" "$2" "$3" "${4:-}"; }
orca_log_error() { _orca_log_emit "error" "$1" "$2" "$3" "${4:-}"; }
+76
View File
@@ -0,0 +1,76 @@
#!/usr/bin/env bash
# orca-verify-render.sh — bash-side render-contract validator (grill C-16).
# Reads a render-bundle JSON file (one Artifact per line, or a JSON array)
# and validates each entry against the orca.emit/v1 schema.
# Exit 0 if all valid; non-zero with a structured error per failure to stderr.
# Source: scripts/lib/orca-log.sh for structured error logging (C-17).
#
# Usage: orca-verify-render.sh <bundle.json>
set -euo pipefail
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
# shellcheck source=lib/orca-log.sh
. "$SCRIPT_DIR/lib/orca-log.sh"
EXPECTED_SCHEMA="orca.emit/v1"
if [ "$#" -lt 1 ]; then
orca_log_error "verify-render" "-" "failed" "missing bundle argument"
echo "usage: $0 <bundle.json>" >&2
exit 2
fi
bundle="$1"
if [ ! -f "$bundle" ]; then
orca_log_error "verify-render" "$bundle" "failed" "bundle file not found"
echo "error: bundle not found: $bundle" >&2
exit 2
fi
errors=0
total=0
# Read the bundle line-by-line. Each line should be a JSON object.
# (The Go emitter writes one Artifact per line for line-delimited parsing.)
while IFS= read -r line; do
# Skip blank lines and comments.
[ -z "$line" ] && continue
case "$line" in \#*) continue ;; esac
total=$((total + 1))
# Validate schema_version field presence and value (crude JSON grep; no jq dep).
# Check schema_version via a simple substring test.
schema_match=0
if printf '%s' "$line" | grep -q "\"schema_version\":\"$EXPECTED_SCHEMA\""; then
schema_match=1
fi
if [ "$schema_match" -eq 1 ]; then
# schema_version matches. Check kind, path, mode presence.
for field in kind path mode; do
if ! printf '%s' "$line" | grep -q "\"$field\":"; then
orca_log_error "verify-render" "$bundle" "failed" "missing field: $field"
echo "error: line $total missing field: $field" >&2
errors=$((errors + 1))
continue 2
fi
done
elif printf '%s' "$line" | grep -q '"schema_version":'; then
orca_log_error "verify-render" "$bundle" "failed" "schema_version mismatch on line $total"
echo "error: line $total schema_version mismatch (expected $EXPECTED_SCHEMA)" >&2
errors=$((errors + 1))
else
orca_log_error "verify-render" "$bundle" "failed" "missing schema_version on line $total"
echo "error: line $total missing schema_version" >&2
errors=$((errors + 1))
fi
done < "$bundle"
if [ "$errors" -gt 0 ]; then
orca_log_error "verify-render" "$bundle" "failed" "$errors of $total artifacts invalid"
echo "verify-render: $errors of $total artifacts invalid" >&2
exit 1
fi
orca_log_info "verify-render" "$bundle" "ok" ""
echo "verify-render: $total artifacts valid"
exit 0
+39
View File
@@ -0,0 +1,39 @@
# Bash Testing Policy (grill C-15)
Every bash script under `scripts/` MUST have at least one bats test covering
the happy path and one covering the failure path. This is the compensating
control for bash being exempt from the Go coverage gate (D-186).
## Framework
- **bats**`bats scripts/tests/*.bash` runs all bash tests.
- **shellcheck**`shellcheck scripts/*.sh scripts/lib/*.sh scripts/tests/*.bash` static analysis.
- **shfmt**`shfmt -d scripts/` formatting check (optional; skip if not installed).
## Install (if missing)
```bash
# bats
npm install -g bats # or: git clone https://github.com/bats-core/bats-core.git && ./bats-core/install.sh /usr/local
# shellcheck
apt-get install -y shellcheck
# shfmt (optional)
mvdan.cc/sh (go install mvdan.cc/sh/v3/cmd/shfmt@latest)
```
## Running
```bash
make test-bash # runs bats (skips gracefully if bats missing)
make lint-bash # runs shellcheck + shfmt (skips gracefully if missing)
make test # runs both Go + bash tests
make lint # runs both Go + bash lint
```
## Test file convention
- Test files live in `scripts/tests/<script-name>_test.bash`.
- Source `load test_helper` at the top of every test file.
- Happy path: `@test "<script> happy path" { ... }`
- Failure path: `@test "<script> failure path" { ... }`
- Use `run <command>` + `assert_status`/`assert_contains`/`assert_not_contains` from test_helper.
+43
View File
@@ -0,0 +1,43 @@
#!/usr/bin/env bats
# Example bats test proving the framework works (C-15 smoke test).
# Real script tests live alongside each script under scripts/tests/.
load test_helper
@test "test_helper assert_status accepts matching status" {
assert_status 0 0
assert_status 1 1
}
@test "test_helper assert_status rejects mismatch" {
run assert_status 0 1
[ "$status" -ne 0 ]
}
@test "test_helper assert_contains finds substrings" {
assert_contains "hello world" "world"
}
@test "test_helper assert_contains rejects missing substrings" {
run assert_contains "hello world" "missing"
[ "$status" -ne 0 ]
}
@test "test_helper assert_not_contains passes when substring absent" {
assert_not_contains "hello world" "missing"
}
@test "test_helper assert_not_contains fails when substring present" {
run assert_not_contains "hello world" "world"
[ "$status" -ne 0 ]
}
@test "test_helper assert_json_field detects JSON fields" {
assert_json_field '{"ts":"2026-01-01T00:00:00Z","level":"info"}' "ts"
assert_json_field '{"ts":"2026-01-01T00:00:00Z","level":"info"}' "level"
}
@test "test_helper SCRIPTS_DIR resolves to scripts/ directory" {
[ -d "$SCRIPTS_DIR" ]
[ -f "$SCRIPTS_DIR/install.sh" ]
}
+69
View File
@@ -0,0 +1,69 @@
#!/usr/bin/env bats
# Tests for scripts/lib/orca-log.sh (C-17 — slog-compatible JSON to syslog).
# Verifies the JSON structure is valid, field set is present, level maps correctly,
# and the logger-fallback-to-stderr path works in test environments (no logger).
load test_helper
@test "orca_log_info emits valid JSON with all required fields" {
export ORCA_LOG_ACTOR="test-actor"
output="$(_orca_log_for_test info test-action test-resource ok "")"
assert_json_field "$output" "ts"
assert_json_field "$output" "level"
assert_json_field "$output" "actor"
assert_json_field "$output" "action"
assert_json_field "$output" "resource"
assert_json_field "$output" "result"
assert_contains "$output" '"level":"info"'
assert_contains "$output" '"action":"test-action"'
assert_contains "$output" '"resource":"test-resource"'
assert_contains "$output" '"result":"ok"'
}
@test "orca_log_warn maps level correctly" {
output="$(_orca_log_for_test warn w-action w-resource warn-result)"
assert_contains "$output" '"level":"warn"'
}
@test "orca_log_error maps level and includes error field when provided" {
output="$(_orca_log_for_test error err-action err-resource failed "something broke")"
assert_contains "$output" '"level":"error"'
assert_contains "$output" '"result":"failed"'
assert_contains "$output" '"error":"something broke"'
}
@test "orca_log_error omits error field when not provided" {
output="$(_orca_log_for_test error err-action err-resource failed)"
assert_contains "$output" '"level":"error"'
assert_not_contains "$output" '"error":'
}
@test "orca_log escapes quotes and backslashes in error field" {
output="$(_orca_log_for_test error a r failed 'has "quote" and \backslash')"
assert_contains "$output" '\"quote\"'
assert_contains "$output" '\\backslash'
}
@test "ORCA_LOG_ACTOR env var overrides the actor field" {
export ORCA_LOG_ACTOR="custom-actor-123"
output="$(_orca_log_for_test info a r ok)"
assert_contains "$output" '"actor":"custom-actor-123"'
}
# Test helper: source orca-log.sh and emit to stderr (force fallback by hiding logger).
_orca_log_for_test() {
local level="$1" action="$2" resource="$3" result="$4" error="${5:-}"
# Source the library in a subshell with logger hidden so it falls back to stderr.
(
PATH="/usr/bin:/bin" # hide logger if it's in /usr/local/bin etc.
# shellcheck disable=SC2317 # logger is overridden below for test capture
logger() { echo "$3"; } # $3 is the message arg (logger -t orca "$line")
# shellcheck disable=SC1091 # path is set at runtime by SCRIPTS_DIR
source "$SCRIPTS_DIR/lib/orca-log.sh"
case "$level" in
info) orca_log_info "$action" "$resource" "$result" "$error" ;;
warn) orca_log_warn "$action" "$resource" "$result" "$error" ;;
error) orca_log_error "$action" "$resource" "$result" "$error" ;;
esac
)
}
@@ -0,0 +1,69 @@
#!/usr/bin/env bats
# Tests for scripts/orca-verify-render.sh (C-16 render-format contract validator).
# Covers happy path (valid input returns 0) and failure paths (schema mismatch,
# missing fields, missing file, missing argument).
load test_helper
VERIFY_RENDER="$SCRIPTS_DIR/orca-verify-render.sh"
TMP_BUNDLE=""
setup() {
TMP_BUNDLE="$(mktemp)"
}
teardown() {
[ -n "$TMP_BUNDLE" ] && rm -f "$TMP_BUNDLE"
}
@test "verify-render happy path: valid artifacts return 0" {
cat >"$TMP_BUNDLE" <<'EOF'
{"schema_version":"orca.emit/v1","kind":"systemd","path":"/etc/systemd/system/x.service","content":"[Service]","mode":"0644"}
{"schema_version":"orca.emit/v1","kind":"traefik","path":"/etc/traefik/dynamic/orca.yml","content":"tls:{}","mode":"0644"}
EOF
run "$VERIFY_RENDER" "$TMP_BUNDLE"
assert_status 0 "$status"
assert_contains "$output" "2 artifacts valid"
}
@test "verify-render failure: schema_version mismatch returns non-zero" {
cat >"$TMP_BUNDLE" <<'EOF'
{"schema_version":"orca.emit/v2","kind":"systemd","path":"/x","mode":"0644"}
EOF
run "$VERIFY_RENDER" "$TMP_BUNDLE"
[ "$status" -ne 0 ]
assert_contains "$output" "schema_version mismatch"
}
@test "verify-render failure: missing schema_version returns non-zero" {
cat >"$TMP_BUNDLE" <<'EOF'
{"kind":"systemd","path":"/x","mode":"0644"}
EOF
run "$VERIFY_RENDER" "$TMP_BUNDLE"
[ "$status" -ne 0 ]
assert_contains "$output" "missing schema_version"
}
@test "verify-render failure: missing bundle argument returns 2" {
run "$VERIFY_RENDER"
assert_status 2 "$status"
assert_contains "$output" "usage:"
}
@test "verify-render failure: non-existent bundle returns 2" {
run "$VERIFY_RENDER" "/nonexistent/bundle.json"
assert_status 2 "$status"
assert_contains "$output" "bundle not found"
}
@test "verify-render skips blank lines and comments" {
cat >"$TMP_BUNDLE" <<'EOF'
# this is a comment
{"schema_version":"orca.emit/v1","kind":"systemd","path":"/x","content":"c","mode":"0644"}
EOF
run "$VERIFY_RENDER" "$TMP_BUNDLE"
assert_status 0 "$status"
assert_contains "$output" "1 artifacts valid"
}
+42
View File
@@ -0,0 +1,42 @@
#!/usr/bin/env bash
# Common helpers for orca bats tests (C-15). Sourced by every test file.
# See scripts/tests/README.md for the bash testing policy.
# Resolve the scripts/ dir relative to this test file.
# BASH_SOURCE[0] is this helper file (scripts/tests/test_helper.bash).
SCRIPTS_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
export SCRIPTS_DIR
# assert_status <expected> <actual> — assert a command's exit status.
assert_status() {
local expected="$1" actual="$2"
[ "$expected" = "$actual" ] || {
echo "expected status $expected, got $actual" >&2
return 1
}
}
# assert_contains <haystack> <needle> — substring assertion.
assert_contains() {
local haystack="$1" needle="$2"
case "$haystack" in
*"$needle"*) return 0 ;;
*) echo "expected [$haystack] to contain [$needle]" >&2; return 1 ;;
esac
}
# assert_not_contains <haystack> <needle> — negative substring assertion.
assert_not_contains() {
local haystack="$1" needle="$2"
case "$haystack" in
*"$needle"*) echo "expected [$haystack] to NOT contain [$needle]" >&2; return 1 ;;
esac
:
}
# assert_json_field <json> <field> — crude JSON field presence check (no jq dep).
# Matches "<field>": present anywhere in the JSON string.
assert_json_field() {
local json="$1" field="$2"
assert_contains "$json" "\"$field\""
}