forked from OpenNeo/impress
49 lines
1.6 KiB
Ruby
49 lines
1.6 KiB
Ruby
namespace :pets do
|
|
desc "Load a pet's viewer data (by name or by color/species/items)"
|
|
task :load, [:first] => [:environment] do |task, args|
|
|
# Collect all arguments (first + extras)
|
|
all_args = [args[:first]] + args.extras
|
|
|
|
# If only one argument, treat it as a pet name
|
|
if all_args.length == 1
|
|
viewer_data = Neopets::CustomPets.fetch_viewer_data(all_args[0])
|
|
else
|
|
# Multiple arguments: color, species, and optional item IDs
|
|
color_name = all_args[0]
|
|
species_name = all_args[1]
|
|
item_ids = all_args[2..]
|
|
|
|
# Look up the PetType to get its image hash
|
|
pet_type = PetType.matching_name(color_name, species_name).first!
|
|
pet_sci = pet_type.image_hash
|
|
|
|
# Fetch the new image hash for this pet+items combination
|
|
response = Neopets::NCMall.fetch_pet_data(pet_sci, item_ids)
|
|
new_sci = response[:newsci]
|
|
|
|
# Output the hash to stderr for debugging
|
|
$stderr.puts "Generated image hash: #{new_sci}"
|
|
|
|
# Load the full viewer data using the new image hash
|
|
viewer_data = Neopets::CustomPets.fetch_viewer_data("@#{new_sci}")
|
|
end
|
|
|
|
puts JSON.pretty_generate(viewer_data)
|
|
end
|
|
|
|
desc "Find pets that were, last we saw, of the given color and species"
|
|
task :find, [:color_name, :species_name] => [:environment] do |task, args|
|
|
begin
|
|
pt = PetType.matching_name(args.color_name, args.species_name).first!
|
|
rescue ActiveRecord::RecordNotFound
|
|
abort "Could not find pet type for " +
|
|
"#{args.color_name} #{args.species_name}"
|
|
end
|
|
|
|
limit = ENV.fetch("LIMIT", 10)
|
|
|
|
pt.pets.limit(limit).order(id: :desc).pluck(:name).each do |pet_name|
|
|
puts "- #{pet_name} (https://www.neopets.com/petlookup.phtml?pet=#{pet_name})"
|
|
end
|
|
end
|
|
end
|