mirror of
https://github.com/go-gitea/gitea.git
synced 2024-08-23 12:50:49 +02:00
160 lines
4.0 KiB
Go
160 lines
4.0 KiB
Go
|
// Copyright 2019 The Gitea Authors. All rights reserved.
|
||
|
// Use of this source code is governed by a MIT-style
|
||
|
// license that can be found in the LICENSE file.
|
||
|
|
||
|
package uploader
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"strings"
|
||
|
|
||
|
"code.gitea.io/git"
|
||
|
"code.gitea.io/gitea/models"
|
||
|
"code.gitea.io/gitea/modules/lfs"
|
||
|
"code.gitea.io/gitea/modules/setting"
|
||
|
)
|
||
|
|
||
|
// UpdateRepoFileOptions holds the repository file update options
|
||
|
type UpdateRepoFileOptions struct {
|
||
|
LastCommitID string
|
||
|
OldBranch string
|
||
|
NewBranch string
|
||
|
OldTreeName string
|
||
|
NewTreeName string
|
||
|
Message string
|
||
|
Content string
|
||
|
IsNewFile bool
|
||
|
}
|
||
|
|
||
|
// UpdateRepoFile adds or updates a file in the given repository
|
||
|
func UpdateRepoFile(repo *models.Repository, doer *models.User, opts *UpdateRepoFileOptions) error {
|
||
|
t, err := NewTemporaryUploadRepository(repo)
|
||
|
defer t.Close()
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if err := t.Clone(opts.OldBranch); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if err := t.SetDefaultIndex(); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
filesInIndex, err := t.LsFiles(opts.NewTreeName, opts.OldTreeName)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("UpdateRepoFile: %v", err)
|
||
|
}
|
||
|
|
||
|
if opts.IsNewFile {
|
||
|
for _, file := range filesInIndex {
|
||
|
if file == opts.NewTreeName {
|
||
|
return models.ErrRepoFileAlreadyExist{FileName: opts.NewTreeName}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//var stdout string
|
||
|
if opts.OldTreeName != opts.NewTreeName && len(filesInIndex) > 0 {
|
||
|
for _, file := range filesInIndex {
|
||
|
if file == opts.OldTreeName {
|
||
|
if err := t.RemoveFilesFromIndex(opts.OldTreeName); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
// Check there is no way this can return multiple infos
|
||
|
filename2attribute2info, err := t.CheckAttribute("filter", opts.NewTreeName)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
content := opts.Content
|
||
|
var lfsMetaObject *models.LFSMetaObject
|
||
|
|
||
|
if filename2attribute2info[opts.NewTreeName] != nil && filename2attribute2info[opts.NewTreeName]["filter"] == "lfs" {
|
||
|
// OK so we are supposed to LFS this data!
|
||
|
oid, err := models.GenerateLFSOid(strings.NewReader(opts.Content))
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
lfsMetaObject = &models.LFSMetaObject{Oid: oid, Size: int64(len(opts.Content)), RepositoryID: repo.ID}
|
||
|
content = lfsMetaObject.Pointer()
|
||
|
}
|
||
|
|
||
|
// Add the object to the database
|
||
|
objectHash, err := t.HashObject(strings.NewReader(content))
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
// Add the object to the index
|
||
|
if err := t.AddObjectToIndex("100644", objectHash, opts.NewTreeName); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
// Now write the tree
|
||
|
treeHash, err := t.WriteTree()
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
// Now commit the tree
|
||
|
commitHash, err := t.CommitTree(doer, treeHash, opts.Message)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
if lfsMetaObject != nil {
|
||
|
// We have an LFS object - create it
|
||
|
lfsMetaObject, err = models.NewLFSMetaObject(lfsMetaObject)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
contentStore := &lfs.ContentStore{BasePath: setting.LFS.ContentPath}
|
||
|
if !contentStore.Exists(lfsMetaObject) {
|
||
|
if err := contentStore.Put(lfsMetaObject, strings.NewReader(opts.Content)); err != nil {
|
||
|
if err2 := repo.RemoveLFSMetaObjectByOid(lfsMetaObject.Oid); err2 != nil {
|
||
|
return fmt.Errorf("Error whilst removing failed inserted LFS object %s: %v (Prev Error: %v)", lfsMetaObject.Oid, err2, err)
|
||
|
}
|
||
|
return err
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Then push this tree to NewBranch
|
||
|
if err := t.Push(doer, commitHash, opts.NewBranch); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
// Simulate push event.
|
||
|
oldCommitID := opts.LastCommitID
|
||
|
if opts.NewBranch != opts.OldBranch {
|
||
|
oldCommitID = git.EmptySHA
|
||
|
}
|
||
|
|
||
|
if err = repo.GetOwner(); err != nil {
|
||
|
return fmt.Errorf("GetOwner: %v", err)
|
||
|
}
|
||
|
err = models.PushUpdate(
|
||
|
opts.NewBranch,
|
||
|
models.PushUpdateOptions{
|
||
|
PusherID: doer.ID,
|
||
|
PusherName: doer.Name,
|
||
|
RepoUserName: repo.Owner.Name,
|
||
|
RepoName: repo.Name,
|
||
|
RefFullName: git.BranchPrefix + opts.NewBranch,
|
||
|
OldCommitID: oldCommitID,
|
||
|
NewCommitID: commitHash,
|
||
|
},
|
||
|
)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("PushUpdate: %v", err)
|
||
|
}
|
||
|
models.UpdateRepoIndexer(repo)
|
||
|
|
||
|
return nil
|
||
|
}
|