module SidekiqSchedulerManager::Web

Constants

VIEW_PATH

Public Class Methods

registered(app) click to toggle source
# File lib/sidekiq-scheduler-manager/web.rb, line 11
def self.registered(app)
  app.get '/schedulers' do
    @schedulers = Sidekiq.get_all_schedules

    # erb File.read(File.join(VIEW_PATH, 'index.html.erb'))
    erb File.read(File.join(VIEW_PATH, 'scheduler.html.erb'))
  end

  app.get '/scheduler/:id' do
    @scheduler = Sidekiq.get_schedule(params[:id])
    erb File.read(File.join(VIEW_PATH, 'scheduler_edit.html.erb'))
  end

  app.put '/scheduler/:id' do
    job_name = params[:id]

    unless job_name.to_s.strip.empty?
      @scheduler = Sidekiq.get_schedule(params[:id])
      data = {
          :class => params[:scheduler]['class'],
          :queue => params[:scheduler]['queue'],
          :description => params[:scheduler]['description'],
          :args => params[:scheduler]['args'],
      }

      # args
      if not params[:scheduler]['args'].to_s.strip.empty?
        args = params[:scheduler]['args'].to_s.strip
        if args.include?('[')
          begin
            data[:args] = JSON.parse(args)
          rescue JSON::ParserError => e
            redirect "#{root_path}scheduler/#{params[:id]}"
          end
        else
          data[:args] = args
        end
      end

      if not params[:scheduler]['cron'].to_s.strip.empty?
        data[:cron] = params[:scheduler]['cron']
      elsif not params[:scheduler]['every'].to_s.strip.empty?
        every = params[:scheduler]['every'].to_s.strip
        if every.include?('[')
          begin
            data[:every] = JSON.parse(every)
          rescue JSON::ParserError => e
            redirect "#{root_path}scheduler/#{params[:id]}"
          end
        else
          data[:every] = every
        end
      end

      Sidekiq.set_schedule(job_name, data)
    end

    redirect "#{root_path}schedulers"
  end

  app.get '/schedulers/new' do
    erb File.read(File.join(VIEW_PATH, 'scheduler_new.html.erb'))
  end

  app.post '/schedulers' do
    job_name = params[:scheduler]['name']

    unless job_name.to_s.strip.empty?
      data = {
          :class => params[:scheduler]['class'],
          :queue => params[:scheduler]['queue'],
          :description => params[:scheduler]['description'],
          :args => params[:scheduler]['args'],
      }

      # args
      if not params[:scheduler]['args'].to_s.strip.empty?
        args = params[:scheduler]['args'].to_s.strip
        if args.include?('[')
          begin
            data[:args] = JSON.parse(args)
          rescue JSON::ParserError => e
            redirect "#{root_path}scheduler/new"
          end
        else
          data[:args] = args
        end
      end

      if not params[:scheduler]['cron'].to_s.strip.empty?
        data[:cron] = params[:scheduler]['cron']
      elsif not params[:scheduler]['every'].to_s.strip.empty?
        every = params[:scheduler]['every'].to_s.strip
        if every.include?('[')
          begin
            data[:every] = JSON.parse(every)
          rescue JSON::ParserError => e
            redirect "#{root_path}schedulers/new"
          end
        else
          data[:every] = every
        end
      end

      Sidekiq.set_schedule(job_name, data)
    end

    redirect "#{root_path}schedulers"
  end

  app.delete '/scheduler/:id' do
    # Sidekiq.set_schedule(params[:id], { noop: nil })
    schedules = Sidekiq.get_all_schedules.reject { |k, v| k == params[:id] }
    Sidekiq.schedule = schedules
    redirect "#{root_path}schedulers"
  end

  app.get '/scheduler/:id/reject' do
    # Sidekiq.set_schedule(params[:id], { noop: nil })
    # schedules = Sidekiq.get_all_schedules.reject { |k, v| k == params[:id] }
    # Sidekiq.schedule = schedules

    Sidekiq.remove_schedule(params[:id])
    redirect "#{root_path}schedulers"
  end

  app.get '/scheduler/:id/enqueue' do
    schedule = Sidekiq.get_schedule(params[:id])
    Sidekiq::Scheduler.enqueue_job(schedule)
    redirect "#{root_path}schedulers"
  end

  app.get '/schedulers/export' do
    result = Array.new
    schedulers = Sidekiq.get_all_schedules
    schedulers.each do |scheduler|
      result.push(scheduler)
    end
    [200, { "Content-Type" => "application/octet-stream", "Content-Disposition" => "attachment", "filename" => "sidekiq_schedule.json" }, [JSON.pretty_generate(result)]]
  end

  app.post '/schedulers/import' do
    # puts params
    tmpfile = params[:file][:tempfile]
    file = tmpfile.read

    data_hash = JSON.parse(file)
    data_hash.each do |job_name, data|
      unless job_name.to_s.strip.empty?
        Sidekiq.set_schedule(job_name, data)
      end
    end

    redirect "#{root_path}schedulers"
  end

end