Skip to content

Commit

Permalink
Merge pull request #2 from tjgillies/master
Browse files Browse the repository at this point in the history
Refactoring
  • Loading branch information
onewheelskyward committed Oct 1, 2017
2 parents c6682cd + a751917 commit 138f46f
Showing 1 changed file with 10 additions and 15 deletions.
25 changes: 10 additions & 15 deletions lib/lita/handlers/onewheel_snowcrash.rb
Original file line number Diff line number Diff line change
Expand Up @@ -16,11 +16,9 @@ class OnewheelSnowcrash < Handler
:generate_words_by_number,
command: true

@@markov = MarkyMarkov::Dictionary.new('dict/snowcrash')

on(:connected) do |load_dict|
get_markov
end
Markov = MarkyMarkov::Dictionary.new('dict/snowcrash')
Markov.parse_file(File.expand_path('../../dict/snowcrash.txt',
File.dirname(__FILE__)))

def generate_random_sentence(response)
response.reply return_sentence_chain(1)
Expand All @@ -38,22 +36,19 @@ def generate_words_by_number(response)
response.reply return_word_chain(response.matches[0][0])
end

def get_markov
@@markov.parse_file(File.expand_path('../../dict/snowcrash.txt', File.dirname(__FILE__)))
end

def return_word_chain(number)
Lita.logger.info "Generating #{number} words."
words = @@markov.generate_n_words(number.to_i)
Lita.logger.info words
words
log_and_return Markov.generate_n_words(number.to_i)
end

def return_sentence_chain(number)
Lita.logger.info "Generating #{number} sentences."
sentences = @@markov.generate_n_sentences(number.to_i)
Lita.logger.info sentences
sentences
log_and_return Markov.generate_n_sentences(number.to_i)
end

def log_and_return(message)
Lita.logger.info message
message
end

Lita.register_handler(self)
Expand Down

0 comments on commit 138f46f

Please sign in to comment.