Scaffold Go nightshiftd daemon

.gitignore

@@ -7,3 +7,8 @@ env/
 build/
 dist/
 infra/.deploy-state
+loadtest/results
+.DS_Store
+
+# Go
+nightshiftd/nightshiftd

nightshiftd/Makefile

@@ -0,0 +1,19 @@
+.PHONY: proto build test clean lint debug
+
+proto:
+	buf generate
+
+debug:
+	go build -gcflags="all=-N -l" -o nightshiftd ./cmd/nightshiftd
+
+build:
+	go build -o nightshiftd ./cmd/nightshiftd
+
+test:
+	go test ./...
+
+clean:
+	rm -f nightshiftd
+
+lint:
+	go vet ./...

nightshiftd/buf.gen.yaml

@@ -0,0 +1,8 @@
+version: v2
+plugins:
+  - remote: buf.build/protocolbuffers/go
+    out: gen
+    opt: paths=source_relative
+  - remote: buf.build/grpc/go
+    out: gen
+    opt: paths=source_relative

nightshiftd/buf.yaml

@@ -0,0 +1,11 @@
+version: v2
+modules:
+  - path: proto
+deps:
+  - buf.build/googleapis/googleapis
+lint:
+  use:
+    - STANDARD
+breaking:
+  use:
+    - FILE

nightshiftd/cmd/nightshiftd/main.go

@@ -0,0 +1,114 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log/slog"
+	"net"
+	"os"
+	"os/signal"
+	"path/filepath"
+	"syscall"
+
+	"github.com/nightshiftco/nightshift/nightshiftd/internal/config"
+	"github.com/nightshiftco/nightshift/nightshiftd/internal/db"
+	"github.com/nightshiftco/nightshift/nightshiftd/internal/server"
+	"google.golang.org/grpc"
+)
+
+func main() {
+	configPath := flag.String("config", "/etc/nightshift/config.toml", "path to config file")
+	flag.Parse()
+
+	if err := run(*configPath); err != nil {
+		fmt.Fprintf(os.Stderr, "nightshiftd: %v\n", err)
+		os.Exit(1)
+	}
+}
+
+func run(configPath string) error {
+	cfg, err := config.Load(configPath)
+	if err != nil {
+		return fmt.Errorf("load config: %w", err)
+	}
+
+	logger := setupLogger(cfg.Log)
+	logger.Info("starting nightshiftd", "config", configPath)
+
+	// state directory
+	if err := os.MkdirAll(cfg.Daemon.StateDir, 0750); err != nil {
+		return fmt.Errorf("create state_dir: %w", err)
+	}
+
+	// opn db and run migrations
+	dbPath := filepath.Join(cfg.Daemon.StateDir, "nightshift.db")
+	database, err := db.Open(dbPath)
+	if err != nil {
+		return fmt.Errorf("open database: %w", err)
+	}
+	defer database.Close()
+	logger.Info("database ready", "path", dbPath)
+
+	// gRPC server and register service
+	gs := grpc.NewServer()
+	srv := server.New(database, logger)
+	srv.Register(gs)
+
+	socketPath := cfg.Daemon.Socket
+	if err := os.MkdirAll(filepath.Dir(socketPath), 0755); err != nil {
+		return fmt.Errorf("create socket dir: %w", err)
+	}
+
+	// Remove stale socket file
+	if err := os.Remove(socketPath); err != nil && !os.IsNotExist(err) {
+		return fmt.Errorf("remove stale socket: %w", err)
+	}
+
+	lis, err := net.Listen("unix", socketPath)
+	if err != nil {
+		return fmt.Errorf("listen on %s: %w", socketPath, err)
+	}
+	logger.Info("listening", "socket", socketPath)
+
+	// shutdown signals
+	sigCh := make(chan os.Signal, 1)
+	signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
+
+	go func() {
+		sig := <-sigCh
+		logger.Info("received signal, shutting down", "signal", sig)
+		gs.GracefulStop()
+	}()
+
+	if err := gs.Serve(lis); err != nil {
+		return fmt.Errorf("serve: %w", err)
+	}
+
+	logger.Info("nightshiftd stopped")
+	return nil
+}
+
+func setupLogger(cfg config.LogConfig) *slog.Logger {
+	var level slog.Level
+	switch cfg.Level {
+	case "debug":
+		level = slog.LevelDebug
+	case "warn":
+		level = slog.LevelWarn
+	case "error":
+		level = slog.LevelError
+	default:
+		level = slog.LevelInfo
+	}
+
+	opts := &slog.HandlerOptions{Level: level}
+
+	var handler slog.Handler
+	if cfg.Format == "json" {
+		handler = slog.NewJSONHandler(os.Stderr, opts)
+	} else {
+		handler = slog.NewTextHandler(os.Stderr, opts)
+	}
+
+	return slog.New(handler)
+}

nightshiftd/config.dev.toml

@@ -0,0 +1,32 @@
+version = 1
+
+[daemon]
+state_dir = "/tmp/nightshiftd"
+socket = "/tmp/nightshiftd.sock"
+
+[pool]
+idle_timeout = "5m"
+default_max_per_agent = 3
+
+[containerd]
+address = "/run/containerd/containerd.sock"
+namespace = "nightshift"
+
+[runtime]
+default = "auto"
+
+[runtime.firecracker]
+name = "aws.firecracker"
+
+[runtime.runc]
+name = "io.containerd.runc.v2"
+
+[network]
+cni_name = "nightshift"
+
+[images]
+base = "nightshift/agent-runtime:latest"
+
+[log]
+level = "debug"
+format = "text"