magpie 0.8.6.2 → 0.8.8

data/lib/magpie/utils.rb

@@ -1,16 +1,28 @@
 # -*- coding: utf-8 -*-
+require 'open-uri'
+require 'hpricot'
+require 'iconv'
 require 'net/https'
 require 'uri'
 
 module Magpie
   module Utils
 
-    private
+    def dig(env)
+      status, header, body = @app.call env
+      req = Rack::Request.new env
+      doc = send_req_to @pay_gateway, req
+      red_text = (doc/@red_xpath).inner_text
+      red_text = (doc/@error_xpath).inner_text if red_text.blank? and @error_xpath
+      red_text = Iconv.iconv("UTF-8//IGNORE","GBK//IGNORE", red_text).to_s
+      return status, header, body, req, red_text
+    end
+
     def send_req_to(gw, req)
       text = case req.request_method
-        when "GET"; get_query(gw, req.query_string)
-        when "POST"; post_query(gw, req.params)
-      end
+             when "GET"; get_query(gw, req.query_string).body
+             when "POST"; post_query(gw, req.params).body
+             end
       doc = Hpricot text
     end
 
@@ -23,31 +35,50 @@ module Magpie
 
     def hash_to_xml(h = { })
       h.inject(""){ |xml, (k, v)|
-        xml << "<#{k}>"
-        Hash === v ? xml << hash_to_xml(v) : xml << v.to_s
-        xml << "</#{k}>"
+        case v
+        when Hash, String
+          xml << "<#{k}>"
+          xml << (Hash === v ? hash_to_xml(v) : v)
+          xml << "</#{k}>"
+        when Array
+          v.each{ |vv| xml << hash_to_xml(k => vv)}
+          xml
+        end
+
       }
     end
 
     def get_xml_body(env, am, red_text)
-      if red_text =~ /错误|\d+/
-        final_error = get_final_error red_text
-        am.valid?
-        xml_body = build_xml(:is_success => "F", :errors => am.errors.merge(:final => final_error))
-        env["magpie.errors.info"] = am.errors.merge(:final => final_error)
+      if red_text.blank?
+        notify_res = get_notify_res(am)
+        xml_body = build_xml(:payment_success => "Yes",  :business => notify_res )
       else
-        begin_at = Time.now
-        notify_res = am.send_notify
-        now = Time.now
-        env["magpie.notify"] = ["POST", am.notify_url, now.strftime("%d/%b/%Y %H:%M:%S"), now - begin_at, am.notify.inspect, notify_res ]
-        xml_body = build_xml(:is_success => "T")
+        am.valid?
+        xml_body = build_xml(:payment_success => "No", :errors => am.errors.merge(:final => red_text))
+        log_errors(am.errors.merge(:final => red_text))
       end
       xml_body
     end
 
-    # 在具体的中间件中重写
-    def get_final_error(red_text)
-      ""
+    def get_notify_res(am)
+      case am
+      when AlipayModel, ChinabankModel
+        notify_res = send_notify("POST", am.notify_url, am.notify).gsub(/<[^>]*>|<\/[^>]*> |\s/m, '')
+        method = "POST"
+      when TenpayModel
+        notify_res = send_notify("GET", am.notify_url, am.notify_string).gsub(/<[^>]*>|<\/[^>]*>|\s/m, '')
+        method = "GET"
+      end
+      log_notify(method, am.notify_url, am.notify, notify_res)
+      notify_res
+    end
+
+    def log_notify(method, notify_url, notify_params, notify_res)
+      Magpie.logger.info(FORMAT_NOTIFY % [method, notify_url, Time.now.strftime("%d/%b/%Y %H:%M:%S"), notify_params.inspect, notify_res])
+    end
+
+    def log_errors(errors = { })
+      Magpie.logger.info(errors.map{ |kv| "%s: %s" % kv }.join("\n"))
     end
 
     def start_http(url, req)
@@ -63,7 +94,6 @@ module Magpie
       url = URI.parse(url + "?" + q_string)
       req = Net::HTTP::Get.new("#{url.path}?#{url.query}")
       res = start_http(url, req)
-      res.body
     end
 
     def post_query(url, params)
@@ -71,7 +101,28 @@ module Magpie
       req = Net::HTTP::Post.new(url.path)
       req.set_form_data params
       res = start_http(url, req)
