wechat-bot 0.1.0.alpha

data/lib/wechat/bot/configuration.rb

@@ -0,0 +1,68 @@
+module WeChat::Bot
+  class Configuration < OpenStruct
+    # 默认配置
+    #
+    # @return [Hash]
+    def self.default_config
+      {
+        # Bot Configurations
+        verbose: false,
+        fireman: 'filehelper',
+
+        # WeChat Configurations
+        app_id: "wx782c26e4c19acffb",
+        auth_url: "https://login.weixin.qq.com",
+        servers: [
+          {
+            index: "wx.qq.com",
+            file: "file.wx.qq.com",
+            push: "webpush.wx.qq.com",
+          },
+          {
+            index: "wx2.qq.com",
+            file: "file.wx2.qq.com",
+            push: "webpush.wx2.qq.com",
+          },
+          {
+            index: "wx8.qq.com",
+            file: "file.wx8.qq.com",
+            push: "webpush.wx8.qq.com",
+          },
+          {
+            index: "wechat.com",
+            file: "file.web.wechat.com",
+            push: "webpush.web.wechat.com",
+          },
+          {
+            index: "web2.wechat.com",
+            file: "file.web2.wechat.com",
+            push: "webpush.web2.wechat.com",
+          },
+        ],
+        cookies: "wechat-bot-cookies.txt",
+        special_users: [
+          'newsapp', 'filehelper', 'weibo', 'qqmail',
+          'fmessage', 'tmessage', 'qmessage', 'qqsync',
+          'floatbottle', 'lbsapp', 'shakeapp', 'medianote',
+          'qqfriend', 'readerapp', 'blogapp', 'facebookapp',
+          'masssendapp', 'meishiapp', 'feedsapp', 'voip',
+          'blogappweixin', 'brandsessionholder', 'weixin',
+          'weixinreminder', 'officialaccounts', 'wxitil',
+          'notification_messages', 'wxid_novlwrv3lqwv11',
+          'gh_22b87fa7cb3c', 'userexperience_alarm',
+        ],
+        user_agent: "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_5) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/59.0.3071.86 Safari/537.36",
+      }
+    end
+
+    def initialize(defaults = nil)
+      defaults ||= self.class.default_config
+      super(defaults)
+    end
+
+    # @return [Hash]
+    def to_h
+      @table.clone
+    end
+  end
+end

data/lib/wechat/bot/contact.rb

