package markup

import (
	"crypto/hmac"
	"crypto/sha256"
	"encoding/hex"
	"fmt"

	"github.com/yuin/goldmark/ast"
)

func GenerateCamoURL(baseURL, secret, imageURL string) string {
	h := hmac.New(sha256.New, []byte(secret))
	h.Write([]byte(imageURL))
	signature := hex.EncodeToString(h.Sum(nil))
	hexURL := hex.EncodeToString([]byte(imageURL))
	return fmt.Sprintf("%s/%s/%s", baseURL, signature, hexURL)
}

func (rctx *RenderContext) camoImageLinkTransformer(dst string) string {
	// don't camo on dev
	if rctx.IsDev {
		return dst
	}

	if rctx.CamoUrl != "" && rctx.CamoSecret != "" {
		return GenerateCamoURL(rctx.CamoUrl, rctx.CamoSecret, dst)
	}

	return dst
}

func (rctx *RenderContext) camoImageLinkAstTransformer(img *ast.Image) {
	dst := string(img.Destination)
	img.Destination = []byte(rctx.camoImageLinkTransformer(dst))
}