// This file is Free Software under the MIT License
// without warranty, see README.md and LICENSES/MIT.txt for details.
//
// SPDX-License-Identifier: MIT
//
// SPDX-FileCopyrightText: 2022 German Federal Office for Information Security (BSI) <https://www.bsi.bund.de>
// Software-Engineering: 2022 Intevation GmbH <https://intevation.de>

package main

import (
	"crypto/tls"
	"errors"
	"fmt"
	"net/http"
	"os"
	"runtime"
	"sync"
	"time"

	"github.com/ProtonMail/gopenpgp/v2/crypto"
	"github.com/csaf-poc/csaf_distribution/v2/csaf"
	"github.com/csaf-poc/csaf_distribution/v2/internal/certs"
	"github.com/csaf-poc/csaf_distribution/v2/internal/filter"
	"github.com/csaf-poc/csaf_distribution/v2/internal/models"
	"github.com/csaf-poc/csaf_distribution/v2/internal/options"
	"github.com/csaf-poc/csaf_distribution/v2/util"
	"golang.org/x/time/rate"
)

const (
	defaultWorkers        = 10
	defaultFolder         = "/var/www"
	defaultWeb            = "/var/www/html"
	defaultDomain         = "https://example.com"
	defaultUpdateInterval = "on best effort"
	defaultLockFile       = "/var/lock/csaf_aggregator/lock"
)

type provider struct {
	Name   string `toml:"name"`
	Domain string `toml:"domain"`
	// Rate gives the provider specific rate limiting (see overall Rate).
	Rate         *float64  `toml:"rate"`
	Insecure     *bool     `toml:"insecure"`
	WriteIndices *bool     `toml:"write_indices"`
	Categories   *[]string `toml:"categories"`
	// ServiceDocument incidates if we should create a service.json document.
	ServiceDocument     *bool                    `toml:"create_service_document"`
	AggregatoryCategory *csaf.AggregatorCategory `toml:"category"`

	// UpdateInterval is as the mandatory `update_interval` if this is a publisher.
	UpdateInterval *string `toml:"update_interval"`

	// IgnorePattern is a list of patterns of advisory URLs to be ignored.
	IgnorePattern []string `toml:"ignorepattern"`

	// ExtraHeader adds extra HTTP header fields to client
	ExtraHeader http.Header `toml:"header"`

	ClientCert       *string `toml:"client_cert"`
	ClientKey        *string `toml:"client_key"`
	ClientPassphrase *string `toml:"client_passphrase"`

	Range *models.TimeRange `toml:"timerange"`

	clientCerts   []tls.Certificate
	ignorePattern filter.PatternMatcher
}

type config struct {
	Verbose bool `toml:"verbose"`
	// Workers is the number of concurrently executed workers for downloading.
	Workers int    `toml:"workers"`
	Folder  string `toml:"folder"`
	Web     string `toml:"web"`
	Domain  string `toml:"domain"`
	// Rate gives the average upper limit of https operations per second.
	Rate                *float64            `toml:"rate"`
	Insecure            *bool               `toml:"insecure"`
	Categories          *[]string           `toml:"categories"`
	WriteIndices        bool                `toml:"write_indices"`
	Aggregator          csaf.AggregatorInfo `toml:"aggregator"`
	Providers           []*provider         `toml:"providers"`
	OpenPGPPrivateKey   string              `toml:"openpgp_private_key"`
	OpenPGPPublicKey    string              `toml:"openpgp_public_key"`
	Passphrase          *string             `toml:"passphrase"`
	AllowSingleProvider bool                `toml:"allow_single_provider"`

	ClientCert       *string `toml:"client_cert"`
	ClientKey        *string `toml:"client_key"`
	ClientPassphrase *string `toml:"client_passphrase"`

	Range *models.TimeRange `long:"timerange" short:"t" description:"RANGE of time from which advisories to download" value-name:"RANGE" toml:"timerange"`

	// LockFile tries to lock to a given file.
	LockFile *string `toml:"lock_file"`

	// Interim performs an interim scan.
	Interim bool `short:"i" long:"interim" description:"Perform an interim scan" toml:"interim"`
	Version bool `long:"version" description:"Display version of the binary" toml:"-"`

	// InterimYears is numbers numbers of years to look back
	// for interim advisories. Less/equal zero means forever.
	InterimYears int `toml:"interim_years"`

	// RemoteValidator configures an optional remote validation.
	RemoteValidatorOptions *csaf.RemoteValidatorOptions `toml:"remote_validator"`

	// ServiceDocument incidates if we should create a service.json document.
	ServiceDocument bool `toml:"create_service_document"`

	// UpdateInterval is used for publishers as the mandatory field
	// 'update_interval'.
	UpdateInterval *string `toml:"update_interval"`

	// IgnorePattern is a list of patterns of advisory URLs to be ignored.
	IgnorePattern []string `toml:"ignorepattern"`

	// ExtraHeader adds extra HTTP header fields to client
	ExtraHeader http.Header `toml:"header"`

	Config string `short:"c" long:"config" description:"Path to config TOML file" value-name:"TOML-FILE" toml:"-"`

	keyMu  sync.Mutex
	key    *crypto.Key
	keyErr error

	clientCerts   []tls.Certificate
	ignorePattern filter.PatternMatcher
}

