// 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. // Code generated by ent, DO NOT EDIT. package user import ( "context" "fmt" "entgo.io/ent" "entgo.io/ent/dialect/sql" "entgo.io/ent/entc/integration/hooks/ent/predicate" ) // Mutation represents an operation that mutates the User nodes in the graph. type Mutation struct { op ent.Op typ string version *int addversion *int name *string worth *uint addworth *int password *string active *bool clearedFields map[string]struct{} cards map[int]struct{} removedcards map[int]struct{} clearedcards bool pets map[int]struct{} removedpets map[int]struct{} clearedpets bool friends map[int]struct{} removedfriends map[int]struct{} clearedfriends bool best_friend *int clearedbest_friend bool predicates []predicate.User } // NewMutation creates a new Mutation for the User entity. func NewMutation(op ent.Op) *Mutation { return &Mutation{ op: op, typ: "User", clearedFields: make(map[string]struct{}), } } // Predicates returns the list of predicates set on the mutation. func (m *Mutation) Predicates() []predicate.User { return m.predicates } // SetVersion sets the "version" field. func (m *Mutation) SetVersion(i int) { m.version = &i m.addversion = nil } // Version returns the value of the "version" field in the mutation. func (m *Mutation) Version() (r int, exists bool) { v := m.version if v == nil { return } return *v, true } // AddVersion adds i to the "version" field. func (m *Mutation) AddVersion(i int) { if m.addversion != nil { *m.addversion += i } else { m.addversion = &i } } // AddedVersion returns the value that was added to the "version" field in this mutation. func (m *Mutation) AddedVersion() (r int, exists bool) { v := m.addversion if v == nil { return } return *v, true } // ResetVersion resets all changes to the "version" field. func (m *Mutation) ResetVersion() { m.version = nil m.addversion = nil } // SetName sets the "name" field. func (m *Mutation) SetName(s string) { m.name = &s } // Name returns the value of the "name" field in the mutation. func (m *Mutation) Name() (r string, exists bool) { v := m.name if v == nil { return } return *v, true } // ResetName resets all changes to the "name" field. func (m *Mutation) ResetName() { m.name = nil } // SetWorth sets the "worth" field. func (m *Mutation) SetWorth(u uint) { m.worth = &u m.addworth = nil } // Worth returns the value of the "worth" field in the mutation. func (m *Mutation) Worth() (r uint, exists bool) { v := m.worth if v == nil { return } return *v, true } // AddWorth adds u to the "worth" field. func (m *Mutation) AddWorth(u int) { if m.addworth != nil { *m.addworth += u } else { m.addworth = &u } } // AddedWorth returns the value that was added to the "worth" field in this mutation. func (m *Mutation) AddedWorth() (r int, exists bool) { v := m.addworth if v == nil { return } return *v, true } // ClearWorth clears the value of the "worth" field. func (m *Mutation) ClearWorth() { m.worth = nil m.addworth = nil m.clearedFields[FieldWorth] = struct{}{} } // WorthCleared returns if the "worth" field was cleared in this mutation. func (m *Mutation) WorthCleared() bool { _, ok := m.clearedFields[FieldWorth] return ok } // ResetWorth resets all changes to the "worth" field. func (m *Mutation) ResetWorth() { m.worth = nil m.addworth = nil delete(m.clearedFields, FieldWorth) } // SetPassword sets the "password" field. func (m *Mutation) SetPassword(s string) { m.password = &s } // Password returns the value of the "password" field in the mutation. func (m *Mutation) Password() (r string, exists bool) { v := m.password if v == nil { return } return *v, true } // ClearPassword clears the value of the "password" field. func (m *Mutation) ClearPassword() { m.password = nil m.clearedFields[FieldPassword] = struct{}{} } // PasswordCleared returns if the "password" field was cleared in this mutation. func (m *Mutation) PasswordCleared() bool { _, ok := m.clearedFields[FieldPassword] return ok } // ResetPassword resets all changes to the "password" field. func (m *Mutation) ResetPassword() { m.password = nil delete(m.clearedFields, FieldPassword) } // SetActive sets the "active" field. func (m *Mutation) SetActive(b bool) { m.active = &b } // Active returns the value of the "active" field in the mutation. func (m *Mutation) Active() (r bool, exists bool) { v := m.active if v == nil { return } return *v, true } // ResetActive resets all changes to the "active" field. func (m *Mutation) ResetActive() { m.active = nil } // AddCardIDs adds the "cards" edge to the Card entity by ids. func (m *Mutation) AddCardIDs(ids ...int) { if m.cards == nil { m.cards = make(map[int]struct{}) } for i := range ids { m.cards[ids[i]] = struct{}{} } } // ClearCards clears the "cards" edge to the Card entity. func (m *Mutation) ClearCards() { m.clearedcards = true } // CardsCleared reports if the "cards" edge to the Card entity was cleared. func (m *Mutation) CardsCleared() bool { return m.clearedcards } // RemoveCardIDs removes the "cards" edge to the Card entity by IDs. func (m *Mutation) RemoveCardIDs(ids ...int) { if m.removedcards == nil { m.removedcards = make(map[int]struct{}) } for i := range ids { delete(m.cards, ids[i]) m.removedcards[ids[i]] = struct{}{} } } // RemovedCards returns the removed IDs of the "cards" edge to the Card entity. func (m *Mutation) RemovedCardsIDs() (ids []int) { for id := range m.removedcards { ids = append(ids, id) } return } // CardsIDs returns the "cards" edge IDs in the mutation. func (m *Mutation) CardsIDs() (ids []int) { for id := range m.cards { ids = append(ids, id) } return } // ResetCards resets all changes to the "cards" edge. func (m *Mutation) ResetCards() { m.cards = nil m.clearedcards = false m.removedcards = nil } // AddPetIDs adds the "pets" edge to the Pet entity by ids. func (m *Mutation) AddPetIDs(ids ...int) { if m.pets == nil { m.pets = make(map[int]struct{}) } for i := range ids { m.pets[ids[i]] = struct{}{} } } // ClearPets clears the "pets" edge to the Pet entity. func (m *Mutation) ClearPets() { m.clearedpets = true } // PetsCleared reports if the "pets" edge to the Pet entity was cleared. func (m *Mutation) PetsCleared() bool { return m.clearedpets } // RemovePetIDs removes the "pets" edge to the Pet entity by IDs. func (m *Mutation) RemovePetIDs(ids ...int) { if m.removedpets == nil { m.removedpets = make(map[int]struct{}) } for i := range ids { delete(m.pets, ids[i]) m.removedpets[ids[i]] = struct{}{} } } // RemovedPets returns the removed IDs of the "pets" edge to the Pet entity. func (m *Mutation) RemovedPetsIDs() (ids []int) { for id := range m.removedpets { ids = append(ids, id) } return } // PetsIDs returns the "pets" edge IDs in the mutation. func (m *Mutation) PetsIDs() (ids []int) { for id := range m.pets { ids = append(ids, id) } return } // ResetPets resets all changes to the "pets" edge. func (m *Mutation) ResetPets() { m.pets = nil m.clearedpets = false m.removedpets = nil } // AddFriendIDs adds the "friends" edge to the User entity by ids. func (m *Mutation) AddFriendIDs(ids ...int) { if m.friends == nil { m.friends = make(map[int]struct{}) } for i := range ids { m.friends[ids[i]] = struct{}{} } } // ClearFriends clears the "friends" edge to the User entity. func (m *Mutation) ClearFriends() { m.clearedfriends = true } // FriendsCleared reports if the "friends" edge to the User entity was cleared. func (m *Mutation) FriendsCleared() bool { return m.clearedfriends } // RemoveFriendIDs removes the "friends" edge to the User entity by IDs. func (m *Mutation) RemoveFriendIDs(ids ...int) { if m.removedfriends == nil { m.removedfriends = make(map[int]struct{}) } for i := range ids { delete(m.friends, ids[i]) m.removedfriends[ids[i]] = struct{}{} } } // RemovedFriends returns the removed IDs of the "friends" edge to the User entity. func (m *Mutation) RemovedFriendsIDs() (ids []int) { for id := range m.removedfriends { ids = append(ids, id) } return } // FriendsIDs returns the "friends" edge IDs in the mutation. func (m *Mutation) FriendsIDs() (ids []int) { for id := range m.friends { ids = append(ids, id) } return } // ResetFriends resets all changes to the "friends" edge. func (m *Mutation) ResetFriends() { m.friends = nil m.clearedfriends = false m.removedfriends = nil } // SetBestFriendID sets the "best_friend" edge to the User entity by id. func (m *Mutation) SetBestFriendID(id int) { m.best_friend = &id } // ClearBestFriend clears the "best_friend" edge to the User entity. func (m *Mutation) ClearBestFriend() { m.clearedbest_friend = true } // BestFriendCleared reports if the "best_friend" edge to the User entity was cleared. func (m *Mutation) BestFriendCleared() bool { return m.clearedbest_friend } // BestFriendID returns the "best_friend" edge ID in the mutation. func (m *Mutation) BestFriendID() (id int, exists bool) { if m.best_friend != nil { return *m.best_friend, true } return } // BestFriendIDs returns the "best_friend" edge IDs in the mutation. // Note that IDs always returns len(IDs) <= 1 for unique edges, and you should use // BestFriendID instead. It exists only for internal usage by the builders. func (m *Mutation) BestFriendIDs() (ids []int) { if id := m.best_friend; id != nil { ids = append(ids, *id) } return } // ResetBestFriend resets all changes to the "best_friend" edge. func (m *Mutation) ResetBestFriend() { m.best_friend = nil m.clearedbest_friend = false } // Where appends a list predicates to the Mutation builder. func (m *Mutation) Where(ps ...predicate.User) { m.predicates = append(m.predicates, ps...) } // WhereP appends storage-level predicates to the Mutation builder. Using this method, // users can use type-assertion to append predicates that do not depend on any generated package. func (m *Mutation) WhereP(ps ...func(*sql.Selector)) { p := make([]predicate.User, len(ps)) for i := range ps { p[i] = ps[i] } m.Where(p...) } // Op returns the operation name. func (m *Mutation) Op() ent.Op { return m.op } // SetOp allows setting the mutation operation. func (m *Mutation) SetOp(op ent.Op) { m.op = op } // Type returns the node type of this mutation (User). func (m *Mutation) Type() string { return m.typ } // Fields returns all fields that were changed during this mutation. Note that in // order to get all numeric fields that were incremented/decremented, call // AddedFields(). func (m *Mutation) Fields() []string { fields := make([]string, 0, 5) if m.version != nil { fields = append(fields, FieldVersion) } if m.name != nil { fields = append(fields, FieldName) } if m.worth != nil { fields = append(fields, FieldWorth) } if m.password != nil { fields = append(fields, FieldPassword) } if m.active != nil { fields = append(fields, FieldActive) } return fields } // Field returns the value of a field with the given name. The second boolean // return value indicates that this field was not set, or was not defined in the // schema. func (m *Mutation) Field(name string) (ent.Value, bool) { switch name { case FieldVersion: return m.Version() case FieldName: return m.Name() case FieldWorth: return m.Worth() case FieldPassword: return m.Password() case FieldActive: return m.Active() } return nil, false } // OldField returns the old value of the field from the database. An error is // returned if the mutation operation is not UpdateOne, or the query to the // database failed. func (m *Mutation) OldField(ctx context.Context, name string) (ent.Value, error) { return nil, fmt.Errorf("unknown User field %s", name) } // SetField sets the value of a field with the given name. It returns an error if // the field is not defined in the schema, or if the type mismatched the field // type. func (m *Mutation) SetField(name string, value ent.Value) error { switch name { case FieldVersion: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetVersion(v) return nil case FieldName: v, ok := value.(string) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetName(v) return nil case FieldWorth: v, ok := value.(uint) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetWorth(v) return nil case FieldPassword: v, ok := value.(string) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetPassword(v) return nil case FieldActive: v, ok := value.(bool) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetActive(v) return nil } return fmt.Errorf("unknown User field %s", name) } // AddedFields returns all numeric fields that were incremented/decremented during // this mutation. func (m *Mutation) AddedFields() []string { var fields []string if m.addversion != nil { fields = append(fields, FieldVersion) } if m.addworth != nil { fields = append(fields, FieldWorth) } return fields } // AddedField returns the numeric value that was incremented/decremented on a field // with the given name. The second boolean return value indicates that this field // was not set, or was not defined in the schema. func (m *Mutation) AddedField(name string) (ent.Value, bool) { switch name { case FieldVersion: return m.AddedVersion() case FieldWorth: return m.AddedWorth() } return nil, false } // AddField adds the value to the field with the given name. It returns an error if // the field is not defined in the schema, or if the type mismatched the field // type. func (m *Mutation) AddField(name string, value ent.Value) error { switch name { case FieldVersion: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.AddVersion(v) return nil case FieldWorth: v, ok := value.(int) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.AddWorth(v) return nil } return fmt.Errorf("unknown User numeric field %s", name) } // ClearedFields returns all nullable fields that were cleared during this // mutation. func (m *Mutation) ClearedFields() []string { var fields []string if m.FieldCleared(FieldWorth) { fields = append(fields, FieldWorth) } if m.FieldCleared(FieldPassword) { fields = append(fields, FieldPassword) } return fields } // FieldCleared returns a boolean indicating if a field with the given name was // cleared in this mutation. func (m *Mutation) FieldCleared(name string) bool { _, ok := m.clearedFields[name] return ok } // ClearField clears the value of the field with the given name. It returns an // error if the field is not defined in the schema. func (m *Mutation) ClearField(name string) error { switch name { case FieldWorth: m.ClearWorth() return nil case FieldPassword: m.ClearPassword() return nil } return fmt.Errorf("unknown User nullable field %s", name) } // ResetField resets all changes in the mutation for the field with the given name. // It returns an error if the field is not defined in the schema. func (m *Mutation) ResetField(name string) error { switch name { case FieldVersion: m.ResetVersion() return nil case FieldName: m.ResetName() return nil case FieldWorth: m.ResetWorth() return nil case FieldPassword: m.ResetPassword() return nil case FieldActive: m.ResetActive() return nil } return fmt.Errorf("unknown User field %s", name) } // AddedEdges returns all edge names that were set/added in this mutation. func (m *Mutation) AddedEdges() []string { edges := make([]string, 0, 4) if m.cards != nil { edges = append(edges, EdgeCards) } if m.pets != nil { edges = append(edges, EdgePets) } if m.friends != nil { edges = append(edges, EdgeFriends) } if m.best_friend != nil { edges = append(edges, EdgeBestFriend) } return edges } // AddedIDs returns all IDs (to other nodes) that were added for the given edge // name in this mutation. func (m *Mutation) AddedIDs(name string) []ent.Value { switch name { case EdgeCards: ids := make([]ent.Value, 0, len(m.cards)) for id := range m.cards { ids = append(ids, id) } return ids case EdgePets: ids := make([]ent.Value, 0, len(m.pets)) for id := range m.pets { ids = append(ids, id) } return ids case EdgeFriends: ids := make([]ent.Value, 0, len(m.friends)) for id := range m.friends { ids = append(ids, id) } return ids case EdgeBestFriend: if id := m.best_friend; id != nil { return []ent.Value{*id} } } return nil } // RemovedEdges returns all edge names that were removed in this mutation. func (m *Mutation) RemovedEdges() []string { edges := make([]string, 0, 4) if m.removedcards != nil { edges = append(edges, EdgeCards) } if m.removedpets != nil { edges = append(edges, EdgePets) } if m.removedfriends != nil { edges = append(edges, EdgeFriends) } return edges } // RemovedIDs returns all IDs (to other nodes) that were removed for the edge with // the given name in this mutation. func (m *Mutation) RemovedIDs(name string) []ent.Value { switch name { case EdgeCards: ids := make([]ent.Value, 0, len(m.removedcards)) for id := range m.removedcards { ids = append(ids, id) } return ids case EdgePets: ids := make([]ent.Value, 0, len(m.removedpets)) for id := range m.removedpets { ids = append(ids, id) } return ids case EdgeFriends: ids := make([]ent.Value, 0, len(m.removedfriends)) for id := range m.removedfriends { ids = append(ids, id) } return ids } return nil } // ClearedEdges returns all edge names that were cleared in this mutation. func (m *Mutation) ClearedEdges() []string { edges := make([]string, 0, 4) if m.clearedcards { edges = append(edges, EdgeCards) } if m.clearedpets { edges = append(edges, EdgePets) } if m.clearedfriends { edges = append(edges, EdgeFriends) } if m.clearedbest_friend { edges = append(edges, EdgeBestFriend) } return edges } // EdgeCleared returns a boolean which indicates if the edge with the given name // was cleared in this mutation. func (m *Mutation) EdgeCleared(name string) bool { switch name { case EdgeCards: return m.clearedcards case EdgePets: return m.clearedpets case EdgeFriends: return m.clearedfriends case EdgeBestFriend: return m.clearedbest_friend } return false } // ClearEdge clears the value of the edge with the given name. It returns an error // if that edge is not defined in the schema. func (m *Mutation) ClearEdge(name string) error { switch name { case EdgeBestFriend: m.ClearBestFriend() return nil } return fmt.Errorf("unknown User unique edge %s", name) } // ResetEdge resets all changes to the edge with the given name in this mutation. // It returns an error if the edge is not defined in the schema. func (m *Mutation) ResetEdge(name string) error { switch name { case EdgeCards: m.ResetCards() return nil case EdgePets: m.ResetPets() return nil case EdgeFriends: m.ResetFriends() return nil case EdgeBestFriend: m.ResetBestFriend() return nil } return fmt.Errorf("unknown User edge %s", name) }