mirror of https://github.com/harness/drone.git
95 lines
2.6 KiB
Go
95 lines
2.6 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 repo
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/harness/gitness/gitrpc"
|
|
"github.com/harness/gitness/internal/auth"
|
|
"github.com/harness/gitness/internal/auth/authz"
|
|
"github.com/harness/gitness/internal/githook"
|
|
"github.com/harness/gitness/internal/store"
|
|
"github.com/harness/gitness/internal/url"
|
|
"github.com/harness/gitness/types"
|
|
"github.com/harness/gitness/types/check"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
type Controller struct {
|
|
defaultBranch string
|
|
db *sqlx.DB
|
|
urlProvider *url.Provider
|
|
uidCheck check.PathUID
|
|
authorizer authz.Authorizer
|
|
pathStore store.PathStore
|
|
repoStore store.RepoStore
|
|
spaceStore store.SpaceStore
|
|
principalStore store.PrincipalStore
|
|
gitRPCClient gitrpc.Interface
|
|
}
|
|
|
|
func NewController(
|
|
defaultBranch string,
|
|
db *sqlx.DB,
|
|
urlProvider *url.Provider,
|
|
uidCheck check.PathUID,
|
|
authorizer authz.Authorizer,
|
|
pathStore store.PathStore,
|
|
repoStore store.RepoStore,
|
|
spaceStore store.SpaceStore,
|
|
principalStore store.PrincipalStore,
|
|
gitRPCClient gitrpc.Interface,
|
|
) *Controller {
|
|
return &Controller{
|
|
defaultBranch: defaultBranch,
|
|
db: db,
|
|
urlProvider: urlProvider,
|
|
uidCheck: uidCheck,
|
|
authorizer: authorizer,
|
|
pathStore: pathStore,
|
|
repoStore: repoStore,
|
|
spaceStore: spaceStore,
|
|
principalStore: principalStore,
|
|
gitRPCClient: gitRPCClient,
|
|
}
|
|
}
|
|
|
|
// CreateRPCWriteParams creates base write parameters for gitrpc write operations.
|
|
// IMPORTANT: session & repo are assumed to be not nil!
|
|
func CreateRPCWriteParams(ctx context.Context, urlProvider *url.Provider,
|
|
session *auth.Session, repo *types.Repository) (gitrpc.WriteParams, error) {
|
|
// generate envars (add everything githook CLI needs for execution)
|
|
envVars, err := githook.GenerateEnvironmentVariables(
|
|
ctx,
|
|
urlProvider.GetAPIBaseURLInternal(),
|
|
repo.ID,
|
|
session.Principal.ID,
|
|
false,
|
|
)
|
|
if err != nil {
|
|
return gitrpc.WriteParams{}, fmt.Errorf("failed to generate git hook environment variables: %w", err)
|
|
}
|
|
|
|
return gitrpc.WriteParams{
|
|
Actor: gitrpc.Identity{
|
|
Name: session.Principal.DisplayName,
|
|
Email: session.Principal.Email,
|
|
},
|
|
RepoUID: repo.GitUID,
|
|
EnvVars: envVars,
|
|
}, nil
|
|
}
|
|
|
|
// CreateRPCReadParams creates base read parameters for gitrpc read operations.
|
|
// IMPORTANT: repo is assumed to be not nil!
|
|
func CreateRPCReadParams(repo *types.Repository) gitrpc.ReadParams {
|
|
return gitrpc.ReadParams{
|
|
RepoUID: repo.GitUID,
|
|
}
|
|
}
|