package poseidon

import (
	"bytes"
	"errors"
	"hash"
)

type digest struct {
	buf       *bytes.Buffer
	frameSize int
}

// Sum returns the Poseidon hash of the input bytes.
// use frame size of 16 inputs by default
func Sum(b []byte) []byte {
	h, _ := New(16)
	h.Write(b)
	return h.Sum(nil)
}

// New returns a new hash.Hash computing the Poseidon hash.
func New(frameSize int) (hash.Hash, error) {
	if frameSize < 2 || frameSize > 16 {
		return nil, errors.New("incorrect frame size")
	}
	return &digest{
		buf:       bytes.NewBuffer([]byte{}),
		frameSize: frameSize,
	}, nil
}

// Write (via the embedded io.Writer interface) adds more data to the running hash.
func (d *digest) Write(p []byte) (n int, err error) {
	return d.buf.Write(p)
}

// Sum returns the Poseidon checksum of the data.
func (d *digest) Sum(b []byte) []byte {
	hahs, err := HashBytesX(d.buf.Bytes(), d.frameSize)
	if err != nil {
		panic(err)
	}
	return append(b, hahs.Bytes()...)
}

// Reset resets the Hash to its initial state.
func (d *digest) Reset() {
	d.buf.Reset()
}

// Size returns the number of bytes Sum will return.
func (d *digest) Size() int {
	return 32
}

// BlockSize returns the hash block size.
func (d *digest) BlockSize() int {
	return spongeChunkSize
}