mirror of https://github.com/tailscale/tailscale/
cmd/tailscale/cli: add login and switch subcommands
Updates #713 Signed-off-by: Maisem Ali <maisem@tailscale.com>pull/6365/head
Maisem Ali
3 years ago
committed by
Maisem Ali
parent
ec1e67b1ab
commit
f3519f7b29
@ -0,0 +1,32 @@
|
|||||||
|
// 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"
|
||||||
|
|
||||||
|
"github.com/peterbourgon/ff/v3/ffcli"
|
||||||
|
)
|
||||||
|
|
||||||
|
var loginArgs upArgsT
|
||||||
|
|
||||||
|
var loginCmd = &ffcli.Command{
|
||||||
|
Name: "login",
|
||||||
|
ShortUsage: "[ALPHA] login [flags]",
|
||||||
|
ShortHelp: "Log in to a Tailscale account",
|
||||||
|
LongHelp: `"tailscale login" logs this machine in to your Tailscale network.
|
||||||
|
This command is currently in alpha and may change in the future.`,
|
||||||
|
UsageFunc: usageFunc,
|
||||||
|
FlagSet: func() *flag.FlagSet {
|
||||||
|
return newUpFlagSet(effectiveGOOS(), &loginArgs, "login")
|
||||||
|
}(),
|
||||||
|
Exec: func(ctx context.Context, args []string) error {
|
||||||
|
if err := localClient.SwitchToEmptyProfile(ctx); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return runUp(ctx, args, loginArgs)
|
||||||
|
},
|
||||||
|
}
|
||||||
@ -0,0 +1,122 @@
|
|||||||
|
// 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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue