wework 0.3.4 → 1.1.0

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

@@ -0,0 +1,16 @@
+require "erb"
+
+module Wework
+  module Api
+    module Methods
+      module Approval
+        def get_approval_data start_time, end_time, next_spnum=nil
+          # https://work.weixin.qq.com/api/doc#11228
+          payload = {starttime: start_time, endtime: end_time}
+          payload[:next_spnum] = next_spnum unless next_spnum.nil?
+          post 'corp/getapprovaldata', payload
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,34 @@
+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
+
+        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/department.rb

@@ -0,0 +1,27 @@
+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
+
+        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,52 @@
+require "erb"
+
+module Wework
+  module Api
+    module Methods
+      module Message
+        def miniprogram_message_send user_id, miniprogram_notice={}
+          notice = miniprogram_notice.merge(appid: agent_id)
+          post 'message/send', { touser: user_id, msgtype: 'miniprogram_notice', miniprogram_notice: notice }
+        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,16 @@
+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}
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,53 @@
+module Wework
+  module Api
+    module Methods
+      module Suite
+        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/user.rb

@@ -0,0 +1,35 @@
+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_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
+      end
+    end
+  end
+end

data/lib/wework/api/provider.rb

@@ -0,0 +1,13 @@
+module Wework
+  module Api
+    class Provider < Base
+      include Methods::Provider
+
+      private
+
+      def token_store
+        @token_store ||= Token::ProviderToken.new self
+      end
+    end
+  end
+end

data/lib/wework/api/suite.rb

@@ -0,0 +1,72 @@
+require 'builder'
+
+module Wework
+  module Api
+    class Suite < Base
+
+      include Wework::Cipher
+      include Methods::Suite
+
+      attr_reader :encoding_aes_key, :suite_id, :suite_secret, :suite_token
+
+      def initialize(options={})
+        @suite_id = options.delete(:suite_id)
+        @suite_secret = options.delete(:suite_secret)
+        @suite_token = options.delete(:suite_token)
+        @encoding_aes_key = options.delete(:encoding_aes_key)
+        super(options)
+      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 = [suite_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
+
+      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

@@ -1,6 +1,9 @@
 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

data/lib/wework/mock_api.rb

@@ -0,0 +1,34 @@
+require 'json'
+require 'sinatra/base'
+
+module Wework
+  class MockApi < Sinatra::Base
+    get '/cgi-bin/media/get' do
+      send_file mock_file_path('files/zhiren.png'), :type => :jpg
+    end
+
+    get '/cgi-bin/*' do |api|
+      json_response "#{api}.json"
+    end
+
+    %i(get post).each do |http_method|
+      send http_method, '/cgi-bin/*/*' do |category, api|
+        json_response "#{category}/#{api}.json"
+      end
+    end
+
+    private
+
+    def json_response file_name
+      content_type :json
+      status 200
+      file_path = mock_file_path(file_name)
+      file_path = mock_file_path('success.json') unless File.exist?(file_path)
+      File.open(file_path, 'rb').read
+    end
+
+    def mock_file_path path
+      File.join(File.expand_path('../../../mock_responses/', __FILE__), path)
+    end
+  end
+end