89 lines
2.6 KiB
Go
89 lines
2.6 KiB
Go
|
// Licensed to Elasticsearch B.V. under one or more contributor
|
||
|
|
// license agreements. See the NOTICE file distributed with
|
||
|
|
// this work for additional information regarding copyright
|
||
|
|
// ownership. Elasticsearch B.V. licenses this file to you 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 file
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"path/filepath"
|
||
|
|
)
|
||
|
|
|
||
|
|
func wildcards(doubleStarPatternDepth uint8, dir string, suffix string) []string {
|
||
|
|
wildcardList := []string{}
|
||
|
|
w := ""
|
||
|
|
i := uint8(0)
|
||
|
|
if dir == "" && suffix == "" {
|
||
|
|
// Don't expand to "" on relative paths
|
||
|
|
w = "*"
|
||
|
|
i = 1
|
||
|
|
}
|
||
|
|
for ; i <= doubleStarPatternDepth; i++ {
|
||
|
|
wildcardList = append(wildcardList, w)
|
||
|
|
w = filepath.Join(w, "*")
|
||
|
|
}
|
||
|
|
return wildcardList
|
||
|
|
}
|
||
|
|
|
||
|
|
// GlobPatterns detects the use of "**" and expands it to standard glob patterns up to a max depth
|
||
|
|
func GlobPatterns(pattern string, doubleStarPatternDepth uint8) ([]string, error) {
|
||
|
|
if doubleStarPatternDepth == 0 {
|
||
|
|
return []string{pattern}, nil
|
||
|
|
}
|
||
|
|
var wildcardList []string
|
||
|
|
var prefix string
|
||
|
|
var suffix string
|
||
|
|
dir, file := filepath.Split(filepath.Clean(pattern))
|
||
|
|
for file != "" && file != "." {
|
||
|
|
if file == "**" {
|
||
|
|
if len(wildcardList) > 0 {
|
||
|
|
return nil, fmt.Errorf("multiple ** in %q", pattern)
|
||
|
|
}
|
||
|
|
wildcardList = wildcards(doubleStarPatternDepth, dir, suffix)
|
||
|
|
prefix = dir
|
||
|
|
} else if len(wildcardList) == 0 {
|
||
|
|
suffix = filepath.Join(file, suffix)
|
||
|
|
}
|
||
|
|
dir, file = filepath.Split(filepath.Clean(dir))
|
||
|
|
}
|
||
|
|
if len(wildcardList) == 0 {
|
||
|
|
return []string{pattern}, nil
|
||
|
|
}
|
||
|
|
var patterns []string
|
||
|
|
for _, w := range wildcardList {
|
||
|
|
patterns = append(patterns, filepath.Join(prefix, w, suffix))
|
||
|
|
}
|
||
|
|
return patterns, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Glob expands '**' patterns into multiple patterns to satisfy https://golang.org/pkg/path/filepath/#Match
|
||
|
|
func Glob(pattern string, doubleStarPatternDepth uint8) ([]string, error) {
|
||
|
|
patterns, err := GlobPatterns(pattern, doubleStarPatternDepth)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
var matches []string
|
||
|
|
for _, p := range patterns {
|
||
|
|
// Evaluate the path as a wildcards/shell glob
|
||
|
|
match, err := filepath.Glob(p)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
matches = append(matches, match...)
|
||
|
|
}
|
||
|
|
return matches, nil
|
||
|
|
}
|