(arg string)
| 25 | } |
| 26 | |
| 27 | func quoteArg(arg string) string { |
| 28 | argBytes := []byte(arg) |
| 29 | |
| 30 | needQuoting := false |
| 31 | hasSingleQuote := false |
| 32 | for _, c := range argBytes { |
| 33 | switch { |
| 34 | case c == '\'': |
| 35 | needQuoting = true |
| 36 | hasSingleQuote = true |
| 37 | case checkNeedQuoting(c): |
| 38 | needQuoting = true |
| 39 | } |
| 40 | } |
| 41 | |
| 42 | if !needQuoting { |
| 43 | return arg |
| 44 | } |
| 45 | |
| 46 | sb := strings.Builder{} |
| 47 | if hasSingleQuote { |
| 48 | sb.WriteByte('"') |
| 49 | for _, c := range argBytes { |
| 50 | if checkNeedQuoting(c) { |
| 51 | sb.WriteByte('\\') |
| 52 | } |
| 53 | sb.WriteByte(c) |
| 54 | } |
| 55 | sb.WriteByte('"') |
| 56 | } else { |
| 57 | sb.WriteByte('\'') |
| 58 | sb.WriteString(arg) |
| 59 | sb.WriteByte('\'') |
| 60 | } |
| 61 | |
| 62 | return sb.String() |
| 63 | } |
| 64 | |
| 65 | func checkNeedQuoting(c byte) bool { |
| 66 | switch c { |
no test coverage detected