wework 0.3.4 → 1.1.0

data/lib/wework/request.rb

@@ -1,5 +1,6 @@
 # Reference this: https://github.com/Eric-Guo/wechat/blob/master/lib/wechat/http_client.rb
 require 'http'
+require 'wework/global_code'
 
 module Wework
   class Request
@@ -51,6 +52,7 @@ module Wework
       parse_response(response, as || :json) do |parse_as, data|
         break data unless parse_as == :json
         result = Wework::Result.new(data)
+        Rails.logger.info "[WEWORK] request path(#{url_base}#{path}): #{result.inspect}" if defined?(Rails)
         raise AccessTokenExpiredError if result.token_expired?
         result
       end
@@ -96,7 +98,7 @@ module Wework
     end
 
     def throw_error
-      raise ResultErrorException.new(message || errmsg) unless success?
+      raise ResultErrorException.new(full_message) unless success?
     end
 
     def success?

data/lib/wework/token/app_token.rb

@@ -0,0 +1,21 @@
+require 'wework/token/base'
+
+module Wework
+  module Token
+    class AppToken < Base
+
+      def redis_key
+        @redis_key ||= Digest::MD5.hexdigest "WX_APP_TOKEN_#{client.corp_id}_#{client.secret}"
+      end
+
+      def token_key
+        'access_token'
+      end
+
+      def refresh_token
+        client.request.get 'gettoken', params: {corpid: client.corp_id, corpsecret: client.secret}
+      end
+
+    end
+  end
+end

data/lib/wework/token/base.rb

@@ -0,0 +1,59 @@
+module Wework
+  module Token
+    class Base
+      attr_accessor :client
+
+      def initialize(client)
+        @client = client
+        raise RedisNotConfigException if redis.nil?
+      end
+
+      def token
+        update_token if expired?
+        redis.hget(redis_key, token_key)
+      end
+
+      def update_token
+        result = refresh_token
+        value = result.send(token_key)
+        if value.nil?
+          puts "#{self.class.name} refresh token error: #{result.inspect}"
+        else
+          expires_at = Time.now.to_i + result.expires_in.to_i - Wework.expired_shift_seconds
+
+          redis.hmset(
+            redis_key,
+            token_key, value,
+            "expires_at", expires_at
+          )
+
+          redis.expireat(redis_key, expires_at)
+        end
+
+        value
+      end
+
+      private
+
+      def redis
+        Wework.redis
+      end
+
+      def redis_key
+        raise NotImplementedError
+      end
+
+      def token_key
+        raise NotImplementedError
+      end
+
+      def refresh_token
+        raise NotImplementedError
+      end
+
+      def expired?
+        redis.hvals(redis_key).empty? || redis.hget(redis_key, 'expires_at').to_i <= Time.now.to_i
+      end
+    end
+  end
+end

data/lib/wework/token/corp_token.rb

@@ -0,0 +1,20 @@
+require 'wework/token/base'
+
+module Wework
+  module Token
+    class CorpToken < Base
+
+      def redis_key
+        @redis_key ||= Digest::MD5.hexdigest "WX_CORP_TOKEN_#{client.corp_id}_#{client.permanent_code}"
+      end
+
+      def token_key
+        'access_token'
+      end
+
+      def refresh_token
+        client.suite.get_corp_token(client.corp_id, client.permanent_code)
+      end
+    end
+  end
+end

data/lib/wework/token/js_ticket.rb

@@ -0,0 +1,25 @@
+require 'wework/token/base'
+
+require 'wework/token/base'
+
+module Wework
+  module Token
+    class JsTicket < Base
+
+      alias_method :ticket, :token
+
+      def redis_key
+        @redis_key ||= Digest::MD5.hexdigest "WX_JS_TICKET_#{client.corp_id}_#{client.agent_id}"
+      end
+
+      def token_key
+        'ticket'
+      end
+
+      def refresh_token
+        client.get 'get_jsapi_ticket'
+      end
+
+    end
+  end
+end

data/lib/wework/token/provider_token.rb

