mirror of https://github.com/tailscale/tailscale/
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
123 lines
2.6 KiB
Go
123 lines
2.6 KiB
Go
2 years ago
|
// Copyright (c) 2022 Tailscale Inc & AUTHORS All rights reserved.
|
||
|
// Use of this source code is governed by a BSD-style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
package cli
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"flag"
|
||
|
"fmt"
|
||
|
"os"
|
||
|
"time"
|
||
|
|
||
|
"github.com/peterbourgon/ff/v3/ffcli"
|
||
|
"tailscale.com/ipn"
|
||
|
)
|
||
|
|
||
|
var switchCmd = &ffcli.Command{
|
||
|
Name: "switch",
|
||
|
ShortHelp: "Switches to a different Tailscale profile",
|
||
|
FlagSet: func() *flag.FlagSet {
|
||
|
fs := flag.NewFlagSet("switch", flag.ExitOnError)
|
||
|
fs.BoolVar(&switchArgs.list, "list", false, "list available profiles")
|
||
|
return fs
|
||
|
}(),
|
||
|
Exec: switchProfile,
|
||
|
UsageFunc: func(*ffcli.Command) string {
|
||
|
return `USAGE
|
||
|
[ALPHA] switch <name>
|
||
|
[ALPHA] switch --list
|
||
|
|
||
|
"tailscale switch" switches between logged in profiles.
|
||
|
This command is currently in alpha and may change in the future.`
|
||
|
},
|
||
|
}
|
||
|
|
||
|
var switchArgs struct {
|
||
|
list bool
|
||
|
}
|
||
|
|
||
|
func listProfiles(ctx context.Context) error {
|
||
|
curP, all, err := localClient.ProfileStatus(ctx)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
for _, prof := range all {
|
||
|
if prof.ID == curP.ID {
|
||
|
fmt.Printf("%s *\n", prof.Name)
|
||
|
} else {
|
||
|
fmt.Println(prof.Name)
|
||
|
}
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func switchProfile(ctx context.Context, args []string) error {
|
||
|
if switchArgs.list {
|
||
|
return listProfiles(ctx)
|
||
|
}
|
||
|
if len(args) != 1 {
|
||
|
outln("usage: tailscale profile switch NAME")
|
||
|
os.Exit(1)
|
||
|
}
|
||
|
cp, all, err := localClient.ProfileStatus(ctx)
|
||
|
if err != nil {
|
||
|
errf("Failed to switch to profile: %v\n", err)
|
||
|
os.Exit(1)
|
||
|
}
|
||
|
var profID ipn.ProfileID
|
||
|
for _, p := range all {
|
||
|
if p.Name == args[0] {
|
||
|
profID = p.ID
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
if profID == "" {
|
||
|
errf("No profile named %q\n", args[0])
|
||
|
os.Exit(1)
|
||
|
}
|
||
|
if profID == cp.ID {
|
||
|
printf("Already on profile %q\n", args[0])
|
||
|
os.Exit(0)
|
||
|
}
|
||
|
if err := localClient.SwitchProfile(ctx, profID); err != nil {
|
||
|
errf("Failed to switch to profile: %v\n", err)
|
||
|
os.Exit(1)
|
||
|
}
|
||
|
printf("Switching to profile %q\n", args[0])
|
||
|
for {
|
||
|
select {
|
||
|
case <-ctx.Done():
|
||
|
errf("Timed out waiting for switch to complete.")
|
||
|
os.Exit(1)
|
||
|
default:
|
||
|
}
|
||
|
st, err := localClient.StatusWithoutPeers(ctx)
|
||
|
if err != nil {
|
||
|
errf("Error getting status: %v", err)
|
||
|
os.Exit(1)
|
||
|
}
|
||
|
switch st.BackendState {
|
||
|
case "NoState", "Starting":
|
||
|
// TODO(maisem): maybe add a way to subscribe to state changes to
|
||
|
// LocalClient.
|
||
|
time.Sleep(100 * time.Millisecond)
|
||
|
continue
|
||
|
case "NeedsLogin":
|
||
|
outln("Logged out.")
|
||
|
outln("To log in, run:")
|
||
|
outln(" tailscale up")
|
||
|
return nil
|
||
|
case "Running":
|
||
|
outln("Success.")
|
||
|
return nil
|
||
|
}
|
||
|
// For all other states, use the default error message.
|
||
|
if msg, ok := isRunningOrStarting(st); !ok {
|
||
|
outln(msg)
|
||
|
os.Exit(1)
|
||
|
}
|
||
|
}
|
||
|
}
|