wechat_payment 0.1.0 → 0.2.0

data/app/models/wechat_payment/refund_order.rb

@@ -0,0 +1,127 @@
+module WechatPayment
+  class RefundOrder < ApplicationRecord
+    belongs_to :payment_order
+
+    before_create :gen_out_refund_no
+
+    belongs_to :goods, polymorphic: true
+    belongs_to :customer, polymorphic: true
+
+    enum state: {
+      pending: "pending",
+      refunded: "refunded",
+      failed: "failed"
+    }, _default: :pending
+
+    # 生成退款编号
+    def gen_out_refund_no
+      loop do
+        out_refund_no = "#{Time.current.to_i}#{SecureRandom.random_number(999_999_999)}"
+        record = WechatPayment::RefundOrder.find_by(out_refund_no: out_refund_no)
+
+        if record.blank?
+          self.out_refund_no = out_refund_no
+          break
+        end
+      end
+    end
+
+    def as_refund_params
+      slice(:out_trade_no, :out_refund_no, :refund_fee, :total_fee).to_options
+    end
+    # 发起退款成功
+    # @param [Hash] result
+    #
+    # result example:
+    #
+    # {
+    #   "return_code"=>"SUCCESS",
+    #   "return_msg"=>"OK",
+    #   "appid"=>"wxc5f2606121234cf",
+    #   "mch_id"=>"1363241234",
+    #   "sub_mch_id"=>"1525912341",
+    #   "nonce_str"=>"RsXVcs0GMg2p5NRD",
+    #   "sign"=>"F10AB3929B900DE4E189CA93B73D9D7A",
+    #   "result_code"=>"SUCCESS",
+    #   "transaction_id"=>"4200001199202106280049902399",
+    #   "out_trade_no"=>"1624867410475591608",
+    #   "out_refund_no"=>"1624867450917685776",
+    #   "refund_id"=>"50301108952021062810183695009",
+    #   "refund_channel"=>"",
+    #   "refund_fee"=>"1",
+    #   "coupon_refund_fee"=>"0",
+    #   "total_fee"=>"1",
+    #   "cash_fee"=>"1",
+    #   "coupon_refund_count"=>"0",
+    #   "cash_refund_fee"=>"1"
+    # }
+    def refund_apply_success(result)
+      if payment_order.goods.respond_to? :refund_apply_success
+        payment_order.goods.refund_apply_success(result)
+      end
+
+      update(
+        refund_id: result["refund_id"],
+        state: :pending
+      )
+
+      result
+    end
+
+    # 发起退款失败
+    def refund_apply_failure(result)
+      # TODO 没遇到过，待补充
+
+      if payment_order.goods.respond_to? :refund_apply_failure
+        payment_order.goods.refund_apply_failure(result)
+      end
+    end
+
+    # 退款成功(回调)
+    # @param [Hash] result
+    #
+    # result example:
+    #
+    # {
+    #   "out_refund_no"=>"1624873658515277479",
+    #   "out_trade_no"=>"1624873575281298144",
+    #   "refund_account"=>"REFUND_SOURCE_RECHARGE_FUNDS",
+    #   "refund_fee"=>"1",
+    #   "refund_id"=>"50301308842021062810182580986",
+    #   "refund_recv_accout"=>"招商银行信用卡4003",
+    #   "refund_request_source"=>"API",
+    #   "refund_status"=>"SUCCESS",
+    #   "settlement_refund_fee"=>"1",
+    #   "settlement_total_fee"=>"1",
+    #   "success_time"=>"2021-06-28 17:47:47",
+    #   "total_fee"=>"1",
+    #   "transaction_id"=>"4200001202202106280268010129"
+    # }
+    def refund_exec_success(result)
+      update(
+        state: :refunded,
+        refunded_at: Time.current
+      )
+
+      if payment_order.total_fee_refunded?
+        payment_order.update(state: :refunded, refunded_at: Time.current)
+      end
+
+      if payment_order.goods.respond_to? :refund_exec_success
+        payment_order.goods.refund_exec_success(result)
+      end
+
+      result
+    end
+
+    # 退款失败(回调)
+    def refund_exec_failure(result)
+      # TODO 待补充
+      if payment_order.goods.respond_to? :refund_exec_failure
+        payment_order.goods.refund_exec_failure(result)
+      end
+    end
+
+
+  end
+end

