feat: add weapon filter

src/components/TeamGenerator.tsx

@@ -4,7 +4,7 @@ import React, { useState } from 'react'
 import { Character } from 'genshin-db'
 import { filterNamed, findTeamType, healers } from '../lib/queries'
 import teamTypes from '../lib/teamTypes'
-import { Gender } from '../types/teams'
+import { Gender, WeaponType, WeaponMap } from '../types/teams'
 import { OptionBuilder } from '../lib/builders/OptionBuilder'
 import classnames from 'classnames'
 
@@ -26,13 +26,19 @@ const TeamGenerator = ({
   const [teamType, setTeamType] = useState('')
   const [gender, setGender] = useState<Gender | undefined>(undefined)
   const [includeHealer, setIncludeHealer] = useState(false)
+  const [weapon, setWeapon] = useState<WeaponType | undefined>(undefined)
   const selectedType = findTeamType(teamType, teamTypes)
 
   const healerTooltip = `Note: Healers list includes ${healers.join(', ')}`
 
   // Trigger option builder
   const generateTeam = () => {
-    const builder = new OptionBuilder(selectedType, gender, includeHealer)
+    const builder = new OptionBuilder(
+      selectedType,
+      gender,
+      includeHealer,
+      weapon
+    )
     const team = builder.build(characters)
     if (!team) {
       alert("Sorry, I couldn't find enough members to make that kind of team")
@@ -112,6 +118,22 @@ const TeamGenerator = ({
             )}
           </div>
 
+          <div className="mb-4">
+            <label className="block text-gray-700 text-sm font-bold mb-2">
+              Weapon Type
+            </label>
+            <select
+              className="w-full p-2"
+              value={weapon}
+              onChange={(e) => setWeapon(e.target.value as WeaponType)}
+            >
+              <option value="">Any weapon</option>
+              {WeaponMap.map((type) => (
+                <option value={type.type}>{type.label}</option>
+              ))}
+            </select>
+          </div>
+
           <div className="mb-4">
             <span className="text-gray-700 text-sm font-bold">Genders</span>
             <div className="mt-2">

src/lib/builders/OptionBuilder.ts

@@ -1,11 +1,12 @@
 import { Builder, Filter } from '../../types/selector'
 import { Character } from 'genshin-db'
-import { CompleteTeam, Gender, TeamType } from '../../types/teams'
+import { CompleteTeam, Gender, TeamType, WeaponType } from '../../types/teams'
 import { RandomBuilder } from './RandomBuilder'
 import { GenderFilter } from '../filters/GenderFilter'
 import { TeamTypeFilter } from '../filters/TeamTypeFilter'
 import { TeamTypeBuilder } from './TeamTypeBuilder'
 import { HealingBuilder } from './HealingBuilder'
+import { WeaponFilter } from '../filters/WeaponFilter'
 
 /**
  * Generate team based on selected options
@@ -14,15 +15,18 @@ export class OptionBuilder implements Builder {
   selectedType: TeamType | undefined
   gender: Gender | undefined
   includesHealer: boolean
+  weapon: WeaponType | undefined
 
   constructor(
     selectedType: TeamType | undefined,
     gender: Gender | undefined,
-    includeHeader: boolean
+    includeHeader: boolean,
+    weapon: WeaponType | undefined
   ) {
     this.selectedType = selectedType
     this.gender = gender
     this.includesHealer = includeHeader
+    this.weapon = weapon
   }
 
   /**
@@ -36,6 +40,9 @@ export class OptionBuilder implements Builder {
     if (this.selectedType) {
       filters.push(new TeamTypeFilter(this.selectedType))
     }
+    if (this.weapon) {
+      filters.push(new WeaponFilter(this.weapon))
+    }
     return filters
   }
 

src/lib/filters/WeaponFilter.ts

@@ -0,0 +1,20 @@
+import { Filter } from '../../types/selector'
+import { Character } from 'genshin-db'
+import { WeaponType } from '../../types/teams'
+
+/**
+ * Filters characters by a specific weapon
+ */
+export class WeaponFilter implements Filter {
+  weapon: WeaponType
+
+  constructor(weapon: WeaponType) {
+    this.weapon = weapon
+  }
+
+  filter(characters: Array<Character>): Array<Character> {
+    return characters.filter(
+      (character) => character.weaponType === this.weapon
+    )
+  }
+}

src/types/teams.ts

@@ -9,6 +9,14 @@ export enum Gender {
   Female = 'Female',
 }
 
+export const WeaponMap: Array<{ type: WeaponType; label: string }> = [
+  { type: 'WEAPON_BOW', label: 'Bow' },
+  { type: 'WEAPON_CATALYST', label: 'Catalyst' },
+  { type: 'WEAPON_CLAYMORE', label: 'Claymore' },
+  { type: 'WEAPON_POLE', label: 'Polearm' },
+  { type: 'WEAPON_SWORD_ONE_HAND', label: 'Sword' },
+]
+
 // Restriction rules for a team member. Can be empty if slot is open to all.
 export type TeamMemberType = {
   // For specific named character