blob: e3f54a6c299ee02d058b28bc820107f212b30dbe [file] [log] [blame]
// Copyright 2017 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package load
import (
"path/filepath"
"strings"
"cmd/go/internal/search"
)
// MatchPackage(pattern, cwd)(p) reports whether package p matches pattern in the working directory cwd.
func MatchPackage(pattern, cwd string) func(*Package) bool {
switch {
case strings.HasPrefix(pattern, "./") || strings.HasPrefix(pattern, "../") || pattern == "." || pattern == "..":
// Split pattern into leading pattern-free directory path
// (including all . and .. elements) and the final pattern.
var dir string
i := strings.Index(pattern, "...")
if i < 0 {
dir, pattern = pattern, ""
} else {
j := strings.LastIndex(pattern[:i], "/")
dir, pattern = pattern[:j], pattern[j+1:]
}
dir = filepath.Join(cwd, dir)
if pattern == "" {
return func(p *Package) bool { return p.Dir == dir }
}
matchPath := search.MatchPattern(pattern)
return func(p *Package) bool {
// Compute relative path to dir and see if it matches the pattern.
rel, err := filepath.Rel(dir, p.Dir)
if err != nil {
// Cannot make relative - e.g. different drive letters on Windows.
return false
}
rel = filepath.ToSlash(rel)
if rel == ".." || strings.HasPrefix(rel, "../") {
return false
}
return matchPath(rel)
}
case pattern == "all":
return func(p *Package) bool { return true }
case pattern == "std":
return func(p *Package) bool { return p.Standard }
case pattern == "cmd":
return func(p *Package) bool { return p.Standard && strings.HasPrefix(p.ImportPath, "cmd/") }
default:
matchPath := search.MatchPattern(pattern)
return func(p *Package) bool { return matchPath(p.ImportPath) }
}
}