// 提供图像感知哈希算法功能（pHash）。
package phash

import (
	"encoding/binary"
	"encoding/hex"
	"fmt"
	"image"
	"os"

	"github.com/azr/phash"
)

// 计算图像的感知哈希值，并返回字节数组。
func Hash(image image.Image) []byte {
	var bashBytes = make([]byte, 8)
	hash := phash.DTC(image)
	binary.BigEndian.PutUint64(bashBytes, hash)
	return bashBytes
}

// 计算图像的感知哈希值，并返回十六进制字符串。
func HashHex(image image.Image) string {
	return hex.EncodeToString(Hash(image))
}

// 计算指定图像文件的感知哈希值，并返回字节数组。
func HashFile(file string) ([]byte, error) {
	f, err := os.Open(file)
	if err != nil {
		return nil, fmt.Errorf("未能打开指定文件，%w", err)
	}
	defer f.Close()

	img, _, err := image.Decode(f)
	if err != nil {
		return nil, fmt.Errorf("未能解码指定图像文件，%w", err)
	}
	return Hash(img), nil
}

// 计算指定图像文件的感知哈希值，并返回十六进制字符串。
func HashFileHex(file string) (string, error) {
	hash, err := HashFile(file)
	if err != nil {
		return "", err
	}
	return hex.EncodeToString(hash), nil
}