--shebang,-!

Closes #1136

Author: mxcl

.github/workflows/ci.yml

@@ -166,6 +166,15 @@ jobs:
             -Xdemangler=rustfilt \
             > lcov.info
 
+      - name: --shebang test 1
+        run: |
+          echo "#!/usr/bin/env -S pkgx --shebang echo" > ./stub.sh
+          chmod +x stub.sh
+          test "$(./stub.sh hi)" = hi
+
+      - name: --shebang test 2
+        run: test $(pkgx -q! echo fail hi) = hi
+
       - uses: coverallsapp/github-action@v2
         with:
           path-to-lcov: lcov.info

crates/cli/Cargo.toml

@@ -14,7 +14,7 @@ regex = "1.11.1"
 indicatif = "0.17.9"
 nix = { version = "0.29.0", features = ["process"] }
 serde_json = "1.0.135"
-libpkgx = { version = "0.3.2", path = "../lib" }
+libpkgx = { version = "0.4.0", path = "../lib" }
 console = { version = "0.15", default-features = false, features = [
   "ansi-parsing",
 ] }

crates/cli/src/args.rs

@@ -11,6 +11,7 @@ pub struct Flags {
     pub silent: bool,
     pub json: bool,
     pub version_n_continue: bool,
+    pub shebang: bool,
 }
 
 pub struct Args {
@@ -31,6 +32,7 @@ pub fn parse() -> Args {
     let mut find_program = false;
     let mut collecting_args = false;
     let mut version_n_continue = false;
+    let mut shebang = false;
 
     for arg in std::env::args().skip(1) {
         if collecting_args {
@@ -51,6 +53,7 @@ pub fn parse() -> Args {
                     }
                     json = true
                 }
+                "--shebang" => shebang = true,
                 "--json=v1" => json = true,
                 "--silent" => silent = true,
                 "--help" => mode = Mode::Help,
@@ -92,6 +95,7 @@ pub fn parse() -> Args {
                     's' => silent = true,
                     'j' => json = true,
                     'v' => version_n_continue = true,
+                    '!' => shebang = true,
                     _ => panic!("unknown argument: -{}", c),
                 }
             }
@@ -108,6 +112,7 @@ pub fn parse() -> Args {
         find_program,
         mode,
         flags: Flags {
+            shebang,
             silent,
             json,
             quiet,

crates/cli/src/help.rs

@@ -18,6 +18,7 @@ examples:
 flags:
   -q,  --quiet    # suppress brief informational messages
   -qq, --silent   # no chat. no errors. just execute.
+  -!,  --shebang  # drops shebang added arg0 for faster pkgx stubs
   -v,  --version
 
 more:

crates/cli/src/main.rs

@@ -227,6 +227,11 @@ async fn main() -> Result<(), Box<dyn Error>> {
 
         clear_progress_bar();
 
+        if flags.shebang {
+            // removes the filename of the shebang script
+            args.remove(0);
+        }
+
         execve(cmd, args, env)
     } else if !plus.is_empty() {
         clear_progress_bar();