-
Notifications
You must be signed in to change notification settings - Fork 41
/
main.go
76 lines (63 loc) · 2.21 KB
/
main.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
// Copyright 2016-2018, Pulumi Corporation.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package tfbridge
import (
"context"
"encoding/json"
"flag"
"fmt"
"io"
"os"
"github.com/pulumi/pulumi/sdk/v3/go/common/util/cmdutil"
"github.com/pulumi/pulumi/sdk/v3/go/common/util/contract"
)
// Main launches the tfbridge plugin for a given package pkg and provider prov.
func Main(pkg string, version string, prov ProviderInfo, pulumiSchema []byte) {
ctx := context.Background()
// Look for a request to dump the provider info to stdout.
flags := flag.NewFlagSet("tf-provider-flags", flag.ContinueOnError)
// Discard print output by default; there might be flags such
// as -tracing that are unrecognized at this phase but will be
// parsed later by `Serve`. We do not want to print errors
// about them. Save `defaultOutput` for help below.
defaultOutput := flags.Output()
flags.SetOutput(io.Discard)
dumpInfo := flags.Bool("get-provider-info", false, "dump provider info as JSON to stdout")
providerVersion := flags.Bool("version", false, "get built provider version")
err := flags.Parse(os.Args[1:])
contract.IgnoreError(err)
// Ensure we do print help message when `--help` is requested.
if err == flag.ErrHelp {
flags.SetOutput(defaultOutput)
err := flags.Parse(os.Args[1:])
if err != nil {
cmdutil.ExitError(err.Error())
}
}
if *dumpInfo {
if err := json.NewEncoder(os.Stdout).Encode(MarshalProviderInfo(&prov)); err != nil {
cmdutil.ExitError(err.Error())
}
os.Exit(0)
}
if *providerVersion {
fmt.Println(version)
os.Exit(0)
}
// Initialize Terraform logging.
prov.P.InitLogging(ctx)
if err := Serve(pkg, version, prov, pulumiSchema); err != nil {
cmdutil.ExitError(err.Error())
}
}