file.go

package model

import (
	"gorm.io/gorm"
	"net/url"
	"strings"
)

type FileType uint

const (
	FILETYPE_INVALID = iota
	// Deprecated: vods can now be downloaded from the edge server using the signed playlist url + ?download=1.
	FILETYPE_VOD
	FILETYPE_ATTACHMENT
	FILETYPE_IMAGE_JPG
	FILETYPE_THUMB_COMB
	FILETYPE_THUMB_CAM
	FILETYPE_THUMB_PRES
	FILETYPE_THUMB_LG_COMB
	FILETYPE_THUMB_LG_CAM
	FILETYPE_THUMB_LG_PRES
	FILETYPE_THUMB_LG_CAM_PRES // generated from CAM and PRES, preferred over the others
)

type File struct {
	gorm.Model

	StreamID uint   `gorm:"not null"`
	Path     string `gorm:"not null"`
	Filename string
	Type     FileType `gorm:"not null; default: 1"`
}

func (f File) GetDownloadFileName() string {
	pts := strings.Split(f.Path, "/")
	if len(pts) == 0 {
		return ""
	}
	return pts[len(pts)-1]
}

func (f File) GetFriendlyFileName() string {
	fn := f.GetDownloadFileName()
	if strings.Contains(strings.ToLower(fn), "cam") {
		return "Camera-view"
	}
	if strings.Contains(strings.ToLower(fn), "pres") {
		return "Presentation"
	}

	if f.Filename != "" {
		return f.Filename
	}
	return "Default view"
}

// GetVodTypeByName infers the type of a video file based on its name.
func (f File) GetVodTypeByName() string {
	if strings.HasSuffix(f.Path, "CAM.mp4") {
		return "CAM"
	}
	if strings.HasSuffix(f.Path, "PRES.mp4") {
		return "PRES"
	}
	return "COMB"
}

func (f File) IsThumb() bool {
	return f.Type == FILETYPE_THUMB_CAM || f.Type == FILETYPE_THUMB_PRES || f.Type == FILETYPE_THUMB_COMB
}

func (f File) IsURL() bool {
	parsedUrl, err := url.Parse(f.Path)
	if err != nil {
		return false
	}
	return parsedUrl.Scheme == "https://" || parsedUrl.Scheme == "http://"
}