torjus/ezshare
1
0
Fork 0
Code Issues Pull Requests Projects Releases 2 Wiki Activity

Reduce code duplication in client actions

Browse Source
This commit is contained in:
Torjus Håkestad 2021-12-08 15:07:39 +01:00
parent 5c858e57fc
commit 3a162ab974
2 changed files with 51 additions and 81 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

131
actions/client.go
View File

@ -28,9 +28,15 @@ import (
"google.golang.org/protobuf/types/known/timestamppb"
)
type contextKey string
const (
contextKeyConfig contextKey = "config"
contextKeyClientConnFunc contextKey = "clientConnFunc"
)
func ActionClientGet(c *cli.Context) error {
addr := c.String("addr")
conn, err := grpc.DialContext(c.Context, addr, grpc.WithInsecure())
conn, err := connFromContext(c)
if err != nil {
return err
}
@ -65,22 +71,7 @@ func ActionClientGet(c *cli.Context) error {
}
func ActionClientUpload(c *cli.Context) error {
cfg, err := getConfig(c)
if err != nil {
return err
}
addr := cfg.Client.DefaultServer
if c.IsSet("addr") {
addr = c.String("addr")
}
clientCreds, err := cfg.Client.Creds()
if err != nil {
return err
}
conn, err := grpc.DialContext(c.Context, addr, grpc.WithTransportCredentials(clientCreds))
conn, err := connFromContext(c)
if err != nil {
return err
}
@ -118,22 +109,7 @@ func ActionClientUpload(c *cli.Context) error {
}
func ActionClientList(c *cli.Context) error {
cfg, err := getConfig(c)
if err != nil {
return err
}
addr := cfg.Client.DefaultServer
if c.IsSet("addr") {
addr = c.String("addr")
}
clientCreds, err := cfg.Client.Creds()
if err != nil {
return err
}
conn, err := grpc.DialContext(c.Context, addr, grpc.WithTransportCredentials(clientCreds))
conn, err := connFromContext(c)
if err != nil {
return err
}
@ -153,22 +129,7 @@ func ActionClientList(c *cli.Context) error {
}
func ActionClientDelete(c *cli.Context) error {
cfg, err := getConfig(c)
if err != nil {
return err
}
addr := cfg.Client.DefaultServer
if c.IsSet("addr") {
addr = c.String("addr")
}
clientCreds, err := cfg.Client.Creds()
if err != nil {
return err
}
conn, err := grpc.DialContext(c.Context, addr, grpc.WithTransportCredentials(clientCreds))
conn, err := connFromContext(c)
if err != nil {
return err
}
@ -362,22 +323,7 @@ func ActionClientLogin(c *cli.Context) error {
}
func ActionClientPassword(c *cli.Context) error {
cfg, err := getConfig(c)
if err != nil {
return err
}
addr := cfg.Client.DefaultServer
if c.IsSet("addr") {
addr = c.String("addr")
}
clientCreds, err := cfg.Client.Creds()
if err != nil {
return err
}
conn, err := grpc.DialContext(c.Context, addr, grpc.WithTransportCredentials(clientCreds))
conn, err := connFromContext(c)
if err != nil {
return err
}
@ -406,21 +352,7 @@ func ActionClientPassword(c *cli.Context) error {
}
func ActionClientUpdate(c *cli.Context) error {
cfg, err := getConfig(c)
if err != nil {
return err
}
addr := cfg.Client.DefaultServer
if c.IsSet("addr") {
addr = c.String("addr")
}
clientCreds, err := cfg.Client.Creds()
if err != nil {
return err
}
conn, err := grpc.DialContext(c.Context, addr, grpc.WithTransportCredentials(clientCreds))
conn, err := connFromContext(c)
if err != nil {
return err
@ -467,3 +399,40 @@ func ActionClientUpdate(c *cli.Context) error {
fmt.Printf("Wrote latest binary to %s", outputPath)
return nil
}
func BeforeClient(c *cli.Context) error {
cfg, err := getConfig(c)
if err != nil {
return nil
}
c.Context = context.WithValue(c.Context, contextKeyConfig, cfg)
addr := cfg.Client.DefaultServer
if c.IsSet("addr") {
addr = c.String("addr")
}
clientCreds, err := cfg.Client.Creds()
if err != nil {
return nil
}
connFunc := func() (*grpc.ClientConn, error) {
return grpc.DialContext(c.Context, addr, grpc.WithTransportCredentials(clientCreds))
}
c.Context = context.WithValue(c.Context, contextKeyClientConnFunc, connFunc)
return nil
}
func connFromContext(c *cli.Context) (*grpc.ClientConn, error) {
connFunc := c.Context.Value(contextKeyClientConnFunc).(func() (*grpc.ClientConn, error))
return connFunc()
}
func configFromContext(c *cli.Context) (*config.Config, error) {
cfg, ok := c.Context.Value(contextKeyConfig).(*config.Config)
if !ok {
return getConfig(c)
}
return cfg, nil
}

1
main.go
View File

@ -62,6 +62,7 @@ func main() {
Usage: "Address of server.",
},
},
Before: actions.BeforeClient,
Subcommands: []*cli.Command{
{
Name: "get",