mirror of
https://github.com/empayre/fleet.git
synced 2024-11-07 09:18:59 +00:00
107 lines
2.9 KiB
Go
107 lines
2.9 KiB
Go
package mysql
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"github.com/kolide/kolide-ose/server/kolide"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// NewInvite generates a new invitation
|
|
func (d *Datastore) NewInvite(i *kolide.Invite) (*kolide.Invite, error) {
|
|
|
|
sql := `
|
|
INSERT INTO invites ( invited_by, email, admin, name, position, token)
|
|
VALUES ( ?, ?, ?, ?, ?, ?)
|
|
`
|
|
|
|
result, err := d.db.Exec(sql, i.InvitedBy, i.Email, i.Admin,
|
|
i.Name, i.Position, i.Token)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "create invite")
|
|
}
|
|
|
|
id, _ := result.LastInsertId()
|
|
i.ID = uint(id)
|
|
|
|
return i, nil
|
|
|
|
}
|
|
|
|
// ListInvites lists all invites in the Kolide database. Supply query options
|
|
// using the opt parameter. See kolide.ListOptions
|
|
func (d *Datastore) ListInvites(opt kolide.ListOptions) ([]*kolide.Invite, error) {
|
|
|
|
invites := []*kolide.Invite{}
|
|
|
|
query := appendListOptionsToSQL("SELECT * FROM invites WHERE NOT deleted", opt)
|
|
err := d.db.Select(&invites, query)
|
|
if err == sql.ErrNoRows {
|
|
return nil, notFound("Invite")
|
|
} else if err != nil {
|
|
return nil, errors.Wrap(err, "select invite by ID")
|
|
}
|
|
return invites, nil
|
|
}
|
|
|
|
// Invite returns Invite identified by id.
|
|
func (d *Datastore) Invite(id uint) (*kolide.Invite, error) {
|
|
var invite kolide.Invite
|
|
err := d.db.Get(&invite, "SELECT * FROM invites WHERE id = ? AND NOT deleted", id)
|
|
if err == sql.ErrNoRows {
|
|
return nil, notFound("Invite").WithID(id)
|
|
} else if err != nil {
|
|
return nil, errors.Wrap(err, "select invite by ID")
|
|
}
|
|
return &invite, nil
|
|
}
|
|
|
|
// InviteByEmail finds an Invite with a particular email, if one exists.
|
|
func (d *Datastore) InviteByEmail(email string) (*kolide.Invite, error) {
|
|
var invite kolide.Invite
|
|
err := d.db.Get(&invite, "SELECT * FROM invites WHERE email = ? AND NOT deleted", email)
|
|
if err == sql.ErrNoRows {
|
|
return nil, notFound("Invite").
|
|
WithMessage(fmt.Sprintf("with email %s", email))
|
|
} else if err != nil {
|
|
return nil, errors.Wrap(err, "sqlx get invite by email")
|
|
}
|
|
return &invite, nil
|
|
}
|
|
|
|
// InviteByToken finds an Invite with a particular token, if one exists.
|
|
func (d *Datastore) InviteByToken(token string) (*kolide.Invite, error) {
|
|
var invite kolide.Invite
|
|
err := d.db.Get(&invite, "SELECT * FROM invites WHERE token = ? AND NOT deleted", token)
|
|
if err == sql.ErrNoRows {
|
|
return nil, notFound("Invite").
|
|
WithMessage(fmt.Sprintf("with token %s", token))
|
|
} else if err != nil {
|
|
return nil, errors.Wrap(err, "sqlx get invite by token")
|
|
}
|
|
return &invite, nil
|
|
}
|
|
|
|
// SaveInvite modifies existing Invite
|
|
func (d *Datastore) SaveInvite(i *kolide.Invite) error {
|
|
sql := `
|
|
UPDATE invites SET invited_by = ?, email = ?, admin = ?,
|
|
name = ?, position = ?, token = ?
|
|
WHERE id = ? AND NOT deleted
|
|
`
|
|
_, err := d.db.Exec(sql, i.InvitedBy, i.Email,
|
|
i.Admin, i.Name, i.Position, i.Token, i.ID,
|
|
)
|
|
if err != nil {
|
|
return errors.Wrap(err, "save invite")
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
func (d *Datastore) DeleteInvite(id uint) error {
|
|
return d.deleteEntity("invites", id)
|
|
}
|