photoprism/internal/config/cli_context.go

package config

import (
	"reflect"
	"strconv"
	"time"

	"github.com/urfave/cli/v2"

	"github.com/photoprism/photoprism/pkg/txt"
)

// ApplyCliContext applies the values of the cli context based on the "flag" annotations.
func ApplyCliContext(c interface{}, ctx *cli.Context) error {
	v := reflect.ValueOf(c).Elem()

	// Iterate through all config fields.
	for i := 0; i < v.NumField(); i++ {
		fieldValue := v.Field(i)

		tagValue := v.Type().Field(i).Tag.Get("flag")

		// Assign value to field with "flag" tag.
		if tagValue != "" && tagValue != "-" {
			switch t := fieldValue.Interface().(type) {
			case time.Duration:
				var s string

				// Get duration string.
				if ctx.IsSet(tagValue) || fieldValue.Interface().(time.Duration) == 0 {
					s = ctx.String(tagValue)
				}

				// Parse duration string.
				if s == "" {
					// Omit.
				} else if sec := txt.UInt(s); sec > 0 {
					fieldValue.Set(reflect.ValueOf(time.Duration(sec) * time.Second)) //nolint:gosec // txt.UInt is bounded; duration uses int64 on supported platforms
				} else if d, err := time.ParseDuration(s); err == nil {
					fieldValue.Set(reflect.ValueOf(d))
				}
			case float64:
				// Only if explicitly set or current value is empty (use default).
				if ctx.IsSet(tagValue) || fieldValue.Float() == 0 {
					f := ctx.Float64(tagValue)
					fieldValue.SetFloat(f)
				}
			case int, int64:
				// Only if explicitly set or current value is empty (use default).
				if ctx.IsSet(tagValue) || fieldValue.Int() == 0 {
					f := ctx.Int64(tagValue)
					fieldValue.SetInt(f)
				}
			case uint, uint64:
				// Only if explicitly set or current value is empty (use default).
				if ctx.IsSet(tagValue) || fieldValue.Uint() == 0 {
					f := ctx.Uint64(tagValue)
					fieldValue.SetUint(f)
				}
			case string:
				// Only if explicitly set or current value is empty (use default)
				if ctx.IsSet(tagValue) || fieldValue.String() == "" {
					f := ctx.String(tagValue)
					fieldValue.SetString(f)
				}
			case []string:
				if ctx.IsSet(tagValue) || fieldValue.Len() == 0 {
					f := reflect.ValueOf(ctx.StringSlice(tagValue))
					fieldValue.Set(f)
				}
			case []float64:
				if ctx.IsSet(tagValue) || fieldValue.Len() == 0 {
					var floats []float64

					for _, s := range ctx.StringSlice(tagValue) {
						if s == "" {
							continue
						}

						if f, err := strconv.ParseFloat(s, 64); err != nil {
							log.Warnf("cannot parse value %q for cli flag %s (%s)", s, tagValue, err)
							continue
						} else {
							floats = append(floats, f)
						}
					}

					fieldValue.Set(reflect.ValueOf(floats))
				}
			case bool:
				if ctx.IsSet(tagValue) {
					f := ctx.Bool(tagValue)
					fieldValue.SetBool(f)
				}
			default:
				log.Warnf("cannot assign value of type %s from cli flag %s", t, tagValue)
			}
		}
	}

	return nil
}