d2go/pkg/data/items.go

package data

import (
	"strings"

	"github.com/hectorgimenez/d2go/pkg/data/item"
	"github.com/hectorgimenez/d2go/pkg/data/stat"
)

type Inventory struct {
	Belt             Belt
	AllItems         []Item
	Gold             int
	StashedGold      [6]int // [0]=personal, [1..5]=shared stash pages
	SharedStashPages int    // Number of shared stash units detected (3=non-DLC, 5+=DLC)
}

func (i Inventory) Find(name item.Name, locations ...item.LocationType) (Item, bool) {
	for _, it := range i.AllItems {
		if strings.EqualFold(string(it.Name), string(name)) {
			// If no locations are specified, return the first item found
			if len(locations) == 0 {
				return it, true
			}

			for _, l := range locations {
				if it.Location.LocationType == l {
					return it, true
				}
			}
		}
	}

	return Item{}, false
}

func (i Inventory) FindByID(unitID UnitID) (Item, bool) {
	for _, it := range i.AllItems {
		if it.UnitID == unitID {
			return it, true
		}
	}

	return Item{}, false
}

func (i Inventory) ByLocation(locations ...item.LocationType) []Item {
	var items []Item

	for _, it := range i.AllItems {
		for _, l := range locations {
			if it.Location.LocationType == l {
				items = append(items, it)
			}
		}
	}

	return items
}

func (i Inventory) Matrix() [4][10]bool {
	invMatrix := [4][10]bool{} // false = empty, true = occupied
	for _, itm := range i.ByLocation(item.LocationInventory) {
		for k := range itm.Desc().InventoryWidth {
			for j := range itm.Desc().InventoryHeight {
				invMatrix[itm.Position.Y+j][itm.Position.X+k] = true
			}
		}
	}

	return invMatrix
}

type UnitID int

type ItemAffixes struct {
	Rare struct {
		Prefix int16
		Suffix int16
	}
	Magic struct {
		Prefixes [3]int16 // Prefix1, Prefix2, Prefix3
		Suffixes [3]int16 // Suffix1, Suffix2, Suffix3
	}
}

type Item struct {
	ID int
	UnitID
	Name                 item.Name
	Quality              item.Quality
	IdentifiedName       string
	RunewordName         item.RunewordName
	LevelReq             int
	Position             Position
	Location             item.Location
	Ethereal             bool
	IsHovered            bool
	BaseStats            stat.Stats
	Stats                stat.Stats
	Affixes              ItemAffixes
	Sockets              []Item
	Identified           bool
	IsRuneword           bool
	IsNamed              bool
	IsStartItem          bool
	IsEar                bool
	IsBroken             bool
	HasBeenEquipped      bool
	HasSockets           bool
	InTradeOrStoreScreen bool
	IsInSocket           bool
	UniqueSetID          int32
	StackedQuantity      int
}

type Drop struct {
	Item         Item
	Rule         string
	RuleFile     string
	DropLocation string
}

func (i Item) Desc() item.Description {
	return item.Desc[i.ID]
}

func (i Item) Type() item.Type {
	return i.Desc().GetType()
}

func (i Item) IsPotion() bool {
	return i.IsHealingPotion() || i.IsManaPotion() || i.IsRejuvPotion()
}

func (i Item) IsHealingPotion() bool {
	return i.Type().IsType(item.TypeHealingPotion)
}

func (i Item) IsManaPotion() bool {
	return i.Type().IsType(item.TypeManaPotion)
}

func (i Item) IsRejuvPotion() bool {
	return i.Type().IsType(item.TypeRejuvPotion)
}

func (i Item) IsFromQuest() bool {
	return i.Type().IsType(item.TypeQuest)
}

func (i Item) FindStat(id stat.ID, layer int) (stat.Data, bool) {
	st, found := i.Stats.FindStat(id, layer)
	if found {
		return st, true
	}

	return i.BaseStats.FindStat(id, layer)
}

func (i Item) HasPrefix(id int16) bool {
	// Check rare prefix
	if i.Affixes.Rare.Prefix == id {
		return true
	}

	// Check magic prefixes
	for _, prefix := range i.Affixes.Magic.Prefixes {
		if prefix == id {
			return true
		}
	}
	return false
}

func (i Item) HasSuffix(id int16) bool {
	// Check rare suffix
	if i.Affixes.Rare.Suffix == id {
		return true
	}

	// Check magic suffixes
	for _, suffix := range i.Affixes.Magic.Suffixes {
		if suffix == id {
			return true
		}
	}
	return false
}

func (a ItemAffixes) GetRarePrefix() (item.RarePrefix, bool) {
	prefix, exists := item.RarePrefixDesc[int(a.Rare.Prefix)]
	return prefix, exists
}

func (a ItemAffixes) GetRareSuffix() (item.RareSuffix, bool) {
	suffix, exists := item.RareSuffixDesc[int(a.Rare.Suffix)]
	return suffix, exists
}

func (a ItemAffixes) GetMagicPrefixes() []item.MagicPrefix {
	prefixes := make([]item.MagicPrefix, 0, 3)
	for _, id := range a.Magic.Prefixes {
		if prefix, exists := item.MagicPrefixDesc[int(id)]; exists && id != 0 {
			prefixes = append(prefixes, prefix)
		}
	}
	return prefixes
}

func (a ItemAffixes) GetMagicSuffixes() []item.MagicSuffix {
	suffixes := make([]item.MagicSuffix, 0, 3)
	for _, id := range a.Magic.Suffixes {
		if suffix, exists := item.MagicSuffixDesc[int(id)]; exists && id != 0 {
			suffixes = append(suffixes, suffix)
		}
	}
	return suffixes
}

func (i Item) GetSocketedItems() []Item {
	return i.Sockets
}
func (i Item) HasSocketedItems() bool {
	return len(i.Sockets) > 0
}