impress/app/models/pet.rb

166 lines
5.3 KiB
Ruby
Raw Normal View History

require 'rocketamf_extensions/remote_gateway'
require 'ostruct'
2011-05-31 07:36:32 -07:00
class Pet < ApplicationRecord
NEOPETS_URL_ORIGIN = ENV['NEOPETS_URL_ORIGIN'] || 'https://www.neopets.com'
GATEWAY_URL = NEOPETS_URL_ORIGIN + '/amfphp/gateway.php'
PET_VIEWER = RocketAMFExtensions::RemoteGateway.new(GATEWAY_URL).
service('CustomPetService').action('getViewerData')
PET_NOT_FOUND_REMOTE_ERROR = 'PHP: Unable to retrieve records from the database.'
WARDROBE_PATH = '/wardrobe'
belongs_to :pet_type
attr_reader :items, :pet_state, :alt_style
scope :with_pet_type_color_ids, ->(color_ids) {
2010-11-27 15:41:06 -08:00
joins(:pet_type).where(PetType.arel_table[:id].in(color_ids))
}
2013-01-13 18:10:01 -08:00
def load!(options={})
options[:locale] ||= I18n.default_locale
2013-12-08 20:59:36 -08:00
I18n.with_locale(options.delete(:locale)) do
use_viewer_data(
self.class.fetch_viewer_data(name, options.delete(:timeout)),
options,
)
2013-01-13 18:10:01 -08:00
end
true
end
2013-12-08 20:59:36 -08:00
def use_viewer_data(viewer_data, options={})
options[:item_scope] ||= Item.all
2013-12-08 20:59:36 -08:00
pet_data = viewer_data[:custom_pet]
raise UnexpectedDataFormat unless pet_data[:species_id]
raise UnexpectedDataFormat unless pet_data[:color_id]
raise UnexpectedDataFormat unless pet_data[:body_id]
has_alt_style = pet_data[:alt_style].present?
self.pet_type = PetType.find_or_initialize_by(
species_id: pet_data[:species_id].to_i,
color_id: pet_data[:color_id].to_i
2013-12-08 20:59:36 -08:00
)
self.pet_type.origin_pet = self
# With an alt style, `body_id` in the biology data refers to the body ID of
# the *alt* style, not the usual pet type. (We have `original_biology` for
# *some* of the pet type's situation, but not it's body ID!)
#
# So, in the alt style case, don't update `body_id` - but if this is our
# first time seeing this pet type and it doesn't *have* a `body_id` yet,
# let's not be creating it without one. We'll need to model it without the
# alt style first. (I don't bother with a clear error message though 😅)
self.pet_type.body_id = pet_data[:body_id] unless has_alt_style
if self.pet_type.body_id.nil?
raise UnexpectedDataFormat,
"can't process alt style on first occurrence of pet type"
end
pet_state_biology = has_alt_style ? pet_data[:original_biology] :
pet_data[:biology_by_zone]
raise UnexpectedDataFormat if pet_state_biology.empty?
pet_state_biology[0] = nil # remove effects if present
@pet_state = self.pet_type.add_pet_state_from_biology! pet_state_biology
if has_alt_style
raise UnexpectedDataFormat unless pet_data[:alt_color]
raise UnexpectedDataFormat if pet_data[:biology_by_zone].empty?
@alt_style = AltStyle.find_or_initialize_by(id: pet_data[:alt_style].to_i)
@alt_style.assign_attributes(
color_id: pet_data[:alt_color].to_i,
species_id: pet_data[:species_id].to_i,
body_id: pet_data[:body_id].to_i,
biology: pet_data[:biology_by_zone],
)
end
2013-12-08 20:59:36 -08:00
@items = Item.collection_from_pet_type_and_registries(self.pet_type,
viewer_data[:object_info_registry], viewer_data[:object_asset_registry],
options[:item_scope])
end
2010-10-10 19:18:42 -07:00
def wardrobe_query
{
name: self.name,
color: self.pet_type.color_id,
species: self.pet_type.species_id,
pose: self.pet_state.pose,
state: self.pet_state.id,
objects: self.items.map(&:id),
2010-10-10 19:18:42 -07:00
}.to_query
end
2010-11-06 16:07:15 -07:00
def contributables
2024-01-24 03:54:43 -08:00
contributables = [pet_type, @pet_state, @alt_style].filter(&:present?)
2010-11-06 16:07:15 -07:00
items.each do |item|
contributables << item
contributables += item.pending_swf_assets
end
contributables
end
2010-11-06 15:08:42 -07:00
before_validation do
2010-11-06 16:07:15 -07:00
pet_type.save!
2011-10-31 14:22:24 -07:00
if @pet_state
@pet_state.save!
@pet_state.handle_assets!
2011-10-31 14:22:24 -07:00
end
if @items
@items.each do |item|
2013-01-28 00:10:25 -08:00
item.save! if item.changed?
item.handle_assets!
end
2010-11-06 15:08:42 -07:00
end
if @alt_style
@alt_style.save!
end
2010-11-06 15:08:42 -07:00
end
2013-01-13 18:10:01 -08:00
def self.load(name, options={})
pet = Pet.find_or_initialize_by(name: name)
2013-01-13 18:10:01 -08:00
pet.load!(options)
pet
end
2013-12-08 20:59:36 -08:00
def self.from_viewer_data(viewer_data, options={})
pet = Pet.find_or_initialize_by(name: viewer_data[:custom_pet][:name])
2013-12-08 20:59:36 -08:00
pet.use_viewer_data(viewer_data, options)
pet
end
# NOTE: Ideally pet requests shouldn't take this long, but Neopets can be
# slow sometimes! Since we're on the Falcon server, long timeouts shouldn't
# slow down the rest of the request queue, like it used to be in the past.
def self.fetch_viewer_data(name, timeout=10, locale=nil)
locale ||= I18n.default_locale
begin
neopets_language_code = I18n.compatible_neopets_language_code_for(locale)
envelope = PET_VIEWER.request([name, 0]).post(
:timeout => timeout,
:headers => {
'Cookie' => "lang=#{neopets_language_code}"
}
)
rescue RocketAMFExtensions::RemoteGateway::AMFError => e
if e.message == PET_NOT_FOUND_REMOTE_ERROR
raise PetNotFound, "Pet #{name.inspect} does not exist"
end
raise DownloadError, e.message
rescue RocketAMFExtensions::RemoteGateway::ConnectionError => e
raise DownloadError, e.message, e.backtrace
end
HashWithIndifferentAccess.new(envelope.messages[0].data.body)
end
class PetNotFound < RuntimeError;end
class DownloadError < RuntimeError;end
class UnexpectedDataFormat < RuntimeError;end
end