ship: v0.1 Foundation milestone complete (#1)

This commit was merged in pull request #1.
This commit is contained in:
2026-06-03 20:08:57 +00:00
parent 55aae5347e
commit be9afa2d2c
59 changed files with 4152 additions and 36 deletions
+60
View File
@@ -0,0 +1,60 @@
package cli
import (
"context"
"fmt"
"time"
"github.com/spf13/cobra"
"git.cloudinit.dev/coreci/orca/internal/store"
)
var (
auditLimit int
)
var auditCmd = &cobra.Command{
Use: "audit",
Short: "View orca audit log",
Long: "Display the most recent audit log entries (security-first observability).",
}
var auditListCmd = &cobra.Command{
Use: "list",
Short: "List recent audit log entries",
RunE: func(cmd *cobra.Command, args []string) error {
ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
defer cancel()
db, closer, err := openDB()
if err != nil {
return err
}
defer closer()
entries, err := store.NewAuditRepo(db).List(ctx, auditLimit)
if err != nil {
return err
}
if jsonOutput {
return printJSON(entries)
}
if len(entries) == 0 {
fmt.Fprintln(cmd.OutOrStdout(), "No audit entries.")
return nil
}
fmt.Fprintf(cmd.OutOrStdout(), "%-22s %-12s %-20s %-30s %-10s\n", "TIMESTAMP", "ACTOR", "ACTION", "RESOURCE", "RESULT")
for _, e := range entries {
fmt.Fprintf(cmd.OutOrStdout(), "%-22s %-12s %-20s %-30s %-10s\n",
e.Timestamp.Format("2006-01-02T15:04:05Z"), e.Actor, e.Action, e.Resource, e.Result)
}
return nil
},
}
func init() {
auditListCmd.Flags().IntVar(&auditLimit, "limit", 50, "max entries to show")
auditCmd.AddCommand(auditListCmd)
rootCmd.AddCommand(auditCmd)
}
+76
View File
@@ -0,0 +1,76 @@
package cli
import (
"context"
"errors"
"fmt"
"net/http"
"os"
"os/signal"
"syscall"
"time"
"github.com/spf13/cobra"
"git.cloudinit.dev/coreci/orca/internal/daemon"
)
var (
daemonAddr string
)
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 and API requests.",
RunE: func(cmd *cobra.Command, args []string) error {
db, closer, err := openDB()
if err != nil {
return err
}
defer closer()
srv := daemon.NewServer(daemon.Options{
DB: db,
Log: newLogger(),
Addr: daemonAddr,
Actor: "daemon",
})
srv.MarkReady()
errCh := make(chan error, 1)
go func() {
err := srv.Start()
if err != nil && !errors.Is(err, http.ErrServerClosed) {
errCh <- err
}
}()
fmt.Fprintf(cmd.OutOrStdout(), "✓ orca daemon listening on %s\n", daemonAddr)
fmt.Fprintln(cmd.OutOrStdout(), " /healthz - liveness")
fmt.Fprintln(cmd.OutOrStdout(), " /readyz - readiness (db + ready flag)")
fmt.Fprintln(cmd.OutOrStdout(), " /v1/status - status JSON")
fmt.Fprintln(cmd.OutOrStdout(), " /v1/jobs - list jobs")
fmt.Fprintln(cmd.OutOrStdout(), " /v1/nodes - list nodes")
fmt.Fprintln(cmd.OutOrStdout(), " /v1/tasks - list tasks")
fmt.Fprintln(cmd.OutOrStdout(), " press Ctrl+C to stop")
ctx, stop := signal.NotifyContext(cmd.Context(), os.Interrupt, syscall.SIGTERM)
defer stop()
select {
case <-ctx.Done():
fmt.Fprintln(cmd.OutOrStdout(), "\nshutting down...")
shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
return srv.Shutdown(shutdownCtx)
case err := <-errCh:
return err
}
},
}
func init() {
daemonCmd.Flags().StringVar(&daemonAddr, "addr", ":8080", "listen address")
rootCmd.AddCommand(daemonCmd)
}
+38
View File
@@ -0,0 +1,38 @@
package cli
import (
"fmt"
"os"
"path/filepath"
"github.com/spf13/cobra"
)
var initCmd = &cobra.Command{
Use: "init",
Short: "Initialize local orca state directory",
Long: "Create the local orca state directory at ~/.orca/ and write a default config file.",
RunE: func(cmd *cobra.Command, args []string) error {
home, err := os.UserHomeDir()
if err != nil {
return fmt.Errorf("get home dir: %w", err)
}
orcaDir := filepath.Join(home, ".orca")
if err := os.MkdirAll(orcaDir, 0o755); err != nil {
return fmt.Errorf("create orca dir: %w", err)
}
result := map[string]string{
"path": orcaDir,
"status": "initialized",
}
if jsonOutput {
return printJSON(result)
}
printText("✓ Initialized orca state at %s\n", orcaDir)
return nil
},
}
func init() {
rootCmd.AddCommand(initCmd)
}
+223
View File
@@ -0,0 +1,223 @@
package cli
import (
"context"
"errors"
"fmt"
"time"
"github.com/google/uuid"
"github.com/spf13/cobra"
"git.cloudinit.dev/coreci/orca/internal/engine"
"git.cloudinit.dev/coreci/orca/internal/jobspec"
"git.cloudinit.dev/coreci/orca/internal/model"
"git.cloudinit.dev/coreci/orca/internal/store"
)
var jobCmd = &cobra.Command{
Use: "job",
Short: "Manage orca jobs",
Long: "Run, list, stop, and inspect orca jobs.",
}
func jobExecutor() (*engine.Executor, func() error, error) {
db, closer, err := openDB()
if err != nil {
return nil, nil, err
}
jobs := store.NewJobRepo(db)
tasks := store.NewTaskRepo(db)
return engine.NewExecutor(jobs, tasks, newLogger()), closer, nil
}
var jobRunCmd = &cobra.Command{
Use: "run <spec.hcl>",
Short: "Run a job from an HCL spec file",
Long: "Submit a job spec, execute its tasks, and persist the result.",
Args: cobra.ExactArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
spec, err := jobspec.ParseFile(args[0])
if err != nil {
return err
}
ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Minute)
defer cancel()
exec, closer, err := jobExecutor()
if err != nil {
return err
}
defer closer()
job := &model.Job{
ID: uuid.NewString(),
Name: spec.Job.Name,
Spec: args[0],
Status: model.JobStatusPending,
}
if err := exec.Run(ctx, job, toTaskSpecs(spec.Tasks)); err != nil {
if jsonOutput {
_ = printJSON(map[string]any{"id": job.ID, "status": "failed", "error": err.Error()})
return err
}
fmt.Fprintf(cmd.ErrOrStderr(), "✗ Job %s failed: %v\n", job.ID, err)
return err
}
if jsonOutput {
return printJSON(map[string]any{"id": job.ID, "name": job.Name, "status": "complete"})
}
fmt.Fprintf(cmd.OutOrStdout(), "✓ Job complete: %s (%s)\n", job.ID, job.Name)
return nil
},
}
var jobListCmd = &cobra.Command{
Use: "list",
Short: "List all jobs",
Long: "Display all jobs and their status.",
RunE: func(cmd *cobra.Command, args []string) error {
ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
defer cancel()
db, closer, err := openDB()
if err != nil {
return err
}
defer closer()
jobs, err := store.NewJobRepo(db).List(ctx)
if err != nil {
return err
}
if jsonOutput {
return printJSON(jobs)
}
if len(jobs) == 0 {
fmt.Fprintln(cmd.OutOrStdout(), "No jobs. Use 'orca job run <spec.hcl>' to submit one.")
return nil
}
fmt.Fprintf(cmd.OutOrStdout(), "%-36s %-20s %-12s %-8s\n", "ID", "NAME", "STATUS", "EXIT")
for _, j := range jobs {
fmt.Fprintf(cmd.OutOrStdout(), "%-36s %-20s %-12s %-8d\n", j.ID, j.Name, j.Status, j.ExitCode)
}
return nil
},
}
var (
stopID string
)
var jobStopCmd = &cobra.Command{
Use: "stop [job-id]",
Short: "Stop a running job",
Long: "Mark a job as stopped. Note: this is a soft stop (cancel context for the daemon).",
Args: cobra.MaximumNArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
id := stopID
if id == "" && len(args) > 0 {
id = args[0]
}
if id == "" {
return fmt.Errorf("job id required (--id or argument)")
}
ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
defer cancel()
db, closer, err := openDB()
if err != nil {
return err
}
defer closer()
repo := store.NewJobRepo(db)
job, err := repo.Get(ctx, id)
if err != nil {
if errors.Is(err, store.ErrNotFound) {
return fmt.Errorf("job not found: %s", id)
}
return err
}
if err := repo.UpdateStatus(ctx, id, model.JobStatusStopped, 130); err != nil {
return err
}
if jsonOutput {
return printJSON(map[string]any{"id": id, "status": "stopped", "previous_status": job.Status})
}
fmt.Fprintf(cmd.OutOrStdout(), "✓ Job stopped: %s\n", id)
return nil
},
}
var jobLogsCmd = &cobra.Command{
Use: "logs [job-id]",
Short: "Show task output for a job",
Long: "Display captured stdout/stderr for all tasks in a job.",
Args: cobra.MaximumNArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
id := stopID
if id == "" && len(args) > 0 {
id = args[0]
}
if id == "" {
return fmt.Errorf("job id required (--id or argument)")
}
ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
defer cancel()
db, closer, err := openDB()
if err != nil {
return err
}
defer closer()
taskRepo := store.NewTaskRepo(db)
tasks, err := taskRepo.ListByJob(ctx, id)
if err != nil {
return err
}
if jsonOutput {
return printJSON(tasks)
}
if len(tasks) == 0 {
fmt.Fprintln(cmd.OutOrStdout(), "No tasks for this job.")
return nil
}
for i, t := range tasks {
fmt.Fprintf(cmd.OutOrStdout(), "--- task[%d] %s (%s) exit=%d ---\n", i, t.Command, t.Status, t.ExitCode)
if t.Stdout != "" {
fmt.Fprintln(cmd.OutOrStdout(), t.Stdout)
}
if t.Stderr != "" {
fmt.Fprintln(cmd.OutOrStderr(), t.Stderr)
}
}
return nil
},
}
func init() {
jobStopCmd.Flags().StringVar(&stopID, "id", "", "job id")
jobLogsCmd.Flags().StringVar(&stopID, "id", "", "job id")
jobCmd.AddCommand(jobRunCmd)
jobCmd.AddCommand(jobListCmd)
jobCmd.AddCommand(jobStopCmd)
jobCmd.AddCommand(jobLogsCmd)
rootCmd.AddCommand(jobCmd)
}
func toTaskSpecs(in []jobspec.TaskSpec) []engine.TaskSpec {
out := make([]engine.TaskSpec, len(in))
for i, t := range in {
out[i] = engine.TaskSpec{
Name: t.Name,
Command: t.Command,
Args: t.Args,
Env: t.Env,
}
}
return out
}
+176
View File
@@ -0,0 +1,176 @@
package cli
import (
"context"
"database/sql"
"fmt"
"log/slog"
"os"
"path/filepath"
"time"
"github.com/google/uuid"
"github.com/spf13/cobra"
"git.cloudinit.dev/coreci/orca/internal/engine"
"git.cloudinit.dev/coreci/orca/internal/model"
"git.cloudinit.dev/coreci/orca/internal/store"
)
func dbPath() string {
if p := os.Getenv("ORCA_DB"); p != "" {
return p
}
home, _ := os.UserHomeDir()
return filepath.Join(home, ".orca", "orca.db")
}
func openDB() (*sql.DB, func() error, error) {
db, err := store.Open(dbPath())
if err != nil {
return nil, nil, err
}
return db, db.Close, nil
}
func newLogger() *slog.Logger {
return slog.New(slog.NewJSONHandler(os.Stderr, &slog.HandlerOptions{Level: slog.LevelInfo}))
}
func nodeRegistry() (*engine.NodeRegistry, func() error, error) {
db, closer, err := openDB()
if err != nil {
return nil, nil, err
}
repo := store.NewNodeRepo(db)
audit := engine.NewAudit(store.NewAuditRepo(db), newLogger())
return engine.NewNodeRegistry(repo, audit, newLogger()), closer, nil
}
var (
joinName string
joinAddr string
leaveID string
)
var nodeCmd = &cobra.Command{
Use: "node",
Short: "Manage orca nodes",
Long: "Join, leave, or list orca nodes in the registry.",
}
var nodeJoinCmd = &cobra.Command{
Use: "join",
Short: "Join a node to the orca registry",
Long: "Register a node in the local orca registry. Persisted to SQLite.",
RunE: func(cmd *cobra.Command, args []string) error {
if joinName == "" {
return fmt.Errorf("--name is required")
}
if joinAddr == "" {
joinAddr = "localhost:8443"
}
ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
defer cancel()
registry, closer, err := nodeRegistry()
if err != nil {
return err
}
defer closer()
node := &model.Node{
ID: uuid.NewString(),
Name: joinName,
Address: joinAddr,
State: model.NodeStateReady,
JoinedAt: time.Now().UTC(),
LastSeen: time.Now().UTC(),
}
if err := registry.Join(ctx, node); err != nil {
return err
}
if jsonOutput {
return printJSON(node)
}
fmt.Fprintf(cmd.OutOrStdout(), "✓ Node joined: %s (%s) at %s\n", node.ID, node.Name, node.Address)
return nil
},
}
var nodeLeaveCmd = &cobra.Command{
Use: "leave [node-id]",
Short: "Remove a node from the orca registry",
Long: "Mark a node as left. Use --id to specify, or pass as argument.",
Args: cobra.MaximumNArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
id := leaveID
if id == "" && len(args) > 0 {
id = args[0]
}
if id == "" {
return fmt.Errorf("node id required (use --id or pass as argument)")
}
ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
defer cancel()
registry, closer, err := nodeRegistry()
if err != nil {
return err
}
defer closer()
if err := registry.Leave(ctx, id); err != nil {
return err
}
if jsonOutput {
return printJSON(map[string]string{"id": id, "state": "left"})
}
fmt.Fprintf(cmd.OutOrStdout(), "✓ Node left: %s\n", id)
return nil
},
}
var nodeListCmd = &cobra.Command{
Use: "list",
Short: "List all nodes in the orca registry",
Long: "Display all registered nodes and their state.",
RunE: func(cmd *cobra.Command, args []string) error {
ctx, cancel := context.WithTimeout(cmd.Context(), 5*time.Second)
defer cancel()
registry, closer, err := nodeRegistry()
if err != nil {
return err
}
defer closer()
nodes, err := registry.List(ctx)
if err != nil {
return err
}
if jsonOutput {
return printJSON(nodes)
}
if len(nodes) == 0 {
fmt.Fprintln(cmd.OutOrStdout(), "No nodes registered. Use 'orca node join' to add one.")
return nil
}
fmt.Fprintf(cmd.OutOrStdout(), "%-36s %-20s %-22s %-10s\n", "ID", "NAME", "ADDRESS", "STATE")
for _, n := range nodes {
fmt.Fprintf(cmd.OutOrStdout(), "%-36s %-20s %-22s %-10s\n", n.ID, n.Name, n.Address, n.State)
}
return nil
},
}
func init() {
nodeJoinCmd.Flags().StringVar(&joinName, "name", "", "node name (required)")
nodeJoinCmd.Flags().StringVar(&joinAddr, "addr", "", "node address (default localhost:8443)")
nodeLeaveCmd.Flags().StringVar(&leaveID, "id", "", "node id")
nodeCmd.AddCommand(nodeJoinCmd)
nodeCmd.AddCommand(nodeLeaveCmd)
nodeCmd.AddCommand(nodeListCmd)
rootCmd.AddCommand(nodeCmd)
}
+52
View File
@@ -0,0 +1,52 @@
package cli
import (
"encoding/json"
"fmt"
"github.com/spf13/cobra"
)
var (
version = "0.1.0-dev"
gitCommit = "unknown"
buildTime = "unknown"
)
var rootCmd = &cobra.Command{
Use: "orca",
Short: "Orca — offline/CLI-first orchestration engine",
Long: `Orca is a minimalist, offline-first, CLI-first orchestration engine
inspired by HashiCorp Nomad, prioritizing stability, security, and simplicity
over feature richness.`,
SilenceUsage: true,
SilenceErrors: true,
}
var jsonOutput bool
func init() {
rootCmd.PersistentFlags().BoolVar(&jsonOutput, "json", false, "output in JSON format")
}
func Execute() error {
return rootCmd.Execute()
}
func printJSON(v any) error {
enc := json.NewEncoder(rootCmd.OutOrStdout())
enc.SetIndent("", " ")
return enc.Encode(v)
}
func printText(format string, args ...any) {
fmt.Fprintf(rootCmd.OutOrStdout(), format, args...)
}
func printResult(text string, jsonObj any) {
if jsonOutput {
_ = printJSON(jsonObj)
return
}
printText("%s\n", text)
}
+67
View File
@@ -0,0 +1,67 @@
package cli
import (
"strings"
"testing"
)
func TestVersionCommandExists(t *testing.T) {
found := false
for _, cmd := range rootCmd.Commands() {
if cmd.Name() == "version" {
found = true
break
}
}
if !found {
t.Fatal("version command not registered")
}
}
func TestRootHasAllSubcommands(t *testing.T) {
expected := []string{"version", "init", "status", "node", "job"}
registered := make(map[string]bool)
for _, cmd := range rootCmd.Commands() {
registered[cmd.Name()] = true
}
for _, name := range expected {
if !registered[name] {
t.Errorf("expected subcommand %q not registered", name)
}
}
}
func TestNodeSubcommands(t *testing.T) {
expected := []string{"join", "leave", "list"}
registered := make(map[string]bool)
for _, cmd := range nodeCmd.Commands() {
registered[cmd.Name()] = true
}
for _, name := range expected {
if !registered[name] {
t.Errorf("expected node subcommand %q not registered", name)
}
}
}
func TestJobSubcommands(t *testing.T) {
expected := []string{"run", "list", "stop", "logs"}
registered := make(map[string]bool)
for _, cmd := range jobCmd.Commands() {
registered[cmd.Name()] = true
}
for _, name := range expected {
if !registered[name] {
t.Errorf("expected job subcommand %q not registered", name)
}
}
}
func TestRootHelpMentionsKeyPillars(t *testing.T) {
help := rootCmd.Long
for _, pillar := range []string{"offline", "CLI", "Nomad", "simplicity"} {
if !strings.Contains(strings.ToLower(help), strings.ToLower(pillar)) {
t.Errorf("root help does not mention pillar %q", pillar)
}
}
}
+36
View File
@@ -0,0 +1,36 @@
package cli
import (
"github.com/spf13/cobra"
)
var statusCmd = &cobra.Command{
Use: "status",
Short: "Show orca daemon status",
Long: "Display the current status of the local orca daemon, including version, uptime, and connection info.",
RunE: func(cmd *cobra.Command, args []string) error {
status := map[string]any{
"version": version,
"daemon": "stopped",
"uptime": "0s",
"api_addr": "https://localhost:8443",
"health": "unknown",
"phase": "1-cli-skeleton",
"milestone": "v0.1",
}
if jsonOutput {
return printJSON(status)
}
printText("orca daemon status\n")
printText(" version: %s\n", version)
printText(" daemon: %s\n", "stopped (daemon not yet implemented in Phase 1)")
printText(" api_addr: %s\n", "https://localhost:8443")
printText(" phase: %s\n", "1-cli-skeleton")
printText(" milestone: %s\n", "v0.1")
return nil
},
}
func init() {
rootCmd.AddCommand(statusCmd)
}
+29
View File
@@ -0,0 +1,29 @@
package cli
import (
"github.com/spf13/cobra"
)
var versionCmd = &cobra.Command{
Use: "version",
Short: "Print version information",
Long: "Print the orca version, git commit, and build time.",
RunE: func(cmd *cobra.Command, args []string) error {
info := map[string]string{
"version": version,
"git_commit": gitCommit,
"build_time": buildTime,
}
if jsonOutput {
return printJSON(info)
}
printText("orca version %s\n", version)
printText(" git commit: %s\n", gitCommit)
printText(" build time: %s\n", buildTime)
return nil
},
}
func init() {
rootCmd.AddCommand(versionCmd)
}
+129
View File
@@ -0,0 +1,129 @@
package daemon
import (
"context"
"encoding/json"
"log/slog"
"net/http"
"time"
)
// handleHealthz reports liveness. It does NOT check dependencies — by design,
// a process that can answer this is "alive" even if its DB is wedged. Use
// /readyz for dependency health.
func (s *Server) handleHealthz(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
writeError(w, http.StatusMethodNotAllowed, "method not allowed")
return
}
writeJSON(w, http.StatusOK, map[string]any{
"status": "alive",
"time": time.Now().UTC().Format(time.RFC3339),
})
}
// handleReadyz reports readiness. Returns 503 if either:
// - MarkReady has not been called, OR
// - the SQLite database cannot be pinged within 2s.
//
// Distinguishing these cases in the response body helps operators triage.
func (s *Server) handleReadyz(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
writeError(w, http.StatusMethodNotAllowed, "method not allowed")
return
}
ctx, cancel := context.WithTimeout(r.Context(), 2*time.Second)
defer cancel()
if !s.ready.Load() {
writeJSON(w, http.StatusServiceUnavailable, map[string]any{
"status": "not_ready",
"reason": "daemon not marked ready",
})
return
}
if err := s.db.PingContext(ctx); err != nil {
s.log.Warn("readyz db ping failed",
slog.String("component", "daemon"),
slog.String("error", err.Error()))
writeJSON(w, http.StatusServiceUnavailable, map[string]any{
"status": "not_ready",
"reason": "db ping failed",
})
return
}
writeJSON(w, http.StatusOK, map[string]any{
"status": "ready",
"db": "ok",
})
}
// handleStatus returns a small diagnostic JSON blob. Cheap to call; does
// NOT touch the database unless we want a DB status check, in which case
// the ping is bounded by 2s.
func (s *Server) handleStatus(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
writeError(w, http.StatusMethodNotAllowed, "method not allowed")
return
}
ctx, cancel := context.WithTimeout(r.Context(), 2*time.Second)
defer cancel()
dbStatus := "ok"
if err := s.db.PingContext(ctx); err != nil {
dbStatus = "error"
s.log.Warn("status db ping failed",
slog.String("component", "daemon"),
slog.String("error", err.Error()))
}
writeJSON(w, http.StatusOK, map[string]any{
"version": Version,
"phase": "5-health-checks",
"milestone": "v0.1",
"db": dbStatus,
"ready": s.ready.Load(),
"time": time.Now().UTC().Format(time.RFC3339),
})
}
// writeJSON encodes body as JSON with the given status code.
// Errors during encoding are logged but not surfaced — we cannot write
// another header after the response has started.
func writeJSON(w http.ResponseWriter, code int, body any) {
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.WriteHeader(code)
_ = json.NewEncoder(w).Encode(body)
}
// writeError emits a uniform error envelope: {"error": "<message>"}.
func writeError(w http.ResponseWriter, code int, msg string) {
writeJSON(w, code, map[string]string{"error": msg})
}
// loggingMiddleware wraps the mux with a structured access log. It does
// NOT log request/response bodies (could contain secrets); just method,
// path, status, and duration.
func loggingMiddleware(log *slog.Logger, next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
start := time.Now()
ww := &statusRecorder{ResponseWriter: w, status: 200}
next.ServeHTTP(ww, r)
log.Info("http",
slog.String("method", r.Method),
slog.String("path", r.URL.Path),
slog.Int("status", ww.status),
slog.Duration("dur", time.Since(start)),
slog.String("remote", r.RemoteAddr),
)
})
}
type statusRecorder struct {
http.ResponseWriter
status int
}
func (s *statusRecorder) WriteHeader(code int) {
s.status = code
s.ResponseWriter.WriteHeader(code)
}
+183
View File
@@ -0,0 +1,183 @@
package daemon
import (
"encoding/json"
"net/http"
"net/http/httptest"
"path/filepath"
"testing"
"git.cloudinit.dev/coreci/orca/internal/store"
)
func newTestServer(t *testing.T) *Server {
t.Helper()
db, err := store.Open(filepath.Join(t.TempDir(), "test.db"))
if err != nil {
t.Fatalf("open db: %v", err)
}
t.Cleanup(func() { _ = db.Close() })
s := NewServer(Options{DB: db, Log: nil, Addr: "127.0.0.1:0"})
s.MarkReady()
return s
}
func TestHealthzReturns200(t *testing.T) {
s := newTestServer(t)
req := httptest.NewRequest("GET", "/healthz", nil)
rr := httptest.NewRecorder()
s.mux().ServeHTTP(rr, req)
if rr.Code != 200 {
t.Errorf("expected 200, got %d", rr.Code)
}
var body map[string]any
_ = json.NewDecoder(rr.Body).Decode(&body)
if body["status"] != "alive" {
t.Errorf("expected status alive, got %v", body["status"])
}
}
func TestReadyzReturns200WhenReady(t *testing.T) {
s := newTestServer(t)
s.MarkReady()
req := httptest.NewRequest("GET", "/readyz", nil)
rr := httptest.NewRecorder()
s.mux().ServeHTTP(rr, req)
if rr.Code != 200 {
t.Errorf("expected 200, got %d", rr.Code)
}
}
func TestReadyzReturns503WhenNotReady(t *testing.T) {
s := newTestServer(t)
s.MarkNotReady()
req := httptest.NewRequest("GET", "/readyz", nil)
rr := httptest.NewRecorder()
s.mux().ServeHTTP(rr, req)
if rr.Code != 503 {
t.Errorf("expected 503, got %d", rr.Code)
}
}
func TestStatusReturns200(t *testing.T) {
s := newTestServer(t)
s.MarkReady()
req := httptest.NewRequest("GET", "/v1/status", nil)
rr := httptest.NewRecorder()
s.mux().ServeHTTP(rr, req)
if rr.Code != 200 {
t.Errorf("expected 200, got %d", rr.Code)
}
var body map[string]any
_ = json.NewDecoder(rr.Body).Decode(&body)
if body["db"] != "ok" {
t.Errorf("expected db ok, got %v", body["db"])
}
if body["milestone"] != "v0.1" {
t.Errorf("expected milestone v0.1, got %v", body["milestone"])
}
}
func TestJobsCollectionEmpty(t *testing.T) {
s := newTestServer(t)
req := httptest.NewRequest("GET", "/v1/jobs", nil)
rr := httptest.NewRecorder()
s.mux().ServeHTTP(rr, req)
if rr.Code != 200 {
t.Errorf("expected 200, got %d", rr.Code)
}
var body map[string]any
_ = json.NewDecoder(rr.Body).Decode(&body)
if body["count"].(float64) != 0 {
t.Errorf("expected count 0, got %v", body["count"])
}
}
func TestJobsCollectionMethodNotAllowed(t *testing.T) {
s := newTestServer(t)
req := httptest.NewRequest("PUT", "/v1/jobs", nil)
rr := httptest.NewRecorder()
s.mux().ServeHTTP(rr, req)
if rr.Code != http.StatusMethodNotAllowed {
t.Errorf("expected 405, got %d", rr.Code)
}
}
func TestJobsItemNotFound(t *testing.T) {
s := newTestServer(t)
req := httptest.NewRequest("GET", "/v1/jobs/nonexistent", nil)
rr := httptest.NewRecorder()
s.mux().ServeHTTP(rr, req)
if rr.Code != http.StatusNotFound {
t.Errorf("expected 404, got %d", rr.Code)
}
}
func TestJobsItemInvalidID(t *testing.T) {
s := newTestServer(t)
req := httptest.NewRequest("GET", "/v1/jobs/has%20space", nil)
rr := httptest.NewRecorder()
s.mux().ServeHTTP(rr, req)
if rr.Code != http.StatusBadRequest {
t.Errorf("expected 400, got %d", rr.Code)
}
}
func TestNodesCollectionEmpty(t *testing.T) {
s := newTestServer(t)
req := httptest.NewRequest("GET", "/v1/nodes", nil)
rr := httptest.NewRecorder()
s.mux().ServeHTTP(rr, req)
if rr.Code != 200 {
t.Errorf("expected 200, got %d", rr.Code)
}
var body map[string]any
_ = json.NewDecoder(rr.Body).Decode(&body)
if body["count"].(float64) != 0 {
t.Errorf("expected count 0, got %v", body["count"])
}
}
func TestTasksCollectionEmpty(t *testing.T) {
s := newTestServer(t)
req := httptest.NewRequest("GET", "/v1/tasks", nil)
rr := httptest.NewRecorder()
s.mux().ServeHTTP(rr, req)
if rr.Code != 200 {
t.Errorf("expected 200, got %d", rr.Code)
}
}
func TestTasksCollectionInvalidLimit(t *testing.T) {
s := newTestServer(t)
req := httptest.NewRequest("GET", "/v1/tasks?limit=abc", nil)
rr := httptest.NewRecorder()
s.mux().ServeHTTP(rr, req)
if rr.Code != http.StatusBadRequest {
t.Errorf("expected 400, got %d", rr.Code)
}
}
func TestValidateID(t *testing.T) {
cases := []struct {
id string
valid bool
}{
{"abc-123", true},
{"550e8400-e29b-41d4-a716-446655440000", true},
{"a", true},
{"", false},
{"has space", false},
{"with/slash", false},
{"../etc/passwd", false},
{string([]byte{0x00, 'a'}), false},
}
for _, c := range cases {
err := validateID(c.id)
if (err == nil) != c.valid {
t.Errorf("validateID(%q): valid=%v, err=%v", c.id, c.valid, err)
}
}
}
+109
View File
@@ -0,0 +1,109 @@
package daemon
import (
"context"
"errors"
"log/slog"
"net/http"
"strings"
"time"
"git.cloudinit.dev/coreci/orca/internal/model"
"git.cloudinit.dev/coreci/orca/internal/store"
)
// handleJobsCollection handles /v1/jobs.
// - GET → list all jobs
// - POST → not yet supported (job submission is CLI-only in v0.1)
func (s *Server) handleJobsCollection(w http.ResponseWriter, r *http.Request) {
ctx, cancel := context.WithTimeout(r.Context(), 5*time.Second)
defer cancel()
switch r.Method {
case http.MethodGet:
jobs, err := store.NewJobRepo(s.db).List(ctx)
if err != nil {
s.log.Error("list jobs",
slog.String("component", "daemon"),
slog.String("error", err.Error()))
writeError(w, http.StatusInternalServerError, "failed to list jobs")
return
}
if jobs == nil {
jobs = []*model.Job{}
}
writeJSON(w, http.StatusOK, map[string]any{"jobs": jobs, "count": len(jobs)})
case http.MethodPost:
// Job submission via HTTP is intentionally not exposed in v0.1.
// The CLI submits jobs to the local store directly; the daemon
// exists for observability and lifecycle control.
writeError(w, http.StatusNotImplemented, "job submission via API is not supported in v0.1; use 'orca job run'")
default:
writeError(w, http.StatusMethodNotAllowed, "method not allowed")
}
}
// handleJobsItem handles /v1/jobs/{id} and /v1/jobs/{id}/tasks.
// - GET /v1/jobs/{id} → job details
// - GET /v1/jobs/{id}/tasks → tasks for a job
func (s *Server) handleJobsItem(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
writeError(w, http.StatusMethodNotAllowed, "method not allowed")
return
}
ctx, cancel := context.WithTimeout(r.Context(), 5*time.Second)
defer cancel()
// Path is /v1/jobs/{id} or /v1/jobs/{id}/tasks
path := strings.TrimPrefix(r.URL.Path, "/v1/jobs/")
parts := strings.Split(path, "/")
if len(parts) == 0 || parts[0] == "" {
writeError(w, http.StatusBadRequest, "job id required")
return
}
id := parts[0]
if err := validateID(id); err != nil {
writeError(w, http.StatusBadRequest, err.Error())
return
}
// /v1/jobs/{id}/tasks
if len(parts) == 2 && parts[1] == "tasks" {
tasks, err := store.NewTaskRepo(s.db).ListByJob(ctx, id)
if err != nil {
s.log.Error("list tasks for job",
slog.String("component", "daemon"),
slog.String("job_id", id),
slog.String("error", err.Error()))
writeError(w, http.StatusInternalServerError, "failed to list tasks")
return
}
if tasks == nil {
tasks = []*model.Task{}
}
writeJSON(w, http.StatusOK, map[string]any{"tasks": tasks, "count": len(tasks), "job_id": id})
return
}
// /v1/jobs/{id} (with no further path)
if len(parts) != 1 {
writeError(w, http.StatusNotFound, "not found")
return
}
job, err := store.NewJobRepo(s.db).Get(ctx, id)
if err != nil {
if errors.Is(err, store.ErrNotFound) {
writeError(w, http.StatusNotFound, "job not found")
return
}
s.log.Error("get job",
slog.String("component", "daemon"),
slog.String("job_id", id),
slog.String("error", err.Error()))
writeError(w, http.StatusInternalServerError, "failed to get job")
return
}
writeJSON(w, http.StatusOK, job)
}
+31
View File
@@ -0,0 +1,31 @@
package daemon
import (
"context"
"net/http"
"time"
"git.cloudinit.dev/coreci/orca/internal/model"
"git.cloudinit.dev/coreci/orca/internal/store"
)
// handleNodesCollection handles /v1/nodes (GET only in v0.1).
// Node registration is CLI-only; the API is read-only for observability.
func (s *Server) handleNodesCollection(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
writeError(w, http.StatusMethodNotAllowed, "method not allowed")
return
}
ctx, cancel := context.WithTimeout(r.Context(), 5*time.Second)
defer cancel()
nodes, err := store.NewNodeRepo(s.db).List(ctx)
if err != nil {
writeError(w, http.StatusInternalServerError, "failed to list nodes")
return
}
if nodes == nil {
nodes = []*model.Node{}
}
writeJSON(w, http.StatusOK, map[string]any{"nodes": nodes, "count": len(nodes)})
}
+120
View File
@@ -0,0 +1,120 @@
// Package daemon implements the orca HTTP daemon.
//
// The daemon exposes health endpoints (/healthz, /readyz), a status endpoint
// (/v1/status), and a v1 resource API for jobs, nodes, and tasks. All handlers
// follow the project conventions:
//
// - context.Context propagated to all I/O
// - errors wrapped with %w
// - structured JSON via writeJSON
// - no secrets in logs
// - input validation on path/query/body
package daemon
import (
"context"
"database/sql"
"errors"
"log/slog"
"net/http"
"sync/atomic"
"time"
)
// Server is the orca HTTP daemon. It holds shared dependencies and lifecycle
// state. Construct it with NewServer, then call Start/Shutdown.
type Server struct {
db *sql.DB
log *slog.Logger
addr string
ready atomic.Bool
httpServer *http.Server
}
// Options configures a new Server.
type Options struct {
DB *sql.DB
Log *slog.Logger
Addr string
Actor string // used for audit logging from API requests
}
// NewServer constructs a Server with the default mux and route table.
func NewServer(opts Options) *Server {
if opts.Log == nil {
opts.Log = slog.Default()
}
if opts.Addr == "" {
opts.Addr = ":8080"
}
if opts.Actor == "" {
opts.Actor = "api"
}
s := &Server{
db: opts.DB,
log: opts.Log,
addr: opts.Addr,
}
s.httpServer = &http.Server{
Addr: opts.Addr,
Handler: s.mux(),
ReadHeaderTimeout: 5 * time.Second,
ReadTimeout: 15 * time.Second,
WriteTimeout: 30 * time.Second,
IdleTimeout: 60 * time.Second,
}
return s
}
// Addr returns the configured listen address.
func (s *Server) Addr() string { return s.addr }
// MarkReady flips the readiness flag to true. The /readyz endpoint returns
// 200 only when this flag is set AND the database is reachable.
func (s *Server) MarkReady() { s.ready.Store(true) }
// MarkNotReady flips the readiness flag to false. Called at shutdown start
// so load balancers stop routing traffic.
func (s *Server) MarkNotReady() { s.ready.Store(false) }
// Ready reports the current readiness flag.
func (s *Server) Ready() bool { return s.ready.Load() }
// mux builds the route table. Handlers are split across files:
// - health.go /healthz, /readyz, /v1/status
// - jobs_handler.go /v1/jobs/*
// - nodes_handler.go /v1/nodes/*
// - tasks_handler.go /v1/tasks/*
func (s *Server) mux() http.Handler {
mux := http.NewServeMux()
mux.HandleFunc("/healthz", s.handleHealthz)
mux.HandleFunc("/readyz", s.handleReadyz)
mux.HandleFunc("/v1/status", s.handleStatus)
mux.HandleFunc("/v1/jobs", s.handleJobsCollection)
mux.HandleFunc("/v1/jobs/", s.handleJobsItem)
mux.HandleFunc("/v1/nodes", s.handleNodesCollection)
mux.HandleFunc("/v1/tasks", s.handleTasksCollection)
return loggingMiddleware(s.log, mux)
}
// Start runs the HTTP server. Returns http.ErrServerClosed on clean shutdown.
func (s *Server) Start() error {
s.log.Info("daemon starting",
slog.String("addr", s.addr),
slog.String("component", "daemon"))
return s.httpServer.ListenAndServe()
}
// Shutdown gracefully stops the server, bounded by ctx. It also flips the
// readiness flag to false so /readyz returns 503 immediately.
func (s *Server) Shutdown(ctx context.Context) error {
s.MarkNotReady()
s.log.Info("daemon shutting down", slog.String("component", "daemon"))
return s.httpServer.Shutdown(ctx)
}
// IsShutdownErr reports whether err is the expected error from a stopped server.
func IsShutdownErr(err error) bool {
return errors.Is(err, http.ErrServerClosed)
}
+150
View File
@@ -0,0 +1,150 @@
package daemon
import (
"context"
"errors"
"io"
"net"
"net/http"
"path/filepath"
"strings"
"testing"
"time"
"git.cloudinit.dev/coreci/orca/internal/store"
)
func TestServerLifecycle(t *testing.T) {
db, err := store.Open(filepath.Join(t.TempDir(), "lifecycle.db"))
if err != nil {
t.Fatalf("open db: %v", err)
}
defer db.Close()
s := NewServer(Options{DB: db, Addr: "127.0.0.1:0"})
s.MarkReady()
if !s.Ready() {
t.Error("expected server ready after MarkReady")
}
s.MarkNotReady()
if s.Ready() {
t.Error("expected server not ready after MarkNotReady")
}
s.MarkReady()
// Bind an ephemeral listener and serve on it directly.
ln, err := net.Listen("tcp", "127.0.0.1:0")
if err != nil {
t.Fatalf("listen: %v", err)
}
addr := ln.Addr().String()
errCh := make(chan error, 1)
go func() {
err := s.httpServer.Serve(ln)
if err != nil && !errors.Is(err, http.ErrServerClosed) {
errCh <- err
}
close(errCh)
}()
// Verify healthz responds.
deadline := time.Now().Add(2 * time.Second)
for time.Now().Before(deadline) {
c := http.Client{Timeout: 200 * time.Millisecond}
r, err := c.Get("http://" + addr + "/healthz")
if err == nil {
_ = r.Body.Close()
if r.StatusCode == 200 {
break
}
}
time.Sleep(20 * time.Millisecond)
}
resp, err := http.Get("http://" + addr + "/healthz")
if err != nil {
t.Fatalf("GET /healthz: %v", err)
}
if resp.StatusCode != 200 {
t.Errorf("expected 200, got %d", resp.StatusCode)
}
body, _ := io.ReadAll(resp.Body)
_ = resp.Body.Close()
if !strings.Contains(string(body), `"alive"`) {
t.Errorf("expected alive status in body, got %s", string(body))
}
// readyz returns 200 when ready.
resp, err = http.Get("http://" + addr + "/readyz")
if err != nil {
t.Fatalf("GET /readyz: %v", err)
}
if resp.StatusCode != 200 {
t.Errorf("expected 200, got %d", resp.StatusCode)
}
_ = resp.Body.Close()
// /v1/jobs returns JSON
resp, err = http.Get("http://" + addr + "/v1/jobs")
if err != nil {
t.Fatalf("GET /v1/jobs: %v", err)
}
if resp.StatusCode != 200 {
t.Errorf("expected 200, got %d", resp.StatusCode)
}
if ct := resp.Header.Get("Content-Type"); !strings.HasPrefix(ct, "application/json") {
t.Errorf("expected JSON content-type, got %s", ct)
}
_ = resp.Body.Close()
// /v1/nodes returns JSON
resp, err = http.Get("http://" + addr + "/v1/nodes")
if err != nil {
t.Fatalf("GET /v1/nodes: %v", err)
}
if resp.StatusCode != 200 {
t.Errorf("expected 200, got %d", resp.StatusCode)
}
_ = resp.Body.Close()
// /v1/tasks returns JSON
resp, err = http.Get("http://" + addr + "/v1/tasks")
if err != nil {
t.Fatalf("GET /v1/tasks: %v", err)
}
if resp.StatusCode != 200 {
t.Errorf("expected 200, got %d", resp.StatusCode)
}
_ = resp.Body.Close()
// Shutdown cleanly.
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()
if err := s.Shutdown(ctx); err != nil {
t.Errorf("shutdown: %v", err)
}
if s.Ready() {
t.Error("expected not-ready after shutdown")
}
// Server should report ErrServerClosed or nil.
select {
case err := <-errCh:
if err != nil && !errors.Is(err, http.ErrServerClosed) {
t.Errorf("expected nil or ErrServerClosed, got %v", err)
}
case <-time.After(2 * time.Second):
t.Error("server did not exit after Shutdown")
}
}
func TestIsShutdownErr(t *testing.T) {
if !IsShutdownErr(http.ErrServerClosed) {
t.Error("expected IsShutdownErr(http.ErrServerClosed) to be true")
}
if IsShutdownErr(errors.New("other")) {
t.Error("expected false for other errors")
}
}
+66
View File
@@ -0,0 +1,66 @@
package daemon
import (
"context"
"log/slog"
"net/http"
"strconv"
"time"
"git.cloudinit.dev/coreci/orca/internal/model"
"git.cloudinit.dev/coreci/orca/internal/store"
)
// handleTasksCollection handles /v1/tasks (GET only).
// Optional query param: ?job_id=<id> to filter by job.
// Optional: ?limit=<n> (default 100, max 1000).
func (s *Server) handleTasksCollection(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodGet {
writeError(w, http.StatusMethodNotAllowed, "method not allowed")
return
}
ctx, cancel := context.WithTimeout(r.Context(), 5*time.Second)
defer cancel()
jobID := r.URL.Query().Get("job_id")
if jobID != "" {
if err := validateID(jobID); err != nil {
writeError(w, http.StatusBadRequest, err.Error())
return
}
}
limit := 100
if v := r.URL.Query().Get("limit"); v != "" {
n, err := strconv.Atoi(v)
if err != nil || n <= 0 {
writeError(w, http.StatusBadRequest, "invalid limit")
return
}
if n > 1000 {
n = 1000
}
limit = n
}
repo := store.NewTaskRepo(s.db)
var tasks []*model.Task
var err error
if jobID != "" {
tasks, err = repo.ListByJob(ctx, jobID)
} else {
tasks, err = repo.ListRecent(ctx, limit)
}
if err != nil {
s.log.Error("list tasks",
slog.String("component", "daemon"),
slog.String("job_id", jobID),
slog.String("error", err.Error()))
writeError(w, http.StatusInternalServerError, "failed to list tasks")
return
}
if tasks == nil {
tasks = []*model.Task{}
}
writeJSON(w, http.StatusOK, map[string]any{"tasks": tasks, "count": len(tasks)})
}
+28
View File
@@ -0,0 +1,28 @@
package daemon
import (
"fmt"
"regexp"
"strings"
)
// idPattern constrains path IDs to a safe subset: alphanumerics, hyphens,
// and underscores. UUIDs and our internal IDs both fit. We reject anything
// that smells like a path-traversal, control character, or shell metachar.
var idPattern = regexp.MustCompile(`^[A-Za-z0-9_-]{1,128}$`)
// validateID checks that an ID is well-formed and within length limits.
// It exists primarily as a defense-in-depth measure against path traversal
// and accidental log-injection when the ID is echoed back in error messages.
func validateID(id string) error {
if id == "" {
return fmt.Errorf("id required")
}
if strings.ContainsAny(id, "\r\n\t\x00") {
return fmt.Errorf("invalid id")
}
if !idPattern.MatchString(id) {
return fmt.Errorf("invalid id format")
}
return nil
}
+5
View File
@@ -0,0 +1,5 @@
package daemon
// Version is the daemon version. It is set at build time via -ldflags by the
// release pipeline, but defaults to a dev marker for local development.
var Version = "0.1.0-dev"
+52
View File
@@ -0,0 +1,52 @@
package engine
import (
"context"
"log/slog"
"git.cloudinit.dev/coreci/orca/internal/store"
)
// Audit wraps a slog.Logger and persists structured audit entries to SQLite.
type Audit struct {
repo *store.AuditRepo
log *slog.Logger
}
func NewAudit(repo *store.AuditRepo, log *slog.Logger) *Audit {
if log == nil {
log = slog.Default()
}
return &Audit{repo: repo, log: log}
}
func (a *Audit) Record(ctx context.Context, actor, action, resource, result string, err error, meta map[string]any) {
entry := &store.AuditEntry{
Actor: actor,
Action: action,
Resource: resource,
Result: result,
Metadata: meta,
}
if err != nil {
entry.Error = err.Error()
}
if persistErr := a.repo.Append(ctx, entry); persistErr != nil {
a.log.Error("audit persist failed",
slog.String("action", action),
slog.String("resource", resource),
slog.String("error", persistErr.Error()))
}
attrs := []any{
slog.String("actor", actor),
slog.String("action", action),
slog.String("resource", resource),
slog.String("result", result),
}
if err != nil {
attrs = append(attrs, slog.String("error", err.Error()))
a.log.Warn("audit", attrs...)
} else {
a.log.Info("audit", attrs...)
}
}
+150
View File
@@ -0,0 +1,150 @@
package engine
import (
"bytes"
"context"
"fmt"
"log/slog"
"os/exec"
"sync"
"time"
"github.com/google/uuid"
"git.cloudinit.dev/coreci/orca/internal/model"
"git.cloudinit.dev/coreci/orca/internal/store"
)
type Executor struct {
jobs *store.JobRepo
tasks *store.TaskRepo
log *slog.Logger
mu sync.Mutex
}
func NewExecutor(jobs *store.JobRepo, tasks *store.TaskRepo, log *slog.Logger) *Executor {
if log == nil {
log = slog.Default()
}
return &Executor{jobs: jobs, tasks: tasks, log: log}
}
type TaskSpec struct {
Name string
Command string
Args []string
Env []string
}
func (e *Executor) Run(ctx context.Context, job *model.Job, specs []TaskSpec) error {
e.mu.Lock()
defer e.mu.Unlock()
// Insert the job first so tasks can reference it via foreign key.
if err := e.jobs.Insert(ctx, job); err != nil {
return err
}
if err := e.jobs.UpdateStatus(ctx, job.ID, model.JobStatusRunning, 0); err != nil {
return err
}
var (
wg sync.WaitGroup
failedCount int
exitCode int
mu sync.Mutex
)
for _, ts := range specs {
wg.Add(1)
go func(ts TaskSpec) {
defer wg.Done()
if err := e.runOne(ctx, job, ts); err != nil {
mu.Lock()
failedCount++
e.log.Error("task failed",
slog.String("job_id", job.ID),
slog.String("task", ts.Name),
slog.String("error", err.Error()))
mu.Unlock()
}
}(ts)
}
wg.Wait()
if failedCount > 0 {
exitCode = 1
if err := e.jobs.UpdateStatus(ctx, job.ID, model.JobStatusFailed, exitCode); err != nil {
return err
}
return fmt.Errorf("%d/%d tasks failed", failedCount, len(specs))
}
if err := e.jobs.UpdateStatus(ctx, job.ID, model.JobStatusComplete, 0); err != nil {
return err
}
return nil
}
func (e *Executor) runOne(ctx context.Context, job *model.Job, ts TaskSpec) error {
task := &model.Task{
ID: uuid.NewString(),
JobID: job.ID,
Command: ts.Command,
Args: ts.Args,
Env: ts.Env,
Status: model.TaskStatusPending,
}
if err := e.tasks.Insert(ctx, task); err != nil {
return err
}
cmd := exec.CommandContext(ctx, ts.Command, ts.Args...)
cmd.Env = append(cmd.Environ(), ts.Env...)
// WaitDelay (Go 1.25+) bounds the time spent waiting on a child process
// that fails to exit after the context is canceled.
cmd.WaitDelay = 5 * time.Second
var stdout, stderr bytes.Buffer
cmd.Stdout = &stdout
cmd.Stderr = &stderr
if err := cmd.Start(); err != nil {
_ = e.tasks.UpdateKilled(ctx, task.ID)
return fmt.Errorf("start: %w", err)
}
if err := e.tasks.UpdateRunning(ctx, task.ID, cmd.Process.Pid); err != nil {
e.log.Warn("update running failed", slog.String("error", err.Error()))
}
e.log.Info("task started",
slog.String("job_id", job.ID),
slog.String("task", ts.Name),
slog.Int("pid", cmd.Process.Pid))
done := make(chan error, 1)
go func() { done <- cmd.Wait() }()
select {
case err := <-done:
exitCode := 0
if err != nil {
if ee, ok := err.(*exec.ExitError); ok {
exitCode = ee.ExitCode()
} else {
exitCode = 1
}
}
_ = e.tasks.UpdateDone(ctx, task.ID, exitCode, stdout.String(), stderr.String())
if err != nil {
return err
}
return nil
case <-ctx.Done():
// WaitDelay (set above) gives the process a grace period to exit
// cleanly before being killed.
_ = e.tasks.UpdateKilled(ctx, task.ID)
return ctx.Err()
}
}
+70
View File
@@ -0,0 +1,70 @@
package engine
import (
"context"
"fmt"
"log/slog"
"git.cloudinit.dev/coreci/orca/internal/model"
"git.cloudinit.dev/coreci/orca/internal/store"
)
type NodeRegistry struct {
repo *store.NodeRepo
audit *Audit
log *slog.Logger
}
func NewNodeRegistry(repo *store.NodeRepo, audit *Audit, log *slog.Logger) *NodeRegistry {
if log == nil {
log = slog.Default()
}
return &NodeRegistry{repo: repo, audit: audit, log: log}
}
func (r *NodeRegistry) Join(ctx context.Context, n *model.Node) error {
if err := r.repo.Insert(ctx, n); err != nil {
r.audit.Record(ctx, "cli", "node.join", n.ID, "failure", err, map[string]any{
"name": n.Name,
"address": n.Address,
})
return fmt.Errorf("join node: %w", err)
}
r.audit.Record(ctx, "cli", "node.join", n.ID, "success", nil, map[string]any{
"name": n.Name,
"address": n.Address,
})
r.log.Info("node joined",
slog.String("node_id", n.ID),
slog.String("name", n.Name),
slog.String("address", n.Address))
return nil
}
func (r *NodeRegistry) Leave(ctx context.Context, id string) error {
if err := r.repo.UpdateState(ctx, id, model.NodeStateLeft); err != nil {
r.audit.Record(ctx, "cli", "node.leave", id, "failure", err, nil)
return fmt.Errorf("leave node: %w", err)
}
r.audit.Record(ctx, "cli", "node.leave", id, "success", nil, nil)
r.log.Info("node left", slog.String("node_id", id))
return nil
}
func (r *NodeRegistry) Forget(ctx context.Context, id string) error {
if err := r.repo.Delete(ctx, id); err != nil {
r.audit.Record(ctx, "cli", "node.forget", id, "failure", err, nil)
return fmt.Errorf("forget node: %w", err)
}
r.audit.Record(ctx, "cli", "node.forget", id, "success", nil, nil)
r.log.Info("node removed from registry", slog.String("node_id", id))
return nil
}
func (r *NodeRegistry) List(ctx context.Context) ([]*model.Node, error) {
return r.repo.List(ctx)
}
func (r *NodeRegistry) Get(ctx context.Context, id string) (*model.Node, error) {
return r.repo.Get(ctx, id)
}
+69
View File
@@ -0,0 +1,69 @@
package jobspec
import (
"fmt"
"os"
"strings"
"github.com/hashicorp/hcl/v2"
"github.com/hashicorp/hcl/v2/gohcl"
"github.com/hashicorp/hcl/v2/hclsimple"
)
type Spec struct {
Job JobSpec `hcl:"job,block"`
Tasks []TaskSpec `hcl:"task,block"`
}
type JobSpec struct {
Name string `hcl:"name,label"`
Type string `hcl:"type,optional"`
}
type TaskSpec struct {
Name string `hcl:"name,label"`
Command string `hcl:"command"`
Args []string `hcl:"args,optional"`
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
}
func (s *Spec) Validate() error {
if strings.TrimSpace(s.Job.Name) == "" {
return fmt.Errorf("job name is required")
}
if len(s.Tasks) == 0 {
return fmt.Errorf("at least one task is required")
}
return nil
}
var _ = hcl.Diagnostics{}
var _ = gohcl.DecodeBody
+60
View File
@@ -0,0 +1,60 @@
package jobspec
import (
"testing"
)
func TestParseValid(t *testing.T) {
hcl := `
job "demo" {
}
task "build" {
command = "/bin/echo"
args = ["hello", "world"]
}
`
spec, err := Parse([]byte(hcl), "test.hcl")
if err != nil {
t.Fatalf("parse: %v", err)
}
if spec.Job.Name != "demo" {
t.Errorf("expected job name 'demo', got %q", spec.Job.Name)
}
if len(spec.Tasks) != 1 {
t.Fatalf("expected 1 task, got %d", len(spec.Tasks))
}
if spec.Tasks[0].Command != "/bin/echo" {
t.Errorf("expected command '/bin/echo', got %q", spec.Tasks[0].Command)
}
if len(spec.Tasks[0].Args) != 2 {
t.Errorf("expected 2 args, got %d", len(spec.Tasks[0].Args))
}
}
func TestParseMissingJob(t *testing.T) {
hcl := `task "x" { command = "/bin/echo" }`
_, err := Parse([]byte(hcl), "test.hcl")
if err == nil {
t.Fatal("expected error for missing job name")
}
}
func TestParseNoTasks(t *testing.T) {
hcl := `job "empty" {}`
_, err := Parse([]byte(hcl), "test.hcl")
if err == nil {
t.Fatal("expected error for no tasks")
}
}
func TestParseTaskMissingCommand(t *testing.T) {
hcl := `
job "x" {}
task "no-cmd" {}
`
_, err := Parse([]byte(hcl), "test.hcl")
if err == nil {
t.Fatal("expected error for missing command")
}
}
+50
View File
@@ -0,0 +1,50 @@
package model
import "time"
type JobStatus string
const (
JobStatusPending JobStatus = "pending"
JobStatusRunning JobStatus = "running"
JobStatusComplete JobStatus = "complete"
JobStatusFailed JobStatus = "failed"
JobStatusStopped JobStatus = "stopped"
)
type Job struct {
ID string `json:"id"`
Name string `json:"name"`
Spec string `json:"spec"`
Status JobStatus `json:"status"`
CreatedAt time.Time `json:"created_at"`
StartedAt *time.Time `json:"started_at,omitempty"`
EndedAt *time.Time `json:"ended_at,omitempty"`
ExitCode int `json:"exit_code"`
}
type TaskStatus string
const (
TaskStatusPending TaskStatus = "pending"
TaskStatusRunning TaskStatus = "running"
TaskStatusComplete TaskStatus = "complete"
TaskStatusFailed TaskStatus = "failed"
TaskStatusKilled TaskStatus = "killed"
)
type Task struct {
ID string `json:"id"`
JobID string `json:"job_id"`
Command string `json:"command"`
Args []string `json:"args"`
Env []string `json:"env,omitempty"`
PID int `json:"pid"`
ExitCode int `json:"exit_code"`
Status TaskStatus `json:"status"`
CreatedAt time.Time `json:"created_at"`
StartedAt *time.Time `json:"started_at,omitempty"`
EndedAt *time.Time `json:"ended_at,omitempty"`
Stdout string `json:"stdout,omitempty"`
Stderr string `json:"stderr,omitempty"`
}
+21
View File
@@ -0,0 +1,21 @@
package model
import "time"
type NodeState string
const (
NodeStatePending NodeState = "pending"
NodeStateReady NodeState = "ready"
NodeStateLeft NodeState = "left"
)
type Node struct {
ID string `json:"id"`
Name string `json:"name"`
Address string `json:"address"`
State NodeState `json:"state"`
JoinedAt time.Time `json:"joined_at"`
LastSeen time.Time `json:"last_seen"`
Metadata map[string]string `json:"metadata,omitempty"`
}
+81
View File
@@ -0,0 +1,81 @@
package store
import (
"context"
"database/sql"
"encoding/json"
"fmt"
"time"
)
type AuditEntry struct {
ID int64 `json:"id"`
Timestamp time.Time `json:"timestamp"`
Actor string `json:"actor"`
Action string `json:"action"`
Resource string `json:"resource"`
Result string `json:"result"`
Error string `json:"error,omitempty"`
Metadata map[string]any `json:"metadata,omitempty"`
}
type AuditRepo struct {
db *sql.DB
}
func NewAuditRepo(db *sql.DB) *AuditRepo {
return &AuditRepo{db: db}
}
func (r *AuditRepo) Append(ctx context.Context, e *AuditEntry) error {
if e.Timestamp.IsZero() {
e.Timestamp = time.Now().UTC()
}
if e.Actor == "" {
e.Actor = "system"
}
metaJSON, _ := json.Marshal(e.Metadata)
if e.Error == "" {
_, err := r.db.ExecContext(ctx,
`INSERT INTO audit_log (timestamp, actor, action, resource, result, metadata) VALUES (?, ?, ?, ?, ?, ?)`,
e.Timestamp, e.Actor, e.Action, e.Resource, e.Result, string(metaJSON))
if err != nil {
return fmt.Errorf("insert audit: %w", err)
}
return nil
}
_, err := r.db.ExecContext(ctx,
`INSERT INTO audit_log (timestamp, actor, action, resource, result, error, metadata) VALUES (?, ?, ?, ?, ?, ?, ?)`,
e.Timestamp, e.Actor, e.Action, e.Resource, e.Result, e.Error, string(metaJSON))
if err != nil {
return fmt.Errorf("insert audit (with error): %w", err)
}
return nil
}
func (r *AuditRepo) List(ctx context.Context, limit int) ([]*AuditEntry, error) {
if limit <= 0 {
limit = 100
}
rows, err := r.db.QueryContext(ctx,
`SELECT id, timestamp, actor, action, resource, result, COALESCE(error, ''), COALESCE(metadata, '') FROM audit_log ORDER BY id DESC LIMIT ?`, limit)
if err != nil {
return nil, fmt.Errorf("list audit: %w", err)
}
defer rows.Close()
var entries []*AuditEntry
for rows.Next() {
var (
e AuditEntry
metaJSON string
)
if err := rows.Scan(&e.ID, &e.Timestamp, &e.Actor, &e.Action, &e.Resource, &e.Result, &e.Error, &metaJSON); err != nil {
return nil, fmt.Errorf("scan audit: %w", err)
}
if metaJSON != "" {
_ = json.Unmarshal([]byte(metaJSON), &e.Metadata)
}
entries = append(entries, &e)
}
return entries, rows.Err()
}
+67
View File
@@ -0,0 +1,67 @@
package store
import (
"context"
"path/filepath"
"testing"
)
func openAuditTestDB(t *testing.T) (*AuditRepo, func()) {
t.Helper()
path := filepath.Join(t.TempDir(), "audit.db")
db, err := Open(path)
if err != nil {
t.Fatalf("open db: %v", err)
}
return NewAuditRepo(db), func() { _ = db.Close() }
}
func TestAuditRepo_AppendAndList(t *testing.T) {
repo, cleanup := openAuditTestDB(t)
defer cleanup()
ctx := context.Background()
for i := 0; i < 5; i++ {
err := repo.Append(ctx, &AuditEntry{
Actor: "cli",
Action: "node.join",
Resource: "node-1",
Result: "success",
})
if err != nil {
t.Fatalf("append[%d]: %v", i, err)
}
}
entries, err := repo.List(ctx, 10)
if err != nil {
t.Fatalf("list: %v", err)
}
if len(entries) != 5 {
t.Errorf("expected 5 entries, got %d", len(entries))
}
}
func TestAuditRepo_WithError(t *testing.T) {
repo, cleanup := openAuditTestDB(t)
defer cleanup()
ctx := context.Background()
err := repo.Append(ctx, &AuditEntry{
Actor: "system",
Action: "task.run",
Resource: "task-1",
Result: "failure",
Error: "exit status 1",
Metadata: map[string]any{"exit_code": 1},
})
if err != nil {
t.Fatalf("append: %v", err)
}
entries, _ := repo.List(ctx, 1)
if len(entries) != 1 {
t.Fatalf("expected 1 entry, got %d", len(entries))
}
if entries[0].Error != "exit status 1" {
t.Errorf("expected error 'exit status 1', got %q", entries[0].Error)
}
}
+246
View File
@@ -0,0 +1,246 @@
package store
import (
"context"
"database/sql"
"encoding/json"
"errors"
"fmt"
"time"
"git.cloudinit.dev/coreci/orca/internal/model"
)
type JobRepo struct {
db *sql.DB
}
func NewJobRepo(db *sql.DB) *JobRepo {
return &JobRepo{db: db}
}
func (r *JobRepo) Insert(ctx context.Context, j *model.Job) error {
if j.CreatedAt.IsZero() {
j.CreatedAt = time.Now().UTC()
}
if j.Status == "" {
j.Status = model.JobStatusPending
}
_, err := r.db.ExecContext(ctx,
`INSERT INTO jobs (id, name, spec, status, exit_code, created_at) VALUES (?, ?, ?, ?, ?, ?)`,
j.ID, j.Name, j.Spec, string(j.Status), j.ExitCode, j.CreatedAt)
if err != nil {
return fmt.Errorf("insert job: %w", err)
}
return nil
}
func (r *JobRepo) Get(ctx context.Context, id string) (*model.Job, error) {
row := r.db.QueryRowContext(ctx,
`SELECT id, name, spec, status, exit_code, created_at, started_at, ended_at FROM jobs WHERE id = ?`, id)
return scanJob(row)
}
func (r *JobRepo) List(ctx context.Context) ([]*model.Job, error) {
rows, err := r.db.QueryContext(ctx,
`SELECT id, name, spec, status, exit_code, created_at, started_at, ended_at FROM jobs ORDER BY created_at DESC`)
if err != nil {
return nil, fmt.Errorf("list jobs: %w", err)
}
defer rows.Close()
var jobs []*model.Job
for rows.Next() {
j, err := scanJob(rows)
if err != nil {
return nil, err
}
jobs = append(jobs, j)
}
return jobs, rows.Err()
}
func (r *JobRepo) UpdateStatus(ctx context.Context, id string, status model.JobStatus, exitCode int) error {
now := time.Now().UTC()
var startedAt, endedAt *time.Time
switch status {
case model.JobStatusRunning:
startedAt = &now
case model.JobStatusComplete, model.JobStatusFailed, model.JobStatusStopped:
endedAt = &now
}
_, err := r.db.ExecContext(ctx,
`UPDATE jobs SET status = ?, exit_code = ?, started_at = COALESCE(?, started_at), ended_at = COALESCE(?, ended_at) WHERE id = ?`,
string(status), exitCode, startedAt, endedAt, id)
if err != nil {
return fmt.Errorf("update job: %w", err)
}
return nil
}
func scanJob(s scanner) (*model.Job, error) {
var (
j model.Job
status string
startedAt sql.NullTime
endedAt sql.NullTime
)
err := s.Scan(&j.ID, &j.Name, &j.Spec, &status, &j.ExitCode, &j.CreatedAt, &startedAt, &endedAt)
if err == sql.ErrNoRows {
return nil, ErrNotFound
}
if err != nil {
return nil, fmt.Errorf("scan job: %w", err)
}
j.Status = model.JobStatus(status)
if startedAt.Valid {
j.StartedAt = &startedAt.Time
}
if endedAt.Valid {
j.EndedAt = &endedAt.Time
}
return &j, nil
}
type TaskRepo struct {
db *sql.DB
}
func NewTaskRepo(db *sql.DB) *TaskRepo {
return &TaskRepo{db: db}
}
func (r *TaskRepo) Insert(ctx context.Context, t *model.Task) error {
if t.CreatedAt.IsZero() {
t.CreatedAt = time.Now().UTC()
}
if t.Status == "" {
t.Status = model.TaskStatusPending
}
argsJSON, _ := json.Marshal(t.Args)
envJSON, _ := json.Marshal(t.Env)
_, err := r.db.ExecContext(ctx,
`INSERT INTO tasks (id, job_id, command, args, env, pid, exit_code, status, created_at, stdout, stderr)
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
t.ID, t.JobID, t.Command, string(argsJSON), string(envJSON),
t.PID, t.ExitCode, string(t.Status), t.CreatedAt, t.Stdout, t.Stderr)
if err != nil {
return fmt.Errorf("insert task: %w", err)
}
return nil
}
func (r *TaskRepo) Get(ctx context.Context, id string) (*model.Task, error) {
row := r.db.QueryRowContext(ctx,
`SELECT id, job_id, command, args, env, pid, exit_code, status, created_at, started_at, ended_at, stdout, stderr FROM tasks WHERE id = ?`, id)
return scanTask(row)
}
func (r *TaskRepo) ListByJob(ctx context.Context, jobID string) ([]*model.Task, error) {
rows, err := r.db.QueryContext(ctx,
`SELECT id, job_id, command, args, env, pid, exit_code, status, created_at, started_at, ended_at, stdout, stderr FROM tasks WHERE job_id = ? ORDER BY created_at ASC`, jobID)
if err != nil {
return nil, fmt.Errorf("list tasks: %w", err)
}
defer rows.Close()
var tasks []*model.Task
for rows.Next() {
t, err := scanTask(rows)
if err != nil {
return nil, err
}
tasks = append(tasks, t)
}
return tasks, rows.Err()
}
func (r *TaskRepo) UpdateRunning(ctx context.Context, id string, pid int) error {
now := time.Now().UTC()
_, err := r.db.ExecContext(ctx,
`UPDATE tasks SET pid = ?, status = ?, started_at = ? WHERE id = ?`,
pid, string(model.TaskStatusRunning), now, id)
if err != nil {
return fmt.Errorf("update task running: %w", err)
}
return nil
}
func (r *TaskRepo) UpdateDone(ctx context.Context, id string, exitCode int, stdout, stderr string) error {
now := time.Now().UTC()
status := model.TaskStatusComplete
if exitCode != 0 {
status = model.TaskStatusFailed
}
_, err := r.db.ExecContext(ctx,
`UPDATE tasks SET status = ?, exit_code = ?, ended_at = ?, stdout = ?, stderr = ? WHERE id = ?`,
string(status), exitCode, now, stdout, stderr, id)
if err != nil {
return fmt.Errorf("update task done: %w", err)
}
return nil
}
func (r *TaskRepo) UpdateKilled(ctx context.Context, id string) error {
now := time.Now().UTC()
_, err := r.db.ExecContext(ctx,
`UPDATE tasks SET status = ?, ended_at = ? WHERE id = ?`,
string(model.TaskStatusKilled), now, id)
if err != nil {
return fmt.Errorf("update task killed: %w", err)
}
return nil
}
// ListRecent returns up to limit tasks ordered by created_at DESC.
// Used by the API to expose recent activity without a job filter.
func (r *TaskRepo) ListRecent(ctx context.Context, limit int) ([]*model.Task, error) {
if limit <= 0 {
limit = 100
}
rows, err := r.db.QueryContext(ctx,
`SELECT id, job_id, command, args, env, pid, exit_code, status, created_at, started_at, ended_at, stdout, stderr FROM tasks ORDER BY created_at DESC LIMIT ?`, limit)
if err != nil {
return nil, fmt.Errorf("list tasks recent: %w", err)
}
defer rows.Close()
var tasks []*model.Task
for rows.Next() {
t, err := scanTask(rows)
if err != nil {
return nil, err
}
tasks = append(tasks, t)
}
return tasks, rows.Err()
}
var _ = errors.New
var _ = json.Marshal
func scanTask(s scanner) (*model.Task, error) {
var (
t model.Task
status string
argsJSON string
envJSON string
startedAt sql.NullTime
endedAt sql.NullTime
)
err := s.Scan(&t.ID, &t.JobID, &t.Command, &argsJSON, &envJSON,
&t.PID, &t.ExitCode, &status, &t.CreatedAt, &startedAt, &endedAt, &t.Stdout, &t.Stderr)
if err == sql.ErrNoRows {
return nil, ErrNotFound
}
if err != nil {
return nil, fmt.Errorf("scan task: %w", err)
}
t.Status = model.TaskStatus(status)
if startedAt.Valid {
t.StartedAt = &startedAt.Time
}
if endedAt.Valid {
t.EndedAt = &endedAt.Time
}
_ = json.Unmarshal([]byte(argsJSON), &t.Args)
_ = json.Unmarshal([]byte(envJSON), &t.Env)
return &t, nil
}
+53
View File
@@ -0,0 +1,53 @@
package store
import (
"context"
"database/sql"
"embed"
"fmt"
"sort"
"strings"
)
//go:embed migrations/*.sql
var migrationsFS embed.FS
func migrate(db *sql.DB) error {
entries, err := migrationsFS.ReadDir("migrations")
if err != nil {
return fmt.Errorf("read migrations dir: %w", err)
}
names := make([]string, 0, len(entries))
for _, e := range entries {
if !e.IsDir() && strings.HasSuffix(e.Name(), ".sql") {
names = append(names, e.Name())
}
}
sort.Strings(names)
if _, err := db.ExecContext(context.Background(), `CREATE TABLE IF NOT EXISTS schema_migrations (name TEXT PRIMARY KEY, applied_at DATETIME NOT NULL)`); err != nil {
return fmt.Errorf("create schema_migrations: %w", err)
}
for _, name := range names {
var existing string
err := db.QueryRowContext(context.Background(), `SELECT name FROM schema_migrations WHERE name = ?`, name).Scan(&existing)
if err == nil {
continue
}
if err != sql.ErrNoRows {
return fmt.Errorf("check migration %s: %w", name, err)
}
sqlBytes, err := migrationsFS.ReadFile("migrations/" + name)
if err != nil {
return fmt.Errorf("read migration %s: %w", name, err)
}
if _, err := db.ExecContext(context.Background(), string(sqlBytes)); err != nil {
return fmt.Errorf("apply migration %s: %w", name, err)
}
if _, err := db.ExecContext(context.Background(), `INSERT INTO schema_migrations (name, applied_at) VALUES (?, datetime('now'))`, name); err != nil {
return fmt.Errorf("record migration %s: %w", name, err)
}
}
return nil
}
+13
View File
@@ -0,0 +1,13 @@
-- Node registry
CREATE TABLE IF NOT EXISTS nodes (
id TEXT PRIMARY KEY,
name TEXT NOT NULL,
address TEXT NOT NULL,
state TEXT NOT NULL DEFAULT 'pending',
joined_at DATETIME NOT NULL,
last_seen DATETIME NOT NULL,
metadata TEXT
);
CREATE INDEX IF NOT EXISTS idx_nodes_state ON nodes(state);
CREATE INDEX IF NOT EXISTS idx_nodes_name ON nodes(name);
@@ -0,0 +1,34 @@
-- Jobs and tasks
CREATE TABLE IF NOT EXISTS jobs (
id TEXT PRIMARY KEY,
name TEXT NOT NULL,
spec TEXT NOT NULL,
status TEXT NOT NULL DEFAULT 'pending',
exit_code INTEGER NOT NULL DEFAULT 0,
created_at DATETIME NOT NULL,
started_at DATETIME,
ended_at DATETIME
);
CREATE INDEX IF NOT EXISTS idx_jobs_status ON jobs(status);
CREATE INDEX IF NOT EXISTS idx_jobs_created ON jobs(created_at);
CREATE TABLE IF NOT EXISTS tasks (
id TEXT PRIMARY KEY,
job_id TEXT NOT NULL,
command TEXT NOT NULL,
args TEXT NOT NULL DEFAULT '[]',
env TEXT NOT NULL DEFAULT '[]',
pid INTEGER NOT NULL DEFAULT 0,
exit_code INTEGER NOT NULL DEFAULT 0,
status TEXT NOT NULL DEFAULT 'pending',
created_at DATETIME NOT NULL,
started_at DATETIME,
ended_at DATETIME,
stdout TEXT NOT NULL DEFAULT '',
stderr TEXT NOT NULL DEFAULT '',
FOREIGN KEY (job_id) REFERENCES jobs(id) ON DELETE CASCADE
);
CREATE INDEX IF NOT EXISTS idx_tasks_job ON tasks(job_id);
CREATE INDEX IF NOT EXISTS idx_tasks_status ON tasks(status);
@@ -0,0 +1,15 @@
-- Audit log for security-first observability
CREATE TABLE IF NOT EXISTS audit_log (
id INTEGER PRIMARY KEY AUTOINCREMENT,
timestamp DATETIME NOT NULL,
actor TEXT NOT NULL DEFAULT 'system',
action TEXT NOT NULL,
resource TEXT NOT NULL,
result TEXT NOT NULL,
error TEXT,
metadata TEXT
);
CREATE INDEX IF NOT EXISTS idx_audit_timestamp ON audit_log(timestamp);
CREATE INDEX IF NOT EXISTS idx_audit_action ON audit_log(action);
CREATE INDEX IF NOT EXISTS idx_audit_resource ON audit_log(resource);
+122
View File
@@ -0,0 +1,122 @@
package store
import (
"context"
"database/sql"
"encoding/json"
"errors"
"fmt"
"time"
"git.cloudinit.dev/coreci/orca/internal/model"
)
var ErrNotFound = errors.New("not found")
type NodeRepo struct {
db *sql.DB
}
func NewNodeRepo(db *sql.DB) *NodeRepo {
return &NodeRepo{db: db}
}
func (r *NodeRepo) Insert(ctx context.Context, n *model.Node) error {
if n.JoinedAt.IsZero() {
n.JoinedAt = time.Now().UTC()
}
if n.LastSeen.IsZero() {
n.LastSeen = n.JoinedAt
}
if n.State == "" {
n.State = model.NodeStateReady
}
metaJSON, err := json.Marshal(n.Metadata)
if err != nil {
return fmt.Errorf("marshal metadata: %w", err)
}
_, err = r.db.ExecContext(ctx,
`INSERT INTO nodes (id, name, address, state, joined_at, last_seen, metadata) VALUES (?, ?, ?, ?, ?, ?, ?)`,
n.ID, n.Name, n.Address, string(n.State), n.JoinedAt, n.LastSeen, string(metaJSON))
if err != nil {
return fmt.Errorf("insert node: %w", err)
}
return nil
}
func (r *NodeRepo) Get(ctx context.Context, id string) (*model.Node, error) {
row := r.db.QueryRowContext(ctx,
`SELECT id, name, address, state, joined_at, last_seen, metadata FROM nodes WHERE id = ?`, id)
return scanNode(row)
}
func (r *NodeRepo) List(ctx context.Context) ([]*model.Node, error) {
rows, err := r.db.QueryContext(ctx,
`SELECT id, name, address, state, joined_at, last_seen, metadata FROM nodes ORDER BY joined_at ASC`)
if err != nil {
return nil, fmt.Errorf("list nodes: %w", err)
}
defer rows.Close()
var nodes []*model.Node
for rows.Next() {
n, err := scanNode(rows)
if err != nil {
return nil, err
}
nodes = append(nodes, n)
}
return nodes, rows.Err()
}
func (r *NodeRepo) UpdateState(ctx context.Context, id string, state model.NodeState) error {
res, err := r.db.ExecContext(ctx,
`UPDATE nodes SET state = ?, last_seen = ? WHERE id = ?`,
string(state), time.Now().UTC(), id)
if err != nil {
return fmt.Errorf("update node state: %w", err)
}
rows, _ := res.RowsAffected()
if rows == 0 {
return ErrNotFound
}
return nil
}
func (r *NodeRepo) Delete(ctx context.Context, id string) error {
res, err := r.db.ExecContext(ctx, `DELETE FROM nodes WHERE id = ?`, id)
if err != nil {
return fmt.Errorf("delete node: %w", err)
}
rows, _ := res.RowsAffected()
if rows == 0 {
return ErrNotFound
}
return nil
}
type scanner interface {
Scan(dest ...any) error
}
func scanNode(s scanner) (*model.Node, error) {
var (
n model.Node
state string
metaJSON sql.NullString
)
err := s.Scan(&n.ID, &n.Name, &n.Address, &state, &n.JoinedAt, &n.LastSeen, &metaJSON)
if err == sql.ErrNoRows {
return nil, ErrNotFound
}
if err != nil {
return nil, fmt.Errorf("scan node: %w", err)
}
n.State = model.NodeState(state)
if metaJSON.Valid && metaJSON.String != "" {
if err := json.Unmarshal([]byte(metaJSON.String), &n.Metadata); err != nil {
return nil, fmt.Errorf("unmarshal metadata: %w", err)
}
}
return &n, nil
}
+102
View File
@@ -0,0 +1,102 @@
package store
import (
"context"
"path/filepath"
"testing"
"time"
"git.cloudinit.dev/coreci/orca/internal/model"
)
func openTestDB(t *testing.T) (*NodeRepo, func()) {
t.Helper()
path := filepath.Join(t.TempDir(), "test.db")
db, err := Open(path)
if err != nil {
t.Fatalf("open db: %v", err)
}
return NewNodeRepo(db), func() { _ = db.Close() }
}
func TestNodeRepo_InsertAndGet(t *testing.T) {
repo, cleanup := openTestDB(t)
defer cleanup()
ctx := context.Background()
n := &model.Node{
ID: "test-id-1",
Name: "alpha",
Address: "localhost:8443",
State: model.NodeStateReady,
JoinedAt: time.Now().UTC(),
LastSeen: time.Now().UTC(),
}
if err := repo.Insert(ctx, n); err != nil {
t.Fatalf("insert: %v", err)
}
got, err := repo.Get(ctx, "test-id-1")
if err != nil {
t.Fatalf("get: %v", err)
}
if got.Name != "alpha" || got.Address != "localhost:8443" {
t.Errorf("unexpected node: %+v", got)
}
if got.State != model.NodeStateReady {
t.Errorf("expected state ready, got %s", got.State)
}
}
func TestNodeRepo_List(t *testing.T) {
repo, cleanup := openTestDB(t)
defer cleanup()
ctx := context.Background()
for _, name := range []string{"a", "b", "c"} {
_ = repo.Insert(ctx, &model.Node{
ID: name, Name: name, Address: "addr",
JoinedAt: time.Now().UTC(), LastSeen: time.Now().UTC(),
})
}
nodes, err := repo.List(ctx)
if err != nil {
t.Fatalf("list: %v", err)
}
if len(nodes) != 3 {
t.Errorf("expected 3 nodes, got %d", len(nodes))
}
}
func TestNodeRepo_UpdateState(t *testing.T) {
repo, cleanup := openTestDB(t)
defer cleanup()
ctx := context.Background()
_ = repo.Insert(ctx, &model.Node{
ID: "x", Name: "x", Address: "a", JoinedAt: time.Now().UTC(), LastSeen: time.Now().UTC(),
})
if err := repo.UpdateState(ctx, "x", model.NodeStateLeft); err != nil {
t.Fatalf("update: %v", err)
}
got, _ := repo.Get(ctx, "x")
if got.State != model.NodeStateLeft {
t.Errorf("expected left, got %s", got.State)
}
}
func TestNodeRepo_Delete(t *testing.T) {
repo, cleanup := openTestDB(t)
defer cleanup()
ctx := context.Background()
_ = repo.Insert(ctx, &model.Node{
ID: "y", Name: "y", Address: "a", JoinedAt: time.Now().UTC(), LastSeen: time.Now().UTC(),
})
if err := repo.Delete(ctx, "y"); err != nil {
t.Fatalf("delete: %v", err)
}
_, err := repo.Get(ctx, "y")
if err != ErrNotFound {
t.Errorf("expected ErrNotFound, got %v", err)
}
}
+36
View File
@@ -0,0 +1,36 @@
package store
import (
"database/sql"
"fmt"
"os"
"path/filepath"
_ "modernc.org/sqlite"
)
func Open(path string) (*sql.DB, error) {
if path == "" {
home, err := os.UserHomeDir()
if err != nil {
return nil, fmt.Errorf("get home dir: %w", err)
}
path = filepath.Join(home, ".orca", "orca.db")
}
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
return nil, fmt.Errorf("create db dir: %w", err)
}
db, err := sql.Open("sqlite", path+"?_pragma=journal_mode(WAL)&_pragma=foreign_keys(ON)")
if err != nil {
return nil, fmt.Errorf("open sqlite: %w", err)
}
if err := db.Ping(); err != nil {
_ = db.Close()
return nil, fmt.Errorf("ping sqlite: %w", err)
}
if err := migrate(db); err != nil {
_ = db.Close()
return nil, fmt.Errorf("migrate: %w", err)
}
return db, nil
}