buildserver.go

// Copyright 2022 Namespace Labs Inc; All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.

package deploy

import (
	"context"
	"fmt"
	"io/fs"

	"namespacelabs.dev/foundation/internal/artifacts/oci"
	"namespacelabs.dev/foundation/internal/build"
	"namespacelabs.dev/foundation/internal/compute"
	"namespacelabs.dev/foundation/internal/console"
	"namespacelabs.dev/foundation/internal/fnfs/memfs"
	"namespacelabs.dev/foundation/internal/fnfs/workspace/wsremote"
	"namespacelabs.dev/foundation/internal/hotreload"
	"namespacelabs.dev/foundation/internal/integrations"
	"namespacelabs.dev/foundation/internal/parsing/platform"
	"namespacelabs.dev/foundation/internal/planning"
	"namespacelabs.dev/foundation/internal/planning/config"
	"namespacelabs.dev/foundation/internal/runtime"
	"namespacelabs.dev/foundation/schema"
	"namespacelabs.dev/foundation/schema/storage"
	"namespacelabs.dev/foundation/std/cfg"
	"namespacelabs.dev/foundation/std/tasks"
)

var RunCodegen = true

func MakeBuildPlan(ctx context.Context, rc runtime.Planner, server planning.PlannedServer, focused bool, spec build.Spec) (build.Plan, error) {
	return tasks.Return(ctx, tasks.Action("planning.prepare-server-image").Scope(server.PackageName()),
		func(ctx context.Context) (build.Plan, error) {
			platforms, err := rc.TargetPlatforms(ctx)
			if err != nil {
				return build.Plan{}, err
			}

			tasks.Attachments(ctx).AddResult("platforms", platform.FormatPlatforms(platforms))

			var ws build.Workspace = server.Module()

			remote := wsremote.Ctx(ctx)

			// If there's no sink, we don't need to keep a full copy of the files.
			digestMode := remote == nil

			fmt.Fprintf(console.Debug(ctx), "prepare-server-image: %s: remoteSink=%v digestMode=%v focused=%v external=%v\n",
				server.PackageName(), remote != nil, digestMode, focused, server.Module().IsExternal())

			observeChanges := focused && !server.Module().IsExternal()

			opts := integrations.IntegrationFor(server.Framework()).PrepareHotReload(ctx, remote, server)
			fmt.Fprintf(console.Debug(ctx), "prepare-server-image: %s: framework=%v opts=%v\n",
				server.PackageName(), server.Framework(), opts != nil)

			if opts == nil {
				opts = &integrations.HotReloadOpts{}
			}

			return build.Plan{
				SourceLabel:   fmt.Sprintf("Server %s", server.PackageName()),
				SourcePackage: server.PackageName(),
				BuildKind:     storage.Build_SERVER,
				Spec:          spec,
				Workspace:     hotreload.NewDevModule(ws, observeChanges, digestMode, *opts, &codegenTrigger{srv: server.Server}),
				Platforms:     platforms,
			}, nil
		})
}

type prepareServerConfig struct {
	planner         runtime.Planner
	serverPackage   schema.PackageName
	env             cfg.Context
	stack           *schema.Stack
	computedConfigs compute.Computable[*schema.ComputedConfigurations]

	compute.LocalScoped[fs.FS]
}

func (c *prepareServerConfig) Inputs() *compute.In {
	return compute.Inputs().
		Indigestible("planner", c.planner).
		JSON("serverPackage", c.serverPackage).
		Indigestible("env", c.env).
		Proto("stack", c.stack).
		Computable("computedConfigs", c.computedConfigs)
}

func (c *prepareServerConfig) Action() *tasks.ActionEvent {
	return tasks.Action("deploy.prepare-server-config").Arg("env", c.env.Environment().Name).Scope(c.serverPackage)
}

func (c *prepareServerConfig) Compute(ctx context.Context, deps compute.Resolved) (fs.FS, error) {
	var fragment []*schema.IngressFragment
	for _, entry := range c.stack.Entry {
		var err error
		fragment, err = runtime.ComputeIngress(ctx, c.env, c.planner, entry, c.stack.Endpoint)
		if err != nil {
			return nil, err
		}
	}

	// XXX These should be scoped down to only the configs provided by this server.
	computedConfigs := compute.MustGetDepValue(deps, c.computedConfigs, "computedConfigs")

	files := &memfs.FS{}
	if err := (config.DehydrateOpts{IncludeTextProto: true}).DehydrateTo(ctx, c.env.Environment(), c.stack, fragment, computedConfigs, files); err != nil {
		return nil, err
	}

	return files, nil
}

func prepareConfigImage(ctx context.Context, planner planning.Planner, server planning.Server, stack *planning.Stack,
	computedConfigs compute.Computable[*schema.ComputedConfigurations]) oci.NamedImage {

	return oci.MakeImageFromScratch(fmt.Sprintf("config %s", server.PackageName()),
		oci.MakeLayer(fmt.Sprintf("config %s", server.PackageName()),
			&prepareServerConfig{
				planner:         planner.Runtime,
				serverPackage:   server.PackageName(),
				env:             planner.Context,
				stack:           stack.Proto(),
				computedConfigs: computedConfigs,
			}))
}