Compare commits

..

3 Commits

3 changed files with 167 additions and 121 deletions

View File

@ -1,35 +1,45 @@
global-variables:
env: &env
image: golang
commands:
- go build -ldflags "-X main.version=${DRONE_TAG##v}" -o release/$GOOS/$GOARCH/ubnt-freifunk-map-api .
- tar -cvzf release/ubnt-freifunk-map-api_$GOOS_$GOARCH.tar.gz -C release/$GOOS/$GOARCH ubnt-freifunk-map-api
kind: pipeline kind: pipeline
type: docker type: docker
name: build go name: build go
steps: steps:
- name: build linux/amd64 - name: build linux/amd64
image: golang <<: *env
commands:
- go build -ldflags "-X main.version=${DRONE_TAG}" -o release/linux/amd64/ubnt-freifunk-map-api .
- tar -cvzf release/ubnt-freifunk-map-api_linux_amd64.tar.gz -C release/linux/amd64 ubnt-freifunk-map-api
environment: environment:
GOARCH: amd64 GOARCH: amd64
GOOS: linux GOOS: linux
- name: build linux/arm64 - name: build linux/arm64
image: golang <<: *env
commands:
- go build -ldflags "-X main.version=${DRONE_TAG}" -o release/linux/arm64/ubnt-freifunk-map-api .
- tar -cvzf release/ubnt-freifunk-map-api_linux_arm64.tar.gz -C release/linux/arm64 ubnt-freifunk-map-api
environment: environment:
GOARCH: arm64 GOARCH: arm64
GOOS: linux GOOS: linux
- name: build windows - name: build windows
image: golang <<: *env
commands:
- go build -ldflags "-X main.version=${DRONE_TAG}" -o release/windows/amd64/ubnt-freifunk-map-api .
- tar -cvzf release/ubnt-freifunk-map-api_windows_amd64.tar.gz -C release/windows/amd64 ubnt-freifunk-map-api
environment: environment:
GOARCH: amd64 GOARCH: amd64
GOOS: windows GOOS: windows
- name: build macos/x64
<<: *env
environment:
GOARCH: amd64
GOOS: darwin
- name: build macos/arm64
<<: *env
environment:
GOARCH: arm64
GOOS: darwin
- name: docker - name: docker
image: plugins/docker image: plugins/docker
settings: settings:

226
main.go
View File

