app/api/controller/user/controller.go - main - githubmirror/gitness

app/api/controller/user/controller.go
93 строки3 KB
Marko Gaćeša
feat: [CODE-2554]: add commit and tag signature parse and verify (#3915)
02 июл 2025, 14:13
02 июл 2025, 14:137d0ffbf
// Copyright 2023 Harness, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package user

import (
	"context"

	"github.com/harness/gitness/app/auth/authz"
	userevents "github.com/harness/gitness/app/events/user"
	"github.com/harness/gitness/app/services/refcache"
	"github.com/harness/gitness/app/store"
	"github.com/harness/gitness/store/database/dbtx"
	"github.com/harness/gitness/types"
	"github.com/harness/gitness/types/check"
	"github.com/harness/gitness/types/enum"

	"golang.org/x/crypto/bcrypt"
)

type Controller struct {
	tx                      dbtx.Transactor
	principalUIDCheck       check.PrincipalUID
	authorizer              authz.Authorizer
	principalStore          store.PrincipalStore
	tokenStore              store.TokenStore
	membershipStore         store.MembershipStore
	publicKeyStore          store.PublicKeyStore
	publicKeySubKeyStore    store.PublicKeySubKeyStore
	gitSignatureResultStore store.GitSignatureResultStore
	eventReporter           *userevents.Reporter
	repoFinder              refcache.RepoFinder
	favoriteStore           store.FavoriteStore
}

func NewController(
	tx dbtx.Transactor,
	principalUIDCheck check.PrincipalUID,
	authorizer authz.Authorizer,
	principalStore store.PrincipalStore,
	tokenStore store.TokenStore,
	membershipStore store.MembershipStore,
	publicKeyStore store.PublicKeyStore,
	publicKeySubKeyStore store.PublicKeySubKeyStore,
	gitSignatureResultStore store.GitSignatureResultStore,
	eventReporter *userevents.Reporter,
	repoFinder refcache.RepoFinder,
	favoriteStore store.FavoriteStore,
) *Controller {
	return &Controller{
		tx:                      tx,
		principalUIDCheck:       principalUIDCheck,
		authorizer:              authorizer,
		principalStore:          principalStore,
		tokenStore:              tokenStore,
		membershipStore:         membershipStore,
		publicKeyStore:          publicKeyStore,
		publicKeySubKeyStore:    publicKeySubKeyStore,
		gitSignatureResultStore: gitSignatureResultStore,
		eventReporter:           eventReporter,
		repoFinder:              repoFinder,
		favoriteStore:           favoriteStore,
	}
}

var hashPassword = bcrypt.GenerateFromPassword

func findUserFromUID(ctx context.Context,
	principalStore store.PrincipalStore, userUID string,
) (*types.User, error) {
	return principalStore.FindUserByUID(ctx, userUID)
}

func findUserFromEmail(ctx context.Context,
	principalStore store.PrincipalStore, email string,
) (*types.User, error) {
	return principalStore.FindUserByEmail(ctx, email)
}

func isUserTokenType(tokenType enum.TokenType) bool {
	return tokenType == enum.TokenTypePAT || tokenType == enum.TokenTypeSession
}