rails_wechat 0.1.0 → 0.1.1

data/app/models/wechat/ext/user_tag.rb

@@ -0,0 +1,38 @@
+module Wechat
+  module Ext::UserTag
+    extend ActiveSupport::Concern
+
+    included do
+      has_many :tags, class_name: 'Wechat::Tag', dependent: :destroy
+
+      after_save_commit :sync_tag_later, if: -> { saved_change_to_name? }
+    end
+
+    def sync_tag_later
+      Wechat::TagJob.perform_later(self)
+    end
+
+    def sync_tag
+      _app = app
+      return unless _app
+      wt = tags.find_or_initialize_by(appid: _app.appid)
+      wt.name = self.name
+      wt.save
+    end
+
+    def app
+      if self.respond_to? :organ_id
+        _organ_id = organ_id
+      else
+        _organ_id = nil
+      end
+
+      if App.column_names.include?('organ_id')
+        App.find_by(organ_id: _organ_id, primary: true)
+      else
+        App.find_by(primary: true)
+      end
+    end
+
+  end
+end

data/app/models/wechat/extraction.rb

@@ -0,0 +1,5 @@
+module Wechat
+  class Extraction < ApplicationRecord
+    include Model::Extraction
+  end
+end

data/app/models/wechat/extractor.rb

@@ -0,0 +1,5 @@
+module Wechat
+  class Extractor < ApplicationRecord
+    include Model::Extractor
+  end
+end

data/app/models/wechat/hook.rb

@@ -0,0 +1,5 @@
+module Wechat
+  class Hook < ApplicationRecord
+    include Model::Hook
+  end
+end

data/app/models/wechat/media.rb

@@ -0,0 +1,5 @@
+module Wechat
+  class Media < ApplicationRecord
+    include Model::Media
+  end
+end

data/app/models/wechat/menu.rb

@@ -0,0 +1,5 @@
+module Wechat
+  class Menu < ApplicationRecord
+    include Model::Menu
+  end
+end

data/app/models/wechat/model/account.rb

@@ -0,0 +1,23 @@
+module Wechat
+  module Model::Account
+    extend ActiveSupport::Concern
+
+    included do
+      has_one :response, as: :effective
+      has_one :wechat_user, class_name: 'Wechat::WechatUser'
+    end
+
+    def invoke_effect(wechat_user)
+      old_account = wechat_user.account
+      if old_account == self
+        "您的微信账号已经绑定到账号 #{identity} 了"
+      elsif old_account
+        old_account.access_tokens.delete_all  # 更改绑定关系后，old account 登陆失效
+        "您的微信账号已更换绑定账号, 之前绑定的账号是#{old_account.identity}, 已经绑定到#{identity}"
+      else
+        "您的微信账号绑定至账号 #{identity} 成功"
+      end
+    end
+
+  end
+end

data/app/models/wechat/model/agency.rb

@@ -0,0 +1,70 @@
+module Wechat
+  module Model::Agency
+    SERVICE_TYPE = {
+      '0' => 'WechatRead',
+      '1' => 'WechatRead',
+      '2' => 'WechatPublic'
+    }.freeze
+    extend ActiveSupport::Concern
+
+    included do
+      attribute :appid, :string
+      attribute :access_token, :string
+      attribute :access_token_expires_at, :datetime
+      attribute :refresh_token, :string
+      attribute :func_infos, :string, array: true
+      attribute :nick_name, :string
+      attribute :head_img, :string
+      attribute :user_name, :string
+      attribute :principal_name, :string
+      attribute :alias_name, :string
+      attribute :qrcode_url, :string
+      attribute :business_info, :json
+      attribute :service_type, :string
+      attribute :verify_type, :string
+
+      belongs_to :platform
+      belongs_to :app, foreign_key: :appid, primary_key: :appid, optional: true
+
+      after_create_commit :store_info_later
+      before_save :init_app, if: -> { appid_changed? && appid }
+    end
+
+    def api
+      return @api if defined? @api
+      @api = Wechat::Api::Public.new(self)
+    end
+
+    def init_app
+      app || build_app
+      app.save
+    end
+
+    def refresh_access_token
+      r = platform.api.authorizer_token(appid, refresh_token)
+      store_access_token(r)
+    end
+
+    def store_info_later
+      AgencyJob.perform_later(self)
+    end
+
+    def store_info
+      r = platform.api.get_authorizer_info(appid)
+      self.assign_attributes r.slice('nick_name', 'head_img', 'user_name', 'principal_name', 'qrcode_url', 'business_info')
+      self.alias_name = r['alias']
+      self.service_type = r.dig('service_type_info', 'id')
+      self.verify_type = r.dig('verify_type_info', 'id')
+      self.save
+    end
+
+    def store_access_token(r)
+      self.access_token = r['authorizer_access_token']
+      self.access_token_expires_at = Time.current + r['expires_in'].to_i
+      self.refresh_token = r['authorizer_refresh_token']
+      self.func_infos = r['func_info'].map { |i| i.dig('funcscope_category', 'id') } if r['func_info'].is_a?(Array)
+      self.save
+    end
+
+  end
+end

