-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
81 lines (67 loc) · 2.17 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
77
78
79
80
81
package main
import (
"context"
"flag"
"log"
"os"
"github.com/hashicorp/terraform-plugin-sdk/v2/plugin"
"github.com/threefoldtech/terraform-provider-grid/internal/provider"
"github.com/threefoldtech/terraform-provider-grid/pkg/state"
)
// Run "go generate" to format example terraform files and generate the docs for the registry/website
// If you do not have terraform installed, you can remove the formatting command, but its suggested to
// ensure the documentation is formatted properly.
//go:generate terraform fmt -recursive ./examples/
// Run the docs generation tool, check its repository for more information on how it works and how docs
// can be customized.
//go:generate go run github.com/hashicorp/terraform-plugin-docs/cmd/tfplugindocs
var (
// these will be set by the goreleaser configuration
// to appropriate values for the compiled binary
version string = "dev"
// goreleaser can also pass the specific commit if you want
// commit string = ""
)
func main() {
var debugMode bool
flag.BoolVar(&debugMode, "debug", false, "set to true to run the provider with support for debuggers like delve")
flag.Parse()
db, err := state.NewLocalStateDB(state.TypeFile)
if err != nil {
log.Fatal(err.Error())
}
err = db.Load()
if err != nil {
log.Fatal(err.Error())
}
network := determineSubstrateNetwork()
subext, err := provider.SubstrateVersion[network](provider.SUBSTRATE_URL[network]).SubstrateExt()
if err != nil {
log.Fatal(err)
}
defer subext.Close()
opts := &plugin.ServeOpts{ProviderFunc: provider.New(version, subext, db.GetState())}
if debugMode {
// TODO: update this string with the full name of your provider as used in your configs
err := plugin.Debug(context.Background(), "registry.terraform.io/hashicorp/scaffolding", opts)
if err != nil {
log.Fatal(err.Error())
}
return
}
plugin.Serve(opts)
err = db.Save()
if err != nil {
log.Fatal(err.Error())
}
}
func determineSubstrateNetwork() string {
network := os.Getenv("NETWORK")
if network == "" {
network = "dev"
}
if network != "dev" && network != "qa" && network != "test" && network != "main" {
log.Fatal("network must be one of dev, qa, test, or main")
}
return network
}