blob: 167da6030f9f8c375b9ae7e91842e4635d5320ab [file] [log] [blame]
/*
* Copyright 2018-present Open Networking Foundation
* Copyright 2018-present Edgecore Networks Corporation
*
* 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.
*/
// Implements global configuration for demotest
package main
import (
"fmt"
flags "github.com/jessevdk/go-flags"
"gopkg.in/yaml.v2"
"io/ioutil"
"log"
"os"
"path"
"path/filepath"
"strings"
)
type OutputType uint8
type GlobalConfigSpec struct {
Kafka string `yaml:"kafka"`
Local string `yaml:"local"`
Importer string `yaml:"importer"`
}
var (
CharReplacer = strings.NewReplacer("\\t", "\t", "\\n", "\n")
GlobalConfig = GlobalConfigSpec{
Kafka: "kafka_ip.sh",
Local: ":9999",
Importer: "localhost:31085",
}
GlobalCommandOptions = make(map[string]map[string]string)
GlobalOptions struct {
Config string `short:"c" long:"config" env:"PROXYCONFIG" value-name:"FILE" default:"" description:"Location of proxy config file"`
Kafka string `short:"k" long:"kafka" default:"" value-name:"SERVER:PORT" description:"IP/Host and port of Kafka"`
Importer string `short:"i" long:"importer" default:"" value-name:"SERVER:PORT" description:"IP/Host and port of Importer"`
Local string `short:"l" long:"local" default:"" value-name:"SERVER:PORT" description:"IP/Host and port to listen on"`
}
Debug = log.New(os.Stdout, "DEBUG: ", 0)
Info = log.New(os.Stdout, "INFO: ", 0)
Warn = log.New(os.Stderr, "WARN: ", 0)
Error = log.New(os.Stderr, "ERROR: ", 0)
)
func ParseCommandLine() {
parser := flags.NewNamedParser(path.Base(os.Args[0]),
flags.HelpFlag|flags.PassDoubleDash|flags.PassAfterNonOption)
_, err := parser.AddGroup("Global Options", "", &GlobalOptions)
if err != nil {
panic(err)
}
_, err = parser.ParseArgs(os.Args[1:])
if err != nil {
_, ok := err.(*flags.Error)
if ok {
real := err.(*flags.Error)
if real.Type == flags.ErrHelp {
os.Stdout.WriteString(err.Error() + "\n")
os.Exit(0)
}
}
fmt.Fprintf(os.Stderr, "%s: %s\n", os.Args[0], err.Error())
os.Exit(1)
}
}
func ProcessGlobalOptions() {
if len(GlobalOptions.Config) == 0 {
home, err := os.UserHomeDir()
if err != nil {
Warn.Printf("Unable to discover the user's home directory: %s", err)
home = "~"
}
GlobalOptions.Config = filepath.Join(home, ".redfish-importer", "demotest-config")
}
if info, err := os.Stat(GlobalOptions.Config); err == nil && !info.IsDir() {
configFile, err := ioutil.ReadFile(GlobalOptions.Config)
if err != nil {
Error.Fatalf("Unable to read the configuration file '%s': %s",
GlobalOptions.Config, err.Error())
}
if err = yaml.Unmarshal(configFile, &GlobalConfig); err != nil {
Error.Fatalf("Unable to parse the configuration file '%s': %s",
GlobalOptions.Config, err.Error())
}
}
if GlobalOptions.Kafka != "" {
GlobalConfig.Kafka = GlobalOptions.Kafka
}
if GlobalOptions.Local != "" {
GlobalConfig.Local = GlobalOptions.Local
}
if GlobalOptions.Importer != "" {
GlobalConfig.Importer = GlobalOptions.Importer
}
}
func ShowGlobalOptions() {
log.Printf("Configuration:")
log.Printf(" Kafka: %v", GlobalConfig.Kafka)
log.Printf(" Listen Address: %v", GlobalConfig.Local)
}