formidable/internal/data/scheme/http/loader_handler.go

40 lines
743 B
Go
Raw Normal View History

package http
import (
"io"
"net/http"
"net/url"
"github.com/pkg/errors"
)
const (
SchemeHTTP = "http"
SchemeHTTPS = "https"
)
type LoaderHandler struct {
client *http.Client
}
func (h *LoaderHandler) Match(url *url.URL) bool {
return url.Scheme == SchemeHTTP || url.Scheme == SchemeHTTPS
}
func (h *LoaderHandler) Open(url *url.URL) (io.ReadCloser, error) {
res, err := h.client.Get(url.String())
if err != nil {
return nil, errors.WithStack(err)
}
if res.StatusCode != http.StatusOK {
return nil, errors.Errorf("unexpected status code '%d (%s)'", res.StatusCode, http.StatusText(res.StatusCode))
}
return res.Body, nil
}
func NewLoaderHandler(client *http.Client) *LoaderHandler {
return &LoaderHandler{client}
}