// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0

package request // import "go.opentelemetry.io/collector/pdata/xpdata/request"

import (
	"context"
	"fmt"

	"go.opentelemetry.io/collector/pdata/internal"
	"go.opentelemetry.io/collector/pdata/ptrace"
)

// MarshalTraces marshals ptrace.Traces along with the context into a byte slice.
func MarshalTraces(ctx context.Context, ld ptrace.Traces) ([]byte, error) {
	tr := internal.TracesRequest{
		FormatVersion:  requestFormatVersion,
		TracesData:     internal.TracesToProto(internal.TracesWrapper(ld)),
		RequestContext: encodeContext(ctx),
	}
	buf := make([]byte, tr.SizeProto())
	tr.MarshalProto(buf)
	return buf, nil
}

// UnmarshalTraces unmarshals a byte slice into ptrace.Traces and the context.
func UnmarshalTraces(buf []byte) (context.Context, ptrace.Traces, error) {
	ctx := context.Background()
	if !isRequestPayloadV1(buf) {
		return ctx, ptrace.Traces{}, ErrInvalidFormat
	}
	tr := internal.TracesRequest{}
	if err := tr.UnmarshalProto(buf); err != nil {
		return ctx, ptrace.Traces{}, fmt.Errorf("failed to unmarshal traces request: %w", err)
	}
	return decodeContext(ctx, tr.RequestContext), ptrace.Traces(internal.TracesFromProto(tr.TracesData)), nil
}