Merge pull request #10 from AngelMunoz/use-token-replacer

Add custom operation `useTokenReplacer`

Author: JordanMarr

src/FSharp.SystemCommandLine/CommandBuilders.fs

@@ -333,6 +333,11 @@ type RootCommandParserBuilder<'A, 'B, 'C, 'D, 'E, 'F, 'G, 'H, 'Output>() =
     member this.UsePipeline (spec: CommandSpec<'Inputs, 'Output>, subCommand: CommandLineBuilder -> CommandLineBuilder) =
         this.CommandLineBuilder <- subCommand this.CommandLineBuilder
         spec
+
+    [<CustomOperation("useTokenReplacer")>]
+    member this.UseTokenReplacer(spec: CommandSpec<'Inputs, 'Output>, replacer: TryReplaceToken) =
+        this.CommandLineBuilder.UseTokenReplacer(replacer) |> ignore
+        spec
 
     /// Executes a Command with a handler that returns unit.
     member this.Run (spec: CommandSpec<'Inputs, unit>) =
@@ -372,6 +377,11 @@ type RootCommandBuilder<'A, 'B, 'C, 'D, 'E, 'F, 'G, 'H, 'Output>(args: string ar
     member this.UsePipeline (spec: CommandSpec<'Inputs, 'Output>, subCommand: CommandLineBuilder -> CommandLineBuilder) =
         this.CommandLineBuilder <- subCommand this.CommandLineBuilder
         spec
+        
+    [<CustomOperation("useTokenReplacer")>]
+    member this.UseTokenReplacer(spec: CommandSpec<'Inputs, 'Output>, replacer: TryReplaceToken) =
+        this.CommandLineBuilder.UseTokenReplacer(replacer) |> ignore
+        spec
 
     /// Executes a Command with a handler that returns unit.
     member this.Run (spec: CommandSpec<'Inputs, unit>) =

src/TestConsole/ProgramTokenReplacer.fs

@@ -0,0 +1,29 @@
+module ProgramTokenReplacer
+
+open FSharp.SystemCommandLine
+open System.CommandLine.Parsing
+
+let app (package: string) =
+    if package.StartsWith("@") then
+        printfn $"{package}"
+        0
+    else
+        eprintfn "The package name does not start with a leading @"
+        1
+
+//[<EntryPoint>]
+let main argv =
+    let package =
+        Input.Option([ "--package"; "-p" ], "A package with a leading @ name")
+
+    rootCommand argv {
+        description "Can be called with a leading @ package"
+
+        useTokenReplacer (
+            // in this case we want to skip @ processing
+            TryReplaceToken (fun _ _ _ -> false)
+        )
+
+        inputs package
+        setHandler app
+    }

src/TestConsole/Properties/launchSettings.json

@@ -3,6 +3,10 @@
     "TestConsole": {
       "commandName": "Project",
       "commandLineArgs": "-w hello -w world"
+    },
+    "Leading @": {
+      "commandName": "Project",
+      "commandLineArgs": "--package @shoelace-style/shoelace"
     }
   }
 }

src/TestConsole/TestConsole.fsproj

@@ -11,11 +11,10 @@
     <Compile Include="ProgramNoArgs.fs" />
     <Compile Include="ProgramSubCommand.fs" />
     <Compile Include="ProgramTask.fs" />
+    <Compile Include="ProgramTokenReplacer.fs" />
     <Compile Include="Program.fs" />
   </ItemGroup>
 
-  <ItemGroup />
-
   <ItemGroup>
     <ProjectReference Include="..\FSharp.SystemCommandLine\FSharp.SystemCommandLine.fsproj" />
   </ItemGroup>