data/app/services/wechat_payment/service.rb

@@ -0,0 +1,74 @@
+
+module WechatPayment
+  class Service
+    attr_reader :client, :payment_order
+
+    def initialize(payment_order)
+      @client = WechatPayment::Client.new
+      @payment_order = payment_order
+    end
+
+    # 下单
+    def order
+      order_result = client.order(payment_order.as_order_params)
+
+      if order_result.success?
+        payment_order.payment_apply_success(order_result.data)
+      else
+        payment_order.payment_apply_failure(order_result.errors)
+      end
+
+      order_result
+    end
+
+    # 退款
+    def refund(refund_fee)
+      if !payment_order.balance_enough_to_refund?(refund_fee)
+        return WechatPayment::ServiceResult.new(message_type: :error, message: "Balance is not enough.")
+      end
+
+      refund_order = payment_order.create_refund_order(refund_fee)
+      refund_result = client.refund(refund_order.as_refund_params)
+
+      if refund_result.success?
+        refund_order.refund_apply_success(refund_result.data)
+      else
+        refund_order.refund_apply_failure(refund_result.errors)
+      end
+
+      refund_result
+    end
+
+    # 处理支付回调
+    def self.handle_payment_notify(notify_data)
+      result = WechatPayment::Client.handle_payment_notify(notify_data)
+      payment_order = WechatPayment::PaymentOrder.find_by(out_trade_no: notify_data["out_trade_no"])
+
+      if result.success? && payment_order.pending?
+        payment_order.with_lock do
+          payment_order.payment_exec_success(result.data)
+        end
+      else
+        payment_order.payment_exec_failure(result.errors)
+      end
+
+      result
+    end
+
+    # 处理退款回调
+    def self.handle_refund_notify(notify_data)
+      result = WechatPayment::Client.handle_refund_notify(notify_data)
+      refund_order = WechatPayment::RefundOrder.find_by(out_refund_no: result.data["out_refund_no"])
+
+      if result.success? && refund_order.pending?
+        refund_order.with_lock do
+          refund_order.refund_exec_success(result.data)
+        end
+      else
+        refund_order.refund_exec_failure(result.errors)
+      end
+
+      result
+    end
+  end
+end

data/app/views/layouts/wechat_payment/application.html.erb

@@ -0,0 +1,15 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>Wechat payment</title>
+  <%= csrf_meta_tags %>
+  <%= csp_meta_tag %>
+
+  <%= stylesheet_link_tag    "wechat_payment/application", media: "all" %>
+</head>
+<body>
+
+<%= yield %>
+
+</body>
+</html>

data/config/routes.rb

@@ -0,0 +1,4 @@
+WechatPayment::Engine.routes.draw do
+  post "/callback/payment", to: "callback#payment"
+  post "/callback/refund", to: "callback#refund"
+end

data/db/migrate/20210706075217_create_wechat_payment_payment_orders.rb

@@ -0,0 +1,22 @@
+class CreateWechatPaymentPaymentOrders < ActiveRecord::Migration[6.1]
+  def change
+    create_table :wechat_payment_payment_orders do |t|
+      t.string :open_id
+      t.string :out_trade_no
+      t.references :goods, polymorphic: true, null: false
+      t.references :customer, polymorphic: true, null: false
+      t.string :transaction_id
+      t.string :body
+      t.integer :total_fee
+      t.string :trade_type
+      t.string :spbill_create_ip
+      t.string :prepay_id
+      t.string :state
+      t.datetime :paid_at
+      t.datetime :refunded_at
+
+      t.timestamps
+    end
+    add_index :wechat_payment_payment_orders, :open_id
+  end
+end

