mirror of https://github.com/tailscale/tailscale/
cmd/tailscale/cli,client,ipn: add appc-routes cli command
Allow the user to access information about routes an app connector has learned, such as how many routes for each domain. Fixes tailscale/corp#32624 Signed-off-by: Fran Bull <fran@tailscale.com>pull/17340/head
Fran Bull
2 months ago
committed by
franbull
parent
976389c0f7
commit
65d6c80695
@ -0,0 +1,153 @@
|
||||
// Copyright (c) Tailscale Inc & AUTHORS
|
||||
// SPDX-License-Identifier: BSD-3-Clause
|
||||
|
||||
package cli
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"flag"
|
||||
"fmt"
|
||||
"slices"
|
||||
"strings"
|
||||
|
||||
"github.com/peterbourgon/ff/v3/ffcli"
|
||||
"tailscale.com/appc"
|
||||
)
|
||||
|
||||
var appcRoutesArgs struct {
|
||||
all bool
|
||||
domainMap bool
|
||||
n bool
|
||||
}
|
||||
|
||||
var appcRoutesCmd = &ffcli.Command{
|
||||
Name: "appc-routes",
|
||||
ShortUsage: "tailscale appc-routes",
|
||||
Exec: runAppcRoutesInfo,
|
||||
ShortHelp: "Print the current app connector routes",
|
||||
FlagSet: (func() *flag.FlagSet {
|
||||
fs := newFlagSet("appc-routes")
|
||||
fs.BoolVar(&appcRoutesArgs.all, "all", false, "Print learned domains and routes and extra policy configured routes.")
|
||||
fs.BoolVar(&appcRoutesArgs.domainMap, "map", false, "Print the map of learned domains: [routes].")
|
||||
fs.BoolVar(&appcRoutesArgs.n, "n", false, "Print the total number of routes this node advertises.")
|
||||
return fs
|
||||
})(),
|
||||
LongHelp: strings.TrimSpace(`
|
||||
The 'tailscale appc-routes' command prints the current App Connector route status.
|
||||
|
||||
By default this command prints the domains configured in the app connector configuration and how many routes have been
|
||||
learned for each domain.
|
||||
|
||||
--all prints the routes learned from the domains configured in the app connector configuration; and any extra routes provided
|
||||
in the the policy app connector 'routes' field.
|
||||
|
||||
--map prints the routes learned from the domains configured in the app connector configuration.
|
||||
|
||||
-n prints the total number of routes advertised by this device, whether learned, set in the policy, or set locally.
|
||||
|
||||
For more information about App Connectors, refer to
|
||||
https://tailscale.com/kb/1281/app-connectors
|
||||
`),
|
||||
}
|
||||
|
||||
func getAllOutput(ri *appc.RouteInfo) (string, error) {
|
||||
domains, err := json.MarshalIndent(ri.Domains, " ", " ")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
control, err := json.MarshalIndent(ri.Control, " ", " ")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
s := fmt.Sprintf(`Learned Routes
|
||||
==============
|
||||
%s
|
||||
|
||||
Routes from Policy
|
||||
==================
|
||||
%s
|
||||
`, domains, control)
|
||||
return s, nil
|
||||
}
|
||||
|
||||
type domainCount struct {
|
||||
domain string
|
||||
count int
|
||||
}
|
||||
|
||||
func getSummarizeLearnedOutput(ri *appc.RouteInfo) string {
|
||||
x := make([]domainCount, len(ri.Domains))
|
||||
i := 0
|
||||
maxDomainWidth := 0
|
||||
for k, v := range ri.Domains {
|
||||
if len(k) > maxDomainWidth {
|
||||
maxDomainWidth = len(k)
|
||||
}
|
||||
x[i] = domainCount{domain: k, count: len(v)}
|
||||
i++
|
||||
}
|
||||
slices.SortFunc(x, func(i, j domainCount) int {
|
||||
if i.count > j.count {
|
||||
return -1
|
||||
}
|
||||
if i.count < j.count {
|
||||
return 1
|
||||
}
|
||||
if i.domain > j.domain {
|
||||
return 1
|
||||
}
|
||||
if i.domain < j.domain {
|
||||
return -1
|
||||
}
|
||||
return 0
|
||||
})
|
||||
s := ""
|
||||
fmtString := fmt.Sprintf("%%-%ds %%d\n", maxDomainWidth) // eg "%-10s %d\n"
|
||||
for _, dc := range x {
|
||||
s += fmt.Sprintf(fmtString, dc.domain, dc.count)
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
func runAppcRoutesInfo(ctx context.Context, args []string) error {
|
||||
prefs, err := localClient.GetPrefs(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if !prefs.AppConnector.Advertise {
|
||||
fmt.Println("not a connector")
|
||||
return nil
|
||||
}
|
||||
|
||||
if appcRoutesArgs.n {
|
||||
fmt.Println(len(prefs.AdvertiseRoutes))
|
||||
return nil
|
||||
}
|
||||
|
||||
routeInfo, err := localClient.GetAppConnectorRouteInfo(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if appcRoutesArgs.domainMap {
|
||||
domains, err := json.Marshal(routeInfo.Domains)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
fmt.Println(string(domains))
|
||||
return nil
|
||||
}
|
||||
|
||||
if appcRoutesArgs.all {
|
||||
s, err := getAllOutput(&routeInfo)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
fmt.Println(s)
|
||||
return nil
|
||||
}
|
||||
|
||||
fmt.Print(getSummarizeLearnedOutput(&routeInfo))
|
||||
return nil
|
||||
}
|
||||
Loading…
Reference in New Issue