-
Notifications
You must be signed in to change notification settings - Fork 234
/
connect.go
115 lines (88 loc) · 2.34 KB
/
connect.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package redis
import (
"context"
"fmt"
"os/exec"
"github.com/spf13/cobra"
"github.com/superfly/flyctl/agent"
"github.com/superfly/flyctl/gql"
"github.com/superfly/flyctl/iostreams"
"github.com/superfly/flyctl/proxy"
"github.com/superfly/flyctl/client"
"github.com/superfly/flyctl/internal/command"
"github.com/superfly/flyctl/internal/flag"
"github.com/superfly/flyctl/internal/prompt"
)
func newConnect() (cmd *cobra.Command) {
const (
long = `Connect to a Redis database using redis-cli`
short = long
usage = "connect"
)
cmd = command.New(usage, short, long, runConnect, command.RequireSession)
flag.Add(cmd,
flag.Org(),
flag.Region(),
)
return cmd
}
func runConnect(ctx context.Context) (err error) {
var (
client = client.FromContext(ctx).API()
io = iostreams.FromContext(ctx)
)
if err != nil {
return err
}
var index int
var options []string
result, err := gql.ListAddOns(ctx, client.GenqClient, "redis")
if err != nil {
return
}
databases := result.AddOns.Nodes
for _, database := range databases {
options = append(options, fmt.Sprintf("%s (%s) %s", database.Name, database.PrimaryRegion, database.Organization.Slug))
}
err = prompt.Select(ctx, &index, "Select a database to connect to", "", options...)
if err != nil {
return
}
response, err := gql.GetAddOn(ctx, client.GenqClient, databases[index].Name)
if err != nil {
return err
}
database := response.AddOn
agentclient, err := agent.Establish(ctx, client)
if err != nil {
return err
}
dialer, err := agentclient.ConnectToTunnel(ctx, database.Organization.Slug)
if err != nil {
return err
}
localProxyPort := "16379"
params := &proxy.ConnectParams{
Ports: []string{localProxyPort, "6379"},
OrganizationSlug: database.Organization.Slug,
Dialer: dialer,
RemoteHost: database.PrivateIp,
}
redisCliPath, err := exec.LookPath("redis-cli")
if err != nil {
fmt.Fprintf(io.Out, "Could not find redis-cli in your $PATH. Install it or point your redis-cli at: %s", "someurl")
return
}
err = proxy.Start(ctx, params)
if err != nil {
return err
}
cmd := exec.CommandContext(ctx, redisCliPath, "-p", localProxyPort)
cmd.Env = append(cmd.Env, fmt.Sprintf("REDISCLI_AUTH=%s", database.Password))
cmd.Stdout = io.Out
cmd.Stderr = io.ErrOut
cmd.Stdin = io.In
cmd.Start()
cmd.Wait()
return
}