Extract modeling ViewerData class into new Pet::ModelingSnapshot file
Both extracted and renamed!
This commit is contained in:
parent
242b85470d
commit
90407403ba
2 changed files with 115 additions and 116 deletions
|
|
@ -5,14 +5,14 @@ class Pet < ApplicationRecord
|
||||||
|
|
||||||
def load!(timeout: nil)
|
def load!(timeout: nil)
|
||||||
viewer_data_hash = Neopets::CustomPets.fetch_viewer_data(name, timeout:)
|
viewer_data_hash = Neopets::CustomPets.fetch_viewer_data(name, timeout:)
|
||||||
use_viewer_data(ViewerData.new(viewer_data_hash))
|
use_modeling_snapshot(ModelingSnapshot.new(viewer_data_hash))
|
||||||
end
|
end
|
||||||
|
|
||||||
def use_viewer_data(viewer_data)
|
def use_modeling_snapshot(snapshot)
|
||||||
self.pet_type = viewer_data.pet_type
|
self.pet_type = snapshot.pet_type
|
||||||
@pet_state = viewer_data.pet_state
|
@pet_state = snapshot.pet_state
|
||||||
@alt_style = viewer_data.alt_style
|
@alt_style = snapshot.alt_style
|
||||||
@items = viewer_data.items
|
@items = snapshot.items
|
||||||
end
|
end
|
||||||
|
|
||||||
def wardrobe_query
|
def wardrobe_query
|
||||||
|
|
@ -50,115 +50,5 @@ class Pet < ApplicationRecord
|
||||||
end
|
end
|
||||||
|
|
||||||
class UnexpectedDataFormat < RuntimeError;end
|
class UnexpectedDataFormat < RuntimeError;end
|
||||||
|
|
||||||
# A representation of a Neopets::CustomPets viewer data response, translated
|
|
||||||
# to DTI's database models!
|
|
||||||
class ViewerData
|
|
||||||
def initialize(viewer_data_hash)
|
|
||||||
@custom_pet = viewer_data_hash[:custom_pet]
|
|
||||||
@object_info_registry = viewer_data_hash[:object_info_registry]
|
|
||||||
@object_asset_registry = viewer_data_hash[:object_asset_registry]
|
|
||||||
end
|
|
||||||
|
|
||||||
def pet_type
|
|
||||||
@pet_type ||= begin
|
|
||||||
raise UnexpectedDataFormat unless @custom_pet[:species_id]
|
|
||||||
raise UnexpectedDataFormat unless @custom_pet[:color_id]
|
|
||||||
raise UnexpectedDataFormat unless @custom_pet[:body_id]
|
|
||||||
|
|
||||||
@custom_pet => {species_id:, color_id:}
|
|
||||||
PetType.find_or_initialize_by(species_id:, color_id:).tap do |pet_type|
|
|
||||||
# Apply the pet's body ID to the pet type, unless it's wearing an alt
|
|
||||||
# style, in which case ignore it, because it's the *alt style*'s body ID.
|
|
||||||
# (This can theoretically cause a problem saving a new pet type when
|
|
||||||
# there's an alt style too!)
|
|
||||||
pet_type.body_id = @custom_pet[:body_id] unless @custom_pet[:alt_style]
|
|
||||||
if pet_type.body_id.nil?
|
|
||||||
raise UnexpectedDataFormat,
|
|
||||||
"can't process alt style on first occurrence of pet type"
|
|
||||||
end
|
|
||||||
|
|
||||||
# Try using this pet for the pet type's thumbnail, but don't worry
|
|
||||||
# if it fails.
|
|
||||||
begin
|
|
||||||
pet_type.consider_pet_image(@custom_pet[:name])
|
|
||||||
rescue => error
|
|
||||||
Rails.logger.warn "Failed to load pet image: #{error.full_message}"
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def pet_state
|
|
||||||
@pet_state ||= begin
|
|
||||||
swf_asset_ids = biology_assets.map(&:remote_id)
|
|
||||||
pet_type.pet_states.find_or_initialize_by(swf_asset_ids:).tap do |pet_state|
|
|
||||||
pet_state.swf_assets = biology_assets
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def alt_style
|
|
||||||
@alt_style ||= begin
|
|
||||||
return nil unless @custom_pet[:alt_style]
|
|
||||||
raise UnexpectedDataFormat unless @custom_pet[:alt_color]
|
|
||||||
|
|
||||||
id = @custom_pet[:alt_style].to_i
|
|
||||||
AltStyle.find_or_initialize_by(id:).tap do |alt_style|
|
|
||||||
alt_style.assign_attributes(
|
|
||||||
color_id: @custom_pet[:alt_color].to_i,
|
|
||||||
species_id: @custom_pet[:species_id].to_i,
|
|
||||||
body_id: @custom_pet[:body_id].to_i,
|
|
||||||
swf_assets: alt_style_assets,
|
|
||||||
)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def items
|
|
||||||
@items ||= begin
|
|
||||||
@object_info_registry.map do |id, item_data|
|
|
||||||
Item.find_or_initialize_by(id:).tap do |item|
|
|
||||||
item.add_origin_registry_info item_data
|
|
||||||
item.swf_assets = item_assets_for id
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
private
|
|
||||||
|
|
||||||
def biology_assets
|
|
||||||
@biology_assets ||= begin
|
|
||||||
biology = @custom_pet[:alt_style].present? ?
|
|
||||||
@custom_pet[:original_biology] :
|
|
||||||
@custom_pet[:biology_by_zone]
|
|
||||||
assets_from_biology(biology)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def item_assets_for(item_id)
|
|
||||||
all_infos = @object_asset_registry.values
|
|
||||||
infos = all_infos.select { |a| a[:obj_info_id].to_i == item_id.to_i }
|
|
||||||
infos.map do |asset_data|
|
|
||||||
remote_id = asset_data[:asset_id].to_i
|
|
||||||
SwfAsset.find_or_initialize_by(type: "object", remote_id:).tap do |swf_asset|
|
|
||||||
swf_asset.origin_pet_type = pet_type
|
|
||||||
swf_asset.origin_object_data = asset_data
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def alt_style_assets
|
|
||||||
raise UnexpectedDataFormat if @custom_pet[:biology_by_zone].empty?
|
|
||||||
assets_from_biology(@custom_pet[:biology_by_zone])
|
|
||||||
end
|
|
||||||
|
|
||||||
def assets_from_biology(biology)
|
|
||||||
raise UnexpectedDataFormat if biology.empty?
|
|
||||||
body_id = @custom_pet[:body_id].to_i
|
|
||||||
biology.values.map { |b| SwfAsset.from_biology_data(body_id, b) }
|
|
||||||
end
|
|
||||||
end
|
|
||||||
end
|
end
|
||||||
|
|
||||||
|
|
|
||||||
109
app/models/pet/modeling_snapshot.rb
Normal file
109
app/models/pet/modeling_snapshot.rb
Normal file
|
|
@ -0,0 +1,109 @@
|
||||||
|
# A representation of a Neopets::CustomPets viewer data response, translated
|
||||||
|
# to DTI's database models!
|
||||||
|
class Pet::ModelingSnapshot
|
||||||
|
def initialize(viewer_data_hash)
|
||||||
|
@custom_pet = viewer_data_hash[:custom_pet]
|
||||||
|
@object_info_registry = viewer_data_hash[:object_info_registry]
|
||||||
|
@object_asset_registry = viewer_data_hash[:object_asset_registry]
|
||||||
|
end
|
||||||
|
|
||||||
|
def pet_type
|
||||||
|
@pet_type ||= begin
|
||||||
|
raise Pet::UnexpectedDataFormat unless @custom_pet[:species_id]
|
||||||
|
raise Pet::UnexpectedDataFormat unless @custom_pet[:color_id]
|
||||||
|
raise Pet::UnexpectedDataFormat unless @custom_pet[:body_id]
|
||||||
|
|
||||||
|
@custom_pet => {species_id:, color_id:}
|
||||||
|
PetType.find_or_initialize_by(species_id:, color_id:).tap do |pet_type|
|
||||||
|
# Apply the pet's body ID to the pet type, unless it's wearing an alt
|
||||||
|
# style, in which case ignore it, because it's the *alt style*'s body ID.
|
||||||
|
# (This can theoretically cause a problem saving a new pet type when
|
||||||
|
# there's an alt style too!)
|
||||||
|
pet_type.body_id = @custom_pet[:body_id] unless @custom_pet[:alt_style]
|
||||||
|
if pet_type.body_id.nil?
|
||||||
|
raise Pet::UnexpectedDataFormat,
|
||||||
|
"can't process alt style on first occurrence of pet type"
|
||||||
|
end
|
||||||
|
|
||||||
|
# Try using this pet for the pet type's thumbnail, but don't worry
|
||||||
|
# if it fails.
|
||||||
|
begin
|
||||||
|
pet_type.consider_pet_image(@custom_pet[:name])
|
||||||
|
rescue => error
|
||||||
|
Rails.logger.warn "Failed to load pet image: #{error.full_message}"
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def pet_state
|
||||||
|
@pet_state ||= begin
|
||||||
|
swf_asset_ids = biology_assets.map(&:remote_id)
|
||||||
|
pet_type.pet_states.find_or_initialize_by(swf_asset_ids:).tap do |pet_state|
|
||||||
|
pet_state.swf_assets = biology_assets
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def alt_style
|
||||||
|
@alt_style ||= begin
|
||||||
|
return nil unless @custom_pet[:alt_style]
|
||||||
|
raise Pet::UnexpectedDataFormat unless @custom_pet[:alt_color]
|
||||||
|
|
||||||
|
id = @custom_pet[:alt_style].to_i
|
||||||
|
AltStyle.find_or_initialize_by(id:).tap do |alt_style|
|
||||||
|
alt_style.assign_attributes(
|
||||||
|
color_id: @custom_pet[:alt_color].to_i,
|
||||||
|
species_id: @custom_pet[:species_id].to_i,
|
||||||
|
body_id: @custom_pet[:body_id].to_i,
|
||||||
|
swf_assets: alt_style_assets,
|
||||||
|
)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def items
|
||||||
|
@items ||= begin
|
||||||
|
@object_info_registry.map do |id, item_data|
|
||||||
|
Item.find_or_initialize_by(id:).tap do |item|
|
||||||
|
item.add_origin_registry_info item_data
|
||||||
|
item.swf_assets = item_assets_for id
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
private
|
||||||
|
|
||||||
|
def biology_assets
|
||||||
|
@biology_assets ||= begin
|
||||||
|
biology = @custom_pet[:alt_style].present? ?
|
||||||
|
@custom_pet[:original_biology] :
|
||||||
|
@custom_pet[:biology_by_zone]
|
||||||
|
assets_from_biology(biology)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def item_assets_for(item_id)
|
||||||
|
all_infos = @object_asset_registry.values
|
||||||
|
infos = all_infos.select { |a| a[:obj_info_id].to_i == item_id.to_i }
|
||||||
|
infos.map do |asset_data|
|
||||||
|
remote_id = asset_data[:asset_id].to_i
|
||||||
|
SwfAsset.find_or_initialize_by(type: "object", remote_id:).tap do |swf_asset|
|
||||||
|
swf_asset.origin_pet_type = pet_type
|
||||||
|
swf_asset.origin_object_data = asset_data
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def alt_style_assets
|
||||||
|
raise Pet::UnexpectedDataFormat if @custom_pet[:biology_by_zone].empty?
|
||||||
|
assets_from_biology(@custom_pet[:biology_by_zone])
|
||||||
|
end
|
||||||
|
|
||||||
|
def assets_from_biology(biology)
|
||||||
|
raise Pet::UnexpectedDataFormat if biology.empty?
|
||||||
|
body_id = @custom_pet[:body_id].to_i
|
||||||
|
biology.values.map { |b| SwfAsset.from_biology_data(body_id, b) }
|
||||||
|
end
|
||||||
|
end
|
||||||
Loading…
Reference in a new issue