garm-provider-edge-connect/config/config.go

// Copyright 2023 Cloudbase Solutions SRL
//
//    Licensed under the Apache License, Version 2.0 (the "License"); you may
//    not use this file except in compliance with the License. You may obtain
//    a copy of the License at
//
//         http://www.apache.org/licenses/LICENSE-2.0
//
//    Unless required by applicable law or agreed to in writing, software
//    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
//    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
//    License for the specific language governing permissions and limitations
//    under the License.

package config

import (
	"fmt"

	"github.com/BurntSushi/toml"
)

// NewConfig returns a new Config
func NewConfig(cfgFile string) (*Config, error) {
	var config Config
	if _, err := toml.DecodeFile(cfgFile, &config); err != nil {
		return nil, fmt.Errorf("error decoding config: %w", err)
	}

	if err := config.Validate(); err != nil {
		return nil, fmt.Errorf("error validating config: %w (note: TOML section must be [edge_connect] not [edge-connect])", err)
	}
	return &config, nil
}

func NewCredentials(credFile string) (*Credentials, error) {
	var creds Credentials
	if _, err := toml.DecodeFile(credFile, &creds); err != nil {
		return nil, fmt.Errorf("error decoding creds: %w", err)
	}

	return &creds, nil
}

type Config struct {
	LogFile         string      `toml:"log_file"`
	CredentialsFile string      `toml:"credentials_file"`
	EdgeConnect     EdgeConnect `toml:"edge_connect"`
}

type Credentials struct {
	Username string `toml:"username"`
	Password string `toml:"password"`
}

type CloudletKey struct {
	Organization string `toml:"organization"`
	Name         string `toml:"name"`
}

type EdgeConnect struct {
	URL           string      `toml:"url"`
	Organization  string      `toml:"organization"`
	Region        string      `toml:"region"`
	DefaultFlavor string      `toml:"default_flavor"`
	CloudletKey   CloudletKey `toml:"cloudlet"`
}

func (c *Config) Validate() error {
	if c.LogFile == "" {
		return fmt.Errorf("log_file is not defined in provider config")
	}

	if c.EdgeConnect.URL == "" {
		return fmt.Errorf("edge_connect.url is not defined in provider config")
	}

	return nil
}