sqlcgen

type BatchUpsertRelationshipsParams struct {
	Subjects []pgtype.UUID
	Objects  []pgtype.UUID
	Kinds    []int32
	Roles    []pgtype.UUID
	Granters []pgtype.UUID
}

type CountGrantedObjectsParams struct {
	Subject pgtype.UUID
	Role    pgtype.UUID
	Objects []pgtype.UUID
}

type DBTX interface {
	Exec(context.Context, string, ...interface{}) (pgconn.CommandTag, error)
	Query(context.Context, string, ...interface{}) (pgx.Rows, error)
	QueryRow(context.Context, string, ...interface{}) pgx.Row
}

type DeleteRelationshipsParams struct {
	Subjects []pgtype.UUID
	Objects  []pgtype.UUID
	Kinds    []int32
}

type LookupRelationshipsByOrgParams struct {
	OriginID     string
	Objects      []pgtype.UUID
	Kinds        []int32
	Roles        []pgtype.UUID
	UseCursor    bool
	AfterSubject pgtype.UUID
	AfterObject  pgtype.UUID
	AfterKind    int32
	AfterRole    pgtype.UUID
	PageLimit    int32
}

type LookupRelationshipsByOrgRow struct {
	Subject          pgtype.UUID
	Object           pgtype.UUID
	Kind             int32
	EdgeType         int32
	Role             pgtype.UUID
	CreatedAt        pgtype.Timestamptz
	SubjectLocalID   string
	GrantedByLocalID pgtype.Text
}

type LookupRelationshipsBySubjectParams struct {
	Subject     pgtype.UUID
	Kinds       []int32
	Roles       []pgtype.UUID
	UseCursor   bool
	AfterObject pgtype.UUID
	AfterKind   int32
	PageLimit   int32
}

type Querier interface {
	BatchUpsertRelationships(ctx context.Context, arg BatchUpsertRelationshipsParams) ([]Relationship, error)
	CountGrantedObjects(ctx context.Context, arg CountGrantedObjectsParams) (int64, error)
	DeleteRelationships(ctx context.Context, arg DeleteRelationshipsParams) error
	// origin = 1 is relationships.v1.Origin ORIGIN_ORGANIZATION; org scoping is by
	// the subject's local identity in that org.
	//
	// Both identities joins and the cursor assume one identities row per principal
	// per org, which identity resolution guarantees today. Principals are moving to
	// several identities per org; this query must change first (see the storage
	// schema in docs/relationships-api.md, gitlab-org/gitlab#635812).
	LookupRelationshipsByOrg(ctx context.Context, arg LookupRelationshipsByOrgParams) ([]LookupRelationshipsByOrgRow, error)
	// edge_type = 1 plus the fixed subject means relationships_direct_grant_unique
	// already makes (object, kind) unique per row, so role breaks no ties and is
	// left out of the order/cursor (unlike LookupRelationshipsByOrg, which scans
	// multiple subjects and does need it).
	LookupRelationshipsBySubject(ctx context.Context, arg LookupRelationshipsBySubjectParams) ([]Relationship, error)
	ReadRelationships(ctx context.Context, arg ReadRelationshipsParams) ([]Relationship, error)
}

type Queries struct {
	// contains filtered or unexported fields
}

type ReadRelationshipsParams struct {
	Subject pgtype.UUID
	Objects []pgtype.UUID
	Kinds   []int32
}

type Relationship struct {
	Subject   pgtype.UUID
	Object    pgtype.UUID
	Kind      int32
	EdgeType  int32
	Role      pgtype.UUID
	CreatedAt pgtype.Timestamptz
	GrantedBy pgtype.UUID
}