package fileutil

import (
	"context"
	"crypto/sha256"
	"encoding/hex"
	"io"
	"os"
	"path/filepath"
)

type HashIndex struct {
	files map[string]string
}

func NewHashIndex(ctx context.Context, root string) (*HashIndex, error) {
	index := &HashIndex{files: make(map[string]string)}

	err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
		if ctx.Err() != nil {
			return ctx.Err()
		}

		if err != nil || info.IsDir() {
			return nil
		}

		file, err := os.Open(path)
		if err != nil {
			return nil
		}
		defer file.Close()

		h := sha256.New()
		if _, err := io.Copy(h, file); err != nil {
			return nil
		}

		sum := hex.EncodeToString(h.Sum(nil))
		index.files[sum] = path

		return nil
	})

	if err != nil {
		return nil, err
	}

	return index, nil
}

func (h *HashIndex) Find(hash string) (string, bool) {
	path, ok := h.files[hash]

	return path, ok
}