HGKeeper

// Convey

// This program is free software: you can redistribute it and/or modify

// it under the terms of the GNU General Public License as published by

// the Free Software Foundation, either version 3 of the License, or

// (at your option) any later version.

// This program is distributed in the hope that it will be useful,

// but WITHOUT ANY WARRANTY; without even the implied warranty of

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License

// along with this program. If not, see <http://www.gnu.org/licenses/>.

package config

import (

"io/ioutil"

"os"

log "github.com/sirupsen/logrus"

"gopkg.in/yaml.v2"

)

func LoadString(data []byte) (*Config, error) {

var cfg rawConfig

err := yaml.Unmarshal(data, &cfg)

if err != nil {

return nil, err

}

return cfg.process()

}

func LoadFile(filename string) (*Config, error) {

_, err := os.Stat(filename)

if err != nil {

// If the user selected the default and it didn't exist, attempt the

// default with a .yaml extension.

if os.IsNotExist(err) && filename == "convey.yml" {

log.Infof("info: configuration file convey.yml not found, attempting convey.yaml instead.")

filename = "convey.yaml"

_, err = os.Stat(filename)

if err != nil {

return nil, err

}

} else {

return nil, err

}

contents, err := ioutil.ReadFile(filename)

if err != nil {

return nil, err

}

return LoadString(contents)

}