mirror of https://github.com/harness/drone.git
122 lines
3.1 KiB
Go
122 lines
3.1 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 server
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/harness/gitness/events"
|
|
git_events "github.com/harness/gitness/gitrpc/events"
|
|
"github.com/harness/gitness/gitrpc/internal/gitea"
|
|
"github.com/harness/gitness/gitrpc/internal/middleware"
|
|
"github.com/harness/gitness/gitrpc/internal/service"
|
|
"github.com/harness/gitness/gitrpc/internal/storage"
|
|
"github.com/harness/gitness/gitrpc/rpc"
|
|
|
|
"code.gitea.io/gitea/modules/setting"
|
|
grpc_middleware "github.com/grpc-ecosystem/go-grpc-middleware"
|
|
grpc_recovery "github.com/grpc-ecosystem/go-grpc-middleware/recovery"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
const (
|
|
repoSubdirName = "repos"
|
|
)
|
|
|
|
type Server struct {
|
|
*grpc.Server
|
|
Bind string
|
|
}
|
|
|
|
func NewServer(config Config, eventsSystem *events.System) (*Server, error) {
|
|
// Create repos folder
|
|
reposRoot := filepath.Join(config.GitRoot, repoSubdirName)
|
|
if _, err := os.Stat(reposRoot); errors.Is(err, os.ErrNotExist) {
|
|
if err = os.MkdirAll(reposRoot, 0o700); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
// TODO: should be subdir of gitRoot? What is it being used for?
|
|
setting.Git.HomePath = "home"
|
|
adapter, err := gitea.New()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// interceptors
|
|
errIntc := middleware.NewErrInterceptor()
|
|
logIntc := middleware.NewLogInterceptor()
|
|
|
|
s := grpc.NewServer(
|
|
grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(
|
|
grpc_recovery.UnaryServerInterceptor(),
|
|
logIntc.UnaryInterceptor(),
|
|
errIntc.UnaryInterceptor(),
|
|
)),
|
|
grpc.StreamInterceptor(grpc_middleware.ChainStreamServer(
|
|
grpc_recovery.StreamServerInterceptor(),
|
|
logIntc.StreamInterceptor(),
|
|
errIntc.StreamInterceptor(),
|
|
)),
|
|
)
|
|
store := storage.NewLocalStore()
|
|
eventReporter, err := git_events.NewReporter(eventsSystem)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("faield to create new event reporter: %w", err)
|
|
}
|
|
|
|
// initialize services
|
|
repoService, err := service.NewRepositoryService(adapter, store, reposRoot)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
refService, err := service.NewReferenceService(adapter, eventReporter, reposRoot)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
httpService, err := service.NewHTTPService(adapter, reposRoot)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
commitFilesService, err := service.NewCommitFilesService(adapter, reposRoot, config.ReposTempPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
diffService, err := service.NewDiffService(adapter, reposRoot)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// register services
|
|
rpc.RegisterRepositoryServiceServer(s, repoService)
|
|
rpc.RegisterReferenceServiceServer(s, refService)
|
|
rpc.RegisterSmartHTTPServiceServer(s, httpService)
|
|
rpc.RegisterCommitFilesServiceServer(s, commitFilesService)
|
|
rpc.RegisterDiffServiceServer(s, diffService)
|
|
|
|
return &Server{
|
|
Server: s,
|
|
Bind: config.Bind,
|
|
}, nil
|
|
}
|
|
|
|
func (s *Server) Start() error {
|
|
lis, err := net.Listen("tcp", s.Bind)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return s.Server.Serve(lis)
|
|
}
|
|
|
|
func (s *Server) Stop() error {
|
|
s.Server.GracefulStop()
|
|
return nil
|
|
}
|