113 lines
2.5 KiB
Go
113 lines
2.5 KiB
Go
// Copyright 2021 Google Inc. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package response
|
|
|
|
import (
|
|
"io"
|
|
"io/ioutil"
|
|
"strings"
|
|
"unicode"
|
|
)
|
|
|
|
const noQuote = '\x00'
|
|
|
|
// ReadRspFile reads a file in Ninja's response file format and returns its contents.
|
|
func ReadRspFile(r io.Reader) ([]string, error) {
|
|
var files []string
|
|
var file []byte
|
|
|
|
buf, err := ioutil.ReadAll(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
isEscaping := false
|
|
quotingStart := byte(noQuote)
|
|
for _, c := range buf {
|
|
switch {
|
|
case isEscaping:
|
|
if quotingStart == '"' {
|
|
if !(c == '"' || c == '\\') {
|
|
// '\"' or '\\' will be escaped under double quoting.
|
|
file = append(file, '\\')
|
|
}
|
|
}
|
|
file = append(file, c)
|
|
isEscaping = false
|
|
case c == '\\' && quotingStart != '\'':
|
|
isEscaping = true
|
|
case quotingStart == noQuote && (c == '\'' || c == '"'):
|
|
quotingStart = c
|
|
case quotingStart != noQuote && c == quotingStart:
|
|
quotingStart = noQuote
|
|
case quotingStart == noQuote && unicode.IsSpace(rune(c)):
|
|
// Current character is a space outside quotes
|
|
if len(file) != 0 {
|
|
files = append(files, string(file))
|
|
}
|
|
file = file[:0]
|
|
default:
|
|
file = append(file, c)
|
|
}
|
|
}
|
|
|
|
if len(file) != 0 {
|
|
files = append(files, string(file))
|
|
}
|
|
|
|
return files, nil
|
|
}
|
|
|
|
func rspUnsafeChar(r rune) bool {
|
|
switch {
|
|
case 'A' <= r && r <= 'Z',
|
|
'a' <= r && r <= 'z',
|
|
'0' <= r && r <= '9',
|
|
r == '_',
|
|
r == '+',
|
|
r == '-',
|
|
r == '.',
|
|
r == '/':
|
|
return false
|
|
default:
|
|
return true
|
|
}
|
|
}
|
|
|
|
var rspEscaper = strings.NewReplacer(`'`, `'\''`)
|
|
|
|
// WriteRspFile writes a list of files to a file in Ninja's response file format.
|
|
func WriteRspFile(w io.Writer, files []string) error {
|
|
for i, f := range files {
|
|
if i != 0 {
|
|
_, err := io.WriteString(w, " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if strings.IndexFunc(f, rspUnsafeChar) != -1 {
|
|
f = `'` + rspEscaper.Replace(f) + `'`
|
|
}
|
|
|
|
_, err := io.WriteString(w, f)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|