cmd: ported code to new client function signature, force one client per session

Signed-off-by: Sam Alba <sam.alba@gmail.com>

cmd/dagger/cmd/compute.go

@@ -168,29 +168,32 @@ var computeCmd = &cobra.Command{
 
 		cl := common.NewClient(ctx, viper.GetBool("no-cache"))
 
-		environment, err := cl.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
+		err := cl.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
 			// check that all inputs are set
 			checkInputs(ctx, env)
 
-			return env.Up(ctx, s)
+			if err := env.Up(ctx, s); err != nil {
+				return err
+			}
+
+			v := compiler.NewValue()
+			if err := v.FillPath(cue.MakePath(), env.Plan()); err != nil {
+				return err
+			}
+			if err := v.FillPath(cue.MakePath(), env.Input()); err != nil {
+				return err
+			}
+			if err := v.FillPath(cue.MakePath(), env.Computed()); err != nil {
+				return err
+			}
+
+			fmt.Println(v.JSON())
+			return nil
 		})
 
 		if err != nil {
 			lg.Fatal().Err(err).Msg("failed to up environment")
 		}
-
-		v := compiler.NewValue()
-		if err := v.FillPath(cue.MakePath(), environment.Plan()); err != nil {
-			lg.Fatal().Err(err).Msg("failed to merge")
-		}
-		if err := v.FillPath(cue.MakePath(), environment.Input()); err != nil {
-			lg.Fatal().Err(err).Msg("failed to merge")
-		}
-		if err := v.FillPath(cue.MakePath(), environment.Computed()); err != nil {
-			lg.Fatal().Err(err).Msg("failed to merge")
-		}
-
-		fmt.Println(v.JSON())
 	},
 }
 

cmd/dagger/cmd/edit.go

