package torrent

import (
	"fmt"
	"os"
	"strings"
)

func getFileIDs(files []File) []string {
	var fileIDs []string
	for _, file := range files {
		if file.ID != 0 {
			fileIDs = append(fileIDs, fmt.Sprintf("%d", file.ID))
		}
	}
	return fileIDs
}

func isStreamable(filePath string) bool {
	return strings.HasSuffix(filePath, ".mkv") ||
		strings.HasSuffix(filePath, ".mp4") ||
		strings.HasSuffix(filePath, ".avi") ||
		strings.HasSuffix(filePath, ".wmv") ||
		strings.HasSuffix(filePath, ".m4v")
}

func ensureDir(dirName string) error {
	if _, err := os.Stat(dirName); os.IsNotExist(err) {
		err := os.Mkdir(dirName, 0755)
		if err != nil {
			return err
		}
	} else if err != nil {
		return err
	}
	return nil
}