a_marmita 0.0.1 → 0.5.8

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    NjMwMTZkMmMzYjQ5YThjMzdjMjk4NjE5MTU2NDhjNDZmNDUzYzgzNA==
+    YjhiMzBiYzFmNTAxZmRkOWUzOWRiMmE3MGUxOTY2NDM3NTE5NGIxNw==
   data.tar.gz: !binary |-
-    MzA3NmYyNDg1MzE4N2U1ZjE3MzI1MzQyNzE1Zjg1OWM1OTMzOWU0Nw==
+    ODc0ZjQ3MjgzNDVlODlhOTkzNTVjYTdlMDY3YzVjMzUzMDBiMDZmNw==
 SHA512:
   metadata.gz: !binary |-
-    ZThlM2E2MDcxYzE4MGU3OWQ5M2FlZjY4MjkzYmZlYTUwMzg4NmJjNWNhMWQ4
-    MDNjOTAwNDNiNTllMmNjZjdkNjAxM2ZhM2Y5N2FmMmYyNzkwZmQ4MjFlYmVh
-    NGE2MTc4NmU0YWU2M2E4MzNjZDU4Zjg0ODY0ZjNjNDZmM2VjMzI=
+    YzEwYzFmOGM2MjRkNzVmNTUwNmZjNDNiY2FkZDEzZDFiY2RjNjBjMzU4ODg5
+    ZmM0ZGZhODU5NmZhZmFkNmQxMzMwMTE0ZTlmZTk3NGUyZmEwNmY4ZWE4MGYx
+    Mzc3ODEzMjcwNDk5ZGRkZTQwNjdmMGEwNTFlOWU0NTg1MDk0YzQ=
   data.tar.gz: !binary |-
-    ZDc1ZGYxZGZhNGJkZTkwYzNmODFkM2I0MDUzYjUxNTBiYzQwNjIzYjlkMWUy
-    NzYyYzk5ODk0MWM3MDY3NTRlZTZhNTY1YmFlZGE5YmZiYTEyYTNiZjUwY2Rh
-    MGM5NzM1YmZmODEwN2EzNjdmNjg2MDE4MmJkYTI0MTZkNzg5Yjc=
+    Y2NjOWM1ZGEyZmM4ZTFjMmJmOTNiNzYwNDNhYzBiYzJlYjUzMWI2NmQxYWZi
+    NjA5ZDhmYjA1YTcxYTFhZmU5YjJjMDBkNDFjY2Q4ZWY1MGY4YzFiNTUyYjlm
+    ODg4YjdhNDNhOWU2Y2YxNWIxMGU2ZTg4NDJjNDQwZmE3ZWM5OWI=

data/.gitignore

@@ -16,3 +16,5 @@ tmp
 .yardoc
 _yardoc
 doc/
+
+run_me.rb

data/.rspec

@@ -0,0 +1,4 @@
+--tty
+--color
+--format documentation
+--format html -o "tmp/rspec_result.html"

data/Gemfile

@@ -6,4 +6,5 @@ gemspec
 group :development, :test do
   gem "rspec", "~> 2.11"
   gem "pry"
+  gem 'webmock'
 end

data/Gemfile.lock

@@ -1,12 +1,15 @@
 PATH
   remote: .
   specs:
-    a_marmita (0.0.1)
+    a_marmita (0.3.5)
 
 GEM
   remote: https://rubygems.org/
   specs:
+    addressable (2.3.6)
     coderay (1.1.0)
+    crack (0.4.2)
+      safe_yaml (~> 1.0.0)
     diff-lcs (1.2.5)
     domain_name (0.5.18)
       unf (>= 0.0.5, < 1.0.0)
@@ -41,10 +44,14 @@ GEM
     rspec-expectations (2.14.5)
       diff-lcs (>= 1.1.3, < 2.0)
     rspec-mocks (2.14.6)
+    safe_yaml (1.0.1)
     slop (3.5.0)
     unf (0.1.3)
       unf_ext
     unf_ext (0.0.6)
+    webmock (1.17.4)
+      addressable (>= 2.2.7)
+      crack (>= 0.3.2)
     webrobots (0.1.1)
 
 PLATFORMS
@@ -56,3 +63,4 @@ DEPENDENCIES
   mechanize
   pry
   rspec (~> 2.11)
+  webmock

data/lib/a_marmita.rb

@@ -1,40 +1,59 @@
+require 'mechanize'
+
+require "a_marmita/client_module"
+require "a_marmita/payment"
 require "a_marmita/version"
+require "a_marmita/helpers"
+require "a_marmita/scrapper"
+require "a_marmita/web_agent"
+require "a_marmita/cart/base"
+require "a_marmita/meals/base"
+require "a_marmita/orders/base"
+require "a_marmita/authentication"
 
 module AMarmita
 
