mirror of https://github.com/jetkvm/kvm.git
feat: add env utils
This commit is contained in:
parent
ac9999171b
commit
a494f2f15f
|
|
@ -0,0 +1,92 @@
|
||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"reflect"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
func MarshalEnv(instance interface{}) ([]string, error) {
|
||||||
|
v := reflect.ValueOf(instance)
|
||||||
|
if v.Kind() == reflect.Ptr {
|
||||||
|
if v.IsNil() {
|
||||||
|
return nil, fmt.Errorf("instance is nil")
|
||||||
|
}
|
||||||
|
v = v.Elem()
|
||||||
|
}
|
||||||
|
|
||||||
|
if v.Kind() != reflect.Struct {
|
||||||
|
return nil, fmt.Errorf("instance must be a struct or pointer to struct")
|
||||||
|
}
|
||||||
|
|
||||||
|
t := v.Type()
|
||||||
|
var result []string
|
||||||
|
|
||||||
|
for i := 0; i < v.NumField(); i++ {
|
||||||
|
field := t.Field(i)
|
||||||
|
fieldValue := v.Field(i)
|
||||||
|
|
||||||
|
// Get the env tag
|
||||||
|
envTag := field.Tag.Get("env")
|
||||||
|
if envTag == "" || envTag == "-" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Skip unexported fields
|
||||||
|
if !fieldValue.CanInterface() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
var valueStr string
|
||||||
|
|
||||||
|
// Handle different types
|
||||||
|
switch fieldValue.Kind() {
|
||||||
|
case reflect.Bool:
|
||||||
|
valueStr = strconv.FormatBool(fieldValue.Bool())
|
||||||
|
|
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
||||||
|
valueStr = strconv.FormatUint(fieldValue.Uint(), 10)
|
||||||
|
|
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
||||||
|
valueStr = strconv.FormatInt(fieldValue.Int(), 10)
|
||||||
|
|
||||||
|
case reflect.Float32, reflect.Float64:
|
||||||
|
valueStr = strconv.FormatFloat(fieldValue.Float(), 'f', -1, 64)
|
||||||
|
|
||||||
|
case reflect.String:
|
||||||
|
valueStr = fieldValue.String()
|
||||||
|
|
||||||
|
case reflect.Ptr:
|
||||||
|
if fieldValue.IsNil() {
|
||||||
|
continue // Skip nil pointers
|
||||||
|
}
|
||||||
|
elem := fieldValue.Elem()
|
||||||
|
// Handle *semver.Version and other pointer types
|
||||||
|
if elem.CanInterface() {
|
||||||
|
if stringer, ok := elem.Interface().(fmt.Stringer); ok {
|
||||||
|
valueStr = stringer.String()
|
||||||
|
} else {
|
||||||
|
valueStr = fmt.Sprintf("%v", elem.Interface())
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
valueStr = fmt.Sprintf("%v", elem.Interface())
|
||||||
|
}
|
||||||
|
|
||||||
|
default:
|
||||||
|
// For other types, try to convert to string
|
||||||
|
if fieldValue.CanInterface() {
|
||||||
|
if stringer, ok := fieldValue.Interface().(fmt.Stringer); ok {
|
||||||
|
valueStr = stringer.String()
|
||||||
|
} else {
|
||||||
|
valueStr = fmt.Sprintf("%v", fieldValue.Interface())
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
valueStr = fmt.Sprintf("%v", fieldValue.Interface())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
result = append(result, fmt.Sprintf("%s=%s", envTag, valueStr))
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,57 @@
|
||||||
|
package utils
|
||||||
|
|
||||||
|
import (
|
||||||
|
"reflect"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/Masterminds/semver/v3"
|
||||||
|
)
|
||||||
|
|
||||||
|
type nativeOptions struct {
|
||||||
|
Disable bool `env:"JETKVM_NATIVE_DISABLE"`
|
||||||
|
SystemVersion *semver.Version `env:"JETKVM_NATIVE_SYSTEM_VERSION"`
|
||||||
|
AppVersion *semver.Version `env:"JETKVM_NATIVE_APP_VERSION"`
|
||||||
|
DisplayRotation uint16 `env:"JETKVM_NATIVE_DISPLAY_ROTATION"`
|
||||||
|
DefaultQualityFactor float64 `env:"JETKVM_NATIVE_DEFAULT_QUALITY_FACTOR"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMarshalEnv(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
instance interface{}
|
||||||
|
want []string
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "basic struct",
|
||||||
|
instance: nativeOptions{
|
||||||
|
Disable: false,
|
||||||
|
SystemVersion: semver.MustParse("1.1.0"),
|
||||||
|
AppVersion: semver.MustParse("1111.0.0"),
|
||||||
|
DisplayRotation: 1,
|
||||||
|
DefaultQualityFactor: 1.0,
|
||||||
|
},
|
||||||
|
want: []string{
|
||||||
|
"JETKVM_NATIVE_DISABLE=false",
|
||||||
|
"JETKVM_NATIVE_SYSTEM_VERSION=1.1.0",
|
||||||
|
"JETKVM_NATIVE_APP_VERSION=1111.0.0",
|
||||||
|
"JETKVM_NATIVE_DISPLAY_ROTATION=1",
|
||||||
|
"JETKVM_NATIVE_DEFAULT_QUALITY_FACTOR=1",
|
||||||
|
},
|
||||||
|
wantErr: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
got, err := MarshalEnv(tt.instance)
|
||||||
|
if (err != nil) != tt.wantErr {
|
||||||
|
t.Errorf("MarshalEnv() error = %v, wantErr %v", err, tt.wantErr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !reflect.DeepEqual(got, tt.want) {
|
||||||
|
t.Errorf("MarshalEnv() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue