rails_wechat 0.1.0 → 0.1.1

data/app/apis/wechat/api/work.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+module Wechat::Api
+  class Work < Base
+    include Agent
+    include Menu
+    include User
+
+    def initialize(app)
+      super
+      @agentid = app.agentid
+    end
+
+  end
+end

data/app/apis/wechat/api.rb

@@ -0,0 +1,7 @@
+require 'wechat/http_client'
+
+require 'wechat/api/base'
+require 'wechat/api/platform'
+require 'wechat/api/public'
+require 'wechat/api/program'
+require 'wechat/api/work'

data/app/apis/wechat/app_not_found.rb

@@ -0,0 +1,3 @@
+module Wechat
+  class AppNotFound < StandardError; end
+end

data/app/apis/wechat/cipher.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+require 'openssl/cipher'
+require 'base64'
+
+module Wechat::Cipher
+  BLOCK_SIZE = 32
+  CIPHER = 'AES-256-CBC'
+  extend self
+  
+  def encrypt(plain, encoding_aes_key)
+    cipher = OpenSSL::Cipher.new(CIPHER)
+    cipher.encrypt
+
+    cipher.padding = 0
+    key_data = Base64.decode64(encoding_aes_key + '=')
+    cipher.key = key_data
+    cipher.iv = [key_data].pack('H*')
+
+    cipher.update(plain) + cipher.final
+  end
+
+  def decrypt(msg, encoding_aes_key)
+    cipher = OpenSSL::Cipher.new(CIPHER)
+    cipher.decrypt
+
+    cipher.padding = 0
+    key_data = Base64.decode64(encoding_aes_key + '=')
+    cipher.key = key_data
+    cipher.iv = [key_data].pack('H*')
+
+    plain = cipher.update(msg) + cipher.final
+    decode_padding(plain)
+  end
+
+  def program_decrypt(encrypted_data, iv, session_key)
+    cipher = OpenSSL::Cipher.new('AES-128-CBC')
+    cipher.decrypt
+
+    cipher.key = Base64.decode64(session_key)
+    cipher.iv = Base64.decode64(iv)
+    decrypted_data = Base64.decode64(encrypted_data)
+    
+    JSON.parse(cipher.update(decrypted_data) + cipher.final)
+  rescue Exception => e
+    { errcode: 41003, errmsg: e.message }
+  end
+
+  def pack(content, app_id)
+    random = SecureRandom.hex(8)
+    text = content.force_encoding('ASCII-8BIT')
+    msg_len = [text.length].pack('N')
+
+    encode_padding("#{random}#{msg_len}#{text}#{app_id}")
+  end
+
+  def unpack(msg)
+    msg = decode_padding(msg)
+    msg_len = msg[16, 4].reverse.unpack('V')[0]
+    content = msg[20, msg_len]
+    app_id = msg[(20 + msg_len)..-1]
+
+    [content, app_id]
+  end
+
+  private
+  def encode_padding(data)
+    length = data.bytes.length
+    amount_to_pad = BLOCK_SIZE - (length % BLOCK_SIZE)
+    amount_to_pad = BLOCK_SIZE if amount_to_pad == 0
+    padding = ([amount_to_pad].pack('c') * amount_to_pad)
+    data + padding
+  end
+
+  def decode_padding(plain)
+    pad = plain.bytes[-1]
+    # no padding
+    pad = 0 if pad < 1 || pad > BLOCK_SIZE
+    plain[0...(plain.length - pad)]
+  end
+  
+end

data/app/apis/wechat/http_client.rb

