request.go

package wiki

import (
	"crypto/tls"
	"encoding/json"
	"fmt"
	"net/http"
	"net/url"
	"strings"
)

// Request sets up the request against the api with the correct parameters
// and has functionality to fetch the data and convert it to a response.
type Request struct {
	*url.URL
}

// NewRequest creates a new request against baseURL for language.
// Language is interpolated in the baseURL if asked, if not it is ignored.
// Query is the title of the page to fetch.
// Returns an error if the URL can not be parsed.
func NewRequest(baseURL, query, language string) (*Request, error) {
	if strings.Contains(baseURL, "%s") {
		baseURL = fmt.Sprintf(baseURL, language)
	}
	url, err := url.Parse(baseURL)
	if err != nil {
		return nil, err
	}
	v := url.Query()
	v.Set("action", "query")
	v.Set("prop", "extracts|info")
	v.Set("format", "json")
	v.Set("exintro", "")
	v.Set("explaintext", "")
	v.Set("inprop", "url")
	v.Set("redirects", "")
	v.Set("converttitles", "")
	v.Set("titles", query)
	url.RawQuery = v.Encode()

	return &Request{url}, nil
}

// Execute fetches the data and decodes it into a Response.
// Returns an error if the data could not be retrieved or the decoding fails.
func (r *Request) Execute(noCheckCert bool) (*Response, error) {
	client := &http.Client{}

	if noCheckCert {
		tr := &http.Transport{
			TLSClientConfig: &tls.Config{InsecureSkipVerify: true}, //nolint:gosec
		}
		client = &http.Client{Transport: tr}
	}

	response, err := client.Get(r.String())
	if err != nil {
		return nil, err
	}
	defer response.Body.Close()

	d := json.NewDecoder(response.Body)
	resp := &Response{}
	err = d.Decode(resp)
	if err != nil {
		return nil, err
	}

	return resp, nil
}