package needle

import (
	"fmt"
	"hash"
	"io"

	"github.com/klauspost/crc32"

	"github.com/chrislusf/seaweedfs/weed/util"
)

var table = crc32.MakeTable(crc32.Castagnoli)

type CRC uint32

func NewCRC(b []byte) CRC {
	return CRC(0).Update(b)
}

func (c CRC) Update(b []byte) CRC {
	return CRC(crc32.Update(uint32(c), table, b))
}

func (c CRC) Value() uint32 {
	return uint32(c>>15|c<<17) + 0xa282ead8
}

func (n *Needle) Etag() string {
	bits := make([]byte, 4)
	util.Uint32toBytes(bits, uint32(n.Checksum))
	return fmt.Sprintf("%x", bits)
}

func NewCRCwriter(w io.Writer) *CRCwriter {

	return &CRCwriter{
		h: crc32.New(table),
		w: w,
	}

}

type CRCwriter struct {
	h hash.Hash32
	w io.Writer
}

func (c *CRCwriter) Write(p []byte) (n int, err error) {
	n, err = c.w.Write(p) // with each write ...
	c.h.Write(p)          // ... update the hash
	return
}

func (c *CRCwriter) Sum() uint32 { return c.h.Sum32() } // final hash