fleet/server/kolide/packs.go

package kolide

import (
	"time"

	"golang.org/x/net/context"
)

type PackStore interface {
	// Pack methods
	NewPack(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(pack *Pack) ([]*Label, error)
	RemoveLabelFromPack(label *Label, pack *Pack) 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)
}

type Pack struct {
	ID        uint      `json:"id" gorm:"primary_key"`
	CreatedAt time.Time `json:"-"`
	UpdatedAt time.Time `json:"-"`
	Name      string    `json:"name" gorm:"not null;unique_index:idx_pack_unique_name"`
	Platform  string    `json:"platform"`
}

type PackQuery struct {
	ID        uint `gorm:"primary_key"`
	CreatedAt time.Time
	UpdatedAt time.Time
	PackID    uint
	QueryID   uint
}

type PackTarget struct {
	ID       uint `gorm:"primary_key"`
	PackID   uint
	Target
}