// Download files via HTTP with support for If-Modified-Since

// Copyright (C) 2019-2021  Simon Ruderich
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

package main

import (
	"crypto/tls"
	"crypto/x509"
	"fmt"
	"io/ioutil"
	"net/http"
	"time"

	"github.com/pkg/errors"
)

// Global variable to permit reuse of connections (keep-alive)
var clients map[string]*http.Client

func init() {
	clients = make(map[string]*http.Client)
	clients[""] = &http.Client{}
}

func fetchIfModified(url, user, pass, ca string, lastModified *time.Time) (int, []byte, error) {
	req, err := http.NewRequest("GET", url, nil)
	if err != nil {
		return 0, nil, err
	}
	if user != "" || pass != "" {
		req.SetBasicAuth(user, pass)
	}
	if !lastModified.IsZero() {
		req.Header.Add("If-Modified-Since",
			lastModified.UTC().Format(http.TimeFormat))
	}

	client, ok := clients[ca]
	if !ok {
		pem, err := ioutil.ReadFile(ca)
		if err != nil {
			return 0, nil, errors.Wrapf(err, "file.ca %q", ca)
		}
		pool := x509.NewCertPool()
		ok := pool.AppendCertsFromPEM(pem)
		if !ok {
			return 0, nil, fmt.Errorf(
				"file.ca %q: no PEM cert found", ca)
		}

		client = &http.Client{
			Transport: &http.Transport{
				TLSClientConfig: &tls.Config{
					RootCAs: pool,
				},
			},
		}
		clients[ca] = client
	}

	resp, err := client.Do(req)
	if err != nil {
		return 0, nil, err
	}
	defer resp.Body.Close()

	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return 0, nil, err
	}

	modified, err := http.ParseTime(resp.Header.Get("Last-Modified"))
	if err == nil {
		*lastModified = modified
	}

	return resp.StatusCode, body, nil
}