Updates #713 Signed-off-by: Maisem Ali <maisem@tailscale.com>main
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