130 lines
2.3 KiB
Go
130 lines
2.3 KiB
Go
/*
|
|
Copyright © 2023 Seednode <seednode@seedno.de>
|
|
*/
|
|
|
|
package types
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"slices"
|
|
"strings"
|
|
)
|
|
|
|
var SupportedFormats = &Types{
|
|
Extensions: make(map[string]string),
|
|
MimeTypes: make(map[string]Type),
|
|
}
|
|
|
|
type Type interface {
|
|
Css() string
|
|
Title(queryParams, fileUri, filePath, fileName, mime string) string
|
|
Body(queryParams, fileUri, filePath, fileName, mime string) string
|
|
Extensions() map[string]string
|
|
MimeTypes() []string
|
|
Validate(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
|
|
}
|
|
|
|
func Register(t Type) {
|
|
SupportedFormats.Add(t)
|
|
}
|
|
|
|
func (t Types) GetExtensions() string {
|
|
var output strings.Builder
|
|
|
|
extensions := make([]string, len(t.Extensions))
|
|
|
|
i := 0
|
|
|
|
for k := range t.Extensions {
|
|
extensions[i] = k
|
|
i++
|
|
}
|
|
|
|
slices.Sort(extensions)
|
|
|
|
for _, v := range extensions {
|
|
output.WriteString(v + "\n")
|
|
}
|
|
|
|
return output.String()
|
|
}
|
|
|
|
func (t Types) GetMimeTypes() string {
|
|
var output strings.Builder
|
|
|
|
mimeTypes := make([]string, len(t.MimeTypes))
|
|
|
|
i := 0
|
|
|
|
for k := range t.MimeTypes {
|
|
mimeTypes[i] = k
|
|
i++
|
|
}
|
|
|
|
slices.Sort(mimeTypes)
|
|
|
|
for _, v := range mimeTypes {
|
|
output.WriteString(v + "\n")
|
|
}
|
|
|
|
return output.String()
|
|
}
|