-      res.body
+    end
+
+    # 向商户系统发送通知
+    # @param [String, String, Hash] url是商户系统用来接收通知的url, notify是支付平台发过来的参数
+    # @return [String] 如果有异常需要你确认url是否有效
+    def send_notify(method, url, notify)
+      times = [4, 6, 2]
+      Rubber.try(3){|i|
+        timeout(times[3-i]) do
+          res = case method.to_s.upcase
+                when "GET"; get_query(url, notify)
+                when "POST"; post_query(url, notify)
+                end
+          case res
+          when Net::HTTPSuccess, Net::HTTPRedirection; res.body
+          else
+            raise "#{res.class}@#{res.code}"
+          end
+        end
+      }
+    rescue Exception => e
+      "发送通知时出现异常#{e}, 请确认#{url}在你的商户系统中可用, 比如#{url}是否可以#{method.upcase}方式接收其他应用的请求"
     end
 
   end

data/lib/middles/alipay.rb

@@ -9,22 +9,12 @@ module Magpie
     def initialize(app, pay_gateway = "https://www.alipay.com/cooperate/gateway.do")
       @app = app
       @pay_gateway = pay_gateway
+      @red_xpath = "//div[@id='Info']/div[@class='ErrorInfo']/div[@class='Todo']"
     end
 
     def call(env)
-      status, header, body = @app.call(env)
-      req = Rack::Request.new(env)
-      doc = send_req_to @pay_gateway, req
-      red_text = (doc/"//div[@id='Info']/div[@class='ErrorInfo']/div[@class='Todo']").inner_text
-      red_text = Iconv.iconv("UTF-8//IGNORE","GBK//IGNORE", red_text).to_s
-      am = AlipayModel.new(req.params)
-      [status, header, get_xml_body(env, am, red_text)]
-    end
-
-    private
-
-    def get_final_error(red_text)
-      red_text
+      status, header, body, req, red_text = dig env
+      [status, header, get_xml_body(env, AlipayModel.new(req.params), red_text)]
     end
 
   end

data/lib/middles/chinabank.rb

@@ -9,23 +9,13 @@ module Magpie
     def initialize(app, pay_gateway = "https://pay3.chinabank.com.cn/PayGate")
       @app = app
       @pay_gateway = pay_gateway
+      @red_xpath = "//strong[@class='red']"
     end
 
     def call(env)
-      status, header, body = @app.call(env)
-      req = Rack::Request.new(env)
-      doc = send_req_to @pay_gateway, req
-      red_text = Iconv.iconv("UTF-8//IGNORE","GBK//IGNORE", (doc/"//strong[@class='red']").inner_text).to_s
-      am = ChinabankModel.new(req.params)
-      [status, header, get_xml_body(env, am, red_text)]
+      status, header, body, req, red_text = dig env
+      [status, header, get_xml_body(env, ChinabankModel.new(req.params), red_text)]
     end
 
-    private
-
-    def get_final_error(red_text)
-      red_text.match(/出错了!(.*)/)[1]
-    end
-
-
   end
 end

data/lib/middles/snake.rb

