step/providers/mime/mime.go

40 lines
898 B
Go

package slice
import "html/template"
import "github.com/gabriel-vasile/mimetype"
import "git.tebibyte.media/sashakoshka/step"
var _ step.FuncProviderFor = new(Provider)
// Provider provides MIME functions.
type Provider struct {
}
// Package fulfills the step.Provider interface.
func (this *Provider) Package () string {
return "mime"
}
// FuncMapFor fulfills the step.FuncProviderFor interface.
func (this *Provider) FuncMapFor (document *step.Document) template.FuncMap {
stat := &state {
document: document,
}
return template.FuncMap {
"fileContentType": stat.funcFileContentType,
}
}
type state struct {
document *step.Document
}
func (this *state) funcFileContentType (name string) (string, error) {
name, err := this.document.Rel(name)
if err != nil { return "", err }
typ, err := mimetype.DetectFile(name)
if err != nil { return "", err }
return typ.String(), err
}