Add function to explictly request a certain provider (#1837)

Signed-off-by: Priya Wadhwa <[email protected]>

pkg/providers/all/all.go

@@ -27,6 +27,7 @@ import (
 
 // Alias these methods, so that folks can import this to get all providers.
 var (
-	Enabled = providers.Enabled
-	Provide = providers.Provide
+	Enabled     = providers.Enabled
+	Provide     = providers.Provide
+	ProvideFrom = providers.ProvideFrom
 )

pkg/providers/interface.go

@@ -83,3 +83,15 @@ func Provide(ctx context.Context, audience string) (string, error) {
 	}
 	return id, err
 }
+
+// ProvideFrom fetches the specified provider
+func ProvideFrom(ctx context.Context, provider string) (Interface, error) {
+	m.Lock()
+	defer m.Unlock()
+
+	p, ok := providers[provider]
+	if !ok {
+		return nil, fmt.Errorf("%s is not a valid provider", provider)
+	}
+	return p, nil
+}