wework 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d7d9c3e6a5439548c70364e4a742552f7a31409d
-  data.tar.gz: 3453fdfbf36889692aeb268f727098832c6d6ff5
+  metadata.gz: da08105f38235bd4f06667e179d9c033a82146f1
+  data.tar.gz: cb7194f21cbb79a9c1ffa8e88e8e7362a167b663
 SHA512:
-  metadata.gz: 6c2d7704089c8e27229aca12e714eabf6ff81855f8e74d2b99704b251e4fc7c23dd6d0960a86ef151a593138fd90ab69a7c3a1cebb589582732f9d2f46d67c19
-  data.tar.gz: 0c0815e57a9e04a8601d5680082b8ff9e5a65421bf2e228c8ecb97b2198798383c5c569b1210b21ebf87bec6ecc4f29c9c758a2bc74291ee99de52aaa2158cdd
+  metadata.gz: 7ca30a8f14e08e9a5e47a22e7bee4235da1849cbbae5044ea81a1385458880a1c6470f5428b66d47d912caa240639995f4a0e21a9ba28dba629870b837e1defb
+  data.tar.gz: 720dc74a741a7e828c577326aafe2f76764bda9077efc2dcd04315e8b627599dae4444365099dad454794412cfcc86303d4af99e572ce3ec0fcc95c7222aa630

data/.gitignore

@@ -7,3 +7,5 @@
 /pkg/
 /spec/reports/
 /tmp/
+*.gem
+config.yml

data/README.md

@@ -20,20 +20,18 @@ Or install it yourself as:
 
 ## Usage
 
-TODO:
-
-## Development
-
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
-
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+TODO: ...
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/seandong/wework. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
-
+* Fork `Wework` on GitHub
+* Make your changes
+* Ensure all tests pass (`bundle exec rake`)
+* Send a pull request
+* If we like them we'll merge them
+* If we've accepted a patch, feel free to ask for commit access!
 
 ## License
 
-The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
+Copyright (c) 2016 MyColorway. See LICENSE.txt for further details.
 

data/lib/wework/api/agent.rb

@@ -0,0 +1,70 @@
+module Wework
+  module Api
+    class Agent < Base
+      # user agent: UA is mozilla/5.0 (iphone; cpu iphone os 10_2 like mac os x) applewebkit/602.3.12 (khtml, like gecko) mobile/14c92 wxwork/1.3.2 micromessenger/6.2
+
+      def get_info
+        get 'agent/get', params: {agentid: agent_id}
+      end
+
+      def set_info data={}
+        post 'agent/set', data.merge(agentid: agent_id)
+      end
+
+      def menu_create menu
+        post 'menu/create', menu, params: {agentid: agent_id}
+      end
+
+      def menu_delete
+        get 'menu/delete', params: {agentid: agent_id}
+      end
+
+      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 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
+
+      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, media_id, title='', description=''
+        message_send user_ids, department_ids, {video: {media_id: media_id, title: 'title', description: description}, msgtype: 'video'}
+      end
+
+      def textcard_message_send user_ids, department_ids, title, description, url, btntxt='详情'
+        message_send user_ids, department_ids, {textcard: {title: title, description: description, url: url, btntxt: btntxt}, msgtype: 'textcard'}
+      end
+
+      def news_message_send user_ids, department_ids, news=[]
+        message_send user_ids, department_ids, {news: {articles: news}, msgtype: 'news'}
+      end
+
+      def agent_id
+        @agent_id.to_i
+      end
+    end
+  end
+end

data/lib/wework/api/base.rb

@@ -0,0 +1,54 @@
+require 'wework/token/store'
+require 'wework/token/redis_store'
+
+module Wework
+  module Api
+    class Base
+      attr_reader :corp_id, :agent_id, :agent_secret
+      attr_accessor :options
+
+      delegate :access_token, to: :token_store
+
+      def initialize(corp_id, agent_id, agent_secret, options={})
+        @corp_id      = corp_id
+        @agent_id     = agent_id
+        @agent_secret = agent_secret
+        @options      = options
+      end
+
+      def token_store
+        @token_store ||= Token::RedisStore.new self
+      end
+
+      def request
+        @request ||= Wework::Request.new(API_ENDPOINT, false)
+      end
+
+      def get(path, headers = {})
+        with_access_token(headers[:params]) do |params|
+          request.get path, headers.merge(params: params)
+        end
+      end
+
+      def post(path, payload, headers = {})
+        with_access_token(headers[:params]) do |params|
+          request.post path, payload, headers.merge(params: params)
+        end
+      end
+
+      def post_file(path, file, headers = {})
+        with_access_token(headers[:params]) do |params|
+          request.post_file path, file, headers.merge(params: params)
+        end
+      end
+
+      def with_access_token(params = {}, tries = 2)
+        params ||= {}
+        yield(params.merge(access_token: access_token))
+      rescue AccessTokenExpiredError
+        token_store.refresh_token
+        retry unless (tries -= 1).zero?
+      end
+    end
+  end
+end

