cu diff

- Add support for `cu diff`
- Only show logs and diffs for changes that happened in the environment
- Move `cu log` code into Repository
- Add `cu log -p` and don't show patches by default
- Show notes in `cu log`

Signed-off-by: Andrea Luzzardi <andrea@luzzardi.com>

Author: aluzzardi

cmd/cu/diff.go

@@ -0,0 +1,30 @@
+package main
+
+import (
+	"os"
+
+	"github.com/dagger/container-use/repository"
+	"github.com/spf13/cobra"
+)
+
+var diffCmd = &cobra.Command{
+	Use:               "diff <env>",
+	Short:             "Show changes between the environment and the local branch",
+	Args:              cobra.ExactArgs(1),
+	ValidArgsFunction: suggestEnvironments,
+	RunE: func(app *cobra.Command, args []string) error {
+		ctx := app.Context()
+
+		// Ensure we're in a git repository
+		repo, err := repository.Open(ctx, ".")
+		if err != nil {
+			return err
+		}
+
+		return repo.Diff(ctx, args[0], os.Stdout)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(diffCmd)
+}

cmd/cu/log.go

@@ -2,8 +2,6 @@ package main
 
 import (
 	"os"
-	"os/exec"
-	"strings"
 
 	"github.com/dagger/container-use/repository"
 	"github.com/spf13/cobra"
@@ -18,22 +16,18 @@ var logCmd = &cobra.Command{
 		ctx := app.Context()
 
 		// Ensure we're in a git repository
-		if _, err := repository.Open(ctx, "."); err != nil {
+		repo, err := repository.Open(ctx, ".")
+		if err != nil {
 			return err
 		}
 
-		env := args[0]
-		// prevent accidental single quotes to mess up command
-		env = strings.Trim(env, "'")
-		cmd := exec.CommandContext(app.Context(), "git", "log", "--patch", "container-use/"+env)
-		cmd.Stderr = os.Stderr
-		cmd.Stdin = os.Stdin
-		cmd.Stdout = os.Stdout
+		patch, _ := app.Flags().GetBool("patch")
 
-		return cmd.Run()
+		return repo.Log(ctx, args[0], patch, os.Stdout)
 	},
 }
 
 func init() {
+	logCmd.Flags().BoolP("patch", "p", false, "Generate patch")
 	rootCmd.AddCommand(logCmd)
 }

repository/git.go

@@ -235,6 +235,33 @@ func (r *Repository) addGitNote(ctx context.Context, env *environment.Environmen
 	return r.propagateGitNotes(ctx, gitNotesLogRef)
 }
 
+func (r *Repository) currentUserBranch(ctx context.Context) (string, error) {
+	return RunGitCommand(ctx, r.userRepoPath, "branch", "--show-current")
+}
+
+func (r *Repository) mergeBase(ctx context.Context, env *environment.EnvironmentInfo) (string, error) {
+	currentBranch, err := r.currentUserBranch(ctx)
+	if err != nil {
+		return "", err
+	}
+	currentBranch = strings.TrimSpace(currentBranch)
+	envGitRef := fmt.Sprintf("%s/%s", containerUseRemote, env.ID)
+	mergeBase, err := RunGitCommand(ctx, r.userRepoPath, "merge-base", currentBranch, envGitRef)
+	if err != nil {
+		return "", err
+	}
+	return strings.TrimSpace(mergeBase), nil
+}
+
+func (r *Repository) revisionRange(ctx context.Context, env *environment.EnvironmentInfo) (string, error) {
+	mergeBase, err := r.mergeBase(ctx, env)
+	if err != nil {
+		return "", err
+	}
+	envGitRef := fmt.Sprintf("%s/%s", containerUseRemote, env.ID)
+	return fmt.Sprintf("%s..%s", mergeBase, envGitRef), nil
+}
+
 func (r *Repository) commitWorktreeChanges(ctx context.Context, worktreePath, name, explanation string) error {
 	status, err := RunGitCommand(ctx, worktreePath, "status", "--porcelain")
 	if err != nil {

repository/repository.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"io"
 	"log/slog"
 	"os"
 	"os/exec"
@@ -359,3 +360,58 @@ func (r *Repository) Checkout(ctx context.Context, id string) (string, error) {
 
 	return branch, err
 }
+
+func (r *Repository) Log(ctx context.Context, id string, patch bool, w io.Writer) error {
+	envInfo, err := r.Info(ctx, id)
+	if err != nil {
+		return err
+	}
+
+	logArgs := []string{
+		"git",
+		"log",
+		fmt.Sprintf("--notes=%s", gitNotesLogRef),
+	}
+
+	if patch {
+		logArgs = append(logArgs, "--patch")
+	}
+
+	revisionRange, err := r.revisionRange(ctx, envInfo)
+	if err != nil {
+		return err
+	}
+
+	logArgs = append(logArgs, revisionRange)
+
+	cmd := exec.CommandContext(ctx, "git")
+	cmd.Args = logArgs
+	cmd.Stdout = w
+
+	return cmd.Run()
+}
+
+func (r *Repository) Diff(ctx context.Context, id string, w io.Writer) error {
+	envInfo, err := r.Info(ctx, id)
+	if err != nil {
+		return err
+	}
+
+	diffArgs := []string{
+		"git",
+		"diff",
+	}
+
+	revisionRange, err := r.revisionRange(ctx, envInfo)
+	if err != nil {
+		return err
+	}
+
+	diffArgs = append(diffArgs, revisionRange)
+
+	cmd := exec.CommandContext(ctx, "git")
+	cmd.Args = diffArgs
+	cmd.Stdout = w
+
+	return cmd.Run()
+}