@@ -0,0 +1,214 @@
+module WeChat::Bot
+  # 微信联系人
+  #
+  # 可以是用户、公众号、群组等
+  class Contact
+    # 联系人分类
+    module Kind
+      User = :user
+      Group = :group
+      MP = :mp
+      Special = :special
+    end
+
+    def self.parse(obj, bot)
+      self.new(bot).parse(obj)
+    end
+
+    def initialize(bot)
+      @bot = bot
+      @data = {}
+    end
+
+    # 用户唯一 ID
+    def username
+      attr(:username)
+    end
+
+    # 用户昵称
+    def nickname
+      attr(:nickname)
+    end
+
+    # 备注名
+    def remarkname
+      attr(:remarkname)
+    end
+
+    # 群聊显示名
+    def displayname
+      attr(:displayname)
+    end
+
+    # 性别
+    def sex
+      attr(:sex)
+    end
+
+    # 个人签名
+    def signature
+      attr(:signature)
+    end
+
+    # 用户类型
+    def kind
+      attr(:kind)
+    end
+
+    # 省份
+    def province
+      attr(:province)
+    end
+
+    # 城市
+    def city
+      attr(:city)
+    end
+
+    # 是否特殊账户
+    def special?
+      kind == Kind::Special
+    end
+
+    # 是否群聊
+    def group?
+      kind == Kind::Group
+    end
+
+    # 是否公众号
+    def mp?
+      kind == Kind::MP
+    end
+
+    # 群组成员列表
+    #
+    # 只有群组才有内容，根据 {#kind} 或 {#group?} 来判断
+    # 不是群组类型的返回空数组
+    #
+    # @return [Hash]
+    def members
+      attr(:members)
+    end
+
+    # 联系人解析
+    #
+    # @param [Hash<Object, Object>] raw
+    # @return [Contact]
+    def parse(raw, update = false)
+      @raw = raw
+
+      parse_kind
+      parse_members
+
+      @raw.each do |key, value|
+        if attribute = mapping[key]
+          next if value.to_s.empty? && update
+
+          sync(attribute, value)
+        end
+      end
+
+      self
+    end
+
+    def update(raw)
+      @raw = raw
+      parse(@raw, true)
+    end
+
+    def to_s
+      "#<#{self.class}:#{object_id.to_s(16)} username='#{username}' nickname='#{nickname}' kind='#{kind}'>"
+    end
+
+    private
+
+    # 更新或新写入变量值
+    #
+    # @param [Symbol] attribute
+    # @param [String, Integer, Hash] value
+    # @param [Boolean] data
+    # @return [void]
+    def sync(attribute, value, data = false)
+      value = value.convert_emoji if attribute.to_sym == :nickname
+
+      # 满足群组类型且 nickname 为空时补充一个默认的群组名（参考微信 App 设计）
+      if attribute.to_sym == :nickname && value.to_s.empty? && @kind == Kind::Group
+        value = members.map {|m| m.nickname }.join("、")
+      end
+
+      if data
+        @data[attribute.to_sym] = value
+      else
+        instance_variable_set("@#{attribute}", value)
+      end
+    end
+
+    # 获取属性
+    #
+    # @param [Symbol] attribute
+    # @param [Boolean] data 默认 false
+    # @return [String, Integer, Hash]
+    def attr(attribute, data = false)
+      if data
+        @data[attribute.to_sym]
+      else
+        instance_variable_get("@#{attribute}")
+      end
+    end
+
+    # 解析联系人类型
+    #
+    # 详见 {Contact::Kind} 成员变量
+    # @return [void]
+    def parse_kind
+      kind = if @bot.config.special_users.include?(@raw["UserName"])
+               # 特殊账户
+               Kind::Special
+             elsif @raw["UserName"].include?("@@")
+               # 群聊
+               Kind::Group
+             elsif @raw["VerifyFlag"] && (@raw["VerifyFlag"] & 8) != 0
+               # 公众号
+               Kind::MP
+             else
+               # 普通用户
+               Kind::User
+             end
+
+      sync(:kind, kind)
+    end
+
+    # 解析群组成员列表
+    #
+    # 只有群组才有内容，根据 {#kind} 或 {#group?} 来判断
+    #
+    # @return [void]
+    def parse_members
+      members = []
+
+      if @raw["MemberList"]
+        @raw["MemberList"].each do |m|
+          members.push(Contact.parse(m, @bot))
+        end
+      end
+
+      sync(:members, members)
+    end
+
+    # 字段映射
+    #
+    # @return [Hash<String, String>]
+    def mapping
+      {
+        "NickName" => "nickname",
+        "UserName" => "username",
+        "RemarkName" => "remarkname",
+        "DisplayName" => "displayname",
+        "Signature" => "signature",
+        "Sex" => "sex",
+        "Province" => "province",
+        "City" => 'city'
+      }
+    end
+  end
+end

data/lib/wechat/bot/contact_list.rb

@@ -0,0 +1,56 @@
+module WeChat::Bot
+  # 微信联系人列表
+  class ContactList < CachedList
+    # 批量同步联系人数据
+    #
+    # 更多查看 {#sync} 接口
+    # @param [Array<Hash>] list 联系人数组
+    # @return [ContactList]
+    def batch_sync(list)
+      list.each do |item|
+        sync(item)
+      end
+
+      self
+    end
+
+    def size
+      @cache.size
+    end
+
+    # 创建用户或更新用户数据
+    #
+    # @param [Hash] data 微信接口返回的单个用户数据
+    # @return [Contact]
+    def sync(data)
+      @mutex.synchronize do
+        contact = Contact.parse(data, @bot)
+        if @cache[contact.username]
+          @cache[contact.username].update(data)
+        else
+          @cache[contact.username] = contact
+        end
+
+        contact
+      end
+    end
+
+    # 查找用户
+    #
+    # @param [Hash] args 接受两个参数:
+    #   - :nickname 昵称
+    #   - :username 用户ID
+    # @return [Contact]
+    def find(**args)
+      @mutex.synchronize do
+        return @cache[args[:username]] if args[:username]
+
+        if args[:nickname]
+          @cache.each do |_, contact|
+            return contact if contact.nickname == args[:nickname]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/wechat/bot/core.rb

