wechate 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a110108b510cc216f3545f8560781460ffec9688
-  data.tar.gz: dcf2e4b471bbe472e0a381fa55bd87f0563f28ff
+  metadata.gz: 23a0e41d3c5fcefc8dbe4a86ed23adeff8e775eb
+  data.tar.gz: f60d74769afdffbaf129826cbe43e580012667e5
 SHA512:
-  metadata.gz: c974f16723bd4d79560fca352c100542abe79dc918917612e4c3de80cc97c150280c6ae0b8211dd18e7bcab06d08f319efa9c7758dac4a95d1ef9e74b1b680d7
-  data.tar.gz: c6b87ebee98420a598b2b19119acc4de8de4287869476460112ae368369dcb0a7b6d6c0ed6065a3f6d5add3f7700fd1373d672b03a45af50f5a53797386d96ca
+  metadata.gz: 9251e27fa2454efde85ff544a5505adad10c78193ba0c42d441d49a3fcec05d8c08aa11fbdb3692d855ffa9be8e62ef9dd09e6ae49f7f701a8f217d41c518637
+  data.tar.gz: 80b69aa06f2186acf2087de14d6c7a549afef46ace84be9ebac8e9a54c61ae5fea082e361bcd877e3b2bcd6a84c6777065c0ca4db17f07ed71fb278b8926f6da

data/.gitignore

@@ -0,0 +1 @@
+*.gem

data/Gemfile

@@ -0,0 +1,2 @@
+source 'https://rubygems.org'
+gemspec

data/lib/wechate.rb

@@ -0,0 +1,16 @@
+require "base64"
+require "openssl"
+require 'digest/sha1'
+require 'faraday'
+require 'nokogiri'
+
+
+require "wechate/client"
+require "wechate/aes"
+require "wechate/http"
+require "wechate/factory"
+require "wechate/worker"
+
+module Wechate
+
+end

data/lib/wechate/aes.rb

@@ -0,0 +1,51 @@
+require "base64"
+require "openssl"
+require 'digest/sha1'
+module Wechate
+  module Aes
+    def pad text
+      text_length = text.length
+      amount_to_pad = 32 - (text_length % 32)
+      amount_to_pad = 32 if amount_to_pad == 0
+      pad = amount_to_pad
+      text + amount_to_pad.chr * amount_to_pad
+    end
+
+    def encode message,timestamp,nonce,from=nil,prefix='2aab967d2fff18c6'
+      msg = (prefix.bytes + [message.bytes.length].pack('N').bytes + "#{message}#{from}".bytes).pack('c*')
+      msg = pad(msg)
+      key = @aeskey
+      key = Base64.decode64(key)
+      cipher = OpenSSL::Cipher::Cipher
+      aes = cipher.new('aes-256-cbc').encrypt
+      aes.padding = 0
+      aes.key = key
+      aes.iv = key[0..15]
+      ret = aes.update(msg) + aes.final
+      msg = Base64.encode64(ret)
+      msg.gsub!("\n","")
+      [msg,sign(@token,timestamp,nonce,msg)]
+    end
+
+    def decode echostr,timestamp,nonce,msg_signature=nil
+      key = @aeskey
+      sign(@token,timestamp,nonce,echostr)
+      cipher = OpenSSL::Cipher::Cipher
+      aes = cipher.new('aes-256-cbc').decrypt
+      aes.padding = 0
+      key = Base64.decode64 key
+      msg = Base64.decode64 echostr
+      aes.key = key
+      aes.iv = key[0..15]
+      result = aes.update msg + aes.final
+      length = result[16,4].unpack('N').first
+      result[20,length].force_encoding('utf-8')
+    end
+
+    def sign(token,timestamp,nonce,msg_encrypt)
+      Digest::SHA1.hexdigest([token,timestamp,nonce,msg_encrypt].sort.join)
+    end
+
+  end
+
+end

data/lib/wechate/client.rb

