Including system metadata when posting to utility API

metadata.go

@@ -0,0 +1,84 @@
+package ctrld
+
+import (
+	"context"
+	"os"
+	"os/user"
+
+	"github.com/cuonglm/osinfo"
+
+	"github.com/Control-D-Inc/ctrld/internal/system"
+)
+
+const (
+	metadataOsKey                = "os"
+	metadataChassisTypeKey       = "chassis_type"
+	metadataChassisVendorKey     = "chassis_vendor"
+	metadataUsernameKey          = "username"
+	metadataDomainOrWorkgroupKey = "domain_or_workgroup"
+	metadataDomainKey            = "domain"
+)
+
+var (
+	chassisType   string
+	chassisVendor string
+)
+
+// SystemMetadata collects system and user-related SystemMetadata and returns it as a map.
+func SystemMetadata(ctx context.Context) map[string]string {
+	logger := LoggerFromCtx(ctx)
+	m := make(map[string]string)
+	oi := osinfo.New()
+	m[metadataOsKey] = oi.String()
+	if chassisType == "" && chassisVendor == "" {
+		ci, err := system.GetChassisInfo()
+		if err != nil {
+			logger.Debug().Err(err).Msg("Failed to get chassis info")
+		} else {
+			chassisType, chassisVendor = ci.Type, ci.Vendor
+		}
+	}
+	m[metadataChassisTypeKey] = chassisType
+	m[metadataChassisVendorKey] = chassisVendor
+	m[metadataUsernameKey] = currentLoginUser(ctx)
+	m[metadataDomainOrWorkgroupKey] = partOfDomainOrWorkgroup(ctx)
+	domain, err := system.GetActiveDirectoryDomain()
+	if err != nil {
+		logger.Debug().Err(err).Msg("Failed to get active directory domain name")
+	}
+	m[metadataDomainKey] = domain
+
+	return m
+}
+
+// currentLoginUser attempts to find the actual login user, even if the process is running as root.
+func currentLoginUser(ctx context.Context) string {
+	logger := LoggerFromCtx(ctx)
+
+	// 1. Check SUDO_USER: This is the most reliable way to find the original user
+	// when a script is run via 'sudo'.
+	if sudoUser := os.Getenv("SUDO_USER"); sudoUser != "" {
+		return sudoUser
+	}
+
+	// 2. Check general user login variables. LOGNAME is often preferred over USER.
+	if logName := os.Getenv("LOGNAME"); logName != "" {
+		return logName
+	}
+
+	// 3. Fallback to USER variable.
+	if userEnv := os.Getenv("USER"); userEnv != "" {
+		return userEnv
+	}
+
+	// 4. Final fallback: Use the standard library function to get the *effective* user.
+	// This will return "root" if the process is running as root.
+	currentUser, err := user.Current()
+	if err != nil {
+		// Handle error gracefully, returning a placeholder
+		logger.Debug().Err(err).Msg("Failed to get current user")
+		return "unknown"
+	}
+
+	return currentUser.Username
+}