ref: 910bfe3e9d2177b8efde268281233d5d1ed333cd
dir: /opossum.go/
package opossum
import (
"bytes"
"golang.org/x/text/encoding/ianaindex"
"io/ioutil"
"mime"
"github.com/psilva261/opossum/logger"
"net/url"
"strings"
)
var log *logger.Logger
func SetLogger(l *logger.Logger) {
log = l
}
type Fetcher interface {
Origin() *url.URL
// LinkedUrl relative to current page
LinkedUrl(string) (*url.URL, error)
Get(*url.URL) ([]byte, ContentType, error)
}
type ContentType struct {
MediaType string
Params map[string]string
}
// NewContentType based on mime type string and url including file extension as fallback
func NewContentType(s string, u *url.URL) (c ContentType, err error) {
if s == "" && u != nil && strings.Contains(u.String(), ".") {
l := strings.Split(u.String(), ".")
ext := l[len(l)-1]
switch ext {
case "jpg":
return NewContentType("image/jpeg", u)
case "png":
return NewContentType("image/png", u)
case "gif":
return NewContentType("image/gif", u)
default:
return ContentType{}, nil
}
}
c.MediaType, c.Params, err = mime.ParseMediaType(s)
return
}
func (c ContentType) IsEmpty() bool {
return c.MediaType == ""
}
func (c ContentType) IsHTML() bool {
return c.MediaType == "text/html"
}
func (c ContentType) IsCSS() bool {
return c.MediaType != "text/html"
}
func (c ContentType) IsJS() bool {
for _, t := range []string{"application/javascript", "application/ecmascript", "text/javascript", "text/ecmascript"} {
if t == c.MediaType {
return true
}
}
return false
}
func (c ContentType) IsPlain() bool {
return c.MediaType == "text/plain"
}
func (c ContentType) IsDownload() bool {
return c.MediaType == "application/octet-stream" ||
c.MediaType == "application/zip"
}
func (c ContentType) IsSvg() bool {
return c.MediaType == "image/svg+xml"
}
func (c ContentType) Utf8(buf []byte) []byte {
charset, ok := c.Params["charset"]
if !ok || charset == "utf8" || charset == "utf-8" {
return buf
}
e, err := ianaindex.IANA.Encoding(charset)
if err != nil {
log.Errorf("get encoding %v: %v", charset, err)
return buf
}
r := bytes.NewReader(buf)
cr := e.NewDecoder().Reader(r)
updated, err := ioutil.ReadAll(cr)
if err == nil {
buf = updated
} else {
log.Errorf("utf8: unable to decode to %v: %v", charset, err)
}
return buf
}