@@ -0,0 +1,21 @@
+require 'wework/token/base'
+
+module Wework
+  module Token
+    class ProviderToken < Base
+
+      def redis_key
+        @redis_key ||= Digest::MD5.hexdigest "WX_PROVIDER_TOKEN_#{client.corp_id}_#{client.secret}"
+      end
+
+      def token_key
+        'provider_access_token'
+      end
+
+      def refresh_token
+        client.request.post 'service/get_provider_token', {corpid: client.corp_id, provider_secret: client.secret}
+      end
+
+    end
+  end
+end

data/lib/wework/token/suite_token.rb

@@ -0,0 +1,21 @@
+require 'wework/token/base'
+
+module Wework
+  module Token
+    class SuiteToken < Base
+
+      def redis_key
+        @redis_key ||= Digest::MD5.hexdigest "WX_SUITE_TOKEN_#{client.suite_id}_#{client.suite_secret}_#{client.suite_token}"
+      end
+
+      def token_key
+        'suite_access_token'
+      end
+
+      def refresh_token
+        client.request.post 'service/get_suite_token', {suite_id: client.suite_id, suite_secret: client.suite_secret, suite_ticket: client.suite_ticket}
+      end
+
+    end
+  end
+end

data/lib/wework/version.rb

@@ -1,3 +1,3 @@
 module Wework
-  VERSION = '0.3.4'.freeze
+  VERSION = '1.1.0'.freeze
 end

data/mock_responses/agent/get.json

@@ -0,0 +1,29 @@
+{
+  "errcode":0,
+  "errmsg": "ok",
+  "agentid": "1",
+  "name": "NAME",
+  "square_logo_url": "xxxxxxxx",
+  "description": "desc",
+  "allow_userinfos": {
+    "user":[
+      {
+                 "userid":"id2"
+             },
+             {
+                 "userid":"id3"
+             }
+      ]
+    },
+   "allow_partys":{
+       "partyid": [1]
+    },
+    "allow_tags":{
+       "tagid": [1,2,3]
+    },
+   "close": 0,
+   "redirect_domain": "www.qq.com",
+   "report_location_flag": 0,
+   "isreportenter": 0,
+   "home_url": "http://www.qq.com"
+}

data/mock_responses/agent/list.json

@@ -0,0 +1,16 @@
+{
+    "errcode":0,
+    "errmsg":"ok" ,
+    "agentlist":[
+        {
+            "agentid":1,
+            "name":"NAME",
+            "square_logo_url":"xxxxxxxx"
+        },
+        {
+            "agentid":2,
+            "name":"NAME",
+            "square_logo_url":"xxxxxxxx"
+        }
+    ]
+}

data/mock_responses/agent/set.json

@@ -0,0 +1,4 @@
+{
+   "errcode":0,
+   "errmsg":"ok"
+}

data/mock_responses/batch/getresult.json

@@ -0,0 +1,8 @@
+{
+    "errcode": 0,
+    "errmsg": "ok",
+    "status": 1,
+    "type": "replace_user",
+    "total": 3,
+    "percentage": 33,
+}

data/mock_responses/department/list.json

@@ -0,0 +1,24 @@
+{
+   "errcode": 0,
+   "errmsg": "ok",
+   "department": [
+      {
+           "id": 1,
+           "name": "杭州知人",
+           "parentid": 0,
+           "order": 10
+       },
+       {
+           "id": 2,
+           "name": "广州研发中心",
+           "parentid": 1,
+           "order": 11
+       },
+       {
+           "id": 3,
+           "name": "邮箱产品部",
+           "parentid": 2,
+           "order": 12
+       }
+   ]
+}

data/mock_responses/error.json

@@ -0,0 +1,4 @@
+{
+   "errcode": -1,
+   "errmsg": "Invalid input"
+}

data/mock_responses/files/party.csv

@@ -0,0 +1,12 @@
+部门名称,部门ID,父部门ID,排序
+A管理组,18,1,0
+A产品组,19,1,-1024
+A系统组,20,1,0
+A客服组,1715,1,0
+A成都销售组,1716,1,0
+ATower组,4693,1,1024
+A上海销售组,5883,1,4096
+A北京销售组,5884,1,5120
+A杭州销售组,5885,1,6144
+A广深销售组,7160,1,7168
+A运营组,11052,1,8192

