// Copyright 2026 The go-ethereum Authors
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.

package engine

import "encoding/json"

// estimateBlobAndProofV1Size returns a rough estimate of the JSON size for a BlobAndProofV1.
func estimateBlobAndProofV1Size(item *BlobAndProofV1) int {
	if item == nil {
		return 4
	}
	return len(item.Blob)*2 + len(item.Proof)*2 + 30
}

// marshalBlobAndProofV1 writes a BlobAndProofV1 as JSON and appends it to buf.
func marshalBlobAndProofV1(buf []byte, item *BlobAndProofV1) []byte {
	if item == nil {
		return append(buf, "null"...)
	}
	buf = append(buf, `{"blob":`...)
	buf = writeHexBytes(buf, item.Blob)

	buf = append(buf, `,"proof":`...)
	buf = writeHexBytes(buf, item.Proof)

	buf = append(buf, '}')
	return buf
}

// estimateBlobAndProofV2Size returns a rough estimate of the JSON size for a BlobAndProofV2.
func estimateBlobAndProofV2Size(item *BlobAndProofV2) int {
	if item == nil {
		return 4
	}
	size := len(item.Blob)*2 + 30
	for _, proof := range item.CellProofs {
		size += len(proof)*2 + 6
	}
	return size
}

// marshalBlobAndProofV2 writes a BlobAndProofV2 as JSON and appends it to buf.
func marshalBlobAndProofV2(buf []byte, item *BlobAndProofV2) []byte {
	if item == nil {
		return append(buf, "null"...)
	}
	buf = append(buf, `{"blob":`...)
	buf = writeHexBytes(buf, item.Blob)

	buf = append(buf, `,"proofs":`...)
	buf = marshalHexBytesArray(buf, item.CellProofs)

	buf = append(buf, '}')
	return buf
}

// MarshalJSON implements json.Marshaler.
func (list BlobAndProofListV1) MarshalJSON() ([]byte, error) {
	// Estimate buffer size.
	size := 2
	for _, item := range list {
		size += estimateBlobAndProofV1Size(item) + 1
	}
	buf := make([]byte, 0, size)

	// Write the array elements to the buffer.
	buf = append(buf, '[')
	for i, item := range list {
		if i > 0 {
			buf = append(buf, ',')
		}
		buf = marshalBlobAndProofV1(buf, item)
	}
	buf = append(buf, ']')
	return buf, nil
}

// UnmarshalJSON implements json.Unmarshaler.
func (list *BlobAndProofListV1) UnmarshalJSON(input []byte) error {
	if isJSONNull(input) {
		*list = nil
		return nil
	}
	items := make(BlobAndProofListV1, 0)
	if err := decodeJSONArray(input, func(value json.RawMessage) error {
		if isJSONNull(value) {
			items = append(items, nil)
			return nil
		}
		item := new(BlobAndProofV1)
		if err := decodeJSONObject(value, func(key string, value json.RawMessage) error {
			switch key {
			case "blob":
				return item.Blob.UnmarshalJSON(value)
			case "proof":
				return item.Proof.UnmarshalJSON(value)
			}
			return nil
		}); err != nil {
			return err
		}
		items = append(items, item)
		return nil
	}); err != nil {
		return err
	}
	*list = items
	return nil
}

// MarshalJSON implements json.Marshaler.
func (list BlobAndProofListV2) MarshalJSON() ([]byte, error) {
	// Estimate buffer size.
	size := 2
	for _, item := range list {
		size += estimateBlobAndProofV2Size(item) + 1
	}
	buf := make([]byte, 0, size)

	// Write the array elements to the buffer.
	buf = append(buf, '[')
	for i, item := range list {
		if i > 0 {
			buf = append(buf, ',')
		}
		buf = marshalBlobAndProofV2(buf, item)
	}
	buf = append(buf, ']')
	return buf, nil
}

// UnmarshalJSON implements json.Unmarshaler.
func (list *BlobAndProofListV2) UnmarshalJSON(input []byte) error {
	if isJSONNull(input) {
		*list = nil
		return nil
	}
	items := make(BlobAndProofListV2, 0)
	if err := decodeJSONArray(input, func(value json.RawMessage) error {
		if isJSONNull(value) {
			items = append(items, nil)
			return nil
		}
		item := new(BlobAndProofV2)
		if err := decodeJSONObject(value, func(key string, value json.RawMessage) error {
			switch key {
			case "blob":
				return item.Blob.UnmarshalJSON(value)
			case "proofs":
				proofs, err := unmarshalHexBytesArray(value)
				if err != nil {
					return err
				}
				item.CellProofs = proofs
			}
			return nil
		}); err != nil {
			return err
		}
		items = append(items, item)
		return nil
	}); err != nil {
		return err
	}
	*list = items
	return nil
}