From fa9d3d4e30bdbb243819daab99b103d0cdd38dfd Mon Sep 17 00:00:00 2001
From: tengge <930372551@qq.com>
Date: Mon, 4 May 2020 08:08:09 +0800
Subject: [PATCH] backup database.

---
 .gitignore                                    |  1 +
 .../server/tools/backup_database/handler.go   | 90 +++++++++++++++++++
 server/server/tools/handle_backup_database.go |  8 --
 3 files changed, 91 insertions(+), 8 deletions(-)
 create mode 100644 server/server/tools/backup_database/handler.go
 delete mode 100644 server/server/tools/handle_backup_database.go

diff --git a/.gitignore b/.gitignore
index 56d187c8..9af82c0b 100644
--- a/.gitignore
+++ b/.gitignore
@@ -65,6 +65,7 @@ typings/
 
 # custom files
 build/
+dump/
 logs/
 server/shadoweditor
 server/shadoweditor.exe
diff --git a/server/server/tools/backup_database/handler.go b/server/server/tools/backup_database/handler.go
new file mode 100644
index 00000000..833a4c68
--- /dev/null
+++ b/server/server/tools/backup_database/handler.go
@@ -0,0 +1,90 @@
+// Copyright 2017-2020 The ShadowEditor Authors. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+//
+// For more information, please visit: https://github.com/tengge1/ShadowEditor
+// You can also visit: https://gitee.com/tengge1/ShadowEditor
+
+package backupdatabase
+
+import (
+	"fmt"
+	"net/http"
+	"net/url"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"time"
+
+	"github.com/tengge1/shadoweditor/helper"
+	"github.com/tengge1/shadoweditor/server"
+)
+
+func init() {
+	server.Mux.UsingContext().Handle(http.MethodPost, "/api/BackupDatabase/Run", handle)
+}
+
+func handle(w http.ResponseWriter, r *http.Request) {
+	db, err := server.Mongo()
+	if err != nil {
+		helper.WriteJSON(w, server.Result{
+			Code: 300,
+			Msg:  err.Error(),
+		})
+		return
+	}
+
+	// Get `mongodump` path.
+	sr, err := db.RunCommand("{ serverStatus: 1, asserts: 0, repl: 0, metrics: 0, locks: 0 }")
+	if err != nil {
+		helper.WriteJSON(w, server.Result{
+			Code: 300,
+			Msg:  err.Error(),
+		})
+		return
+	}
+
+	var status []map[string]interface{}
+	sr.Decode(&status)
+
+	path := status[0]["process"].(string)
+	dir := filepath.Dir(path)
+	dump := filepath.Join(dir, "mongodump.exe")
+
+	if _, err := os.Stat(dump); os.IsNotExist(err) {
+		helper.WriteJSON(w, server.Result{
+			Code: 300,
+			Msg:  "mongodump.exe is not existed.",
+		})
+		return
+	}
+
+	// Get mongodb information.
+	uri, _ := url.Parse(db.ConnectionString)
+	host := uri.Host
+	port := uri.Port()
+	dbName := db.DatabaseName
+	now := time.Now()
+
+	backupDir := server.MapPath("/backup/database/dump" + helper.TimeToString(now, "yyyyMMddHHmmss"))
+
+	if _, err := os.Stat(backupDir); os.IsNotExist(err) {
+		os.MkdirAll(backupDir, 0755)
+	}
+
+	// Start a process to backup MongoDB.
+	cmd := fmt.Sprintf("--host %v --port %v --db %v --out %v", host, port, dbName, backupDir)
+	process := exec.Command(dump, cmd)
+	process.Start()
+
+	result := backupResult{}
+	result.Code = 200
+	result.Msg = "Backup database successfully!"
+	result.Path = backupDir
+	helper.WriteJSON(w, result)
+}
+
+type backupResult struct {
+	server.Result
+	Path string
+}
diff --git a/server/server/tools/handle_backup_database.go b/server/server/tools/handle_backup_database.go
deleted file mode 100644
index 87865c02..00000000
--- a/server/server/tools/handle_backup_database.go
+++ /dev/null
@@ -1,8 +0,0 @@
-// Copyright 2017-2020 The ShadowEditor Authors. All rights reserved.
-// Use of this source code is governed by a MIT-style
-// license that can be found in the LICENSE file.
-//
-// For more information, please visit: https://github.com/tengge1/ShadowEditor
-// You can also visit: https://gitee.com/tengge1/ShadowEditor
-
-package tools