mirror of
https://github.com/empayre/fleet.git
synced 2024-11-06 17:05:18 +00:00
103 lines
2.4 KiB
Go
103 lines
2.4 KiB
Go
package cli
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/WatchBeam/clock"
|
|
kitlog "github.com/go-kit/kit/log"
|
|
"github.com/kolide/kolide-ose/server/config"
|
|
"github.com/kolide/kolide-ose/server/datastore"
|
|
"github.com/kolide/kolide-ose/server/kolide"
|
|
"github.com/kolide/kolide-ose/server/service"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func createPrepareCmd(configManager config.Manager) *cobra.Command {
|
|
|
|
var prepareCmd = &cobra.Command{
|
|
Use: "prepare",
|
|
Short: "Subcommands for initializing kolide infrastructure",
|
|
Long: `
|
|
Subcommands for initializing kolide infrastructure
|
|
|
|
To setup kolide infrastructure, use one of the available commands.
|
|
`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
cmd.Help()
|
|
},
|
|
}
|
|
|
|
var dbCmd = &cobra.Command{
|
|
Use: "db",
|
|
Short: "Given correct database configurations, prepare the databases for use",
|
|
Long: ``,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
config := configManager.LoadConfig()
|
|
connString := datastore.GetMysqlConnectionString(config.Mysql)
|
|
|
|
ds, err := datastore.New("gorm-mysql", connString)
|
|
if err != nil {
|
|
initFatal(err, "creating db connection")
|
|
}
|
|
|
|
if err := ds.Drop(); err != nil {
|
|
initFatal(err, "dropping db tables")
|
|
}
|
|
|
|
if err := ds.Migrate(); err != nil {
|
|
initFatal(err, "migrating db schema")
|
|
}
|
|
},
|
|
}
|
|
|
|
prepareCmd.AddCommand(dbCmd)
|
|
|
|
var testDataCmd = &cobra.Command{
|
|
Use: "test-data",
|
|
Short: "Generate test data",
|
|
Long: ``,
|
|
Run: func(cmd *cobra.Command, arg []string) {
|
|
config := configManager.LoadConfig()
|
|
connString := datastore.GetMysqlConnectionString(config.Mysql)
|
|
|
|
ds, err := datastore.New("gorm-mysql", connString)
|
|
if err != nil {
|
|
initFatal(err, "creating db connection")
|
|
}
|
|
if err != nil {
|
|
initFatal(err, "creating new service")
|
|
}
|
|
var (
|
|
name = "admin"
|
|
username = "admin"
|
|
password = "secret"
|
|
email = "admin@kolide.co"
|
|
enabled = true
|
|
isAdmin = true
|
|
)
|
|
admin := kolide.UserPayload{
|
|
Name: &name,
|
|
Username: &username,
|
|
Password: &password,
|
|
Email: &email,
|
|
Enabled: &enabled,
|
|
Admin: &isAdmin,
|
|
}
|
|
svc, err := service.NewService(ds, kitlog.NewNopLogger(), config, nil, clock.C)
|
|
if err != nil {
|
|
initFatal(err, "creating service")
|
|
}
|
|
|
|
_, err = svc.NewUser(context.Background(), admin)
|
|
if err != nil {
|
|
initFatal(err, "saving new user")
|
|
}
|
|
},
|
|
}
|
|
|
|
prepareCmd.AddCommand(testDataCmd)
|
|
|
|
return prepareCmd
|
|
|
|
}
|