hotel_beds 0.3.2 → 0.4.0

data/lib/hotel_beds/model/contract.rb

@@ -0,0 +1,15 @@
+require "hotel_beds/model"
+require "hotel_beds/model/comment"
+
+module HotelBeds
+  module Model
+    class Contract
+      include HotelBeds::Model
+
+      # attributes
+      attribute :name, String
+      attribute :incoming_office_code, String
+      attribute :comments, Array[HotelBeds::Model::Comment]
+    end
+  end
+end

data/lib/hotel_beds/model/customer.rb

@@ -0,0 +1,21 @@
+require "hotel_beds/model"
+
+module HotelBeds
+  module Model
+    class Customer
+      include HotelBeds::Model
+
+      # attributes
+      attribute :id, Integer
+      attribute :type, Symbol
+      attribute :age, Integer
+      attribute :name, Integer
+      attribute :last_name, Integer
+
+      # validation
+      validates :id, :name, :last_name, presence: true
+      validates :age, numericality: { greater_than_or_equal_to: 0 }
+      validates :type, inclusion: { in: %i(adult child) }
+    end
+  end
+end

data/lib/hotel_beds/model/destination.rb

@@ -0,0 +1,13 @@
+require "hotel_beds/model"
+
+module HotelBeds
+  module Model
+    class Destination
+      include HotelBeds::Model
+
+      # attributes
+      attribute :code, String
+      attribute :name, String
+    end
+  end
+end

data/lib/hotel_beds/model/hotel.rb

@@ -1,5 +1,8 @@
 require "hotel_beds/model"
-require_relative "search_result"
+require "hotel_beds/model/available_room"
+require "hotel_beds/model/contract"
+require "hotel_beds/model/destination"
+require "hotel_beds/parser/room_grouper"
 
 module HotelBeds
   module Model
@@ -7,17 +10,19 @@ module HotelBeds
       include HotelBeds::Model
 
       # attributes
-      attribute :id, Integer
+      attribute :code, String
       attribute :availability_token, String
       attribute :name, String
       attribute :images, Array[String]
-      attribute :stars, Integer
       attribute :longitude, BigDecimal
       attribute :latitude, BigDecimal
-      attribute :results, Array[SearchResult]
-      attribute :destination_code, String
-      attribute :contract_name, String
-      attribute :contract_incoming_office_code, String
+      attribute :available_rooms, Array[HotelBeds::Model::AvailableRoom]
+      attribute :contract, HotelBeds::Model::Contract
+      attribute :destination, HotelBeds::Model::Destination
+
+      def grouped_rooms(requested_rooms)
+        HotelBeds::Parser::RoomGrouper.new(requested_rooms, available_rooms).groups
+      end
     end
   end
 end

data/lib/hotel_beds/model/price.rb

@@ -0,0 +1,18 @@
+require "hotel_beds/model"
+
+module HotelBeds
+  module Model
+    class Price
+      include HotelBeds::Model
+
+      # attributes
+      attribute :from, Date
+      attribute :to, Date
+      attribute :amount, BigDecimal
+
+      def dates
+        (from..to).to_a
+      end
+    end
+  end
+end

data/lib/hotel_beds/model/purchase.rb

@@ -1,5 +1,6 @@
 require "hotel_beds/model"
 require "hotel_beds/model/service"
+require "hotel_beds/model/customer"
 
 module HotelBeds
   module Model
@@ -7,10 +8,14 @@ module HotelBeds
       include HotelBeds::Model
 
       # attributes
-      attribute :id, String
-      attribute :services, Array[Service]
+      attribute :token, String
+      attribute :time_to_expiration, Integer
+      attribute :agency_reference, String
+      attribute :status, String
       attribute :currency, String
       attribute :amount, BigDecimal
+      attribute :services, Array[HotelBeds::Model::Service]
+      attribute :holder, HotelBeds::Model::Customer
     end
   end
 end

data/lib/hotel_beds/model/room.rb

@@ -1,5 +1,4 @@
 require "hotel_beds/model"
-require_relative "room"
 
 module HotelBeds
   module Model

data/lib/hotel_beds/model/service.rb

@@ -1,4 +1,5 @@
 require "hotel_beds/model"
+require "hotel_beds/model/contract"
 
 module HotelBeds
   module Model
@@ -7,10 +8,11 @@ module HotelBeds
 
       # attributes
       attribute :id, String