@@ -0,0 +1,157 @@
+require 'nokogiri'
+require_relative 'http'
+require_relative 'aes'
+
+module Wechate
+  OpenSSL::SSL::VERIFY_PEER = OpenSSL::SSL::VERIFY_NONE
+  class Client
+    include Wechate::Http
+    include Wechate::Aes
+
+    def initialize aeskey,token,appid=nil,secret=nil
+      @host_api = 'https://qyapi.weixin.qq.com'
+      @aeskey = aeskey
+      @token = token
+      @appid = appid
+      @secret = secret
+      @access_token = nil
+      @worker = nil
+
+      @urls = {
+          :user_info  => "/cgi-bin/user/info",
+          :menu       => "/cgi-bin/menu/create",
+          :menu_get   => "/cgi-bin/menu/get",
+          :user_list  => "/cgi-bin/user/get",
+          :group_list => "/cgi-bin/groups/get",
+          :user_change_group => "/cgi-bin/groups/members/update",
+          :send => "/cgi-bin/message/custom/send",
+          :media => "/cgi-bin/media/get",
+          :message_send => "/cgi-bin/message/send",
+          :token => "/cgi-bin/gettoken",
+          :department_list => "/cgi-bin/department/list",
+      }
+      
+    end
+
+    def worker=(worker)
+      @worker = worker
+    end
+
+    def department_list
+      data = api_get(:department_list)
+      data["department"]
+    end
+
+    def api_get_user_list(next_openid=nil)
+      args = {}
+      args = {next_openid:next_openid} if next_openid
+      data = api_get(:user_list,args)
+      return nil unless data
+      data["data"]["openid"]
+    end
+
+    def ping message,users=nil,group=nil
+        params = {"touser"=> users ? users.join('|') : "@all","agentid"=>"1","safe"=>"0"}
+        case message
+        when String
+          params["msgtype"] = "text" 
+          params["text"] = {        
+            "content"=> message
+          }
+        when File
+          params["msgtype"] = "text" 
+          params["file"] = {
+            "media_id"=> "MEDIA_ID"
+          }    
+        when Array
+          params["msgtype"] = "news"
+          params["news"] = {
+             "articles"=>message
+          }
+        end
+        
+        api_post(:message_send,params.to_json.gsub(/\\u([0-9a-z]{4})/) {|s| [$1.to_i(16)].pack("U")})      
+    end
+
+    def pong message,timestamp,nonce
+      xml = Nokogiri::XML(message)
+      fromUserName = xml.xpath("//FromUserName").text
+      toUserName = xml.xpath("//ToUserName").text
+      timestamp = Time.now.to_i.to_s
+      response = get_response(xml)
+      msg_encrypt = case response
+      when String
+        build_text(fromUserName,toUserName,timestamp,response)
+      when Array
+        build_news(fromUserName,toUserName,timestamp,response)
+      when URI
+        build_image(fromUserName,toUserName,timestamp,response)
+      end
+      encode(msg_encrypt,timestamp,nonce,toUserName)
+    end
+
+    def get_response xml
+      raise "Invalid worker" unless @worker
+      create_time = xml.xpath("//CreateTime").text
+      msg_type = xml.xpath("//MsgType").text
+      fromUserName = xml.xpath("//FromUserName").text
+      agent_id =  xml.xpath("//AgentID").text
+
+      ret = case msg_type
+      when "event"
+        event = xml.xpath("//Event").text
+        event_key = xml.xpath("//EventKey").text
+        ret = case event 
+        when "click"
+          @worker.click_callback(fromUserName,event_key,create_time)
+        when "scancode_push"
+        when "scancode_waitmsg"      
+          @worker.scancode_callback(fromUserName,event_key,xml.xpath("//ScanResult").text)
+        when "location_select"
+        when "pic_photo_or_album"
+          @worker.unknown_event(fromUserName,event_key)
+        else
+          @worker.unknown_event(fromUserName,event_key)
+        end
+      else
+        msg_id = xml.xpath("//MsgId").text
+        case msg_type 
+        when "text"
+          @worker.text_callback(fromUserName,msg_id,xml.xpath("//PicUrl").text)
+        when "image"
+          args = {:pic=>xml.xpath("//PicUrl").text,
+            :media_id=>xml.xpath("//MediaId").text}
+          @worker.image_callback(fromUserName,msg_id,args)
+        when "location"
+          args = {:location=>{:lat=>xml.xpath("//Location_X").text,
+            :lng=>xml.xpath("//Location_Y").text},
+          :scale=>xml.xpath("//Scale").text,
+          :lable=>xml.xpath("//Label").text}
+          @worker.location_callback(fromUserName,msg_id,args)
+        when "voice"
+          args = {:format=>xml.xpath("//Format").text,
+            :media_id=>xml.xpath("//MediaId").text}
+          @worker.voice_callback(fromUserName,msg_id,args)
+        end
+      end
+    end
+
+    def build_text fromUserName,toUserName,timestamp,response
+      "<xml><ToUserName><![CDATA[#{fromUserName}]]></ToUserName><FromUserName><![CDATA[#{toUserName}]]></FromUserName><CreateTime>#{timestamp}</CreateTime><MsgType><![CDATA[text]]></MsgType><Content><![CDATA[#{response}]]></Content></xml>"
+    end
+
+    def build_image fromUserName,toUserName,timestamp,response
+      "<xml><ToUserName><![CDATA[toUser]]></ToUserName><FromUserName><![CDATA[fromUser]]></FromUserName><CreateTime>12345678</CreateTime><MsgType><![CDATA[news]]></MsgType><ArticleCount>2</ArticleCount><Articles><item><Title><![CDATA[title1]]></Title> <Description><![CDATA[description1]]></Description><PicUrl><![CDATA[picurl]]></PicUrl><Url><![CDATA[url]]></Url></item></Articles></xml>"
+    end
+
+    def build_news fromUserName,toUserName,timestamp,response
+      items = response.map do |item|
+        "<item><Title><![CDATA[#{item.title}]]></Title><Description><![CDATA[#{item.description}]]></Description><PicUrl><![CDATA[#{item.picurl}]]></PicUrl><Url><![CDATA[#{item.url}]]></Url></item>"
+      end.join
+
+      "<xml><ToUserName><![CDATA[#{fromUserName}]]></ToUserName><FromUserName><![CDATA[#{toUserName}]]></FromUserName><CreateTime>#{timestamp}</CreateTime><MsgType><![CDATA[news]]></MsgType><ArticleCount>#{response.length}</ArticleCount><Articles>#{items}</Articles></xml>"
+    end
+
+  end
+
+end