@@ -0,0 +1,138 @@
+require "wechat/bot/configuration"
+require "wechat/bot/cached_list"
+require "wechat/bot/contact_list"
+require "wechat/bot/contact"
+require "wechat/bot/logger"
+
+require "wechat/bot/handler_list"
+require "wechat/bot/handler"
+require "wechat/bot/message"
+require "wechat/bot/pattern"
+require "wechat/bot/callback"
+
+require "wechat/bot/http/adapter/js"
+require "wechat/bot/http/adapter/xml"
+require "wechat/bot/http/session"
+
+require "logger"
+
+module WeChat::Bot
+  # 机器人的核心类
+  class Core
+    # 微信 API 客户端
+    #
+    # @return [Client]
+    attr_reader :client
+
+    # 当前登录用户信息
+    #
+    # @return [Contact]
+    attr_reader :profile
+
+    # 联系人列表
+    #
+    # @return [ContactList]
+    attr_reader :contact_list
+
+    # @return [Logger]
+    attr_accessor :logger
+
+    # @return [HandlerList]
+    attr_reader :handlers
+
+    # @return [Configuration]
+    attr_reader :config
+
+    # @return [Callback]
+    # @api private
+    attr_reader :callback
+
+    def initialize(&block)
+      # defaults_logger
+      @logger = Logger.new(STDOUT, self)
+      @config = Configuration.new
+      @handlers = HandlerList.new
+      @callback = Callback.new(self)
+
+      @client = Client.new(self)
+      @profile = Contact.new(self)
+      @contact_list = ContactList.new(self)
+
+      instance_eval(&block) if block_given?
+    end
+
+    # 消息触发器
+    #
+    # @param [String, Symbol, Integer] event
+    # @param [Regexp, Pattern, String] regexp
+    # @param [Array<Object>] args
+    # @yieldparam [Array<String>]
+    # @return [Handler]
+    def on(event, regexp = //, *args, &block)
+      event = event.to_s.to_sym
+
+      pattern = case regexp
+                when Pattern
+                  regexp
+                when Regexp
+                  Pattern.new(nil, regexp, nil)
+                else
+                  if event == :ctcp
+                    Pattern.generate(:ctcp, regexp)
+                  else
+                    Pattern.new(/^/, /#{Regexp.escape(regexp.to_s)}/, /$/)
+                  end
+                end
+
+      handler = Handler.new(self, event, pattern, {args: args, execute_in_callback: true}, &block)
+      @handlers.register(handler)
+
+      handler
+    end
+
+    # 用于设置 WeChat::Bot 的配置
+    # 默认无需配置，需要定制化 yield {Core#config} 进行配置
+    #
+    # @yieldparam [Struct] config
+    # @return [void] 没有返回值
+    def configure
+      yield @config
+    end
+
+    # 运行机器人
+    #
+    # @return [void]
+    def start
+      @client.login
+      @client.contacts
+
+      @contact_list.each do |c|
+        @logger.debug "Contact: #{c}"
+      end
+
+      while true
+        break unless @client.logged? || @client.alive?
+        sleep 1
+      end
+    rescue Interrupt => e
+      message = "你使用 Ctrl + C 终止了运行"
+      @logger.warn(message)
+      @client.send_text(@config.fireman, "[告警] 意外下线\n#{message}\n#{e.backtrace.join("\n")}") if @client.logged? && @client.alive?
+    rescue Exception => e
+      message = e.message
+      @logger.fatal(e)
+      @client.send_text(@config.fireman, "[告警] 意外下线\n#{message}\n#{e.backtrace.join("\n")}") if @client.logged? && @client.alive?
+    ensure
+      @client.logout if @client.logged? && @client.alive?
+    end
+
+    private
+
+    # def defaults_logger
+    #   @logger = Logger.new($stdout)
+    #   @logger.formatter = proc do |severity, datetime, progname, msg|
+    #     "#{severity}\t[#{datetime.strftime("%Y-%m-%d %H:%M:%S.%2N")}]: #{msg}\n"
+    #   end
+    # end
+  end
+end