feat(proxy): validate sampler info (#48)

src/app/command/bootstrap.go

@@ -64,6 +64,7 @@ type Bootstrap struct {
 	Container   *assistant.CobraContainer
 	optionsInfo ConfigureOptionsInfo
 	profiles    proxy.ConfiguredProfiles
+	sampler     proxy.SamplerInfo
 }
 
 type ConfigureOptionsInfo struct {

src/app/command/root-cmd.go

@@ -81,16 +81,6 @@ func (b *Bootstrap) buildRootCommand(container *assistant.CobraContainer) {
 	container.MustRegisterParamSet(WorkerPoolFamName, workerPoolFam)
 	container.MustRegisterParamSet(FoldersFamName, foldersFam)
 	container.MustRegisterParamSet(ProfileFamName, profileFam)
-
-	// This needs to be implemented in extendio, so we can't bind
-	// here with a validated binder, we just perform a check here
-	// after the bind as a temporary measure. Actually, this
-	// needs to be implemented inside cross field validation.
-
-	// Ideally, the ProfileParameterSet would perform a check against
-	// the config, but extendio is not aware of config, so it can't
-	// check. Instead, we can put this check into
-	b.profiles = b.optionsInfo.Config.Viper.GetStringMapStringSlice("profiles")
 }
 
 func (b *Bootstrap) getRootInputs() *proxy.RootCommandInputs {

src/app/command/shrink-cmd.go

@@ -131,6 +131,14 @@ func (b *Bootstrap) buildShrinkCommand(container *assistant.CobraContainer) *cob
 						return err
 					}
 
+					// validate the scheme
+					//
+					_ = inputs.ParamSet.Native.Scheme
+
+					if err := b.sampler.Validate(inputs.ParamSet.Native.Scheme, &b.profiles); err != nil {
+						return err
+					}
+
 					appErr = proxy.EnterShrink(
 						inputs,
 						b.optionsInfo.Program,
@@ -201,8 +209,8 @@ func (b *Bootstrap) buildShrinkCommand(container *assistant.CobraContainer) *cob
 		defaultScheme = ""
 	)
 
-	samplers := b.optionsInfo.Config.Viper.Get("samplers")
-	_ = samplers
+	sampler := b.optionsInfo.Config.Viper.Get("sampler")
+	_ = sampler
 
 	// can we validate that it is present in the config?
 	paramSet.BindValidatedString(
@@ -212,9 +220,9 @@ func (b *Bootstrap) buildShrinkCommand(container *assistant.CobraContainer) *cob
 		),
 		&paramSet.Native.Scheme,
 		func(s string, f *pflag.Flag) error {
-			return lo.TernaryF(samplers != nil,
+			return lo.TernaryF(sampler != nil,
 				func() error { return nil },
-				func() error { return fmt.Errorf("samplers not found (%v)", s) },
+				func() error { return fmt.Errorf("sampler not found (%v)", s) },
 			)
 		},
 	)
@@ -334,6 +342,8 @@ func (b *Bootstrap) buildShrinkCommand(container *assistant.CobraContainer) *cob
 		"quality":         "q",
 	}
 
+	b.load()
+
 	// 📌A note about cobra args validation: cmd.ValidArgs lets you define
 	// a list of all allowable tokens for positional args. Just define
 	// ValidArgs, eg:
@@ -374,3 +384,16 @@ func (b *Bootstrap) getShrinkInputs() *proxy.ShrinkCommandInputs {
 		).(*assistant.ParamSet[store.FilesFilterParameterSet]),
 	}
 }
+
+func (b *Bootstrap) load() {
+	// Ideally, the ProfileParameterSet would perform a check against
+	// the config, but extendio is not aware of config, so it can't
+	// check. Instead, we can check here.
+	//
+	b.profiles = b.optionsInfo.Config.Viper.GetStringMapStringSlice("profiles")
+	err := b.optionsInfo.Config.Viper.UnmarshalKey("sampler", &b.sampler)
+
+	if err != nil {
+		panic(err)
+	}
+}