data/db/migrate/20210706095205_create_wechat_payment_refund_orders.rb

@@ -0,0 +1,20 @@
+class CreateWechatPaymentRefundOrders < ActiveRecord::Migration[6.1]
+  def change
+    create_table :wechat_payment_refund_orders do |t|
+      t.integer :payment_order_id
+      t.references :goods, polymorphic: true, null: false
+      t.references :customer, polymorphic: true, null: false
+      t.integer :refund_fee
+      t.integer :total_fee
+      t.string :out_trade_no
+      t.string :out_refund_no
+      t.string :refund_id
+      t.string :state
+      t.datetime :refunded_at
+
+      t.timestamps
+
+      t.index :payment_order_id, name: "payment_id_on_refund_orders"
+    end
+  end
+end

data/lib/generators/wechat_payment/goods/USAGE

@@ -0,0 +1,8 @@
+Description:
+    Explain the generator
+
+Example:
+    bin/rails generate goods Thing
+
+    This will create:
+        what/will/it/create

data/lib/generators/wechat_payment/goods/goods_generator.rb

@@ -0,0 +1,5 @@
+class WechatPayment::GoodsGenerator < Rails::Generators::NamedBase
+  source_root File.expand_path('templates', __dir__)
+
+
+end

data/lib/generators/wechat_payment/install/USAGE

@@ -0,0 +1,8 @@
+Description:
+    Explain the generator
+
+Example:
+    bin/rails generate install Thing
+
+    This will create:
+        what/will/it/create

data/lib/generators/wechat_payment/install/install_generator.rb

@@ -0,0 +1,86 @@
+class WechatPayment::InstallGenerator < Rails::Generators::Base
+  source_root File.expand_path('templates', __dir__)
+
+  argument :goods, type: :string
+  argument :user, type: :string, default: :User
+
+
+  # 生成 initializer 文件
+  def gen_initializer_file
+    copy_file "initializer.rb", "config/initializers/wechat_payment.rb"
+  end
+
+  # 挂载 engine 到路由上
+  def mount_payment_engine
+    route %Q(mount WechatPayment::Engine => "/wechat_payment")
+  end
+
+  # 安装迁移文件
+  def copy_migration
+    rake "wechat_payment:install:migrations"
+  end
+
+  def add_concern_to_goods
+    goods_model_head_one = "class #{goods_model_name} < ApplicationRecord"
+    inject_into_file goods_model_file, after: goods_model_head_one do <<-GOODS_CONCERN
+
+  include WechatPayment::Concern::Goods
+#{def_custom_user_model}
+    GOODS_CONCERN
+    end
+  end
+
+  def add_concern_to_users
+    user_model_head_one = "class #{user_model_name} < ApplicationRecord"
+    inject_into_file user_model_file, after: user_model_head_one do <<-'USERS_CONCERN'
+
+  include WechatPayment::Concern::User
+    USERS_CONCERN
+    end
+  end
+
+  def add_concern_to_user_goods
+    user_goods_model_head_one = "class #{user_goods_model_name} < ApplicationRecord"
+    inject_into_file user_goods_model_file, after: user_goods_model_head_one do <<-'USER_GOOD_CONCERN'
+
+  include WechatPayment::Concern::UserGoods
+    USER_GOOD_CONCERN
+    end
+  end
+
+  private
+
+  def goods_model_file
+    "app/models/#{goods.to_s.underscore}.rb"
+  end
+
+  def user_model_file
+    "app/models/#{user.to_s.underscore}.rb"
+  end
+
+  def user_goods_model_file
+    "app/models/#{user.to_s.underscore}_#{goods.to_s.underscore}.rb"
+  end
+
+  def goods_model_name
+    goods.to_s.camelize
+  end
+
+  def user_model_name
+    user.to_s.camelize
+  end
+
+  def user_goods_model_name
+    user_model_name + goods_model_name
+  end
+
+  def def_custom_user_model
+    if user_model_name != 'User'
+      <<-DEF
+  self.user_model = "#{user_model_name}"
+  self.user_ref_field = "#{user_model_name.underscore}"
+  self.user_goods_model = "#{user_model_name}#{goods_model_name}"
+      DEF
+    end
+  end
+end

