yanic/api/aliases.go

package api

import (
	"encoding/json"
	"fmt"
	"github.com/FreifunkBremen/respond-collector/models"
	"github.com/julienschmidt/httprouter"
	"net/http"
)

// 7 nachkommerstellen sollten genug sein (7cm genau)
// http://blog.3960.org/post/7309573249/genauigkeit-bei-geo-koordinaten

const GEOROUND = 0.0000001

func geoEqual(a, b float64) bool {
	if (a-b) < GEOROUND && (b-a) < GEOROUND {
		return true
	}
	return false
}

type ApiAliases struct {
	aliases *models.Aliases
	config  *models.Config
	nodes   *models.Nodes
}

func NewAliases(config *models.Config, router *httprouter.Router, prefix string, nodes *models.Nodes) {
	api := &ApiAliases{
		aliases: models.NewAliases(config),
		nodes:   nodes,
		config:  config,
	}
	router.GET(prefix, api.GetAll)
	router.GET(prefix+"/ansible", api.AnsibleDiff)
	router.GET(prefix+"/cleanup", api.Cleanup)
	router.GET(prefix+"/auth", BasicAuth(api.Cleanup, []byte(config.Webserver.Api.Passphrase)))
	router.GET(prefix+"/alias/:nodeid", api.GetOne)
	router.POST(prefix+"/alias/:nodeid", BasicAuth(api.SaveOne, []byte(config.Webserver.Api.Passphrase)))
}

// clean up the aliases by correct values in nodes
func (api *ApiAliases) cleaner() {
	for key, alias := range api.aliases.List {
		if node := api.nodes.List[key]; node != nil {
			//counter for the diffrent attribute
			count := 0
			if nodeinfo := node.Nodeinfo; nodeinfo != nil {
				if len(alias.Hostname) > 0 {
					count += 1
					if alias.Hostname == nodeinfo.Hostname {
						count -= 1
						alias.Hostname = ""
					}
				}
				if len(alias.Owner) > 0 {
					count += 1
					if nodeinfo.Owner != nil && alias.Owner == nodeinfo.Owner.Contact {
						count -= 1
						alias.Owner = ""
					}
				}
				if alias.Location != nil {
					count += 2
					if nodeinfo.Location != nil {
						if geoEqual(alias.Location.Latitude, nodeinfo.Location.Latitude) {
							count -= 1
							if geoEqual(alias.Location.Longtitude, nodeinfo.Location.Longtitude) {
								count -= 1
								alias.Location = nil
							}
						} else {
							if geoEqual(alias.Location.Longtitude, nodeinfo.Location.Longtitude) {
								count -= 1
							}
						}
					}
				}
				if nodeinfo.Wireless != nil && alias.Wireless != nil {
					count += 4
					if alias.Wireless.Channel24 == nodeinfo.Wireless.Channel24 {
						count -= 1
					}
					if alias.Wireless.TxPower24 == nodeinfo.Wireless.TxPower24 {
						count -= 1
					}
					if alias.Wireless.Channel5 == nodeinfo.Wireless.Channel5 {
						count -= 1
					}
					if alias.Wireless.TxPower5 == nodeinfo.Wireless.TxPower5 {
						count -= 1
					}
					if alias.Wireless.Channel24 == nodeinfo.Wireless.Channel24 && alias.Wireless.TxPower24 == nodeinfo.Wireless.TxPower24 && alias.Wireless.Channel5 == nodeinfo.Wireless.Channel5 && alias.Wireless.TxPower5 == nodeinfo.Wireless.TxPower5 {
						alias.Wireless = nil
					}
				}
			}

			//delete element
			if count <= 0 {
				delete(api.aliases.List, key)
				fmt.Print("[api] node updated '", key, "'\n")
			}
		}
	}
}
func (api *ApiAliases) GetAll(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
	jsonOutput(w, r, api.aliases.List)
}

func (api *ApiAliases) GetOne(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
	if alias := api.aliases.List[ps.ByName("nodeid")]; alias != nil {
		jsonOutput(w, r, alias)
		return
	}
	fmt.Fprint(w, "Not found: ", ps.ByName("nodeid"), "\n")
}

func (api *ApiAliases) SaveOne(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
	var alias models.Alias

	err := json.NewDecoder(r.Body).Decode(&alias)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
		fmt.Fprint(w, "Decode: ", ps.ByName("nodeid"), "\n")
		return
	}
	api.aliases.Update(ps.ByName("nodeid"), &alias)
	fmt.Print("[api] node updated '", ps.ByName("nodeid"), "'\n")
	jsonOutput(w, r, alias)
}

func (api *ApiAliases) Cleanup(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
	api.cleaner()
	jsonOutput(w, r, api.aliases.List)
}
func (api *ApiAliases) AnsibleDiff(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
	fmt.Print("[api] ansible\n")
	api.cleaner()
	jsonOutput(w, r, models.GenerateAnsible(api.nodes, api.aliases.List))
}