photoprism/pkg/fs/mediatype.go
2020-12-05 02:18:45 +01:00

57 lines
1.1 KiB
Go

package fs
type MediaType string
const (
MediaImage MediaType = "image"
MediaSidecar MediaType = "sidecar"
MediaRaw MediaType = "raw"
MediaVideo MediaType = "video"
MediaOther MediaType = "other"
)
var MediaTypes = map[FileType]MediaType{
TypeRaw: MediaRaw,
TypeJpeg: MediaImage,
TypePng: MediaImage,
TypeGif: MediaImage,
TypeTiff: MediaImage,
TypeBitmap: MediaImage,
TypeHEIF: MediaImage,
TypeAvi: MediaVideo,
TypeMP4: MediaVideo,
TypeMov: MediaVideo,
TypeXMP: MediaSidecar,
TypeXML: MediaSidecar,
TypeAAE: MediaSidecar,
TypeYaml: MediaSidecar,
TypeText: MediaSidecar,
TypeJson: MediaSidecar,
TypeToml: MediaSidecar,
TypeMarkdown: MediaSidecar,
TypeOther: MediaOther,
}
func GetMediaType(fileName string) MediaType {
if fileName == "" {
return MediaOther
}
result, ok := MediaTypes[GetFileType(fileName)]
if !ok {
result = MediaOther
}
return result
}
func IsMedia(fileName string) bool {
switch GetMediaType(fileName) {
case MediaRaw, MediaImage, MediaVideo:
return true
default:
return false
}
}