func()

in lib/ec2macosinit/userdata.go [23:61]
31 lines of code
8 McCabe index (conditional complexity)

func (m *UserDataModule) Do(mctx *ModuleContext) (message string, err error) {
	const scriptFileName = "userdata"
	userdataScript := filepath.Join(mctx.InstanceHistoryPath(), scriptFileName)

	// Get user data from IMDS
	ud, respCode, err := mctx.IMDS.getIMDSProperty("user-data")
	if err != nil {
		return "", fmt.Errorf("ec2macosinit: error getting user data from IMDS: %s\n", err)
	}
	if respCode == 404 { // 404 = no user data provided, exit nicely
		return "no user data provided through IMDS", nil
	}
	if respCode != 200 { // 200 = ok
		return "", fmt.Errorf("ec2macosinit: received an unexpected response code from IMDS: %d - %s\n", respCode, err)
	}

	err = writeShellScript(userdataScript, userdataReader(ud))
	if err != nil {
		return "", fmt.Errorf("userdata script: %w", err)
	}

	// If we don't want to execute the user data, exit nicely - we're done
	if !m.ExecuteUserData {
		return "successfully handled user data with no execution request", nil
	}

	// Execute user data script
	out, err := executeCommand([]string{userdataScript}, "", []string{})
	if err != nil {
		if strings.Contains(err.Error(), "exec format error") {
			contentType := http.DetectContentType([]byte(ud))
			return fmt.Sprintf("provided user data is not executable (detected type: %s)", contentType), nil
		} else {
			return fmt.Sprintf("error while running user data with stdout: [%s] and stderr: [%s]", out.stdout, out.stderr), err
		}
	}

	return fmt.Sprintf("successfully ran user data with stdout: [%s] and stderr: [%s]", out.stdout, out.stderr), nil
}