@ -23,118 +23,131 @@ const (
// flags // flags
var token = flag.String("token", "", "Defines the x-auth-token") var token = flag.String("token", "", "Defines the x-auth-token")
var version = "development" var version = "development"
var delay time.Duration = 60 * time.Second
func main() { func main() {
log.Printf("starting version %s...\n", version)
log.Println("starting version", version, "...")
// parse all flags // parse all flags
flag.Parse() flag.Parse()
// check if token is set // check if token is set
if *token == "" { if *token == "" {
log.Fatalln("Please specify an API token via the flag '-token'") log.Fatalln("Please specify an API token via the flag '-token'")
} }
// Variables for runtime // start API processing (runs in a loop)
var links []link go processAPIs()
var nodes []node
d, err := getDevices() // start webserver
if err != nil { serveJSON()
log.Fatalln(err) }
}
// API CALL 1 func processAPIs() {
log.Println("calling API 1") tick := time.Tick(delay)
var u []unifiAPIResponse for range tick {
err = callUnifiAPI("/devices", &u) // Variables for runtime
if err != nil { var links []link
log.Fatalln(err) var nodes []node
}
for i := range d.Devices { d, err := getDevices()
var dev unifiAPIResponse if err != nil {
var currentDevice device log.Fatalln(err)
for j := range u { }
if strings.ToUpper(u[j].Identification.MAC) == strings.ToUpper(d.Devices[i].MAC) {
dev = u[j] // API CALL 1
currentDevice = d.Devices[i] log.Println("calling API 1")
var u []unifiAPIResponse
err = callUnifiAPI("/devices", &u)
if err != nil {
log.Fatalln(err)
}
for i := range d.Devices {
var dev unifiAPIResponse
var currentDevice device
for j := range u {
if strings.ToUpper(u[j].Identification.MAC) == strings.ToUpper(d.Devices[i].MAC) {
dev = u[j]
currentDevice = d.Devices[i]
}
} }
}
var isOnline bool = false var isOnline bool = false
if dev.Overview.Status == "active" { if dev.Overview.Status == "active" {
isOnline = true isOnline = true
}
// END OF API CALL 1
// API CALL 2
log.Println("calling API 2 for device", d.Devices[i].Name)
var details unifiAPIDetails
callUnifiAPI("/devices/erouters/"+dev.Identification.ID, &details)
// END OF API CALL 2
// API CALL 3
log.Println("calling API 3 for device", d.Devices[i].Name)
var airmaxes []unifiAPIAirmax
callUnifiAPI("/devices/airmaxes/"+dev.Identification.ID+"/stations", &airmaxes)
// check if remote mac address is part of our published network
for i := range airmaxes {
if isRemoteMACpublished(airmaxes[i].DeviceIdentification.MAC, d.Devices) == true {
links = addLink(dev, airmaxes[i], links)
} }
// END OF API CALL 1
// API CALL 2
log.Println("calling API 2 for device", d.Devices[i].Name)
var details unifiAPIDetails
callUnifiAPI("/devices/erouters/"+dev.Identification.ID, &details)
// END OF API CALL 2
// API CALL 3
log.Println("calling API 3 for device", d.Devices[i].Name)
var airmaxes []unifiAPIAirmax
callUnifiAPI("/devices/airmaxes/"+dev.Identification.ID+"/stations", &airmaxes)
// check if remote mac address is part of our published network
for i := range airmaxes {
if isRemoteMACpublished(airmaxes[i].DeviceIdentification.MAC, d.Devices) == true {
links = addLink(dev, airmaxes[i], links)
}
}
// END OF API CALL 3
// Get info from json file (static)
nodes = append(nodes, node{
Firstseen: dev.Overview.CreatedAt.Format(iso8601),
Lastseen: dev.Overview.LastSeen.Format(iso8601),
IsOnline: isOnline,
IsGateway: false,
Clients: 0,
ClientsWifi24: 0,
ClientsWifi5: 0,
ClientsOther: 0,
RootFSUsage: 0,
LoadAVG: details.Overview.CPU / 100,
MemoryUsage: details.Overview.RAM / 100,
Uptime: dev.Identification.Started.Format(iso8601),
GatewayNexthop: currentDevice.GatewayNexthop,
Gateway: currentDevice.Gateway,
Location: currentDevice.Location,
NodeID: strings.ReplaceAll(dev.Identification.MAC, ":", ""),
MAC: dev.Identification.MAC,
Adresses: getAddresses(details.IPAddress),
Domain: currentDevice.Domain,
Hostname: "[RiFu] " + details.Identification.Name,
Owner: "Freifunk Rhein-Sieg",
Firmware: firmware{
Base: "Ubiquiti - Stock",
Release: details.Firmware.Current,
},
Autoupdater: autoupdater{
Enabled: false,
Branch: "stable",
},
NProc: 1,
Model: details.Identification.Model,
})
} }
// END OF API CALL 3
// Get info from json file (static) // assemble final struct
nodes = append(nodes, node{ o := output{
Firstseen: dev.Overview.CreatedAt.Format(iso8601), Timestamp: time.Now().Format(iso8601),
Lastseen: dev.Overview.LastSeen.Format(iso8601), Nodes: nodes,
IsOnline: isOnline, Links: links,
IsGateway: false, }
Clients: 0,
ClientsWifi24: 0, // create file output
ClientsWifi5: 0, log.Println("writing json file")
ClientsOther: 0,
RootFSUsage: 0, if err := o.writeToFile(); err != nil {
LoadAVG: details.Overview.CPU / 100, log.Fatalln(err)
MemoryUsage: details.Overview.RAM / 100, }
Uptime: dev.Identification.Started.Format(iso8601),
GatewayNexthop: currentDevice.GatewayNexthop, // we're done here
Gateway: currentDevice.Gateway, log.Println("...done")
Location: currentDevice.Location,
NodeID: strings.ReplaceAll(dev.Identification.MAC, ":", ""),
MAC: dev.Identification.MAC,
Adresses: getAddresses(details.IPAddress),
Domain: currentDevice.Domain,
Hostname: "[RiFu] " + details.Identification.Name,
Owner: "Freifunk Rhein-Sieg",
Firmware: firmware{
Base: "Ubiquiti - Stock",
Release: details.Firmware.Current,
},
Autoupdater: autoupdater{
Enabled: false,
Branch: "stable",
},
NProc: 1,
Model: details.Identification.Model,
})
} }
// assemble final struct
o := output{
Timestamp: time.Now().Format(iso8601),
Nodes: nodes,
Links: links,
}
// create file output
log.Println("writing json file")
writeJSONtoFile(o)
// we're done here
log.Println("...done")
} }
func getDevices() (devices, error) { func getDevices() (devices, error) {
@ -213,22 +226,19 @@ func addLink(dev unifiAPIResponse, airmaxes unifiAPIAirmax, links []link) []link
return links return links
} }
func writeJSONtoFile(o output) error {
file, err := json.MarshalIndent(o, "", " ")
if err != nil {
return fmt.Errorf("can't marshal to json: %+v", o)
}
// write to file
err = ioutil.WriteFile("example.json", file, 0644)
if err != nil {
return fmt.Errorf("can't write to json file example.json")
}
return nil
}
func getAddresses(ip string) []string { func getAddresses(ip string) []string {
var adresses []string var adresses []string
adresses = append(adresses, strings.Split(ip, "/")[0]) adresses = append(adresses, strings.Split(ip, "/")[0])
return adresses return adresses
} }
func serveJSON() {
fs := http.FileServer(http.Dir("./output"))
http.Handle("/", fs)
log.Println("Listening on :3000...")
err := http.ListenAndServe(":3000", nil)
if err != nil {
log.Fatalln(err)
}
}

View File

@ -1,6 +1,12 @@
package main package main
import "time" import (
"encoding/json"
"fmt"
"io/ioutil"
"os"
"time"
)
type device struct { type device struct {
Name string `json:"name"` Name string `json:"name"`
@ -111,6 +117,26 @@ type output struct {
Links []link `json:"links"` Links []link `json:"links"`
} }
func (o *output) writeToFile() error {
file, err := json.MarshalIndent(o, "", " ")
if err != nil {
return fmt.Errorf("can't marshal to json: %+v", o)
}
// write to file
if _, err := os.Stat("output"); os.IsNotExist(err) {
err = os.Mkdir("output", 0755)
if err != nil {
return fmt.Errorf("can't create output folder")
}
}
err = ioutil.WriteFile("output/meshviewer.json", file, 0644)
if err != nil {
return fmt.Errorf("can't write to json file meshviewer.json")
}
return nil
}
type apiResponse struct { type apiResponse struct {
StatusCode int `json:"statusCode"` StatusCode int `json:"statusCode"`
Error string `json:"error"` Error string `json:"error"`