// configPaths are the potential file locations of the config file.
var configPaths = []string{
	// TODO: Make symmetric to checker and downloader.
	"aggregator.toml",
}

// parseArgsConfig parse the command arguments and loads configuration
// from a configuration file.
func parseArgsConfig() ([]string, *config, error) {
	p := options.Parser[config]{
		DefaultConfigLocations: configPaths,
		ConfigLocation: func(cfg *config) string {
			return cfg.Config
		},
		HasVersion: func(cfg *config) bool { return cfg.Version },
		// Establish default values if not set.
		EnsureDefaults: (*config).setDefaults,
	}
	return p.Parse()
}

// tooOldForInterims returns a function that tells if a given
// time is too old for the configured interims interval.
func (c *config) tooOldForInterims() func(time.Time) bool {
	if c.InterimYears <= 0 {
		return func(time.Time) bool { return false }
	}
	from := time.Now().AddDate(-c.InterimYears, 0, 0)
	return func(t time.Time) bool { return t.Before(from) }
}

// ageAccept returns a function which checks if a given time
// is in the accepted download interval of the provider or
// the global config.
func (p *provider) ageAccept(c *config) func(time.Time) bool {
	switch {
	case p.Range != nil:
		return p.Range.Contains
	case c.Range != nil:
		return c.Range.Contains
	default:
		return nil
	}
}

// ignoreFile returns true if the given URL should not be downloaded.
func (p *provider) ignoreURL(u string, c *config) bool {
	return p.ignorePattern.Matches(u) || c.ignorePattern.Matches(u)
}

// updateInterval returns the update interval of a publisher.
func (p *provider) updateInterval(c *config) string {
	if p.UpdateInterval != nil {
		return *p.UpdateInterval
	}
	if c.UpdateInterval != nil {
		return *c.UpdateInterval
	}
	return defaultUpdateInterval
}

// serviceDocument tells if we should generate a service document for a
// given provider.
func (p *provider) serviceDocument(c *config) bool {
	if p.ServiceDocument != nil {
		return *p.ServiceDocument
	}
	return c.ServiceDocument
}

// writeIndices tells if we should write index.txt and changes.csv.
func (p *provider) writeIndices(c *config) bool {
	if p.WriteIndices != nil {
		return *p.WriteIndices
	}
	return c.WriteIndices
}

func (p *provider) runAsMirror(c *config) bool {
	if p.AggregatoryCategory != nil {
		return *p.AggregatoryCategory == csaf.AggregatorAggregator
	}
	return c.runAsMirror()
}

// atLeastNMirrors checks if there are at least n mirrors configured.
func (c *config) atLeastNMirrors(n int) bool {
	var mirrors int
	for _, p := range c.Providers {
		if p.runAsMirror(c) {
			if mirrors++; mirrors >= n {
				return true
			}
		}
	}
	return false
}

// runAsMirror determines if the aggregator should run in mirror mode.
func (c *config) runAsMirror() bool {
	return c.Aggregator.Category != nil &&
		*c.Aggregator.Category == csaf.AggregatorAggregator
}

func (c *config) privateOpenPGPKey() (*crypto.Key, error) {
	if c.OpenPGPPrivateKey == "" {
		return nil, nil
	}
	c.keyMu.Lock()
	defer c.keyMu.Unlock()
	if c.key != nil || c.keyErr != nil {
		return c.key, c.keyErr
	}
	var f *os.File
	if f, c.keyErr = os.Open(c.OpenPGPPrivateKey); c.keyErr != nil {
		return nil, c.keyErr
	}
	defer f.Close()
	c.key, c.keyErr = crypto.NewKeyFromArmoredReader(f)
	return c.key, c.keyErr
}

