Initial commit: DSH Web Manager

internal/api/intents.go

@@ -0,0 +1,212 @@
+package api
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"io"
+	"net/http"
+	"strings"
+	"time"
+
+	"git.misaka.ren/m1saka/dsh-web/internal/store"
+)
+
+func (a *API) intents(w http.ResponseWriter, r *http.Request) {
+	u, _, err := a.user(r)
+	if err != nil {
+		fail(w, 401, "authentication required")
+		return
+	}
+	if err = a.s.EnsureIntents(); err != nil {
+		fail(w, 500, "database error")
+		return
+	}
+	switch r.Method {
+	case http.MethodGet:
+		list, err := a.s.Intents(u.ID)
+		if err != nil {
+			fail(w, 500, "database error")
+			return
+		}
+		writeJSON(w, list)
+	case http.MethodPost:
+		var in struct {
+			Name    string `json:"name"`
+			Version string `json:"version"`
+		}
+		dec := json.NewDecoder(http.MaxBytesReader(w, r.Body, 4096))
+		dec.DisallowUnknownFields()
+		if dec.Decode(&in) != nil || dec.Decode(new(any)) != io.EOF || strings.TrimSpace(in.Name) == "" || len(in.Name) > 200 || !validBuildVersion(in.Version) {
+			fail(w, 400, "name and fixed version required")
+			return
+		}
+		i := store.Intent{ID: store.ID(), UID: u.ID, Name: strings.TrimSpace(in.Name), Version: in.Version, Host: a.runtimeHost(r), Status: "waiting"}
+		if err = a.s.ReserveIntent(i); err != nil {
+			if errors.Is(err, store.ErrInstanceLimit) {
+				fail(w, 409, err.Error())
+			} else {
+				fail(w, 500, "database error")
+			}
+			return
+		}
+		w.Header().Set("Content-Type", "application/json")
+		w.WriteHeader(http.StatusAccepted)
+		writeJSON(w, i)
+	default:
+		fail(w, 405, "method not allowed")
+	}
+}
+func (a *API) intent(w http.ResponseWriter, r *http.Request) {
+	u, _, err := a.user(r)
+	if err != nil {
+		fail(w, 401, "authentication required")
+		return
+	}
+	if r.Method != http.MethodDelete {
+		fail(w, 405, "method not allowed")
+		return
+	}
+	if err = a.s.EnsureIntents(); err != nil {
+		fail(w, 500, "database error")
+		return
+	}
+	id := strings.TrimPrefix(r.URL.Path, "/api/intents/")
+	if err = a.s.CancelIntent(id, u.ID); err != nil {
+		if errors.Is(err, store.ErrIntentState) {
+			fail(w, 409, "intent not found or no longer waiting")
+		} else {
+			fail(w, 500, "database error")
+		}
+		return
+	}
+	writeJSON(w, map[string]string{"status": "cancelled"})
+}
+
+// Call once at startup, after the build queue has recovered interrupted jobs.
+func (a *API) StartIntents(ctx context.Context) error {
+	if err := a.s.EnsureIntents(); err != nil {
+		return err
+	}
+	if err := a.recoverIntents(ctx); err != nil {
+		return err
+	}
+	go func() {
+		ticker := time.NewTicker(time.Second)
+		defer ticker.Stop()
+		for {
+			select {
+			case <-ctx.Done():
+				return
+			default:
+			}
+			a.processIntents(ctx)
+			select {
+			case <-ctx.Done():
+				return
+			case <-ticker.C:
+			}
+		}
+	}()
+	return nil
+}
+
+func (a *API) finishIntent(id, status, detail string) {
+	_, _ = a.s.DB.Exec(`UPDATE creation_intents SET status=?,error=? WHERE id=? AND status IN ('waiting','creating')`, status, detail, id)
+}
+func (a *API) recoverIntents(ctx context.Context) error {
+	list, err := a.s.Intents("")
+	if err != nil {
+		return err
+	}
+	for _, in := range list {
+		if in.Status != "creating" {
+			continue
+		}
+		i, err := a.s.GetInstance(in.InstanceID)
+		if err == nil && i.Status == "running" {
+			a.finishIntent(in.ID, "success", "")
+			continue
+		}
+		cleanup, cancel := context.WithTimeout(ctx, 30*time.Second)
+		// Docker may have created the container before its ID reached SQLite.
+		err = a.removeRuntimeContainer(cleanup, runtimeNetwork(in.InstanceID))
+		if err == nil {
+			err = a.removeRuntimeNetwork(cleanup, in.InstanceID)
+		}
+		cancel()
+		if err != nil {
+			return err
+		}
+		a.s.DeleteInstance(in.InstanceID)
+		a.finishIntent(in.ID, "failed", "instance creation interrupted by process restart")
+	}
+	return nil
+}
+
+func (a *API) processIntents(ctx context.Context) {
+	list, err := a.s.Intents("")
+	if err != nil {
+		return
+	}
+	for _, in := range list {
+		if ctx.Err() != nil {
+			return
+		}
+		if in.Status != "waiting" {
+			continue
+		}
+		if in.BuildID == "" {
+			b, err := a.enqueueBuild(ctx, in.UID, in.Version)
+			if err != nil {
+				a.finishIntent(in.ID, "failed", "build submission failed")
+				continue
+			}
+			if _, err = a.s.DB.Exec(`UPDATE creation_intents SET build_id=? WHERE id=? AND status='waiting'`, b.ID, in.ID); err != nil {
+				continue
+			}
+			in.BuildID = b.ID
+		}
+		var status, imageID, version string
+		err := a.s.DB.QueryRow(`SELECT status,image_id,version FROM image_builds WHERE id=?`, in.BuildID).Scan(&status, &imageID, &version)
+		if err != nil {
+			continue
+		}
+		if version != in.Version || status == "failed" || status == "cancelled" || status == "missing" {
+			a.finishIntent(in.ID, "failed", "build failed; requested version was not created")
+			continue
+		}
+		if status != "success" {
+			continue
+		}
+		work, cancel := context.WithTimeout(ctx, 90*time.Second)
+		verified, err := a.verifiedImages(work)
+		if err != nil {
+			cancel()
+			continue
+		}
+		if !runtimeImageID.MatchString(imageID) || verified[imageID] != in.Version {
+			cancel()
+			a.finishIntent(in.ID, "failed", "verified image for requested version unavailable")
+			continue
+		}
+		u, err := a.s.UserByID(in.UID)
+		if err != nil {
+			cancel()
+			a.finishIntent(in.ID, "failed", "owner unavailable")
+			continue
+		}
+		i := store.Instance{ID: store.ID(), UID: in.UID, Name: in.Name, Version: in.Version, ImageID: imageID, Status: "creating", Created: time.Now()}
+		if err = a.s.ClaimIntent(in.ID, i); err != nil {
+			cancel()
+			continue
+		}
+		_, err = a.createRuntime(work, u, i, in.Host)
+		cancel()
+		if err != nil {
+			a.finishIntent(in.ID, "failed", err.Error())
+		} else {
+			a.finishIntent(in.ID, "success", "")
+		}
+	}
+}