-  extend self
-
-  def get_meals
-    
-    login('joao.goncalves@linkedcare.com', 'yEJHVd')
+  class Client
 
-    scrap_meals
+    include WebAgent
+    include Authentication
 
-  end
+    def cart
+      @cart ||= Cart::Base.new(self)
+    end
 
+    def meals
+      @meals ||= Meals::Base.new(self)
+    end
 
-  protected #################### PROTECTED ##################
+    def orders
+      @orders ||= Orders::Base.new(self)
+    end
 
-  def scrap_meals
-    page = agent.get "http://www.amarmita.com/ementa.php"
+    def payment
+      @payment ||= Payment.new(self)
+    end
 
-    page.parser.css('#conteudo table tr')
-    binding.pry
   end
 
-  def login(email, pass)
-    page = agent.get "http://www.amarmita.com/login.php?user=#{email}&psw=#{pass}"
+  module ClassMethods
 
-    page.body == "1"
-  end
+    def new_client
+      Client.new
+    end
 
-  def agent
-    return @agent if defined?(@agent)
+    def singleton
+      @singleton ||= Client.new
+    end
 
-    @agent = Mechanize.new.tap do |agent|
-      agent.log = Logger.new "mech.log"
-      agent.user_agent_alias = 'Mac Safari'
+    def method_missing(method, *args, &block)
+      return super unless singleton.respond_to?(method)
+      
+      singleton.send(method, *args)
     end
+
   end
 
+  extend ClassMethods
+  
 end

data/lib/a_marmita/authentication.rb

@@ -0,0 +1,29 @@
+module AMarmita
+
+  module Authentication
+
+    def self.include(base)
+      base.class_eval { attr_reader :email, :pass }
+    end
+
+    def set_credentials(email, pass)
+      @email, @pass = email, pass
+    end
+
+    def login(email = nil, pass = nil)
+      set_credentials(email, pass) unless email.nil?
+      
+      @logged = (get_page_body("http://www.amarmita.com/login.php?user=#{@email}&psw=#{@pass}") == "1")
+    end
+
+    def logged?(force = false)
+      @logged = nil if force
+
+      return @logged unless @logged.nil?
+
+      @logged = (get_page_body("http://www.amarmita.com/logged.php") == "1")
+    end
+
+  end
+
+end

data/lib/a_marmita/cart/base.rb

@@ -0,0 +1,66 @@
+require "a_marmita/cart/scrapper"
+
+module AMarmita
+  module Cart
+
+    class Base < ClientModule
+
+      def items
+        return :bad_login if cant_log_in?
+
+        page, last_date = get_page("http://www.amarmita.com/enc.php?op=4"), ''
+
+        content_rows = page.parser.css('table > tr')
+        
+        content_rows.map do |row|
+          new_date = row.css('.em_marmitalist_data')
+
+          last_date = Helpers.cart_date_parser(new_date.first.text) unless new_date.empty?
+
+          Cart::Scrapper.new(row).scrap(date: last_date)
+        end.compact
+      end
+
+      def add_item(id, date = nil, quantity = 1)
+        return :bad_login if cant_log_in?
+        
+        date = Helpers.format_date(date)
+
+        server_response = get_page_body("http://www.amarmita.com/enc.php?id=#{id}&qtd=#{quantity}&data=#{date}&op=1")
+
+        { "1" => :ok, "-2" => :bad_date }[server_response] || :error
+      end
+
+      def remove_item(id)
+        return :bad_login if cant_log_in?
+        
+        server_response = get_page_body("http://www.amarmita.com/enc.php?id=#{id}&op=2")
+
+        server_response == "1" ? :ok : :error
+      end
+
+      def total
+        return :bad_login if cant_log_in?
+        
+        total = get_page_body("http://www.amarmita.com/enc.php?op=5")
+
+        Helpers.to_float(total).first
+      end
+
+      def checkout(payment_method)
+        return :bad_login if cant_log_in?
+
+        case payment_method
+        when :through_mb
+          client.payment.through_mb
+        when :end_of_month
+          client.payment.end_of_month
+        else
+          raise 'Unknown payment method!'
+        end
+      end
+
+    end
+
+  end
+end

data/lib/a_marmita/cart/scrapper.rb

@@ -0,0 +1,50 @@
+module AMarmita
+  module Cart
+
+    class Scrapper < AMarmita::Scrapper
+      
+      DESCRIPTION, QUANTITY = 0, 1
+
+      
+      def run(options = {})
+        @attributes = {
+          date: options[:date],
+          id: Helpers.to_int(get_id),
+          description: get_description,
+          quantity: Helpers.to_int(get_quantity)
+        }
+      end
+      
+
+      protected ############## PROTECTED ##################
+
+      def get_id
+        get_xml_value(css_parser.css('a').first) do |xml_object|
+          get_id_from_javascript_function xml_object.attr('href')
+        end
+      end
+
+      def get_description
+        get_xml_value similar_entries[DESCRIPTION]
+      end
+
+      def get_quantity
+        get_xml_value similar_entries[QUANTITY]
+      end
+
+
+      private ################# PRIVATE ##################
+
+      def similar_entries
+        @similar_entries ||= css_parser.css('.em_marmitalist_desc')
+      end
+
+      def get_id_from_javascript_function(javascript_function)
+        id = javascript_function[(javascript_function.index('(') + 1)..-1]
+        id[0..(id.index(')') - 1)]
+      end
+
+    end
+
+  end
+end

