Merge pull request #92 from jasl/refactor-multi_account_in_sandbox_model

向后兼容性

Merge pull request #92 from jasl/refactor-multi_account_in_sandbox_model
向后兼容性
Jun Jiang · GitHub
2 parents 0a174775 5f14dc16
Showing 3 changed files with 20 additions and 4 deletions Show diff stats
lib/wx_pay.rb
lib/wx_pay/service.rb
lib/wx_pay/sign.rb
@@ -11,7 +11,7 @@ module WxPay
   class<< self
     attr_accessor :appid, :mch_id, :key, :appsecret, :extra_rest_client_options, :debug_mode
-    attr_accessor :sandbox_mode
+    attr_accessor :sandbox_mode, :sandbox_key
     attr_reader :apiclient_cert, :apiclient_key
     def set_apiclient_by_pkcs12(str, pass)
@@ -27,7 +27,7 @@ module WxPay
       ), quirks_mode: true)
     end
-    def self.get_sandbox_signkey(mch_id)
+    def self.get_sandbox_signkey(mch_id = WxPay.mch_id)
       params = {
         mch_id: mch_id,
         nonce_str: SecureRandom.uuid.tr('-', '')
@@ -453,11 +453,24 @@ module WxPay
       def xmlify_payload(params, sign_type = WxPay::Sign::SIGN_TYPE_MD5)
         sign = WxPay::Sign.generate(params, sign_type)
-        params.delete(:key) if params[:key]
-        "<xml>#{params.map { |k, v| "<#{k}>#{v}</#{k}>" }.join}<sign>#{sign}</sign></xml>"
+        "<xml>#{params.except(:key).map { |k, v| "<#{k}>#{v}</#{k}>" }.join}<sign>#{sign}</sign></xml>"
       end
       def make_payload(params, sign_type = WxPay::Sign::SIGN_TYPE_MD5)
+        # TODO: Move this out
+        if WxPay.sandbox_mode? && !params[:key]
+          r = get_sandbox_signkey
+          if r['return_code'] == WxPay::Result::SUCCESS_FLAG
+            params = params.merge(
+              mch_id: r['mch_id'] || WxPay.mch_id,
+              key: r['sandbox_signkey']
+            )
+            WxPay.sandbox_key = r['sandbox_signkey']
+          else
+            warn("WxPay Warn: fetch sandbox sign key failed #{r['return_msg']}")
+          end
+        end
+
         xmlify_payload(params, sign_type)
       end
@@ -27,6 +27,9 @@ module WxPay
     def self.verify?(params, options = {})
       params = params.dup
       params = params.merge(options)
+      if WxPay.sandbox_mode? && !params[:key]
+        params[:key] = WxPay.sandbox_key
+      end
       sign = params.delete('sign') || params.delete(:sign)
       generate(params) == sign
     end
	@@ -11,7 +11,7 @@ module WxPay		@@ -11,7 +11,7 @@ module WxPay
11		11
12	class<< self	12	class<< self
13	attr_accessor :appid, :mch_id, :key, :appsecret, :extra_rest_client_options, :debug_mode	13	attr_accessor :appid, :mch_id, :key, :appsecret, :extra_rest_client_options, :debug_mode
14	- attr_accessor :sandbox_mode	14	+ attr_accessor :sandbox_mode, :sandbox_key
15	attr_reader :apiclient_cert, :apiclient_key	15	attr_reader :apiclient_cert, :apiclient_key
16		16
17	def set_apiclient_by_pkcs12(str, pass)	17	def set_apiclient_by_pkcs12(str, pass)