Initial commit

DBManager.go

@@ -0,0 +1,187 @@
+// Copyright 2019 ihciah <ihciah@gmail.com>
+//
+// Licensed under the GNU Affero General Public License, Version 3.0
+// (the "License"); you may not use this file except in compliance with the
+// License.
+// You may obtain a copy of the License at
+//
+//     https://www.gnu.org/licenses/agpl-3.0.html
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package main
+
+import (
+	"bufio"
+	"fmt"
+	"github.com/gfhelper/GFHelper/cipher"
+	"github.com/gfhelper/GFHelper/usercenter"
+	"github.com/gfhelper/GFHelper/utils"
+	"io/ioutil"
+	"log"
+	"os"
+	"strconv"
+	"strings"
+)
+
+func DeleteUsers(uc *usercenter.UserCenter) {
+	// 给定用户列表文件，从数据库中删除这些用户
+	filepath := utils.GetInput("Input file path:")
+	file, err := os.Open(filepath)
+	if err != nil {
+		log.Println("Unable to load given file.")
+		return
+	}
+	scanner := bufio.NewScanner(file)
+	for scanner.Scan() {
+		line := scanner.Text()
+		sep := " "
+		if strings.Contains(line, ",") {
+			sep = ","
+		}
+		if strings.Contains(line, "\t") {
+			sep = "\t"
+		}
+		parts := strings.Split(line, sep)
+		if len(parts) > 0 {
+			account := strings.Trim(parts[0], " ,\t\r\n")
+			if strings.HasSuffix(account, "com") {
+				err := uc.DelUser(account)
+				if err != nil {
+					log.Printf("Error when delete user %s: %s.", account, err)
+				} else {
+					log.Printf("User %s deleted.", account)
+				}
+			}
+		}
+	}
+}
+
+func AddUser(uc *usercenter.UserCenter) {
+	userID := utils.GetInput("ID:")
+	password := utils.GetInput("Password:")
+	channel := utils.GetInput("Channel(ios, android, tw):")
+	uc.AddUser(userID, password, channel)
+	log.Println("Done!")
+}
+
+func ExportAllUsers(uc *usercenter.UserCenter) {
+	// 导出所有用户到json文件
+	err := uc.ExportToFile("export.json")
+	if err != nil {
+		log.Printf("Error when export: %s", err)
+	}
+	log.Println("Done!")
+}
+
+func writeFileChannel(users []usercenter.UserData, suffix string) error {
+	outputFullPath := fmt.Sprintf("export_%s_full.txt", suffix)
+	outputPath := fmt.Sprintf("export_%s.txt", suffix)
+	linesFull := make([]string, 0, len(users))
+	lines := make([]string, 0, len(users))
+	for _, u := range users {
+		lines = append(lines, fmt.Sprintf("%s,%s", u.ID, u.Password))
+		linesFull = append(linesFull, fmt.Sprintf("%s %s %d %d %d %d %d %d %d", u.ID, u.Password, u.MP,
+			u.AMMO, u.MRE, u.PART, u.CORE, u.BuildCard, u.FastBuildCard))
+	}
+	err := ioutil.WriteFile(outputPath, []byte(strings.Join(lines, "\n")), 0644)
+	if err != nil {
+		return err
+	}
+	err = ioutil.WriteFile(outputFullPath, []byte(strings.Join(linesFull, "\n")), 0644)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func writeFile(users []usercenter.UserData, suffix string) error {
+	users_ios := make([]usercenter.UserData, 0, len(users))
+	users_android := make([]usercenter.UserData, 0, len(users))
+	for _, u := range users {
+		if u.ChannelId == "ios" {
+			users_ios = append(users_ios, u)
+		} else {
+			users_android = append(users_android, u)
+		}
+	}
+	var err error
+	err = writeFileChannel(users_ios, suffix+"_ios")
+	if err != nil {
+		return err
+	}
+	err = writeFileChannel(users_android, suffix+"_android")
+	return err
+}
+
+func ExportFilteredUsers(uc *usercenter.UserCenter) {
+	// 导出过滤后的用户
+	exportLevels := [][]int{{600, 1150000}, {500, 1100000}, {400, 1050000}, {300, 800000}}
+
+	users, err := uc.ReadAll()
+	if err != nil {
+		log.Printf("Error when read users: %s", err)
+	}
+	leveledAccounts := make(map[int][]usercenter.UserData)
+	for _, level := range exportLevels {
+		leveledAccounts[level[0]] = make([]usercenter.UserData, 0, len(users))
+	}
+
+	for _, u := range users {
+		resources, build, fastBuild := u.Summary()
+		for _, level := range exportLevels {
+			if build >= int64(level[0]) && fastBuild >= int64(level[0]) && resources >= int64(level[1]) {
+				leveledAccounts[level[0]] = append(leveledAccounts[level[0]], u)
+				break
+			}
+		}
+	}
+
+	for level, usersExport := range leveledAccounts {
+		writeFile(usersExport, strconv.Itoa(level))
+	}
+	log.Println("Done!")
+}
+
+func Interact(uc *usercenter.UserCenter) {
+	choices := map[string]func(*usercenter.UserCenter){
+		"1": ExportAllUsers,
+		"2": ExportFilteredUsers,
+		"3": DeleteUsers,
+		"4": AddUser,
+	}
+	num := utils.GetInput("1: Export all users\n2: Export filtered users\n3: Delete given users\n4: Add user\nEnter a choice: ")
+	f, ok := choices[num]
+	if !ok {
+		fmt.Println("Wrong choice!")
+		return
+	}
+	f(uc)
+}
+
+func main() {
+	author := []byte("\x57\x66\xe8\xec\x62\xc4\xfd\x40\x3a\xc0\x80\x29\xaf\xe5\x60\xc2\x9a\xb3\x6c\x0a\x41\x70\x0c\x7d\xbd\xf8\x9b\xab\x42\x43\xa3\xc6\xde\xe9\x91\xa9\xdc\x48\xc4\x1f\xd1\xeb\xa9\xc3\xda\xbb\xf8\x59\xb0\x63\x33\x99\x03\x13\x6d\x38\xcf\x1c\xa2\x39\x73\x2c\xc1\xde\x27\x5e\x4e\x19\x21\x50\xe0\xb3")
+	authorPlain, err := cipher.XXTEADecryptToString(author)
+	if err != nil{
+		return
+	}
+	fmt.Println(authorPlain)
+	uc, err := usercenter.MakeUserCenter(usercenter.MaxResourceDatabase, false)
+	if err != nil {
+		fmt.Println(err)
+		return
+	}
+	existed, err := uc.ReadAll()
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	fmt.Println("Existed user:", len(existed))
+	for {
+		Interact(&uc)
+	}
+}

DBMigrator.go

@@ -0,0 +1,116 @@
+// Copyright 2019 ihciah <ihciah@gmail.com>
+//
+// Licensed under the GNU Affero General Public License, Version 3.0
+// (the "License"); you may not use this file except in compliance with the
+// License.
+// You may obtain a copy of the License at
+//
+//     https://www.gnu.org/licenses/agpl-3.0.html
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/asdine/storm"
+	"github.com/coreos/bbolt"
+	"github.com/gfhelper/GFHelper/cipher"
+	"log"
+	"strings"
+	"time"
+)
+
+type UserDataOld struct {
+	ID            string
+	Password      string
+	AccessToken   string
+	OpenID        string
+	UID           string
+	Sign          string
+	IsReady       bool
+	MP            int64
+	AMMO          int64
+	MRE           int64
+	PART          int64
+	CORE          int64
+	BuildCard     int64
+	FastBuildCard int64
+}
+
+type UserData struct {
+	ID            string
+	Password      string
+	AccessToken   string
+	OpenID        string
+	UID           string
+	Sign          string
+	ChannelId     string
+	MP            int64
+	AMMO          int64
+	MRE           int64
+	PART          int64
+	CORE          int64
+	BuildCard     int64
+	FastBuildCard int64
+}
+
+func main() {
+	author := []byte("\x57\x66\xe8\xec\x62\xc4\xfd\x40\x3a\xc0\x80\x29\xaf\xe5\x60\xc2\x9a\xb3\x6c\x0a\x41\x70\x0c\x7d\xbd\xf8\x9b\xab\x42\x43\xa3\xc6\xde\xe9\x91\xa9\xdc\x48\xc4\x1f\xd1\xeb\xa9\xc3\xda\xbb\xf8\x59\xb0\x63\x33\x99\x03\x13\x6d\x38\xcf\x1c\xa2\x39\x73\x2c\xc1\xde\x27\x5e\x4e\x19\x21\x50\xe0\xb3")
+	authorPlain, err := cipher.XXTEADecryptToString(author)
+	if err != nil{
+		return
+	}
+	fmt.Println(authorPlain)
+	db, err := storm.Open("database.db", storm.BoltOptions(0600, &bolt.Options{Timeout: 10 * time.Second}))
+	if err != nil {
+		log.Fatal("Cannot open database")
+	}
+
+	oldUsers := make([]UserDataOld, 0, 0)
+	db.Bolt.View(func(tx *bolt.Tx) error {
+		b := tx.Bucket([]byte("UserData"))
+		b.ForEach(func(k, v []byte) error {
+			if strings.HasPrefix(string(k), "_") {
+				return nil
+			}
+			var user UserDataOld
+			if err := json.Unmarshal(v, &user); err == nil {
+				oldUsers = append(oldUsers, user)
+			}
+			return nil
+		})
+		return nil
+	})
+	log.Printf("Read %d users from old database.", len(oldUsers))
+
+	dbMig, err := storm.Open("database_migrated.db", storm.BoltOptions(0600, &bolt.Options{Timeout: 10 * time.Second}))
+	if err != nil {
+		log.Fatal("Cannot open database")
+	}
+	for _, user := range oldUsers {
+		newUser := UserData{
+			user.ID,
+			user.Password,
+			user.AccessToken,
+			user.OpenID,
+			user.UID,
+			user.Sign,
+			"ios",
+			user.MP,
+			user.AMMO,
+			user.MRE,
+			user.PART,
+			user.CORE,
+			user.BuildCard,
+			user.FastBuildCard,
+		}
+		dbMig.Save(&newUser)
+	}
+	log.Println("Done.")
+}