You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.0 KiB
54 lines
1.0 KiB
package openuri
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
// Client type
|
|
type Client struct {
|
|
httpClient *http.Client
|
|
}
|
|
|
|
// ClientOption type
|
|
type ClientOption func(*Client) error
|
|
|
|
// New returns a Client struct
|
|
func New(options ...ClientOption) (*Client, error) {
|
|
c := &Client{httpClient: http.DefaultClient}
|
|
for _, option := range options {
|
|
if err := option(c); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return c, nil
|
|
}
|
|
|
|
// Open an io.ReadCloser from a local file or URL
|
|
func Open(name string, options ...ClientOption) (io.ReadCloser, error) {
|
|
c, err := New(options...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.Open(name)
|
|
}
|
|
|
|
func WithHTTPClient(v *http.Client) ClientOption {
|
|
return func(c *Client) error {
|
|
c.httpClient = v
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func (c *Client) Open(name string) (io.ReadCloser, error) {
|
|
if strings.HasPrefix(name, "http://") || strings.HasPrefix(name, "https://") {
|
|
resp, err := c.httpClient.Get(name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return resp.Body, nil
|
|
}
|
|
return os.Open(name)
|
|
}
|