77 lines
1.6 KiB
Go
77 lines
1.6 KiB
Go
/*
|
|
Copyright © 2023 Seednode <seednode@seedno.de>
|
|
*/
|
|
|
|
package types
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
type Type struct {
|
|
Css func() string
|
|
Title func(queryParams, fileUri, filePath, fileName, mime string) string
|
|
Body func(queryParams, fileUri, filePath, fileName, mime string) string
|
|
Extensions map[string]string
|
|
MimeTypes []string
|
|
Validate func(filePath string) bool
|
|
}
|
|
|
|
type Types struct {
|
|
Extensions map[string]string
|
|
MimeTypes map[string]*Type
|
|
}
|
|
|
|
func (s *Types) Add(t *Type) {
|
|
for k, v := range t.Extensions {
|
|
_, exists := s.Extensions[k]
|
|
if !exists {
|
|
s.Extensions[k] = v
|
|
}
|
|
}
|
|
|
|
for _, v := range t.MimeTypes {
|
|
_, exists := s.Extensions[v]
|
|
if !exists {
|
|
s.MimeTypes[v] = t
|
|
}
|
|
}
|
|
}
|
|
|
|
func FileType(path string, registeredFormats *Types) (bool, *Type, string, error) {
|
|
file, err := os.Open(path)
|
|
switch {
|
|
case errors.Is(err, os.ErrNotExist):
|
|
return false, nil, "", nil
|
|
case err != nil:
|
|
return false, nil, "", err
|
|
}
|
|
defer file.Close()
|
|
|
|
head := make([]byte, 512)
|
|
file.Read(head)
|
|
|
|
mimeType := http.DetectContentType(head)
|
|
|
|
// try identifying files by mime types first
|
|
fileType, exists := registeredFormats.MimeTypes[mimeType]
|
|
if exists {
|
|
return fileType.Validate(path), fileType, mimeType, nil
|
|
}
|
|
|
|
// if mime type detection fails, use the file extension
|
|
mimeType, exists = registeredFormats.Extensions[filepath.Ext(path)]
|
|
if exists {
|
|
fileType, exists := registeredFormats.MimeTypes[mimeType]
|
|
|
|
if exists {
|
|
return fileType.Validate(path), fileType, mimeType, nil
|
|
}
|
|
}
|
|
|
|
return false, nil, "", nil
|
|
}
|