ruby-cqhttp 0.0.7 → 0.1.0

data/lib/Core/WebSocket/API.rb

@@ -0,0 +1,73 @@
+module Onebot
+  # WebSocket连接处理部分
+  module WebSocket
+    class API
+      attr_accessor :queueList
+
+      def initialize(ws, eventLogger)
+        @ws = ws
+        @eventLogger = eventLogger
+        @queueList = {}
+      end
+
+      def get_msg(message_id, _url = @apiUrl)
+        ret = send('get_msg', { message_id: })
+        if parseRet(ret)
+          @eventLogger.log "获取消息成功 (#{message_id})"
+        else
+          @eventLogger.log "获取消息失败，错误码: #{ret.msg}, 错误消息: #{ret.wording}", Logger::WARN
+        end
+        ret[:data]
+      end
+
+      # 发送私聊消息
+      #
+      # @param msg [String]
+      # @param user_id [Number]
+      # @return [Hash]
+      def sendPrivateMessage(message, user_id)
+        ret = send('send_private_msg', { user_id:, message: })
+        if parseRet(ret)
+          @eventLogger.log "发送至私聊 #{user_id} 的消息: #{message} (#{ret.data.message_id})"
+        else
+          @eventLogger.log "发送私聊消息失败，错误码: #{ret.msg}, 错误消息: #{ret.wording}", Logger::WARN
+        end
+        ret[:data]
+      end
+
+      # 发送群聊消息
+      #
+      # @param msg [String]
+      # @param group_id [Number]
+      # @return [Hash]
+      def sendGroupMessage(message, group_id)
+        ret = send('send_group_msg', { group_id:, message: })
+        if parseRet(ret)
+          @eventLogger.log "发送至群 #{group_id} 的消息: #{message} (#{ret.data.message_id})"
+        else
+          @eventLogger.log "发送群消息失败，错误码: #{ret.msg}, 错误消息: #{ret.wording}", Logger::WARN
+        end
+        ret[:data]
+      end
+
+      private
+
+      #
+      #  解析API返回
+      #
+      def parseRet(ret)
+        return true if ret.status == 'ok'
+        return false if ret.status == 'failed'
+      end
+
+      def send(action, params)
+        echo = Time.now.to_i.to_s
+        @ws.send({ action:, params:, echo: }.to_json)
+        @queueList[echo] = Queue.new
+        ret = @queueList[echo].pop
+        @queueList.delete(echo)
+        ret
+      end
+    end
+  end
+end

data/lib/Core/WebSocket/Bot.rb

@@ -0,0 +1,69 @@
+module Onebot
+  # WebSocket连接处理部分
+  module WebSocket
+    class Bot
+      include EventEmitter
+      # @return [Number] self QQ id
+      attr_accessor :selfID
+
+      def initialize(**args); end
+
+      def method_missing(name, *args, &)
+        return @api.send(name, *args, &) if !@api.nil? && @api.respond_to?(name)
+
+        super
+      end
+
+      def respond_to_missing?(method_name, include_private = false)
+        (!@api.nil? && @api.respond_to?(name)) || super
+      end
+
+      # 发送消息
+      def sendMessage(msg, session)
+        return sendGroupMessage msg, session.group_id if session.message_type == 'group'
+        return sendPrivateMessage msg, session.user_id if session.message_type == 'private'
+      end
+
+      private
+
+      #
+      #  消息解析部分
+      #
+      def dataParse(data)
+        msg = JSON.parse(data, symbolize_names: true)
+        @eventLogger.dataParse(msg)
+        # 连接成功
+        if msg.meta_event_type == 'lifecycle' && msg.sub_type == 'connect'
+          @selfID = msg.self_id
+          emit :logged, @selfID
+        end
+        #
+        # 函数回调
+        #
+        @api.queueList[msg.echo] << msg if msg.include?(:echo) # 往API模块回调返回消息
+        case msg.post_type
+        #
+        # 请求事件
+        #
+        when 'request'
+          emit :request, msg.request_type, msg
+          #
+          # 提醒事件
+          #
+        when 'notice'
+          emit :notice, msg.notice_type, msg
+          #
+          # 消息事件
+          #
+        when 'message'
+          if msg.message_type == 'group' # 判断是否为群聊
+            emit :groupMessage, msg
+          else
+            emit :privateMessage, msg
+          end
+          emit :message, msg
+        end
+      end
+    end
+  end
+end