data/lib/a_marmita/client_module.rb

@@ -0,0 +1,24 @@
+module AMarmita
+
+  class ClientModule
+
+    extend Forwardable
+
+    attr_reader :client
+
+    def initialize(client)
+      @client = client
+    end
+
+    def_delegators :client, :get_page_body, :get_page, :logged?, :login
+
+    
+    protected ################ PROTECTED #################
+
+    def cant_log_in?
+      !logged? && !login
+    end
+
+  end
+
+end

data/lib/a_marmita/helpers.rb

@@ -0,0 +1,62 @@
+# encoding: utf-8
+
+module AMarmita
+
+  module Helpers
+
+    extend self
+
+    MONTH_NAMES_PT = ["", "janeiro", "fevereiro", "março", "abril", "maio", "junho", "julho", "agosto", "setembro", "outubro", "novembro", "dezembro"]
+
+
+    def try(object, *a, &b)
+      if a.empty? && block_given?
+        yield object
+      else
+        object.respond_to?(a.first) ? object.public_send(*a, &b) : nil
+      end
+    end
+
+    def to_int(string)
+      string.nil? ? 0 : string.gsub(/[^0-9]/, '').to_i
+    end
+
+    def to_float(string)
+      return 0 if string.nil?
+      
+      values = string.scan(/\d+[,.]*\d*/).flatten.map(&:to_f)
+
+      values.length > 1 ? values : values.first
+    end
+
+    def cart_date_parser(string)
+      return nil if string.nil?
+
+      day, month = *string.scan(/.* - ([0-9]*) (.*)/).flatten
+
+      format_date "#{Date.today.year}-#{get_month_number(month)}-#{day}"
+    end
+
+    def date_parser(date)
+      if date.is_a?(String)
+        date = Date.parse(date) rescue nil
+      end
+
+      date
+    end
+
+    def format_date(date)
+      date = date_parser(date)
+
+      date = (Date.today + 1) unless date.respond_to?(:strftime)
+
+      date.strftime('%Y-%m-%d')
+    end
+
+    def get_month_number(month)
+      MONTH_NAMES_PT.index(month.downcase)
+    end
+    
+  end
+
+end

data/lib/a_marmita/meals/base.rb

@@ -0,0 +1,64 @@
+require "a_marmita/meals/scrapper"
+
+module AMarmita
+  module Meals
+
+    SOUP, MAIN_COURSE, DESSERT = 1, 2, 3
+
+    class Base < ClientModule
+    
+      def today_menu
+        page = get_page "http://www.amarmita.com/ementa.php"
+        
+        content_rows = page.parser.css('#conteudo > table > tr')
+        
+        { available_days: parse_links(page), list: parse_meals(content_rows) }
+      end
+
+      def soups(date = nil, week_number = nil)
+        get_meals(SOUP, date, week_number)
+      end
+
+      def main_courses(date = nil, week_number = nil)
+        get_meals(MAIN_COURSE, date, week_number)
+      end
+
+      def desserts(date = nil, week_number = nil)
+        get_meals(DESSERT, date, week_number)
+      end
+
+
+      protected ################### PROTECTED ################
+
+      def get_meals(type, date = nil, week_number = nil)
+        date = Helpers.date_parser(date) || Date.today
+        week_number ||= date.strftime('%U').to_i + 1
+
+        page = get_page "http://www.amarmita.com/encomendamain.php?week=#{week_number}&tipo=#{type}&ano=#{date.year}&mes=#{date.month}&dia=#{date.day}"
+        
+        content_rows = page.parser.css('table')[0].css('> tr')
+
+        parse_meals(content_rows)
+      end
+
+      def parse_links(page)
+        page.links.inject([]) do |links, link|
+          links << get_date_and_week_number_from_javascript_function(link.href) if link.href.include?('em_setday')
+          links
+        end
+      end
+
+      def parse_meals(content_rows)
+        content_rows.map { |row| Meals::Scrapper.new(row).scrap }.compact
+      end
+
+      def get_date_and_week_number_from_javascript_function(javascript_function)
+        args = javascript_function[(javascript_function.index('(') + 1)..-2].split(',')
+        
+        { date: args[0..2].join('-'), week_number: args[3] }
+      end
+
+    end
+
+  end
+end