@@ -0,0 +1,116 @@
+# -*- coding: utf-8 -*-
+require 'erb'
+
+module Magpie
+
+  class Snake
+    include Utils
+
+    def self.reg(snake, target, state)
+      proc{ instance_method("#{target}_#{state}").bind(snake).call}
+    end
+
+    def initialize(app, &block)
+      @app = app
+      @block = block
+    end
+
+    def call(env)
+      state, header, body = @app.call(env)
+      @block.call(self)
+      @req = Rack::Request.new(env)
+      @urls[@req.request_method].each { |path, lamb|
+        if @req.path_info =~ Regexp.new("^#{path}$")
+          body = lamb.call
+          break
+        end
+      }
+      [state, header, body]
+      rescue Exception => e
+      Magpie.logger.info(e.inspect + ":\n" + e.backtrace[0..8].join("\n"))
+      [500, header, "500, 请查看日志,了解异常原因"]
+    end
+
+    def tongue(target, contents = { })
+      @urls ||= { "GET" => { }, "POST" => { }}
+      states = [contents[:states]].flatten.compact
+      route("GET", target, states)
+      actions = [contents[:actions]].flatten.compact
+      route("POST", target, actions)
+    end
+
+    def reg(target, state)
+      self.class.reg(self, target, state)
+    end
+
+    def route(method, target, states)
+      routes = states.inject({ }){ |h, state|
+        url_path = "/#{target}/#{state}"
+        h[url_path] = reg(target, state)
+        h["/#{target}"] = reg(target, state) if state.to_s == "index"
+        h
+      }
+      @urls[method.to_s.upcase].merge!(routes)
+    end
+
+    def alipay_index
+      @am = AlipayModel.new(@req.params)
+      @title = "支付宝-收银台"
+      render_success_or_fail
+    end
+
+    def chinabank_index
+      @am = ChinabankModel.new(@req.params)
+      @title = "网银在线-收银台"
+      render_success_or_fail
+    end
+
+    def tenpay_index
+      @am = TenpayModel.new(@req.params)
+      @title = "财付通-收银台"
+      render_success_or_fail
+    end
+
+    def order_pay
+      return "支付失败, 缺少足够的参数" if @req.params.blank?
+      case @req.params["notify_kind"]
+      when "alipay", "chinabank"
+        notify_res = send_notify("POST", @req.params["notify_url"], query_to_hash(@req.params["notify"]))
+        method = "POST"
+      when "tenpay"
+        notify_res = send_notify("GET", @req.params["notify_url"], @req.params["notify"])
+        method = "GET"
+      end
+      log_notify(method, @req.params["notify_url"], query_to_hash(@req.params["notify"]), notify_res)
+      notify_res
+    end
+
+    private
+
+    def render(file_name, options = { })
+      layout = options["layout"] || "layouts/app.html.erb"
+      file_path = File.join(File.dirname(__FILE__), "../..", "lib", "views", "#{file_name}.html.erb")
+      layout_path = File.join(File.dirname(__FILE__), "../..", "lib", "views", layout)
+      template = ERB.new(File.read(file_path))
+      layout = ERB.new(File.read(layout_path))
+      @yield = template.result(binding)
+      layout.result(binding)
+    end
+
+    def query_to_hash(query)
+      hash_params = query.split("&").inject({ }){ |h, q| qs = q.split("="); h[qs[0]] = qs[1]; h }
+    end
+
+    def render_success_or_fail
+      if @am.valid?
+        @dung = Dung.new(@am)
+        render("success")
+      else
+        log_errors(@am.errors)
+        render("fail")
+      end
+    end
+
+  end
+
+end

data/lib/middles/tenpay.rb

@@ -0,0 +1,22 @@
+
+require 'models/tenpay'
+
+module Magpie
+
+  class Tenpay
+    include Utils
+
+    def initialize(app, pay_gateway = "http://service.tenpay.com/cgi-bin/v3.0/payservice.cgi")
+      @app = app
+      @pay_gateway = pay_gateway
+      @red_xpath = "//div[@id='error-info']"
+      @error_xpath = "//td[@class='font_14']"
+    end
+
+    def call(env)
+      status, header, body, req, red_text = dig env
+      [status, header, get_xml_body(env, TenpayModel.new(req.params), red_text)]
+    end
+
+  end
+end

data/lib/models/alipay.rb

@@ -1,151 +1,92 @@
 # -*- coding: utf-8 -*-
 
-class AlipayModel
-  include ActiveModel::Validations
-  attr_accessor :service,
-  :partner,
-  :notify_url,
-  :return_url,
-  :sign,
-  :sign_type,
-  :subject,
-  :out_trade_no,
-  :payment_type,
-  :show_url,
-  :body,
-  :price,
-  :total_fee,
-  :quantity,
-  :seller_email,
-  :seller_id,
-  :_input_charset
-
-  validates_presence_of :service, :partner, :notify_url, :return_url, :sign, :sign_type, :subject, :out_trade_no, :payment_type
-  validates_length_of :partner, :maximum => 16
-  validates_length_of :notify_url, :return_url, :maximum => 190
-  validates_length_of :show_url, :maximum => 400
-  validates_length_of :body, :maximum => 1000
-  validates_length_of :out_trade_no, :maximum => 64
-  validates_length_of :payment_type, :maximum => 4
-  validates_format_of :price, :total_fee,
-  :with => /^[0-9]{1,9}\.[0-9]{1,2}$/,
-  :allow_blank => true,
-  :message => "format should be Number(13, 2)"
-  validates_numericality_of :price, :total_fee,
-  :greater_than_or_equal_to => 0.01,
-  :less_than_or_equal_to => 100000000.00,
-  :allow_blank => true,
-  :message => "should between 0.01~100000000.00"
-  validates_numericality_of :quantity,
-  :only_integer => true,
-  :greater_than => 0,
-  :less_than => 1000000,
-  :allow_blank => true,
-  :message => "should be integer and between 1~999999"
-  validates_inclusion_of :_input_charset, :in => %w(utf-8 gb2312), :message => "should be utf-8 or gb2312", :allow_blank => true
-
-  validate do |am|
-    am.errors[:money] << "price和total_fee不能同时出现" if am.repeat_money?
-    am.errors[:money] << "price and total_fee can not both be blank" if am.money_blank?
-    am.errors[:quantity] << "if price is not blank, must input quantity" if am.price_missing_quantity?
-    am.errors[:seller] << "seller_email and seller_id can not both be blank" if am.seller_blank?
-    am.errors[:sign] << "invalid sign" if am.invalid_sign?
-    am.errors[:partner] << "not exist" if am.missing_partner?
-  end
+module Magpie
 
