badass 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4ab95b2d4678a5c3700ff190aa6c7ff496c6a01b61623f2dcef876f6a1f1855e
-  data.tar.gz: 360a3107f9d5211315f3e07eda902327971212965021856d14f2d015d8feb1b7
+  metadata.gz: e48b4cc6729c2e2b669be4c10796aa438da25c9b58eaee979c3d1e4dbae0f89a
+  data.tar.gz: f8b76cd02126f0d0696b8c169cd21f832ea1c51b8a3f16d79f9d845e18e291f1
 SHA512:
-  metadata.gz: ae0e9040e652e7f7eb70f6c9aeb5f9b902055f5b8995f473e11834f0d34ce698b670fb3de93e07500a925f342c4b9fa3f634ae79000b944444a6cbd05d3b8251
-  data.tar.gz: 29d60696eeaa0ef609bbe7f7dc83ea71524f835c909342948f8d829be43fd5d26bc25f905b691d64e1a2874ebf6dba0234428ad6a8bfe81dc6cdac6de3abd1c6
+  metadata.gz: 468602146563064d4755dda01ecc741a28615fa22fb2be25cad2778061baebe7a29bf8cac723e62f9419172d08ce1857229d283ffd5cc576109243ae5f71cea8
+  data.tar.gz: 857e993e52a020ea95b1f0d890a22fe8656a02514a122dfb20df8cb06a188996e058935f9f7656015d9318fb593afb8df785b6aaea68daa72f3da71c3835551f

data/lib/badass.rb

@@ -2,13 +2,15 @@ require 'net/http'
 require 'json'
 
 module BadASS
-  @api_root = 'https://bad-dragon.com/api'
+  FIRMNESSES = { '2' => 'Extra Soft', '3' => 'Soft', '5' => 'Medium', '8' => 'Firm', '3/5' => 'Soft Shaft, Med Base', '5/3' => 'Soft Shaft, Med Base', '3/8' => 'Soft Shaft, Firm Base', '8/3' => 'Soft Shaft, Firm Base', '5/8' => 'Med Shaft, Firm Base', '8/5' => 'Med Shaft, Firm Base' }.freeze
+  SIZES = { 'onesize' => 'One-Size', 'mini' => 'Mini', 'small' => 'Small', 'medium' => 'Medium', 'large' => 'Large', 'extralarge' => 'Extra Large' }.freeze
 
-  def self.test
-    return 'your gay lol'
+  @baddragon_skus = {}
+  JSON.parse(Net::HTTP.get(URI('https://bad-dragon.com/api/inventory-toy/product-list'))).each do |toy|
+    @baddragon_skus[toy['sku']] = toy['name']
   end
-
+  BAD_DRAGON_SKUS = @baddragon_skus.freeze
 end
 
 require 'badass/client'
-require 'badass/fast'
+require 'badass/toy'

data/lib/badass/client.rb

@@ -1,32 +1,25 @@
 class BadASS::Client
   # Creates a client to do operations.
+  # Takes a :refresh_time argument to define how often the toy list should be refreshed.
   def initialize(refresh_time: 600)
     Thread.new do
-      @baddragon_skus = {}
-      JSON.parse(Net::HTTP.get(URI('https://bad-dragon.com/api/inventory-toy/product-list'))).each do |toy|
-        @baddragon_skus[toy['sku']] = toy['name']
-      end
+      @toys = []
       loop do
-        @toys_finished = false
         page = 1
-        @toys = []
+        toy_list = []
         loop do
           newtoys = JSON.parse(Net::HTTP.get(URI("https://bad-dragon.com/api/inventory-toys?price[min]=0&price[max]=300&noAccessories=false&cumtube=false&suctionCup=false&sort[field]=price&&sort[direction]=asc&page=#{page}&limit=60")))
+          page += 1
           newtoys['toys'].each do |toy|
-            toy['name'] = @baddragon_skus[toy['sku']]
+            toy_list << BadASS::Toy.new(toy)
           end
-          page += 1
-          @toys.concat(newtoys['toys'])
           break if page > newtoys['totalPages']
         end
-        @toys_finished = true
+        @toys = toy_list
         sleep(refresh_time)
       end
     end
   end
 
-  def toys
-    sleep(0.1) until @toys_finished
-    @toys
-  end
+  attr_reader :toys
 end

data/lib/badass/toy.rb

@@ -1,17 +1,40 @@
 class BadASS::Toy
+  # Create a Toy object using a hash from the API.
   def initialize(toy_hash)
-    @id = toy_hash['id'] || 000000
-    @title = video_hash[:title] || 'N/A'
-    @url = video_hash[:webpage_url] || 'N/A'
-    @thumbnail_url = video_hash[:thumbnail] || Bot::HBOT.profile.avatar_url
-    @like_count = video_hash[:like_count] || 'N/A'
-    @dislike_count = video_hash[:dislike_count] || 'N/A'
-    @view_count = video_hash[:view_count] || 'N/A'
-    @length = video_hash[:duration] || 0
-    @location = video_hash[:filename] + '.mp4'
-    @loop = false
-    @event = event
-    @skipped_time = 0
-    @filters = video_hash[:filters] || []
+    @id = toy_hash['id']
+    @sku = toy_hash['sku']
+    @size = toy_hash['size']
+    @price = toy_hash['price']
+    @weight = toy_hash['weight']
+    @color = toy_hash['color']
+    @colors = [toy_hash['color1'], toy_hash['color2'], toy_hash['color3']]
+    @flop_reason = toy_hash['flop_reason'].capitalize
+    @type = toy_hash['type'].capitalize
+    @cumtube = toy_hash['cumtube']
+    @suction_cup = toy_hash['suction_cup']
+    @images = toy_hash['images'].map { |toy| toy['fullFilename'] }
   end
+
+  def name
+    BadASS::BAD_DRAGON_SKUS[sku]
+  end
+
+  def cumtube?
+    @cumtube == 1
+  end
+
+  def suction_cup?
+    @suction_cup == 1
+  end
+
+  attr_reader :id
+  attr_reader :sku
+  attr_reader :size
+  attr_reader :price
+  attr_reader :weight
+  attr_reader :color
+  attr_reader :colors
+  attr_reader :flop_reason
+  attr_reader :type
+  attr_reader :images
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: badass
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - charagarlnad