data/lib/wework/api/contact.rb

@@ -0,0 +1,54 @@
+module Wework
+  module Api
+    class Contact < Base
+
+      def initialize(corp_id, corp_secret)
+        super(corp_id, CONTACT_AGENT_ID, corp_secret)
+      end
+
+      def user_create userid, name, mobile, department, data={}
+        post 'user/create', data.merge(userid: userid, name: name, mobile: mobile, department: department)
+      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
+
+      def department_create name, parentid=0, data={}
+        post 'department/create', data.merge(name: name, parentid: parentid)
+      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=0
+        get 'department/list', params: {id: department_id}
+      end
+    end
+  end
+end

data/lib/wework/config.rb

@@ -0,0 +1,25 @@
+module Wework
+
+  class << self
+    attr_accessor :config
+    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: 5, connect: 5, read: 5}
+    end
+  end
+
+  class Config
+    attr_accessor :redis, :http_timeout_options
+  end
+end

data/lib/wework/engine.rb

@@ -0,0 +1,31 @@
+module Wework
+  class Engine
+
+    attr_reader :corp_id, :corp_secret, :app_id, :app_secret
+
+    def initialize(options={})
+      @corp_id = options[:corp_id]
+      @corp_secret = options[:corp_secret]
+      @app_id = options[:app_id]
+      @app_secret = options[:app_secret]
+    end
+
+    def contract
+      @contract ||= Wework::Api::Contact.new(corp_id, corp_secret) if contract?
+    end
+
+    def agent
+      @agent ||= Wework::Api::Agent.new(corp_id, app_id, app_secret) if agent?
+    end
+
+    private
+
+    def agent?
+      corp_id.present? && app_id.present? && app_secret.present?
+    end
+
+    def contract?
+      corp_id.present? && corp_secret.present?
+    end
+  end
+end

data/lib/wework/request.rb

@@ -0,0 +1,101 @@
+# Reference this: https://github.com/Eric-Guo/wechat/blob/master/lib/wechat/http_client.rb
+require 'http'
+
+module Wework
+  class Request
+    attr_reader :base, :ssl_context, :httprb
+
+    def initialize(base, skip_verify_ssl)
+      @base = base
+      @httprb = HTTP.timeout(:global, **Wework.http_timeout_options)
+      @ssl_context = OpenSSL::SSL::SSLContext.new
+      @ssl_context.ssl_version = :TLSv1_client
+      @ssl_context.verify_mode = OpenSSL::SSL::VERIFY_NONE if skip_verify_ssl
+    end
+
+    def get(path, get_header = {})
+      request(path, get_header) do |url, header|
+        params = header.delete(:params)
+        httprb.headers(header).get(url, params: params, ssl_context: ssl_context)
+      end
+    end
+
+    def post(path, post_body, post_header = {})
+      request(path, post_header) do |url, header|
+        params = header.delete(:params)
+        httprb.headers(header).post(url, params: params, json: post_body, ssl_context: ssl_context)
+      end
+    end
+
+    def post_file(path, file, post_header = {})
+      request(path, post_header) do |url, header|
+        params = header.delete(:params)
+        httprb.headers(header)
+          .post(url, params: params,
+                     form: { media: HTTP::FormData::File.new(file),
+                             hack: 'X' }, # Existing here for http-form_data 1.0.1 handle single param improperly
+                     ssl_context: ssl_context)
+      end
+    end
+
+    private
+
+    def request(path, header = {}, &_block)
+      url_base = header.delete(:base) || base
+      as = header.delete(:as)
+      header['Accept'] = 'application/json'
+      response = yield("#{url_base}#{path}", header)
+
+      raise ResponseError.new(response.status) unless HTTP_OK_STATUS.include?(response.status)
+
+      parse_response(response, as || :json) do |parse_as, data|
+        break data unless parse_as == :json && data['errcode'].present?
+        result = Wework::Result.new(data)
+        raise AccessTokenExpiredError if result.token_expired?
+        result
+      end
+    end
+
+    def parse_response(response, as)
+      content_type = response.headers[:content_type]
+      parse_as = {
+        %r{^application\/json} => :json,
+        %r{^image\/.*} => :file
+      }.each_with_object([]) { |match, memo| memo << match[1] if content_type =~ match[0] }.first || as || :text
+
+      case parse_as
+      when :file
+        file = Tempfile.new('tmp')
+        file.binmode
+        file.write(response.body)
+        file.close
+        data = file
+
+      when :json
+        data = JSON.parse response.body.to_s.gsub(/[\u0000-\u001f]+/, '')
+      else
+        data = response.body
+      end
+
+      yield(parse_as, data)
+    end
+  end
+
+  class Result < OpenStruct
+    def initialize(data)
+      data['full_message'] = "(#{data['errcode']}) #{data['errmsg']}"
+      super data
+    end
+
+    def token_expired?
+      # 42001: access_token timeout
+      # 40014: invalid access_token
+      # 40001, invalid credential, access_token is invalid or not latest hint
+      [42001, 40014, 40001].include?(errcode)
+    end
+
+    def success?
+      errcode == SUCCESS_CODE
+    end
+  end
+end

