commands

Documentation

Index

• Constants
• Variables

Constants

const (
	MigrationsDirectory = "db/migrations"
)

Variables

var DBMigrate = &cobra.Command{
	Use:     "db:migrate",
	Aliases: []string{"dbm"},
	Short:   "Run database migrations",
	Run: func(cmd *cobra.Command, _ []string) {
		var dir string
		databaseURL := f.GetEnvOrString("DATABASE_URL", "")

		if f.IsProductionEnv() {
			dir = cmd.Flag("dir").Value.String()
			if dir == "" {
				log.Fatal("You should specify the directory containing migrations with the `--dir` flag")
			}
		} else {
			if !h.BuiltOnFoundation() {
				log.Fatal("This command must be run from inside a Foundation service")
			}

			dir = h.AtServiceRoot(MigrationsDirectory)
		}

		_, err := os.Stat(dir)
		if os.IsNotExist(err) {
			log.Fatalf("Migrations directory `%s` does not exist", dir)
		}

		if databaseURL == "" {
			log.Fatal("`DATABASE_URL` environment variable is not set")
		}

		m, err := migrate.New(fmt.Sprintf("file://%s", dir), databaseURL)
		if err != nil {
			log.Fatal(err)
		}

		if err = m.Up(); err != nil {
			log.Fatal(err)
		}
	},
}

var DBRollback = &cobra.Command{
	Use:     "db:rollback",
	Aliases: []string{"dbr"},
	Short:   "Rollback database migrations",
	Long:    "Rollback database migrations by a given number of steps, e.g.: `foundation db:rollback --steps 2`",
	Run: func(cmd *cobra.Command, _ []string) {
		var dir string
		databaseURL := f.GetEnvOrString("DATABASE_URL", "")

		if f.IsProductionEnv() {
			dir = cmd.Flag("dir").Value.String()
			if dir == "" {
				log.Fatal("You should specify the directory containing migrations with the `--dir` flag")
			}
		} else {
			if !h.BuiltOnFoundation() {
				log.Fatal("This command must be run from inside a Foundation service")
			}

			dir = h.AtServiceRoot(MigrationsDirectory)
		}

		_, err := os.Stat(dir)
		if os.IsNotExist(err) {
			log.Fatalf("Migrations directory `%s` does not exist", dir)
		}

		steps, err := cmd.Flags().GetInt("step")
		if err != nil || steps <= 0 {
			log.Fatal("You should set `--steps` flag to a positive integer")
		}

		if databaseURL == "" {
			log.Fatal("`DATABASE_URL` environment variable is not set")
		}

		m, err := migrate.New(fmt.Sprintf("file://%s", dir), databaseURL)
		if err != nil {
			log.Fatal(err)
		}

		if err = m.Steps(-1 * steps); err != nil {
			log.Fatal(err)
		}
	},
}

var New = &cobra.Command{
	Use:     "new",
	Aliases: []string{"n"},
	Short:   "Create a new Foundation application or service",
	Run: func(cmd *cobra.Command, _ []string) {
		serviceFlag, err := cmd.Flags().GetBool("service")
		if err != nil {
			log.Fatal().Err(err).Msg("Failed to get `--service` flag")
		}

		input := &newInput{
			FoundationVersion: f.Version,
			Name:              cmd.Flag("name").Value.String(),
		}

		if serviceFlag {
			newService(input)
		} else {
			newApplication(input)
		}
	},
}

var Start = &cobra.Command{
	Use:     "start",
	Aliases: []string{"s"},
	Short:   "Start a service",
	Run: func(_ *cobra.Command, _ []string) {
		if !h.BuiltOnFoundation() {
			log.Fatal("This command must be run from inside a Foundation service")
		}

		files, err := os.ReadDir(h.AtServiceRoot("cmd"))
		if err != nil {
			log.Fatal(err)
		}

		// Collect all directory names
		var binaries []string
		for _, f := range files {
			if f.IsDir() {
				binaries = append(binaries, f.Name())
			}
		}

		prompt := &survey.Select{
			Message: "Choose a service to start:",
			Options: binaries,
		}
		var binaryName string
		if err = survey.AskOne(prompt, &binaryName); err != nil {
			log.Fatal(err)
		}

		svc := exec.Command("go", "run", h.AtServiceRoot("cmd", binaryName))
		svc.Stdout = os.Stdout
		svc.Stderr = os.Stderr
		if err = svc.Run(); err != nil {
			log.Fatal(err)
		}
	},
}

var Test = &cobra.Command{
	Use:     "test",
	Aliases: []string{"t"},
	Short:   "Run tests",
	Run: func(cmd *cobra.Command, _ []string) {
		if !h.BuiltOnFoundation() {
			log.Fatal("This command must be run from inside a Foundation service")
		}

		if f.IsProductionEnv() {
			log.Fatal("You're trying to run tests in production environment")
		}

		env := os.Environ()
		env = append(env, "FOUNDATION_ENV=test")

		opts := []string{
			"test",
		}
		if cmd.Flag("opts") != nil {
			opts = append(opts, strings.Split(cmd.Flag("opts").Value.String(), " ")...)
		}
		opts = append(opts, h.AtServiceRoot("..."))

		test := exec.Command("go", opts...)
		test.Stdout = os.Stdout
		test.Stderr = os.Stderr
		test.Env = env
		if err := test.Run(); err != nil {
			log.Fatal(err)
		}
	},
}