-
Notifications
You must be signed in to change notification settings - Fork 9
/
handler.go
66 lines (53 loc) · 1.83 KB
/
handler.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
// 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 parsing
import (
"context"
"google.golang.org/protobuf/proto"
"google.golang.org/protobuf/types/known/anypb"
"namespacelabs.dev/foundation/internal/fnerrors"
"namespacelabs.dev/foundation/schema"
"namespacelabs.dev/foundation/std/pkggraph"
)
type ServerFrameworkExt struct {
Include []schema.PackageName
FrameworkSpecific *anypb.Any
}
type ServerInputs struct {
Services []*schema.GrpcExportService
}
// XXX we're injection Location in these, which allows for loading arbitrary files for the workspace;
// Ideally we'd pass a PackageLoader instead.
type FrameworkHandler interface {
PreParseServer(context.Context, pkggraph.Location, *ServerFrameworkExt) error
PostParseServer(context.Context, *Sealed) error
// List of packages that should be added as server dependencies, when the target environment's purpose is DEVELOPMENT.
DevelopmentPackages() []schema.PackageName
}
type CueService struct {
ProtoTypename string `json:"protoTypename"`
GoPackage string `json:"goPackage"`
}
var (
FrameworkHandlers = map[schema.Framework]FrameworkHandler{}
)
func RegisterFrameworkHandler(framework schema.Framework, handler FrameworkHandler) {
FrameworkHandlers[framework] = handler
}
func GetExtension(extensions []*anypb.Any, msg proto.Message) bool {
for _, ext := range extensions {
if ext.MessageIs(msg) {
if ext.UnmarshalTo(msg) == nil {
return true
}
}
}
return false
}
func MustExtension(extensions []*anypb.Any, msg proto.Message) error {
if !GetExtension(extensions, msg) {
return fnerrors.InternalError("didn't find required extension: %s", msg.ProtoReflect().Descriptor().FullName())
}
return nil
}