lita-tox/lib/lita/adapters/tox.rb

require 'tox'

##
# Lita module.
#
module Lita
  ##
  # Lita adapters module.
  #
  module Adapters
    ##
    # Tox adapter for the Lita chat bot.
    #
    class Tox < Adapter
      config :savedata_filename, type: String
      config :status, type: String

      def initialize(robot)
        super

        options = ::Tox::Options.new

        if config.savedata_filename && File.exist?(config.savedata_filename)
          savedata_file = open(config.savedata_filename)
          options.data = savedata_file.read
          savedata_file.close
        end

        @tox = ::Tox.new(options)

        log.info("ID: #{@tox.id}")

        @tox.name = robot.name if robot.name
        @tox.status_message = config.status if config.status

        @tox.on_friend_request do |key|
          @tox.friend_add_norequest(key)
        end

        @tox.on_friend_message do |friend_number, text|
          user = User.new(friend_number)
          source = Source.new(user: user)
          message = Message.new(robot, text, source)

          message.command!
          robot.receive(message)
        end
      end

      def run
        @tox.loop
      end

      def shut_down
        if config.savedata_filename
          savedata_file = open(config.savedata_filename, 'w')
          savedata_file.write(@tox.savedata)
          savedata_file.close
        end

        @tox.kill
      end

      def send_messages(target, messages)
        messages.reject(&:empty?).each do |message|
          @tox.friend_send_message(target.user.id.to_i, message)
        end
      end
    end

    Lita.register_adapter(:tox, Tox)
  end
end