-  def initialize(attributes = {})
-    @attributes = attributes
-    attributes.each do |name, value|
-      send("#{name}=", value) if respond_to? name
-    end
-  end
+  class AlipayModel
 
-  def repeat_money?
-    self.price.to_s.length > 0 and self.total_fee.to_s.length > 0
-  end
+    include Goose
+    include Mouse
 
-  def price_missing_quantity?
-    self.price.to_s.length > 0 and self.quantity.to_s.length == 0
-  end
+    set_accounts_kind :alipay
 
-  def missing_partner?
-    return if self.partner.to_s.length == 0
-    self.account == [] ? true : false
-  end
+    attr_accessor :service, :partner, :notify_url, :return_url, :sign, :sign_type, :subject, :out_trade_no
 
-  def seller_blank?
-    self.seller_id.to_s.length == 0 and self.seller_email.to_s.length == 0
-  end
+    attr_accessor :payment_type, :show_url, :body, :price, :total_fee, :quantity, :seller_email, :seller_id
 
-  def money_blank?
-    self.price.to_s.length == 0 and self.total_fee.to_s.length == 0
-  end
+    attr_accessor :_input_charset
 
-  def invalid_sign?
-    attrs = @attributes.dup
-    attrs.delete("sign")
-    attrs.delete("sign_type")
-    text = attrs.delete_if{ |k, v| v.to_s.length == 0 }.sort.collect{ |s| s[0] + "=" + URI.decode(s[1]) }.join("&") + self.key
-    self.sign == Digest::MD5.hexdigest(text) ? false : true
-  end
+    goose_validate :presence_attributes, :length_attributes, :format_attributes do |am|
+      am.errors[:money] << "price和total_fee不能同时出现" if am.repeat_money?
+      am.errors[:money] << "price and total_fee can not both be blank" if am.money_blank?
+      am.errors[:quantity] << "if price is not blank, must input quantity" if am.price_missing_quantity?
+      am.errors[:seller] << "seller_email and seller_id can not both be blank" if am.seller_blank?
+      am.errors[:sign] << "invalid sign" if am.invalid_sign?
+      am.errors[:partner] << "not exist" if !am.partner.blank? and am.missing_partner?
+      am.errors[:_input_charset] << "should be utf-8 or gb2312" unless am._input_charset.blank? or %w(utf-8 gb2312).member?(am._input_charset)
+    end
 
-  def account
-    @account ||= self.class.accounts.assoc self.partner
-    @account ||= []
-  end
+    def repeat_money?
+      self.price.to_s.length > 0 and self.total_fee.to_s.length > 0
+    end
 
-  def key
-    self.account[1].to_s
-  end
+    def price_missing_quantity?
+      self.price.to_s.length > 0 and self.quantity.blank?
+    end
 
-  def self.accounts
-    @accounts ||= YAML.load_file('test/partner.yml')['alipay'] if ENV['magpie'] == 'test'
-    @accounts ||= Magpie.yml_db['alipay']
-  end
+    def seller_blank?
+      self.seller_id.blank? and self.seller_email.blank?
+    end
 
+    def money_blank?
+      self.price.blank? and self.total_fee.blank?
+    end
 