data/app/models/wechat/model/annunciate.rb

@@ -0,0 +1,32 @@
+module Wechat
+  module Model::Annunciate
+    extend ActiveSupport::Concern
+
+    included do
+      after_create_commit :to_wechat_later
+    end
+
+    def to_wechat_later
+      AnnunciateJob.perform_later(self)
+    end
+
+    def to_wechat
+      app = PublicApp.valid.find_by(organ_id: annunciation.organ_id, primary: true)
+
+      return unless app
+
+      tag_ids = app.tags.where(user_tag_id: user_tag_id).pluck(:tag_id)
+      if tag_ids.present?
+        api = app.api
+        tag_ids.each do |tag_id|
+          api.message_mass_sendall(body, tag_id)
+        end
+      end
+    end
+
+    def body
+      annunciation.body
+    end
+
+  end
+end

data/app/models/wechat/model/app/mobile_app.rb

@@ -0,0 +1,11 @@
+module Wechat
+  module Model::App::MobileApp
+    extend ActiveSupport::Concern
+
+    included do
+
+    end
+
+
+  end
+end

data/app/models/wechat/model/app/program_app.rb

@@ -0,0 +1,18 @@
+module Wechat
+  module Model::App::ProgramApp
+    extend ActiveSupport::Concern
+
+    included do
+    end
+
+    def api
+      return @api if defined? @api
+      @api = Wechat::Api::Program.new(self)
+    end
+
+    def template_messenger(template)
+      Wechat::Message::Template::Program.new(self, template)
+    end
+
+  end
+end

data/app/models/wechat/model/app/public_app.rb

@@ -0,0 +1,46 @@
+module Wechat
+  module Model::App::PublicApp
+
+    def sync_templates
+      templates = api.templates
+      templates.each do |template|
+        template = templates.find_or_initialize_by(template_id: template['template_id'])
+        template.assign_attributes template.slice('title', 'content', 'example')
+        template.save
+      end
+    end
+
+    def oauth2_url(scope = 'snsapi_userinfo', state: SecureRandom.hex(16), **host_options)
+      h = {
+        appid: appid,
+        redirect_uri: Rails.application.routes.url_for(controller: 'wechat/apps', action: 'login', id: id, **host_options),
+        response_type: 'code',
+        scope: scope,
+        state: state
+      }
+      logger.debug "\e[35m  Detail: #{h}  \e[0m"
+      "https://open.weixin.qq.com/connect/oauth2/authorize?#{h.to_query}#wechat_redirect"
+    end
+
+    def oauth2_data_url(scope = 'snsapi_userinfo', **host_options)
+      r = oauth2_url(scope, **host_options)
+      QrcodeHelper.data_url(r)
+    end
+
+    def generate_wechat_user(code)
+      h = {
+        appid: appid,
+        secret: secret,
+        code: code,
+        grant_type: 'authorization_code'
+      }
+      r = HTTPX.get "https://api.weixin.qq.com/sns/oauth2/access_token?#{h.to_query}"
+      result = JSON.parse(r.body.to_s)
+      wechat_user = wechat_users.find_or_initialize_by(uid: result['openid'])
+      wechat_user.assign_attributes result.slice('access_token', 'refresh_token', 'unionid')
+      wechat_user.expires_at = Time.current + result['expires_in'].to_i
+      wechat_user
+    end
+
+  end
+end

data/app/models/wechat/model/app/read_app.rb

