Compute schema-minimal.json

.gitignore

@@ -19,7 +19,7 @@ yarn.lock
 
 ci-scripts
 provider/shim/coverage.txt
-provider/**/schema-embed.json
+provider/**/*-embed.json
 **/version.txt
 **/nuget
 **/dist

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

@@ -19,45 +19,69 @@ package main
 import (
 	"encoding/json"
 	"errors"
+	"fmt"
 	"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")
-	}
+type compressAndVersionSchemaFileOptions struct {
+	sourceFile string
+	destFile   string
+	version    string
+}
 
-	schemaContents, err := ioutil.ReadFile("./schema.json")
+func compressAndVersionSchemaFile(opts compressAndVersionSchemaFileOptions) error {
+	schemaContents, err := os.ReadFile(opts.sourceFile)
 	if err != nil {
-		log.Fatal(err)
+		return fmt.Errorf("cannot read sourceFile: %w", err)
 	}
-
 	var packageSpec schema.PackageSpec
 	err = json.Unmarshal(schemaContents, &packageSpec)
 	if err != nil {
-		log.Fatalf("cannot deserialize schema: %v", err)
+		return fmt.Errorf("cannot deserialize schema: %w", err)
 	}
-
-	packageSpec.Version = version
+	packageSpec.Version = opts.version
 	versionedContents, err := json.Marshal(packageSpec)
 	if err != nil {
-		log.Fatalf("cannot reserialize schema: %v", err)
+		return fmt.Errorf("cannot reserialize schema: %w", err)
+	}
+	err = os.WriteFile(opts.destFile, versionedContents, 0600)
+	if err != nil {
+		return fmt.Errorf("cannot write destFile: %w", err)
 	}
+	return nil
+}
 
+func main() {
 	// 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")
+	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)
+	version, found := os.LookupEnv("VERSION")
+	if !found {
+		log.Fatal("version not found")
+	}
+
+	for _, opts := range []compressAndVersionSchemaFileOptions{
+		{
+			sourceFile: "schema.json",
+			destFile:   "schema-embed.json",
+			version:    version,
+		},
+		{
+			sourceFile: "schema-light.json",
+			destFile:   "schema-light-embed.json",
+			version:    version,
+		},
+	} {
+		err = compressAndVersionSchemaFile(opts)
+		if err != nil {
+			log.Fatal(err)
+		}
 	}
 }

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

@@ -19,16 +19,27 @@ package main
 import (
 	"context"
 	_ "embed"
+	"os"
 
 	aws "github.com/pulumi/pulumi-aws/provider/v6"
 	pf "github.com/pulumi/pulumi-terraform-bridge/pf/tfbridge"
+	"github.com/pulumi/pulumi/sdk/v3/go/common/util/cmdutil"
 )
 
 //go:embed schema-embed.json
 var pulumiSchema []byte
 
+//go:embed schema-light-embed.json
+var pulumiSchemaLight []byte
+
 func main() {
 	ctx := context.Background()
 	info := aws.Provider()
-	pf.MainWithMuxer(ctx, "aws", *info, pulumiSchema)
+
+	s := pulumiSchema
+	if cmdutil.IsTruthy(os.Getenv("PULUMI_AWS_LIGHT_SCHEMA")) {
+		s = pulumiSchemaLight
+	}
+
+	pf.MainWithMuxer(ctx, "aws", *info, s)
 }