-      attribute :contract_name, String
-      attribute :contract_incoming_office_code, String
-      attribute :check_in_date, Date
-      attribute :check_out_date, Date
+      attribute :type, String
+      attribute :status, String
+      attribute :contract, HotelBeds::Model::Contract
+      attribute :date_from, Date
+      attribute :date_to, Date
       attribute :currency, String
       attribute :amount, BigDecimal
     end

data/lib/hotel_beds/parser.rb

@@ -0,0 +1,94 @@
+module HotelBeds
+  module Parser
+    def self.included(base)
+      base.extend ClassMethods
+      base.send(:include, InstanceMethods)
+    end
+
+    module ClassMethods
+      class Attribute
+        attr_accessor :name, :parser, :selector, :attr, :multiple, :custom
+        private :name=, :parser=, :selector=, :attr=, :multiple=, :custom=
+
+        def initialize(name, parser: nil, selector: nil, attr: :content, multiple: false, &block)
+          self.name = name.to_sym
+          self.parser = parser
+          self.selector = selector
+          self.attr = attr
+          self.multiple = !!multiple
+          self.custom = block
+          freeze
+        end
+
+        def retrieve(doc)
+          if custom
+            custom.yield(doc)
+          elsif multiple
+            doc.css(selector).map(&method(:read_element))
+          elsif selector
+            (element = doc.at_css(selector)) && read_element(element) or nil
+          else
+            read_element(doc)
+          end
+        end
+
+        private
+        def read_element(element)
+          if parser
+            parser.new(element).to_h
+          elsif attr == :content
+            element.content
+          else
+            element.attr(attr)
+          end
+        end
+      end
+
+      def attributes
+        @attributes ||= Array.new
+      end
+
+      protected def attribute(*args, &block)
+        (@attributes ||= Array.new).push(Attribute.new(*args, &block))
+      end
+
+      def default_model_class(klass = nil)
+        # set the class, if given
+        @default_model_class = klass if klass
+        # return the class, defaulting to HotelBeds::Model::ClassName
+        @default_model_class ||= begin
+          klass_name = name.gsub(/^.*\:\:(.*?)$/, '\1')
+          file_name = klass_name.gsub(/(?<=.)([A-Z]+)/, '_\1').downcase
+          require "hotel_beds/model/#{file_name}"
+          ::HotelBeds.const_get("Model").const_get(klass_name)
+        end
+      end
+    end
+
+    module InstanceMethods
+      def self.included(base)
+        base.class_eval do
+          attr_accessor :doc
+          private :doc=
+        end
+      end
+
+      def initialize(doc)
+        self.doc = doc
+        freeze
+      end
+
+      # parses the document into a hash of attributes
+      def to_h
+        self.class.attributes.inject(Hash.new) do |result, attribute|
+          result.merge(attribute.name => attribute.retrieve(doc))
+        end
+      end
+
+      # parses the document into a HotelBeds::Model instance
+      def to_model(klass = self.class.default_model_class)
+        klass.new(to_h)
+      end
+    end
+  end
+end

data/lib/hotel_beds/parser/available_room.rb

@@ -0,0 +1,37 @@
+require "hotel_beds/parser"
+require "hotel_beds/parser/cancellation_policy"
+require "hotel_beds/parser/price"
+require "hotel_beds/parser/customer"
+
+module HotelBeds
+  module Parser
+    class AvailableRoom
+      include HotelBeds::Parser
+
+      # attributes
+      attribute :id, selector: "HotelRoom", attr: "SHRUI"
+      attribute :room_count, selector: "HotelOccupancy RoomCount"
+      attribute :adult_count, selector: "HotelOccupancy AdultCount"
+      attribute :child_count, selector: "HotelOccupancy ChildCount"
+      attribute :number_available, selector: "HotelRoom", attr: "availCount"
+      attribute :description, selector: "HotelRoom RoomType"
+      attribute :board, selector: "HotelRoom Board"
+      attribute :board_code, selector: "HotelRoom Board", attr: "code"
+      attribute :room_type_code, selector: "HotelRoom RoomType", attr: "code"
+      attribute :room_type_characteristic,
+        selector: "HotelRoom RoomType", attr: "characteristic"
+      attribute :price do |element|
+        ((element.at_css("HotelRoom") > "Price") > "Amount").first.content
+      end
+      attribute :cancellation_policies,
+        selector: "HotelRoom CancellationPolicy", multiple: true,
+        parser: HotelBeds::Parser::CancellationPolicy
+      attribute :rates,
+        selector: "Price PriceList Price", multiple: true,
+        parser: HotelBeds::Parser::Price
+      attribute :customers,
+        selector: "HotelOccupancy GuestList Customer", multiple: true,
+        parser: HotelBeds::Parser::Customer
+    end
+  end
+end

