feat: add snapshot ID as a starting point of a new branch (#599)

Author: agneum

engine/cmd/cli/commands/branch/actions.go

@@ -182,6 +182,11 @@ func create(cliCtx *cli.Context) error {
 	branchName := cliCtx.Args().First()
 
 	baseBranch := cliCtx.String("parent-branch")
+	snapshotID := cliCtx.String("snapshot-id")
+
+	if baseBranch != "" && snapshotID != "" {
+		return commands.NewActionError("either --parent-branch or --snapshot-id must be specified")
+	}
 
 	if baseBranch == "" {
 		baseBranch = getBaseBranch(cliCtx)
@@ -190,6 +195,7 @@ func create(cliCtx *cli.Context) error {
 	branchRequest := types.BranchCreateRequest{
 		BranchName: branchName,
 		BaseBranch: baseBranch,
+		SnapshotID: snapshotID,
 	}
 
 	branch, err := dblabClient.CreateBranch(cliCtx.Context, branchRequest)

engine/cmd/cli/commands/branch/command_list.go

@@ -21,7 +21,12 @@ func List() []*cli.Command {
 					Aliases: []string{"d"},
 				},
 				&cli.StringFlag{
-					Name: "parent-branch",
+					Name:  "parent-branch",
+					Usage: "specify branch name as starting point for new branch; cannot be used together with --snapshot-id",
+				},
+				&cli.StringFlag{
+					Name:  "snapshot-id",
+					Usage: "specify snapshot ID is starting point for new branch; cannot be used together with --parent-branch",
 				},
 			},
 			ArgsUsage: "BRANCH_NAME",