wework-next 1.2.6

data/lib/wework/api/methods/batch.rb

@@ -0,0 +1,38 @@
+module Wework
+  module Api
+    module Methods
+      module Batch
+        def batch_syncuser media_id, callback_url=nil, token=nil, encodingaeskey=nil
+          post 'batch/syncuser', batch_params(media_id, callback_url, token, encodingaeskey)
+        end
+
+        def batch_replaceuser media_id, callback_url=nil, token=nil, encodingaeskey=nil
+          post 'batch/replaceuser', batch_params(media_id, callback_url, token, encodingaeskey)
+        end
+
+        def batch_replaceparty media_id, callback_url=nil, token=nil, encodingaeskey=nil
+          post 'batch/replaceparty', batch_params(media_id, callback_url, token, encodingaeskey)
+        end
+
+        def batch_getresult job_id
+          get 'batch/getresult', params: {jobid: job_id}
+        end
+
+        def batch_invite user=[], party=[], tag=[]
+          post 'batch/invite', {user: user, party: party, tag: tag}
+        end
+
+        private
+
+        def batch_params media_id, callback_url, token, encodingaeskey
+          params = {media_id: media_id}
+          if callback_url.present? && token.present? && encodingaeskey.present?
+            params[:callback] = {url: callback_url, token: token, encodingaeskey: encodingaeskey}
+          end
+
+          params
+        end
+      end
+    end
+  end
+end

data/lib/wework/api/methods/checkin.rb

@@ -0,0 +1,19 @@
+require "erb"
+
+module Wework
+  module Api
+    module Methods
+      module Checkin
+        def get_checkin_data start_time, end_time, userid_list=[], checkin_type=3
+          # https://work.weixin.qq.com/api/doc#11196
+          post 'checkin/getcheckindata', {
+             opencheckindatatype: checkin_type,
+             starttime: start_time,
+             endtime: end_time,
+             useridlist: userid_list,
+          }
+        end
+      end
+    end
+  end
+end

data/lib/wework/api/methods/crm.rb

@@ -0,0 +1,11 @@
+module Wework
+  module Api
+    module Methods
+      module Crm
+        def crm_get_external_contact external_userid
+          get 'crm/get_external_contact', params: {external_userid: external_userid}
+        end
+      end
+    end
+  end
+end

data/lib/wework/api/methods/department.rb

@@ -0,0 +1,36 @@
+module Wework
+  module Api
+    module Methods
+      module Department
+        def department_create(data = {})
+          post 'department/create', data
+        end
+
+        def department_update(department_id, data = {})
+          post 'department/update', data.merge(id: department_id)
+        end
+
+        def department_delete(department_id)
+          get 'department/delete', params: { id: department_id }
+        end
+
+        # @param [nil] department_id
+        def simple_list(department_id = nil)
+          if department_id.nil?
+            get 'department/simplelist', params: {}
+          else
+            get 'department/simplelist', params: { id: department_id }
+          end
+        end
+
+        def department_list(department_id = nil)
+          if department_id.nil?
+            get 'department/list'
+          else
+            get 'department/list', params: { id: department_id }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/wework/api/methods/media.rb

@@ -0,0 +1,20 @@
+module Wework
+  module Api
+    module Methods
+      module Media
+        # public API
+        def media_upload(type, file)
+          post_file 'media/upload', file, params: { type: type }
+        end
+
+        def media_get(media_id)
+          get 'media/get', params: { media_id: media_id }, as: :file
+        end
+
+        def get_media_url(media_id)
+          "#{API_ENDPOINT}media/get?access_token=#{access_token}&media_id=#{media_id}"
+        end
+      end
+    end
+  end
+end

data/lib/wework/api/methods/menu.rb

@@ -0,0 +1,21 @@
+require "erb"
+
+module Wework
+  module Api
+    module Methods
+      module Menu
+        def menu_create menu
+          post 'menu/create', menu, params: {agentid: agent_id}
+        end
+
+        def menu_get
+          get 'menu/get', params: {agentid: agent_id}
+        end
+
+        def menu_delete
+          get 'menu/delete', params: {agentid: agent_id}
+        end
+      end
+    end
+  end
+end

