Ensure that we embed the version in the schema before packaging to the binary (#86)

Author: stack72

.gitignore

@@ -13,6 +13,7 @@ ci-scripts
 sdk/dotnet/version.txt
 .DS_Store
 /provider/cmd/pulumi-resource-command/schema.go
+/provider/cmd/pulumi-resource-command/schema-embed.json
 dist/
 
 rsa*

provider/cmd/pulumi-resource-command/generate.go

@@ -13,25 +13,50 @@
 // limitations under the License.
 
 //go:build ignore
-// +build ignore
 
 package main
 
 import (
-	"fmt"
+	"encoding/json"
+	"errors"
+	"io/fs"
 	"io/ioutil"
 	"log"
+	"os"
+
+	"github.com/pulumi/pulumi/pkg/v3/codegen/schema"
 )
 
 func main() {
+	version, found := os.LookupEnv("VERSION")
+	if !found {
+		log.Fatal("version not found")
+	}
+
 	schemaContents, err := ioutil.ReadFile("./schema.json")
 	if err != nil {
 		log.Fatal(err)
 	}
 
-	err = ioutil.WriteFile("./schema.go", []byte(fmt.Sprintf(`package main
-var pulumiSchema = %#v
-`, schemaContents)), 0600)
+	var packageSpec schema.PackageSpec
+	err = json.Unmarshal(schemaContents, &packageSpec)
+	if err != nil {
+		log.Fatalf("cannot deserialize schema: %v", err)
+	}
+
+	packageSpec.Version = version
+	versionedContents, err := json.Marshal(packageSpec)
+	if err != nil {
+		log.Fatalf("cannot reserialize schema: %v", err)
+	}
+
+	// Clean up schema.go as it may be present & gitignored and tolerate an error if the file isn't present.
+	err = os.Remove("./schema.go")
+	if err != nil && !errors.Is(err, fs.ErrNotExist) {
+		log.Fatal(err)
+	}
+
+	err = ioutil.WriteFile("./schema-embed.json", versionedContents, 0600)
 	if err != nil {
 		log.Fatal(err)
 	}

provider/cmd/pulumi-resource-command/main.go

@@ -17,12 +17,15 @@
 package main
 
 import (
+	_ "embed"
+
 	"github.com/pulumi/pulumi-command/provider/pkg/provider"
 	"github.com/pulumi/pulumi-command/provider/pkg/version"
 )
 
-var providerName = "command"
+//go:embed schema-embed.json
+var pulumiSchema []byte
 
 func main() {
-	provider.Serve(providerName, version.Version, pulumiSchema)
+	provider.Serve("command", version.Version, pulumiSchema)
 }

provider/go.mod

@@ -1,6 +1,6 @@
 module github.com/pulumi/pulumi-command/provider
 
-go 1.17
+go 1.18
 
 require (
 	github.com/golang/protobuf v1.5.2