func (c *config) httpClient(p *provider) util.Client {

	hClient := http.Client{}

	var tlsConfig tls.Config
	if p.Insecure != nil && *p.Insecure || c.Insecure != nil && *c.Insecure {
		tlsConfig.InsecureSkipVerify = true
	}

	// Use client certs if needed.
	switch {
	// Provider has precedence over global.
	case len(p.clientCerts) != 0:
		tlsConfig.Certificates = p.clientCerts
	case len(c.clientCerts) != 0:
		tlsConfig.Certificates = c.clientCerts
	}

	hClient.Transport = &http.Transport{
		TLSClientConfig: &tlsConfig,
	}

	client := util.Client(&hClient)

	// Add extra headers.
	switch {
	// Provider has precedence over global.
	case len(p.ExtraHeader) > 0:
		client = &util.HeaderClient{
			Client: client,
			Header: p.ExtraHeader,
		}
	case len(c.ExtraHeader) > 0:
		client = &util.HeaderClient{
			Client: client,
			Header: c.ExtraHeader,
		}
	}

	if c.Verbose {
		client = &util.LoggingClient{Client: client}
	}

	if p.Rate == nil && c.Rate == nil {
		return client
	}

	var r float64
	if c.Rate != nil {
		r = *c.Rate
	}
	if p.Rate != nil {
		r = *p.Rate
	}
	return &util.LimitingClient{
		Client:  client,
		Limiter: rate.NewLimiter(rate.Limit(r), 1),
	}
}

func (c *config) checkProviders() error {

	if !c.AllowSingleProvider && len(c.Providers) < 2 {
		return errors.New("need at least two providers")
	}

	already := util.Set[string]{}

	for _, p := range c.Providers {
		if p.Name == "" {
			return errors.New("no name given for provider")
		}
		if p.Domain == "" {
			return errors.New("no domain given for provider")
		}
		if already.Contains(p.Name) {
			return fmt.Errorf("provider '%s' is configured more than once", p.Name)
		}
		already.Add(p.Name)
	}
	return nil
}

func (c *config) checkMirror() error {
	if c.runAsMirror() {
		if !c.AllowSingleProvider && !c.atLeastNMirrors(2) {
			return errors.New("at least 2 providers need to be mirrored")
		} else if c.AllowSingleProvider && !c.atLeastNMirrors(1) {
			return errors.New("at least one provider must be mirrored")
		}
	} else if !c.AllowSingleProvider && c.atLeastNMirrors(1) {
		return errors.New("found mirrors in a lister aggregator")
	}

	return nil
}

func (c *config) setDefaults() {
	if c.Folder == "" {
		c.Folder = defaultFolder
	}

	if c.Web == "" {
		c.Web = defaultWeb
	}

	if c.Domain == "" {
		c.Domain = defaultDomain
	}

	switch {
	case c.LockFile == nil:
		lockFile := defaultLockFile
		c.LockFile = &lockFile
	case *c.LockFile == "":
		c.LockFile = nil
	}

	if c.Workers <= 0 {
		if n := runtime.NumCPU(); n > defaultWorkers {
			c.Workers = defaultWorkers
		} else {
			c.Workers = n
		}
	}

	if c.Workers > len(c.Providers) {
		c.Workers = len(c.Providers)
	}
}

// compileIgnorePatterns compiles the configured patterns to be ignored.
func (p *provider) compileIgnorePatterns() error {
	pm, err := filter.NewPatternMatcher(p.IgnorePattern)
	if err != nil {
		return fmt.Errorf("invalid ignore patterns for %q: %w", p.Name, err)
	}
	p.ignorePattern = pm
	return nil
}

// compileIgnorePatterns compiles the configured patterns to be ignored.
func (c *config) compileIgnorePatterns() error {
	// Compile the top level patterns.
	pm, err := filter.NewPatternMatcher(c.IgnorePattern)
	if err != nil {
		return err
	}
	c.ignorePattern = pm
	// Compile the patterns of the providers.
	for _, p := range c.Providers {
		if err := p.compileIgnorePatterns(); err != nil {
			return err
		}
	}
	return nil
}

// prepareCertificates loads the provider specific client side certificates
// used by the HTTP client.
func (p *provider) prepareCertificates() error {
	cert, err := certs.LoadCertificate(
		p.ClientCert, p.ClientKey, p.ClientPassphrase)
	if err != nil {
		return fmt.Errorf("invalid certificates for %q: %w", p.Name, err)
	}
	p.clientCerts = cert
	return nil
}

// prepareCertificates loads the client side certificates used by the HTTP client.
func (c *config) prepareCertificates() error {
	// Global certificates
	cert, err := certs.LoadCertificate(
		c.ClientCert, c.ClientKey, c.ClientPassphrase)
	if err != nil {
		return err
	}
	c.clientCerts = cert
	// Provider certificates
	for _, p := range c.Providers {
		if err := p.prepareCertificates(); err != nil {
			return err
		}
	}
	return nil
}

// prepare prepares internal state of a loaded configuration.
func (c *config) prepare() error {

	if len(c.Providers) == 0 {
		return errors.New("no providers given in configuration")
	}

	for _, prepare := range []func() error{
		c.prepareCertificates,
		c.compileIgnorePatterns,
		c.Aggregator.Validate,
		c.checkProviders,
		c.checkMirror,
	} {
		if err := prepare(); err != nil {
			return err
		}
	}
	return nil
}