guild_member service created in /internal/service

internal/service/guild_member.go

@@ -0,0 +1,169 @@
+package service
+
+import (
+	"erupe-ce/utils/db"
+	"erupe-ce/utils/logger"
+	"fmt"
+	"time"
+
+	"github.com/jmoiron/sqlx"
+	"go.uber.org/zap"
+)
+
+type GuildMember struct {
+	GuildID         uint32     `db:"guild_id"`
+	CharID          uint32     `db:"character_id"`
+	JoinedAt        *time.Time `db:"joined_at"`
+	Souls           uint32     `db:"souls"`
+	RPToday         uint16     `db:"rp_today"`
+	RPYesterday     uint16     `db:"rp_yesterday"`
+	Name            string     `db:"name"`
+	IsApplicant     bool       `db:"is_applicant"`
+	OrderIndex      uint16     `db:"order_index"`
+	LastLogin       uint32     `db:"last_login"`
+	Recruiter       bool       `db:"recruiter"`
+	AvoidLeadership bool       `db:"avoid_leadership"`
+	IsLeader        bool       `db:"is_leader"`
+	HR              uint16     `db:"hr"`
+	GR              uint16     `db:"gr"`
+	WeaponID        uint16     `db:"weapon_id"`
+	WeaponType      uint8      `db:"weapon_type"`
+}
+
+func (gm *GuildMember) CanRecruit() bool {
+	if gm.Recruiter {
+		return true
+	}
+	if gm.OrderIndex <= 3 {
+		return true
+	}
+	if gm.IsLeader {
+		return true
+	}
+	return false
+}
+
+func (gm *GuildMember) IsSubLeader() bool {
+	return gm.OrderIndex <= 3
+}
+
+func (gm *GuildMember) Save() error {
+	db, err := db.GetDB()
+	logger := logger.Get()
+
+	if err != nil {
+		logger.Fatal(fmt.Sprintf("Failed to get database instance: %s", err))
+	}
+	_, err = db.Exec("UPDATE guild_characters SET avoid_leadership=$1, order_index=$2 WHERE character_id=$3", gm.AvoidLeadership, gm.OrderIndex, gm.CharID)
+
+	if err != nil {
+		logger.Error(
+			"failed to update guild member data",
+			zap.Error(err),
+			zap.Uint32("charID", gm.CharID),
+			zap.Uint32("guildID", gm.GuildID),
+		)
+		return err
+	}
+	return nil
+}
+
+const guildMembersSelectSQL = `
+SELECT * FROM (
+	SELECT
+		g.id AS guild_id,
+		joined_at,
+		COALESCE((SELECT SUM(souls) FROM festa_submissions fs WHERE fs.character_id=c.id), 0) AS souls,
+		COALESCE(rp_today, 0) AS rp_today,
+		COALESCE(rp_yesterday, 0) AS rp_yesterday,
+		c.name,
+		c.id AS character_id,
+		COALESCE(order_index, 0) AS order_index,
+		c.last_login,
+		COALESCE(recruiter, false) AS recruiter,
+		COALESCE(avoid_leadership, false) AS avoid_leadership,
+		c.hr,
+		c.gr,
+		c.weapon_id,
+		c.weapon_type,
+		EXISTS(SELECT 1 FROM guild_applications ga WHERE ga.character_id=c.id AND application_type='applied') AS is_applicant,
+		CASE WHEN g.leader_id = c.id THEN true ELSE false END AS is_leader
+		FROM guild_characters gc
+		LEFT JOIN characters c ON c.id = gc.character_id
+		LEFT JOIN guilds g ON g.id = gc.guild_id
+) AS subquery
+`
+
+func GetGuildMembers(guildID uint32, applicants bool) ([]*GuildMember, error) {
+	db, err := db.GetDB()
+	logger := logger.Get()
+
+	if err != nil {
+		logger.Fatal(fmt.Sprintf("Failed to get database instance: %s", err))
+	}
+	rows, err := db.Queryx(fmt.Sprintf(`
+			%s
+			WHERE guild_id = $1 AND is_applicant = $2
+	`, guildMembersSelectSQL), guildID, applicants)
+
+	if err != nil {
+		logger.Error("failed to retrieve membership data for guild", zap.Error(err), zap.Uint32("guildID", guildID))
+		return nil, err
+	}
+
+	defer rows.Close()
+
+	members := make([]*GuildMember, 0)
+
+	for rows.Next() {
+		member, err := buildGuildMemberObjectFromDBResult(rows, err)
+
+		if err != nil {
+			return nil, err
+		}
+
+		members = append(members, member)
+	}
+
+	return members, nil
+}
+
+func GetCharacterGuildData(charID uint32) (*GuildMember, error) {
+	db, err := db.GetDB()
+	logger := logger.Get()
+
+	if err != nil {
+		logger.Fatal(fmt.Sprintf("Failed to get database instance: %s", err))
+	}
+	rows, err := db.Queryx(fmt.Sprintf("%s	WHERE character_id=$1", guildMembersSelectSQL), charID)
+
+	if err != nil {
+		logger.Error(fmt.Sprintf("failed to retrieve membership data for character '%d'", charID))
+		return nil, err
+	}
+
+	defer rows.Close()
+
+	hasRow := rows.Next()
+
+	if !hasRow {
+		return nil, nil
+	}
+
+	return buildGuildMemberObjectFromDBResult(rows, err)
+}
+
+func buildGuildMemberObjectFromDBResult(rows *sqlx.Rows, err error) (*GuildMember, error) {
+	logger := logger.Get()
+
+	memberData := &GuildMember{}
+
+	err = rows.StructScan(&memberData)
+
+	if err != nil {
+		logger.Error("failed to retrieve guild data from database", zap.Error(err))
+		return nil, err
+	}
+
+	return memberData, nil
+}