set json writer to PACKAGE_MANAGER enum type based on json schema

Signed-off-by: Brandon Lum <lumjjb@gmail.com>
spdx · Mar 24, 2023 · a2418c6 · a2418c6
1 parent a45344d
commit a2418c6
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/spdx/v2/v2_2/package.go b/spdx/v2/v2_2/package.go
@@ -182,3 +182,18 @@ func (r *PackageExternalReference) UnmarshalJSON(b []byte) error {
 
 	return nil
 }
+
+var _ json.Marshaler = (*PackageExternalReference)(nil)
+
+func (r *PackageExternalReference) MarshalJSON() ([]byte, error) {
+	type ref PackageExternalReference
+	var rr ref
+
+	rr = ref(*r)
+
+	if rr.Category == common.CategoryPackageManager {
+		r.Category = common.CategoryPackageManagerJson
+	}
+
+	return json.Marshal(&rr)
+}
diff --git a/spdx/v2/v2_3/package.go b/spdx/v2/v2_3/package.go
@@ -196,3 +196,18 @@ func (r *PackageExternalReference) UnmarshalJSON(b []byte) error {
 	*r = PackageExternalReference(rr)
 	return nil
 }
+
+var _ json.Marshaler = (*PackageExternalReference)(nil)
+
+func (r *PackageExternalReference) MarshalJSON() ([]byte, error) {
+	type ref PackageExternalReference
+	var rr ref
+
+	rr = ref(*r)
+
+	if rr.Category == common.CategoryPackageManagerJson {
+		r.Category = common.CategoryPackageManager
+	}
+
+	return json.Marshal(&rr)
+}