mirror of
https://github.com/empayre/fleet.git
synced 2024-11-07 09:18:59 +00:00
a036c5da9f
* Adds created_by attribute to packs This PR also updated the distributed query code to use the pattern established here (service checks context) * add enable/disable state to packs * add query_count to packs API responses * add host_count to packs API responses (very, very poorly) * pack description should not be required * counting hosts in packs via mysql * removing extraneous newline in test * Switch case instead of if/if else * add description to update query for SavePack method * change AND to WHERE in query as per @zwass * add ordering and list options as per @murphybytes' suggestion
80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package kolide
|
|
|
|
import (
|
|
"time"
|
|
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
type PackStore interface {
|
|
// Pack methods
|
|
NewPack(pack *Pack) (*Pack, error)
|
|
SavePack(pack *Pack) error
|
|
DeletePack(pid uint) error
|
|
Pack(pid uint) (*Pack, error)
|
|
ListPacks(opt ListOptions) ([]*Pack, error)
|
|
|
|
// Modifying the queries in packs
|
|
AddQueryToPack(qid uint, pid uint) error
|
|
ListQueriesInPack(pack *Pack) ([]*Query, error)
|
|
RemoveQueryFromPack(query *Query, pack *Pack) error
|
|
|
|
// Modifying the labels for packs
|
|
AddLabelToPack(lid uint, pid uint) error
|
|
ListLabelsForPack(pid uint) ([]*Label, error)
|
|
RemoveLabelFromPack(label *Label, pack *Pack) error
|
|
|
|
ListHostsInPack(pid uint, opt ListOptions) ([]*Host, error)
|
|
}
|
|
|
|
type PackService interface {
|
|
ListPacks(ctx context.Context, opt ListOptions) ([]*Pack, error)
|
|
GetPack(ctx context.Context, id uint) (*Pack, error)
|
|
NewPack(ctx context.Context, p PackPayload) (*Pack, error)
|
|
ModifyPack(ctx context.Context, id uint, p PackPayload) (*Pack, error)
|
|
DeletePack(ctx context.Context, id uint) error
|
|
|
|
AddQueryToPack(ctx context.Context, qid, pid uint) error
|
|
ListQueriesInPack(ctx context.Context, id uint) ([]*Query, error)
|
|
RemoveQueryFromPack(ctx context.Context, qid, pid uint) error
|
|
|
|
AddLabelToPack(ctx context.Context, lid, pid uint) error
|
|
ListLabelsForPack(ctx context.Context, pid uint) ([]*Label, error)
|
|
RemoveLabelFromPack(ctx context.Context, lid, pid uint) error
|
|
|
|
ListPacksForHost(ctx context.Context, hid uint) ([]*Pack, error)
|
|
ListHostsInPack(ctx context.Context, pid uint, opt ListOptions) ([]*Host, error)
|
|
}
|
|
|
|
type Pack struct {
|
|
UpdateCreateTimestamps
|
|
DeleteFields
|
|
ID uint `json:"id"`
|
|
Name string `json:"name"`
|
|
Description string `json:"description"`
|
|
Platform string `json:"platform"`
|
|
CreatedBy uint `json:"created_by" db:"created_by"`
|
|
Disabled bool `json:"disabled"`
|
|
}
|
|
|
|
type PackPayload struct {
|
|
Name *string
|
|
Description *string
|
|
Platform *string
|
|
Disabled *bool
|
|
}
|
|
|
|
type PackQuery struct {
|
|
ID uint
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
PackID uint
|
|
QueryID uint
|
|
}
|
|
|
|
type PackTarget struct {
|
|
ID uint
|
|
PackID uint
|
|
Target
|
|
}
|