From f80be6fb7b6783419cc4663645f890e4f78799fe Mon Sep 17 00:00:00 2001 From: Damandeep Singh Date: Fri, 7 Mar 2025 14:34:02 +0530 Subject: [PATCH] feat: Added disclaimer for branch checkout --- .../doltcore/sqle/dprocedures/dolt_checkout.go | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/go/libraries/doltcore/sqle/dprocedures/dolt_checkout.go b/go/libraries/doltcore/sqle/dprocedures/dolt_checkout.go index 71f4208eb6..6dcaa1b356 100644 --- a/go/libraries/doltcore/sqle/dprocedures/dolt_checkout.go +++ b/go/libraries/doltcore/sqle/dprocedures/dolt_checkout.go @@ -17,10 +17,11 @@ package dprocedures import ( "errors" "fmt" - + "os" + "strings" "github.com/dolthub/go-mysql-server/sql" "github.com/dolthub/go-mysql-server/sql/types" - + "github.com/fatih/color" "github.com/dolthub/dolt/go/cmd/dolt/cli" "github.com/dolthub/dolt/go/cmd/dolt/errhand" "github.com/dolthub/dolt/go/libraries/doltcore/doltdb" @@ -34,6 +35,7 @@ import ( ) var ErrEmptyBranchName = errors.New("error: cannot checkout empty string") +var os_args=os.Args[1:] var doltCheckoutSchema = []*sql.Column{ { @@ -176,6 +178,10 @@ func doDoltCheckout(ctx *sql.Context, args []string) (statusCode int, successMes if err != nil { return 1, "", err } + terminal_command:=strings.Join(os_args, " ") + if strings.Contains(terminal_command, "sql") { + cli.PrintErrln(color.YellowString("WARNING: The checked out branch is only changed for this current session.")) + } return 0, generateSuccessMessage(branchName, ""), nil }