komainu 0.0.1

data/.gitignore

@@ -0,0 +1,5 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+*.swp

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in komainu.gemspec
+gemspec

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+
+require 'rake/testtask'
+Rake::TestTask.new(:spec) do |task|
+  task.test_files = FileList['spec/**/*_spec.rb']
+end

data/komainu.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "komainu/version"
+
+Gem::Specification.new do |s|
+  s.name        = "komainu"
+  s.version     = Komainu::VERSION
+  s.authors     = ["Andrew Vos"]
+  s.email       = ["andrew.vos@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{}
+  s.description = %q{}
+
+  s.rubyforge_project = "komainu"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_development_dependency "minitest"
+  # s.add_runtime_dependency "rest-client"
+end

data/lib/komainu.rb

@@ -0,0 +1,4 @@
+require "komainu/version"
+
+module Komainu
+end

data/lib/komainu/levenshtein.rb

@@ -0,0 +1,51 @@
+module Komainu
+  class Levenshtein
+    def initialize(words)
+      @trie = TrieNode.new
+      words.each do |word|
+        @trie.insert(word)
+      end
+    end
+
+    def best_match word
+      "Batman"
+    end
+
+    def search word, maximum_distance
+      current_row = (0..word.length).to_a
+      results = {}
+      @trie.children.keys.each do |letter|
+        search_recursive(@trie.children[letter], letter, word, current_row, results, maximum_distance)
+      end
+      results
+    end
+
+    def search_recursive node, letter, word, previous_row, results, maximum_distance
+      columns = word.length + 1
+      current_row = [previous_row.first + 1]
+
+      (1...columns).each do |column|
+        insert_cost = current_row[column - 1] + 1
+        delete_cost = previous_row[column] + 1
+
+        if word[column - 1] != letter
+          replace_cost = previous_row[column - 1] + 1
+        else
+          replace_cost = previous_row[column - 1]
+        end
+
+        current_row << [insert_cost, delete_cost, replace_cost].min
+      end
+
+      if current_row.last <= maximum_distance && node.word
+        results[node.word] = current_row.last
+      end
+
+      if current_row.min <= maximum_distance
+        node.children.keys.each do |letter|
+          search_recursive(node.children[letter], letter, word, current_row, results, maximum_distance)
+        end
+      end
+    end
+  end
+end

data/lib/komainu/search_result.rb

@@ -0,0 +1,10 @@
+module Komainu
+  class SearchResult
+    attr_accessor :name, :text
+
+    def initialize name, text
+      @name = name
+      @text = text
+    end
+  end
+end

data/lib/komainu/search_results.rb

@@ -0,0 +1,11 @@
+require "komainu/search_result"
+
+module Komainu
+  class SearchResults
+    attr_accessor :items, :suggestion
+
+    def initialize
+      @items = []
+    end
+  end
+end

data/lib/komainu/searches_text.rb

@@ -0,0 +1,50 @@
+require "komainu/levenshtein"
+require "komainu/search_results"
+
+module Komainu
+  class SearchesText
+    def initialize data_to_search
+      @data_to_search = data_to_search
+    end
+
+    def search query
+      results = SearchResults.new
+      @data_to_search.each do |name, text|
+        if text_includes_string(text, query)
+          results.items << SearchResult.new(name, text)
+        elsif text_includes_words_from_string(text, query)
+          results.items << SearchResult.new(name, text)
+        end
+      end
+
+      results.suggestion = calculate_suggestion(query)
+      results
+    end
+
+    private
+
+    def calculate_suggestion(query)
+      words = split_into_words(@data_to_search.values.join(" "))
+      levenshtein = Levenshtein.new(words)
+      suggestion = split_into_words(query).map do |word|
+        matches = levenshtein.search(word, 2)
+        matches.keys.first || word
+      end
+      suggestion.join(" ")
+    end
+
+    def split_into_words(string)
+      string.scan(/\b\w+\b/)
+    end
+
+    def text_includes_string text, string
+      text.downcase.include? string.downcase
+    end
+
+    def text_includes_words_from_string text, string
+      string.split(" ").any? do |word|
+        text_includes_string(text, word)
+      end
+    end
+  end
+end

data/lib/komainu/trie_node.rb

@@ -0,0 +1,20 @@
+module Komainu
+  class TrieNode
+    attr_accessor :word, :children
+
+    def initialize
+      @children = {}
+    end
+
+    def insert word
+      node = self
+      word.each_char do |letter|
+        unless node.children[letter]
+          node.children[letter] = TrieNode.new
+        end
+        node = node.children[letter]
+      end
+      node.word = word
+    end
+  end
+end

data/lib/komainu/version.rb

@@ -0,0 +1,3 @@
+module Komainu
+  VERSION = "0.0.1"
+end

data/spec/komainu/levenshtein_spec.rb

@@ -0,0 +1,16 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "..", "spec_helper"))
+require "komainu/levenshtein"
+
+module Komainu
+  describe Levenshtein do
+    it "finds words with a distance less the maximum distance" do
+      levenshtein = Levenshtein.new(["hello", "there", "good", "sirs"])
+      levenshtein.search("hell", 14).must_equal({
+        "hello" => 1,
+        "there" => 3,
+        "good"  => 4,
+        "sirs"  => 4
+      })
+    end
+  end
+end