data/lib/hotel_beds/parser/cancellation_policy.rb

@@ -0,0 +1,13 @@
+require "hotel_beds/parser"
+
+module HotelBeds
+  module Parser
+    class CancellationPolicy
+      include HotelBeds::Parser
+
+      # attributes
+      attribute :amount, attr: "amount"
+      attribute :from, attr: "dateFrom"
+    end
+  end
+end

data/lib/hotel_beds/parser/comment.rb

@@ -0,0 +1,13 @@
+require "hotel_beds/parser"
+
+module HotelBeds
+  module Parser
+    class Comment
+      include HotelBeds::Parser
+
+      # attributes
+      attribute :type, attr: "type"
+      attribute :content
+    end
+  end
+end

data/lib/hotel_beds/parser/contract.rb

@@ -0,0 +1,17 @@
+require "hotel_beds/parser"
+require "hotel_beds/parser/comment"
+
+module HotelBeds
+  module Parser
+    class Contract
+      include HotelBeds::Parser
+
+      # attributes
+      attribute :name, selector: "Name"
+      attribute :incoming_office_code, selector: "IncomingOffice", attr: "code"
+      attribute :comments,
+        selector: "CommentList Comment", multiple: true,
+        parser: HotelBeds::Parser::Comment
+    end
+  end
+end

data/lib/hotel_beds/parser/customer.rb

@@ -0,0 +1,18 @@
+require "hotel_beds/parser"
+
+module HotelBeds
+  module Parser
+    class Customer
+      include HotelBeds::Parser
+
+      # attributes
+      attribute :id, selector: "CustomerId"
+      attribute :type do |element|
+        element.attr("type") == "CH" ? :child : :adult
+      end
+      attribute :name, selector: "Name"
+      attribute :last_name, selector: "LastName"
+      attribute :age, selector: "Age"
+    end
+  end
+end

data/lib/hotel_beds/parser/destination.rb

@@ -0,0 +1,13 @@
+require "hotel_beds/parser"
+
+module HotelBeds
+  module Parser
+    class Destination
+      include HotelBeds::Parser
+
+      # attributes
+      attribute :code, attr: "code"
+      attribute :name, selector: "Name"
+    end
+  end
+end

data/lib/hotel_beds/parser/errors.rb

@@ -0,0 +1,39 @@
+require "active_model/errors"
+
+module HotelBeds
+  module Parser
+    class Errors
+      attr_accessor :response
+      private :response=
+
+      def initialize(response)
+        self.response = response
+        freeze
+      end
+
+      def to_model(object)
+        ActiveModel::Errors.new(object).tap do |errors|
+          if response.http_error?
+            errors.add(:base, "HTTP error")
+          elsif response.soap_fault?
+            errors.add(:base, "SOAP error")
+          elsif !response.success?
+            errors.add(:base, "Request failed")
+          end
+
+          body.css("ErrorList Error").each do |error|
+            errors.add(:base, [
+              (sm = error.at_css("Message")) && sm.content,
+              (dm = error.at_css("DetailedMessage")) && dm.content
+            ].compact.join("\n"))
+          end
+        end
+      end
+
+      private
+      def body
+        Nokogiri::XML(response.body.values.first)
+      end
+    end
+  end
+end

data/lib/hotel_beds/parser/hotel.rb

@@ -0,0 +1,27 @@
+require "hotel_beds/parser"
+require "hotel_beds/parser/contract"
+require "hotel_beds/parser/destination"
+require "hotel_beds/parser/available_room"
+
+module HotelBeds
+  module Parser
+    class Hotel
+      include HotelBeds::Parser
+
+      # attributes
+      attribute :code, selector: "HotelInfo > Code"
+      attribute :availability_token, attr: "availToken"
+      attribute :name, selector: "HotelInfo > Name"
+      attribute :images, selector: "HotelInfo > ImageList > Image > Url",
+        multiple: true
+      attribute :longitude, selector: "Position", attr: "longitude"
+      attribute :latitude, selector: "Position", attr: "latitude"
+      attribute :contract, selector: "ContractList > Contract",
+        parser: HotelBeds::Parser::Contract
+      attribute :destination, selector: "HotelInfo > Destination",
+        parser: HotelBeds::Parser::Destination
+      attribute :available_rooms, selector: "AvailableRoom", multiple: true,
+        parser: HotelBeds::Parser::AvailableRoom
+    end
+  end
+end