drone/gitrpc/diff.go

// Copyright 2022 Harness Inc. All rights reserved.
// Use of this source code is governed by the Polyform Free Trial License
// that can be found in the LICENSE.md file for this repository.

package gitrpc

import (
	"context"
	"fmt"
	"io"

	"github.com/harness/gitness/gitrpc/internal/streamio"
	"github.com/harness/gitness/gitrpc/rpc"
)

type RawDiffRequest struct {
	RepoID        string
	LeftCommitID  string
	RightCommitID string
}

func (c *Client) RawDiff(ctx context.Context, in *RawDiffRequest, w io.Writer) error {
	diff, err := c.diffService.RawDiff(ctx, &rpc.RawDiffRequest{
		RepoId:        in.RepoID,
		LeftCommitId:  in.LeftCommitID,
		RightCommitId: in.RightCommitID,
	})
	if err != nil {
		return err
	}

	reader := streamio.NewReader(func() ([]byte, error) {
		var resp *rpc.RawDiffResponse
		resp, err = diff.Recv()
		return resp.GetData(), err
	})

	if _, err = io.Copy(w, reader); err != nil {
		return fmt.Errorf("copy rpc data: %w", err)
	}

	return nil
}