Migrate legacyAuthor

This commit enables users to run the main file in the directory of a git repository with git-bug and migrates any legacyAuthors found to the new identity.Identity type. This allows for users to update to the latest version of git-bug, bridging over the breaking changes.

Author: vincetiu8

main.go

@@ -4,36 +4,110 @@ import (
 	"fmt"
 	"os"
 
-	"github.com/MichaelMure/git-bug-migration/migration1/bug"
-	"github.com/MichaelMure/git-bug-migration/migration1/repository"
+	mg1b "github.com/MichaelMure/git-bug-migration/migration1/bug"
+	mg1i "github.com/MichaelMure/git-bug-migration/migration1/identity"
+	mg1r "github.com/MichaelMure/git-bug-migration/migration1/repository"
 )
 
 const rootCommandName = "git-bug-migration"
 
-var repo repository.ClockedRepo
+var repo mg1r.ClockedRepo
 
 func main() {
 	cwd, err := os.Getwd()
 	if err != nil {
 		panic(fmt.Errorf("unable to get the current working directory: %q", err))
 	}
 
-	repo, err = repository.NewGitRepo(cwd, []repository.ClockLoader{bug.ClockLoader})
-	if err == repository.ErrNotARepo {
+	repo, err = mg1r.NewGitRepo(cwd, []mg1r.ClockLoader{mg1b.ClockLoader})
+	if err == mg1r.ErrNotARepo {
 		panic(fmt.Errorf("%s must be run from within a git repo", rootCommandName))
 	}
 
+	identities := []*mg1i.Identity{}
+	for streamedIdentity := range mg1i.ReadAllLocalIdentities(repo) {
+		if streamedIdentity.Err != nil {
+			fmt.Print(fmt.Errorf("Got error when reading identity: %q\n", err))
+			continue
+		}
+		identities = append(identities, streamedIdentity.Identity)
+	}
+
 	if err != nil {
 		panic(err)
 	}
 
-	for streamedBug := range bug.ReadAllLocalBugs(repo) {
+	// Iterating through all the bugs in the repo
+	for streamedBug := range mg1b.ReadAllLocalBugs(repo) {
 		if streamedBug.Err != nil {
-			fmt.Print(fmt.Errorf("Got error when reading bug: %q", err))
+			fmt.Print(fmt.Errorf("Got error when reading bug: %q\n", err))
 			continue
 		}
 
-		//switch streamedBug.Bug
-		fmt.Print(streamedBug.Bug.FirstOp().GetAuthor())
+		// Getting the old bug
+		oldBug := streamedBug.Bug
+
+		b := mg1b.NewBug()
+		bugChange := false
+
+		// Iterating over each operation in the bug
+		it := mg1b.NewOperationIterator(oldBug)
+		for it.Next() {
+			operation := it.Value()
+			oldAuthor := operation.GetAuthor()
+
+			// Checking if the author is of the legacy (bare) type
+			switch oldAuthor.(type) {
+			case *mg1i.Bare:
+				fmt.Print("Detected legacyAuthor!\n")
+
+				bugChange = true
+
+				// Search existing identities for any traces of this old identity
+				var newAuthor *mg1i.Identity = nil
+				for _, identity := range identities {
+					if oldAuthor.Name() == identity.Name() {
+						newAuthor = identity
+					}
+				}
+
+				// If no existing identity is found, create a new one
+				if newAuthor == nil {
+					newAuthor = mg1i.NewIdentityFull(
+						oldAuthor.Name(),
+						oldAuthor.Email(),
+						oldAuthor.Login(),
+						oldAuthor.AvatarUrl(),
+					)
+				}
+
+				// Set the author of the operation to the new identity
+				operation.SetAuthor(newAuthor)
+				b.Append(operation)
+				continue
+
+			// If the author's identity is a new identity type, its fine. Just append it to the cache
+			case *mg1i.Identity:
+				b.Append(operation)
+				continue
+
+			// This should not be reached
+			default:
+				fmt.Printf("Unknown author type: %T\n", operation.GetAuthor())
+			}
+		}
+
+		// If the bug has been changed, remove the old bug and commit the new one
+		if bugChange {
+			err = b.Commit(repo)
+			if err != nil {
+				fmt.Printf("Got error when attempting to commit new bug: %q\n", err)
+			}
+
+			err = mg1b.RemoveLocalBug(repo, oldBug.Id())
+			if err != nil {
+				fmt.Printf("Got error when attempting to remove bug: %q\n", err)
+			}
+		}
 	}
 }

migration1/bug/bug.go

@@ -243,6 +243,11 @@ func readBug(repo repository.ClockedRepo, ref string) (*Bug, error) {
 	return &bug, nil
 }
 
+func RemoveLocalBug(repo repository.ClockedRepo, id entity.Id) error {
+	ref := bugsRefPattern + id.String()
+	return repo.RemoveRef(ref)
+}
+
 type StreamedBug struct {
 	Bug *Bug
 	Err error

migration1/bug/operation.go

@@ -50,6 +50,8 @@ type Operation interface {
 	AllMetadata() map[string]string
 	// GetAuthor return the author identity
 	GetAuthor() identity.Interface
+	// GetAuthor sets the author identity
+	SetAuthor(identity.Interface)
 
 	// sign-post method for gqlgen
 	IsOperation()
@@ -217,3 +219,8 @@ func (op *OpBase) AllMetadata() map[string]string {
 func (op *OpBase) GetAuthor() identity.Interface {
 	return op.Author
 }
+
+// SetAuthor return author identity
+func (op *OpBase) SetAuthor(author identity.Interface) {
+	op.Author = author
+}

migration1/repository/config_testing.go

@@ -283,6 +283,13 @@ func (repo *GitRepo) UpdateRef(ref string, hash git.Hash) error {
 	return err
 }
 
+// RemoveRef will remove a Git reference
+func (repo *GitRepo) RemoveRef(ref string) error {
+	_, err := repo.runGitCommand("update-ref", "-d", ref)
+
+	return err
+}
+
 // ListRefs will return a list of Git ref matching the given refspec
 func (repo *GitRepo) ListRefs(refspec string) ([]string, error) {
 	stdout, err := repo.runGitCommand("for-each-ref", "--format=%(refname)", refspec)