forked from Shiloh/githaven
e5723d6556
* Move restore repo to internal router and invoke from command to avoid open the same db file or queues files * Follow @zeripath's review * set no timeout for resotre repo private request * make restore repo cancelable
65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
// 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 cmd
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
|
|
"code.gitea.io/gitea/modules/log"
|
|
"code.gitea.io/gitea/modules/private"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
// CmdRestoreRepository represents the available restore a repository sub-command.
|
|
var CmdRestoreRepository = cli.Command{
|
|
Name: "restore-repo",
|
|
Usage: "Restore the repository from disk",
|
|
Description: "This is a command for restoring the repository data.",
|
|
Action: runRestoreRepository,
|
|
Flags: []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "repo_dir, r",
|
|
Value: "./data",
|
|
Usage: "Repository dir path to restore from",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "owner_name",
|
|
Value: "",
|
|
Usage: "Restore destination owner name",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "repo_name",
|
|
Value: "",
|
|
Usage: "Restore destination repository name",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "units",
|
|
Value: "",
|
|
Usage: `Which items will be restored, one or more units should be separated as comma.
|
|
wiki, issues, labels, releases, release_assets, milestones, pull_requests, comments are allowed. Empty means all units.`,
|
|
},
|
|
},
|
|
}
|
|
|
|
func runRestoreRepository(ctx *cli.Context) error {
|
|
setting.NewContext()
|
|
|
|
statusCode, errStr := private.RestoreRepo(
|
|
ctx.String("repo_dir"),
|
|
ctx.String("owner_name"),
|
|
ctx.String("repo_name"),
|
|
ctx.StringSlice("units"),
|
|
)
|
|
if statusCode == http.StatusOK {
|
|
return nil
|
|
}
|
|
|
|
log.Fatal("Failed to restore repository: %v", errStr)
|
|
return errors.New(errStr)
|
|
}
|