snapshot of https://github.com/ent/ent.git (master, 2026-09-15; blobs > 131072B omitted)

entc/integration/ent/role/role.go

@@ -0,0 +1,78 @@
+// Copyright 2019-present Facebook Inc. All rights reserved.
+// This source code is licensed under the Apache 2.0 license found
+// in the LICENSE file in the root directory of this source tree.
+
+package role
+
+import (
+	"database/sql/driver"
+)
+
+type Role string
+
+const (
+	Admin     Role = "ADMIN"
+	Owner     Role = "OWNER"
+	User      Role = "USER"
+	Read      Role = "READ"
+	Write     Role = "WRITE"
+	ReadWrite Role = "READ+WRITE"
+)
+
+func (Role) Values() (roles []string) {
+	for _, r := range []Role{Admin, Owner, User, Read, Write, ReadWrite} {
+		roles = append(roles, string(r))
+	}
+	return
+}
+
+type Priority int
+
+const (
+	Unknown Priority = iota
+	Low
+	High
+)
+
+func (p Priority) String() string {
+	switch p {
+	case Low:
+		return "LOW"
+	case High:
+		return "HIGH"
+	default:
+		return "UNKNOWN"
+	}
+}
+
+func (Priority) Values() []string {
+	return []string{Unknown.String(), Low.String(), High.String()}
+}
+
+func (p Priority) Value() (driver.Value, error) {
+	return p.String(), nil
+}
+
+func (p *Priority) Scan(val any) error {
+	var s string
+
+	switch v := val.(type) {
+	case nil:
+		return nil
+	case string:
+		s = v
+	case []uint8:
+		s = string(v)
+	}
+
+	switch s {
+	case "LOW":
+		*p = Low
+	case "HIGH":
+		*p = High
+	default:
+		*p = Unknown
+	}
+
+	return nil
+}