Rakefile

#!/usr/bin/env ruby

begin
require 'rspec/core/rake_task'

  RSpec::Core::RakeTask.new(:spec) {|t| }
  task :default => :spec
rescue LoadError
end

$stdout.sync = true

$LOAD_PATH.unshift File.expand_path(File.dirname(__FILE__))

require 'rake'
require 'ext'

desc "Pry console"
task :console do
  require 'pry'

  require 'models'
  require 'bart_api'
  require 'my_rollbar'

  Models.setup

  Pry.start
end

desc "Dump raw outage csv"
task :dump_csv do
  require 'models'
  Models.setup

  outages = Hash[
    Models::Outage.all.group_by(&:elevator_id)
      .map {|k,v| [Models::Elevator.get(k).name, v]}
  ]

  outages_to_print = outages.map do |k,v|
    length = v.map(&:length).reduce(:+).round(2)
    avg = (length / v.size).round(2)
    { :elevator => k,
      :number_outages => v.size,
      :length_outages => length,
      :avg_outage_length => avg
    }
  end.sort_by {|r| r[:length_outages]}
     .reverse # Put the most-out elevators at the top

  csv = CSV.generate do |c|
    c << ['Elevator', 'Number of Outages', 'Total hours out', 'Average outage length']
    outages_to_print.each {|o| c << o.values}
  end
  puts csv
end

task :worker do
  require 'worker'
  require 'models'
  require 'my_rollbar'

  Models.setup

  loop do
    begin
      puts "Start loop ..."
      Worker.run!
      puts "About to sleep ..."
    rescue StandardError => e
      puts "ERROR: #{e.class}, #{e.message}, #{e.backtrace}"
      Rollbar.error(e)
    end

    sleep 60
  end
end

desc "Get current state"
task :current do
  require'models'
  require 'my_rollbar'
  Models.setup

  puts "Worker count: #{Models::Metric.first(:name => "bartworker").try(:counter)}"
  puts "Current system count: #{Models::System.count}"
  puts "Current station count: #{Models::Station.count}"
  puts "Current elevator count: #{Models::Elevator.count}"
  puts "Current outages: #{Models::Outage.all_open.count}, #{Models::Outage.all_open.to_a.map(&:elevator).map(&:name).join(", ")}"
  puts "Unparseables: #{Models::Unparseable.count}"
  puts "Users: #{Models::User.count}"
end

namespace :dev do
  desc "Check that all required env keys are set"
  task :check_environment do
    require 'environment'
    Environment.check_env
  end

  desc "Generate a new session secret"
  task :session_secret do
    require 'environment'
    puts Environment.new_session_secret
  end

  desc "Dump seed json from database - this contains systems, stations, and elevators"
  task :dump_seed_json do
    require 'models'
    Models.setup

    puts JSON.pretty_generate(Models.seed_hash)
  end

  desc "Load from seed.json"
  task :load_seed_json, [:file] do |t, args|
    require 'models'
    Models.setup

    unless File.exists?(args[:file])
      warn "You must provide a json file like that generated by dev:dump_seed_hash"
      exit 1
    end

    json = File.read(args[:file])
    seed_hash = JSON.parse(json)
    Models.from_seed_hash(seed_hash)
  end
end

namespace :migrations do
  desc "Add Muni"
  task :m_001_add_muni do
    require 'models'
    require 'my_rollbar'
    Models.setup

    orig_adding_elevators = ENV['ADDING_ELEVATORS']
    ENV['ADDING_ELEVATORS'] = '1'

    if Models::System.first(:name => "SF Muni")
      puts "You already added the SF Muni system!"
      exit 1
    end

    bart = Models::System.first_or_create(:name => "BART")
    Models::Station.all.each do |s|
      next unless s.systems.empty?
      s.systems << bart
      s.save
    end
    muni = Models::System.first_or_create(:name => "SF Muni")

    ["Castro", "Church", "Civic Center", "Embarcadero", "Forest Hill",
"Montgomery", "Powell", "Van Ness", "West Portal"].each do |name|
      station = Models::Station.create(:name => "Muni: #{name}", :systems => [muni])
      elevator = Models::Elevator.create(:name => "Muni: #{name}", :station => station)
    end

    ENV['ADDING_ELEVATORS'] = orig_adding_elevators
  end

  desc "Remove ' Station' from elevator names"
  task :m_002_remove_station_from_elevator_names do
    require 'models'
    require 'my_rollbar'
    Models.setup

    Models::Elevator.all(:name.like => "Station").each do |e|
      e.name.gsub!(/ Station/, '')
      e.save
    end
  end
end

namespace :export do
  namespace :bart do
    desc "Raw outage data to csv"
    task :raw do
      require 'exports'
      require 'models'
      Models.setup

      Exports.bart_raw_outage_dump
    end

    desc "Elevator data"
    task :elevator do
      require 'exports'
      require 'models'
      Models.setup

      Exports.bart_elevator_dump
    end
  end
end