YumeMichi
cb0200ff29
- Add user_accessory as the per-user accessory inventory table. - Migrate old accessory wear records to user-owned accessory ids, including legacy common_accessory_m lookups for old installs. - Allocate default accessories for newly created users and update accessory reads to use user_accessory at runtime. - Validate accessory ownership when wearing accessories and keep existing accessory-based responses working. Signed-off-by: Sean Du <do4suki@gmail.com>
76 lines
2.3 KiB
Go
76 lines
2.3 KiB
Go
package session
|
|
|
|
import (
|
|
accessorymodel "honoka-chan/internal/model/accessory"
|
|
usermodel "honoka-chan/internal/model/user"
|
|
liverecordschema "honoka-chan/internal/schema/liverecord"
|
|
)
|
|
|
|
func (ss *Session) GetUserAccessoryWearByUnitOwningUserID(unitOwningUserID int) (bool, *usermodel.UserAccessoryWear) {
|
|
wearData := usermodel.UserAccessoryWear{}
|
|
has, err := ss.UserEng.Table(new(usermodel.UserAccessoryWear)).
|
|
Where("unit_owning_user_id = ?", unitOwningUserID).Get(&wearData)
|
|
if ss.CheckErr(err) {
|
|
return false, nil
|
|
}
|
|
|
|
return has, &wearData
|
|
}
|
|
|
|
func (ss *Session) GetUserAccessoryByAccessoryOwningUserID(accessoryOwningUserID int) (bool, *usermodel.UserAccessory) {
|
|
accessoryData := usermodel.UserAccessory{}
|
|
has, err := ss.UserEng.Table(new(usermodel.UserAccessory)).
|
|
Where("accessory_owning_user_id = ?", accessoryOwningUserID).
|
|
Get(&accessoryData)
|
|
if ss.CheckErr(err) {
|
|
return false, nil
|
|
}
|
|
|
|
return has, &accessoryData
|
|
}
|
|
|
|
func (ss *Session) GetAccessoryByAccessoryOwningUserID(accessoryOwningUserID int) (bool, *accessorymodel.Accessory) {
|
|
has, userAccessoryData := ss.GetUserAccessoryByAccessoryOwningUserID(accessoryOwningUserID)
|
|
if !has {
|
|
return false, nil
|
|
}
|
|
|
|
accessoryData := accessorymodel.Accessory{}
|
|
has, err := ss.MainEng.Table("accessory_m").
|
|
Where("accessory_id = ?", userAccessoryData.AccessoryID).
|
|
Get(&accessoryData)
|
|
if ss.CheckErr(err) {
|
|
return false, nil
|
|
}
|
|
if !has {
|
|
return false, nil
|
|
}
|
|
|
|
accessoryData.Exp = userAccessoryData.Exp
|
|
|
|
return has, &accessoryData
|
|
}
|
|
|
|
func (ss *Session) GetUserAccessoryInfoByUnitOwningUserID(unitOwningUserID int) (bool, *liverecordschema.AccessoryInfo) {
|
|
has, wearData := ss.GetUserAccessoryWearByUnitOwningUserID(unitOwningUserID)
|
|
if !has {
|
|
return false, nil
|
|
}
|
|
|
|
has, accessoryData := ss.GetAccessoryByAccessoryOwningUserID(wearData.AccessoryOwningUserID)
|
|
if !has {
|
|
return false, nil
|
|
}
|
|
|
|
return has, &liverecordschema.AccessoryInfo{
|
|
AccessoryOwningUserID: wearData.AccessoryOwningUserID,
|
|
AccessoryID: accessoryData.AccessoryID,
|
|
Exp: accessoryData.Exp,
|
|
NextExp: 0,
|
|
Level: accessoryData.MaxLevel,
|
|
MaxLevel: accessoryData.MaxLevel,
|
|
RankUpCount: accessoryData.MaxLevel - accessoryData.DefaultMaxLevel,
|
|
FavoriteFlag: true,
|
|
}
|
|
}
|