feat(mock): quick way to add packages (#417)

Author: reubeno

internal/app/azldev/cmds/advanced/mock.go

@@ -78,6 +78,9 @@ type ShellOptions struct {
 
 	// Whether or not to enable external network access from within the mock root the shell is executed in.
 	EnableNetwork bool
+
+	// Packages to add to the mock root before starting the shell.
+	PackagesToAdd []string
 }
 
 // Constructs a [cobra.Command] for the 'mock shell' command.
@@ -97,6 +100,9 @@ func NewShellCmd() *cobra.Command {
 
 	cmd.Flags().StringVarP(&options.MockConfigPath, "config", "c", "", "Path to the mock .cfg file")
 	cmd.Flags().BoolVar(&options.EnableNetwork, "enable-network", false, "Enable network access in the mock root")
+	cmd.Flags().StringArrayVarP(&options.PackagesToAdd, "add-package", "p", []string{},
+		"Package to add to the mock root before starting the shell",
+	)
 
 	return cmd
 }
@@ -135,6 +141,13 @@ func RunShell(env *azldev.Env, options *ShellOptions, extraArgs []string) error
 		runner.EnableNetwork()
 	}
 
+	if len(options.PackagesToAdd) > 0 {
+		err = runner.InstallPackages(env, options.PackagesToAdd)
+		if err != nil {
+			return fmt.Errorf("failed to install packages in mock root (%v): %w", options.PackagesToAdd, err)
+		}
+	}
+
 	cmd, err := runner.CmdInChroot(env, extraArgs, true /*interactive*/)
 	if err != nil {
 		return fmt.Errorf("failed to create shell command: %w", err)

internal/rpm/mock/mock.go

@@ -507,6 +507,36 @@ func (r *Runner) CmdInChroot(ctx context.Context, args []string, interactive boo
 	return cmd, nil
 }
 
+// InstallPackages installs the specified packages into the mock root.
+func (r *Runner) InstallPackages(ctx context.Context, packages []string) error {
+	err := r.ensureMockPresentAndConfigured()
+	if err != nil {
+		return err
+	}
+
+	mockArgs := r.getBaseArgs()
+	mockArgs = append(mockArgs, "--install")
+	mockArgs = append(mockArgs, packages...)
+
+	cmd := exec.CommandContext(ctx, MockBinary, mockArgs...)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	extcmd, err := r.cmdFactory.Command(cmd)
+	if err != nil {
+		return fmt.Errorf("failed to create external command for mock:\n%w", err)
+	}
+
+	extcmd = extcmd.SetLongRunning("Waiting for mock (installing packages)...")
+
+	err = extcmd.Run(ctx)
+	if err != nil {
+		return fmt.Errorf("mock failed to install packages (%v):\n%w", packages, err)
+	}
+
+	return nil
+}
+
 // ScrubRoot removes the mock root.
 func (r *Runner) ScrubRoot(ctx context.Context) error {
 	err := r.ensureMockPresentAndConfigured()