forked from OpenNeo/impress
133 lines
3.7 KiB
Ruby
133 lines
3.7 KiB
Ruby
require 'yaml'
|
|
|
|
class ClosetPage
|
|
include ActiveModel::Conversion
|
|
extend ActiveModel::Naming
|
|
|
|
SELECTORS = {
|
|
:items => "form[action=\"process_closet.phtml\"] tr[bgcolor!=silver][bgcolor!=\"#E4E4E4\"]",
|
|
:item_thumbnail => "img",
|
|
:item_name => "td:nth-child(2)",
|
|
:item_quantity => "td:nth-child(5)",
|
|
:item_remove => "input",
|
|
:page_select => "select[name=page]",
|
|
:selected => "option[selected]"
|
|
}
|
|
|
|
attr_reader :hangers, :index, :source, :total_pages, :unknown_item_names, :user
|
|
|
|
def initialize(user)
|
|
raise ArgumentError, "Expected #{user.inspect} to be a User", caller unless user.is_a?(User)
|
|
@user = user
|
|
end
|
|
|
|
def last?
|
|
@index == @total_pages
|
|
end
|
|
|
|
def persisted?
|
|
false
|
|
end
|
|
|
|
def save_hangers!
|
|
counts = {:created => 0, :updated => 0}
|
|
ClosetHanger.transaction do
|
|
@hangers.each do |hanger|
|
|
if hanger.new_record?
|
|
counts[:created] += 1
|
|
hanger.save!
|
|
elsif hanger.changed?
|
|
counts[:updated] += 1
|
|
hanger.save!
|
|
end
|
|
end
|
|
end
|
|
counts
|
|
end
|
|
|
|
def source=(source)
|
|
@source = source
|
|
parse_source!(source)
|
|
end
|
|
|
|
protected
|
|
|
|
def element(selector_name, parent)
|
|
parent.at_css(SELECTORS[selector_name]) ||
|
|
raise(ParseError, "Closet #{selector_name} element not found in #{parent.inspect}")
|
|
end
|
|
|
|
def elements(selector_name, parent)
|
|
parent.css(SELECTORS[selector_name])
|
|
end
|
|
|
|
def parse_source!(source)
|
|
doc = Nokogiri::HTML(source)
|
|
|
|
page_selector = element(:page_select, doc)
|
|
@total_pages = page_selector.children.size
|
|
@index = element(:selected, page_selector)['value'].to_i
|
|
|
|
items_data = {
|
|
:id => {},
|
|
:thumbnail_url => {}
|
|
}
|
|
|
|
# Go through the items, and find the ID/thumbnail for each and data with it
|
|
elements(:items, doc).each do |row|
|
|
# For normal items, the td contains essentially:
|
|
# <b>NAME<br/><span>OPTIONAL ADJECTIVE</span></b>
|
|
# For PB items, the td contains:
|
|
# NAME<br/><span>OPTIONAL ADJECTIVE</span>
|
|
# So, we want the first text node. If it's a PB item, that's the first
|
|
# child. If it's a normal item, it's the first child <b>'s child.
|
|
name_el = element(:item_name, row).children[0]
|
|
name_el = name_el.children[0] if name_el.name == 'b'
|
|
|
|
data = {
|
|
:name => name_el.text,
|
|
:quantity => element(:item_quantity, row).text.to_i
|
|
}
|
|
|
|
if id = element(:item_remove, row)['name']
|
|
id = id.to_i
|
|
items_data[:id][id] = data
|
|
else # if this is a pb item, which does not give ID, go by thumbnail
|
|
thumbnail_url = element(:item_thumbnail, row)['src']
|
|
items_data[:thumbnail_url][thumbnail_url] = data
|
|
end
|
|
end
|
|
|
|
# Find items with either a matching ID or matching thumbnail URL
|
|
# Check out that single-query beauty :)
|
|
i = Item.arel_table
|
|
items = Item.where(
|
|
i[:id].in(items_data[:id].keys).
|
|
or(
|
|
i[:thumbnail_url].in(items_data[:thumbnail_url].keys)
|
|
)
|
|
)
|
|
|
|
# Create closet hanger from each item, and remove them from the reference
|
|
# lists
|
|
@hangers = items.map do |item|
|
|
data = items_data[:id].delete(item.id) ||
|
|
items_data[:thumbnail_url].delete(item.thumbnail_url)
|
|
hanger = @user.closet_hangers.find_or_initialize_by_item_id(item.id)
|
|
hanger.quantity = data[:quantity]
|
|
hanger
|
|
end
|
|
|
|
# Take the names of the items remaining in the reference lists, meaning
|
|
# that they weren't found
|
|
@unknown_item_names = []
|
|
items_data.each do |type, data_by_key|
|
|
data_by_key.each do |key, data|
|
|
@unknown_item_names << data[:name]
|
|
end
|
|
end
|
|
end
|
|
|
|
class ParseError < RuntimeError;end
|
|
end
|
|
|