roulette/types/types.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
}