data/lib/wework/api/methods/message.rb

@@ -0,0 +1,51 @@
+require "erb"
+
+module Wework
+  module Api
+    module Methods
+      module Message
+        def miniprogram_message_send user_id, miniprogram_notice={}, options={}
+          post 'message/send', { touser: user_id, msgtype: 'miniprogram_notice', miniprogram_notice: miniprogram_notice, **options }
+        end
+
+        def text_message_send user_ids, department_ids, content
+          message_send user_ids, department_ids, {text: {content: content}, msgtype: 'text'}
+        end
+
+        def image_message_send user_ids, department_ids, media_id
+          message_send user_ids, department_ids, {image: {media_id: media_id}, msgtype: 'image'}
+        end
+
+        def voice_message_send user_ids, department_ids, media_id
+          message_send user_ids, department_ids, {voice: {media_id: media_id}, msgtype: 'voice'}
+        end
+
+        def file_message_send user_ids, department_ids, media_id
+          message_send user_ids, department_ids, {file: {media_id: media_id}, msgtype: 'file'}
+        end
+
+        def video_message_send user_ids, department_ids, video={}
+          message_send user_ids, department_ids, {video: video, msgtype: 'video'}
+        end
+
+        def textcard_message_send user_ids, department_ids, textcard={}
+          message_send user_ids, department_ids, {textcard: textcard, msgtype: 'textcard'}
+        end
+
+        def news_message_send user_ids, department_ids, news=[]
+          message_send user_ids, department_ids, {news: {articles: news}, msgtype: 'news'}
+        end
+
+        private
+
+        def message_send user_ids, department_ids, payload={}
+          payload[:agentid] = agent_id
+          payload[:touser] = Array.wrap(user_ids).join('|') if user_ids.present?
+          payload[:toparty] = Array.wrap(department_ids).join('|') if department_ids.present?
+
+          post 'message/send', payload
+        end
+      end
+    end
+  end
+end

data/lib/wework/api/methods/provider.rb

@@ -0,0 +1,38 @@
+module Wework
+  module Api
+    module Methods
+      module Provider
+        def sso_authorize_url(redirect_uri, user_type='admin', state='qywxlogin')
+          uri = ERB::Util.url_encode(redirect_uri)
+          "#{SSO_AUTHORIZE_ENDPOINT}?appid=#{corp_id}&redirect_uri=#{uri}&state=#{state}&usertype=#{user_type}"
+        end
+
+        def get_login_info auth_code
+          post 'service/get_login_info', {auth_code: auth_code, access_token: access_token}
+        end
+
+        def get_register_code template_id, options={}
+          params = {template_id: template_id}
+          post 'service/get_register_code', params.merge(options)
+        end
+
+        def get_register_url template_id, options={}
+          register_code = get_register_code(template_id, options).register_code
+          "#{REGISTER_ENDPOINT}?register_code=#{register_code}"
+        end
+
+        def service_media_upload(type, file)
+          post_file 'service/media/upload', file, params: { type: type }
+        end
+
+        def id_translate auth_corpid, media_id
+          post 'service/contact/id_translate', { auth_corpid: auth_corpid, media_id_list: [ media_id ] }
+        end
+
+        def service_batch_getresult job_id
+          get 'service/batch/getresult', params: { jobid: job_id }
+        end
+      end
+    end
+  end
+end

data/lib/wework/api/methods/service.rb

