83 lines
2.4 KiB
Go
83 lines
2.4 KiB
Go
|
// Copyright (c) Mainflux
|
||
|
|
// SPDX-License-Identifier: Apache-2.0
|
||
|
|
|
||
|
|
package provision
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
|
||
|
|
"github.com/mainflux/mainflux/errors"
|
||
|
|
"github.com/pelletier/go-toml"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Service represents service config.
|
||
|
|
type ServiceConf struct {
|
||
|
|
Port string `toml:"port"`
|
||
|
|
LogLevel string `toml:"log_level"`
|
||
|
|
TLS bool `toml:"tls"`
|
||
|
|
CACerts string `toml:"ca_certs"`
|
||
|
|
ServerCert string `toml:"server_cert"`
|
||
|
|
ServerKey string `toml:"server_key"`
|
||
|
|
ThingsLocation string `toml:"things_location"`
|
||
|
|
UsersLocation string `toml:"users_location"`
|
||
|
|
MQTTURL string `toml:"mqtt_url"`
|
||
|
|
HTTPPort string `toml:"http_port"`
|
||
|
|
MfUser string `toml:"mf_user"`
|
||
|
|
MfPass string `toml:"mf_pass"`
|
||
|
|
MfAPIKey string `toml:"mf_api_key"`
|
||
|
|
MfBSURL string `toml:"mf_bs_url"`
|
||
|
|
MfWhiteListURL string `toml:"mf_whit_list"`
|
||
|
|
MfCertsURL string `toml:"mf_certs_url"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Bootstrap struct {
|
||
|
|
X509Provision bool `toml:"x509_provision"`
|
||
|
|
Provision bool `toml:"provision"`
|
||
|
|
AutoWhiteList bool `toml:"autowhite_list"`
|
||
|
|
Content string `toml:"content"`
|
||
|
|
}
|
||
|
|
type Channel struct {
|
||
|
|
Name string `toml:"name"`
|
||
|
|
Metadata map[string]interface{} `toml:"metadata" mapstructure:"metadata"`
|
||
|
|
}
|
||
|
|
type Thing struct {
|
||
|
|
Name string `toml:"name"`
|
||
|
|
Metadata map[string]interface{} `toml:"metadata" mapstructure:"metadata"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// Config struct of Provision
|
||
|
|
type Config struct {
|
||
|
|
File string `toml:"file"`
|
||
|
|
Server ServiceConf `toml:"server" mapstructure:"server"`
|
||
|
|
Bootstrap Bootstrap `toml:"bootstrap" mapstructure:"boostrap"`
|
||
|
|
Things []Thing `toml:"things" mapstructure:"things"`
|
||
|
|
Channels []Channel `toml:"channels" mapstructure:"channels"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// Save - store config in a file
|
||
|
|
func Save(c Config, file string) error {
|
||
|
|
b, err := toml.Marshal(c)
|
||
|
|
if err != nil {
|
||
|
|
return errors.New(fmt.Sprintf("Error reading config file: %s", err))
|
||
|
|
}
|
||
|
|
if err := ioutil.WriteFile(file, b, 0644); err != nil {
|
||
|
|
return errors.New(fmt.Sprintf("Error writing toml: %s", err))
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Read - retrieve config from a file
|
||
|
|
func Read(file string) (Config, error) {
|
||
|
|
data, err := ioutil.ReadFile(file)
|
||
|
|
c := Config{}
|
||
|
|
if err != nil {
|
||
|
|
return c, errors.New(fmt.Sprintf("Error reading config file: %s", err))
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := toml.Unmarshal(data, &c); err != nil {
|
||
|
|
return Config{}, errors.New(fmt.Sprintf("Error unmarshaling toml: %s", err))
|
||
|
|
}
|
||
|
|
return c, nil
|
||
|
|
}
|