|
5 | 5 | package cmd
|
6 | 6 |
|
7 | 7 | import (
|
8 |
| - "context" |
9 |
| - "strings" |
| 8 | + "errors" |
| 9 | + "net/http" |
10 | 10 |
|
11 | 11 | "code.gitea.io/gitea/modules/log"
|
12 |
| - "code.gitea.io/gitea/modules/migrations" |
13 |
| - "code.gitea.io/gitea/modules/migrations/base" |
| 12 | + "code.gitea.io/gitea/modules/private" |
14 | 13 | "code.gitea.io/gitea/modules/setting"
|
15 |
| - "code.gitea.io/gitea/modules/storage" |
16 |
| - pull_service "code.gitea.io/gitea/services/pull" |
17 | 14 |
|
18 | 15 | "github.com/urfave/cli"
|
19 | 16 | )
|
@@ -50,70 +47,18 @@ wiki, issues, labels, releases, release_assets, milestones, pull_requests, comme
|
50 | 47 | }
|
51 | 48 |
|
52 | 49 | func runRestoreRepository(ctx *cli.Context) error {
|
53 |
| - if err := initDB(); err != nil { |
54 |
| - return err |
55 |
| - } |
56 |
| - |
57 |
| - log.Trace("AppPath: %s", setting.AppPath) |
58 |
| - log.Trace("AppWorkPath: %s", setting.AppWorkPath) |
59 |
| - log.Trace("Custom path: %s", setting.CustomPath) |
60 |
| - log.Trace("Log path: %s", setting.LogRootPath) |
61 |
| - setting.InitDBConfig() |
62 |
| - |
63 |
| - if err := storage.Init(); err != nil { |
64 |
| - return err |
65 |
| - } |
66 |
| - |
67 |
| - if err := pull_service.Init(); err != nil { |
68 |
| - return err |
69 |
| - } |
70 |
| - |
71 |
| - var opts = base.MigrateOptions{ |
72 |
| - RepoName: ctx.String("repo_name"), |
73 |
| - } |
74 |
| - |
75 |
| - if len(ctx.String("units")) == 0 { |
76 |
| - opts.Wiki = true |
77 |
| - opts.Issues = true |
78 |
| - opts.Milestones = true |
79 |
| - opts.Labels = true |
80 |
| - opts.Releases = true |
81 |
| - opts.Comments = true |
82 |
| - opts.PullRequests = true |
83 |
| - opts.ReleaseAssets = true |
84 |
| - } else { |
85 |
| - units := strings.Split(ctx.String("units"), ",") |
86 |
| - for _, unit := range units { |
87 |
| - switch strings.ToLower(unit) { |
88 |
| - case "wiki": |
89 |
| - opts.Wiki = true |
90 |
| - case "issues": |
91 |
| - opts.Issues = true |
92 |
| - case "milestones": |
93 |
| - opts.Milestones = true |
94 |
| - case "labels": |
95 |
| - opts.Labels = true |
96 |
| - case "releases": |
97 |
| - opts.Releases = true |
98 |
| - case "release_assets": |
99 |
| - opts.ReleaseAssets = true |
100 |
| - case "comments": |
101 |
| - opts.Comments = true |
102 |
| - case "pull_requests": |
103 |
| - opts.PullRequests = true |
104 |
| - } |
105 |
| - } |
106 |
| - } |
| 50 | + setting.NewContext() |
107 | 51 |
|
108 |
| - if err := migrations.RestoreRepository( |
109 |
| - context.Background(), |
| 52 | + statusCode, errStr := private.RestoreRepo( |
110 | 53 | ctx.String("repo_dir"),
|
111 | 54 | ctx.String("owner_name"),
|
112 | 55 | ctx.String("repo_name"),
|
113 |
| - ); err != nil { |
114 |
| - log.Fatal("Failed to restore repository: %v", err) |
115 |
| - return err |
| 56 | + ctx.StringSlice("units"), |
| 57 | + ) |
| 58 | + if statusCode == http.StatusOK { |
| 59 | + return nil |
116 | 60 | }
|
117 | 61 |
|
118 |
| - return nil |
| 62 | + log.Fatal("Failed to restore repository: %v", errStr) |
| 63 | + return errors.New(errStr) |
119 | 64 | }
|
0 commit comments