refactor: Removed line preventing azldev to run as root (#369)

* Removed line preventing azldev to run as root

* Added cobra annotation to conditionally allow execution as root

* Fixed minor lint issue

Author: Antonio Salinas

cmd/azldev/azldev.go

@@ -4,7 +4,6 @@
 package main
 
 import (
-	"fmt"
 	"os"
 
 	"github.com/microsoft/azldev/internal/app/azldev"
@@ -18,13 +17,6 @@ import (
 )
 
 func main() {
-	// Make sure we're *not* running as root.
-	if os.Geteuid() == 0 {
-		fmt.Fprintln(os.Stderr,
-			"error: this tool may not be run as root; it will internally invoke sub-commands as sudo as needed")
-		os.Exit(1)
-	}
-
 	// Instantiate the main CLI app instance.
 	app := InstantiateApp()
 

internal/app/azldev/app.go

@@ -107,6 +107,12 @@ func NewApp(fsFactory opctx.FileSystemFactory, osEnvFactory opctx.OSEnvFactory)
 		Long:    usageInfo,
 		Version: displayVersion,
 		PersistentPreRunE: func(command *cobra.Command, _ []string) error {
+			slog.Debug("Command annotations", "annotations", command.Annotations)
+
+			if _, ok := command.Annotations[CommandAnnotationRootOK]; !ok && os.Geteuid() == 0 {
+				return errors.New("this command may not be run as root")
+			}
+
 			env, err := GetEnvFromCommand(command)
 			if err != nil {
 				return err

internal/app/azldev/cmds/component/preparesources.go

@@ -37,6 +37,9 @@ func NewPrepareSourcesCmd() *cobra.Command {
 			return nil, PrepareComponentSources(env, &options)
 		}),
 		ValidArgsFunction: components.GenerateComponentNameCompletions,
+		Annotations: map[string]string{
+			azldev.CommandAnnotationRootOK: "true",
+		},
 	}
 
 	components.AddComponentFilterOptionsToCommand(cmd, &options.ComponentFilter)

internal/app/azldev/command.go

@@ -16,6 +16,12 @@ import (
 	"github.com/spf13/cobra"
 )
 
+const (
+	// CommandAnnotationRootOK is a [cobra.Command.Annotations] key used to indicate that a command
+	// is allowed to be run as root.
+	CommandAnnotationRootOK = "rootOK"
+)
+
 const (
 	// CommandGroupMeta represents the command group for meta commands like help, completion, and version.
 	CommandGroupMeta = "meta"