data/spec/komainu/searches_text_spec.rb

@@ -0,0 +1,38 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "..", "spec_helper"))
+require "komainu/searches_text"
+
+module Komainu
+  describe SearchesText do
+    subject { SearchesText.new(data_to_search) }
+
+    let :data_to_search do
+      {
+        :item1 => "This is some text",
+        :item2 => "Batman has no parents"
+      }
+    end
+
+    it "finds exact matches" do
+      result = subject.search("Batman has")
+      result.items.size.must_equal 1
+      result.items.first.name.must_equal :item2
+    end
+
+    it "finds matches in any case" do
+      result = subject.search("BATMAN has")
+      result.items.size.must_equal 1
+      result.items.first.name.must_equal :item2
+    end
+
+    it "finds matches if the text is not in order" do
+      result = subject.search("has batman")
+      result.items.size.must_equal 1
+      result.items.first.name.must_equal :item2
+    end
+
+    it "suggests an alternate query" do
+      result = subject.search("btman")
+      result.suggestion.must_equal "Batman"
+    end
+  end
+end

data/spec/komainu/trie_node_spec.rb

@@ -0,0 +1,28 @@
+require File.expand_path(File.join(File.dirname(__FILE__), "..", "spec_helper"))
+require "komainu/trie_node"
+
+module Komainu
+  describe TrieNode do
+    it "stores a word" do
+      trie_node = TrieNode.new
+      trie_node.insert("abc")
+      trie_node.children["a"].wont_be_nil
+      trie_node.children["a"].children["b"].wont_be_nil
+      trie_node.children["a"].children["b"].children["c"].wont_be_nil
+    end
+
+    it "stores multiple words" do
+      trie_node = TrieNode.new
+      trie_node.insert("ac")
+      trie_node.insert("ad")
+      trie_node.children["a"].children["c"].wont_be_nil
+      trie_node.children["a"].children["d"].wont_be_nil
+    end
+
+    it "sets the last node word" do
+      trie_node = TrieNode.new
+      trie_node.insert("ac")
+      trie_node.children["a"].children["c"].word.must_equal "ac"
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,4 @@
+$: << File.expand_path(File.join(File.dirname(__FILE__), "..", "lib"))
+require "minitest/pride"
+require "minitest/autorun"
+require "minitest/spec"

metadata

@@ -0,0 +1,75 @@
+--- !ruby/object:Gem::Specification
+name: komainu
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Andrew Vos
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-11-22 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: &70327818908280 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70327818908280
+description: ''
+email:
+- andrew.vos@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- komainu.gemspec
+- lib/komainu.rb
+- lib/komainu/levenshtein.rb
+- lib/komainu/search_result.rb
+- lib/komainu/search_results.rb
+- lib/komainu/searches_text.rb
+- lib/komainu/trie_node.rb
+- lib/komainu/version.rb
+- spec/komainu/levenshtein_spec.rb
+- spec/komainu/searches_text_spec.rb
+- spec/komainu/trie_node_spec.rb
+- spec/spec_helper.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: komainu
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: ''
+test_files:
+- spec/komainu/levenshtein_spec.rb
+- spec/komainu/searches_text_spec.rb
+- spec/komainu/trie_node_spec.rb
+- spec/spec_helper.rb