data/lib/wechate/factory.rb

@@ -0,0 +1,7 @@
+module Wechate
+  class Factory
+    def self.create_worker(worker)
+      worker.new
+    end
+  end
+end

data/lib/wechate/http.rb

@@ -0,0 +1,50 @@
+require 'faraday'
+require "openssl"
+
+module Wechate
+  module Http
+  def api_get(url,args={})
+    get_api_token unless @access_token
+    url = "#{@urls[url]}?"
+    args.each{|k,v| url = url + "&#{k.to_s}=#{v.to_s}"}
+    
+    if !@access_token then
+      get_api_token
+      return nil unless @access_token
+    end
+    
+    conn = Faraday.new(:url =>@host_api) 
+    req = url + "&access_token=#{@access_token}"
+    response = conn.get req
+    ret = JSON.parse(response.body)
+    if ret["errcode"]!=0 then #expired
+      if "42001"==ret["errcode"] then
+        return nil unless get_api_token
+        response = conn.get url + "&access_token=#{@access_token}"
+      else
+        return nil
+      end
+    end
+    return ret
+  end
+
+  def api_post(url,body)
+    get_api_token unless @access_token
+    conn = Faraday.new(:url => @host_api)
+    response = conn.post "#{@urls[url]}?access_token=#{@access_token}" do |req|
+      req.body = body
+    end
+    return response
+  end
+
+  def get_api_token
+    conn = Faraday.new(:url =>@host_api)
+    response = conn.get "/cgi-bin/gettoken" do |req|
+      req.params['corpid'] = @appid
+      req.params['corpsecret'] = @secret
+    end
+    @access_token = JSON.parse(response.body)["access_token"]
+  end
+end
+
+end

data/lib/wechate/version.rb

@@ -0,0 +1,3 @@
+module Wechate
+  VERSION = "0.1.4"
+end

data/lib/wechate/worker.rb

@@ -0,0 +1,32 @@
+module Wechate
+  class Worker
+
+    def text_callback(user,msg_id,text)
+      "hi"
+    end
+
+    def image_callback(user,msg_id,args)
+      "hi"
+    end
+
+    def location_callback(user,msg_id,args)
+      "hi"
+    end
+
+    def voice_callback(user,msg_id,args)
+      "hi"
+    end
+
+    def unknown_event(user,event,event_key)
+      "hi"
+    end
+
+    def click_callback(user,event_key,create_time)
+        "hi"
+    end
+
+    def scancode_callback(user,event_key,args)
+      "hi"
+    end
+  end
+end

data/wechate.gemspec

@@ -0,0 +1,24 @@
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "wechate/version"
+Gem::Specification.new do |spec|
+    spec.name        = 'wechate'
+    spec.version = Wechate::VERSION
+    spec.date        = '2010-04-28'
+    spec.summary     = ""
+    spec.description = ""
+    spec.authors   = ["goxplanet"]
+    spec.email     = 'otec.teng@gmail.com'
+    spec.license   = 'MIT'
+
+    spec.files         = `git ls-files`.split($/)
+    spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+    spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+    spec.require_paths = ["lib"]
+
+    spec.add_dependency "nokogiri"
+    spec.add_dependency "faraday"
+
+    spec.add_development_dependency "bundler", "~> 1.2"
+    spec.add_development_dependency "rake"    
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: wechate
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - goxplanet
@@ -38,12 +38,50 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.2'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: ''
 email: otec.teng@gmail.com
 executables: []
 extensions: []
 extra_rdoc_files: []
-files: []
+files:
+- .gitignore
+- Gemfile
+- lib/wechate.rb
+- lib/wechate/aes.rb
+- lib/wechate/client.rb
+- lib/wechate/factory.rb
+- lib/wechate/http.rb
+- lib/wechate/version.rb
+- lib/wechate/worker.rb
+- wechate.gemspec
 homepage: 
 licenses:
 - MIT