-  def notify
-    @notify ||= notify_attrs.inject({ }){ |notify, attr|
-      notify[attr] = self.send(attr)
-      notify
-    }.merge("sign_type" => sign_type, "sign" => notify_sign)
-  end
+    def invalid_sign?
+      attrs = @attributes.dup
+      attrs.delete("sign")
+      attrs.delete("sign_type")
+      text = attrs.delete_if{ |k, v| v.blank? }.sort.collect{ |s| s[0] + "=" + URI.decode(s[1]) }.join("&") + self.key
+      self.sign == Digest::MD5.hexdigest(text) ? false : true
+    end
 
-  def send_notify
-    url = URI.parse notify_url
-    res = Net::HTTP.post_form url, self.notify
-    res.body
-  end
+    def notify
+      @notify ||= notify_attrs.inject({ }){ |notify, attr|
+        notify[attr] = self.send(attr)
+        notify
+      }.merge("sign_type" => sign_type, "sign" => notify_sign)
+    end
 
-  private
-  def notify_id
-    @notify_id ||= Time.now.to_i
-  end
 
-  def notify_time
-    @notify_time ||= Time.now.strftime("%Y-%m-%d %H:%M:%S")
-  end
+    private
 
-  def notify_sign
-    @notify_sign ||= Digest::MD5.hexdigest(notify_text).downcase
-  end
+    def notify_id
+      @notify_id ||= Time.now.to_i
+    end
 
-  def notify_text
-     @notify_text ||= notify_attrs.sort.collect{ |attr|
-      "#{attr}=#{self.send(attr)}"
-    }.join("&") + self.key
-  end
+    def notify_time
+      @notify_time ||= Time.now.strftime("%Y-%m-%d %H:%M:%S")
+    end
 
-  def trade_no
-    @trade_no ||= Time.now.to_i.to_s + rand(1000000).to_s
-  end
+    def notify_sign
+      @notify_sign ||= Digest::MD5.hexdigest(notify_text).downcase
+    end
 
-  def trade_status
-    @trade_status ||= %w(TRADE_FINISHED TRADE_SUCCESS)[rand(2)]
-  end
+    def notify_text
+      @notify_text ||= notify_attrs.sort.collect{ |attr|
+        "#{attr}=#{self.send(attr)}"
+      }.join("&") + self.key
+    end
+
+    def trade_no
+      @trade_no ||= Time.now.to_i.to_s + rand(1000000).to_s
+    end
 
-  def notify_attrs
-    @notify_attrs ||= %w{ notify_id
+    def trade_status
+      @trade_status ||= %w(TRADE_FINISHED TRADE_SUCCESS)[rand(2)]
+    end
+
+    def notify_attrs
+      @notify_attrs ||= %w{ notify_id
         notify_time
         trade_no
         out_trade_no
@@ -166,7 +107,35 @@ class AlipayModel
         gmt_refund
         use_coupon
       }.select{ |attr| self.respond_to?(attr, true) && self.send(attr).to_s.length > 0 }
-  end
+    end
+
+
+    def presence_attributes
+      [:service, :partner, :notify_url, :return_url, :sign, :sign_type, :subject, :out_trade_no, :payment_type].each {|attr|
+        self.errors[attr] << "can't be blank" if self.send(attr).blank?
+      }
+    end
+
+    def length_attributes
+      errors[:partner] << length_error_msg(16) if self.partner.to_s.length > 16
+      [:notify_url, :return_url].each { |attr| self.errors[attr] << length_error_msg(190) if self.send(attr).to_s.length > 190}
+      errors[:show_url] << length_error_msg(400) if self.show_url.to_s.length > 400
+      errors[:body] << length_error_msg(1000) if body.to_s.length > 1000
+      errors[:out_trade_no] << length_error_msg(64) if out_trade_no.to_s.length > 64
+      errors[:payment_type] << length_error_msg(4) if payment_type.to_s.length > 4
+    end
 
+    def format_attributes
+      [:price, :total_fee].each { |attr|
+        self.errors[attr] << "format should be Number(13, 2)" unless self.send(attr).blank? or self.send(attr) =~  /^[0-9]{1,9}\.[0-9]{1,2}$/
+      }
+      self.errors[:quantity] << "should be integer and between 1~999999" unless self.quantity.blank? or self.quantity =~ /^[1-9]{1,6}$/
+    end
+
+    def length_error_msg(length)
+      "is too long (maximum is #{length} characters)"
+    end
+
+  end
 
 end