data/lib/generators/wechat_payment/install/templates/initializer.rb

@@ -0,0 +1,19 @@
+
+WechatPayment.setup do |config|
+  # 接收回调的域名
+  config.host = "https://xxx.com"
+
+  # 下面所有参数都需要改成你自己的小程序配置
+  config.appid = "wxc5c26065c6123456"
+  config.app_secret = "123456784aa91bb538867d3d2790b308"
+  config.mch_id = "112241802"
+  config.key = "123456723erivPO09irNNbh78u8udwFer"
+
+  # 证书可以在微信支付后台获取到，路径是相对于项目根路径，如果需要退款的话，则必须要证书
+  # config.cert_path = "config/apiclient_cert.p12"
+  config.cert_path = nil
+
+  config.sub_appid = "wx8f9f912623456789"
+  config.sub_mch_id = "1234911291"
+  config.sub_app_secret = "88888231e2f3a21152d163f61b99999"
+end

data/lib/generators/wechat_payment/routes/USAGE

@@ -0,0 +1,8 @@
+Description:
+    Explain the generator
+
+Example:
+    bin/rails generate routes Thing
+
+    This will create:
+        what/will/it/create

data/lib/generators/wechat_payment/routes/routes_generator.rb

@@ -0,0 +1,7 @@
+class WechatPayment::RoutesGenerator < Rails::Generators::NamedBase
+  source_root File.expand_path('templates', __dir__)
+
+  def mount_wechat_payment
+    route %Q(mount WechatPayment::Engine => "/wechat_payment")
+  end
+end

data/lib/tasks/wechat_payment_tasks.rake

@@ -2,3 +2,13 @@
 # task :wechat_payment do
 #   # Task goes here
 # end
+
+desc "Install Wechat Payment Engine"
+namespace :wechat_payment do
+  task :install, [:goods, :user] => :environment do |task, args|
+    Rake::Task["wechat_payment:install:migrations"].invoke
+    sh "rails g wechat_payment:initializer wechat_payment"
+    sh "rails g wechat_payment:routes wechat_payment"
+    sh "rails g wechat_payment:goods #{args.goods} user:#{args.user.presence || 'user'}"
+  end
+end

data/lib/wechat_payment.rb

@@ -1,6 +1,44 @@
 require "wechat_payment/version"
-require "wechat_payment/railtie"
+require "wechat_payment/engine"
 
 module WechatPayment
-  # Your code goes here...
+
+  class << self
+    attr_reader :apiclient_cert, :apiclient_key
+    attr_accessor :appid, :app_secret, :mch_id, :sub_appid, :sub_app_secret, :sub_mch_id, :key, :cert_path, :host
+  end
+
+  def self.setup
+    yield self if block_given?
+
+    if cert_path
+      set_apiclient_by_pkcs12(File.binread(cert_path), mch_id)
+    end
+  end
+
+  def self.set_apiclient_by_pkcs12(str, pass)
+    pkcs12 = OpenSSL::PKCS12.new(str, pass)
+    @apiclient_cert = pkcs12.certificate
+    @apiclient_key = pkcs12.key
+
+    pkcs12
+  end
+
+  def apiclient_cert=(cert)
+    @apiclient_cert = OpenSSL::X509::Certificate.new(cert)
+  end
+
+  def apiclient_key=(key)
+    @apiclient_key = OpenSSL::PKey::RSA.new(key)
+  end
+
+  def self.as_payment_params
+    {
+      appid: appid,
+      mch_id: mch_id,
+      sub_appid: sub_appid,
+      sub_mch_id: sub_mch_id
+    }
+  end
+
 end