data/mock_responses/files/sample.txt

@@ -0,0 +1 @@
+Wework is a ruby API wrapper for work wechat

data/mock_responses/files/user.csv

@@ -0,0 +1,2 @@
+姓名,帐号,手机号,邮箱,所在部门,职位
+李丽,lili1983,13438802101,22705030@qq.com,1,财务

data/mock_responses/get_jsapi_ticket.json

@@ -0,0 +1,6 @@
+{
+    "errcode":0,
+    "errmsg":"ok",
+    "ticket":"bxLdikRXVbTPdHSM05e5u5sUoXNKd8-41ZO3MhKoyN5OfkWITDGgnr2fwJ0m9E8NYzWKVZvdVtaUgWvsdshFKA",
+    "expires_in":7200
+}

data/mock_responses/gettoken.json

@@ -0,0 +1,6 @@
+{
+   "errcode":0,
+   "errmsg":"",
+   "access_token": "accesstoken000001",
+   "expires_in": 7200
+}

data/mock_responses/menu/get.json

@@ -0,0 +1,24 @@
+{
+   "button":[
+       {
+           "type":"click",
+           "name":"今日歌曲",
+           "key":"V1001_TODAY_MUSIC"
+       },
+       {
+           "name":"菜单",
+           "sub_button":[
+               {
+                   "type":"view",
+                   "name":"搜索",
+                   "url":"http://www.soso.com/"
+               },
+               {
+                   "type":"click",
+                   "name":"赞一下我们",
+                   "key":"V1001_GOOD"
+               }
+           ]
+      }
+   ]
+}

data/mock_responses/service/get_corp_token.json

@@ -0,0 +1,4 @@
+{
+    "access_token": "xxxxxx",
+    "expires_in": 7200
+}

data/mock_responses/service/get_login_info.json

@@ -0,0 +1,34 @@
+{
+  "usertype": 1,
+  "user_info": {
+    "userid": "xxxx",
+    "name": "xxxx",
+    "avatar": "xxxx",
+    "email": "xxxx"
+  },
+  "corp_info": {
+    "corpid": "wx6c698d13f7a409a4"
+  },
+  "agent": [
+    {
+      "agentid": 0,
+      "auth_type": 1
+    },
+    {
+      "agentid": 1,
+      "auth_type": 1
+    },
+    {
+      "agentid": 2,
+      "auth_type": 1
+    }
+  ],
+  "auth_info": {
+    "department": [
+      {
+        "id": "2",
+        "writable": "true"
+      }
+    ]
+  }
+}

data/mock_responses/service/get_permanent_code.json

@@ -0,0 +1,54 @@
+{
+    "access_token": "xxxxxx",
+    "expires_in": 7200,
+    "permanent_code": "xxxx",
+    "auth_corp_info":
+    {
+        "corpid": "xxxx",
+        "corp_name": "name",
+        "corp_type": "verified",
+        "corp_square_logo_url": "yyyyy",
+        "corp_user_max": 50,
+        "corp_agent_max": 30,
+        "corp_full_name":"full_name",
+        "verified_end_time":1431775834,
+        "subject_type": 1,
+        "corp_wxqrcode": "zzzzz"
+    },
+    "auth_info":
+    {
+        "agent" :
+        [
+            {
+                "agentid":1,
+                "name":"NAME",
+                "round_logo_url":"xxxxxx",
+                "square_logo_url":"yyyyyy",
+                "appid":1,
+                "privilege":
+                {
+                    "level":1,
+                    "allow_party":[1,2,3],
+                    "allow_user":["zhansan","lisi"],
+                    "allow_tag":[1,2,3],
+                    "extra_party":[4,5,6],
+                    "extra_user":["wangwu"],
+                    "extra_tag":[4,5,6]
+                }
+            },
+            {
+                "agentid":2,
+                "name":"NAME2",
+                "round_logo_url":"xxxxxx",
+                "square_logo_url":"yyyyyy",
+                "appid":5
+            }
+        ]
+    },
+    "auth_user_info":
+    {
+        "email":"xxxx@aaa.com",
+        "mobile":"1234567890",
+        "userid":"aa"
+    }
+}