Print a warning on an unrecognised CDI hook

Signed-off-by: Carlos Eduardo Arango Gutierrez <[email protected]>

Author: ArangoGutierrez

cmd/nvidia-cdi-hook/main.go

@@ -19,6 +19,7 @@ package main
 import (
 	"context"
 	"os"
+	"strings"
 
 	"github.com/sirupsen/logrus"
 
@@ -72,6 +73,35 @@ func main() {
 			commands.IssueUnsupportedHookWarning(logger, cmd)
 			return nil
 		},
+		// Handle usage errors to catch unrecognized hooks
+		OnUsageError: func(ctx context.Context, cmd *cli.Command, err error, isSubcommand bool) error {
+			errMsg := err.Error()
+
+			// Check if this is a "No help topic for" error (unrecognized
+			// subcommand without flags)
+			if strings.HasPrefix(errMsg, "No help topic for") {
+				commands.IssueUnsupportedHookWarning(logger, cmd)
+				return nil // Suppress the error and exit cleanly
+			}
+
+			// Check if this is a "flag provided but not defined: -" error
+			// (unrecognized subcommand with flags)
+			if strings.HasPrefix(errMsg, "flag provided but not defined: -") {
+				// Get the arguments to check if the first one is an unrecognized command
+				args := cmd.Args().Slice()
+				if len(args) > 0 {
+					potentialCmd := args[0]
+					// Check if it's not a recognized subcommand
+					if cmd.Command(potentialCmd) == nil {
+						// This is an unrecognized hook with flags
+						commands.IssueUnsupportedHookWarning(logger, cmd)
+						return nil // Suppress the error and exit cleanly
+					}
+				}
+			}
+			// For other errors, return the error as-is
+			return err
+		},
 		// Define the subcommands
 		Commands: commands.New(logger),
 		Flags: []cli.Flag{