diff --git a/main.go b/main.go index 482305f..2d41bf8 100644 --- a/main.go +++ b/main.go @@ -25,12 +25,6 @@ func main() { } log.Printf("Network interface: %s", link.Attrs().Name) - upstream, err := getUpstream(link.Attrs().Name) - if err != nil { - log.Fatal(err) - } - log.Printf("Connected to %s on %s", upstream.Name, upstream.Port) - api, err := findAPI(link) if err != nil { log.Fatal(err) @@ -47,6 +41,12 @@ func main() { log.Fatal(err) } + upstream, err := getUpstream(link.Attrs().Name) + if err != nil { + log.Fatal(err) + } + log.Printf("Connected to %s on %s", upstream.Name, upstream.Port) + devices, err := unifi.ListDevices() if err != nil { log.Fatal(err) diff --git a/unifi.go b/unifi.go index fc69535..c29681a 100644 --- a/unifi.go +++ b/unifi.go @@ -210,7 +210,11 @@ func (d *Device) PathHop(path []*Device, visited map[string]*Device) { } func (d *Device) LogHop(prefix string, last bool) { - log.Printf("%s+- %s", prefix, d.Name) + if last { + log.Printf("%s└─ %s", prefix, d.Name) + } else { + log.Printf("%s├─ %s", prefix, d.Name) + } names := []string{} for _, next := range d.Beyond { @@ -224,7 +228,7 @@ func (d *Device) LogHop(prefix string, last bool) { if last { next.LogHop(fmt.Sprintf("%s ", prefix), i == len(names) - 1) } else { - next.LogHop(fmt.Sprintf("%s| ", prefix), i == len(names) - 1) + next.LogHop(fmt.Sprintf("%s│ ", prefix), i == len(names) - 1) } i++ }