energylink/pkg/services/provisioning/datasources/config_reader.go

package datasources

import (
	"io/ioutil"
	"path/filepath"
	"strings"

	"github.com/grafana/grafana/pkg/log"
	"gopkg.in/yaml.v2"
)

type configReader struct {
	log log.Logger
}

func (cr *configReader) readConfig(path string) ([]*DatasourcesAsConfig, error) {
	var datasources []*DatasourcesAsConfig

	files, err := ioutil.ReadDir(path)
	if err != nil {
		cr.log.Error("cant read datasource provisioning files from directory", "path", path)
		return datasources, nil
	}

	for _, file := range files {
		if strings.HasSuffix(file.Name(), ".yaml") || strings.HasSuffix(file.Name(), ".yml") {
			filename, _ := filepath.Abs(filepath.Join(path, file.Name()))
			yamlFile, err := ioutil.ReadFile(filename)
			if err != nil {
				return nil, err
			}

			datasource, err := parseDatasourceConfig(yamlFile)
			if err != nil {
				return nil, err
			}

			if datasource != nil {
				datasources = append(datasources, datasource)
			}
		}
	}

	err = validateDefaultUniqueness(datasources)
	if err != nil {
		return nil, err
	}

	return datasources, nil
}

func parseDatasourceConfig(yamlFile []byte) (*DatasourcesAsConfig, error) {
	var apiVersion *ConfigVersion
	err := yaml.Unmarshal(yamlFile, &apiVersion)
	if err != nil {
		return nil, err
	}

	if apiVersion.ApiVersion > 0 {
		var v1 *DatasourcesAsConfigV1
		err = yaml.Unmarshal(yamlFile, &v1)
		if err != nil {
			return nil, err
		}
		return v1.mapToDatasourceFromConfig(apiVersion.ApiVersion), nil

	}

	var v0 *DatasourcesAsConfigV0
	err = yaml.Unmarshal(yamlFile, &v0)
	if err != nil {
		return nil, err
	}
	return v0.mapToDatasourceFromConfig(apiVersion.ApiVersion), nil
}

func validateDefaultUniqueness(datasources []*DatasourcesAsConfig) error {
	defaultCount := 0
	for i := range datasources {
		if datasources[i].Datasources == nil {
			continue
		}

		for _, ds := range datasources[i].Datasources {
			if ds.OrgId == 0 {
				ds.OrgId = 1
			}

			if ds.IsDefault {
				defaultCount++
				if defaultCount > 1 {
					return ErrInvalidConfigToManyDefault
				}
			}
		}

		for _, ds := range datasources[i].DeleteDatasources {
			if ds.OrgId == 0 {
				ds.OrgId = 1
			}
		}
	}

	return nil
}