gopenfusion/db/inventory.go

package db

import (
	"database/sql"

	"github.com/CPunch/gopenfusion/protocol"
)

type Inventory struct {
	PlayerID  int
	Slot      int
	ID        int
	Type      int
	Opt       int
	TimeLimit int
}

// start && end are both inclusive
func (db *DBHandler) GetPlayerInventorySlots(PlayerID int, start int, end int) ([]protocol.SItemBase, error) {
	rows, err := db.Query("SELECT PlayerID, Slot, ID, Type, Opt, TimeLimit FROM Inventory WHERE Slot BETWEEN ? AND ? AND PlayerID = ?", start, end, PlayerID)
	if err != nil {
		return nil, err
	}

	var inven []Inventory
	for rows.Next() {
		item := Inventory{}

		if err := rows.Scan(
			&item.PlayerID, &item.Slot, &item.ID, &item.Type, &item.Opt, &item.TimeLimit); err != nil {
			return nil, err
		}

		inven = append(inven, item)
	}

	// populate an SItemBase array
	items := make([]protocol.SItemBase, end-start)
	for _, item := range inven {
		items[item.Slot-start] = protocol.SItemBase{
			IID:        int16(item.ID),
			IType:      int16(item.Type),
			IOpt:       int32(item.Opt),
			ITimeLimit: int32(item.TimeLimit),
		}
	}
	return items, nil
}

// start is inclusive
func (db *DBHandler) SetPlayerInventorySlots(PlayerID int, start int, items []protocol.SItemBase) error {
	return db.Transaction(func(tx *sql.Tx) error {
		// delete inventory slots
		_, err := db.Query("DELETE FROM Inventory WHERE Slot BETWEEN ? AND ? AND PlayerID = ?", start, start+len(items)-1, PlayerID)
		if err != nil {
			return err
		}

		// insert inventory
		for i, item := range items {
			if item.IID != 0 {
				_, err := db.Query("INSERT INTO Inventory (PlayerID, Slot, ID, Type, Opt, TimeLimit) VALUES (?, ?, ?, ?, ?, ?)", PlayerID, start+i, item.IID, item.IType, item.IOpt, item.ITimeLimit)
				if err != nil {
					return err
				}
			}
		}

		return nil
	})
}