Allow forking w/o group flag

Author: m0hossam

internal/cmd/db_create.go

@@ -74,9 +74,37 @@ func CreateDatabase(name string) error {
 		return err
 	}
 
-	group, err := groupFromFlag(groups)
-	if err != nil {
-		return err
+	var seed *turso.DBSeed
+	var group turso.Group
+
+	isFromDB := checkIfFromDBFlag()
+	if isFromDB {
+		seed, err = parseDBSeedFlags(client, false)
+		if err != nil {
+			return err
+		}
+
+		db, err := getDatabase(client, seed.Name)
+		if err != nil {
+			return err
+		}
+
+		for _, g := range groups {
+			if g.Name == db.Group {
+				group = g
+			}
+		}
+
+		if groupFlag != "" && groupFlag != group.Name {
+			return fmt.Errorf("group %s does not match group %s of the source database. Please fork within the same group", groupFlag, group.Name)
+		}
+	}
+
+	if !isFromDB {
+		group, err = groupFromFlag(groups)
+		if err != nil {
+			return err
+		}
 	}
 	groupName := group.Name
 
@@ -85,10 +113,12 @@ func CreateDatabase(name string) error {
 		return err
 	}
 
-	isAWS := strings.HasPrefix(group.Primary, "aws-")
-	seed, err := parseDBSeedFlags(client, isAWS)
-	if err != nil {
-		return err
+	if !isFromDB {
+		isAWS := strings.HasPrefix(group.Primary, "aws-")
+		seed, err = parseDBSeedFlags(client, isAWS)
+		if err != nil {
+			return err
+		}
 	}
 
 	version := "latest"

internal/cmd/group_flag.go

@@ -49,6 +49,10 @@ func addFromDBFlag(cmd *cobra.Command) {
 	cmd.Flags().StringVar(&timestampFlag, "timestamp", "", "Set a point in time in the past to copy data from the selected database. Must be used with the 'from-db' flag. Must be in RFC3339 format like '2023-09-29T10:16:13-03:00'")
 }
 
+func checkIfFromDBFlag() bool {
+	return fromDBFlag != ""
+}
+
 func parseTimestampFlag() (*time.Time, error) {
 	if timestampFlag == "" {
 		return nil, nil