@@ -0,0 +1,53 @@
+module Wework
+  module Api
+    module Methods
+      module Service
+        def corp_authorize_url(redirect_uri, state="corp_authorize")
+          "#{APP_AUTHORIZE_ENDPOINT}?suite_id=#{suite_id}&pre_auth_code=#{get_pre_auth_code}&redirect_uri=#{redirect_uri}&state=#{state}"
+        end
+
+        def get_pre_auth_code
+          result = get 'service/get_pre_auth_code'
+          return result.pre_auth_code if result.success?
+        end
+
+        def set_session_info pre_auth_code, session_info={}
+          post 'service/set_session_info', {pre_auth_code: pre_auth_code, session_info: session_info}
+        end
+
+        def get_permanent_code auth_code
+          post 'service/get_permanent_code', {auth_code: auth_code}
+        end
+
+        def get_auth_info auth_corpid, permanent_code
+          post 'service/get_auth_info', {auth_corpid: auth_corpid, permanent_code: permanent_code}
+        end
+
+        def get_corp_token auth_corpid, permanent_code
+          post 'service/get_corp_token', {suite_id: suite_id, auth_corpid: auth_corpid, permanent_code: permanent_code}
+        end
+
+        def get_admin_list auth_corpid, agentid
+          post 'service/get_admin_list', {auth_corpid: auth_corpid, agentid: agentid}
+        end
+
+        def authorize_url(redirect_uri, scope="snsapi_base", state="wxwork")
+          uri = ERB::Util.url_encode(redirect_uri)
+          "#{AUTHORIZE_ENDPOINT}?appid=#{suite_id}&redirect_uri=#{uri}&response_type=code&scope=#{scope}&state=#{state}#wechat_redirect"
+        end
+
+        def get_oauth_userinfo(code)
+          get 'service/getuserinfo3rd', params: {access_token: access_token, code: code}
+        end
+
+        def get_user_detail(user_ticket)
+          post "service/getuserdetail3rd?access_token=#{access_token}", {user_ticket: user_ticket}
+        end
+
+        def get_session_with_jscode(js_code, grant_type='authorization_code')
+          post 'service/miniprogram/jscode2session', {}, params: {js_code: js_code, grant_type: grant_type}
+        end
+      end
+    end
+  end
+end

data/lib/wework/api/methods/tag.rb

@@ -0,0 +1,37 @@
+module Wework
+  module Api
+    module Methods
+      module Tag
+        def tag_create(tagname, tagid=nil)
+          data = {tagname: tagname}
+          data[:tagid] = tagid unless tagid.nil?
+          post 'tag/create', data
+        end
+
+        def tag_update(tagid, tagname)
+          post 'tag/update', {tagid: tagid, tagname: tagname}
+        end
+
+        def tag_delete tagid
+          get 'user/delete', params: {tagid: tagid}
+        end
+
+        def tag_get tagid
+          get 'user/get', params: {tagid: tagid}
+        end
+
+        def tag_addtagusers tagid, userlist=[], partylist=[]
+          post 'tag/addtagusers', {tagid: tagid, userlist: userlist, partylist: partylist}
+        end
+
+        def tag_deltagusers tagid, userlist=[], partylist=[]
+          post 'tag/deltagusers', {tagid: tagid, userlist: userlist, partylist: partylist}
+        end
+
+        def tag_list
+          get 'user/list'
+        end
+      end
+    end
+  end
+end

data/lib/wework/api/methods/user.rb

@@ -0,0 +1,55 @@
+module Wework
+  module Api
+    module Methods
+      module User
+        def user_create data={}
+          post 'user/create', data
+        end
+
+        def user_get userid
+          get 'user/get', params: {userid: userid}
+        end
+
+        def user_getuserid mobile
+          post 'user/getuserid', {mobile: mobile}
+        end
+
+        def user_update userid, data={}
+          post 'user/update', data.merge(userid: userid)
+        end
+
+        def user_delete userid
+          get 'user/delete', params: {userid: userid}
+        end
+
+        def user_batchdelete useridlist=[]
+          post 'user/batchdelete', {useridlist: useridlist}
+        end
+
+        def user_simplelist department_id, fetch_child=0
+          get 'user/simplelist', params: {department_id: department_id, fetch_child: fetch_child}
+        end
+
+        def user_list department_id, fetch_child=0
+          get 'user/list', params: {department_id: department_id, fetch_child: fetch_child}
+        end
+
+        def user_list_id
+          get 'user/list_id',params:{limit:10000}
+        end
+        def convert_to_openid userid
+          post 'user/convert_to_openid', {userid: userid}
+        end
+
+        def convert_to_userid openid
+          post 'user/convert_to_userid', {openid: openid}
+        end
+
+
+        def authsucc userid
+          get 'user/authsucc', params: {userid: userid}
+        end
+      end
+    end
+  end
+end

