module Piedesaint

# File lib/piedesaint.rb, line 74
def self.app ( options = {} )
  ::Rack::Builder.app do 
    use ::Rack::CommonLogger
    use ::Rack::ShowExceptions
    if ( !options[:key].nil? and !options[:key].empty? )
      use ::Rack::SslEnforcer, http_port: options[:http_port], https_port: options[:https_port]
    end
    use ::Rack::Deflater

    if options[:username].nil? or options[:username].empty?
      puts "Service without Basic Authentication"
    else
      use ::Rack::Auth::Basic, "Icecreamland" do |username, password|
        ( options[:username] == username ) && ( options[:password] == password )
      end
    end

    use ::Rack::Deflater

    if options[:metastore].nil? or options[:metastore].empty?
      puts "Service without cache"
    else
      use ::Rack::Cache, verbose: true,
          metastore: options[:metastore],
          entitystore: options[:entitystore],
          default_ttl: options[:freshness]
      puts "Service cache at #{options[:metastore]} / #{options[:entitystore]}"
    end

    map "/" do
      run Rack::DirectoriesTraversal.new(options)
    end
    puts "Serving tar'ed folders" if options[:tar]
  end
end

# File lib/piedesaint.rb, line 110
def self.execute ( options = {} )
  event = ::Puma::Events.new STDOUT, STDERR
  puma = ::Puma::Server.new self.app( options ), event

  ## For Puma 2.0.1 (there is a bug regarding SSL and at least Ruby 1.9.3)
  ## Puma server doesn't receive 'event' (that's left to Puma::Binder)
  #binder = ::Puma::Binder.new event
  #puma.binder = binder
  #ctx = ::Puma::MiniSSL::SSLContext.new
  #ctx.key = "./server.key"
  #ctx.cert = "./server.crt"
  #ctx.verify_mode = ::Puma::MiniSSL::VERIFY_NONE

  puma.add_tcp_listener options[:iface], options[:http_port]

  if ( !options[:key].nil? and !options[:key].empty? )
    ctx = ::OpenSSL::SSL::SSLContext.new
    ctx.key = OpenSSL::PKey::RSA.new File.read(options[:key])
    ctx.cert = OpenSSL::X509::Certificate.new File.read(options[:cert])
    ctx.verify_mode = ::OpenSSL::SSL::VERIFY_NONE
    puma.add_ssl_listener options[:iface], options[:https_port], ctx
  end

  puma.min_threads = 1
  puma.max_threads = 10

  begin
    Signal.trap "SIGUSR2" do
      @restart = true
      puma.begin_restart
    end
  rescue Exception
    p "*** Sorry signal SIGUSR2 not implemented, restart feature disabled!"
  end

  begin
    Signal.trap "SIGTERM" do
      p " - Gracefully stopping, waiting for requests to finish"
      puma.stop false
    end
  rescue Exception
    p "*** Sorry signal SIGTERM not implemented, gracefully stopping feature disabled!"
  end
  
  begin
    puts "Service starting at #{options[:http_port]} -> #{options[:https_port]}"
    puma.run.join
  rescue Interrupt
    graceful_stop puma
  end

  if @restart
    p "* Restarting..."
    @status.stop true if @status
    restart!
  end
end

# File lib/piedesaint.rb, line 169
def self.graceful_stop(puma)
  p " - Gracefully stopping, waiting for requests to finish"
  @status.stop(true) if @status
  puma.stop(true)
  p " - Goodbye!"
end

# File lib/piedesaint.rb, line 176
def self.tar(path)
  tar = StringIO.new
  Gem::Package::TarWriter.new(tar) do |tarwriter|
    Dir[File.join(path, "**/{*,.*}")].each do |file|
      mode = File.stat(file).mode
      relative_file = File.join(File.basename(path), file.sub(/^#{Regexp::escape path}\/?/, ''))

      if File.directory? file
        next if [ ".", ".."].include? File.basename(file)
        tarwriter.mkdir relative_file, mode
      else
        tarwriter.add_file(relative_file, mode)  do |filepart|
          File.open(file, "rb") { |f| filepart.write f.read }
        end
      end
    end
  end
  tar.rewind
  tar
end