lita-markov 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 80906c8495bb1dcbbc33b970047390e606b429c8
+  data.tar.gz: 85f84a17b45289e6c86334318322eebba6724b8f
+SHA512:
+  metadata.gz: f7c18adeba628b7e6cc48a22922fd745595104ce6846c919b773ab475e7c22b4d2e852d747055adadaec74b0dc881cab36254b305ab4c2df179c6a77e1b118e3
+  data.tar.gz: e4fce6fb0f3d70d04d2153fb202d1740c223abadc475a18a6629ba5080f211219dc58a0817bd36d58d6529801b8da7a2eaeb700585b7bb69d7ff1c20daf45145

data/.gitignore

@@ -0,0 +1,2 @@
+Gemfile.lock
+tmp

data/Gemfile

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

data/LICENSE

@@ -0,0 +1,27 @@
+Copyright (c) 2015 Dirk Gadsden
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+
+* Neither the name of Roost nor the names of its contributors may be used to
+  endorse or promote products derived from this software without specific
+  prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

data/README.md

@@ -0,0 +1,26 @@
+# Markov chains for Lita
+
+Listens to your public chat rooms and generates Markov chain databases
+for each user.
+
+## Installation
+
+Add `lita-markov` to your Lita instance's Gemfile:
+
+``` ruby
+gem 'lita-markov'
+```
+
+## Usage
+
+The bot will automatically ingest all messages into the Redis-backed Markov
+chain database. You can then query the bot for a generated chain:
+
+```
+user> mybot markov dirk
+mybot> I love cookies!
+```
+
+## License
+
+Licensed under the 3-clause BSD license. See [LICENSE](LICENSE) for details.

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-markov.rb

@@ -0,0 +1,13 @@
+require "lita"
+
+Lita.load_locales Dir[File.expand_path(
+  File.join("..", "..", "locales", "*.yml"), __FILE__
+)]
+
+require 'marky_markov/persistent_json_dictionary'
+require 'lita/handlers/markov'
+
+# Lita::Handlers::Markov.template_root File.expand_path(
+#   File.join("..", "..", "templates"),
+#  __FILE__
+# )

data/lib/lita/handlers/markov.rb

@@ -0,0 +1,64 @@
+module Lita::Handlers
+  class Markov < Lita::Handler
+    Dictionary = MarkyMarkov::PersistentJSONDictionary
+
+    REDIS_KEY_PREFIX = 'lita-markov:'
+    
+    route(/.+/, :ingest, command: false)
+
+    route(/markov (.+)/, :generate, command: true, help: {
+      'markov USER' => 'Generate a markov chain from the given user.'
+    })
+
+    def ingest(chat)
+      # Don't ingest messages addressed to ourselves
+      return if chat.command?
+
+      message = chat.matches[0].strip
+
+      # Get the mention name (ie. 'dirk') of the user
+      name       = chat.user.mention_name
+      dictionary = dictionary_for_user name
+
+      # Passing `false` to indicate it's a string and not a file name
+      dictionary.parse_source message, false
+
+      save_dictionary name, dictionary
+    end
+
+    def generate(chat)
+      name = chat.matches[0][0].strip
+
+      dictionary = dictionary_for_user name
+      generator  = MarkovSentenceGenerator.new dictionary
+
+      begin
+        sentence = generator.generate_sentence 1
+
+        chat.reply sentence
+      rescue EmptyDictionaryError
+        chat.reply "Looks like #{name} hasn't said anything!"
+      end
+    end
+
+    def save_dictionary(name, dictionary)
+      redis.set key_for_user(name), dictionary.to_json
+    end
+
+    def dictionary_for_user(name)
+      key        = key_for_user name
+      dictionary = Dictionary.new name
+      json       = redis.get key
+
+      dictionary.load_json(json) if json
+
+      dictionary
+    end
+
+    def key_for_user(name)
+      REDIS_KEY_PREFIX+name.downcase
+    end
+
+    Lita.register_handler self
+  end
+end

data/lib/marky_markov/persistent_json_dictionary.rb

@@ -0,0 +1,34 @@
+require 'marky_markov'
+require 'oj'
+
+module MarkyMarkov
+  class PersistentJSONDictionary < ::PersistentDictionary
+    def initialize(*args)
+      super(*args)
+
+      @dictionary        = {}
+      @capitalized_words = []
+    end
+
+    # No-op instead of reading from the filesystem
+    def open_dictionary
+      nil
+    end
+
+    def load_json(json)
+      data = Oj.load json
+
+      @depth             = data['depth']
+      @dictionary        = data['dictionary']
+      @capitalized_words = data['capitalized_words']
+    end
+
+    def to_json
+      Oj.dump(
+        'depth'             => @depth,
+        'dictionary'        => @dictionary,
+        'capitalized_words' => @capitalized_words
+      )
+    end
+  end
+end

