backends_toml.go

// Copyright 2018 The OpenPitrix Authors. All rights reserved.
// Use of this source code is governed by a Apache license
// that can be found in the LICENSE file.

package libconfd

import (
	"fmt"
	"strings"

	"github.com/BurntSushi/toml"
)

const TomlBackendType = "libconfd-backend-toml"

var _ BackendClient = (*TomlBackend)(nil)

type TomlBackend struct {
	TOMLFile string
}

func init() {
	RegisterBackendClient(
		(*TomlBackend)(nil).Type(),
		func(cfg *BackendConfig) (BackendClient, error) {
			p := NewTomlBackendClient(cfg)
			return p, nil
		},
	)
}

func NewTomlBackendClient(cfg *BackendConfig) *TomlBackend {
	GetLogger().Assert(cfg.Type == (*TomlBackend)(nil).Type())
	return &TomlBackend{TOMLFile: cfg.Host[0]}
}

func (_ *TomlBackend) Close() error {
	return nil
}

func (_ *TomlBackend) Type() string {
	return TomlBackendType
}

func (_ *TomlBackend) WatchEnabled() bool {
	return false
}

func (_ *TomlBackend) WatchPrefix(prefix string, keys []string, waitIndex uint64, stopChan chan bool) (uint64, error) {
	return 0, fmt.Errorf("do not support watch")
}

func (p *TomlBackend) GetValues(keys []string) (m map[string]string, err error) {
	var dataMap map[string]string
	_, err = toml.DecodeFile(p.TOMLFile, &dataMap)
	if err != nil {
		return nil, err
	}

	// skip invalid key
	m = make(map[string]string)
	for k, v := range dataMap {
		if strings.HasPrefix(k, "/") {
			m[k] = v
		}
	}

	return m, nil
}