src/app/proxy/image-defs.go

@@ -15,10 +15,10 @@ type RootParameterSet struct { // should contain RootCommandInputs
 
 type ConfiguredProfiles map[string][]string
 
-func (ps ConfiguredProfiles) Validate(profile string) error {
-	if profile != "" {
-		if _, found := ps[profile]; !found {
-			return fmt.Errorf("no such profile: '%v'", profile)
+func (ps ConfiguredProfiles) Validate(name string) error {
+	if name != "" {
+		if _, found := ps[name]; !found {
+			return fmt.Errorf("no such profile: '%v'", name)
 		}
 	}
 

src/app/proxy/sampler-info.go

@@ -0,0 +1,43 @@
+package proxy
+
+import (
+	"fmt"
+)
+
+type (
+	ConfiguredSamplingScheme struct {
+		Profiles []string `mapstructure:"profiles"`
+	}
+
+	ConfiguredSchemes map[string]ConfiguredSamplingScheme
+
+	SamplerSchemes map[string][]ConfiguredSamplingScheme
+	SamplerInfo    struct {
+		Files   int            `mapstructure:"files"`
+		Folders int            `mapstructure:"folders"`
+		Schemes SamplerSchemes `mapstructure:"schemes"`
+	}
+)
+
+func (cs SamplerInfo) Validate(name string, profiles *ConfiguredProfiles) error {
+	if name == "" {
+		return nil
+	}
+
+	var (
+		found  bool
+		scheme []ConfiguredSamplingScheme
+	)
+
+	if scheme, found = cs.Schemes[name]; !found {
+		return fmt.Errorf("scheme: '%v' not found in config", scheme)
+	}
+
+	for _, p := range scheme[0].Profiles {
+		if _, found := (*profiles)[p]; !found {
+			return fmt.Errorf("profile(scheme: '%v'): '%v' not found in config", name, p)
+		}
+	}
+
+	return nil
+}

src/app/proxy/sampler-runner_test.go

@@ -95,7 +95,6 @@ var _ = Describe("SamplerRunner", Ordered, func() {
 				// "--sample-folders", "3"
 				"--dry-run",
 				"--mode", "tidy",
-				"--scheme", entry.scheme,
 			}
 
 			bootstrap := command.Bootstrap{}
@@ -124,13 +123,24 @@ var _ = Describe("SamplerRunner", Ordered, func() {
 
 		Entry(nil, &samplerTE{
 			runnerTE: runnerTE{
-				given:    "scheme",
-				should:   "bar",
+				given:    "profile",
+				should:   "sample using the defined profile",
 				relative: "nasa/interstellar/Dark Energy Explorers/sessions/scan-01",
 				args: []string{
 					"--strip", "--interlace", "plane", "--quality", "85", "--profile", "adaptive",
 				},
 			},
 		}),
+
+		Entry(nil, &samplerTE{
+			runnerTE: runnerTE{
+				given:    "scheme",
+				should:   "sample all profiles in the scheme",
+				relative: "nasa/interstellar/Dark Energy Explorers/sessions/scan-01",
+				args: []string{
+					"--strip", "--interlace", "plane", "--quality", "85", "--scheme", "blur-sf",
+				},
+			},
+		}),
 	)
 })

test/data/configuration/pixa-test.yml

@@ -21,11 +21,13 @@ profiles:
       "--adaptive-resize",
       "60",
     ]
-samplers:
+sampler:
+  files: 3
+  folders: 1
   schemes:
-    - foo:
+    blur-sf:
       profiles: ["blur", "sf"]
-    - bar:
+    adaptive-sf:
       profiles: ["adaptive", "sf"]
-    - baz:
+    adaptive-blur:
       profiles: ["adaptive", "blur"]