Add relationship filter function

Signed-off-by: Ian Ling <ian@iancaling.com>

1 files changed, 66 insertions, 0 deletions

diff --git a/spdxlib/documents.go b/spdxlib/documents.go
new file mode 100644
index 0000000..0f90a02
--- /dev/null
+++ b/spdxlib/documents.go
@@ -0,0 +1,66 @@
+package spdxlib
+
+import (
+    "fmt"
+    "github.com/spdx/tools-golang/spdx"
+)
+
+// ValidateDocument2_1 returns an error if the Document is found to be invalid, or nil if the Document is valid.
+// Currently, this only verifies that all Element IDs mentioned in Relationships exist in the Document as either a
+// Package or an UnpackagedFile.
+func ValidateDocument2_1(doc *spdx.Document2_1) error {
+    // cache a map of valid package IDs for quick lookups
+    validElementIDs := make(map[spdx.ElementID]bool)
+    for _, docPackage := range doc.Packages {
+        validElementIDs[docPackage.PackageSPDXIdentifier] = true
+    }
+
+    for _, unpackagedFile := range doc.UnpackagedFiles {
+        validElementIDs[unpackagedFile.FileSPDXIdentifier] = true
+    }
+
+    // add the Document element ID
+    validElementIDs[spdx.MakeDocElementID("", "DOCUMENT").ElementRefID] = true
+
+    for _, relationship := range doc.Relationships {
+        if !validElementIDs[relationship.RefA.ElementRefID] {
+            return fmt.Errorf("%s used in relationship but no such package exists", string(relationship.RefA.ElementRefID))
+        }
+
+        if !validElementIDs[relationship.RefB.ElementRefID] {
+            return fmt.Errorf("%s used in relationship but no such package exists", string(relationship.RefB.ElementRefID))
+        }
+    }
+
+    return nil
+}
+
+// ValidateDocument2_2 returns an error if the Document is found to be invalid, or nil if the Document is valid.
+// Currently, this only verifies that all Element IDs mentioned in Relationships exist in the Document as either a
+// Package or an UnpackagedFile.
+func ValidateDocument2_2(doc *spdx.Document2_2) error {
+    // cache a map of package IDs for quick lookups
+    validElementIDs := make(map[spdx.ElementID]bool)
+    for _, docPackage := range doc.Packages {
+        validElementIDs[docPackage.PackageSPDXIdentifier] = true
+    }
+
+    for _, unpackagedFile := range doc.UnpackagedFiles {
+        validElementIDs[unpackagedFile.FileSPDXIdentifier] = true
+    }
+
+    // add the Document element ID
+    validElementIDs[spdx.MakeDocElementID("", "DOCUMENT").ElementRefID] = true
+
+    for _, relationship := range doc.Relationships {
+        if !validElementIDs[relationship.RefA.ElementRefID] {
+            return fmt.Errorf("%s used in relationship but no such package exists", string(relationship.RefA.ElementRefID))
+        }
+
+        if !validElementIDs[relationship.RefB.ElementRefID] {
+            return fmt.Errorf("%s used in relationship but no such package exists", string(relationship.RefB.ElementRefID))
+        }
+    }
+
+    return nil
+}