first golang version
This commit is contained in:
parent
ff96affcbe
commit
b293cbd66b
26
devices.json
Normal file
26
devices.json
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
{
|
||||||
|
"devices":[
|
||||||
|
{
|
||||||
|
"name": "Rathaus2Bahnhof",
|
||||||
|
"mac": "18:E8:29:8E:C6:4D",
|
||||||
|
"gateway_nexthop": "18e8292f7de6",
|
||||||
|
"gateway": "a28cae6ff604",
|
||||||
|
"domain": "rifu",
|
||||||
|
"location": {
|
||||||
|
"longitude":7.148406208,
|
||||||
|
"latitude":50.817093402
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Bahnhof2Rathaus",
|
||||||
|
"mac": "18:e8:29:dc:c3:7e",
|
||||||
|
"gateway_nexthop": "18e8292f7de6",
|
||||||
|
"gateway": "a28cae6ff604",
|
||||||
|
"domain": "rifu",
|
||||||
|
"location": {
|
||||||
|
"longitude":7.150436640,
|
||||||
|
"latitude":50.814456507
|
||||||
|
}
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
3
go.mod
Normal file
3
go.mod
Normal file
@ -0,0 +1,3 @@
|
|||||||
|
module git.freifunk-rhein-sieg.net/Freifunk-Troisdorf/ubnt-freifunk-map-api
|
||||||
|
|
||||||
|
go 1.16
|
||||||
212
main.go
Normal file
212
main.go
Normal file
@ -0,0 +1,212 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"flag"
|
||||||
|
"io/ioutil"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// types
|
||||||
|
|
||||||
|
const (
|
||||||
|
baseURL = "https://unifi.freifunk-troisdorf.de/v2.1"
|
||||||
|
iso8601 = "2006-01-02T15:04:05-0700"
|
||||||
|
)
|
||||||
|
|
||||||
|
// flags
|
||||||
|
var token = flag.String("token", "", "Defines the x-auth-token")
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// parse all flags
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
// check if token is set
|
||||||
|
if *token == "" {
|
||||||
|
log.Println("Please specify an API token via the flag '-token'")
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Variables for runtime
|
||||||
|
var links []link
|
||||||
|
var nodes []node
|
||||||
|
|
||||||
|
d := getDevices()
|
||||||
|
|
||||||
|
// API CALL 1
|
||||||
|
var u []unifiAPIResponse
|
||||||
|
callUnifiAPI("/devices", &u)
|
||||||
|
|
||||||
|
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]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// fmt.Println(time.Now().Format(iso8601))
|
||||||
|
var isOnline bool = false
|
||||||
|
if dev.Overview.Status == "active" {
|
||||||
|
isOnline = true
|
||||||
|
}
|
||||||
|
|
||||||
|
// uptime := dev.Identification.Started.Format(iso8601)
|
||||||
|
// nodeID := strings.ReplaceAll(dev.Identification.MAC, ":", "")
|
||||||
|
// END OF API CALL 1
|
||||||
|
|
||||||
|
// API CALL 2
|
||||||
|
var details unifiAPIDetails
|
||||||
|
callUnifiAPI("/devices/erouters/"+dev.Identification.ID, &details)
|
||||||
|
// END OF API CALL 2
|
||||||
|
|
||||||
|
// API CALL 3
|
||||||
|
var airmaxes []unifiAPIAirmax
|
||||||
|
callUnifiAPI("/devices/airmaxes/"+dev.Identification.ID+"/stations", &airmaxes)
|
||||||
|
// check if remote mac address is part of our published network
|
||||||
|
// if isRemoteMACpublished()
|
||||||
|
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,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// assemble final struct
|
||||||
|
o := output{
|
||||||
|
Timestamp: time.Now().Format(iso8601),
|
||||||
|
Nodes: nodes,
|
||||||
|
Links: links,
|
||||||
|
}
|
||||||
|
|
||||||
|
// create file output
|
||||||
|
writeJSONtoFile(o)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getDevices() devices {
|
||||||
|
// get devices from JSON file
|
||||||
|
jsonFile, err := os.Open("devices.json")
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
defer jsonFile.Close()
|
||||||
|
|
||||||
|
// read file to bytes
|
||||||
|
byteValue, _ := ioutil.ReadAll(jsonFile)
|
||||||
|
|
||||||
|
// variable for d
|
||||||
|
var d devices
|
||||||
|
// unmarshal to struct
|
||||||
|
json.Unmarshal(byteValue, &d)
|
||||||
|
return d
|
||||||
|
}
|
||||||
|
|
||||||
|
func callUnifiAPI(url string, i interface{}) {
|
||||||
|
request, err := http.NewRequest(http.MethodGet, baseURL+url, nil)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
request.Header.Set("x-auth-token", *token)
|
||||||
|
client := &http.Client{}
|
||||||
|
response, err := client.Do(request)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := ioutil.ReadAll(response.Body)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
// fmt.Println(string(data))
|
||||||
|
|
||||||
|
json.Unmarshal(data, &i)
|
||||||
|
}
|
||||||
|
|
||||||
|
func isRemoteMACpublished(mac string, devices []device) bool {
|
||||||
|
for i := range devices {
|
||||||
|
if devices[i].MAC == mac {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func addLink(dev unifiAPIResponse, airmaxes unifiAPIAirmax, links []link) []link {
|
||||||
|
for i := range links {
|
||||||
|
if links[i].SourceAddr == airmaxes.DeviceIdentification.MAC {
|
||||||
|
// link already exists
|
||||||
|
return links
|
||||||
|
}
|
||||||
|
}
|
||||||
|
links = append(links, link{
|
||||||
|
Type: "wifi",
|
||||||
|
Source: strings.ReplaceAll(dev.Identification.MAC, ":", ""),
|
||||||
|
Target: strings.ReplaceAll(airmaxes.DeviceIdentification.MAC, ":", ""),
|
||||||
|
SourceTQ: airmaxes.Statistics.LinkScore,
|
||||||
|
TargetTQ: airmaxes.Statistics.LinkScore,
|
||||||
|
SourceAddr: dev.Identification.MAC,
|
||||||
|
TargetAddr: airmaxes.DeviceIdentification.MAC,
|
||||||
|
})
|
||||||
|
return links
|
||||||
|
}
|
||||||
|
|
||||||
|
func writeJSONtoFile(o output) {
|
||||||
|
file, err := json.MarshalIndent(o, "", " ")
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// write to file
|
||||||
|
err = ioutil.WriteFile("example.json", file, 0644)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func getAddresses(ip string) []string {
|
||||||
|
var adresses []string
|
||||||
|
adresses = append(adresses, strings.Split(ip, "/")[0])
|
||||||
|
return adresses
|
||||||
|
}
|
||||||
112
types.go
Normal file
112
types.go
Normal file
@ -0,0 +1,112 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
type device struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
MAC string `json:"mac"`
|
||||||
|
GatewayNexthop string `json:"gateway_nexthop"`
|
||||||
|
Gateway string `json:"gateway"`
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
Location struct {
|
||||||
|
Longitude float64 `json:"longitude"`
|
||||||
|
Latitude float64 `json:"latitude"`
|
||||||
|
} `json:"location"`
|
||||||
|
}
|
||||||
|
type devices struct {
|
||||||
|
Devices []device `json:"devices"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type unifiAPIResponse struct {
|
||||||
|
Identification struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
MAC string `json:"mac"`
|
||||||
|
Started time.Time `json:"started"`
|
||||||
|
} `json:"identification"`
|
||||||
|
Overview struct {
|
||||||
|
LastSeen time.Time `json:"lastSeen"`
|
||||||
|
CreatedAt time.Time `json:"createdAt"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
} `json:"overview"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type unifiAPIDetails struct {
|
||||||
|
Identification struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Model string `json:"model"`
|
||||||
|
} `json:"identification"`
|
||||||
|
Firmware struct {
|
||||||
|
Current string `json:"current"`
|
||||||
|
} `json:"firmware"`
|
||||||
|
Overview struct {
|
||||||
|
CPU float64 `json:"cpu"`
|
||||||
|
RAM float64 `json:"ram"`
|
||||||
|
} `json:"overview"`
|
||||||
|
IPAddress string `json:"ipAddress"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type unifiAPIAirmax struct {
|
||||||
|
DeviceIdentification struct {
|
||||||
|
MAC string `json:"mac"`
|
||||||
|
} `json:"deviceidentification"`
|
||||||
|
Statistics struct {
|
||||||
|
LinkScore float64 `json:"linkScore"`
|
||||||
|
} `json:"statistics"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type link struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
Source string `json:"source"`
|
||||||
|
Target string `json:"target"`
|
||||||
|
SourceTQ float64 `json:"source_tq"`
|
||||||
|
TargetTQ float64 `json:"target_tq"`
|
||||||
|
SourceAddr string `json:"source_addr"`
|
||||||
|
TargetAddr string `json:"target_addr"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type node struct {
|
||||||
|
Firstseen string `json:"firstseen"`
|
||||||
|
Lastseen string `json:"lastseen"`
|
||||||
|
IsOnline bool `json:"is_online"`
|
||||||
|
IsGateway bool `json:"is_gateway"`
|
||||||
|
Clients int `json:"clients"`
|
||||||
|
ClientsWifi24 int `json:"clients_wifi24"`
|
||||||
|
ClientsWifi5 int `json:"clients_wifi5"`
|
||||||
|
ClientsOther int `json:"clients_other"`
|
||||||
|
RootFSUsage int `json:"rootfs_usage"`
|
||||||
|
LoadAVG float64 `json:"loadavg"`
|
||||||
|
MemoryUsage float64 `json:"memory_usage"`
|
||||||
|
Uptime string `json:"uptime"`
|
||||||
|
GatewayNexthop string `json:"gateway_nexthop"`
|
||||||
|
Gateway string `json:"gateway"`
|
||||||
|
Location struct {
|
||||||
|
Longitude float64 `json:"longitude"`
|
||||||
|
Latitude float64 `json:"latitude"`
|
||||||
|
} `json:"location"`
|
||||||
|
NodeID string `json:"node_id"`
|
||||||
|
MAC string `json:"mac"`
|
||||||
|
Adresses []string `json:"addresses"`
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
Hostname string `json:"hostname"`
|
||||||
|
Owner string `json:"owner"`
|
||||||
|
Firmware firmware `json:"firmware"`
|
||||||
|
Autoupdater autoupdater `json:"autoupdater"`
|
||||||
|
NProc int `json:"nproc"`
|
||||||
|
Model string `json:"model"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type firmware struct {
|
||||||
|
Base string `json:"base"`
|
||||||
|
Release string `json:"release"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type autoupdater struct {
|
||||||
|
Enabled bool `json:"enabled"`
|
||||||
|
Branch string `json:"branch"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type output struct {
|
||||||
|
Timestamp string `json:"timestamp"`
|
||||||
|
Nodes []node `json:"nodes"`
|
||||||
|
Links []link `json:"links"`
|
||||||
|
}
|
||||||
137
unifi.ps1
137
unifi.ps1
@ -1,137 +0,0 @@
|
|||||||
|
|
||||||
$URI = "https://unifi.freifunk-troisdorf.de/v2.1/devices"
|
|
||||||
$headers = @{
|
|
||||||
"x-auth-token" = ""
|
|
||||||
}
|
|
||||||
|
|
||||||
$devices = (Get-Content -Path .\devices.json | ConvertFrom-Json).devices
|
|
||||||
|
|
||||||
$request = Invoke-Restmethod -URI $URI -Method GET -Headers $headers
|
|
||||||
|
|
||||||
# Get timestamp
|
|
||||||
$timestamp = Get-Date -UFormat "%Y-%m-%dT%R:00%Z00"
|
|
||||||
$timeDiffFromUTC = Get-Date -UFormat "%Z00"
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# create empty nodes array
|
|
||||||
$nodes = @()
|
|
||||||
$links = @()
|
|
||||||
|
|
||||||
# loop through each device
|
|
||||||
foreach ($device in $devices) {
|
|
||||||
$dev = $request | Where-Object { $_.identification.mac -eq $device.mac}
|
|
||||||
# implement function that reports not found mac addresses in API
|
|
||||||
|
|
||||||
## Get info from API
|
|
||||||
$firstseen = Get-Date -Date $dev.overview.createdAt -UFormat "%Y-%m-%dT%R:00$timeDiffFromUTC"
|
|
||||||
$lastseen = Get-Date -Date $dev.overview.lastSeen -UFormat "%Y-%m-%dT%R:00$timeDiffFromUTC"
|
|
||||||
if ($dev.overview.status -eq "active") {
|
|
||||||
$is_online = $true
|
|
||||||
} else {
|
|
||||||
$is_online = $false
|
|
||||||
}
|
|
||||||
|
|
||||||
$uptime = Get-Date -Date $dev.identification.started -UFormat "%Y-%m-%dT%R:00$timeDiffFromUTC"
|
|
||||||
$mac = $dev.identification.mac
|
|
||||||
$node_id = $mac.replace(":","")
|
|
||||||
|
|
||||||
|
|
||||||
## End of API call 1
|
|
||||||
|
|
||||||
|
|
||||||
# get more info from each device
|
|
||||||
## Get more info from second API call
|
|
||||||
$id = $dev.identification.id
|
|
||||||
$URI = "https://unifi.freifunk-troisdorf.de/v2.1/devices/erouters/$id"
|
|
||||||
$details = Invoke-Restmethod -URI $URI -Method GET -Headers $headers
|
|
||||||
|
|
||||||
$loadavg = $details.overview.cpu / 100
|
|
||||||
$memory_usage = $details.overview.ram / 100
|
|
||||||
|
|
||||||
$addresses = @()
|
|
||||||
# cut away network mask
|
|
||||||
$addresses += $details.ipAddress.split("/")[0]
|
|
||||||
|
|
||||||
$hostname = "[RiFu] " + $details.identification.name
|
|
||||||
|
|
||||||
$firmware = @{
|
|
||||||
base = "Ubiquiti - Stock"
|
|
||||||
release = $details.firmware.current
|
|
||||||
}
|
|
||||||
$model = $details.identification.model
|
|
||||||
|
|
||||||
## End of API call 2
|
|
||||||
|
|
||||||
## API Call 3 (Links)
|
|
||||||
$URI = "https://unifi.freifunk-troisdorf.de/v2.1/devices/airmaxes/$id/stations"
|
|
||||||
$airmaxes = Invoke-Restmethod -URI $URI -Method GET -Headers $headers
|
|
||||||
$remoteMAC = $airmaxes.deviceIdentification.mac
|
|
||||||
|
|
||||||
|
|
||||||
# check if remote mac address is part of our published network
|
|
||||||
if ($devices | Where-Object -Property mac -EQ $remoteMAC) {
|
|
||||||
# this is a public link, let's advertise it
|
|
||||||
if (!($links | Where-Object -Property source_addr -EQ $remoteMAC)) {
|
|
||||||
# this link does not exist already, let's add it
|
|
||||||
$links += @{
|
|
||||||
type = "wifi"
|
|
||||||
source = $mac.replace(":","")
|
|
||||||
target = $remoteMAC.replace(":","")
|
|
||||||
source_tq = $airmaxes.statistics.linkScore
|
|
||||||
target_tq = $airmaxes.statistics.linkScore
|
|
||||||
source_addr = $mac
|
|
||||||
target_addr = $remoteMAC
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
## Get info from json file (static)
|
|
||||||
$currentDev = $devices | Where-Object -Property mac -EQ $dev.identification.mac
|
|
||||||
$gateway_nexthop = $currentDev.gateway_nexthop
|
|
||||||
$gateway = $currentDev.gateway
|
|
||||||
$domain = $currentDev.domain
|
|
||||||
$location = $currentDev.location
|
|
||||||
|
|
||||||
# assemble all into nodes an links
|
|
||||||
$nodes += @{
|
|
||||||
firstseen = $firstseen
|
|
||||||
lastseen = $lastseen
|
|
||||||
is_online = $is_online
|
|
||||||
is_gateway = $false
|
|
||||||
clients = 0
|
|
||||||
clients_wifi24 = 0
|
|
||||||
clients_wifi5 = 0
|
|
||||||
clients_other = 0
|
|
||||||
rootfs_usage = 0
|
|
||||||
loadavg = $loadavg
|
|
||||||
memory_usage = $memory_usage
|
|
||||||
uptime = $uptime
|
|
||||||
gateway_nexthop = $gateway_nexthop
|
|
||||||
gateway = $gateway
|
|
||||||
location = $location
|
|
||||||
node_id = $node_id
|
|
||||||
mac = $mac
|
|
||||||
addresses = $addresses
|
|
||||||
domain = $domain
|
|
||||||
hostname = $hostname
|
|
||||||
owner = "Freifunk Rhein-Sieg"
|
|
||||||
firmware = $firmware
|
|
||||||
autoupdater = @{
|
|
||||||
enabled = $false
|
|
||||||
branch = "stable"
|
|
||||||
}
|
|
||||||
nproc = 1
|
|
||||||
model = $model
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# create file output
|
|
||||||
$output = @{
|
|
||||||
timestamp = $timestamp
|
|
||||||
nodes = $nodes
|
|
||||||
links = $links
|
|
||||||
}
|
|
||||||
|
|
||||||
$output | ConvertTo-Json -Depth 4 | Out-File example.json
|
|
||||||
Loading…
Reference in New Issue
Block a user