1use common_base::{enum_iter, struct_iter};
2use common_i18n::Content;
3use rand::{seq::SliceRandom, thread_rng};
4use serde::{Deserialize, Serialize};
5use strum::{Display, EnumString};
6
7struct_iter! {
8 #[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
9 pub struct Body {
10 pub species: Species,
11 pub body_type: BodyType,
12 }
13}
14
15impl Body {
16 pub fn random() -> Self {
17 let mut rng = thread_rng();
18 let species = *ALL_SPECIES.choose(&mut rng).unwrap();
19 Self::random_with(&mut rng, &species)
20 }
21
22 #[inline]
23 pub fn random_with(rng: &mut impl rand::Rng, &species: &Species) -> Self {
24 let body_type = *ALL_BODY_TYPES.choose(rng).unwrap();
25 Self { species, body_type }
26 }
27
28 pub fn localize_npc(&self) -> Option<Content> {
33 let key = match &self.species {
34 Species::Ogre => match self.body_type {
35 BodyType::Male => "body-npc-speech-biped_large-ogre-male",
36 BodyType::Female => "body-npc-speech-biped_large-ogre-female",
37 },
38 Species::Cyclops => "body-npc-speech-biped_large-cyclops",
39 Species::Wendigo => "body-npc-speech-biped_large-wendigo",
40 Species::Werewolf => "body-npc-speech-biped_large-werewolf",
41 Species::Cavetroll => "body-npc-speech-biped_large-cave_troll",
42 Species::Mountaintroll => "body-npc-speech-biped_large-mountain_troll",
43 Species::Swamptroll => "body-npc-speech-biped_large-swamp_troll",
44 Species::Blueoni => "body-npc-speech-biped_large-blue_oni",
45 Species::Redoni => "body-npc-speech-biped_large-red_oni",
46 Species::Tursus => "body-npc-speech-biped_large-tursus",
47 Species::Dullahan => "body-npc-speech-biped_large-dullahan",
48 Species::Occultsaurok => "body-npc-speech-biped_large-occult_saurok",
49 Species::Mightysaurok => "body-npc-speech-biped_large-mighty_saurok",
50 Species::Slysaurok => "body-npc-speech-biped_large-sly_saurok",
51 Species::Mindflayer => "body-npc-speech-biped_large-mindflayer",
52 Species::Minotaur => "body-npc-speech-biped_large-minotaur",
53 Species::Tidalwarrior => "body-npc-speech-biped_large-tidal_warrior",
54 Species::Yeti => "body-npc-speech-biped_large-yeti",
55 Species::Harvester => "body-npc-speech-biped_large-harvester",
56 Species::Cultistwarlord => "body-npc-speech-biped_large-cultist_warlord",
57 Species::Cultistwarlock => "body-npc-speech-biped_large-cultist_warlock",
58 Species::Huskbrute => "body-npc-speech-biped_large-husk_brute",
59 Species::Gigasfrost => "body-npc-speech-biped_large-gigas_frost",
60 Species::Gigasfire => "body-npc-speech-biped_large-gigas_fire",
61 Species::AdletElder => "body-npc-speech-biped_large-adlet_elder",
62 Species::SeaBishop => "body-npc-speech-biped_large-sea_bishop",
63 Species::HaniwaGeneral => "body-npc-speech-biped_large-haniwa_general",
64 Species::TerracottaBesieger => "body-npc-speech-biped_large-terracotta_besieger",
65 Species::TerracottaDemolisher => "body-npc-speech-biped_large-terracotta_demolisher",
66 Species::TerracottaPunisher => "body-npc-speech-biped_large-terracotta_punisher",
67 Species::TerracottaPursuer => "body-npc-speech-biped_large-terracotta_pursuer",
68 Species::Cursekeeper => "body-npc-speech-biped_large-cursekeeper",
69 Species::Forgemaster => "body-npc-speech-biped_large-forgemaster",
70 Species::Strigoi => "body-npc-speech-biped_large-strigoi",
71 Species::Executioner => "body-npc-speech-biped_large-executioner",
72 };
73
74 Some(Content::localized(key))
75 }
76}
77
78impl From<Body> for super::Body {
79 fn from(body: Body) -> Self { super::Body::BipedLarge(body) }
80}
81
82enum_iter! {
83 ~const_array(ALL)
84 #[derive(
85 Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize)]
86 #[repr(u32)]
87 pub enum Species {
88 Ogre = 0,
89 Cyclops = 1,
90 Wendigo = 2,
91 Cavetroll = 3,
92 Mountaintroll = 4,
93 Swamptroll = 5,
94 Dullahan = 6,
95 Werewolf = 7,
96 Occultsaurok = 8,
97 Mightysaurok = 9,
98 Slysaurok = 10,
99 Mindflayer = 11,
100 Minotaur = 12,
101 Tidalwarrior = 13,
102 Yeti = 14,
103 Harvester = 15,
104 Blueoni = 16,
105 Redoni = 17,
106 Cultistwarlord = 18,
107 Cultistwarlock = 19,
108 Huskbrute = 20,
109 Tursus = 21,
110 Gigasfrost = 22,
111 AdletElder = 23,
112 SeaBishop = 24,
113 HaniwaGeneral = 25,
114 TerracottaBesieger = 26,
115 TerracottaDemolisher = 27,
116 TerracottaPunisher = 28,
117 TerracottaPursuer = 29,
118 Cursekeeper = 30,
119 Forgemaster = 31,
120 Strigoi = 32,
121 Executioner = 33,
122 Gigasfire = 34,
123 }
124}
125
126#[derive(Clone, Debug, Deserialize, Serialize)]
128pub struct AllSpecies<SpeciesMeta> {
129 pub ogre: SpeciesMeta,
130 pub cyclops: SpeciesMeta,
131 pub wendigo: SpeciesMeta,
132 pub troll_cave: SpeciesMeta,
133 pub troll_mountain: SpeciesMeta,
134 pub troll_swamp: SpeciesMeta,
135 pub dullahan: SpeciesMeta,
136 pub werewolf: SpeciesMeta,
137 pub saurok_occult: SpeciesMeta,
138 pub saurok_mighty: SpeciesMeta,
139 pub saurok_sly: SpeciesMeta,
140 pub mindflayer: SpeciesMeta,
141 pub minotaur: SpeciesMeta,
142 pub tidalwarrior: SpeciesMeta,
143 pub yeti: SpeciesMeta,
144 pub harvester: SpeciesMeta,
145 pub oni_blue: SpeciesMeta,
146 pub oni_red: SpeciesMeta,
147 pub cultist_warlord: SpeciesMeta,
148 pub cultist_warlock: SpeciesMeta,
149 pub husk_brute: SpeciesMeta,
150 pub tursus: SpeciesMeta,
151 pub gigas_frost: SpeciesMeta,
152 pub adlet_elder: SpeciesMeta,
153 pub sea_bishop: SpeciesMeta,
154 pub haniwa_general: SpeciesMeta,
155 pub terracotta_besieger: SpeciesMeta,
156 pub terracotta_demolisher: SpeciesMeta,
157 pub terracotta_punisher: SpeciesMeta,
158 pub terracotta_pursuer: SpeciesMeta,
159 pub cursekeeper: SpeciesMeta,
160 pub forgemaster: SpeciesMeta,
161 pub strigoi: SpeciesMeta,
162 pub executioner: SpeciesMeta,
163 pub gigas_fire: SpeciesMeta,
164}
165
166impl<'a, SpeciesMeta> core::ops::Index<&'a Species> for AllSpecies<SpeciesMeta> {
167 type Output = SpeciesMeta;
168
169 #[inline]
170 fn index(&self, &index: &'a Species) -> &Self::Output {
171 match index {
172 Species::Ogre => &self.ogre,
173 Species::Cyclops => &self.cyclops,
174 Species::Wendigo => &self.wendigo,
175 Species::Cavetroll => &self.troll_cave,
176 Species::Mountaintroll => &self.troll_mountain,
177 Species::Swamptroll => &self.troll_swamp,
178 Species::Dullahan => &self.dullahan,
179 Species::Werewolf => &self.werewolf,
180 Species::Occultsaurok => &self.saurok_occult,
181 Species::Mightysaurok => &self.saurok_mighty,
182 Species::Slysaurok => &self.saurok_sly,
183 Species::Mindflayer => &self.mindflayer,
184 Species::Minotaur => &self.minotaur,
185 Species::Tidalwarrior => &self.tidalwarrior,
186 Species::Yeti => &self.yeti,
187 Species::Harvester => &self.harvester,
188 Species::Blueoni => &self.oni_blue,
189 Species::Redoni => &self.oni_red,
190 Species::Cultistwarlord => &self.cultist_warlord,
191 Species::Cultistwarlock => &self.cultist_warlock,
192 Species::Huskbrute => &self.husk_brute,
193 Species::Tursus => &self.tursus,
194 Species::Gigasfrost => &self.gigas_frost,
195 Species::AdletElder => &self.adlet_elder,
196 Species::SeaBishop => &self.sea_bishop,
197 Species::HaniwaGeneral => &self.haniwa_general,
198 Species::TerracottaBesieger => &self.terracotta_besieger,
199 Species::TerracottaDemolisher => &self.terracotta_demolisher,
200 Species::TerracottaPunisher => &self.terracotta_punisher,
201 Species::TerracottaPursuer => &self.terracotta_pursuer,
202 Species::Cursekeeper => &self.cursekeeper,
203 Species::Forgemaster => &self.forgemaster,
204 Species::Strigoi => &self.strigoi,
205 Species::Executioner => &self.executioner,
206 Species::Gigasfire => &self.gigas_fire,
207 }
208 }
209}
210
211pub const ALL_SPECIES: [Species; Species::NUM_KINDS] = Species::ALL;
212
213impl<'a, SpeciesMeta: 'a> IntoIterator for &'a AllSpecies<SpeciesMeta> {
214 type IntoIter = std::iter::Copied<std::slice::Iter<'static, Self::Item>>;
215 type Item = Species;
216
217 fn into_iter(self) -> Self::IntoIter { ALL_SPECIES.iter().copied() }
218}
219
220enum_iter! {
221 ~const_array(ALL)
222 #[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash, Serialize, Deserialize, EnumString, Display)]
223 #[repr(u32)]
224 pub enum BodyType {
225 Female = 0,
226 Male = 1,
227 }
228}
229pub const ALL_BODY_TYPES: [BodyType; BodyType::NUM_KINDS] = BodyType::ALL;