mirror of
https://github.com/Mezeporta/Erupe.git
synced 2026-03-21 23:22:34 +01:00
RE'd putDisplayed_achievement from ZZ client DLL via Ghidra: the packet sends opcode + 1 zero byte with no achievement ID, acting as a blanket "I saw everything" signal. Server changes: - Track per-character last-displayed levels in new displayed_levels column (migration 0008) - GetAchievement compares current vs displayed levels per entry - DisplayedAchievement snapshots current levels to clear notifications - Repo, service, mock, and 3 new service tests Protbot changes: - New --action achievement: fetches achievements, shows rank-up markers, sends DISPLAYED_ACHIEVEMENT, re-fetches to verify notifications clear - Packet builders for GET/ADD/DISPLAYED_ACHIEVEMENT
60 lines
2.5 KiB
Go
60 lines
2.5 KiB
Go
package channelserver
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
// AchievementRepository centralizes all database access for the achievements table.
|
|
type AchievementRepository struct {
|
|
db *sqlx.DB
|
|
}
|
|
|
|
// NewAchievementRepository creates a new AchievementRepository.
|
|
func NewAchievementRepository(db *sqlx.DB) *AchievementRepository {
|
|
return &AchievementRepository{db: db}
|
|
}
|
|
|
|
// EnsureExists creates an achievements record for the character if one doesn't exist.
|
|
func (r *AchievementRepository) EnsureExists(charID uint32) error {
|
|
_, err := r.db.Exec("INSERT INTO achievements (id) VALUES ($1) ON CONFLICT DO NOTHING", charID)
|
|
return err
|
|
}
|
|
|
|
// GetAllScores returns all 33 achievement scores for a character.
|
|
func (r *AchievementRepository) GetAllScores(charID uint32) ([33]int32, error) {
|
|
var scores [33]int32
|
|
err := r.db.QueryRow("SELECT * FROM achievements WHERE id=$1", charID).Scan(&scores[0],
|
|
&scores[0], &scores[1], &scores[2], &scores[3], &scores[4], &scores[5], &scores[6], &scores[7], &scores[8],
|
|
&scores[9], &scores[10], &scores[11], &scores[12], &scores[13], &scores[14], &scores[15], &scores[16],
|
|
&scores[17], &scores[18], &scores[19], &scores[20], &scores[21], &scores[22], &scores[23], &scores[24],
|
|
&scores[25], &scores[26], &scores[27], &scores[28], &scores[29], &scores[30], &scores[31], &scores[32])
|
|
return scores, err
|
|
}
|
|
|
|
// IncrementScore increments the score for a specific achievement column.
|
|
// achievementID must be in the range [0, 32] to prevent SQL injection.
|
|
func (r *AchievementRepository) IncrementScore(charID uint32, achievementID uint8) error {
|
|
if achievementID > 32 {
|
|
return fmt.Errorf("achievement ID %d out of range [0, 32]", achievementID)
|
|
}
|
|
_, err := r.db.Exec(fmt.Sprintf("UPDATE achievements SET ach%d=ach%d+1 WHERE id=$1", achievementID, achievementID), charID)
|
|
return err
|
|
}
|
|
|
|
// GetDisplayedLevels returns the last-displayed achievement levels for a character.
|
|
// Returns nil if never displayed (all rank-ups should trigger notifications).
|
|
func (r *AchievementRepository) GetDisplayedLevels(charID uint32) ([]byte, error) {
|
|
var levels []byte
|
|
err := r.db.QueryRow("SELECT displayed_levels FROM achievements WHERE id=$1", charID).Scan(&levels)
|
|
return levels, err
|
|
}
|
|
|
|
// SaveDisplayedLevels stores the current achievement levels as "displayed",
|
|
// so future GET_ACHIEVEMENT responses only notify on new rank-ups.
|
|
func (r *AchievementRepository) SaveDisplayedLevels(charID uint32, levels []byte) error {
|
|
_, err := r.db.Exec("UPDATE achievements SET displayed_levels=$1 WHERE id=$2", levels, charID)
|
|
return err
|
|
}
|