mirror of
https://github.com/SecurityBrewery/catalyst.git
synced 2025-12-06 07:12:46 +01:00
37 lines
795 B
Go
37 lines
795 B
Go
package app
|
|
|
|
import (
|
|
"github.com/pocketbase/pocketbase/core"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/SecurityBrewery/catalyst/fakedata"
|
|
)
|
|
|
|
func fakeDataCmd(app core.App) *cobra.Command {
|
|
var userCount, ticketCount int
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "fake-data",
|
|
RunE: func(_ *cobra.Command, _ []string) error {
|
|
return fakedata.Generate(app, userCount, ticketCount)
|
|
},
|
|
}
|
|
|
|
cmd.PersistentFlags().IntVar(&userCount, "users", 10, "Number of users to generate")
|
|
|
|
cmd.PersistentFlags().IntVar(&ticketCount, "tickets", 100, "Number of tickets to generate")
|
|
|
|
return cmd
|
|
}
|
|
|
|
func defaultDataCmd(app core.App) *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "default-data",
|
|
RunE: func(_ *cobra.Command, _ []string) error {
|
|
return fakedata.GenerateDefaultData(app)
|
|
},
|
|
}
|
|
|
|
return cmd
|
|
}
|