1// Copyright 2018 The CUE Authors 2// 3// Licensed under the Apache License, Version 2.0 (the "License"); 4// you may not use this file except in compliance with the License. 5// You may obtain a copy of the License at 6// 7// http://www.apache.org/licenses/LICENSE-2.0 8// 9// Unless required by applicable law or agreed to in writing, software 10// distributed under the License is distributed on an "AS IS" BASIS, 11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12// See the License for the specific language governing permissions and 13// limitations under the License. 14 15package load 16 17import ( 18 "io/ioutil" 19 "path/filepath" 20 "strings" 21 22 "cuelang.org/go/cue/build" 23 "cuelang.org/go/cue/errors" 24 "cuelang.org/go/cue/token" 25) 26 27var errExclude = errors.New("file rejected") 28 29type cueError = errors.Error 30type excludeError struct { 31 cueError 32} 33 34func (e excludeError) Is(err error) bool { return err == errExclude } 35 36// matchFile determines whether the file with the given name in the given directory 37// should be included in the package being constructed. 38// It returns the data read from the file. 39// If returnImports is true and name denotes a CUE file, matchFile reads 40// until the end of the imports (and returns that data) even though it only 41// considers text until the first non-comment. 42// If allTags is non-nil, matchFile records any encountered build tag 43// by setting allTags[tag] = true. 44func matchFile(cfg *Config, file *build.File, returnImports, allFiles bool, allTags map[string]bool) (match bool, data []byte, err errors.Error) { 45 if fi := cfg.fileSystem.getOverlay(file.Filename); fi != nil { 46 if fi.file != nil { 47 file.Source = fi.file 48 } else { 49 file.Source = fi.contents 50 } 51 } 52 53 if file.Encoding != build.CUE { 54 return false, nil, nil // not a CUE file, don't record. 55 } 56 57 if file.Filename == "-" { 58 b, err2 := ioutil.ReadAll(cfg.stdin()) 59 if err2 != nil { 60 err = errors.Newf(token.NoPos, "read stdin: %v", err) 61 return 62 } 63 file.Source = b 64 return true, b, nil // don't check shouldBuild for stdin 65 } 66 67 name := filepath.Base(file.Filename) 68 if !cfg.filesMode && strings.HasPrefix(name, ".") { 69 return false, nil, &excludeError{ 70 errors.Newf(token.NoPos, "filename starts with a '.'"), 71 } 72 } 73 74 if strings.HasPrefix(name, "_") { 75 return false, nil, &excludeError{ 76 errors.Newf(token.NoPos, "filename starts with a '_"), 77 } 78 } 79 80 f, err := cfg.fileSystem.openFile(file.Filename) 81 if err != nil { 82 return false, nil, err 83 } 84 85 data, err = readImports(f, false, nil) 86 f.Close() 87 if err != nil { 88 return false, nil, 89 errors.Newf(token.NoPos, "read %s: %v", file.Filename, err) 90 } 91 92 return true, data, nil 93} 94