-
Notifications
You must be signed in to change notification settings - Fork 9
/
selector.go
108 lines (86 loc) · 3.09 KB
/
selector.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
// 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 runtime
import (
"encoding/json"
"strings"
"github.com/tailscale/hujson"
"namespacelabs.dev/foundation/internal/fnerrors"
"namespacelabs.dev/foundation/schema"
runtimepb "namespacelabs.dev/foundation/schema/runtime"
)
func SelectService(rt *runtimepb.RuntimeConfig, ref *schema.ServiceRef) (*runtimepb.Server_Service, error) {
if ref == nil {
return nil, fnerrors.BadInputError("missing required service endpoint")
}
allServers := append(rt.StackEntry, rt.Current)
for _, srv := range allServers {
if srv.PackageName == ref.GetServerRef().GetPackageName() {
for _, service := range srv.Service {
if service.Name == ref.ServiceName {
return service, nil
}
}
return nil, fnerrors.BadInputError("the required service %q is not exported by %q",
ref.ServiceName, ref.GetServerRef().GetPackageName())
}
}
return nil, fnerrors.BadInputError("the required server %q is not present in the stack", ref.GetServerRef().GetPackageName())
}
func SelectServiceValue(rt *runtimepb.RuntimeConfig, ref *schema.ServiceRef, selector func(*runtimepb.Server_Service) (string, error)) (string, error) {
svc, err := SelectService(rt, ref)
if err != nil {
return "", err
}
return selector(svc)
}
func SelectServiceEndpoint(svc *runtimepb.Server_Service) (string, error) {
return svc.Endpoint, nil
}
func SelectServiceIngress(service *runtimepb.Server_Service) (string, error) {
if service.Ingress == nil || len(service.Ingress.Domain) == 0 {
return "", fnerrors.BadInputError("service %s has no ingress, %v", service.Name, service)
}
// TODO: introduce a concept of the "default" ingress, use it here.
return service.Ingress.Domain[0].BaseUrl, nil
}
func SelectInstance(rt *runtimepb.RuntimeConfig, instance *schema.ResolvableSource) (any, error) {
switch {
case instance.Service != nil:
return SelectService(rt, instance.Service)
case instance.SelectInternalEndpointByKind != "":
var matches []*runtimepb.Server_InternalEndpoint
for _, m := range rt.Current.GetInternalEndpoint() {
matching := false
for _, x := range m.Exported {
if x.Kind == instance.SelectInternalEndpointByKind {
matching = true
break
}
}
if matching {
matches = append(matches, m)
}
}
if len(matches) != 1 {
return nil, fnerrors.BadInputError("%s: expected 1 match, got %d", instance.SelectInternalEndpointByKind, len(matches))
}
return matches[0], nil
case instance.UntypedJson != nil:
contents := instance.UntypedJson.Contents
if strings.HasSuffix(instance.UntypedJson.Path, ".jsonc") {
var err error
contents, err = hujson.Standardize(instance.UntypedJson.Contents)
if err != nil {
return nil, err
}
}
var m any
if err := json.Unmarshal(contents, &m); err != nil {
return nil, fnerrors.BadInputError("instance: failed to parse json: %w", err)
}
return m, nil
}
return nil, fnerrors.BadInputError("instance: can't construct a value")
}