package tv

import (
	"errors"
	"strings"

	"github.com/ollien/gobbler/categorize/match"
)

var ErrNotTvShow = errors.New("no tv-like information found in show name")

// FindShowFolder finds the show that the given filename likey belongs to, among the given show folders.
// Returns ErrNotTvShow if no match can be found
func FindShowFolder(filename string, showFolders []string) (string, error) {
	possibleShowName, err := extractShowName(filename)
	if err != nil {
		return "", ErrNotTvShow
	}

	return matchToFolder(possibleShowName, showFolders)
}

func extractShowName(filename string) (string, error) {
	seasonString, err := extractSeasonNumberString(filename)
	if err != nil {
		return "", errNoSeasonInfo
	}

	seasonStartPos := strings.Index(filename, seasonString)
	if seasonStartPos == -1 {
		// This should never really happen; if the call to extractSeasonNumberString returns a match, we should always
		// be able to find it in the string. However, I don't know that it deserves a panic either...
		return "", errors.New("failed to find season information in filename")
	}

	possibleShowName := filename[:seasonStartPos]

	return strings.TrimFunc(possibleShowName, isTrimmableSymbol), nil
}

func matchToFolder(candidate string, showFolders []string) (string, error) {
	return match.FindBestMatch(candidate, showFolders)
}