@@ -0,0 +1,24 @@
+module Wechat
+  module Model::App::ReadApp
+    extend ActiveSupport::Concern
+
+    included do
+
+    end
+
+
+    def oauth2_url(scope = 'snsapi_userinfo', **host_options)
+      options = Rails.application.routes.default_url_options
+      h = {
+        appid: appid,
+        redirect_uri: url_helpers.app_url(id, **options),
+        response_type: 'code',
+        scope: scope,
+        state: SecureRandom.hex(16)
+      }
+      "https://open.weixin.qq.com/connect/oauth2/authorize?#{h.to_query}#wechat_redirect"
+    end
+
+
+  end
+end

data/app/models/wechat/model/app/work_app.rb

@@ -0,0 +1,33 @@
+module Wechat
+  module Model::App::WorkApp
+    extend ActiveSupport::Concern
+
+    included do
+      validates :agentid, presence: true
+
+      alias_attribute :corpid, :appid
+      alias_attribute :corpsecret, :secret
+    end
+
+    def api
+      return @api if defined? @api
+      @api = Wechat::Api::Work.new(self)
+    end
+
+    def wechat_corp_oauth2(oauth2_params, account = nil)
+      userid   = cookies.signed_or_encrypted[:we_userid]
+      deviceid = cookies.signed_or_encrypted[:we_deviceid]
+      if userid.present? && deviceid.present?
+        yield userid, { 'UserId' => userid, 'DeviceId' => deviceid }
+      elsif params[:code].present? && params[:state] == oauth2_params[:state]
+        userinfo = wechat(account).getuserinfo(params[:code])
+        cookies.signed_or_encrypted[:we_userid] = { value: userinfo['UserId'], expires: self.class.oauth2_cookie_duration.from_now }
+        cookies.signed_or_encrypted[:we_deviceid] = { value: userinfo['DeviceId'], expires: self.class.oauth2_cookie_duration.from_now }
+        yield userinfo['UserId'], userinfo
+      else
+        redirect_to generate_oauth2_url(oauth2_params)
+      end
+    end
+
+  end
+end

data/app/models/wechat/model/app.rb

