aboutsummaryrefslogtreecommitdiff
path: root/internal/object/item.go
blob: b0865d6bd578b1d659dbc014f47327b8a272b1fb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
package object

import (
	"fmt"
	"strings"

	"gopkg.in/yaml.v3"
)

func WordPrefixMatch(input, name string) bool {
	inputWords := strings.Fields(strings.ToLower(input))
	if len(inputWords) == 0 {
		return false
	}
	nameWords := strings.Fields(strings.ToLower(name))
	for _, iw := range inputWords {
		found := false
		for _, nw := range nameWords {
			if strings.HasPrefix(nw, iw) || strings.HasPrefix(iw, nw) {
				found = true
				break
			}
		}
		if !found {
			return false
		}
	}
	return true
}

type EquipSlot string

const (
	SlotHead      EquipSlot = "head"
	SlotNeck      EquipSlot = "neck"
	SlotTorso     EquipSlot = "torso"
	SlotLegs      EquipSlot = "legs"
	SlotHands     EquipSlot = "hands"
	SlotFeet      EquipSlot = "feet"
	SlotBack      EquipSlot = "back"
	SlotAmmo      EquipSlot = "ammo"
	SlotMainHand  EquipSlot = "main_hand"
	SlotOffHand   EquipSlot = "off_hand"
	SlotRing      EquipSlot = "ring"
)

type WeaponType string

const (
	WeaponMelee      WeaponType = "melee"
	WeaponRanged     WeaponType = "ranged"
	WeaponScience    WeaponType = "science"
)

type ItemDef struct {
	ID          string         `yaml:"id"`
	Name        string         `yaml:"name"`
	Color       string         `yaml:"color"`
	Description string         `yaml:"description"`
	Value       int            `yaml:"value"`
	Stackable   bool           `yaml:"stackable"`
	EquipSlot   EquipSlot      `yaml:"equip_slot"`
	WeaponType  WeaponType     `yaml:"weapon_type"`
	AttackType  string         `yaml:"attack_type"`
	Stats       ItemStats      `yaml:"stats"`
	Speed       float64        `yaml:"speed"`
	Requirements map[string]int `yaml:"requirements,omitempty"`
	ToolType    string         `yaml:"tool_type"`
	ToolSpeed   float64        `yaml:"tool_speed"`
	BurnTicks   float64        `yaml:"burn_ticks"`
	FireLevel   int            `yaml:"fire_level"`
	FireXP      int            `yaml:"fire_xp"`
	Quality     int            `yaml:"quality"`
	MaxQuality  int            `yaml:"max_quality"`
	SearchTable     string     `yaml:"search_table"`
	SearchMiscTable string     `yaml:"search_misc_table"`
	SearchTicks     float64    `yaml:"search_ticks"`
	SearchMessage   string     `yaml:"search_message"`
	HealValue       int        `yaml:"heal_value"`
	EatMessage      string     `yaml:"eat_message"`
	Craft           CraftList       `yaml:"craft,omitempty"`
	ProvidesJunk    string          `yaml:"provides_junk,omitempty"`
	FarmPatchType   string          `yaml:"farm_patch_type"`
	FarmLevel       int             `yaml:"farm_level"`
	FarmPlantXP     int             `yaml:"farm_plant_xp"`
	FarmHarvestXP   int             `yaml:"farm_harvest_xp"`
	FarmStages      int             `yaml:"farm_stages"`
	FarmProduct     string          `yaml:"farm_product"`
	FarmMinYield    int             `yaml:"farm_min_yield"`
	FarmMaxYield    int             `yaml:"farm_max_yield"`
	PotionEffect   string          `yaml:"potion_effect"`
	PotionBonus    int             `yaml:"potion_bonus"`
	PotionDuration float64         `yaml:"potion_duration"`
}

type CraftList []CraftDef

func (cl *CraftList) UnmarshalYAML(value *yaml.Node) error {
	switch value.Kind {
	case yaml.SequenceNode:
		return value.Decode((*[]CraftDef)(cl))
	case yaml.MappingNode:
		var single CraftDef
		if err := value.Decode(&single); err != nil {
			return err
		}
		*cl = []CraftDef{single}
		return nil
	}
	return fmt.Errorf("craft: expected mapping or sequence")
}

func (d *ItemDef) FirstCraft() *CraftDef {
	if len(d.Craft) == 0 {
		return nil
	}
	return &d.Craft[0]
}

func (d *ItemDef) FindCraft(fn func(CraftDef) bool) *CraftDef {
	for i := range d.Craft {
		if fn(d.Craft[i]) {
			return &d.Craft[i]
		}
	}
	return nil
}

func (d *ItemDef) MatchingCrafts(fn func(CraftDef) bool) []*CraftDef {
	var result []*CraftDef
	for i := range d.Craft {
		if fn(d.Craft[i]) {
			result = append(result, &d.Craft[i])
		}
	}
	return result
}

type ConsumeEntry struct {
	Items      []string `yaml:"items"`
	Quantity   int      `yaml:"quantity"`
	Byproducts []string `yaml:"byproducts"`
}

type CraftStep struct {
	Tick    float64 `yaml:"tick"`
	Message string  `yaml:"message"`
}

type SuccessFormula struct {
	Base     float64 `yaml:"base"`
	PerLevel float64 `yaml:"per_level"`
	Cap      float64 `yaml:"cap"`
}

type CraftDef struct {
	Type         string            `yaml:"type"`
	Skill        string            `yaml:"skill"`
	Level        int               `yaml:"level"`
	XP           int               `yaml:"xp"`
	Wait         float64           `yaml:"wait"`
	Station      []string          `yaml:"station"`
	Tool         string            `yaml:"tool"`
	Consume      []ConsumeEntry    `yaml:"consume"`
	OutputQty    int               `yaml:"output_qty"`
	Fail         string            `yaml:"fail"`
	Message      string            `yaml:"message"`
	FailMessage  string            `yaml:"fail_message"`
	StartMessage string            `yaml:"start_message"`
	EndMessage   string            `yaml:"end_message"`
	Steps        []CraftStep       `yaml:"steps"`
	Success      *SuccessFormula   `yaml:"success"`
}

func (c *CraftDef) EffectiveSkill() string {
	if c.Skill != "" {
		return c.Skill
	}
	return c.Type
}

type ItemStats struct {
	StabAttack    int `yaml:"stab_attack"`
	SlashAttack   int `yaml:"slash_attack"`
	CrushAttack   int `yaml:"crush_attack"`
	ScienceAttack int `yaml:"science_attack"`
	RangedAttack  int `yaml:"ranged_attack"`

	StabDefense    int `yaml:"stab_defense"`
	SlashDefense   int `yaml:"slash_defense"`
	CrushDefense   int `yaml:"crush_defense"`
	ScienceDefense int `yaml:"science_defense"`
	RangedDefense  int `yaml:"ranged_defense"`

	StrengthBonus   int `yaml:"strength_bonus"`
	RangedStrength  int `yaml:"ranged_strength"`
	ScienceDamage   int `yaml:"science_damage"`
	TechnologyBonus int `yaml:"technology_bonus"`
}

func (d *ItemDef) MatchesName(input string) bool {
	lower := strings.ToLower(strings.TrimSpace(input))
	if lower == "" {
		return false
	}
	if strings.ToLower(d.Name) == lower {
		return true
	}
	return WordPrefixMatch(lower, d.Name)
}