package identity import ( "fmt" "strings" "github.com/MichaelMure/git-bug/repository" "github.com/pkg/errors" ) // Fetch retrieve updates from a remote // This does not change the local identities state func Fetch(repo repository.Repo, remote string) (string, error) { remoteRefSpec := fmt.Sprintf(identityRemoteRefPattern, remote) fetchRefSpec := fmt.Sprintf("%s*:%s*", identityRefPattern, remoteRefSpec) return repo.FetchRefs(remote, fetchRefSpec) } // Push update a remote with the local changes func Push(repo repository.Repo, remote string) (string, error) { return repo.PushRefs(remote, identityRefPattern+"*") } // Pull will do a Fetch + MergeAll // This function will return an error if a merge fail func Pull(repo repository.ClockedRepo, remote string) error { _, err := Fetch(repo, remote) if err != nil { return err } for merge := range MergeAll(repo, remote) { if merge.Err != nil { return merge.Err } if merge.Status == MergeStatusInvalid { return errors.Errorf("merge failure: %s", merge.Reason) } } return nil } // MergeAll will merge all the available remote identity func MergeAll(repo repository.ClockedRepo, remote string) <-chan MergeResult { out := make(chan MergeResult) go func() { defer close(out) remoteRefSpec := fmt.Sprintf(identityRemoteRefPattern, remote) remoteRefs, err := repo.ListRefs(remoteRefSpec) if err != nil { out <- MergeResult{Err: err} return } for _, remoteRef := range remoteRefs { refSplitted := strings.Split(remoteRef, "/") id := refSplitted[len(refSplitted)-1] remoteIdentity, err := read(repo, remoteRef) if err != nil { out <- newMergeInvalidStatus(id, errors.Wrap(err, "remote identity is not readable").Error()) continue } // Check for error in remote data if err := remoteIdentity.Validate(); err != nil { out <- newMergeInvalidStatus(id, errors.Wrap(err, "remote identity is invalid").Error()) continue } localRef := identityRefPattern + remoteIdentity.Id() localExist, err := repo.RefExist(localRef) if err != nil { out <- newMergeError(err, id) continue } // the identity is not local yet, simply create the reference if !localExist { err := repo.CopyRef(remoteRef, localRef) if err != nil { out <- newMergeError(err, id) return } out <- newMergeStatus(MergeStatusNew, id, remoteIdentity) continue } localIdentity, err := read(repo, localRef) if err != nil { out <- newMergeError(errors.Wrap(err, "local identity is not readable"), id) return } updated, err := localIdentity.Merge(repo, remoteIdentity) if err != nil { out <- newMergeInvalidStatus(id, errors.Wrap(err, "merge failed").Error()) return } if updated { out <- newMergeStatus(MergeStatusUpdated, id, localIdentity) } else { out <- newMergeStatus(MergeStatusNothing, id, localIdentity) } } }() return out } // MergeStatus represent the result of a merge operation of a bug type MergeStatus int const ( _ MergeStatus = iota MergeStatusNew MergeStatusInvalid MergeStatusUpdated MergeStatusNothing ) // Todo: share a generalized MergeResult with the bug package ? type MergeResult struct { // Err is set when a terminal error occur in the process Err error Id string Status MergeStatus // Only set for invalid status Reason string // Not set for invalid status Identity *Identity } func (mr MergeResult) String() string { switch mr.Status { case MergeStatusNew: return "new" case MergeStatusInvalid: return fmt.Sprintf("invalid data: %s", mr.Reason) case MergeStatusUpdated: return "updated" case MergeStatusNothing: return "nothing to do" default: panic("unknown merge status") } } func newMergeError(err error, id string) MergeResult { return MergeResult{ Err: err, Id: id, } } func newMergeStatus(status MergeStatus, id string, identity *Identity) MergeResult { return MergeResult{ Id: id, Status: status, // Identity is not set for an invalid merge result Identity: identity, } } func newMergeInvalidStatus(id string, reason string) MergeResult { return MergeResult{ Id: id, Status: MergeStatusInvalid, Reason: reason, } }