data/lita-markov.gemspec

@@ -0,0 +1,25 @@
+Gem::Specification.new do |spec|
+  spec.name          = "lita-markov"
+  spec.version       = "0.0.1"
+  spec.authors       = ["Dirk Gadsden"]
+  spec.email         = ["dirk@dirk.to"]
+  spec.description   = "Markov chains for Lita."
+  spec.summary       = spec.description
+  spec.homepage      = "http://github.com/dirk/lita-markov"
+  spec.metadata      = { "lita_plugin_type" => "handler" }
+
+  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.6"
+  spec.add_runtime_dependency "marky_markov", "~> 0.3.5"
+  spec.add_runtime_dependency "oj", "~> 2.13.1"
+
+  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:
+    handlers:
+      whoami:

data/spec/lita/handlers/markov_spec.rb

@@ -0,0 +1,47 @@
+require 'spec_helper'
+require 'pry'
+
+describe Lita::Handlers::Markov, lita_handler: true do
+  before(:each) do
+    Lita.redis.flushall
+  end
+
+  it "won't call #ingest for non-command messages" do
+    expect(subject).to_not receive(:ingest)
+
+    send_message "#{robot.name} foo"
+    send_command 'bar'
+  end
+
+  it "will ingest a message into that person's dictionary" do
+    send_message 'hello markov world'
+    send_message 'hello markov planet'
+
+    dictionary = subject.dictionary_for_user user.mention_name
+
+    # Check that the messages made it into the dictionary
+    expect(dictionary.dictionary[['hello', 'markov']]).to eql ['world', 'planet']
+  end
+
+  it 'will build a sentence' do
+    send_message 'I love cookies!'
+    send_message 'I love pancakes!'
+
+    send_command "#{robot.name} markov #{user.mention_name}"
+
+    expect(replies.count).to eql 1
+
+    possible_replies = [
+      'I love cookies!',
+      'I love pancakes!'
+    ]
+    expect(possible_replies).to include replies[0]
+  end
+
+  it "will complain if the user hasn't said anything" do
+    send_command "#{robot.name} markov #{user.mention_name}"
+
+    expect(replies.count).to eql 1
+    expect(replies[0]).to eql "Looks like Test User hasn't said anything!"
+  end
+end

data/spec/marky_markov/persistent_json_dictionary_spec.rb

@@ -0,0 +1,28 @@
+require 'spec_helper'
+
+describe MarkyMarkov::PersistentJSONDictionary do
+  subject { MarkyMarkov::PersistentJSONDictionary.new 'whoa' }
+  
+  it 'initializes with sensible defaults' do
+    expect(subject.dictionary).to eq({})
+  end
+
+  it 'saves a dictionary' do
+    subject.add_word ['a', 'b'], 'c'
+
+    json = subject.to_json
+
+    expect(json).to eql '{"depth":2,"dictionary":{"^#1":[["a","b"],["c"]]},"capitalized_words":[]}'
+  end
+
+  it 'saves and loads a dictionary' do
+    subject.add_word ['a', 'b'], 'c'
+
+    json = subject.to_json
+
+    new_dictionary = MarkyMarkov::PersistentJSONDictionary.new 'whoa-another-one'
+    new_dictionary.load_json json
+
+    expect(new_dictionary.dictionary).to eql(['a', 'b'] => ['c'])
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,6 @@
+require 'lita-markov'
+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,173 @@
+--- !ruby/object:Gem::Specification
+name: lita-markov
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Dirk Gadsden
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-11-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: lita
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.6'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.6'
+- !ruby/object:Gem::Dependency
+  name: marky_markov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.3.5
+- !ruby/object:Gem::Dependency
+  name: oj
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.13.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.13.1
+- !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: Markov chains for Lita.
+email:
+- dirk@dirk.to
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- lib/lita-markov.rb
+- lib/lita/handlers/markov.rb
+- lib/marky_markov/persistent_json_dictionary.rb
+- lita-markov.gemspec
+- locales/en.yml
+- spec/lita/handlers/markov_spec.rb
+- spec/marky_markov/persistent_json_dictionary_spec.rb
+- spec/spec_helper.rb
+- templates/.gitkeep
+homepage: http://github.com/dirk/lita-markov
+licenses: []
+metadata:
+  lita_plugin_type: handler
+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.4.5.1
+signing_key: 
+specification_version: 4
+summary: Markov chains for Lita.
+test_files:
+- spec/lita/handlers/markov_spec.rb
+- spec/marky_markov/persistent_json_dictionary_spec.rb
+- spec/spec_helper.rb