// Code generated by ent, DO NOT EDIT. package ent import ( "context" "errors" "fmt" "sync" "entgo.io/ent" "entgo.io/ent/dialect/sql" "git.rgst.io/homelab/klefki/internal/db/ent/machine" "git.rgst.io/homelab/klefki/internal/db/ent/predicate" ) const ( // Operation types. OpCreate = ent.OpCreate OpDelete = ent.OpDelete OpDeleteOne = ent.OpDeleteOne OpUpdate = ent.OpUpdate OpUpdateOne = ent.OpUpdateOne // Node types. TypeMachine = "Machine" ) // MachineMutation represents an operation that mutates the Machine nodes in the graph. type MachineMutation struct { config op Op typ string id *string public_key *string clearedFields map[string]struct{} done bool oldValue func(context.Context) (*Machine, error) predicates []predicate.Machine } var _ ent.Mutation = (*MachineMutation)(nil) // machineOption allows management of the mutation configuration using functional options. type machineOption func(*MachineMutation) // newMachineMutation creates new mutation for the Machine entity. func newMachineMutation(c config, op Op, opts ...machineOption) *MachineMutation { m := &MachineMutation{ config: c, op: op, typ: TypeMachine, clearedFields: make(map[string]struct{}), } for _, opt := range opts { opt(m) } return m } // withMachineID sets the ID field of the mutation. func withMachineID(id string) machineOption { return func(m *MachineMutation) { var ( err error once sync.Once value *Machine ) m.oldValue = func(ctx context.Context) (*Machine, error) { once.Do(func() { if m.done { err = errors.New("querying old values post mutation is not allowed") } else { value, err = m.Client().Machine.Get(ctx, id) } }) return value, err } m.id = &id } } // withMachine sets the old Machine of the mutation. func withMachine(node *Machine) machineOption { return func(m *MachineMutation) { m.oldValue = func(context.Context) (*Machine, error) { return node, nil } m.id = &node.ID } } // Client returns a new `ent.Client` from the mutation. If the mutation was // executed in a transaction (ent.Tx), a transactional client is returned. func (m MachineMutation) Client() *Client { client := &Client{config: m.config} client.init() return client } // Tx returns an `ent.Tx` for mutations that were executed in transactions; // it returns an error otherwise. func (m MachineMutation) Tx() (*Tx, error) { if _, ok := m.driver.(*txDriver); !ok { return nil, errors.New("ent: mutation is not running in a transaction") } tx := &Tx{config: m.config} tx.init() return tx, nil } // SetID sets the value of the id field. Note that this // operation is only accepted on creation of Machine entities. func (m *MachineMutation) SetID(id string) { m.id = &id } // ID returns the ID value in the mutation. Note that the ID is only available // if it was provided to the builder or after it was returned from the database. func (m *MachineMutation) ID() (id string, exists bool) { if m.id == nil { return } return *m.id, true } // IDs queries the database and returns the entity ids that match the mutation's predicate. // That means, if the mutation is applied within a transaction with an isolation level such // as sql.LevelSerializable, the returned ids match the ids of the rows that will be updated // or updated by the mutation. func (m *MachineMutation) IDs(ctx context.Context) ([]string, error) { switch { case m.op.Is(OpUpdateOne | OpDeleteOne): id, exists := m.ID() if exists { return []string{id}, nil } fallthrough case m.op.Is(OpUpdate | OpDelete): return m.Client().Machine.Query().Where(m.predicates...).IDs(ctx) default: return nil, fmt.Errorf("IDs is not allowed on %s operations", m.op) } } // SetPublicKey sets the "public_key" field. func (m *MachineMutation) SetPublicKey(s string) { m.public_key = &s } // PublicKey returns the value of the "public_key" field in the mutation. func (m *MachineMutation) PublicKey() (r string, exists bool) { v := m.public_key if v == nil { return } return *v, true } // OldPublicKey returns the old "public_key" field's value of the Machine entity. // If the Machine object wasn't provided to the builder, the object is fetched from the database. // An error is returned if the mutation operation is not UpdateOne, or the database query fails. func (m *MachineMutation) OldPublicKey(ctx context.Context) (v string, err error) { if !m.op.Is(OpUpdateOne) { return v, errors.New("OldPublicKey is only allowed on UpdateOne operations") } if m.id == nil || m.oldValue == nil { return v, errors.New("OldPublicKey requires an ID field in the mutation") } oldValue, err := m.oldValue(ctx) if err != nil { return v, fmt.Errorf("querying old value for OldPublicKey: %w", err) } return oldValue.PublicKey, nil } // ResetPublicKey resets all changes to the "public_key" field. func (m *MachineMutation) ResetPublicKey() { m.public_key = nil } // Where appends a list predicates to the MachineMutation builder. func (m *MachineMutation) Where(ps ...predicate.Machine) { m.predicates = append(m.predicates, ps...) } // WhereP appends storage-level predicates to the MachineMutation builder. Using this method, // users can use type-assertion to append predicates that do not depend on any generated package. func (m *MachineMutation) WhereP(ps ...func(*sql.Selector)) { p := make([]predicate.Machine, len(ps)) for i := range ps { p[i] = ps[i] } m.Where(p...) } // Op returns the operation name. func (m *MachineMutation) Op() Op { return m.op } // SetOp allows setting the mutation operation. func (m *MachineMutation) SetOp(op Op) { m.op = op } // Type returns the node type of this mutation (Machine). func (m *MachineMutation) 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 *MachineMutation) Fields() []string { fields := make([]string, 0, 1) if m.public_key != nil { fields = append(fields, machine.FieldPublicKey) } 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 *MachineMutation) Field(name string) (ent.Value, bool) { switch name { case machine.FieldPublicKey: return m.PublicKey() } 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 *MachineMutation) OldField(ctx context.Context, name string) (ent.Value, error) { switch name { case machine.FieldPublicKey: return m.OldPublicKey(ctx) } return nil, fmt.Errorf("unknown Machine 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 *MachineMutation) SetField(name string, value ent.Value) error { switch name { case machine.FieldPublicKey: v, ok := value.(string) if !ok { return fmt.Errorf("unexpected type %T for field %s", value, name) } m.SetPublicKey(v) return nil } return fmt.Errorf("unknown Machine field %s", name) } // AddedFields returns all numeric fields that were incremented/decremented during // this mutation. func (m *MachineMutation) AddedFields() []string { return nil } // 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 *MachineMutation) AddedField(name string) (ent.Value, bool) { 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 *MachineMutation) AddField(name string, value ent.Value) error { switch name { } return fmt.Errorf("unknown Machine numeric field %s", name) } // ClearedFields returns all nullable fields that were cleared during this // mutation. func (m *MachineMutation) ClearedFields() []string { return nil } // FieldCleared returns a boolean indicating if a field with the given name was // cleared in this mutation. func (m *MachineMutation) 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 *MachineMutation) ClearField(name string) error { return fmt.Errorf("unknown Machine 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 *MachineMutation) ResetField(name string) error { switch name { case machine.FieldPublicKey: m.ResetPublicKey() return nil } return fmt.Errorf("unknown Machine field %s", name) } // AddedEdges returns all edge names that were set/added in this mutation. func (m *MachineMutation) AddedEdges() []string { edges := make([]string, 0, 0) return edges } // AddedIDs returns all IDs (to other nodes) that were added for the given edge // name in this mutation. func (m *MachineMutation) AddedIDs(name string) []ent.Value { return nil } // RemovedEdges returns all edge names that were removed in this mutation. func (m *MachineMutation) RemovedEdges() []string { edges := make([]string, 0, 0) return edges } // RemovedIDs returns all IDs (to other nodes) that were removed for the edge with // the given name in this mutation. func (m *MachineMutation) RemovedIDs(name string) []ent.Value { return nil } // ClearedEdges returns all edge names that were cleared in this mutation. func (m *MachineMutation) ClearedEdges() []string { edges := make([]string, 0, 0) return edges } // EdgeCleared returns a boolean which indicates if the edge with the given name // was cleared in this mutation. func (m *MachineMutation) EdgeCleared(name string) bool { 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 *MachineMutation) ClearEdge(name string) error { return fmt.Errorf("unknown Machine 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 *MachineMutation) ResetEdge(name string) error { return fmt.Errorf("unknown Machine edge %s", name) }