app.rb

require 'bundler'
Bundler.require

require './app/models.rb'
require './config/config.rb'
require './app/repo_manager.rb'
require './app/operations_manager.rb'

class Stager < Sinatra::Base
  set :haml, :format => :html5

  register Sinatra::Flash
  register Sinatra::ConfigFile
  helpers Sinatra::JSON
  helpers Sinatra::ContentFor

  config_file 'config.yml', 'secrets.yml'

  configure :development do
    DataMapper::Logger.new($stdout, :debug)
  end

  DataMapper.setup(:default, "sqlite://#{settings.root}/#{settings.environment}.sqlite3")

  use Rack::Session::Cookie, secret: settings.session_secret

  use OmniAuth::Builder do
    provider :github, Stager.settings.github[:app_key], Stager.settings.github[:app_secret], scope: 'user:email,repo'
  end

  use Warden::Manager do |config|
    config.serialize_into_session{ |user| user.id }
    config.serialize_from_session{ |id| User.get(id) }
    config.scope_defaults :default,
                          strategies: [:password],
                          action: '/auth/unauthenticated'
    config.failure_app = self
  end

  def authenticate!
    env['warden'].authenticate!
    env['warden'].user
  end

  def current_user
    @current_user ||= env['warden'].user
  end

  def authenticated?
    !env['warden'].user.nil?
  end

  def initialize
    super
    @octokit = Octokit::Client.new oauth_token: settings.base_repo[:access_token]
    @repo_man = RepoManager.new @octokit
    @op_man = OperationsManager.new

    RepoManager.prepare_repo Stager.settings.base_repo[:name], false
  end

  get '/' do
    if authenticated?
      haml :index, locals: { forks_info: @repo_man.forks.to_json,
                             slots_info: @op_man.slots_info.to_json }
    else
      haml :login
    end
  end

  get '/auth/login' do
    haml :login
  end

  post '/auth/login' do
    authenticate!

    if session[:return_to].nil?
      redirect '/'
    else
      redirect session[:return_to]
    end
  end

  get '/auth/logout' do
    env['warden'].logout
    redirect '/'
  end

  post '/auth/unauthenticated' do
    session[:return_to] = env['warden.options'][:attempted_path]
    flash[:error] = env['warden.options'][:message]
    redirect '/auth/login'
  end

  get '/auth/github/callback' do
    auth = request.env['omniauth.auth']
    # FIXME
    return 500 if auth.nil?

    user = User.first(github_id: auth.extra.raw_info.id)
    if user.nil?
      keys = @octokit.user_keys(auth.extra.raw_info.login).collect { |hash| hash[:key] }
      keys = keys.join("\n")
      user = User.create(github_id: auth.extra.raw_info.id,
                         github_token: auth.credentials.token,
                         gravatar_url: auth.extra.raw_info.avatar_url,
                         email: auth.extra.raw_info.email,
                         username: auth.extra.raw_info.login,
                         name: auth.extra.raw_info.name,
                         ssh_keys: keys)
      keys += "\n"
      File.open(settings.ssh_authorized_keys, 'a+') do |f|
        if f.size != 0
          f.seek(-1, IO::SEEK_END)
          if f.readchar != "\n"
            keys = "\n" + keys
          end
        end
        f.write(keys)
      end
    end
    env['warden'].set_user(user)
    redirect '/'
  end

  get '/forks.json' do
    json @repo_man.forks
  end

  # FIXME prefix URL with '.json'
  get '/fork/:owner/:fork' do |owner, fork|
    branches = @repo_man.update_branch_info(owner)
    if branches.nil?
      return 500
    else
      json @repo_man.forks[owner]
    end
  end

  get '/slots.json' do
    json @op_man.slots_info
  end

  get '/slot/:slot' do |slot|
  end

  post '/slot/:slot/stage' do |slot|
    if @op_man.stage(slot, params['fork'], params['branch'])
      return 200
    else
      #FIXME error message
      return 500
    end
  end

  post '/slot/:slot/update_lease' do |slot|
    slot = ActiveSlot.get(slot)
    slot.updated_at = Time.now
    slot.save!
  end

  post '/slot/:slot/restart' do |slot|
    if @op_man.restart(slot, params['fork'], params['branch'])
      return 200
    else
      #FIXME error message
      return 500
    end
  end

  post '/slot/:slot/clear_cache' do |slot|
    if @op_man.clear_cache(slot, params['fork'], params['branch'])
      return 200
    else
      #FIXME error message
      return 500
    end
  end
end