81 lines
2.3 KiB
Go
81 lines
2.3 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"encoding/json"
|
|
"github.com/julienschmidt/httprouter"
|
|
"github.com/FreifunkBremen/respond-collector/models"
|
|
)
|
|
|
|
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 {
|
|
if nodeinfo := node.Nodeinfo; nodeinfo !=nil {
|
|
//counter for the diffrent attribute
|
|
count := 1
|
|
if alias.Hostname == nodeinfo.Hostname {
|
|
count -= 1
|
|
}
|
|
//delete element
|
|
if count <= 0 {
|
|
delete(api.aliases.List,key)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
func (api *ApiAliases) GetAll(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
|
|
jsonOutput(w,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,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)
|
|
jsonOutput(w,alias)
|
|
}
|
|
|
|
func (api *ApiAliases) Cleanup(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
|
|
api.cleaner()
|
|
jsonOutput(w,api.aliases.List)
|
|
}
|
|
func (api *ApiAliases) AnsibleDiff(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
|
|
api.cleaner()
|
|
jsonOutput(w,models.GenerateAnsible(api.nodes,api.aliases.List))
|
|
}
|