85 lines
2.1 KiB
Go
85 lines
2.1 KiB
Go
package fs
|
|
|
|
import (
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// StripExt removes the file type extension from a file name (if any).
|
|
func StripExt(name string) string {
|
|
if end := strings.LastIndex(name, "."); end != -1 {
|
|
name = name[:end]
|
|
}
|
|
|
|
return name
|
|
}
|
|
|
|
// StripKnownExt removes a known file type extension from a file name (if any).
|
|
func StripKnownExt(name string) string {
|
|
if FileExt.Known(name) {
|
|
name = StripExt(name)
|
|
}
|
|
|
|
return name
|
|
}
|
|
|
|
// Ext returns all extension of a file name including the dots.
|
|
func Ext(name string) string {
|
|
ext := filepath.Ext(name)
|
|
name = StripExt(name)
|
|
|
|
if FileExt.Known(name) {
|
|
ext = filepath.Ext(name) + ext
|
|
}
|
|
|
|
return ext
|
|
}
|
|
|
|
// StripSequence removes common sequence patterns at the end of file names.
|
|
func StripSequence(name string) string {
|
|
// Strip numeric extensions like .00000, .00001, .4542353245,.... (at least 5 digits).
|
|
if dot := strings.LastIndex(name, "."); dot != -1 && len(name[dot+1:]) >= 5 {
|
|
if i, err := strconv.Atoi(name[dot+1:]); err == nil && i >= 0 {
|
|
name = name[:dot]
|
|
}
|
|
}
|
|
|
|
// Other common sequential naming schemes.
|
|
if end := strings.Index(name, "("); end != -1 {
|
|
// Copies created by Chrome & Windows, example: IMG_1234 (2).
|
|
name = name[:end]
|
|
} else if end := strings.Index(name, " copy"); end != -1 {
|
|
// Copies created by OS X, example: IMG_1234 copy 2.
|
|
name = name[:end]
|
|
}
|
|
|
|
name = strings.TrimSpace(name)
|
|
|
|
return name
|
|
}
|
|
|
|
// BasePrefix returns the filename base without any extensions and path.
|
|
func BasePrefix(fileName string, stripSequence bool) string {
|
|
name := StripKnownExt(StripExt(filepath.Base(fileName)))
|
|
|
|
if !stripSequence {
|
|
return name
|
|
}
|
|
|
|
return StripSequence(name)
|
|
}
|
|
|
|
// RelPrefix returns the relative filename.
|
|
func RelPrefix(fileName, dir string, stripSequence bool) string {
|
|
if name := RelName(fileName, dir); name != "" {
|
|
return AbsPrefix(name, stripSequence)
|
|
}
|
|
|
|
return BasePrefix(fileName, stripSequence)
|
|
}
|
|
|
|
// AbsPrefix returns the directory and base filename without any extensions.
|
|
func AbsPrefix(fileName string, stripSequence bool) string {
|
|
return filepath.Join(filepath.Dir(fileName), BasePrefix(fileName, stripSequence))
|
|
}
|