feat: register the local repo for shell and run by default

Traditionally, only `build` allows oma to use the local repo. This
patch ports this feature to `shell` and `run` for a better debugging
experience.

Author: chenx97

src/actions/container.rs

@@ -16,7 +16,7 @@ use crate::{
     config, error, info,
     machine::{self, get_container_ns_name, inspect_instance, spawn_container},
     network::download_file_progress,
-    overlayfs, warn,
+    overlayfs, repo, warn,
 };
 
 use super::{for_each_instance, APT_UPDATE_SCRIPT};
@@ -298,6 +298,28 @@ pub fn start_container(instance: &str) -> Result<String> {
     Ok(ns_name)
 }
 
+/// Prepare the local repository for the container
+pub fn prepare_local_repo(instance: &str) -> Result<()> {
+    let conf = config::read_config();
+    if conf.is_err() {
+        return Err(anyhow!("Please configure this workspace first!"));
+    }
+    let conf = conf.unwrap();
+    if !conf.local_repo {
+        return Err(anyhow!("Please enable local packages repository first!"));
+    }
+    let ns_name = get_instance_ns_name(instance)?;
+    let inst = inspect_instance(instance, &ns_name)?;
+    let output_dir = get_output_directory(conf.sep_mount);
+    let root = std::env::current_dir()?.join(output_dir);
+    if !inst.mounted {
+        mount_fs(instance)?;
+    }
+    info!("Refreshing local repository...");
+    repo::init_repo(root.as_ref(), Path::new(&instance))?;
+    Ok(())
+}
+
 /// Execute the specified command in the container
 pub fn run_in_container<S: AsRef<OsStr>>(instance: &str, args: &[S]) -> Result<i32> {
     let ns_name = start_container(instance)?;

src/cli.rs

@@ -91,13 +91,29 @@ pub fn build_cli() -> Command {
             Command::new("shell")
                 .alias("sh")
                 .arg(instance_arg.clone().help("Instance to be used"))
+                .arg(
+                    Arg::new("NO_LOCAL_REPO")
+                        .long("no-local-repo")
+                        .action(clap::ArgAction::SetTrue)
+                        .help(
+                            "Avoid preparing the local repository for the interactive environment",
+                        ),
+                )
                 .arg(Arg::new("COMMANDS").required(false).num_args(1..))
                 .about("Start an interactive shell"),
         )
         .subcommand(
             Command::new("run")
                 .alias("exec")
                 .arg(instance_arg.clone().help("Instance to run command in"))
+                .arg(
+                    Arg::new("NO_LOCAL_REPO")
+                        .long("no-local-repo")
+                        .action(clap::ArgAction::SetTrue)
+                        .help(
+                            "Avoid preparing the local repository for the interactive environment",
+                        ),
+                )
                 .arg(Arg::new("COMMANDS").required(true).num_args(1..))
                 .about("Lower-level version of 'shell', without login environment, without sourcing ~/.bash_profile"),
         )

src/main.rs

@@ -278,13 +278,19 @@ fn main() -> Result<()> {
         }
         ("run", args) => {
             let instance = get_instance_option(args)?;
+            if !args.get_flag("NO_LOCAL_REPO") {
+                actions::prepare_local_repo(&instance)?;
+            }
             let args = args.get_many::<String>("COMMANDS").unwrap();
             let status =
                 actions::run_in_container(&instance, &args.into_iter().collect::<Vec<_>>())?;
             process::exit(status);
         }
         ("shell", args) => {
             let instance = get_instance_option(args)?;
+            if !args.get_flag("NO_LOCAL_REPO") {
+                actions::prepare_local_repo(&instance)?;
+            }
             if let Some(cmd) = args.get_many::<String>("COMMANDS") {
                 let command = cmd
                     .into_iter()