options.go

package runtime

import (
	"time"

	"google.golang.org/protobuf/types/known/timestamppb"
)

// Option is a functional option for configuring the App.
// It allows for applying configurations to the App instance at creation time.
type Option func(*App)

// WithEnv sets the environment for the application.
func WithEnv(env string) Option {
	return func(a *App) {
		// This option will be applied after the appInfo is initialized.
		a.appInfo.Env = env
	}
}

// WithID sets a custom instance ID.
func WithID(id string) Option {
	return func(a *App) {
		// This option will be applied after the appInfo is initialized.
		a.appInfo.Id = id
	}
}

// WithStartTime sets a custom start time.
func WithStartTime(startTime time.Time) Option {
	return func(a *App) {
		// This option will be applied after the appInfo is initialized.
		a.appInfo.StartTime = timestamppb.New(startTime)
	}
}

// WithMetadata adds a key-value pair to the application's metadata.
func WithMetadata(key, value string) Option {
	return func(a *App) {
		// This option will be applied after the appInfo is initialized.
		if a.appInfo.Metadata == nil {
			a.appInfo.Metadata = make(map[string]string)
		}
		a.appInfo.Metadata[key] = value
	}
}

// WithMetadataMap adds a map of key-value pairs to the application's metadata.
func WithMetadataMap(metadata map[string]string) Option {
	return func(a *App) {
		// This option will be applied after the appInfo is initialized.
		a.appInfo.Metadata = metadata
	}
}