turbot · pskrbasu · Sep 16, 2025 · Aug 28, 2025 · Sep 10, 2025 · Sep 11, 2025
diff --git a/cmd/collect.go b/cmd/collect.go
@@ -17,14 +17,14 @@ import (
 	"github.com/turbot/pipe-fittings/v2/cmdconfig"
 	pconstants "github.com/turbot/pipe-fittings/v2/constants"
 	"github.com/turbot/pipe-fittings/v2/contexthelpers"
-	"github.com/turbot/pipe-fittings/v2/error_helpers"
 	"github.com/turbot/pipe-fittings/v2/modconfig"
 	"github.com/turbot/pipe-fittings/v2/parse"
 	localcmdconfig "github.com/turbot/tailpipe/internal/cmdconfig"
 	"github.com/turbot/tailpipe/internal/collector"
 	"github.com/turbot/tailpipe/internal/config"
 	"github.com/turbot/tailpipe/internal/constants"
 	"github.com/turbot/tailpipe/internal/database"
+	error_helpers "github.com/turbot/tailpipe/internal/error_helpers"
 	"github.com/turbot/tailpipe/internal/plugin"
 	"golang.org/x/exp/maps"
 )
@@ -71,8 +71,8 @@ func runCollectCmd(cmd *cobra.Command, args []string) {
 		}
 
 		if err != nil {
-			if errors.Is(err, context.Canceled) {
-				fmt.Println("Collection cancelled.") //nolint:forbidigo // ui output
+			if error_helpers.IsCancelledError(err) {
+				fmt.Println("tailpipe collect command cancelled.") //nolint:forbidigo // ui output
 			} else {
 				error_helpers.ShowError(ctx, err)
 			}
@@ -364,14 +364,13 @@ func setExitCodeForCollectError(err error) {
 	if exitCode != 0 || err == nil {
 		return
 	}
-	// TODO Set exit code for cancellation
-	if errors.Is(err, context.Canceled) {
-		exitCode = 0
+	// set exit code for cancellation
+	if error_helpers.IsCancelledError(err) {
+		exitCode = pconstants.ExitCodeOperationCancelled
 		return
 	}
 
-	// TODO #errors - assign exit codes https://github.com/turbot/tailpipe/issues/496
-	exitCode = 1
+	exitCode = pconstants.ExitCodeCollectionFailed
 }
 
 // parse the from time

diff --git a/cmd/compact.go b/cmd/compact.go
@@ -2,7 +2,6 @@ package cmd
 
 import (
 	"context"
-	"errors"
 	"fmt"
 	"log/slog"
 	"os"
@@ -15,11 +14,11 @@ import (
 	"github.com/turbot/pipe-fittings/v2/cmdconfig"
 	pconstants "github.com/turbot/pipe-fittings/v2/constants"
 	"github.com/turbot/pipe-fittings/v2/contexthelpers"
-	"github.com/turbot/pipe-fittings/v2/error_helpers"
 	localcmdconfig "github.com/turbot/tailpipe/internal/cmdconfig"
 	"github.com/turbot/tailpipe/internal/config"
 	"github.com/turbot/tailpipe/internal/constants"
 	"github.com/turbot/tailpipe/internal/database"
+	error_helpers "github.com/turbot/tailpipe/internal/error_helpers"
 	"golang.org/x/exp/maps"
 )
 
@@ -49,9 +48,9 @@ func runCompactCmd(cmd *cobra.Command, args []string) {
 		if err != nil {
 			setExitCodeForCompactError(err)
 
-			if errors.Is(err, context.Canceled) {
+			if error_helpers.IsCancelledError(err) {
 				//nolint:forbidigo // ui
-				fmt.Println("Compact cancelled")
+				fmt.Println("tailpipe compact command cancelled.")
 			} else {
 				error_helpers.ShowError(ctx, err)
 			}
@@ -126,5 +125,11 @@ func setExitCodeForCompactError(err error) {
 	if exitCode != 0 || err == nil {
 		return
 	}
-	exitCode = 1
+	// set exit code for cancellation
+	if error_helpers.IsCancelledError(err) {
+		exitCode = pconstants.ExitCodeOperationCancelled
+		return
+	}
+
+	exitCode = pconstants.ExitCodeCompactFailed
 }
diff --git a/cmd/connect.go b/cmd/connect.go
@@ -4,27 +4,29 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
-	"golang.org/x/exp/maps"
 	"log"
 	"os"
 	"path/filepath"
 	"strings"
 	"time"
 
+	"golang.org/x/exp/maps"
+
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 	"github.com/thediveo/enumflag/v2"
 	"github.com/turbot/go-kit/helpers"
 	"github.com/turbot/pipe-fittings/v2/cmdconfig"
 	"github.com/turbot/pipe-fittings/v2/connection"
 	pconstants "github.com/turbot/pipe-fittings/v2/constants"
-	"github.com/turbot/pipe-fittings/v2/error_helpers"
+	"github.com/turbot/pipe-fittings/v2/contexthelpers"
 	pfilepaths "github.com/turbot/pipe-fittings/v2/filepaths"
 	"github.com/turbot/pipe-fittings/v2/parse"
 	localcmdconfig "github.com/turbot/tailpipe/internal/cmdconfig"
 	"github.com/turbot/tailpipe/internal/config"
 	"github.com/turbot/tailpipe/internal/constants"
 	"github.com/turbot/tailpipe/internal/database"
+	error_helpers "github.com/turbot/tailpipe/internal/error_helpers"
 )
 
 // variable used to assign the output mode flag
@@ -94,13 +96,21 @@ The generated script can be used with DuckDB:
 func runConnectCmd(cmd *cobra.Command, _ []string) {
 	var err error
 	var initFilePath string
-	ctx := cmd.Context()
+	ctx, cancel := context.WithCancel(cmd.Context())
+	contexthelpers.StartCancelHandler(cancel)
 
 	defer func() {
 		if r := recover(); r != nil {
 			err = helpers.ToError(r)
 		}
-		setExitCodeForConnectError(err)
+		if err != nil {
+			if error_helpers.IsCancelledError(err) {
+				fmt.Println("tailpipe connect command cancelled.") //nolint:forbidigo // ui output
+			} else {
+				error_helpers.ShowError(ctx, err)
+			}
+			setExitCodeForConnectError(err)
+		}
 		displayOutput(ctx, initFilePath, err)
 	}()
 
@@ -409,8 +419,11 @@ func setExitCodeForConnectError(err error) {
 	if exitCode != 0 || err == nil || viper.GetString(pconstants.ArgOutput) == pconstants.OutputFormatJSON {
 		return
 	}
-
-	exitCode = 1
+	if error_helpers.IsCancelledError(err) {
+		exitCode = pconstants.ExitCodeOperationCancelled
+		return
+	}
+	exitCode = pconstants.ExitCodeConnectFailed
 }
 
 // generateInitFilename generates a temporary filename with a timestamp

diff --git a/cmd/format.go b/cmd/format.go
@@ -12,12 +12,12 @@ import (
 	"github.com/turbot/pipe-fittings/v2/cmdconfig"
 	pconstants "github.com/turbot/pipe-fittings/v2/constants"
 	"github.com/turbot/pipe-fittings/v2/contexthelpers"
-	"github.com/turbot/pipe-fittings/v2/error_helpers"
 	"github.com/turbot/pipe-fittings/v2/printers"
 	"github.com/turbot/pipe-fittings/v2/utils"
 	localcmdconfig "github.com/turbot/tailpipe/internal/cmdconfig"
 	"github.com/turbot/tailpipe/internal/constants"
 	"github.com/turbot/tailpipe/internal/display"
+	error_helpers "github.com/turbot/tailpipe/internal/error_helpers"
 )
 
 // variable used to assign the output mode flag
@@ -73,11 +73,20 @@ func runFormatListCmd(cmd *cobra.Command, args []string) {
 	ctx, cancel := context.WithCancel(cmd.Context())
 	contexthelpers.StartCancelHandler(cancel)
 	utils.LogTime("runFormatListCmd start")
+	var err error
 	defer func() {
 		utils.LogTime("runFormatListCmd end")
 		if r := recover(); r != nil {
-			error_helpers.ShowError(ctx, helpers.ToError(r))
-			exitCode = pconstants.ExitCodeUnknownErrorPanic
+			err = helpers.ToError(r)
+		}
+		if err != nil {
+			if error_helpers.IsCancelledError(err) {
+				//nolint:forbidigo // ui output
+				fmt.Println("tailpipe format list command cancelled.")
+			} else {
+				error_helpers.ShowError(ctx, err)
+			}
+			setExitCodeForFormatError(err,1)
 		}
 	}()
 
@@ -99,8 +108,8 @@ func runFormatListCmd(cmd *cobra.Command, args []string) {
 	// Print
 	err = printer.PrintResource(ctx, printableResource, cmd.OutOrStdout())
 	if err != nil {
-		error_helpers.ShowError(ctx, err)
-		exitCode = pconstants.ExitCodeUnknownErrorPanic
+		exitCode = pconstants.ExitCodeOutputRenderingFailed
+		return
 	}
 }
 
@@ -128,11 +137,20 @@ func runFormatShowCmd(cmd *cobra.Command, args []string) {
 	ctx, cancel := context.WithCancel(cmd.Context())
 	contexthelpers.StartCancelHandler(cancel)
 	utils.LogTime("runFormatShowCmd start")
+	var err error
 	defer func() {
 		utils.LogTime("runFormatShowCmd end")
 		if r := recover(); r != nil {
-			error_helpers.ShowError(ctx, helpers.ToError(r))
-			exitCode = pconstants.ExitCodeUnknownErrorPanic
+			err = helpers.ToError(r)
+		}
+		if err != nil {
+			if error_helpers.IsCancelledError(err) {
+				//nolint:forbidigo // ui output
+				fmt.Println("tailpipe format show command cancelled.")
+			} else {
+				error_helpers.ShowError(ctx, err)
+			}
+			setExitCodeForFormatError(err, 1)
 		}
 	}()
 
@@ -149,7 +167,21 @@ func runFormatShowCmd(cmd *cobra.Command, args []string) {
 	// Print
 	err = printer.PrintResource(ctx, printableResource, cmd.OutOrStdout())
 	if err != nil {
-		error_helpers.ShowError(ctx, err)
-		exitCode = pconstants.ExitCodeUnknownErrorPanic
+		exitCode = pconstants.ExitCodeOutputRenderingFailed
+		return
+	}
+}
+
+func setExitCodeForFormatError(err error, nonCancelCode int) {
+	// set exit code only if an error occurred and no exit code is already set
+	if exitCode != 0 || err == nil {
+		return
+	}
+	// set exit code for cancellation
+	if error_helpers.IsCancelledError(err) {
+		exitCode = pconstants.ExitCodeOperationCancelled
+		return
 	}
+	// no dedicated format exit code exists yet; use generic nonzero failure
+	exitCode = nonCancelCode
 }