88 lines
1.9 KiB
Go
88 lines
1.9 KiB
Go
package filetype
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"gopkg.in/h2non/filetype.v1/matchers"
|
|
"gopkg.in/h2non/filetype.v1/types"
|
|
)
|
|
|
|
// Types stores a map of supported types
|
|
var Types = types.Types
|
|
|
|
// NewType creates and registers a new type
|
|
var NewType = types.NewType
|
|
|
|
// Unknown represents an unknown file type
|
|
var Unknown = types.Unknown
|
|
|
|
// ErrEmptyBuffer represents an empty buffer error
|
|
var ErrEmptyBuffer = errors.New("Empty buffer")
|
|
|
|
// ErrUnknownBuffer represents a unknown buffer error
|
|
var ErrUnknownBuffer = errors.New("Unknown buffer type")
|
|
|
|
// AddType registers a new file type
|
|
func AddType(ext, mime string) types.Type {
|
|
return types.NewType(ext, mime)
|
|
}
|
|
|
|
// Is checks if a given buffer matches with the given file type extension
|
|
func Is(buf []byte, ext string) bool {
|
|
kind, ok := types.Types[ext]
|
|
if ok {
|
|
return IsType(buf, kind)
|
|
}
|
|
return false
|
|
}
|
|
|
|
// IsExtension semantic alias to Is()
|
|
func IsExtension(buf []byte, ext string) bool {
|
|
return Is(buf, ext)
|
|
}
|
|
|
|
// IsType checks if a given buffer matches with the given file type
|
|
func IsType(buf []byte, kind types.Type) bool {
|
|
matcher := matchers.Matchers[kind]
|
|
if matcher == nil {
|
|
return false
|
|
}
|
|
return matcher(buf) != types.Unknown
|
|
}
|
|
|
|
// IsMIME checks if a given buffer matches with the given MIME type
|
|
func IsMIME(buf []byte, mime string) bool {
|
|
for _, kind := range types.Types {
|
|
if kind.MIME.Value == mime {
|
|
matcher := matchers.Matchers[kind]
|
|
return matcher(buf) != types.Unknown
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// IsSupported checks if a given file extension is supported
|
|
func IsSupported(ext string) bool {
|
|
for name := range Types {
|
|
if name == ext {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// IsMIMESupported checks if a given MIME type is supported
|
|
func IsMIMESupported(mime string) bool {
|
|
for _, m := range Types {
|
|
if m.MIME.Value == mime {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// GetType retrieves a Type by file extension
|
|
func GetType(ext string) types.Type {
|
|
return types.Get(ext)
|
|
}
|