Allow specifying vector version when creating groups (#869)

Also add vector groups to the CLI completion script

Author: Athos

internal/cmd/canary_flag.go

@@ -6,4 +6,5 @@ var canaryFlag bool
 
 func addCanaryFlag(cmd *cobra.Command) {
 	cmd.Flags().BoolVar(&canaryFlag, "canary", false, "Use database canary build.")
+	_ = cmd.Flags().MarkHidden("canary")
 }

internal/cmd/group.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/spf13/cobra"
 	"github.com/tursodatabase/turso-cli/internal"
+	"github.com/tursodatabase/turso-cli/internal/flags"
 	"github.com/tursodatabase/turso-cli/internal/prompt"
 	"github.com/tursodatabase/turso-cli/internal/turso"
 )
@@ -23,6 +24,7 @@ func init() {
 	addLocationFlag(groupsCreateCmd, "Create the group primary in the specified location")
 	addWaitFlag(groupsCreateCmd, "Wait for group to be ready")
 	addCanaryFlag(groupsCreateCmd)
+	flags.AddVersion(groupsCreateCmd, "Version of the group. Valid values: 'latest', 'canary' or 'vector'")
 	groupCmd.AddCommand(groupsDestroyCmd)
 	addYesFlag(groupsDestroyCmd, "Confirms the destruction of the group, with all its locations and databases.")
 	groupCmd.AddCommand(groupShowCmd)
@@ -94,8 +96,11 @@ var groupsCreateCmd = &cobra.Command{
 			return fmt.Errorf("location '%s' is not a valid one", location)
 		}
 
-		version := "latest"
+		version := flags.Version()
 		if canaryFlag {
+			if version != "" {
+				return fmt.Errorf("cannot specify both --canary and --version flags")
+			}
 			version = "canary"
 		}
 

internal/cmd/group_update.go

@@ -13,7 +13,7 @@ import (
 func init() {
 	groupCmd.AddCommand(groupUpdateCmd)
 	flags.AddYes(groupUpdateCmd, "Confirms the update")
-	flags.AddVersion(groupUpdateCmd, "Version to update to. Valid values: 'latest' or 'canary'")
+	flags.AddVersion(groupUpdateCmd, "Version to update to. Valid values: 'latest', 'canary' or 'vector'")
 	flags.AddExtensions(groupUpdateCmd, "Extensions to enable. Valid values: 'all' or 'none'")
 }
 

internal/flags/version.go

@@ -9,7 +9,7 @@ var versionFlag string
 func AddVersion(cmd *cobra.Command, desc string) {
 	cmd.Flags().StringVar(&versionFlag, "version", "", desc)
 	_ = cmd.RegisterFlagCompletionFunc("version", func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
-		return []string{"latest", "canary"}, cobra.ShellCompDirectiveNoFileComp
+		return []string{"latest", "canary", "vector"}, cobra.ShellCompDirectiveNoFileComp
 	})
 }