1 files changed, 75 insertions, 0 deletions
diff --git a/vendor/github.com/authzed/spicedb/pkg/tuple/onr.go b/vendor/github.com/authzed/spicedb/pkg/tuple/onr.go
new file mode 100644
index 0000000..83ad512
--- /dev/null
+++ b/vendor/github.com/authzed/spicedb/pkg/tuple/onr.go
@@ -0,0 +1,75 @@
+package tuple
+
+import (
+	"fmt"
+	"regexp"
+	"slices"
+)
+
+var (
+	onrRegex     = regexp.MustCompile(fmt.Sprintf("^%s$", onrExpr))
+	subjectRegex = regexp.MustCompile(fmt.Sprintf("^%s$", subjectExpr))
+)
+
+var (
+	onrSubjectRelIndex   = slices.Index(subjectRegex.SubexpNames(), "subjectRel")
+	onrSubjectTypeIndex  = slices.Index(subjectRegex.SubexpNames(), "subjectType")
+	onrSubjectIDIndex    = slices.Index(subjectRegex.SubexpNames(), "subjectID")
+	onrResourceTypeIndex = slices.Index(onrRegex.SubexpNames(), "resourceType")
+	onrResourceIDIndex   = slices.Index(onrRegex.SubexpNames(), "resourceID")
+	onrResourceRelIndex  = slices.Index(onrRegex.SubexpNames(), "resourceRel")
+)
+
+// ParseSubjectONR converts a string representation of a Subject ONR to an ObjectAndRelation. Unlike
+// ParseONR, this method allows for objects without relations. If an object without a relation
+// is given, the relation will be set to ellipsis.
+func ParseSubjectONR(subjectOnr string) (ObjectAndRelation, error) {
+	groups := subjectRegex.FindStringSubmatch(subjectOnr)
+	if len(groups) == 0 {
+		return ObjectAndRelation{}, fmt.Errorf("invalid subject ONR: %s", subjectOnr)
+	}
+
+	relation := Ellipsis
+	if len(groups[onrSubjectRelIndex]) > 0 {
+		relation = groups[onrSubjectRelIndex]
+	}
+
+	return ObjectAndRelation{
+		ObjectType: groups[onrSubjectTypeIndex],
+		ObjectID:   groups[onrSubjectIDIndex],
+		Relation:   relation,
+	}, nil
+}
+
+// MustParseSubjectONR converts a string representation of a Subject ONR to an ObjectAndRelation.
+// Panics on error.
+func MustParseSubjectONR(subjectOnr string) ObjectAndRelation {
+	parsed, err := ParseSubjectONR(subjectOnr)
+	if err != nil {
+		panic(err)
+	}
+	return parsed
+}
+
+// ParseONR converts a string representation of an ONR to an ObjectAndRelation object.
+func ParseONR(onr string) (ObjectAndRelation, error) {
+	groups := onrRegex.FindStringSubmatch(onr)
+	if len(groups) == 0 {
+		return ObjectAndRelation{}, fmt.Errorf("invalid ONR: %s", onr)
+	}
+
+	return ObjectAndRelation{
+		ObjectType: groups[onrResourceTypeIndex],
+		ObjectID:   groups[onrResourceIDIndex],
+		Relation:   groups[onrResourceRelIndex],
+	}, nil
+}
+
+// MustParseONR converts a string representation of an ONR to an ObjectAndRelation object. Panics on error.
+func MustParseONR(onr string) ObjectAndRelation {
+	parsed, err := ParseONR(onr)
+	if err != nil {
+		panic(err)
+	}
+	return parsed
+}