@@ -0,0 +1,80 @@
+require 'httpx'
+require 'http/form_data'
+
+module Wechat
+  class HttpClient
+    attr_reader :http
+
+    def initialize
+      @http = HTTPX.with(**RailsWechat.config.httpx)
+    end
+
+    def get(path, headers: {}, params: {}, base: nil, **options)
+      headers['Accept'] ||= 'application/json'
+      url = base + path
+
+      response = @http.with_headers(headers).get(url, params: params)
+      parse_response(response, options[:as])
+    end
+
+    def post(path, payload, headers: {}, params: {}, base: nil, **options)
+      headers['Accept'] ||= 'application/json'
+      headers['Content-Type'] ||= 'application/json'
+      url = base + path
+
+      response = @http.with_headers(headers).post(url, params: params, body: payload)
+      parse_response(response, options[:as])
+    end
+
+    def post_file(path, file, headers: {}, params: {}, base: nil, **options)
+      headers['Accept'] ||= 'application/json'
+      url = base + path
+
+      form_file = file.is_a?(HTTP::FormData::File) ? file : HTTP::FormData::File.new(file)
+      response = @http.plugin(:multipart).with_headers(headers).post(
+        url,
+        params: params,
+        form: { media: form_file }
+      )
+      parse_response(response, options[:as])
+    end
+
+    private
+    def parse_response(response, parse_as)
+      raise "Request get fail, response status #{response.status}" if response.status != 200
+
+      content_type = response.content_type.mime_type
+      body = response.body.to_s
+
+      if content_type =~ /image|audio|video/
+        data = Tempfile.new('tmp')
+        data.binmode
+        data.write(body)
+        data.rewind
+        return data
+      elsif content_type =~ /html|xml/
+        data = Hash.from_xml(body)
+      else
+        data = JSON.parse body.gsub(/[\u0000-\u001f]+/, '')
+      end
+
+      case data['errcode']
+      when 0 # for request didn't expect results
+        data
+        # 42001: access_token timeout
+        # 40014: invalid access_token
+        # 40001, invalid credential, access_token is invalid or not latest hint
+        # 48001, api unauthorized hint, should not handle here # GH-230
+      when 42001, 40014, 40001, 41001
+        raise Wechat::AccessTokenExpiredError
+        # 40029, invalid code for mp # GH-225
+        # 43004, require subscribe hint # GH-214
+      when 2
+        raise Wechat::ResponseError.new(data['errcode'], data['errmsg'])
+      else
+        data
+      end
+    end
+
+  end
+end

data/app/apis/wechat/invalid_credential_error.rb

@@ -0,0 +1,3 @@
+module Wechat
+  class InvalidCredentialError < WechatError; end
+end

data/app/apis/wechat/message/base.rb

@@ -0,0 +1,17 @@
+class Wechat::Message::Base
+
+
+  def content_blank?
+    case @message_hash['MsgType']
+    when 'image', 'voice', 'video', 'music'
+      !@message_hash.key?(@message_hash['MsgType'].classify)
+    when 'news'
+      !@message_hash.key?('Articles')
+    when 'text'
+      @message_hash[:Content].blank?
+    else
+      false
+    end
+  end
+
+end

data/app/apis/wechat/message/mass/public.rb

@@ -0,0 +1,40 @@
+# https://developers.weixin.qq.com/doc/offiaccount/Message_Management/Batch_Sends_and_Originality_Checks.html
+class Wechat::Message::Mass::Public < Wechat::Message::Mass
+  MSG_TYPE = {
+    mpnews: 'media_id',
+    voice: 'media_id',
+    image: 'media_id',
+    mpvideo: 'media_id',
+    text: 'content',
+    wxcard: 'card_id'
+  }.freeze
+  
+  def to_mass(tag_id = nil, **options)
+    if tag_id
+      @message_hash.merge!(filter: { is_to_all: false, tag_id: tag_id }, **options)
+    else
+      @message_hash.merge!(filter: { is_to_all: true }, **options)
+    end
+    @send_to = 'message/mass/sendall'
+    self
+  end
+  
+  def do_send
+    api.post @send_to, @message_hash
+  end
+
+  def to(*openid, **options)
+    raise 'must send to more than 2 uses!' if openid.size < 2
+    @message_hash.merge!(touser: openid, **options)
+    @send_to = 'message/mass/send'
+    self
+  end
+
+  def content(msgtype, body)
+    @message_hash.merge! msgtype: msgtype
+    @message_hash[msgtype] = {
+      MSG_TYPE[msgtype.to_sym] => body
+    }
+  end
+  
+end

data/app/apis/wechat/message/mass/work.rb

