chore: hide update banner when pinned

speakeasy-api · Jan 22, 2025 · d30accf · d30accf
1 parent f7d2170
commit d30accf
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/cmd/root.go b/cmd/root.go
@@ -130,7 +130,7 @@ func setupRootCmd(version, artifactArch string) {
 		cmd.SetContext(ctx)
 
 		if !slices.Contains([]string{"update", "language-server"}, cmd.Name()) {
-			checkForUpdate(ctx, version, artifactArch)
+			checkForUpdate(ctx, version, artifactArch, cmd)
 		}
 
 		return setLogLevel(cmd)
@@ -143,7 +143,7 @@ func GetRootCommand() *cobra.Command {
 	return rootCmd
 }
 
-func checkForUpdate(ctx context.Context, currentVersion, artifactArch string) {
+func checkForUpdate(ctx context.Context, currentVersion, artifactArch string, cmd *cobra.Command) {
 	// Don't display if piping to a file for example
 	if !utils.IsInteractive() {
 		return
@@ -153,6 +153,11 @@ func checkForUpdate(ctx context.Context, currentVersion, artifactArch string) {
 		return
 	}
 
+	// When using the --pinned flag, don't display update notifications
+	if flag := cmd.Flag("pinned"); flag != nil && flag.Value.String() == "true" {
+		return
+	}
+
 	newerVersion, err := updates.GetNewerVersion(ctx, artifactArch, currentVersion)
 	if err != nil {
 		return // Don't display error to user