Reorganize directory structures

Signed-off-by: Sean Du <do4suki@gmail.com>

internal/tools/init.go

@@ -0,0 +1,150 @@
+package tools
+
+import (
+	"fmt"
+	"honoka-chan/internal/model"
+	"honoka-chan/internal/utils"
+	"honoka-chan/pkg/db"
+	"time"
+)
+
+type UserData struct {
+	ID            int    `xorm:"id pk autoincr"`
+	Phone         string `xorm:"phone"`
+	Password      string `xorm:"password"`
+	Autokey       string `xorm:"autokey"`
+	Ticket        string `xorm:"ticket"`
+	UserID        int    `xorm:"userid"`
+	LastLoginTime int64  `xorm:"last_login_time"`
+}
+
+type UserPref struct {
+	ID               int    `xorm:"id pk autoincr"`
+	UserID           int    `xorm:"user_id"`
+	AwardID          int    `xorm:"award_id"`
+	BackgroundID     int    `xorm:"background_id"`
+	UnitOwningUserID int    `xorm:"unit_owning_user_id"`
+	UserName         string `xorm:"user_name"`
+	UserLevel        int    `xorm:"user_level"`
+	UserDesc         string `xorm:"user_desc"`
+	UpdateTime       int64  `xorm:"update_time"`
+}
+
+func InitUserData(userId int) {
+	userList := []UserData{}
+	if userId != 0 {
+		err := db.UserEng.Table("users").Where("userid = ?", userId).Find(&userList)
+		utils.CheckErr(err)
+	} else {
+		err := db.UserEng.Table("users").Asc("id").Find(&userList)
+		utils.CheckErr(err)
+	}
+
+	session := db.UserEng.NewSession()
+	defer session.Close()
+
+	if err := session.Begin(); err != nil {
+		panic(err)
+	}
+
+	for _, user := range userList {
+		// 检查用户配置
+		exists, err := db.UserEng.Table("user_preference_m").Where("user_id = ?", user.UserID).Exist()
+		utils.CheckErr(err)
+
+		if !exists {
+			// 默认中心成员（）
+			var oId int
+			_, err = db.MainEng.Table("common_unit_m").Cols("unit_owning_user_id").Where("unit_id = ?", 31).Get(&oId)
+			utils.CheckErr(err)
+			fmt.Println("Center UnitOwningUserID:", oId)
+			userPref := UserPref{
+				UserID:           user.UserID,
+				AwardID:          1, // 音乃木坂学生
+				BackgroundID:     1, // 初始背景
+				UnitOwningUserID: oId,
+				UserName:         "音乃木坂学生",
+				UserLevel:        1,
+				UserDesc:         "你好。",
+				UpdateTime:       time.Now().Unix(),
+			}
+			_, err = session.Table("user_preference_m").Insert(&userPref)
+			utils.CheckErr(err)
+			fmt.Println("UserPref ID", userPref.ID)
+		}
+
+		// 检查用户卡组配置
+		exists, err = db.UserEng.Table("user_deck_m").Where("user_id = ?", user.UserID).Asc("deck_id").Exist()
+		utils.CheckErr(err)
+		// fmt.Println("UserDeck exists:", exists)
+
+		if !exists {
+			userDeck := model.UserDeckData{
+				DeckID:     1,
+				MainFlag:   1,
+				DeckName:   "队伍A",
+				UserID:     user.UserID,
+				InsertDate: time.Now().Unix(),
+			}
+
+			// 默认队伍
+			_, err = session.Table("user_deck_m").Insert(&userDeck)
+			if err != nil {
+				session.Rollback()
+				panic(err)
+			}
+			userDeckId := userDeck.ID
+			fmt.Println("New UserDeck:", userDeckId)
+
+			// 默认卡组
+			unitIds := []int{}
+			err = db.MainEng.Table("unit_m").Cols("unit_id").Where("album_series_id = ?", 615).Find(&unitIds)
+			if err != nil {
+				session.Rollback()
+				panic(err)
+			}
+
+			unitData := []model.UnitData{}
+			err = db.MainEng.Table("common_unit_m").In("unit_id", unitIds).Find(&unitData)
+			if err != nil {
+				session.Rollback()
+				panic(err)
+			}
+			// fmt.Println(unitData)
+
+			position := 1
+			for _, unit := range unitData {
+				unitDeckData := model.UnitDeckData{
+					UserDeckID:       userDeckId,
+					UnitOwningUserID: unit.UnitOwningUserID,
+					UnitID:           unit.UnitID,
+					Position:         position,
+					Level:            100,
+					LevelLimitID:     1,
+					DisplayRank:      2,
+					Love:             1000,
+					UnitSkillLevel:   8,
+					IsRankMax:        true,
+					IsLoveMax:        true,
+					IsLevelMax:       true,
+					IsSigned:         unit.IsSigned,
+					BeforeLove:       1000,
+					MaxLove:          1000,
+					InsertData:       time.Now().Unix(),
+				}
+				_, err = session.Table("deck_unit_m").Insert(&unitDeckData)
+				if err != nil {
+					session.Rollback()
+					panic(err)
+				}
+				fmt.Println("New DeckUnit:", unitDeckData.ID)
+
+				position++
+			}
+
+			if err = session.Commit(); err != nil {
+				panic(err)
+			}
+		}
+	}
+}