gopenfusion/db/players.go

package db

import (
	"database/sql"
	"fmt"

	"github.com/CPunch/GopenFusion/config"
	"github.com/CPunch/GopenFusion/protocol"
	"github.com/blockloop/scan"
)

type Player struct {
	PlayerID           int
	AccountID          int
	FirstName          string
	LastName           string
	NameCheck          int
	Slot               int
	Created            int
	LastLogin          int
	Level              int
	Nano1              int
	Nano2              int
	Nano3              int
	AppearanceFlag     int
	TutorialFlag       int
	PayZoneFlag        int
	XCoordinate        int
	YCoordinate        int
	ZCoordinate        int
	Angle              int
	HP                 int
	FusionMatter       int
	Taros              int
	BatteryW           int
	BatteryN           int
	Mentor             int
	CurrentMissionID   int
	WarpLocationFlag   int
	SkywayLocationFlag []byte
	FirstUseFlag       []byte
	Quests             []byte
	/* Appearance tbl */
	Body      int `db:"Body"`
	EyeColor  int `db:"EyeColor"`
	FaceStyle int `db:"FaceStyle"`
	Gender    int `db:"Gender"`
	HairColor int `db:"HairColor"`
	HairStyle int `db:"HairStyle"`
	Height    int `db:"Height"`
	SkinColor int `db:"SkinColor"`
}

// returns PlayerID, error
func (db *DBHandler) NewPlayer(AccountID int, FirstName, LastName string, slot int) (int, error) {
	nameCheck := 1 // for now, we approve all names
	QuestFlag := make([]byte, 128)
	SkywayLocationFlag := make([]byte, 16)
	FirstUseFlag := make([]byte, 16)

	var PlayerID int
	if err := db.Transaction(func(tx *sql.Tx) error {
		// create player
		row, err := tx.Query(
			"INSERT INTO Players (AccountID, Slot, FirstName, LastName, XCoordinate, YCoordinate, ZCoordinate, Angle, HP, NameCheck, Quests, SkywayLocationFlag, FirstUseFlag) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?) RETURNING PlayerID",
			AccountID, slot, FirstName, LastName, config.SPAWN_X, config.SPAWN_Y, config.SPAWN_Z, 0, config.GetMaxHP(1), nameCheck, QuestFlag, SkywayLocationFlag, FirstUseFlag)
		if err != nil {
			return err
		}

		if err := scan.Row(&PlayerID, row); err != nil {
			return err
		}

		// create appearance
		if _, err := tx.Exec("INSERT INTO Appearances (PlayerID) VALUES (?)", PlayerID); err != nil {
			return err
		}

		return nil
	}); err != nil {
		return -1, nil
	}

	return PlayerID, nil
}

// TODO: should this operate on the raw packet? should we do validation here or prior?
func (db *DBHandler) FinishPlayer(character *protocol.SP_CL2LS_REQ_CHAR_CREATE, AccountId int) error {
	return db.Transaction(func(tx *sql.Tx) error {
		// update AppearanceFlag
		_, err := tx.Exec("UPDATE Players SET AppearanceFlag = 1 WHERE PlayerID = ? AND AccountID = ? AND AppearanceFlag = 0", character.PCStyle.IPC_UID, AccountId)
		if err != nil {
			return err
		}

		// update Appearance
		_, err = tx.Exec("UPDATE Appearances SET Body = ?, EyeColor = ?, FaceStyle = ?, Gender = ?, HairColor = ?, HairStyle = ?, Height = ?, SkinColor = ? WHERE PlayerID = ?",
			character.PCStyle.IBody,
			character.PCStyle.IEyeColor,
			character.PCStyle.IFaceStyle,
			character.PCStyle.IGender,
			character.PCStyle.IHairColor,
			character.PCStyle.IHairStyle,
			character.PCStyle.IHeight,
			character.PCStyle.ISkinColor,
			character.PCStyle.IPC_UID)
		if err != nil {
			return err
		}

		// update Inventory
		items := [3]int16{character.SOn_Item.IEquipUBID, character.SOn_Item.IEquipLBID, character.SOn_Item.IEquipFootID}
		for i := 0; i < len(items); i++ {
			_, err = tx.Exec("INSERT INTO Inventory (PlayerID, Slot, ID, Type, Opt) VALUES (?, ?, ?, ?, 1)", character.PCStyle.IPC_UID, i+1, items[i], i+1)
			if err != nil {
				return err
			}
		}

		return nil
	})
}

func (db *DBHandler) FinishTutorial(PlayerID, AccountID int) error {
	_, err := db.Exec("UPDATE Players SET TutorialFlag = 1 WHERE PlayerID = ? AND AccountID = ? AND TutorialFlag = 0", PlayerID, AccountID)
	if err != nil {
		return err
	}

	// TODO: reference openfusion's finishTutorial for their academy specific patches
	return nil
}