data/lib/Core/WebSocket/Client.rb

@@ -0,0 +1,44 @@
+module Onebot
+  # WebSocket连接处理部分
+  module WebSocket
+    class Client < Bot
+      # @return [URI] WS URL
+      attr_accessor :url
+      # @return [Faye::WebSocket::Client] WS Conn
+      attr_accessor :ws
+      attr_accessor :api
+
+      # 设置 WS URL
+      def initialize(url: nil, logger: nil, **args)
+        super
+        @eventLogger = Logging::EventLogger.new(logger)
+        @url = url
+      end
+
+      # 连接 WS
+      def connect(protocols = nil, options = {})
+        @eventLogger.log '正在连接到 ' << @url
+        EM.run do
+          @ws = Faye::WebSocket::Client.new(@url, protocols, options)
+          @api = API.new(@ws, @eventLogger)
+
+          @ws.on :message do |event|
+            Thread.new { dataParse(event.data) }
+          end
+
+          @ws.on :close do |event|
+            emit :close, event
+            @eventLogger.log '连接断开'
+            @ws = nil
+            EM.stop
+          end
+
+          @ws.on :error do |event|
+            emit :error, event
+            @ws = nil
+          end
+        end
+      end
+    end
+  end
+end

data/lib/Core/WebSocket/Server.rb

@@ -0,0 +1,35 @@
+module Onebot
+  # WebSocket连接处理部分
+  module WebSocket
+    class Server < Bot
+      # @return [Faye::WebSocket] ws server
+      attr_accessor :ws
+      attr_accessor :api
+
+      require 'rack'
+
+      def initialize(env:, logger: nil, options: { ping: 5 }, &block)
+        super
+        @eventLogger = Logging::EventLogger.new(logger)
+        @ws = Faye::WebSocket.new(env, %w[irc xmpp], options)
+        @api = API.new(@ws, @eventLogger)
+        @eventLogger.log(['客户端', '连接', @ws.url, @ws.version], ::Logger::INFO, 'Puma')
+
+        @ws.on :message do |event|
+          Thread.new { dataParse(event.data) }
+        end
+
+        @ws.on :close do |event|
+          @eventLogger.log(['客户端', '断开', event.code].to_s, ::Logger::INFO, 'Puma')
+          @ws = nil
+        end
+
+        yield self if block_given?
+      end
+
+      def rack_response
+        @ws.rack_response
+      end
+    end
+  end
+end

data/lib/onebot-ruby.rb

@@ -0,0 +1,44 @@
+require 'faye/websocket'
+require 'json'
+require 'event_emitter'
+require 'net/http'
+require 'logger'
+
+class Hash
+  def method_missing(m, *a)
+    if m.to_s =~ /=$/
+      self[Regexp.last_match.pre_match] = a[0]
+    elsif a.empty?
+      self[m]
+    else
+      super
+    end
+  end
+
+  def respond_to_missing?(m, include_private = false)
+    super unless m.to_s =~ /=$/
+  end
+end
+
+# 一个基于 OneBot 标准的 QQ 机器人框架
+module Onebot
+  module Logging
+    autoload :Logger, File.expand_path('Core/Logging/Logger', __dir__)
+    autoload :EventLogger, File.expand_path('Core/Logging/EventLogger', __dir__)
+  end
+
+  module WebSocket
+    autoload :Session, File.expand_path('Core/Websocket/Session', __dir__)
+    autoload :API, File.expand_path('Core/Websocket/API', __dir__)
+    autoload :Bot, File.expand_path('Core/Websocket/Bot', __dir__)
+    autoload :Client, File.expand_path('Core/Websocket/Client', __dir__)
+    autoload :Server, File.expand_path('Core/Websocket/Server', __dir__)
+  end
+
+  module Http
+    autoload :API, File.expand_path('Core/Http/API', __dir__)
+  end
+
+  autoload :Core, File.expand_path('Core/Core', __dir__)
+  autoload :Utils, File.expand_path('Core/Utils', __dir__)
+end

