Fix homedir issue

Author: owenthereal

cmd/candy/cmd/root.go

@@ -25,34 +25,33 @@ var (
 )
 
 func init() {
-	var err error
-	homeDir, err = userHomeDir()
-	if err != nil {
-		candy.Log().Fatal("error getting user home directory", zap.Error(err))
-	}
-
-	rootCmd.PersistentFlags().StringVar(&flagRootCfgFile, "config", filepath.Join(homeDir, ".candyconfig"), "Config file")
+	rootCmd.PersistentFlags().StringVar(&flagRootCfgFile, "config", filepath.Join(userHomeDir(), ".candyconfig"), "Config file")
 }
 
-func userHomeDir() (string, error) {
+func userHomeDir() string {
+	if homeDir != "" {
+		return homeDir
+	}
+
 	var (
 		sudo = os.Getenv("SUDO_USER")
 		euid = os.Geteuid()
+		err  error
 	)
 
 	if sudo != "" && euid == 0 {
 		u, err := user.Lookup(sudo)
 		if err != nil {
-			return "", nil
+			candy.Log().Fatal("error looking up sudo user", zap.String("user", sudo), zap.Error(err))
 		}
 
-		return u.HomeDir, nil
+		return u.HomeDir
 	}
 
-	homeDir, err := os.UserHomeDir()
+	homeDir, err = os.UserHomeDir()
 	if err != nil {
-		return "", err
+		candy.Log().Fatal("error getting user home directory", zap.Error(err))
 	}
 
-	return homeDir, nil
+	return homeDir
 }

cmd/candy/cmd/run.go

@@ -32,7 +32,7 @@ func init() {
 }
 
 func addServerFlags(cmd *cobra.Command) {
-	cmd.Flags().String("host-root", filepath.Join(homeDir, ".candy"), "Path to the directory containing applications that will be served by Candy")
+	cmd.Flags().String("host-root", filepath.Join(userHomeDir(), ".candy"), "Path to the directory containing applications that will be served by Candy")
 	cmd.Flags().StringSlice("domain", defaultDomains, "The top-level domains for which Candy will respond to DNS queries")
 	cmd.Flags().String("http-addr", ":28080", "The Proxy server HTTP address")
 	cmd.Flags().String("https-addr", ":28443", "The Proxy server HTTPS address")
@@ -66,7 +66,7 @@ func startServer(c *cobra.Command, ctx context.Context) error {
 	candy.Log().Info("using config", zap.Any("cfg", cfg))
 
 	if err := os.MkdirAll(cfg.HostRoot, 0o0755); err != nil {
-		return fmt.Errorf("failed to create host directory: %w", err)
+		return fmt.Errorf("failed to create host directory %s: %w", cfg.HostRoot, err)
 	}
 
 	svr := server.New(cfg)