123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162 |
- // 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 (
- "context"
- "errors"
- "strings"
-
- "code.gitea.io/gitea/modules/convert"
- "code.gitea.io/gitea/modules/log"
- "code.gitea.io/gitea/modules/migrations"
- "code.gitea.io/gitea/modules/migrations/base"
- "code.gitea.io/gitea/modules/setting"
- "code.gitea.io/gitea/modules/structs"
-
- "github.com/urfave/cli"
- )
-
- // CmdDumpRepository represents the available dump repository sub-command.
- var CmdDumpRepository = cli.Command{
- Name: "dump-repo",
- Usage: "Dump the repository from git/github/gitea/gitlab",
- Description: "This is a command for dumping the repository data.",
- Action: runDumpRepository,
- Flags: []cli.Flag{
- cli.StringFlag{
- Name: "git_service",
- Value: "",
- Usage: "Git service, git, github, gitea, gitlab. If clone_addr could be recognized, this could be ignored.",
- },
- cli.StringFlag{
- Name: "repo_dir, r",
- Value: "./data",
- Usage: "Repository dir path to store the data",
- },
- cli.StringFlag{
- Name: "clone_addr",
- Value: "",
- Usage: "The URL will be clone, currently could be a git/github/gitea/gitlab http/https URL",
- },
- cli.StringFlag{
- Name: "auth_username",
- Value: "",
- Usage: "The username to visit the clone_addr",
- },
- cli.StringFlag{
- Name: "auth_password",
- Value: "",
- Usage: "The password to visit the clone_addr",
- },
- cli.StringFlag{
- Name: "auth_token",
- Value: "",
- Usage: "The personal token to visit the clone_addr",
- },
- cli.StringFlag{
- Name: "owner_name",
- Value: "",
- Usage: "The data will be stored on a directory with owner name if not empty",
- },
- cli.StringFlag{
- Name: "repo_name",
- Value: "",
- Usage: "The data will be stored on a directory with repository name if not empty",
- },
- cli.StringFlag{
- Name: "units",
- Value: "",
- Usage: `Which items will be migrated, 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 runDumpRepository(ctx *cli.Context) error {
- if err := initDB(); err != nil {
- return err
- }
-
- log.Trace("AppPath: %s", setting.AppPath)
- log.Trace("AppWorkPath: %s", setting.AppWorkPath)
- log.Trace("Custom path: %s", setting.CustomPath)
- log.Trace("Log path: %s", setting.LogRootPath)
- setting.InitDBConfig()
-
- var (
- serviceType structs.GitServiceType
- cloneAddr = ctx.String("clone_addr")
- serviceStr = ctx.String("git_service")
- )
-
- if strings.HasPrefix(strings.ToLower(cloneAddr), "https://github.com/") {
- serviceStr = "github"
- } else if strings.HasPrefix(strings.ToLower(cloneAddr), "https://gitlab.com/") {
- serviceStr = "gitlab"
- } else if strings.HasPrefix(strings.ToLower(cloneAddr), "https://gitea.com/") {
- serviceStr = "gitea"
- }
- if serviceStr == "" {
- return errors.New("git_service missed or clone_addr cannot be recognized")
- }
- serviceType = convert.ToGitServiceType(serviceStr)
-
- var opts = base.MigrateOptions{
- GitServiceType: serviceType,
- CloneAddr: cloneAddr,
- AuthUsername: ctx.String("auth_username"),
- AuthPassword: ctx.String("auth_password"),
- AuthToken: ctx.String("auth_token"),
- RepoName: ctx.String("repo_name"),
- }
-
- if len(ctx.String("units")) == 0 {
- opts.Wiki = true
- opts.Issues = true
- opts.Milestones = true
- opts.Labels = true
- opts.Releases = true
- opts.Comments = true
- opts.PullRequests = true
- opts.ReleaseAssets = true
- } else {
- units := strings.Split(ctx.String("units"), ",")
- for _, unit := range units {
- switch strings.ToLower(unit) {
- case "wiki":
- opts.Wiki = true
- case "issues":
- opts.Issues = true
- case "milestones":
- opts.Milestones = true
- case "labels":
- opts.Labels = true
- case "releases":
- opts.Releases = true
- case "release_assets":
- opts.ReleaseAssets = true
- case "comments":
- opts.Comments = true
- case "pull_requests":
- opts.PullRequests = true
- }
- }
- }
-
- if err := migrations.DumpRepository(
- context.Background(),
- ctx.String("repo_dir"),
- ctx.String("owner_name"),
- opts,
- ); err != nil {
- log.Fatal("Failed to dump repository: %v", err)
- return err
- }
-
- log.Trace("Dump finished!!!")
-
- return nil
- }
|