fix yaml archives for conformance

cmd/registry/cmd/apply/apply_test.go

@@ -229,8 +229,6 @@ func TestArtifactStorage(t *testing.T) {
 		}
 
 		want := &apihub.Lifecycle{
-			Id:          "lifecycle", // deprecated field
-			Kind:        "Lifecycle", // deprecated field
 			DisplayName: "Lifecycle",
 			Description: "A series of stages that an API typically moves through in its lifetime",
 			Stages: []*apihub.Lifecycle_Stage{

cmd/registry/cmd/compute/conformance/conformance.go

@@ -31,7 +31,7 @@
 	"github.com/spf13/cobra"
 )
 
-var styleguideFilter = fmt.Sprintf("mime_type.contains('%s')", mime.MimeTypeForKind("StyleGuide"))
+var styleguideFilter = fmt.Sprintf("mime_type.contains('%s') || mime_type.contains('%s')", mime.MimeTypeForKind("StyleGuide"), mime.YamlMimeTypeForKind("StyleGuide"))
 
 func Command() *cobra.Command {
 	var filter string
@@ -82,6 +82,11 @@
 					log.FromContext(ctx).WithError(err).Debugf("Unmarshal() to StyleGuide failed on artifact: %s", artifact.GetName())
 					return nil
 				}
+				name, err := names.ParseArtifact(artifact.GetName())
+				if err != nil {
+					return err
+				}
+				guide.Id = name.ArtifactID()
 				guides = append(guides, guide)
 				return nil
 			}); err != nil {

cmd/registry/patch/artifact.go

@@ -16,8 +16,6 @@
 
 import (
 	"context"
-	"encoding/json"
-	"errors"
 	"fmt"
 	"strings"
 
@@ -115,54 +113,29 @@
 }
 
 func applyArtifactPatch(ctx context.Context, client connection.RegistryClient, content *encoding.Artifact, parent string, filename string) error {
-	// Restyle the YAML representation so that yaml.Marshal will marshal it as JSON.
-	encoding.StyleForJSON(&content.Data)
-	// Marshal the YAML representation into the JSON serialization.
-	j, err := yaml.Marshal(content.Data)
-	if err != nil {
-		return err
-	}
-	// Populate Id and Kind fields in the contents of the artifact
-	jWithIdAndKind, err := populateIdAndKind(j, content.Kind, content.Metadata.Name)
-	if err != nil {
-		return err
-	}
 	var mimeType string
 	var bytes []byte
-	// Unmarshal the JSON serialization into the message struct.
-	var m proto.Message
-	m, err = mime.MessageForKind(content.Kind)
-	if err == nil {
-		err = protojson.Unmarshal(jWithIdAndKind, m)
+	msg, err := mime.MessageForKind(content.Kind)
+	if storeArchivesAsYaml(ctx) || err != nil {
+		mimeType = mime.YamlMimeTypeForKind(content.Kind)
+		encoding.StyleForYAML(&content.Data)
+		bytes, err = yaml.Marshal(content.Data)
 		if err != nil {
-			if strings.Contains(err.Error(), "unknown field") {
-				// Try unmarshaling the original YAML (without the additional Id and Kind fields).
-				err = protojson.Unmarshal(j, m)
-				if err != nil {
-					return err
-				}
-			}
-		}
-		if storeArchivesAsYaml(ctx) {
-			mimeType = mime.YamlMimeTypeForKind(content.Kind)
-			encoding.StyleForYAML(&content.Data)
-			bytes, err = yaml.Marshal(content.Data)
-			if err != nil {
-				return err
-			}
-		} else {
-			mimeType = mime.MimeTypeForKind(content.Kind)
-			// Marshal the message struct to bytes.
-			bytes, err = proto.Marshal(m)
-			if err != nil {
-				return err
-			}
+			return err
 		}
 	} else {
-		// If there was no struct defined for the type, marshal it struct as YAML
+		// Convert YAML to JSON for protojson
+		encoding.StyleForJSON(&content.Data)
+		j, err := yaml.Marshal(content.Data)
+		if err != nil {
+			return err
+		}
+		err = protojson.Unmarshal(j, msg)
+		if err != nil {
+			return err
+		}
 		mimeType = mime.MimeTypeForKind(content.Kind)
-		encoding.StyleForYAML(&content.Data)
-		bytes, err = yaml.Marshal(content.Data)
+		bytes, err = proto.Marshal(msg)
 		if err != nil {
 			return err
 		}
@@ -196,27 +169,6 @@
 	return nil
 }
 
-// populateIdAndKind inserts the "id" and "kind" fields in the supplied json bytes.
-func populateIdAndKind(bytes []byte, kind, id string) ([]byte, error) {
-	var jsonData map[string]interface{}
-	err := json.Unmarshal(bytes, &jsonData)
-	if err != nil {
-		return nil, err
-	}
-	if jsonData == nil {
-		return nil, errors.New("missing data")
-	}
-	jsonData["id"] = id
-	jsonData["kind"] = kind
-
-	rBytes, err := json.Marshal(jsonData)
-	if err != nil {
-		return nil, err
-	}
-
-	return rBytes, nil
-}
-
 func UnmarshalContents(contents []byte, mimeType string, message proto.Message) error {
 	if !mime.IsYamlKind(mimeType) {
 		return proto.Unmarshal(contents, message)

cmd/registry/patch/patch_test.go

@@ -1185,9 +1185,6 @@ func TestInvalidArtifactPatches(t *testing.T) {
 		{
 			artifactID: "references-no-data",
 		},
-		{
-			artifactID: "struct-invalid-structure",
-		},
 		{
 			artifactID: "struct-no-metadata",
 		},