package webshared

import (
	"crypto/ed25519"
	"crypto/sha256"
	"crypto/x509"
	"encoding/base64"
	"io"
	"net/http"
	"strings"
	"time"

	"git.mstar.dev/mstar/goutils/maputils"
	"github.com/go-ap/httpsig"
	"github.com/rs/zerolog/log"

	"git.mstar.dev/mstar/linstrom/config"
)

// No init needed, zero value is good

var RequestClient = http.Client{}

const xRandomHeader = "X-Auth-Random"

// Sign a given outbound request for authorized fetch.
// At the end, the Signature header will have the signature needed,
// nothing else is modified.
// If the request is POST, the postBody must contain the raw body of
// the request and the Digest header will also be added
func SignRequest(r *http.Request, keyId string, privateKeyBytes, postBody []byte) error {
	method := r.Method
	headers := r.Header
	var nowString string
	if dateString := headers.Get("Date"); dateString != "" {
		nowString = dateString
	} else {
		nowString = time.Now().Format("Mon, 02 Jan 2006 15:04:05 MST")
		headers.Set("Date", nowString)
	}
	var host string
	if hostString := headers.Get("Host"); hostString != "" {
		host = hostString
	} else {
		host = config.GlobalConfig.General.GetFullDomain()
		headers.Set("Host", host)
	}
	applyBodyHash(headers, postBody)
	mappedHeaders := maputils.MapNewKeys(headers, func(k string, v []string) (string, string) {
		if len(v) > 0 {
			return strings.ToLower(k), v[0]
		} else {
			return strings.ToLower(k), ""
		}
	})
	// Filter for only the date, host, digest and request-target headers
	mappedHeaders = maputils.FilterMap(mappedHeaders, func(k, v string) bool {
		k = strings.ToLower(k)
		return k == "date" || k == "host" || k == "digest" || k == "(request-target)"
	})
	var signedString string
	var usedHeaders []string
	if config.GlobalConfig.Experimental.UseEd25519Keys {
		tmp, tmp2, err := CreateSignatureED(method, r.URL.Path, mappedHeaders, privateKeyBytes)
		if err != nil {
			return err
		}
		signedString = tmp
		usedHeaders = tmp2
	} else {
		tmp, tmp2, err := CreateSignatureRSA(method, r.URL.Path, mappedHeaders, privateKeyBytes)
		if err != nil {
			return err
		}
		signedString = tmp
		usedHeaders = tmp2
	}
	signature := CreateSignatureHeaderContent(
		keyId,
		signedString,
		usedHeaders...,
	)
	log.Debug().Str("signature-header", signature).Send()
	headers.Set("Signature", signature)
	return nil
}

func SignRequestWithHttpsig(
	r *http.Request,
	keyId string,
	privateKeyBytes, postBody []byte,
) error {
	keyId = config.GlobalConfig.General.GetFullPublicUrl() + "/api/activitypub/user/" + keyId
	if config.GlobalConfig.Experimental.UseEd25519Keys {
		key := ed25519.PrivateKey(privateKeyBytes)
		signer := httpsig.NewEd25519Signer(keyId, key, nil)
		if err := signer.Sign(r); err != nil {
			return err
		}
	} else {
		key, err := x509.ParsePKCS1PrivateKey(privateKeyBytes)
		if err != nil {
			return err
		}
		signer := httpsig.NewRSASHA256Signer(keyId, key, nil)
		if err = signer.Sign(r); err != nil {
			return err
		}
		// r.Header.Add("Signature", strings.TrimPrefix(r.Header.Get("Authorization"), "Signature "))
	}
	return nil
}

func applyBodyHash(headers http.Header, body []byte) error {
	if body == nil {
		return nil
	}
	hash := sha256.Sum256(body)
	based := base64.StdEncoding.EncodeToString(hash[:])
	headers.Set("Digest", "SHA-256="+based)
	return nil
}

func NewRequest(method string, url string, body io.Reader) (*http.Request, error) {
	req, err := http.NewRequest(method, url, body)
	if err != nil {
		return nil, err
	}
	req.Header.Add(
		"User-Agent",
		"Linstrom v0.0.0-pre-alpha ("+config.GlobalConfig.General.GetFullDomain()+")",
	)
	req.Header.Add("Date", time.Now().Format(time.RFC1123))
	req.Header.Add("Host", config.GlobalConfig.General.GetFullDomain())
	return req, nil
}