123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277 |
- // Copyright 2020 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 migrations
-
- import (
- "context"
- "fmt"
- "os"
- "path/filepath"
- "strconv"
-
- base "code.gitea.io/gitea/modules/migration"
-
- "gopkg.in/yaml.v2"
- )
-
- // RepositoryRestorer implements an Downloader from the local directory
- type RepositoryRestorer struct {
- base.NullDownloader
- ctx context.Context
- baseDir string
- repoOwner string
- repoName string
- validation bool
- }
-
- // NewRepositoryRestorer creates a repository restorer which could restore repository from a dumped folder
- func NewRepositoryRestorer(ctx context.Context, baseDir, owner, repoName string, validation bool) (*RepositoryRestorer, error) {
- baseDir, err := filepath.Abs(baseDir)
- if err != nil {
- return nil, err
- }
- return &RepositoryRestorer{
- ctx: ctx,
- baseDir: baseDir,
- repoOwner: owner,
- repoName: repoName,
- validation: validation,
- }, nil
- }
-
- func (r *RepositoryRestorer) commentDir() string {
- return filepath.Join(r.baseDir, "comments")
- }
-
- func (r *RepositoryRestorer) reviewDir() string {
- return filepath.Join(r.baseDir, "reviews")
- }
-
- // SetContext set context
- func (r *RepositoryRestorer) SetContext(ctx context.Context) {
- r.ctx = ctx
- }
-
- func (r *RepositoryRestorer) getRepoOptions() (map[string]string, error) {
- p := filepath.Join(r.baseDir, "repo.yml")
- bs, err := os.ReadFile(p)
- if err != nil {
- return nil, err
- }
-
- opts := make(map[string]string)
- err = yaml.Unmarshal(bs, &opts)
- if err != nil {
- return nil, err
- }
- return opts, nil
- }
-
- // GetRepoInfo returns a repository information
- func (r *RepositoryRestorer) GetRepoInfo() (*base.Repository, error) {
- opts, err := r.getRepoOptions()
- if err != nil {
- return nil, err
- }
-
- isPrivate, _ := strconv.ParseBool(opts["is_private"])
-
- return &base.Repository{
- Owner: r.repoOwner,
- Name: r.repoName,
- IsPrivate: isPrivate,
- Description: opts["description"],
- OriginalURL: opts["original_url"],
- CloneURL: filepath.Join(r.baseDir, "git"),
- DefaultBranch: opts["default_branch"],
- }, nil
- }
-
- // GetTopics return github topics
- func (r *RepositoryRestorer) GetTopics() ([]string, error) {
- p := filepath.Join(r.baseDir, "topic.yml")
-
- topics := struct {
- Topics []string `yaml:"topics"`
- }{}
-
- bs, err := os.ReadFile(p)
- if err != nil {
- if os.IsNotExist(err) {
- return nil, nil
- }
- return nil, err
- }
-
- err = yaml.Unmarshal(bs, &topics)
- if err != nil {
- return nil, err
- }
- return topics.Topics, nil
- }
-
- // GetMilestones returns milestones
- func (r *RepositoryRestorer) GetMilestones() ([]*base.Milestone, error) {
- milestones := make([]*base.Milestone, 0, 10)
- p := filepath.Join(r.baseDir, "milestone.yml")
- err := base.Load(p, &milestones, r.validation)
- if err != nil {
- if os.IsNotExist(err) {
- return nil, nil
- }
- return nil, err
- }
-
- return milestones, nil
- }
-
- // GetReleases returns releases
- func (r *RepositoryRestorer) GetReleases() ([]*base.Release, error) {
- releases := make([]*base.Release, 0, 10)
- p := filepath.Join(r.baseDir, "release.yml")
- _, err := os.Stat(p)
- if err != nil {
- if os.IsNotExist(err) {
- return nil, nil
- }
- return nil, err
- }
-
- bs, err := os.ReadFile(p)
- if err != nil {
- return nil, err
- }
-
- err = yaml.Unmarshal(bs, &releases)
- if err != nil {
- return nil, err
- }
- for _, rel := range releases {
- for _, asset := range rel.Assets {
- if asset.DownloadURL != nil {
- *asset.DownloadURL = "file://" + filepath.Join(r.baseDir, *asset.DownloadURL)
- }
- }
- }
- return releases, nil
- }
-
- // GetLabels returns labels
- func (r *RepositoryRestorer) GetLabels() ([]*base.Label, error) {
- labels := make([]*base.Label, 0, 10)
- p := filepath.Join(r.baseDir, "label.yml")
- _, err := os.Stat(p)
- if err != nil {
- if os.IsNotExist(err) {
- return nil, nil
- }
- return nil, err
- }
-
- bs, err := os.ReadFile(p)
- if err != nil {
- return nil, err
- }
-
- err = yaml.Unmarshal(bs, &labels)
- if err != nil {
- return nil, err
- }
- return labels, nil
- }
-
- // GetIssues returns issues according start and limit
- func (r *RepositoryRestorer) GetIssues(page, perPage int) ([]*base.Issue, bool, error) {
- issues := make([]*base.Issue, 0, 10)
- p := filepath.Join(r.baseDir, "issue.yml")
- err := base.Load(p, &issues, r.validation)
- if err != nil {
- if os.IsNotExist(err) {
- return nil, true, nil
- }
- return nil, false, err
- }
-
- for _, issue := range issues {
- issue.Context = base.BasicIssueContext(issue.Number)
- }
- return issues, true, nil
- }
-
- // GetComments returns comments according issueNumber
- func (r *RepositoryRestorer) GetComments(opts base.GetCommentOptions) ([]*base.Comment, bool, error) {
- comments := make([]*base.Comment, 0, 10)
- p := filepath.Join(r.commentDir(), fmt.Sprintf("%d.yml", opts.Context.ForeignID()))
- _, err := os.Stat(p)
- if err != nil {
- if os.IsNotExist(err) {
- return nil, false, nil
- }
- return nil, false, err
- }
-
- bs, err := os.ReadFile(p)
- if err != nil {
- return nil, false, err
- }
-
- err = yaml.Unmarshal(bs, &comments)
- if err != nil {
- return nil, false, err
- }
- return comments, false, nil
- }
-
- // GetPullRequests returns pull requests according page and perPage
- func (r *RepositoryRestorer) GetPullRequests(page, perPage int) ([]*base.PullRequest, bool, error) {
- pulls := make([]*base.PullRequest, 0, 10)
- p := filepath.Join(r.baseDir, "pull_request.yml")
- _, err := os.Stat(p)
- if err != nil {
- if os.IsNotExist(err) {
- return nil, true, nil
- }
- return nil, false, err
- }
-
- bs, err := os.ReadFile(p)
- if err != nil {
- return nil, false, err
- }
-
- err = yaml.Unmarshal(bs, &pulls)
- if err != nil {
- return nil, false, err
- }
- for _, pr := range pulls {
- pr.PatchURL = "file://" + filepath.Join(r.baseDir, pr.PatchURL)
- pr.Context = base.BasicIssueContext(pr.Number)
- }
- return pulls, true, nil
- }
-
- // GetReviews returns pull requests review
- func (r *RepositoryRestorer) GetReviews(context base.IssueContext) ([]*base.Review, error) {
- reviews := make([]*base.Review, 0, 10)
- p := filepath.Join(r.reviewDir(), fmt.Sprintf("%d.yml", context.ForeignID()))
- _, err := os.Stat(p)
- if err != nil {
- if os.IsNotExist(err) {
- return nil, nil
- }
- return nil, err
- }
-
- bs, err := os.ReadFile(p)
- if err != nil {
- return nil, err
- }
-
- err = yaml.Unmarshal(bs, &reviews)
- if err != nil {
- return nil, err
- }
- return reviews, nil
- }
|