package wiiudownloader

import (
	"crypto/aes"
	"crypto/cipher"
	"crypto/sha1"
	"errors"
	"fmt"
	"os"
	"path/filepath"
	"reflect"
)

var commonKey = []byte{0xD7, 0xB0, 0x04, 0x02, 0x65, 0x9B, 0xA2, 0xAB, 0xD2, 0xCB, 0x0D, 0xB2, 0x7F, 0xA2, 0xB6, 0x56}

func checkContentHashes(path string, content contentInfo, cipherHashTree cipher.Block) error {
	h3Data, err := os.ReadFile(filepath.Join(path, fmt.Sprintf("%s.h3", content.ID)))
	if err != nil {
		return fmt.Errorf("failed to read H3 hash tree file: %w", err)
	}
	encryptedFile, err := os.Open(filepath.Join(path, fmt.Sprintf("%s.app", content.ID)))
	if err != nil {
		return fmt.Errorf("failed to open encrypted file: %w", err)
	}
	defer encryptedFile.Close()

	h3Hash := sha1.Sum(h3Data)
	if !reflect.DeepEqual(h3Hash[:8], content.Hash[:8]) {
		return errors.New("h3 Hash mismatch")
	}

	chunkCount := int(content.Size / 0x10000)
	decryptedContent := make([]byte, 0x400)

	h0HashNum := 0
	h1HashNum := 0
	h2HashNum := 0
	h3HashNum := 0

	buffer := make([]byte, 0x400)
	iv := make([]byte, aes.BlockSize)
	for chunkNum := 0; chunkNum < chunkCount; chunkNum++ {
		encryptedFile.Read(buffer)
		cipher.NewCBCDecrypter(cipherHashTree, iv).CryptBlocks(decryptedContent, buffer)

		h0Hashes := decryptedContent[0:0x140]
		h1Hashes := decryptedContent[0x140:0x280]
		h2Hashes := decryptedContent[0x280:0x3c0]

		h1Hash := h1Hashes[(h1HashNum * 0x14):((h1HashNum + 1) * 0x14)]
		h2Hash := h2Hashes[(h2HashNum * 0x14):((h2HashNum + 1) * 0x14)]
		h3Hash := h3Data[(h3HashNum * 0x14):((h3HashNum + 1) * 0x14)]

		h0HashesHash := sha1.Sum(h0Hashes)
		h1HashesHash := sha1.Sum(h1Hashes)
		h2HashesHash := sha1.Sum(h2Hashes)

		if !reflect.DeepEqual(h0HashesHash[:], h1Hash) {
			return errors.New("h0 Hashes Hash mismatch")
		}
		if !reflect.DeepEqual(h1HashesHash[:], h2Hash) {
			return errors.New("h1 Hashes Hash mismatch")
		}
		if !reflect.DeepEqual(h2HashesHash[:], h3Hash) {
			return errors.New("h2 Hashes Hash mismatch")
		}
		encryptedFile.Seek(0xFC00, 1)
		h0HashNum++
		if h0HashNum >= 16 {
			h0HashNum = 0
			h1HashNum++
		}
		if h1HashNum >= 16 {
			h1HashNum = 0
			h2HashNum++
		}
		if h2HashNum >= 16 {
			h2HashNum = 0
			h3HashNum++
		}
	}
	return nil
}

type contentInfo struct {
	ID   string
	Hash []byte
	Size int64
}