@@ -0,0 +1,198 @@
+module Wechat
+  module Model::App
+    extend ActiveSupport::Concern
+
+    included do
+      attribute :type, :string, default: 'Wechat::PublicApp'
+      attribute :name, :string
+      attribute :enabled, :boolean, default: true
+      attribute :shared, :boolean, default: false
+      attribute :appid, :string
+      attribute :secret, :string
+      attribute :token, :string, default: -> { SecureRandom.hex }
+      attribute :agentid, :string, comment: '企业微信所用'
+      attribute :mch_id, :string, comment: '支付专用、商户号'
+      attribute :key, :string, comment: '支付专用'
+      attribute :key_v3, :string, comment: '支付通知解密'
+      attribute :encrypt_mode, :boolean, default: true
+      attribute :encoding_aes_key, :string
+      attribute :access_token, :string
+      attribute :access_token_expires_at, :datetime
+      attribute :jsapi_ticket, :string
+      attribute :oauth2_state, :string
+      attribute :jsapi_ticket_expires_at, :datetime
+      attribute :user_name, :string
+      attribute :oauth_enable, :boolean, default: true
+      attribute :apiclient_cert, :string
+      attribute :apiclient_key, :string
+      attribute :serial_no, :string
+
+      belongs_to :organ, class_name: 'Org::Organ', optional: true
+
+      has_many :post_syncs, as: :synced, dependent: :delete_all
+      has_many :posts, through: :post_syncs
+
+      has_one :organ_domain, -> { where(default: true) }, class_name: 'Org::OrganDomain', foreign_key: :appid, primary_key: :appid
+      has_many :organ_domains, class_name: 'Org::OrganDomain', foreign_key: :appid, primary_key: :appid
+
+      has_one :agency, foreign_key: :appid, primary_key: :appid
+      has_many :agencies, foreign_key: :appid, primary_key: :appid
+      has_many :scenes, foreign_key: :appid, primary_key: :appid
+      has_many :tags, foreign_key: :appid, primary_key: :appid
+      has_many :templates, foreign_key: :appid, primary_key: :appid
+
+      scope :valid, -> { where(enabled: true) }
+      scope :shared, -> { where(shared: true) }
+
+      validates :appid, presence: true, uniqueness: true
+
+      before_validation do
+        self.encoding_aes_key ||= SecureRandom.alphanumeric(43) if encrypt_mode
+      end
+    end
+
+    def url
+      Rails.application.routes.url_for(controller: 'wechat/wechats', action: 'show', id: self.id, host: host)
+    end
+
+    def sync_menu
+      api.menu_delete
+      api.menu_create menu
+    end
+
+    def menu
+      {
+        button: default_menus + within_menus
+      }
+    end
+
+    def default_menus
+      if organ && organ.respond_to?(:limit_wechat_menu)
+        limit = 3 - organ.limit_wechat_menu
+      else
+        limit = 3
+      end
+      Menu.where(parent_id: nil, appid: nil).limit(limit).as_json
+    end
+
+    def within_menus
+      if organ && organ.respond_to?(:limit_wechat_menu)
+        self.menus.limit(organ.limit_wechat_menu).where(parent_id: nil).order(position: :asc).as_json
+      else
+        self.menus.where(parent_id: nil).order(position: :asc).as_json
+      end
+    end
+
+    def refresh_access_token
+      r = api.token
+      if r['access_token']
+        store_access_token(r)
+      else
+        logger.debug "\e[35m  #{r['errmsg']}  \e[0m"
+      end
+    end
+
+    def store_access_token(token_hash)
+      self.access_token = token_hash['access_token']
+      self.access_token_expires_at = Time.current + token_hash['expires_in'].to_i
+      self.save
+    end
+
+    def jsapi_ticket
+      if jsapi_ticket_valid?
+        super
+      else
+        refresh_jsapi_ticket
+      end
+    end
+
+    def jsapi_ticket_valid?
+      return false unless jsapi_ticket_expires_at.acts_like?(:time)
+      jsapi_ticket_expires_at > Time.current
+    end
+
+    def refresh_jsapi_ticket
+      r = api.jsapi_ticket
+      store_jsapi_ticket(r)
+      jsapi_ticket
+    end
+
+    def store_jsapi_ticket(ticket_hash)
+      self.jsapi_ticket = ticket_hash['ticket']
+      self.jsapi_ticket_expires_at = Time.current + ticket_hash['expires_in'].to_i
+      self.save
+    end
+
+    def api
+      return @api if defined? @api
+      if secret.present?
+        @api = Wechat::Api::Public.new(self)
+      elsif agency
+        @api = Wechat::Api::Public.new(agency)
+      else
+        raise 'Must has secret or under agency'
+      end
+    end
+
+    def oauth2_qrcode_url(**host_options)
+      q = {
+        appid: appid,
+        redirect_uri: Rails.application.routes.url_for(controller: 'wechat/apps', action: 'show', id: id, **host_options),
+        scope: 'snsapi_login',
+        response_type: 'code',
+        state: SecureRandom.hex(16)
+      }
+
+      "https://open.weixin.qq.com/connect/qrconnect?#{q.to_query}#wechat_redirect"
+    end
+
+    def sync_tags
+      results = api.tags
+      results.fetch('tags', []).each do |result|
+        tag = tags.find_or_initialize_by(name: result['name'])
+        tag.count = result['count']
+        tag.tag_id = result['id']
+        tag.save
+      end
+      tags.where(tag_id: nil).each do |tag|
+        tag.sync_to_wechat_later
+      end
+    end
+
+    # 小程序
+    def sync_templates
+      templates = api.templates
+      templates.each do |template|
+        template = templates.find_or_initialize_by(template_id: template['priTmplId'])
+        template.template_type = template['type']
+        template.assign_attributes template.slice('title', 'content', 'example')
+        template.save
+      end
+    end
+
+    # 公众号
+    def sync_template_configs
+      templates = api.templates
+      templates.each do |template|
+        template_config = TemplatePublic.new(title: template['title'])
+        data_keys = Template.new(content: template['content']).data_keys
+        data_keys.each do |key|
+          template_config.template_key_words.build(name: key)
+        end
+        template_config.save
+      end
+    end
+
+    def template_ids(notifiable_type, *code)
+      ids = TemplateConfig.where(notifiable_type: notifiable_type, code: code).pluck(:id)
+      templates.where(template_config_id: ids).pluck(:template_id)
+    end
+
+    def host
+      if oauth_enable
+        organ_domain&.identifier || organ_domains.first&.identifier
+      end
+    end
+
+  end
+end