Make cog build logs follow by default

This is a user's intention when using the command. They might
want to not follow if it's a long-running process or they are
using it in a script. The former is not the case with builds and
the latter they can use the flag.

Signed-off-by: Ben Firshman <ben@firshman.co.uk>

Author: bfirsh

pkg/cli/build.go

@@ -8,6 +8,8 @@ import (
 	"github.com/replicate/cog/pkg/util/console"
 )
 
+var buildNoFollow bool
+
 func newBuildCommand() *cobra.Command {
 	cmd := &cobra.Command{
 		Use:   "build",
@@ -30,26 +32,23 @@ func newBuildLogCommand() *cobra.Command {
 
 	addModelFlag(cmd)
 
-	cmd.Flags().BoolP("follow", "f", false, "Follow streaming logs")
+	cmd.Flags().BoolVarP(&buildNoFollow, "no-follow", "", false, "Exit immediately instead of waiting until output finishes")
 	// TODO(andreas): tail
 
 	return cmd
 }
 
 func showBuildLogs(cmd *cobra.Command, args []string) error {
 	buildID := args[0]
-	follow, err := cmd.Flags().GetBool("follow")
-	if err != nil {
-		return err
-	}
 
 	model, err := getModel()
 	if err != nil {
 		return err
 	}
 
 	c := client.NewClient()
-	logChan, err := c.GetBuildLogs(model, buildID, follow)
+	// FIXME(bfirsh): why isn't this a logger.Logger?
+	logChan, err := c.GetBuildLogs(model, buildID, !buildNoFollow)
 	if err != nil {
 		return err
 	}