// returns the deleted player row
func (db *DBHandler) DeletePlayer(PlayerID, AccountID int) (*Player, error) {
	row, err := db.Query("DELETE FROM Players WHERE AccountID = ? AND PlayerID = ? RETURNING *")
	if err != nil {
		return nil, err
	}

	var plr Player
	if err := scan.Row(&plr, row); err != nil {
		return nil, err
	}

	return &plr, nil
}

func (db *DBHandler) GetPlayer(PlayerID int) (*Player, error) {
	rows, err := db.Query(`SELECT
			p.PlayerID, p.AccountID, p.Slot, p.FirstName, p.LastName,
			p.Level, p.Nano1, p.Nano2, p.Nano3,
			p.AppearanceFlag, p.TutorialFlag, p.PayZoneFlag,
			p.XCoordinate, p.YCoordinate, p.ZCoordinate, p.NameCheck,
			p.Angle, p.HP, p.FusionMatter, p.Taros, p.Quests,
			p.BatteryW, p.BatteryN, p.Mentor, p.WarpLocationFlag,
			p.SkywayLocationFlag, p.CurrentMissionID, p.FirstUseFlag,
			a.Body, a.EyeColor, a.FaceStyle, a.Gender, a.HairColor, a.HairStyle,
			a.Height, a.SkinColor
		FROM Players as p
		INNER JOIN Appearances as a ON p.PlayerID = a.PlayerID
		WHERE p.PlayerID = ?`, PlayerID)
	if err != nil {
		return nil, err
	}

	fmt.Print(PlayerID)
	var plr Player
	for rows.Next() {
		if err := rows.Scan(
			&plr.PlayerID, &plr.AccountID, &plr.Slot, &plr.FirstName, &plr.LastName,
			&plr.Level, &plr.Nano1, &plr.Nano2, &plr.Nano3,
			&plr.AppearanceFlag, &plr.TutorialFlag, &plr.PayZoneFlag,
			&plr.XCoordinate, &plr.YCoordinate, &plr.ZCoordinate, &plr.NameCheck,
			&plr.Angle, &plr.HP, &plr.FusionMatter, &plr.Taros, &plr.Quests,
			&plr.BatteryW, &plr.BatteryN, &plr.Mentor, &plr.WarpLocationFlag,
			&plr.SkywayLocationFlag, &plr.CurrentMissionID, &plr.FirstUseFlag,
			&plr.Body, &plr.EyeColor, &plr.FaceStyle, &plr.Gender, &plr.HairColor, &plr.HairStyle,
			&plr.Height, &plr.SkinColor); err != nil {
			panic(err)
		}
	}

	return &plr, nil
}

func (db *DBHandler) GetPlayers(AccountID int) ([]Player, error) {
	rows, err := db.Query(`SELECT
			p.PlayerID, p.AccountID, p.Slot, p.FirstName, p.LastName,
			p.Level, p.Nano1, p.Nano2, p.Nano3,
			p.AppearanceFlag, p.TutorialFlag, p.PayZoneFlag,
			p.XCoordinate, p.YCoordinate, p.ZCoordinate, p.NameCheck,
			p.Angle, p.HP, p.FusionMatter, p.Taros, p.Quests,
			p.BatteryW, p.BatteryN, p.Mentor, p.WarpLocationFlag,
			p.SkywayLocationFlag, p.CurrentMissionID, p.FirstUseFlag,
			a.Body, a.EyeColor, a.FaceStyle, a.Gender, a.HairColor, a.HairStyle,
			a.Height, a.SkinColor
		FROM Players as p
		INNER JOIN Appearances as a ON p.PlayerID = a.PlayerID
		WHERE p.AccountID = ?`, AccountID)
	if err != nil {
		return nil, err
	}

	var plrs []Player
	for rows.Next() {
		plr := Player{}

		if err := rows.Scan(
			&plr.PlayerID, &plr.AccountID, &plr.Slot, &plr.FirstName, &plr.LastName,
			&plr.Level, &plr.Nano1, &plr.Nano2, &plr.Nano3,
			&plr.AppearanceFlag, &plr.TutorialFlag, &plr.PayZoneFlag,
			&plr.XCoordinate, &plr.YCoordinate, &plr.ZCoordinate, &plr.NameCheck,
			&plr.Angle, &plr.HP, &plr.FusionMatter, &plr.Taros, &plr.Quests,
			&plr.BatteryW, &plr.BatteryN, &plr.Mentor, &plr.WarpLocationFlag,
			&plr.SkywayLocationFlag, &plr.CurrentMissionID, &plr.FirstUseFlag,
			&plr.Body, &plr.EyeColor, &plr.FaceStyle, &plr.Gender, &plr.HairColor, &plr.HairStyle,
			&plr.Height, &plr.SkinColor); err != nil {
			return nil, err
		}

		plrs = append(plrs, plr)
	}

	return plrs, nil
}