117 lines
3 KiB
Go
117 lines
3 KiB
Go
// Copyright 2017 Frédéric Guillot. All rights reserved.
|
|
// Use of this source code is governed by the Apache 2.0
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package http
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/miniflux/miniflux/helper"
|
|
"github.com/miniflux/miniflux/logger"
|
|
)
|
|
|
|
// Note: Some websites have a user agent filter.
|
|
const userAgent = "Mozilla/5.0 (like Gecko, like Safari, like Chrome) - Miniflux <https://miniflux.net/>"
|
|
const requestTimeout = 300
|
|
|
|
// Client is a HTTP Client :)
|
|
type Client struct {
|
|
url string
|
|
etagHeader string
|
|
lastModifiedHeader string
|
|
username string
|
|
password string
|
|
Insecure bool
|
|
}
|
|
|
|
// Get execute a GET HTTP request.
|
|
func (c *Client) Get() (*Response, error) {
|
|
defer helper.ExecutionTime(time.Now(), fmt.Sprintf("[HttpClient:Get] url=%s", c.url))
|
|
|
|
client := c.buildClient()
|
|
resp, err := client.Do(c.buildRequest())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
response := &Response{
|
|
Body: resp.Body,
|
|
StatusCode: resp.StatusCode,
|
|
EffectiveURL: resp.Request.URL.String(),
|
|
LastModified: resp.Header.Get("Last-Modified"),
|
|
ETag: resp.Header.Get("ETag"),
|
|
ContentType: resp.Header.Get("Content-Type"),
|
|
}
|
|
|
|
logger.Debug("[HttpClient:Get] OriginalURL=%s, StatusCode=%d, ETag=%s, LastModified=%s, EffectiveURL=%s",
|
|
c.url,
|
|
response.StatusCode,
|
|
response.ETag,
|
|
response.LastModified,
|
|
response.EffectiveURL,
|
|
)
|
|
|
|
return response, err
|
|
}
|
|
|
|
func (c *Client) buildRequest() *http.Request {
|
|
link, _ := url.Parse(c.url)
|
|
request := &http.Request{
|
|
URL: link,
|
|
Method: http.MethodGet,
|
|
Header: c.buildHeaders(),
|
|
}
|
|
|
|
if c.username != "" && c.password != "" {
|
|
request.SetBasicAuth(c.username, c.password)
|
|
}
|
|
|
|
return request
|
|
}
|
|
|
|
func (c *Client) buildClient() http.Client {
|
|
client := http.Client{Timeout: time.Duration(requestTimeout * time.Second)}
|
|
if c.Insecure {
|
|
client.Transport = &http.Transport{
|
|
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
|
}
|
|
}
|
|
|
|
return client
|
|
}
|
|
|
|
func (c *Client) buildHeaders() http.Header {
|
|
headers := make(http.Header)
|
|
headers.Add("User-Agent", userAgent)
|
|
headers.Add("Accept", "text/html,application/xhtml+xml,application/xml,application/json")
|
|
|
|
if c.etagHeader != "" {
|
|
headers.Add("If-None-Match", c.etagHeader)
|
|
}
|
|
|
|
if c.lastModifiedHeader != "" {
|
|
headers.Add("If-Modified-Since", c.lastModifiedHeader)
|
|
}
|
|
|
|
return headers
|
|
}
|
|
|
|
// NewClient returns a new HTTP client.
|
|
func NewClient(url string) *Client {
|
|
return &Client{url: url, Insecure: false}
|
|
}
|
|
|
|
// NewClientWithCredentials returns a new HTTP client that require authentication.
|
|
func NewClientWithCredentials(url, username, password string) *Client {
|
|
return &Client{url: url, Insecure: false, username: username, password: password}
|
|
}
|
|
|
|
// NewClientWithCacheHeaders returns a new HTTP client that send cache headers.
|
|
func NewClientWithCacheHeaders(url, etagHeader, lastModifiedHeader string) *Client {
|
|
return &Client{url: url, etagHeader: etagHeader, lastModifiedHeader: lastModifiedHeader, Insecure: false}
|
|
}
|