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