lita-discord 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: a37c01b611457e695c4a7f73d5c46b79058ee502
+  data.tar.gz: 75d7e3923aead4abb4ecc9010c53a906c3f1da70
+SHA512:
+  metadata.gz: 48dcfc7e93bf466930fd4cd209cb384ce3824ad86c5986d625ef1975d994402fe8818e0b70be27582844bac60e85ec1eab9f2acada0dd2e79ac3a501da973d1d
+  data.tar.gz: 2dbd6a02802ef02058a49c02469743da17eee803426656e2561e57408bbb6e596e71b05f654a83a208b6f1c78dbd14f290419f7ace08504cfef770c9dbbb1b34

data/.gitignore

@@ -0,0 +1,20 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+lita_config.rb
+lib/lita/adapters/discord/template.rb
+*.json

data/.travis.yml

@@ -0,0 +1,8 @@
+language: ruby
+rvm:
+  - 2.0.0
+script: bundle exec rake
+before_install:
+  - gem update --system
+services:
+  - redis-server

data/Gemfile

@@ -0,0 +1,3 @@
+source "https://rubygems.org"
+
+gemspec

data/README.md

@@ -0,0 +1,31 @@
+# lita-discord
+
+[![Build Status](https://travis-ci.org/kyleboe/lita-discord.png?branch=master)](https://travis-ci.org/kyleboe/lita-discord)
+
+**lita-discord** is an adapter for [Lita](https://www.lita.io/) that allows you to use the robot with [Discord](https://discordapp.com/)
+
+## Installation
+
+Add lita-discord to your Lita instance's Gemfile:
+
+``` ruby
+gem "lita-discord"
+```
+
+## Configuration
+
+You will need to go to the discord website and make an account that will be the account that your bot will function as. (Fix this awful wording later)
+
+### Required attributes
+
+* `email` (String) - The email of the account of the bot you created
+* `password` (String) - The password of the account of the bot you created
+
+
+### Optional attributes
+
+TODO: Put the optional attributes here
+
+## Usage
+
+TODO: Describe the plugin's features and how to use them.

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task default: :spec

data/lib/lita/adapters/discord/api.rb

@@ -0,0 +1,64 @@
+require 'rest-client'
+require 'json'
+
+module Lita
+	module Adapters
+		class Discord < Adapter
+			class API
+				APIBASE = 'https://discordapp.com/api'.freeze
+
+				def user_agent
+				    # This particular string is required by the Discord devs.
+				    required = "lita-discord (https://github.com/kyleboe/lita-discord, v0.1.0)"
+					@bot_name ||= ''
+
+					"rest-client/#{RestClient::VERSION} #{RUBY_ENGINE}/#{RUBY_VERSION}p#{RUBY_PATCHLEVEL} lita-discord/0.1.0 #{required} #{@bot_name}"
+				end
+
+				# Referenced https://github.com/meew0/discordrb/blob/9897dad08370d4de5de738c8f6c27b8c7764c429/lib/discordrb/api.rb#L35
+				def raw_request(type, attributes)
+				    RestClient.send(type, *attributes)
+					rescue RestClient::Forbidden
+					raise Lita.logger.debug("The bot doesn't have the required permission to do this!")
+				end
+
+				def request(type, *attributes)
+					# Add a custom user agent
+				    attributes.last[:user_agent] = user_agent if attributes.last.is_a? Hash
+				    response = raw_request(type, attributes)
+
+				    while response.code == 429
+				    	wait_seconds = response[:retry_after].to_i / 1000.0
+				    	Lita.logger.debug("WARNING: Discord rate limiting will cause a delay of #{wait_seconds} seconds for the request: #{type} #{attributes}")
+				    	sleep wait_seconds / 1000.0
+				    	response = raw_request(type, attributes)
+					end
+					response				
+				end
+
+				# Make an avatar URL from the user and avatar IDs
+				def avatar_url(user_id, avatar_id)
+					"#{APIBASE}/users/#{user_id}/avatars/#{avatar_id}.jpg"
+				end
+
+				# Login to the server
+				def login(email, password)
+					request( :post, "#{APIBASE}/auth/login", email: email, password: password )
+				end
+
+				# Logout from the server
+				def logout(token)
+					request( :post, "#{APIBASE}/auth/logout", nil, Authorization: token )
+				end
+
+				def validate_token(token)
+					request( :post, "#{APIBASE}/auth/login",{}.to_json, Authorization: token, content_type: :json )
+				end
+
+				def user(token, user_id)
+					request( :get, "#{APIBASE}/users/#{user_id}", Authorization: token )
+				end
+			end
+		end
+	end
+end

data/lib/lita/adapters/discord/client.rb

@@ -0,0 +1,87 @@
+require 'rest-client'
+require 'faye/websocket'
+require 'eventmachine'
+require 'lita/adapters/discord/api'
+require 'lita/adapters/discord/token_cache'
+
+module Lita
+	module Adapters
+		class Discord < Adapter
+			class Client
+				def initialize(email, password)
+					@email = email
+					@password = password
+					@token_cache = TokenCache.new
+					@channels = {}
+					@users = {}
+					@restricted_channels = []
+					@event_threads = []
+					@current_thread = 0
+				end
+
+				def connect
+					@token = login
+				end
+
+				def disconnect
+					API.logout
+				end
+
+
+				private
+
+				def login
+					if @email == :token
+						Lita.logger.debug('Logging in using static token')
+						# The password is the token!
+						return @password
+					end
+
+					Lita.logger.debug('Logging in')
+					login_attempts ||= 0
+
+					# First, attempt to get the token from the cache
+					token = @token_cache.token(@email, @password)
+
+					if token
+						Lita.logger.debug('Token successfully obtained from cache!')
+						return token
+					end
+
+					# Login
+					login_response = API.new.login(@email, @password)
+					# raise Discordrb::Errors::HTTPStatusError, login_response.code if login_response.code >= 400
+
+					# Parse response
+					login_response_object = JSON.parse(login_response)
+					# raise Discordrb::Errors::InvalidAuthenticationError unless login_response_object['token']
+					
+					Lita.logger.debug('Received token from Discord!')
+
+					# Cache the token
+					@token_cache.store_token(@email, @password, login_response_object['token'])
+
+					login_response_object['token']
+					# rescue Exception => e
+					# response_code = login_response.nil? ? 0 : login_response.code ######## mackmm145
+					# if login_attempts < 100 && (e.inspect.include?('No such host is known.') || response_code == 523)
+					# 	Lita.logger.debug("Login failed! Reattempting in 5 seconds. #{100 - login_attempts} attempts remaining.")
+					# 	Lita.logger.debug("Error was: #{e.inspect}")
+					# 	sleep 5
+					# 	login_attempts += 1
+					# 	retry
+					# else
+					# 	Lita.logger.debug("Login failed permanently after #{login_attempts + 1} attempts")
+					# 	# Apparently we get a 400 if the password or username is incorrect. In that case, tell the user
+					# 	Lita.logger.debug("Are you sure you're using the correct username and password?") if e.class == RestClient::BadRequest
+					# 	# raise $ERROR_INFO
+					# end
+				end
+
+				def logout
+					
+				end
+			end
+		end
+	end
+end

data/lib/lita/adapters/discord/token_cache.rb

@@ -0,0 +1,147 @@
+require 'base64'
+require 'json'
+require 'openssl'
+require 'lita/adapters/discord/api'
+
+module Lita
+	module Adapters
+		class Discord < Adapter
+			KEYLEN = 32
+			CACHE_PATH = Dir.pwd + '/.discord_token_cache.json'
+			class CachedToken
+				def initialize(data = nil)
+					if data
+						@verify_salt = Base64.decode64(data['verify_salt'])
+						@password_hash = Base64.decode64(data['password_hash'])
+						@encrypt_salt = Base64.decode64(data['encrypt_salt'])
+						@iv = Base64.decode64(data['iv'])
+						@encrypted_token = Base64.decode64(data['encrypted_token'])
+					else
+						generate_salts
+					end
+				end
+
+				# @return [Hash<Symbol => String>] the data representing the token and encryption data, all encrypted and base64-encoded
+			    def data
+					{
+						verify_salt: Base64.encode64(@verify_salt),
+						password_hash: Base64.encode64(@password_hash),
+						encrypt_salt: Base64.encode64(@encrypt_salt),
+						iv: Base64.encode64(@iv),
+						encrypted_token: Base64.encode64(@encrypted_token)
+					}
+				end
+
+				def verify_password(password)
+					hash_password(password) == @password_hash
+				end
+
+				def generate_verify_hash(password)
+					@password_hash = hash_password(password)
+				end
+
+				def obtain_key(password)
+					@key = OpenSSL::PKCS5.pbkdf2_hmac_sha1(password, @encrypt_salt, 300_000, KEYLEN)
+				end
+
+				def generate_salts
+					@verify_salt = OpenSSL::Random.random_bytes(KEYLEN)
+					@encrypt_salt = OpenSSL::Random.random_bytes(KEYLEN)
+				end
+
+				def decrypt_token(password)
+					key = obtain_key(password)
+					decipher = OpenSSL::Cipher::AES256.new(:CBC)
+					decipher.decrypt
+					decipher.key = key
+					decipher.iv = @iv
+					decipher.update(@encrypted_token) + decipher.final
+				end
+
+				def encrypt_token(password, token)
+					key = obtain_key(password)
+					cipher = OpenSSL::Cipher::AES256.new(:CBC)
+					cipher.encrypt
+					cipher.key = key
+					@iv = cipher.random_iv
+					@encrypted_token = cipher.update(token) + cipher.final
+				end
+
+				def test_token(token)
+					API.validate_token(token)
+				end
+
+				def hash_password(password)
+					OpenSSL::PKCS5.pbkdf2_hmac_sha1(password, @verify_salt, 300_000, KEYLEN)
+				end
+			end
+
+			class TokenCache
+
+				def initialize
+					if File.file? CACHE_PATH
+						@data = JSON.parse(File.read(CACHE_PATH))
+					else
+						Lita.logger.debug("Cache file #{CACHE_PATH} not found. Using empty cache")
+						@data = {}
+					end
+				rescue => e
+					Lita.logger.debug('Exception occurred while parsing token cache file:')
+					# I guess start here if stuff is super broken
+					Lita.logger.debug('Continuing with empty cache')
+					@data = {}
+				end
+
+				def token(email, password)
+					if @data[email]
+						begin
+							cached = CachedToken.new(@data[email])
+							if cached.verify_password(password)
+								token = cached.decrypt_token(password)
+								if token
+									begin
+										cached.test_token
+										token
+									rescue => e
+										fail_token('Token cached, verified and decrypted, but rejected by Discord', email)
+										sleep 1
+										nil
+									end
+								else
+									fail_token('Token cached and verified, but decryption failed', email)
+								end
+							else
+								fail_token('Token verification failed', email)
+							end
+						rescue => e
+							fail_token('Token cached but invalid', email)
+						end
+					else
+						fail_token('Token not cached at all')
+					end
+				end
+
+				def store_token(email, password, token)
+					cached = CachedToken.new
+					cached.generate_verify_hash(password)
+					cached.encrypt_token(password, token)
+					@data[email] = cached.data
+					write_cache
+				end
+
+				def write_cache
+					File.write(CACHE_PATH, @data.to_json)
+				end
+
+				private
+
+				def fail_token(msg, email = nil)
+					Lita.logger.warn("Token not retrieved from cache - #{msg}")
+					@data.delete(email) if email
+					nil
+				end
+				
+			end
+		end
+	end
+end

data/lib/lita/adapters/discord.rb

@@ -0,0 +1,39 @@
+require 'lita'
+require "lita/adapters/discord/client"
+
+module Lita
+	module Adapters
+		class Discord < Adapter
+			config :email, type: String, required: true
+			config :password, type: String, required: true
+
+			def initialize(robot)
+				super
+				## TIME TO REBUILD THIS SHIZZZZZZ
+				@client = Client.new(config.email, config.password)
+			end
+
+			def run
+				@client.connect  		
+			end
+
+			def shut_down
+				@client.disconnect
+			end
+
+			# def mention_format(user_id)
+			# 	"@<#{user_id}>"
+			# end
+
+			# Referenced https://github.com/litaio/lita-slack/blob/master/lib/lita/adapters/slack.rb#L46
+			# Trying to build out basic commands for max plugin compatibility 
+			# Update when api is built out
+			# def set_topic(target, topic)
+
+
+			# end
+
+			Lita.register_adapter(:discord, self)
+		end
+	end
+end

data/lib/lita-discord.rb

@@ -0,0 +1,7 @@
+require "lita"
+
+Lita.load_locales Dir[File.expand_path(
+  File.join("..", "..", "locales", "*.yml"), __FILE__
+)]
+
+require "lita/adapters/discord"

data/lita-discord.gemspec

@@ -0,0 +1,29 @@
+Gem::Specification.new do |spec|
+  spec.name          = "lita-discord"
+  spec.version       = "0.1.0"
+  spec.authors       = ["Kyle Boe"]
+  spec.email         = ["ksboe1@gmail.com"]
+  spec.description   = "A Discord adapter for Lita"
+  spec.summary       = "lita-discord is an adapter for Lita that allows you to use the robot to manage a Discord server."
+  spec.homepage      = "https://github.com/kyleboe/lita-discord"
+  spec.license       = "MIT"
+  spec.metadata      = { "lita_plugin_type" => "adapter" }
+
+  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_runtime_dependency "lita", ">= 4.7"
+
+  spec.add_dependency 'rest-client'
+  spec.add_dependency 'faye-websocket'
+  spec.add_dependency 'websocket-client-simple'
+  spec.add_dependency 'eventmachine'
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "pry-byebug"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rack-test"
+  spec.add_development_dependency "rspec", ">= 3.0.0"
+end

data/locales/en.yml

@@ -0,0 +1,4 @@
+en:
+  lita:
+    adapters:
+      discord:

data/spec/lita/adapters/discord_spec.rb

@@ -0,0 +1,4 @@
+require "spec_helper"
+
+describe Lita::Adapters::Discord, lita: true do
+end

data/spec/spec_helper.rb

@@ -0,0 +1,6 @@
+require "lita-discord"
+require "lita/rspec"
+
+# A compatibility mode is provided for older plugins upgrading from Lita 3. Since this plugin
+# was generated with Lita 4, the compatibility mode should be left disabled.
+Lita.version_3_compatibility_mode = false

metadata

@@ -0,0 +1,202 @@
+--- !ruby/object:Gem::Specification
+name: lita-discord
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Kyle Boe
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-04-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: lita
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.7'
+- !ruby/object:Gem::Dependency
+  name: rest-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: faye-websocket
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: websocket-client-simple
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: pry-byebug
+  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'
+- !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'
+- !ruby/object:Gem::Dependency
+  name: rack-test
+  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'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+description: A Discord adapter for Lita
+email:
+- ksboe1@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- Gemfile
+- README.md
+- Rakefile
+- lib/lita-discord.rb
+- lib/lita/adapters/discord.rb
+- lib/lita/adapters/discord/api.rb
+- lib/lita/adapters/discord/client.rb
+- lib/lita/adapters/discord/token_cache.rb
+- lita-discord.gemspec
+- locales/en.yml
+- spec/lita/adapters/discord_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/kyleboe/lita-discord
+licenses:
+- MIT
+metadata:
+  lita_plugin_type: adapter
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: lita-discord is an adapter for Lita that allows you to use the robot to manage
+  a Discord server.
+test_files:
+- spec/lita/adapters/discord_spec.rb
+- spec/spec_helper.rb