Added enbled flag to APIs in config file
This commit is contained in:
parent
e133a1e95d
commit
861d1c5d5e
@ -1,12 +1,12 @@
|
||||
{
|
||||
"unms": {
|
||||
"active": "true",
|
||||
"enabled": true,
|
||||
"unmsAPIUrl": "https://unifi.freifunk-troisdorf.de/v2.1",
|
||||
"APItoken": "API TOKEN",
|
||||
"devicesURL": "https://git.freifunk-rhein-sieg.net/Freifunk-Troisdorf/ubnt-freifunk-map-api/raw/branch/master/"
|
||||
},
|
||||
"unifi": {
|
||||
"active": "true",
|
||||
"enabled": true,
|
||||
"APIUrl": "https://unifi.freifunk-troisdorf.de:8443",
|
||||
"user": "APIuser",
|
||||
"password": "PASSWORD",
|
||||
|
24
main.go
24
main.go
@ -25,13 +25,13 @@ const (
|
||||
// flags
|
||||
var configPath = flag.String("configPath", "config.json", "Path to config.json")
|
||||
var version = "development"
|
||||
var delay time.Duration = 5 * time.Second
|
||||
var delay time.Duration = 60 * time.Second
|
||||
|
||||
func main() {
|
||||
log.Printf("starting version %s...\n", version)
|
||||
// parse all flags
|
||||
flag.Parse()
|
||||
c := loadConfig(*configPath)
|
||||
c := loadconfig(*configPath)
|
||||
|
||||
// check if flags are set
|
||||
if *configPath == "" {
|
||||
@ -44,8 +44,8 @@ func main() {
|
||||
serveJSON()
|
||||
}
|
||||
|
||||
func loadConfig(file string) Config {
|
||||
var config Config
|
||||
func loadconfig(file string) config {
|
||||
var config config
|
||||
configFile, err := os.Open(file)
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
@ -64,7 +64,7 @@ func itob(i int) bool {
|
||||
}
|
||||
|
||||
//Unifi Controller API processing
|
||||
func processUcAPIs(c Config) ([]node, []link) {
|
||||
func processUcAPIs(c config) ([]node, []link) {
|
||||
//get list of Unifi devices to display
|
||||
var nodes []node
|
||||
var links []link
|
||||
@ -148,7 +148,7 @@ func processUcAPIs(c Config) ([]node, []link) {
|
||||
}
|
||||
|
||||
//UNMS API processing (Richtfunk)
|
||||
func processUNMSAPI(c Config) ([]node, []link) {
|
||||
func processUNMSAPI(c config) ([]node, []link) {
|
||||
// Variables for runtime
|
||||
var links []link
|
||||
var nodes []node
|
||||
@ -238,18 +238,22 @@ func processUNMSAPI(c Config) ([]node, []link) {
|
||||
return nodes, links
|
||||
}
|
||||
|
||||
func processAPIs(c Config) {
|
||||
func processAPIs(c config) {
|
||||
tick := time.Tick(delay)
|
||||
for range tick {
|
||||
var nodes []node
|
||||
var links []link
|
||||
|
||||
if c.Unms.Enabled == true {
|
||||
unmsNodes, unmsLinks := processUNMSAPI(c)
|
||||
ucNodes, ucLinks := processUcAPIs(c)
|
||||
nodes = append(nodes, unmsNodes...)
|
||||
nodes = append(nodes, ucNodes...)
|
||||
links = append(links, unmsLinks...)
|
||||
}
|
||||
if c.Unifi.Enabled == true {
|
||||
ucNodes, ucLinks := processUcAPIs(c)
|
||||
nodes = append(nodes, ucNodes...)
|
||||
links = append(links, ucLinks...)
|
||||
}
|
||||
|
||||
// assemble final struct
|
||||
o := output{
|
||||
@ -295,7 +299,7 @@ func getDevices(url string) (devices, error) {
|
||||
return d, nil
|
||||
}
|
||||
|
||||
func callUnifiAPI(c Config, url string, i interface{}) error {
|
||||
func callUnifiAPI(c config, url string, i interface{}) error {
|
||||
request, err := http.NewRequest(http.MethodGet, c.Unms.UnmsAPIURL+url, nil)
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprint("can't set request", c.Unms.UnmsAPIURL+url))
|
||||
|
6
types.go
6
types.go
@ -10,15 +10,15 @@ import (
|
||||
"time"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
type config struct {
|
||||
Unms struct {
|
||||
Active string `json:"active"`
|
||||
Enabled bool `json:"enabled"`
|
||||
UnmsAPIURL string `json:"unmsAPIUrl"`
|
||||
APItoken string `json:"APItoken"`
|
||||
DevicesURL string `json:"devicesURL"`
|
||||
} `json:"unms"`
|
||||
Unifi struct {
|
||||
Active string `json:"active"`
|
||||
Enabled bool `json:"enabled"`
|
||||
APIURL string `json:"APIUrl"`
|
||||
User string `json:"user"`
|
||||
Password string `json:"password"`
|
||||
|
Loading…
Reference in New Issue
Block a user