From 0c0843f44973a5e0fc91839fce1dc0fdf3cd77e1 Mon Sep 17 00:00:00 2001
From: iyear <ljyngup@gmail.com>
Date: Thu, 9 Apr 2020 15:47:48 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=20/export=20=E5=91=BD?=
 =?UTF-8?q?=E4=BB=A4=E7=94=A8=E4=BA=8E=E5=AF=BC=E5=87=BA=E4=BF=A1=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 handle.go | 79 +++++++++++++++++++++++++++++++++++++++++++++++--------
 main.go   |  1 +
 2 files changed, 69 insertions(+), 11 deletions(-)

diff --git a/handle.go b/handle.go
index 0b037c2..863b341 100644
--- a/handle.go
+++ b/handle.go
@@ -1,10 +1,12 @@
 package main
 
 import (
-	"fmt"
+	"encoding/json"
 	"github.com/fsnotify/fsnotify"
 	"github.com/spf13/viper"
 	tb "gopkg.in/tucnak/telebot.v2"
+	"io/ioutil"
+	"os"
 	"path/filepath"
 	"strconv"
 	"strings"
@@ -19,6 +21,7 @@ const (
 	/my 查看已绑定账户信息
 	/bind  绑定新账户
 	/unbind 解绑账户
+	/export 导出账户信息(json格式)
 	/help 帮助
 	/task 手动执行一次任务(管理员)
 	/log 获取最近日志文件(管理员)
@@ -113,7 +116,9 @@ func bLog(m *tb.Message) {
 		inlineKeys = append(inlineKeys, []tb.InlineButton{inlineBtn})
 	}
 	_, err := bot.Send(m.Chat, "选择一个日志", &tb.ReplyMarkup{InlineKeyboard: inlineKeys})
-	fmt.Println(err)
+	if err != nil {
+		logger.Println(err)
+	}
 }
 func bLogsInlineBtn(c *tb.Callback) {
 	//fmt.Println(c.Data)
@@ -130,11 +135,12 @@ func bBind1(m *tb.Message) {
 	logger.Println("ReApp: " + strconv.FormatInt(m.Chat.ID, 10))
 	bot.Send(m.Chat, "应用注册： [点击直达]("+MSGetReAppUrl()+")", tb.ModeMarkdown)
 	_, err := bot.Send(m.Chat, "请回复client_id+空格+client_secret", &tb.ReplyMarkup{ForceReply: true})
-	if err == nil {
-		UserStatus[m.Chat.ID] = USBind1
-		UserCid[m.Chat.ID] = m.Text
+	if err != nil {
+		logger.Println(err)
+		return
 	}
-
+	UserStatus[m.Chat.ID] = USBind1
+	UserCid[m.Chat.ID] = m.Text
 }
 func bBind2(m *tb.Message) {
 	logger.Println("Auth: " + strconv.FormatInt(m.Chat.ID, 10))
@@ -149,11 +155,13 @@ func bBind2(m *tb.Message) {
 	cse := tmp[1]
 	bot.Send(m.Chat, "授权账户： [点击直达]("+MSGetAuthUrl(cid)+")", tb.ModeMarkdown)
 	_, err := bot.Send(m.Chat, "请回复http://localhost/…… + 空格 + 别名(用于管理)", &tb.ReplyMarkup{ForceReply: true})
-	if err == nil {
-		UserStatus[m.Chat.ID] = USBind2
-		UserCid[m.Chat.ID] = cid
-		UserCSecret[m.Chat.ID] = cse
+	if err != nil {
+		logger.Println(err)
+		return
 	}
+	UserStatus[m.Chat.ID] = USBind2
+	UserCid[m.Chat.ID] = cid
+	UserCSecret[m.Chat.ID] = cse
 }
 
 func bUnBind(m *tb.Message) {
@@ -182,7 +190,56 @@ func bUnBindInlineBtn(c *tb.Callback) {
 	bot.Send(c.Message.Chat, "解绑成功!")
 	bot.Respond(c)
 }
-
+func bExport(m *tb.Message) {
+	type MsMiniData struct {
+		Alias        string
+		ClientId     string
+		ClientSecret string
+		RefreshToken string
+		Other        string
+	}
+	var MsMini []MsMiniData
+	data := QueryDataByTG(db, m.Chat.ID)
+	if len(data) == 0 {
+		bot.Send(m.Chat, "你还没有绑定过账户嗷~")
+		return
+	}
+	for _, u := range data {
+		var ms MsMiniData
+		ms.RefreshToken = u.refreshToken
+		ms.Alias = u.alias
+		ms.ClientId = u.clientId
+		ms.ClientSecret = u.clientSecret
+		ms.Other = u.other
+		MsMini = append(MsMini, ms)
+	}
+	//MarshalIndent是为json+美化,/t表缩进
+	export, err := json.MarshalIndent(MsMini, "", "\t")
+	if err != nil {
+		logger.Println(err)
+		bot.Send(m.Chat, "获取JSON失败~\n"+err.Error())
+		return
+	}
+	//fmt.Println(string(export))
+	fileName := "./" + strconv.FormatInt(m.Chat.ID, 10) + "_export_tmp.json"
+	if err = ioutil.WriteFile(fileName, export, 0644); err != nil {
+		logger.Println(err)
+		bot.Send(m.Chat, "写入临时文件失败~\n"+err.Error())
+		return
+	}
+	exportFile := &tb.Document{File: tb.FromDisk(fileName), FileName: strconv.FormatInt(m.Chat.ID, 10) + ".json", MIME: "text/plain"}
+	_, err = bot.Send(m.Chat, exportFile)
+	if err != nil {
+		logger.Println(err)
+		return
+	}
+	//不遗留本地文件
+	if exportFile.InCloud() == true && os.Remove(fileName) == nil {
+		logger.Println(fileName + " Has Removed")
+	} else {
+		logger.Println(fileName + " Removed ERROR")
+	}
+}
 func bHelp(m *tb.Message) {
 	bot.Send(m.Sender, bHelpContent+"\n"+notice, &tb.SendOptions{DisableWebPagePreview: false})
 }
diff --git a/main.go b/main.go
index 242f765..68cc1af 100644
--- a/main.go
+++ b/main.go
@@ -51,6 +51,7 @@ func MakeHandle() {
 	bot.Handle("/my", bMy)
 	bot.Handle("/bind", bBind1)
 	bot.Handle("/unbind", bUnBind)
+	bot.Handle("/export", bExport)
 	bot.Handle("/help", bHelp)
 	bot.Handle(tb.OnText, bOnText)
 	//管理员