data/lib/wework/token/redis_store.rb

@@ -0,0 +1,40 @@
+module Wework
+  module Token
+    class RedisStore < Store
+
+      def initialize(agent)
+        raise RedisNotConfigException if redis.nil?
+        super
+      end
+
+      def access_token
+        super
+        redis.hget(key, "access_token")
+      end
+
+      def expired?
+        redis.hvals(key).empty?
+      end
+
+      def refresh_token
+        result = super
+
+        expires_at = Time.now.to_i + result['expires_in'].to_i - 100
+        redis.hmset(
+          key,
+          "access_token", result['access_token'],
+          "expires_at", expires_at
+        )
+
+        redis.expireat(key, expires_at)
+      end
+
+      private
+
+      def redis
+        Wework.redis
+      end
+
+    end
+  end
+end

data/lib/wework/token/store.rb

@@ -0,0 +1,31 @@
+module Wework
+  module Token
+    class Store
+
+      attr_accessor :agent
+
+      def initialize(agent)
+        @agent = agent
+      end
+
+      def access_token
+        refresh_token if expired?
+      end
+
+      def expired?
+        raise NotImplementedError, "Subclasses must implement a token_expired? method"
+      end
+
+      def refresh_token
+        agent.request.get 'gettoken', params: {corpid: agent.corp_id, corpsecret: agent.agent_secret}
+      end
+
+      private
+
+      def key
+        @key ||= Digest::MD5.hexdigest("#{ACCESS_TOKEN_PREFIX}_#{agent.agent_id}_#{agent.agent_secret}")
+      end
+
+    end
+  end
+end

data/lib/wework/version.rb

@@ -1,3 +1,3 @@
 module Wework
-  VERSION = "0.1.0"
+  VERSION = '0.1.2'.freeze
 end

data/lib/wework.rb

@@ -1,5 +1,30 @@
-require "wework/version"
+require 'redis'
+require 'active_support/all'
+#require 'active_support/core_ext/object/blank'
+
+Dir["#{File.dirname(__FILE__)}/wework/*.rb"].each do |path|
+  require path
+end
+
+require 'wework/api/base'
+require 'wework/api/agent'
+require 'wework/api/contact'
 
 module Wework
-  # Your code goes here...
+  API_ENDPOINT        = 'https://qyapi.weixin.qq.com/cgi-bin/'.freeze
+  ACCESS_TOKEN_PREFIX = 'WEWORK'.freeze
+  CONTACT_AGENT_ID    = 'CONTACT'.freeze
+  HTTP_OK_STATUS      = [200, 201].freeze
+  SUCCESS_CODE        = 0
+
+  # Exceptions
+  class RedisNotConfigException < RuntimeError; end
+  class AccessTokenExpiredError < RuntimeError; end
+  class ResponseError < StandardError
+    attr_reader :error_code
+    def initialize(errcode, errmsg='')
+      @error_code = errcode
+      super "(#{error_code}) #{errmsg}"
+    end
+  end
 end

data/wework.gemspec

@@ -20,7 +20,11 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
+  spec.add_dependency 'http', '>= 1.0.4', '< 3'
+  spec.add_dependency 'activesupport', '~> 5.0'
+  spec.add_dependency 'redis', '~>3.2'
+
   spec.add_development_dependency "bundler", "~> 1.13"
   spec.add_development_dependency "rake", "~> 10.0"
-  spec.add_development_dependency "minitest", "~> 5.0"
+  spec.add_development_dependency "minitest", "~> 5.10"
 end

metadata

@@ -1,15 +1,63 @@
 --- !ruby/object:Gem::Specification
 name: wework
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - seandong
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-12-16 00:00:00.000000000 Z
+date: 2016-12-19 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: http
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.4
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.4
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3'
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.0'
+- !ruby/object:Gem::Dependency
+  name: redis
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -44,14 +92,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '5.0'
+        version: '5.10'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '5.0'
+        version: '5.10'
 description: Ruby API wrapper for work wechat.
 email:
 - sindon@gmail.com
@@ -69,6 +117,14 @@ files:
 - bin/console
 - bin/setup
 - lib/wework.rb
+- lib/wework/api/agent.rb
+- lib/wework/api/base.rb
+- lib/wework/api/contact.rb
+- lib/wework/config.rb
+- lib/wework/engine.rb
+- lib/wework/request.rb
+- lib/wework/token/redis_store.rb
+- lib/wework/token/store.rb
 - lib/wework/version.rb
 - wework.gemspec
 homepage: https://github.com/seandong/wework