treat \ as escape only before a separator

This commit is contained in:
David Shiflet
2022-05-03 21:19:51 -04:00
committed by Alec Thomas
parent 5538b7f045
commit 29685e7da6
5 changed files with 86 additions and 20 deletions
+5 -2
View File
@@ -721,12 +721,15 @@ func SplitEscaped(s string, sep rune) (out []string) {
} }
escaped := false escaped := false
token := "" token := ""
for _, ch := range s { for i, ch := range s {
switch { switch {
case escaped: case escaped:
if ch != sep {
token += `\`
}
token += string(ch) token += string(ch)
escaped = false escaped = false
case ch == '\\': case ch == '\\' && i < len(s)-1:
escaped = true escaped = true
case ch == sep && !escaped: case ch == sep && !escaped:
out = append(out, token) out = append(out, token)
+26
View File
@@ -0,0 +1,26 @@
//go:build !windows
// +build !windows
package kong_test
import (
"os"
"testing"
"github.com/stretchr/testify/require"
)
func TestPathMapper(t *testing.T) {
var cli struct {
Path string `arg:"" type:"path"`
}
p := mustNew(t, &cli)
_, err := p.Parse([]string{"/an/absolute/path"})
require.NoError(t, err)
require.Equal(t, "/an/absolute/path", cli.Path)
_, err = p.Parse([]string{"-"})
require.NoError(t, err)
require.Equal(t, "-", cli.Path)
}
+6 -16
View File
@@ -9,6 +9,7 @@ import (
"net/url" "net/url"
"os" "os"
"reflect" "reflect"
"runtime"
"strings" "strings"
"testing" "testing"
"time" "time"
@@ -412,27 +413,16 @@ func TestFileMapper(t *testing.T) {
_ = cli.File.Close() _ = cli.File.Close()
_, err = p.Parse([]string{"testdata/missing.txt"}) _, err = p.Parse([]string{"testdata/missing.txt"})
require.Error(t, err) require.Error(t, err)
require.Contains(t, err.Error(), "missing.txt: no such file or directory") if runtime.GOOS == "windows" {
require.Contains(t, err.Error(), "missing.txt: The system cannot find the file specified.")
} else {
require.Contains(t, err.Error(), "missing.txt: no such file or directory")
}
_, err = p.Parse([]string{"-"}) _, err = p.Parse([]string{"-"})
require.NoError(t, err) require.NoError(t, err)
require.Equal(t, os.Stdin, cli.File) require.Equal(t, os.Stdin, cli.File)
} }
func TestPathMapper(t *testing.T) {
var cli struct {
Path string `arg:"" type:"path"`
}
p := mustNew(t, &cli)
_, err := p.Parse([]string{"/an/absolute/path"})
require.NoError(t, err)
require.Equal(t, "/an/absolute/path", cli.Path)
_, err = p.Parse([]string{"-"})
require.NoError(t, err)
require.Equal(t, "-", cli.Path)
}
func TestMapperPlaceHolder(t *testing.T) { func TestMapperPlaceHolder(t *testing.T) {
var cli struct { var cli struct {
Flag string Flag string
+44
View File
@@ -0,0 +1,44 @@
//go:build windows
// +build windows
package kong_test
import (
"os"
"testing"
"github.com/stretchr/testify/require"
)
func TestWindowsPathMapper(t *testing.T) {
var cli struct {
Path string `short:"p" type:"path"`
Files []string `short:"f" type:"path"`
}
wd, err := os.Getwd()
require.NoError(t, err, "Getwd failed")
p := mustNew(t, &cli)
_, err = p.Parse([]string{`-p`, `c:\an\absolute\path`, `-f`, `c:\second\absolute\path\`, `-f`, `relative\path\file`})
require.NoError(t, err)
require.Equal(t, `c:\an\absolute\path`, cli.Path)
require.Equal(t, []string{`c:\second\absolute\path\`, wd + `\relative\path\file`}, cli.Files)
}
func TestWindowsFileMapper(t *testing.T) {
type CLI struct {
File *os.File `arg:""`
}
var cli CLI
p := mustNew(t, &cli)
_, err := p.Parse([]string{"testdata\\file.txt"})
require.NoError(t, err)
require.NotNil(t, cli.File)
_ = cli.File.Close()
_, err = p.Parse([]string{"testdata\\missing.txt"})
require.Error(t, err)
require.Contains(t, err.Error(), "missing.txt: The system cannot find the file specified.")
_, err = p.Parse([]string{"-"})
require.NoError(t, err)
require.Equal(t, os.Stdin, cli.File)
}
+5 -2
View File
@@ -4,6 +4,7 @@ import (
"io/ioutil" "io/ioutil"
"os" "os"
"path/filepath" "path/filepath"
"runtime"
"strings" "strings"
"testing" "testing"
@@ -62,7 +63,9 @@ func TestChangeDirFlag(t *testing.T) {
p := Must(&cli) p := Must(&cli)
_, err = p.Parse([]string{"-C", dir, "out.txt"}) _, err = p.Parse([]string{"-C", dir, "out.txt"})
require.NoError(t, err) require.NoError(t, err)
file, err = filepath.EvalSymlinks(file) // Needed because OSX uses a symlinked tmp dir. if runtime.GOOS != "windows" {
require.NoError(t, err) file, err = filepath.EvalSymlinks(file) // Needed because OSX uses a symlinked tmp dir.
require.NoError(t, err)
}
require.Equal(t, file, cli.Path) require.Equal(t, file, cli.Path)
} }