drone/internal/api/controller/repo/update.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"

	apiauth "github.com/harness/gitness/internal/api/auth"
	"github.com/harness/gitness/internal/auth"
	"github.com/harness/gitness/types"
	"github.com/harness/gitness/types/enum"
)

// UpdateInput is used for updating a repo.
type UpdateInput struct {
	Description *string `json:"description"`
	IsPublic    *bool   `json:"is_public"`
}

// Update updates a repository.
func (c *Controller) Update(ctx context.Context, session *auth.Session,
	repoRef string, in *UpdateInput) (*types.Repository, error) {
	repo, err := c.repoStore.FindRepoFromRef(ctx, repoRef)
	if err != nil {
		return nil, err
	}

	if err = apiauth.CheckRepo(ctx, c.authorizer, session, repo, enum.PermissionRepoEdit, false); err != nil {
		return nil, err
	}

	// update values only if provided
	if in.Description != nil {
		repo.Description = *in.Description
	}
	if in.IsPublic != nil {
		repo.IsPublic = *in.IsPublic
	}

	// ensure provided values are valid
	if err = c.repoCheck(repo); err != nil {
		return nil, err
	}

	err = c.repoStore.Update(ctx, repo)
	if err != nil {
		return nil, err
	}

	// populate repo url
	repo.GitURL, err = GenerateRepoGitURL(c.gitBaseURL, repo.Path)
	if err != nil {
		return nil, err
	}

	return repo, nil
}