@@ -71,7 +71,7 @@ var editCmd = &cobra.Command{
 		st.Inputs = newState.Inputs
 
 		cl := common.NewClient(ctx, false)
-		_, err = cl.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
+		err = cl.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
 			// check for cue errors by scanning all the inputs
 			_, err := env.ScanInputs(ctx, true)
 			if err != nil {

cmd/dagger/cmd/input/list.go

@@ -45,7 +45,7 @@ var listCmd = &cobra.Command{
 			lg.Fatal().Err(err).Msg("unable to create client")
 		}
 
-		_, err = c.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
+		err = c.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
 			inputs, err := env.ScanInputs(ctx, false)
 			if err != nil {
 				return err

cmd/dagger/cmd/input/root.go

@@ -43,7 +43,7 @@ func updateEnvironmentInput(ctx context.Context, cl *client.Client, target strin
 
 	st.SetInput(target, input)
 
-	_, err := cl.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
+	err := cl.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
 		// the inputs are set, check for cue errors by scanning all the inputs
 		_, err := env.ScanInputs(ctx, true)
 		if err != nil {

cmd/dagger/cmd/output/list.go

@@ -6,12 +6,10 @@ import (
 	"os"
 	"text/tabwriter"
 
-	"go.dagger.io/dagger/client"
 	"go.dagger.io/dagger/cmd/dagger/cmd/common"
 	"go.dagger.io/dagger/cmd/dagger/logger"
 	"go.dagger.io/dagger/environment"
 	"go.dagger.io/dagger/solver"
-	"go.dagger.io/dagger/state"
 
 	"github.com/rs/zerolog/log"
 	"github.com/spf13/cobra"
@@ -36,23 +34,25 @@ var listCmd = &cobra.Command{
 		workspace := common.CurrentWorkspace(ctx)
 		st := common.CurrentEnvironmentState(ctx, workspace)
 
-		ListOutputs(ctx, st, true)
+		cl := common.NewClient(ctx, false)
+		err := cl.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
+			return ListOutputs(ctx, env, true)
+		})
+
+		if err != nil {
+			lg.Fatal().Err(err).Msg("failed to scan outputs")
+		}
 	},
 }
 
-func ListOutputs(ctx context.Context, st *state.State, isTTY bool) {
+func ListOutputs(ctx context.Context, env *environment.Environment, isTTY bool) error {
 	lg := log.Ctx(ctx).With().
-		Str("environment", st.Name).
+		Str("environment", env.Name()).
 		Logger()
 
-	c, err := client.New(ctx, "", false)
-	if err != nil {
-		lg.Fatal().Err(err).Msg("unable to create client")
-	}
-
-	_, err = c.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
 	outputs, err := env.ScanOutputs(ctx)
 	if err != nil {
+		lg.Error().Err(err).Msg("failed to scan outputs")
 		return err
 	}
 
@@ -78,9 +78,4 @@ func ListOutputs(ctx context.Context, st *state.State, isTTY bool) {
 
 	w.Flush()
 	return nil
-	})
-
-	if err != nil {
-		lg.Fatal().Err(err).Msg("failed to query environment")
-	}
 }

cmd/dagger/cmd/query.go

@@ -1,13 +1,15 @@
 package cmd
 
 import (
+	"context"
 	"fmt"
 
 	"cuelang.org/go/cue"
-	"go.dagger.io/dagger/client"
 	"go.dagger.io/dagger/cmd/dagger/cmd/common"
 	"go.dagger.io/dagger/cmd/dagger/logger"
 	"go.dagger.io/dagger/compiler"
+	"go.dagger.io/dagger/environment"
+	"go.dagger.io/dagger/solver"
 
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
@@ -42,29 +44,27 @@ var queryCmd = &cobra.Command{
 			cuePath = cue.ParsePath(args[0])
 		}
 
-		c, err := client.New(ctx, "", false)
+		cl := common.NewClient(ctx, false)
-		if err != nil {
-			lg.Fatal().Err(err).Msg("unable to create client")
-		}
-
-		environment, err := c.Do(ctx, state, nil)
-		if err != nil {
-			lg.Fatal().Err(err).Msg("failed to query environment")
-		}
-
 		cueVal := compiler.NewValue()
 
+		err := cl.Do(ctx, state, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
 			if !viper.GetBool("no-plan") {
-			if err := cueVal.FillPath(cue.MakePath(), environment.Plan()); err != nil {
+				if err := cueVal.FillPath(cue.MakePath(), env.Plan()); err != nil {
-				lg.Fatal().Err(err).Msg("failed to merge plan")
+					return err
 				}
 			}
 
 			if !viper.GetBool("no-input") {
-			if err := cueVal.FillPath(cue.MakePath(), environment.Input()); err != nil {
+				if err := cueVal.FillPath(cue.MakePath(), env.Input()); err != nil {
-				lg.Fatal().Err(err).Msg("failed to merge plan with output")
+					return err
 				}
 			}
+			return nil
+		})
+
+		if err != nil {
+			lg.Fatal().Err(err).Msg("failed to query environment")
+		}
 
 		if !viper.GetBool("no-computed") && state.Computed != "" {
 			computed, err := compiler.DecodeJSON("", []byte(state.Computed))

cmd/dagger/cmd/up.go

@@ -38,34 +38,39 @@ var upCmd = &cobra.Command{
 
 		cl := common.NewClient(ctx, viper.GetBool("no-cache"))
 
-		result, err := cl.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
+		err := cl.Do(ctx, st, func(ctx context.Context, env *environment.Environment, s solver.Solver) error {
 			// check that all inputs are set
-			checkInputs(ctx, env)
+			if err := checkInputs(ctx, env); err != nil {
+				return err
+			}
 
-			return env.Up(ctx, s)
+			if err := env.Up(ctx, s); err != nil {
+				return err
+			}
+
+			st.Computed = env.Computed().JSON().PrettyString()
+			if err := workspace.Save(ctx, st); err != nil {
+				return err
+			}
+
+			return output.ListOutputs(ctx, env, term.IsTerminal(int(os.Stdout.Fd())))
 		})
 
 		if err != nil {
 			lg.Fatal().Err(err).Msg("failed to up environment")
 		}
-
-		st.Computed = result.Computed().JSON().PrettyString()
-		if err := workspace.Save(ctx, st); err != nil {
-			lg.Fatal().Err(err).Msg("failed to update environment")
-		}
-
-		output.ListOutputs(ctx, st, term.IsTerminal(int(os.Stdout.Fd())))
 	},
 }
 
-func checkInputs(ctx context.Context, env *environment.Environment) {
+func checkInputs(ctx context.Context, env *environment.Environment) error {
 	lg := log.Ctx(ctx)
 	warnOnly := viper.GetBool("force")
 
 	notConcreteInputs := []*compiler.Value{}
 	inputs, err := env.ScanInputs(ctx, true)
 	if err != nil {
-		lg.Fatal().Err(err).Msg("failed to scan inputs")
+		lg.Error().Err(err).Msg("failed to scan inputs")
+		return err
 	}
 
 	for _, i := range inputs {
@@ -83,8 +88,10 @@ func checkInputs(ctx context.Context, env *environment.Environment) {
 	}
 
 	if !warnOnly && len(notConcreteInputs) > 0 {
-		lg.Fatal().Int("missing", len(notConcreteInputs)).Msg("some required inputs are not set, please re-run with `--force` if you think it's a mistake")
+		lg.Error().Int("missing", len(notConcreteInputs)).Msg("some required inputs are not set, please re-run with `--force` if you think it's a mistake")
 	}
+
+	return nil
 }
 
 func init() {