data/ruby-cqhttp.gemspec

@@ -3,22 +3,21 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |s|
   s.name = 'ruby-cqhttp'
-  s.version       = '0.0.7'
-  s.summary       = '一个基于 OneBot 标准的 QQ 机器人框架'
-  s.description   = '一个基于 OneBot 标准的 QQ 机器人框架'
-  s.authors       = ['fantasyzhjk']
-  s.email         = 'fantasyzhjk@outlook.com'
-  s.platform    = Gem::Platform::RUBY
-  s.homepage      = 'https://github.com/fantasyzhjk/ruby-cqhttp/'
-  s.license       = 'MIT'
-  s.files         = `git ls-files -z`.split("\x0")
+  s.version = '0.1.0'
+  s.summary = '一个基于 OneBot 标准的 QQ 机器人框架'
+  s.description = '一个基于 OneBot 标准的 QQ 机器人框架'
+  s.authors = ['fantasyzhjk']
+  s.email = 'fantasyzhjk@outlook.com'
+  s.platform = Gem::Platform::RUBY
+  s.homepage = 'https://github.com/fantasyzhjk/ruby-cqhttp/'
+  s.license = 'MIT'
+  s.files = `git ls-files -z`.split("\x0")
   s.require_paths = ['lib']
   s.required_ruby_version = '>= 2.7.0'
 
-  s.add_runtime_dependency  'json'
-  s.add_runtime_dependency  'logger'
-  s.add_runtime_dependency  'uri', '~> 0.10.1'
-  s.add_runtime_dependency  'eventmachine'
-  s.add_runtime_dependency  'faye-websocket'
-  s.add_runtime_dependency  'event_emitter'
+  s.add_runtime_dependency 'event_emitter'
+  s.add_runtime_dependency 'faye-websocket'
+  s.add_runtime_dependency 'json'
+  s.add_runtime_dependency 'logger'
+  s.add_runtime_dependency 'rack'
 end

data/tests/client.rb

@@ -0,0 +1,21 @@
+require '../lib/onebot-ruby'
+
+logger = Onebot::Logging::Logger.new.setLoggerLevel(Logger::INFO)
+api = Onebot::Http::API.new.setLogger(logger)
+
+Onebot::Core.connect url: 'ws://localhost:7700', logger: logger, options: { headers: { 'Authorization' => 'Bearer xxxxxxxxxxxx' } } do |bot|
+  bot.on :logged do |_botqq|
+    logger.log('我开了欸')
+  end
+
+  bot.on :privateMessage do |session|
+    bot.sendMessage(session.message, session)
+  end
+
+  bot.on :notice do |notice_type, data|
+    if notice_type == 'friend_recall'
+      req = bot.get_msg(data.message_id)
+      bot.sendPrivateMessage req.message, req.sender.user_id
+    end
+  end
+end

data/tests/rainbows.conf

@@ -0,0 +1,3 @@
+Rainbows! do
+  use :EventMachine
+end

data/tests/server.rb

