MeBox/internal/service/strm_tree_paths.go

package service

import (
	"errors"
	"fmt"
	"path"
	"path/filepath"
	"strings"
)

func strmTreeRelativeSource(source, sourceRoot string) string {
	source = normalizeSTRMTreeSource(source)
	root := normalizeRemotePath(sourceRoot)
	if root != "/" && strings.HasPrefix(source, root+"/") {
		return strings.TrimPrefix(strings.TrimPrefix(source, root), "/")
	}
	return strings.TrimPrefix(source, "/")
}

func strmTreeCloudRef(source, sourceRoot string) string {
	source = normalizeSTRMTreeSource(source)
	if strings.HasPrefix(source, "/") {
		return source
	}
	if strings.TrimSpace(sourceRoot) != "" {
		return joinRemotePath(sourceRoot, source)
	}
	return normalizeRemotePath(source)
}

func strmTreeOutputRelativePath(source string) (string, error) {
	return strmTreeOutputRelativePathWithLinkExtension(source, videoExtensions, ".strm", false)
}

func strmTreeOutputSubtitleLinkRelativePath(source string) (string, error) {
	return strmTreeOutputRelativePathWithLinkExtension(source, strmTreeSubtitleExtensions, ".strm", true)
}

func strmTreeOutputRelativePathWithLinkExtension(source string, allowedExtensions map[string]struct{}, linkExtension string, appendLinkExtension bool) (string, error) {
	parts := strings.Split(strings.Trim(strings.ReplaceAll(source, "\\", "/"), "/"), "/")
	if len(parts) == 0 {
		return "", errors.New("empty source path")
	}
	out := make([]string, 0, len(parts))
	for i, part := range parts {
		part = strings.TrimSpace(part)
		if part == "" || part == "." || part == ".." {
			return "", fmt.Errorf("unsafe path segment %q", part)
		}
		if i == len(parts)-1 {
			ext := strings.ToLower(path.Ext(part))
			if _, ok := allowedExtensions[ext]; !ok {
				return "", fmt.Errorf("unsupported media extension %q", ext)
			}
			if linkExtension != "" {
				if appendLinkExtension {
					part += linkExtension
				} else {
					part = strings.TrimSuffix(part, path.Ext(part)) + linkExtension
				}
			}
		}
		safe := sanitizeFilename(part)
		if safe == "" {
			return "", errors.New("empty sanitized path segment")
		}
		out = append(out, safe)
	}
	return filepath.Join(out...), nil
}

func strmTreeSubtitleMatchesVideo(subtitle strmTreeSource, videos []strmTreeSource) bool {
	subDir, subBase := strmTreeDirAndBase(subtitle.Path)
	if subBase == "" {
		return false
	}
	for _, video := range videos {
		if video.Kind != "" && video.Kind != strmTreeSourceKindVideo {
			continue
		}
		if !strings.EqualFold(subtitle.Provider, video.Provider) {
			continue
		}
		videoDir, videoBase := strmTreeDirAndBase(video.Path)
		if !strings.EqualFold(subDir, videoDir) || videoBase == "" {
			continue
		}
		if strings.EqualFold(subBase, videoBase) || strings.HasPrefix(strings.ToLower(subBase), strings.ToLower(videoBase)+".") {
			return true
		}
	}
	return false
}

func strmTreeDirAndBase(source string) (string, string) {
	source = normalizeSTRMTreeSource(source)
	dir := path.Dir(source)
	name := path.Base(source)
	base := strings.TrimSuffix(name, path.Ext(name))
	return strings.ToLower(strings.Trim(dir, "/")), strings.ToLower(base)
}

func strmTreeOutputPrefixPath(prefix string) (string, error) {
	parts := strings.Split(strings.Trim(strings.ReplaceAll(prefix, "\\", "/"), "/"), "/")
	out := make([]string, 0, len(parts))
	for _, part := range parts {
		part = strings.TrimSpace(part)
		if part == "" {
			continue
		}
		if part == "." || part == ".." {
			return "", fmt.Errorf("unsafe output prefix segment %q", part)
		}
		safe := sanitizeFilename(part)
		if safe == "" {
			return "", errors.New("empty sanitized output prefix segment")
		}
		out = append(out, safe)
	}
	return filepath.Join(out...), nil
}