// Package hash 实现 TCS-IPTV 的内容哈希核心：
// 文件 SHA-256、分段 Merkle Tree、感知哈希。
// 对应需求：需求1（母版哈希生成）。
package hash

import (
	"crypto/sha256"
	"encoding/hex"
	"errors"
	"io"
	"os"
)

// DefaultSegmentSize 是分段哈希的默认分段大小（10 MiB）。
const DefaultSegmentSize = 10 * 1024 * 1024

// ErrEmptyInput 表示输入为空。
var ErrEmptyInput = errors.New("hash: empty input")

// SHA256Hex 计算字节切片的 SHA-256，返回十六进制字符串。
func SHA256Hex(data []byte) string {
	sum := sha256.Sum256(data)
	return hex.EncodeToString(sum[:])
}

// FileSHA256 流式计算文件的整体 SHA-256，避免一次性载入大文件。
func FileSHA256(path string) (string, error) {
	f, err := os.Open(path)
	if err != nil {
		return "", err
	}
	defer f.Close()

	h := sha256.New()
	if _, err := io.Copy(h, f); err != nil {
		return "", err
	}
	return hex.EncodeToString(h.Sum(nil)), nil
}

// SegmentHashes 按 segmentSize 分段计算文件各段的 SHA-256。
// 用于构建 Merkle Tree 的叶子节点。
func SegmentHashes(path string, segmentSize int) ([]string, error) {
	if segmentSize <= 0 {
		segmentSize = DefaultSegmentSize
	}
	f, err := os.Open(path)
	if err != nil {
		return nil, err
	}
	defer f.Close()

	var segments []string
	buf := make([]byte, segmentSize)
	for {
		n, err := io.ReadFull(f, buf)
		if n > 0 {
			segments = append(segments, SHA256Hex(buf[:n]))
		}
		if err == io.EOF || err == io.ErrUnexpectedEOF {
			break
		}
		if err != nil {
			return nil, err
		}
	}
	return segments, nil
}