@@ -0,0 +1,77 @@
+require '../lib/onebot-ruby'
+require 'faye/websocket'
+require 'rack'
+
+logger = Onebot::Logging::Logger.new.setLoggerLevel(Logger::INFO)
+static  = Rack::File.new(File.dirname(__FILE__))
+options = { ping: 5 }
+
+App = lambda do |env|
+  if Faye::WebSocket.websocket?(env)
+    bot = Onebot::WebSocket::Server.new(env:, logger:, options:)
+
+    bot.on :logged do |_botqq|
+      logger.log('我开了欸')
+    end
+
+    bot.on :privateMessage do |session|
+      # p session.message
+      # p session
+      p Onebot::Utils.cqParse(session.message)
+      bot.sendPrivateMessage(session.message, session.userId)
+    end
+
+    bot.rack_response
+  else
+    static.call(env)
+  end
+end
+
+def App.log(message); end
+
+port   = ARGV[0] || 7000
+secure = ARGV[1] == 'tls'
+engine = ARGV[2] || 'thin'
+spec   = File.expand_path(__dir__)
+
+Faye::WebSocket.load_adapter(engine)
+
+case engine
+
+when 'goliath'
+  class WebSocketServer < Goliath::API
+    def response(env)
+      App.call(env)
+    end
+  end
+
+when 'puma'
+  require 'puma/binder'
+  require 'puma/events'
+  events = Puma::Events.new($stdout, $stderr)
+  binder = Puma::Binder.new(events)
+  binder.parse(["tcp://0.0.0.0:#{port}"], App)
+  server = Puma::Server.new(App, events)
+  server.binder = binder
+  server.run.join
+
+when 'rainbows'
+  rackup = Unicorn::Configurator::RACKUP
+  rackup[:port] = port
+  rackup[:set_listener] = true
+  options = rackup[:options]
+  options[:config_file] = File.expand_path('rainbows.conf', __dir__)
+  Rainbows::HttpServer.new(App, options).start.join
+
+when 'thin'
+  thin = Rack::Handler.get('thin')
+  thin.run(App, Host: '0.0.0.0', Port: port) do |server|
+    if secure
+      server.ssl_options = {
+        private_key_file: spec + '/server.key',
+        cert_chain_file: spec + '/server.crt'
+      }
+      server.ssl = true
+    end
+  end
+end

metadata

@@ -1,17 +1,17 @@
 --- !ruby/object:Gem::Specification
 name: ruby-cqhttp
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.1.0
 platform: ruby
 authors:
 - fantasyzhjk
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-03-24 00:00:00.000000000 Z
+date: 2022-05-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: json
+  name: event_emitter
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -25,7 +25,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: logger
+  name: faye-websocket
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -39,21 +39,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: uri
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.10.1
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - "~>"
-      - !ruby/object:Gem::Version
-        version: 0.10.1
-- !ruby/object:Gem::Dependency
-  name: eventmachine
+  name: json
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -67,7 +53,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: faye-websocket
+  name: logger
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -81,7 +67,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: event_emitter
+  name: rack
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -100,19 +86,31 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".bundle/config"
 - ".gitignore"
+- ".rubocop.yml"
 - Gemfile
 - README.md
-- lib/Bot/Api.rb
-- lib/Bot/Bot.rb
-- lib/Bot/Utils.rb
-- lib/ruby-cqhttp.rb
+- icon.png
+- lib/Core/Core.rb
+- lib/Core/Http/API.rb
+- lib/Core/Logging/EventLogger.rb
+- lib/Core/Logging/Logger.rb
+- lib/Core/Utils.rb
+- lib/Core/WebSocket/API.rb
+- lib/Core/WebSocket/Bot.rb
+- lib/Core/WebSocket/Client.rb
+- lib/Core/WebSocket/Server.rb
+- lib/onebot-ruby.rb
 - ruby-cqhttp.gemspec
+- tests/client.rb
+- tests/rainbows.conf
+- tests/server.rb
 homepage: https://github.com/fantasyzhjk/ruby-cqhttp/
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -127,8 +125,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.4
-signing_key: 
+rubygems_version: 3.3.7
+signing_key:
 specification_version: 4
 summary: 一个基于 OneBot 标准的 QQ 机器人框架
 test_files: []