|
|
@@ -0,0 +1,129 @@
|
|
|
+// Copyright 2017 The Gogs 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 (
|
|
|
+ "os"
|
|
|
+ "path"
|
|
|
+
|
|
|
+ "github.com/Unknwon/cae/zip"
|
|
|
+ "github.com/Unknwon/com"
|
|
|
+ "github.com/mcuadros/go-version"
|
|
|
+ "github.com/urfave/cli"
|
|
|
+ log "gopkg.in/clog.v1"
|
|
|
+ "gopkg.in/ini.v1"
|
|
|
+
|
|
|
+ "github.com/gogits/gogs/models"
|
|
|
+ "github.com/gogits/gogs/modules/setting"
|
|
|
+)
|
|
|
+
|
|
|
+var Restore = cli.Command{
|
|
|
+ Name: "restore",
|
|
|
+ Usage: "Restore files and database from backup",
|
|
|
+ Description: `Restore imports all related files and database from a backup archive.
|
|
|
+The backup version must lower or equal to current Gogs version. You can also import
|
|
|
+backup from other database engines, which is useful for database migrating.
|
|
|
+
|
|
|
+If corresponding files or database tables are not presented in the archive, they will
|
|
|
+be skipped and remian unchanged.`,
|
|
|
+ Action: runRestore,
|
|
|
+ Flags: []cli.Flag{
|
|
|
+ stringFlag("config, c", "custom/conf/app.ini", "Custom configuration file path"),
|
|
|
+ boolFlag("verbose, v", "Show process details"),
|
|
|
+ stringFlag("tempdir, t", os.TempDir(), "Temporary directory path"),
|
|
|
+ stringFlag("from", "", "Path to backup archive"),
|
|
|
+ boolFlag("database-only", "Only import database"),
|
|
|
+ boolFlag("exclude-repos", "Exclude repositories"),
|
|
|
+ },
|
|
|
+}
|
|
|
+
|
|
|
+func runRestore(c *cli.Context) error {
|
|
|
+ zip.Verbose = c.Bool("verbose")
|
|
|
+
|
|
|
+ tmpDir := c.String("tempdir")
|
|
|
+ if !com.IsExist(tmpDir) {
|
|
|
+ log.Fatal(0, "'--tempdir' does not exist: %s", tmpDir)
|
|
|
+ }
|
|
|
+
|
|
|
+ log.Info("Restore backup from: %s", c.String("from"))
|
|
|
+ if err := zip.ExtractTo(c.String("from"), tmpDir); err != nil {
|
|
|
+ log.Fatal(0, "Fail to extract backup archive: %v", err)
|
|
|
+ }
|
|
|
+ archivePath := path.Join(tmpDir, _ARCHIVE_ROOT_DIR)
|
|
|
+
|
|
|
+ // Check backup version
|
|
|
+ metaFile := path.Join(archivePath, "metadata.ini")
|
|
|
+ if !com.IsExist(metaFile) {
|
|
|
+ log.Fatal(0, "File 'metadata.ini' is missing")
|
|
|
+ }
|
|
|
+ metadata, err := ini.Load(metaFile)
|
|
|
+ if err != nil {
|
|
|
+ log.Fatal(0, "Fail to load metadata '%s': %v", metaFile, err)
|
|
|
+ }
|
|
|
+ backupVersion := metadata.Section("").Key("GOGS_VERSION").MustString("999.0")
|
|
|
+ if version.Compare(setting.AppVer, backupVersion, "<") {
|
|
|
+ log.Fatal(0, "Current Gogs version is lower than backup version: %s < %s", setting.AppVer, backupVersion)
|
|
|
+ }
|
|
|
+
|
|
|
+ // If config file is not present in backup, user must set this file via flag.
|
|
|
+ // Otherwise, it's optional to set config file flag.
|
|
|
+ configFile := path.Join(archivePath, "custom/conf/app.ini")
|
|
|
+ if c.IsSet("config") {
|
|
|
+ setting.CustomConf = c.String("config")
|
|
|
+ } else if !com.IsExist(configFile) {
|
|
|
+ log.Fatal(0, "'--config' is not specified and custom config file is not found in backup")
|
|
|
+ } else {
|
|
|
+ setting.CustomConf = configFile
|
|
|
+ }
|
|
|
+ setting.NewContext()
|
|
|
+ models.LoadConfigs()
|
|
|
+ models.SetEngine()
|
|
|
+
|
|
|
+ // Database
|
|
|
+ dbDir := path.Join(archivePath, "db")
|
|
|
+ if err = models.ImportDatabase(dbDir); err != nil {
|
|
|
+ log.Fatal(0, "Fail to import database: %v", err)
|
|
|
+ }
|
|
|
+
|
|
|
+ // Custom files
|
|
|
+ if !c.Bool("database-only") {
|
|
|
+ if com.IsExist(setting.CustomPath) {
|
|
|
+ if err = os.Rename(setting.CustomPath, setting.CustomPath+".bak"); err != nil {
|
|
|
+ log.Fatal(0, "Fail to backup current 'custom': %v", err)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if err = os.Rename(path.Join(archivePath, "custom"), setting.CustomPath); err != nil {
|
|
|
+ log.Fatal(0, "Fail to import 'custom': %v", err)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // Data files
|
|
|
+ if !c.Bool("database-only") {
|
|
|
+ for _, dir := range []string{"attachments", "avatars"} {
|
|
|
+ dirPath := path.Join(setting.AppDataPath, dir)
|
|
|
+ if com.IsExist(dirPath) {
|
|
|
+ if err = os.Rename(dirPath, dirPath+".bak"); err != nil {
|
|
|
+ log.Fatal(0, "Fail to backup current 'data': %v", err)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if err = os.Rename(path.Join(archivePath, "data", dir), dirPath); err != nil {
|
|
|
+ log.Fatal(0, "Fail to import 'data': %v", err)
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // Repositories
|
|
|
+ reposPath := path.Join(archivePath, "repositories.zip")
|
|
|
+ if !c.Bool("exclude-repos") && !c.Bool("database-only") && com.IsExist(reposPath) {
|
|
|
+ if err := zip.ExtractTo(reposPath, path.Dir(setting.RepoRootPath)); err != nil {
|
|
|
+ log.Fatal(0, "Fail to extract 'repositories.zip': %v", err)
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ os.RemoveAll(path.Join(tmpDir, _ARCHIVE_ROOT_DIR))
|
|
|
+ log.Info("Restore succeed!")
|
|
|
+ log.Shutdown()
|
|
|
+ return nil
|
|
|
+}
|
Unknwon