29 lines
611 B
Go
29 lines
611 B
Go
package classifier
|
|
|
|
|
|
import (
|
|
"path/filepath"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
var movieYear = regexp.MustCompile(`(?i)(19|20)\d{2}`)
|
|
|
|
func classifyMovie(filename string) (bool, string, int) {
|
|
base := strings.TrimSuffix(filepath.Base(filename), filepath.Ext(filename))
|
|
|
|
// Find year
|
|
yearMatch := movieYear.FindString(base)
|
|
if yearMatch == "" {
|
|
return false, "", 0
|
|
}
|
|
|
|
year := parseInt(yearMatch)
|
|
|
|
// Extract title (everything before year)
|
|
title := strings.TrimSpace(strings.Replace(base, yearMatch, "", 1))
|
|
title = cleanupTitle(title)
|
|
|
|
return true, title, year
|
|
}
|