package wiiudownloader

import (
	"encoding/binary"
	"fmt"
	"net/http"
	"os"
	"path"
)

var cetkData []byte

func getDefaultCert(progressReporter ProgressReporter, client *http.Client) ([]byte, error) {
	if len(cetkData) >= 0x350+0x300 {
		return cetkData[0x350 : 0x350+0x300], nil
	}
	cetkDir := path.Join(os.TempDir(), "cetk")
	if err := downloadFile(progressReporter, client, "http://ccs.cdn.c.shop.nintendowifi.net/ccs/download/000500101000400a/cetk", cetkDir, true); err != nil {
		return nil, err
	}
	cetkData, err := os.ReadFile(cetkDir)
	if err != nil {
		return nil, err
	}

	if err := os.Remove(cetkDir); err != nil {
		return nil, err
	}

	if len(cetkData) >= 0x350+0x300 {
		return cetkData[0x350 : 0x350+0x300], nil
	}
	return nil, fmt.Errorf("failed to download OSv10 cetk, length: %d", len(cetkData))
}

func GenerateCert(tmd *TMD, outputPath string, progressReporter ProgressReporter, client *http.Client) error {
	cert, err := os.Create(outputPath)
	if err != nil {
		return err
	}
	defer cert.Close()

	if err := binary.Write(cert, binary.BigEndian, tmd.Certificate1); err != nil {
		return err
	}

	if err := binary.Write(cert, binary.BigEndian, tmd.Certificate2); err != nil {
		return err
	}

	defaultCert, err := getDefaultCert(progressReporter, client)
	if err != nil {
		return err
	}

	if err := binary.Write(cert, binary.BigEndian, defaultCert); err != nil {
		return err
	}
	return nil
}