data/lib/wework/api/provider.rb

@@ -0,0 +1,28 @@
+module Wework
+  module Api
+    class Provider < Base
+      include Methods::Provider
+      include Wework::Cipher
+
+      attr_reader :encoding_aes_key, :token
+
+      def initialize(options={})
+        @token = options.delete(:token)
+        @encoding_aes_key = options.delete(:encoding_aes_key)
+        super(options)
+      end
+
+      private
+
+      def token_store
+        @token_store ||= Token::ProviderToken.new self
+      end
+
+      private
+
+      def token_params
+        {provider_access_token: access_token}
+      end
+    end
+  end
+end

data/lib/wework/api/suite.rb

@@ -0,0 +1,48 @@
+require 'builder'
+
+module Wework
+  module Api
+    class Suite < Base
+
+      include Wework::Cipher
+      include Methods::Service
+
+      attr_reader :encoding_aes_key, :suite_id, :suite_secret, :suite_token, :token
+
+      def initialize(options={})
+        @suite_id = options.delete(:suite_id)
+        @suite_secret = options.delete(:suite_secret)
+        @token = @suite_token = options.delete(:suite_token)
+        @encoding_aes_key = options.delete(:encoding_aes_key)
+        super(options)
+      end
+
+      def suite_ticket= ticket
+        Wework.redis.set ticket_key, ticket
+      end
+
+      def suite_ticket
+        Wework.redis.get ticket_key
+      end
+
+      def corp(corp_id, permanent_code)
+        Wework::Api::Corp.new(suite: self, corp_id: corp_id, permanent_code: permanent_code)
+      end
+
+      private
+
+      def token_params
+        {suite_access_token: access_token}
+      end
+
+      def ticket_key
+        "SUITE_TICKET_#{suite_id}"
+      end
+
+      def token_store
+        @token_store ||= Token::SuiteToken.new self
+      end
+
+    end
+  end
+end

data/lib/wework/cipher.rb

@@ -0,0 +1,98 @@
+require 'openssl/cipher'
+require 'base64'
+
+# the file copy from here
+# https://github.com/Eric-Guo/wechat/blob/master/lib/wechat/cipher.rb
+
+module Wework
+  module Cipher
+    extend ActiveSupport::Concern
+
+    BLOCK_SIZE = 32
+    CIPHER = 'AES-256-CBC'.freeze
+
+    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
+
+    # app_id or corp_id
+    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
+
+    def msg_decrypt message
+      unpack(decrypt(Base64.decode64(message), encoding_aes_key))[0]
+    end
+
+    def msg_encrypt message
+      Base64.strict_encode64(encrypt(pack(message, corp_id), encoding_aes_key))
+    end
+
+    def signature(timestamp, nonce, encrypt)
+      array = [token, timestamp, nonce]
+      array << encrypt unless encrypt.nil?
+      Digest::SHA1.hexdigest array.compact.collect(&:to_s).sort.join
+    end
+
+    def generate_xml(msg, timestamp, nonce)
+      encrypt = msg_encrypt(msg)
+      {
+        Encrypt: encrypt,
+        MsgSignature: signature(timestamp, nonce, encrypt),
+        TimeStamp: timestamp,
+        Nonce: nonce
+      }.to_xml(root: 'xml', children: 'item', skip_instruct: true, skip_types: true)
+    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
+end

data/lib/wework/config.rb

@@ -0,0 +1,28 @@
+module Wework
+
+  class << self
+    def configure
+      yield config
+    end
+
+    def config
+      @config ||= Config.new
+    end
+
+    def redis
+      config.redis
+    end
+
+    def http_timeout_options
+      config.http_timeout_options || {write: 2, connect: 5, read: 10}
+    end
+
+    def expired_shift_seconds
+      config.expired_shift_seconds || 100
+    end
+  end
+
+  class Config
+    attr_accessor :redis, :http_timeout_options, :expired_shift_seconds
+  end
+end