module Trinidad
  module Lifecycle
    class Base
      include Trinidad::Tomcat::LifecycleListener
      attr_reader :webapp

      def initialize(webapp)
        @webapp = webapp
        @configured_logger = false
      end

      def lifecycleEvent(event)
        if Trinidad::Tomcat::Lifecycle::BEFORE_START_EVENT == event.type
          context = event.lifecycle
          configure_defaults(context)
        end
      end

      def configure_defaults(context)
        remove_defaults(context)
        configure_context_runtime(context)
        configure_logging
      end

      def configure_context_runtime(context)
        if @webapp.solo? && @webapp.threadsafe?
          context.servlet_context.set_attribute('jruby.runtime', JRuby.runtime)
        end
      end

      def remove_defaults(context)
        context.remove_welcome_file('index.jsp')
        context.remove_welcome_file('index.htm')
        context.remove_welcome_file('index.html')

        jsp_servlet = context.find_child('jsp')
        context.remove_child(jsp_servlet) if jsp_servlet

        context.remove_servlet_mapping('*.jspx')
        context.remove_servlet_mapping('*.jsp')

        context.process_tlds = false
        context.xml_validation = false
      end

      def configure_logging
        return if @configured_logger

        log_path = File.join(@webapp.work_dir, 'log', "#{@webapp.environment}.log")
        log_file = java.io.File.new(log_path)

        unless log_file.exists
          log_file.parent_file.mkdirs
          log_file.create_new_file
        end

        jlogging = java.util.logging

        log_handler = jlogging.FileHandler.new(log_path, true)
        logger = jlogging.Logger.get_logger("")

        log_level = @webapp.log
        unless %w{ALL CONFIG FINE FINER FINEST INFO OFF SEVERE WARNING}.include?(log_level)
          puts "Invalid log level #{log_level}, using default: INFO"
          log_level = 'INFO'
        end

        level = jlogging.Level.parse(log_level)

        logger.handlers.each do |handler|
          handler.level = level
        end

        logger.level = level

        log_handler.formatter = jlogging.SimpleFormatter.new
        logger.add_handler(log_handler)

        @configured_logger = true
      end
    end
  end
end