go/task
go/task
1
0
Fork 0
mirror of https://github.com/go-task/task.git synced 2024-12-12 10:45:49 +02:00
Code Issues Releases Activity

Merge pull request #613 from Peter554/fix-quoting-cli-args

Browse Source 
Fix quoting of CLI_ARGS
This commit is contained in:
Andrey Nering 2021-11-28 15:51:03 -03:00 committed by GitHub
commit 846c27d579
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
cmd/task/task.go
View File

@ -12,6 +12,7 @@ import (
"syscall" "syscall"
"github.com/spf13/pflag" "github.com/spf13/pflag"
"mvdan.cc/sh/v3/syntax"
"github.com/go-task/task/v3" "github.com/go-task/task/v3"
"github.com/go-task/task/v3/args" "github.com/go-task/task/v3/args"
@ -159,18 +160,22 @@ func main() {
} }
var ( var (
calls []taskfile.Call calls []taskfile.Call
globals *taskfile.Vars globals *taskfile.Vars
tasksAndVars, cliArgs = getArgs()
) )
tasksAndVars, cliArgs, err := getArgs()
if err != nil {
log.Fatal(err)
}
if v >= 3.0 { if v >= 3.0 {
calls, globals = args.ParseV3(tasksAndVars...) calls, globals = args.ParseV3(tasksAndVars...)
} else { } else {
calls, globals = args.ParseV2(tasksAndVars...) calls, globals = args.ParseV2(tasksAndVars...)
} }
globals.Set("CLI_ARGS", taskfile.Var{Static: strings.Join(cliArgs, " ")}) globals.Set("CLI_ARGS", taskfile.Var{Static: cliArgs})
e.Taskfile.Vars.Merge(globals) e.Taskfile.Vars.Merge(globals)
ctx := context.Background() ctx := context.Background()
@ -191,20 +196,25 @@ func main() {
} }
} }
func getArgs() (tasksAndVars, cliArgs []string) { func getArgs() ([]string, string, error) {
var ( var (
args = pflag.Args() args = pflag.Args()
doubleDashPos = pflag.CommandLine.ArgsLenAtDash() doubleDashPos = pflag.CommandLine.ArgsLenAtDash()
) )
if doubleDashPos != -1 { if doubleDashPos == -1 {
tasksAndVars = args[:doubleDashPos] return args, "", nil
cliArgs = args[doubleDashPos:]
} else {
tasksAndVars = args
} }
return quotedCliArgs := []string{}
for _, arg := range args[doubleDashPos:] {
quotedCliArg, err := syntax.Quote(arg, syntax.LangBash)
if err != nil {
return []string{}, "", err
}
quotedCliArgs = append(quotedCliArgs, quotedCliArg)
}
return args[:doubleDashPos], strings.Join(quotedCliArgs, " "), nil
} }
func getSignalContext() context.Context { func getSignalContext() context.Context {