59 lines
1.0 KiB
Go
59 lines
1.0 KiB
Go
|
/*
|
||
|
|
Copyright © 2023 Seednode <seednode@seedno.de>
|
||
|
|
*/
|
||
|
|
|
||
|
|
package cmd
|
||
|
|
|
||
|
|
type SupportedType struct {
|
||
|
|
title func(queryParams, filePath, mime, fileName string, width, height int) string
|
||
|
|
body func(queryParams, filePath, mime, fileName string, width, height int) string
|
||
|
|
extensions []string
|
||
|
|
validator func([]byte) bool
|
||
|
|
}
|
||
|
|
|
||
|
|
func (i *SupportedType) Extensions() []string {
|
||
|
|
return i.extensions
|
||
|
|
}
|
||
|
|
|
||
|
|
type SupportedTypes struct {
|
||
|
|
types []*SupportedType
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *SupportedTypes) Add(t *SupportedType) {
|
||
|
|
s.types = append(s.types, t)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *SupportedTypes) Extensions() []string {
|
||
|
|
var r []string
|
||
|
|
|
||
|
|
for _, t := range s.types {
|
||
|
|
r = append(r, t.Extensions()...)
|
||
|
|
}
|
||
|
|
|
||
|
|
return r
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *SupportedTypes) Type(extension string) *SupportedType {
|
||
|
|
for i := range s.types {
|
||
|
|
for _, e := range s.types[i].Extensions() {
|
||
|
|
if extension == e {
|
||
|
|
return s.types[i]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *SupportedTypes) IsSupported(head []byte) bool {
|
||
|
|
r := false
|
||
|
|
|
||
|
|
for i := range s.types {
|
||
|
|
if s.types[i].validator(head) {
|
||
|
|
r = true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return r
|
||
|
|
}
|