package putsvc

import (
	"github.com/nspcc-dev/neofs-api-go/v2/object"
	putsvc "github.com/nspcc-dev/neofs-node/pkg/services/object/put"
	"github.com/pkg/errors"
)

type streamer struct {
	stream *putsvc.Streamer
}

func (s *streamer) Send(req *object.PutRequest) (err error) {
	switch v := req.GetBody().GetObjectPart().(type) {
	case *object.PutObjectPartInit:
		if err = s.stream.Init(toInitPrm(v, req.GetMetaHeader().GetSessionToken(), req.GetMetaHeader().GetTTL())); err != nil {
			err = errors.Wrapf(err, "(%T) could not init object put stream", s)
		}
	case *object.PutObjectPartChunk:
		if err = s.stream.SendChunk(toChunkPrm(v)); err != nil {
			err = errors.Wrapf(err, "(%T) could not send payload chunk", s)
		}
	default:
		err = errors.Errorf("(%T) invalid object put stream part type %T", s, v)
	}

	return
}

func (s *streamer) CloseAndRecv() (*object.PutResponse, error) {
	resp, err := s.stream.Close()
	if err != nil {
		return nil, errors.Wrapf(err, "(%T) could not object put stream", s)
	}

	return fromPutResponse(resp), nil
}