-
Notifications
You must be signed in to change notification settings - Fork 9
/
buildserver.go
129 lines (105 loc) · 4.58 KB
/
buildserver.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
// 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,
}))
}