gogs/models/update.go

// Copyright 2014 The Gogs 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 models

import (
	"container/list"
	"fmt"
	"os/exec"
	"strings"

	git "github.com/gogits/git-module"
)

// CommitToPushCommit transforms a git.Commit to PushCommit type.
func CommitToPushCommit(commit *git.Commit) *PushCommit {
	return &PushCommit{
		Sha1:           commit.ID.String(),
		Message:        commit.Message(),
		AuthorEmail:    commit.Author.Email,
		AuthorName:     commit.Author.Name,
		CommitterEmail: commit.Committer.Email,
		CommitterName:  commit.Committer.Name,
		Timestamp:      commit.Committer.When,
	}
}

func ListToPushCommits(l *list.List) *PushCommits {
	if l == nil {
		return &PushCommits{}
	}

	commits := make([]*PushCommit, 0)
	var actEmail string
	for e := l.Front(); e != nil; e = e.Next() {
		commit := e.Value.(*git.Commit)
		if actEmail == "" {
			actEmail = commit.Committer.Email
		}
		commits = append(commits, CommitToPushCommit(commit))
	}
	return &PushCommits{l.Len(), commits, "", nil}
}

type PushUpdateOptions struct {
	OldCommitID  string
	NewCommitID  string
	RefFullName  string
	PusherID     int64
	PusherName   string
	RepoUserName string
	RepoName     string
}

// PushUpdate must be called for any push actions in order to
// generates necessary push action history feeds.
func PushUpdate(opts PushUpdateOptions) (err error) {
	isNewRef := opts.OldCommitID == git.EMPTY_SHA
	isDelRef := opts.NewCommitID == git.EMPTY_SHA
	if isNewRef && isDelRef {
		return fmt.Errorf("Old and new revisions are both %s", git.EMPTY_SHA)
	}

	repoPath := RepoPath(opts.RepoUserName, opts.RepoName)

	gitUpdate := exec.Command("git", "update-server-info")
	gitUpdate.Dir = repoPath
	if err = gitUpdate.Run(); err != nil {
		return fmt.Errorf("Fail to call 'git update-server-info': %v", err)
	}

	gitRepo, err := git.OpenRepository(repoPath)
	if err != nil {
		return fmt.Errorf("OpenRepository: %v", err)
	}

	owner, err := GetUserByName(opts.RepoUserName)
	if err != nil {
		return fmt.Errorf("GetUserByName: %v", err)
	}

	repo, err := GetRepositoryByName(owner.ID, opts.RepoName)
	if err != nil {
		return fmt.Errorf("GetRepositoryByName: %v", err)
	}

	if err = repo.UpdateSize(); err != nil {
		return fmt.Errorf("UpdateSize: %v", err)
	}

	// Push tags
	if strings.HasPrefix(opts.RefFullName, git.TAG_PREFIX) {
		if err := CommitRepoAction(CommitRepoActionOptions{
			PusherName:  opts.PusherName,
			RepoOwnerID: owner.ID,
			RepoName:    repo.Name,
			RefFullName: opts.RefFullName,
			OldCommitID: opts.OldCommitID,
			NewCommitID: opts.NewCommitID,
			Commits:     &PushCommits{},
		}); err != nil {
			return fmt.Errorf("CommitRepoAction.(tag): %v", err)
		}
		return nil
	}

	var l *list.List
	// Skip read parent commits when delete branch
	if !isDelRef {
		var oldCommit *git.Commit

		// Push new branch
		newCommit, err := gitRepo.GetCommit(opts.NewCommitID)
		if err != nil {
			return fmt.Errorf("GetCommit [commit_id: %s]: %v", opts.NewCommitID, err)
		}

		if isNewRef {
			l, err = newCommit.CommitsBefore()
			if err != nil {
				return fmt.Errorf("CommitsBefore [commit_id: %s]: %v", newCommit.ID, err)
			}
		} else {
			oldCommit, err = gitRepo.GetCommit(opts.OldCommitID)
			if err != nil {
				return fmt.Errorf("GetCommit [commit_id: %s]: %v", opts.OldCommitID, err)
			}
			l, err = gitRepo.CommitsBefore(newCommit, oldCommit)
			if err != nil {
				return fmt.Errorf("CommitsBetween [before: %s, until: %s]: %v", opts.NewCommitID, opts.OldCommitID, err)
			}
		}

		searchCommits := make([]*Commit, 0)
		for e := l.Front(); e != nil; e = e.Next() {
			commit := e.Value.(*git.Commit)
			searchCommits = append(searchCommits, GitCommitToSearchCommit(repo.ID, commit))
		}
		x.Insert(searchCommits)

		if !isNewRef {
			// Remove deleted commits if any
			var d *list.List
			d, err = gitRepo.CommitsBefore(oldCommit, newCommit)
			if err != nil {
				return fmt.Errorf("CommitsBetween [before: %s, until: %s]: %v", opts.OldCommitID, opts.NewCommitID, err)
			}
			rebasedCommits := make([]string, 0)
			for e := d.Front(); e != nil; e = e.Next() {
				commit := e.Value.(*git.Commit)
				rebasedCommits = append(rebasedCommits, commit.ID.String())
			}
			if len(rebasedCommits) > 0 {
				d, err = gitRepo.CommitsUnreachable(rebasedCommits)
				if err != nil {
					return fmt.Errorf("CommitsUnreachable: %v", err)
				}
				rebasedCommits = make([]string, 0)
				for e := d.Front(); e != nil; e = e.Next() {
					commit := e.Value.(*git.Commit)
					rebasedCommits = append(rebasedCommits, commit.ID.String())
				}
				if len(rebasedCommits) > 0 {
					x.In("sha", rebasedCommits).Delete(new(Commit))
				}
			}
		}
	}

	if err := CommitRepoAction(CommitRepoActionOptions{
		PusherName:  opts.PusherName,
		RepoOwnerID: owner.ID,
		RepoName:    repo.Name,
		RefFullName: opts.RefFullName,
		OldCommitID: opts.OldCommitID,
		NewCommitID: opts.NewCommitID,
		Commits:     ListToPushCommits(l),
	}); err != nil {
		return fmt.Errorf("CommitRepoAction.(branch): %v", err)
	}
	return nil
}

func ReindexCommits(RepoUserName string, RepoName string) (err error) {
	repoPath := RepoPath(RepoUserName, RepoName)

	gitRepo, err := git.OpenRepository(repoPath)
	if err != nil {
		return fmt.Errorf("OpenRepository: %v", err)
	}

	owner, err := GetUserByName(RepoUserName)
	if err != nil {
		return fmt.Errorf("GetUserByName: %v", err)
	}

	repo, err := GetRepositoryByName(owner.ID, RepoName)
	if err != nil {
		return fmt.Errorf("GetRepositoryByName: %v", err)
	}

	x.Delete(&Commit{RepoID: repo.ID})

	for page := 0; ; page++ {
		commits, err := gitRepo.CommitsByRangeSize("--all", page, 1000)
		if err != nil {
			return fmt.Errorf("CommitsByRange: %v", err)
		}
		searchCommits := make([]*Commit, 0)
		for e := commits.Front(); e != nil; e = e.Next() {
			commit := e.Value.(*git.Commit)
			searchCommits = append(searchCommits, GitCommitToSearchCommit(repo.ID, commit))
		}
		x.Insert(searchCommits)
		if len(searchCommits) < 1000 {
			break
		}
	}

	return nil
}