require 'jruby/rack'

module JRuby
  module Rack
    class MerbServletHelper < ::JRuby::Rack::ServletHelper
      attr_reader :merb_environment, :merb_root

      def initialize(servlet_context = nil)
        super

        @merb_root = @servlet_context.getInitParameter('merb.root')
        @merb_root ||= '/WEB-INF'
        @merb_root = expand_root_path @merb_root

        @merb_environment = @servlet_context.getInitParameter('merb.environment')
        @merb_environment ||= 'production'
      end

      def load_merb
        load_merb_gems
        register_servlet_adapter
        load_servlet_sessions
        start_merb
      end

      def load_merb_gems
        logger.debug('Loading merb-core gem')
        require 'rubygems'
        require 'merb-core'
        require 'merb-core/rack'
      end

      def load_servlet_sessions
        logger.debug('Loading Merb servlet sessions')
        require 'jruby/rack/merb_servlet_session'
      end

      def register_servlet_adapter
        logger.debug('Registering Merb servlet adapter')
        Merb::Rack::Adapter.register %w{servlet}, :Servlet
      end

      def start_merb
        logger.debug('Starting Merb')

        Merb::Config.setup({
          :merb_root => @merb_root,
          :environment => @merb_environment,
          :adapter => 'servlet',
          :fork_for_class_load => false,
          :verbose => true
        })
        Merb.environment = Merb::Config[:environment]
        Merb.root = Merb::Config[:merb_root]

        logger.debug("Running bootloaders")
        Merb::BootLoader.run

        logger.debug("Starting Rack adapter")
        Merb.adapter.start(Merb::Config.to_hash)
      end
    end

    class MerbFactory
      def self.new
        MerbServletHelper.instance.load_merb
        ::Rack::Builder.new { run ::Merb::Config[:app] }.to_app
      end
    end
  end
end

# Merb likes to hardcode things into the Merb:: namespace.
module Merb
  module Rack
    class Servlet
      def self.start(opts={})
        ::Merb.logger.warn!("Using Java servlet adapter")
      end
    end
  end
end