mirror of
https://github.com/harness/drone.git
synced 2025-05-05 05:39:10 +08:00
71 lines
1.7 KiB
Go
71 lines
1.7 KiB
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 service
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/harness/gitness/gitrpc/internal/streamio"
|
|
"github.com/harness/gitness/gitrpc/internal/types"
|
|
"github.com/harness/gitness/gitrpc/rpc"
|
|
|
|
"code.gitea.io/gitea/modules/git"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
)
|
|
|
|
type DiffService struct {
|
|
rpc.UnimplementedDiffServiceServer
|
|
adapter GitAdapter
|
|
reposRoot string
|
|
}
|
|
|
|
func NewDiffService(adapter GitAdapter, reposRoot string) (*DiffService, error) {
|
|
return &DiffService{
|
|
adapter: adapter,
|
|
reposRoot: reposRoot,
|
|
}, nil
|
|
}
|
|
|
|
func (s DiffService) RawDiff(request *rpc.RawDiffRequest, stream rpc.DiffService_RawDiffServer) error {
|
|
err := validateDiffRequest(request)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ctx := stream.Context()
|
|
base := request.GetBase()
|
|
|
|
sw := streamio.NewWriter(func(p []byte) error {
|
|
return stream.Send(&rpc.RawDiffResponse{Data: p})
|
|
})
|
|
|
|
repoPath := getFullPathForRepo(s.reposRoot, base.GetRepoUid())
|
|
|
|
cmd := git.NewCommand(ctx, "diff", "--full-index", request.LeftCommitId, request.RightCommitId)
|
|
cmd.SetDescription(fmt.Sprintf("GetDiffRange [repo_path: %s]", repoPath))
|
|
return cmd.Run(&git.RunOpts{
|
|
Timeout: time.Duration(setting.Git.Timeout.Default) * time.Second,
|
|
Dir: repoPath,
|
|
Stderr: os.Stderr,
|
|
Stdout: sw,
|
|
})
|
|
}
|
|
|
|
func validateDiffRequest(in *rpc.RawDiffRequest) error {
|
|
if in.GetBase() == nil {
|
|
return types.ErrBaseCannotBeEmpty
|
|
}
|
|
if in.GetLeftCommitId() == "" {
|
|
return types.ErrEmptyLeftCommitID
|
|
}
|
|
if in.GetRightCommitId() == "" {
|
|
return types.ErrEmptyRightCommitID
|
|
}
|
|
|
|
return nil
|
|
}
|