Support a config file to use instead of commandline arguments

Signed-off-by: Dave Henderson <dhenderson@gmail.com>

1 files changed, 24 insertions, 0 deletions

diff --git a/data/datasource.go b/data/datasource.go
index 1fcccf98..456b5a12 100644
--- a/data/datasource.go
+++ b/data/datasource.go
@@ -16,6 +16,7 @@ import (
 
 	"github.com/pkg/errors"
 
+	"github.com/hairyhenderson/gomplate/v3/internal/config"
 	"github.com/hairyhenderson/gomplate/v3/libkv"
 	"github.com/hairyhenderson/gomplate/v3/vault"
 )
@@ -125,6 +126,29 @@ func NewData(datasourceArgs, headerArgs []string) (*Data, error) {
 	return data, nil
 }
 
+// FromConfig - internal use only!
+func FromConfig(cfg *config.Config) *Data {
+	sources := map[string]*Source{}
+	for alias, d := range cfg.DataSources {
+		sources[alias] = &Source{
+			Alias:  alias,
+			URL:    d.URL,
+			header: d.Header,
+		}
+	}
+	for alias, d := range cfg.Context {
+		sources[alias] = &Source{
+			Alias:  alias,
+			URL:    d.URL,
+			header: d.Header,
+		}
+	}
+	return &Data{
+		Sources:      sources,
+		extraHeaders: cfg.ExtraHeaders,
+	}
+}
+
 // Source - a data source
 type Source struct {
 	Alias             string