@@ -0,0 +1,84 @@
+class Wechat::Message::Mass::Work < Wechat::Message::Mass
+  # see: https://work.weixin.qq.com/api/doc#90000/90135/90236/消息类型
+  MSG_TYPE = {
+    text: 'content',
+    markdown: 'content',
+    voice: 'media_id',
+    image: 'media_id',
+    file: 'media_id',
+    video: ['media_id', 'title', 'description'],
+    textcard: ['title', 'description', 'url', 'btntxt'],
+    news: ['articles', 'title', 'url', 'description', 'picurl'],
+    mpnews:  ['articles', 'title', 'thumb_media_id', 'content', 'author', 'content_source_url', 'digest'],
+    miniprogram_notice: ['title', 'description', 'url', 'btntxt'],
+    taskcard: ['title', 'description', 'url', 'btntxt']
+  }.freeze
+
+  def restore
+    case msgtype
+    when 'news', 'mpnews'
+      @message_hash[msgtype] = { articles: @message_hash.delete('articles') }
+    end
+    
+    super
+  end
+
+  def to(*user)
+    @message_hash.merge!(touser: user.join('|'))
+  end
+
+  def to_all
+    @message_hash.merge!(touser: '@all')
+  end
+  
+  def to_party(*party)
+    @message_hash.merge!(toparty: party.join('|'))
+  end
+  
+  def to_tag(*tag)
+    @message_hash.merge!(totag: tag.join('|'))
+  end
+
+  def agent_id(agentid)
+    @message_hash.merge!(AgentId: agentid)
+  end
+
+  def textcard(title, description, url, btntxt = nil)
+    data = {
+      title: title,
+      description: description,
+      url: url
+    }
+    data[:btntxt] = btntxt if btntxt.present?
+    @message_hash.merge!(MsgType: 'textcard', TextCard: data)
+  end
+
+  def markdown(content)
+    @message_hash.merge!(MsgType: 'markdown', Markdown: {
+      content: content
+    })
+  end
+
+  def ref_mpnews(media_id)
+    @message_hash.merge!(MsgType: 'ref_mpnews', MpNews: { MediaId: media_id })
+  end
+
+  def mpnews(collection, &_block)
+    if block_given?
+      article = MpNewsArticleBuilder.new
+      collection.take(8).each_with_index { |item, index| yield(article, item, index) }
+      items = article.items
+    else
+      items = collection.collect do |item|
+        camelize_hash_keys(item.symbolize_keys.slice(:thumb_media_id, :title, :content, :author, :content_source_url, :digest, :show_cover_pic).reject { |_k, v| v.nil? })
+      end
+    end
+
+    @message_hash.merge!(MsgType: 'mpnews', Articles: items.collect { |item| camelize_hash_keys(item) })
+  end
+
+  def file(media_id)
+    @message_hash.merge!(MsgType: 'file', File: { MediaId: media_id })
+  end
+
+end

data/app/apis/wechat/message/mass.rb

@@ -0,0 +1,22 @@
+class Wechat::Message::Mass < Wechat::Message::Base
+  attr_reader :msgtype
+
+  def initialize(app, msg = {})
+    super
+    @send_to = nil
+    @msgtype = @message_hash['msgtype'].to_s
+    restore
+  end
+
+  def do_send
+
+  end
+
+  def restore
+    case msgtype
+    when 'text', 'markdown'
+      @message_hash[msgtype] = { content: @message_hash.delete('content') }
+    end
+  end
+
+end

data/app/apis/wechat/message.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+module Wechat::Message; end
+require 'wechat/message/base'
+
+require 'wechat/message/mass'
+require 'wechat/message/mass/public'
+require 'wechat/message/mass/work'

data/app/apis/wechat/response_error.rb

@@ -0,0 +1,10 @@
+module Wechat
+  class ResponseError < StandardError
+    attr_reader :error_code
+
+    def initialize(errcode, errmsg)
+      @error_code = errcode
+      super "#{errmsg}(#{error_code})"
+    end
+  end
+end

data/app/apis/wechat/signature.rb

@@ -0,0 +1,32 @@
+module Wechat
+  module Signature
+    extend self
+
+    def hexdigest(token, timestamp, nonce, msg_encrypt = nil)
+      array = [token, timestamp, nonce]
+      array << msg_encrypt if msg_encrypt
+      dev_msg_signature = array.compact.collect(&:to_s).sort.join
+      Digest::SHA1.hexdigest(dev_msg_signature)
+    end
+
+    # Obtain the wechat jssdk config signature parameter and return below hash
+    #  params = {
+    #    noncestr: noncestr,
+    #    timestamp: timestamp,
+    #    jsapi_ticket: ticket,
+    #    url: url,
+    #    signature: signature
+    #  }
+    def signature(ticket, url)
+      params = {
+        noncestr: SecureRandom.base64(16),
+        timestamp: Time.current.to_i,
+        jsapi_ticket: ticket,
+        url: url
+      }
+      result = Digest::SHA1.hexdigest params.to_qeury
+      params.merge(signature: result)
+    end
+
+  end
+end

data/app/apis/wechat/wechat_error.rb

@@ -0,0 +1,3 @@
+module Wechat
+  class WechatError < StandardError; end
+end

data/app/controllers/wechat/admin/accounts_controller.rb

@@ -0,0 +1,25 @@
+module Wechat
+  class Admin::AccountsController < Admin::BaseController
+    before_action :set_account, only: [:qrcode]
+
+    def qrcode
+      @account.qrcode
+    end
+
+    private
+    def set_account
+      @account = Account.find(params[:id])
+    end
+
+    def account_params
+      params.fetch(:account, {}).permit(
+        :user_id,
+        :type,
+        :identity,
+        :confirmed,
+        :primary
+      )
+    end
+
+  end
+end