From efa526a770f63b1d3f906522078a6f8207170d7d Mon Sep 17 00:00:00 2001
From: Justin Edmund <justin@jedmund.com>
Date: Wed, 16 Aug 2023 02:53:34 -0700
Subject: [PATCH] Add mapping for bullets

---
 app/helpers/granblue_wiki.rb | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/app/helpers/granblue_wiki.rb b/app/helpers/granblue_wiki.rb
index 3898c82..666d652 100644
--- a/app/helpers/granblue_wiki.rb
+++ b/app/helpers/granblue_wiki.rb
@@ -11,6 +11,7 @@ class GranblueWiki
   class_attribute :rarities
   class_attribute :genders
   class_attribute :races
+  class_attribute :bullets
   class_attribute :boolean
 
   self.base_uri = 'https://gbf.wiki/api.php'
@@ -59,11 +60,23 @@ class GranblueWiki
     'mf' => 3
   }.freeze
 
+  self.bullets = {
+    'cartridge' => 1,
+    'rifle' => 2,
+    'parabellum' => 3,
+    'aetherial' => 4
+  }.freeze
+
   self.boolean = {
     'yes' => true,
     'no' => false
   }.freeze
 
+  def initialize(props: ['wikitext'], debug: false)
+    @debug = debug
+    @props = props.join('|')
+  end
+
   def fetch(page)
     query_params = params(page).map do |key, value|
       "#{key}=#{value}"