Rakefile

require './app'

desc 'Empties the entire database'
task :reset do
  start_time = Time.now
  puts 'Emptying redis...'
  $redis.flushall
  puts "Completed in #{Time.now - start_time} seconds"
end

desc 'Ingest the history of one or more Slack channels into the database'
task :ingest do
  if ENV['CHANNELS'].nil?
    puts 'You need to specify the name of the channels you wish to import, e.g. rake ingest CHANNELS=\'#random\''
  else
    options = {}
    start_time = Time.now
    channels = ENV['CHANNELS'].split(',')
    options[:user_id] = get_slack_user_id(ENV['USERNAME']) unless ENV['USERNAME'].nil?
    options[:oldest] = (start_time - (60 * 60 * 24 * ENV['DAYS'].to_i)).to_i unless ENV['DAYS'].nil?
    $redis.pipelined do
      channels.each do |channel|
        puts "\nImporting channel #{channel.strip} to #{ENV['RACK_ENV']} (this will take a while)\n\n"
        channel_id = get_channel_id(channel.strip)
        import_history(channel_id, options)
      end
    end
    puts "Completed in #{Time.now - start_time} seconds"
  end
end

desc 'Empties the database and reingests the channel or channels'
task :reingest => ['reset', 'ingest']

task :topic do
  if ENV['CHANNEL'].nil?
    puts 'You need to specify the name of the channel you wish to set the topic for, e.g. rake topic CHANNEL=\'#random\''
  else
    start_time = Time.now
    channel_id = get_channel_id(ENV['CHANNEL'].strip)
    markov_topic(channel_id)
    puts "Completed in #{Time.now - start_time} seconds"
  end
end

task :chat do
  if ENV['CHANNEL'].nil?
    puts 'You need to specify the name of the channel you wish to chat in, e.g. rake chat CHANNEL=\'#random\''
  else
    start_time = Time.now
    channel_id = get_channel_id(ENV['CHANNEL'].strip)
    markov_chat(channel_id)
    puts "Completed in #{Time.now - start_time} seconds"
  end
end