package launchpad import ( "context" "fmt" "time" "github.com/MichaelMure/git-bug/bridge/core" "github.com/MichaelMure/git-bug/bug" "github.com/MichaelMure/git-bug/cache" "github.com/MichaelMure/git-bug/entity" ) type launchpadImporter struct { conf core.Configuration } func (li *launchpadImporter) Init(repo *cache.RepoCache, conf core.Configuration) error { li.conf = conf return nil } func (li *launchpadImporter) ensurePerson(repo *cache.RepoCache, owner LPPerson) (*cache.IdentityCache, error) { // Look first in the cache i, err := repo.ResolveIdentityImmutableMetadata(metaKeyLaunchpadLogin, owner.Login) if err == nil { return i, nil } if entity.IsErrMultipleMatch(err) { return nil, err } return repo.NewIdentityRaw( owner.Name, "", "", map[string]string{ metaKeyLaunchpadLogin: owner.Login, }, ) } func (li *launchpadImporter) ImportAll(ctx context.Context, repo *cache.RepoCache, since time.Time) (<-chan core.ImportResult, error) { out := make(chan core.ImportResult) lpAPI := new(launchpadAPI) err := lpAPI.Init() if err != nil { return nil, err } lpBugs, err := lpAPI.SearchTasks(ctx, li.conf["project"]) if err != nil { return nil, err } go func() { for _, lpBug := range lpBugs { select { case <-ctx.Done(): return default: lpBugID := fmt.Sprintf("%d", lpBug.ID) b, err := repo.ResolveBugCreateMetadata(metaKeyLaunchpadID, lpBugID) if err != nil && err != bug.ErrBugNotExist { out <- core.NewImportError(err, entity.Id(lpBugID)) return } owner, err := li.ensurePerson(repo, lpBug.Owner) if err != nil { out <- core.NewImportError(err, entity.Id(lpBugID)) return } if err == bug.ErrBugNotExist { createdAt, _ := time.Parse(time.RFC3339, lpBug.CreatedAt) b, _, err = repo.NewBugRaw( owner, createdAt.Unix(), lpBug.Title, lpBug.Description, nil, map[string]string{ core.MetaKeyOrigin: target, metaKeyLaunchpadID: lpBugID, }, ) if err != nil { out <- core.NewImportError(err, entity.Id(lpBugID)) return } out <- core.NewImportBug(b.Id()) } /* Handle messages */ if len(lpBug.Messages) == 0 { return } // The Launchpad API returns the bug description as the first // comment, so skip it. for _, lpMessage := range lpBug.Messages[1:] { _, err := b.ResolveOperationWithMetadata(metaKeyLaunchpadID, lpMessage.ID) if err != nil && err != cache.ErrNoMatchingOp { out <- core.NewImportError(err, entity.Id(lpMessage.ID)) return } // If this comment already exists, we are probably // updating an existing bug. We do not want to duplicate // the comments, so let us just skip this one. // TODO: Can Launchpad comments be edited? if err == nil { continue } owner, err := li.ensurePerson(repo, lpMessage.Owner) if err != nil { out <- core.NewImportError(err, "") return } // This is a new comment, we can add it. createdAt, _ := time.Parse(time.RFC3339, lpMessage.CreatedAt) op, err := b.AddCommentRaw( owner, createdAt.Unix(), lpMessage.Content, nil, map[string]string{ metaKeyLaunchpadID: lpMessage.ID, }) if err != nil { out <- core.NewImportError(err, op.Id()) return } out <- core.NewImportComment(op.Id()) } if !b.NeedCommit() { out <- core.NewImportNothing(b.Id(), "no imported operation") } else if err := b.Commit(); err != nil { out <- core.NewImportError(err, "") return } } } }() return out, nil }