require 'awestruct/util/exception_helper'
require 'awestruct/handlers/base_handler'

require 'tilt'

module Awestruct
  module Handlers

    class TiltMatcher
      # Returns the Tilt template class if a portion of the path is registered
      # to a Tilt template and the Tilt template can be loaded. Returns false
      # if no portions of the path are registered to a Tilt template or the
      # Tilt template cannot be loaded.
      def match(path)
        matcher = ::Tilt[File.basename(path)]
        if matcher.nil?
	  $LOG.warn(%(Copying #{path}))
          return false
        end

        # We have our own extra integration with Asciidoctor
        if matcher.name.include? 'Asciidoctor'
          return false
        end
        matcher
      end
    end

    class BaseTiltHandler < BaseHandler

      def initialize(site, delegate)
        super(site, delegate)
      end

      def source_file_name
        File.basename path
      end

      # HACK: if this is a double file name extension type
      # e.g. html.haml, xml.erb
      # move to default-site.yml?
      def double_extension?
        return true if input_extension =~ /haml|slim|erb|mustache/
      end

      def simple_name
        base = File.basename(source_file_name, File.extname(source_file_name))
        return File.basename(base, File.extname(base)) if double_extension?
        return base
      end

      def output_filename
        simple_name + output_extension
      end

      def input_extension
        File.extname(source_file_name)
      end

      def output_extension
        return File.extname(File.basename(source_file_name, File.extname(source_file_name))) if double_extension?

        template = ::Tilt[path]
        if !template.nil?
          mime = template.default_mime_type
          if !mime.nil?
            return '.js' if mime.eql? 'application/javascript'
            return '.html' if mime.eql? 'text/html'
            return '.xml' if mime.eql? 'text/xml'
            return '.css' if mime.eql? 'text/css'
            return '.html' # if all else falls trough
          end
        end
        return '.html'
      end

      def content_syntax
        # Check configuration for override, else convert extension to sym
        extension = input_extension[1..-1]
        if (!site[:content_syntax].nil? && !site[:content_syntax].empty?)
          syntax = site[:content_syntax][extension]
          return syntax.to_sym unless syntax.nil? or syntax.empty?
        end

        return extension.to_sym
      end

      def options
        opts = {}

        in_ext = input_extension[1..-1].to_sym
        out_ext = output_extension[1..-1].to_sym

        engine_name = ::Tilt[path].name.gsub(/(Awestruct)?(Tilt|:|Template)/i, '').downcase.to_sym

        # example: slim
        engine_opts = site[engine_name]
        unless engine_opts.nil?
          opts.merge! engine_opts
        end

        # example: slim|xml
        engine_opts_for_output = site["#{engine_name}|#{out_ext}"]
        unless engine_opts_for_output.nil?
          opts.merge! engine_opts_for_output
        end

        # config overrides for specific file extension if different from engine name
        unless engine_name == in_ext
          # example: adoc
          in_ext_opts = site[in_ext]
          unless in_ext_opts.nil?
            opts.merge! in_ext_opts
          end

          # example: adoc|xml
          in_ext_opts_for_output = site["#{in_ext}|#{out_ext}"]
          unless in_ext_opts_for_output.nil?
            opts.merge! in_ext_opts_for_output
          end
        end

        return opts
      end

      def rendered_content(context, with_layouts=true)
        $LOG.debug "invoking tilt for #{delegate.path} with_layouts = #{with_layouts}" if $LOG.debug?

        begin
          c = delegate.rendered_content(context, with_layouts)
          return "" if c.nil? or c.empty?
          template = ::Tilt::new(delegate.path.to_s, delegate.content_line_offset + 1, options) { |engine|
            c
          }
          return template.render(context)
        rescue LoadError => e
          error_page = context[:page]
          ExceptionHelper.log_message "Could not load template library required for rendering #{File.join site.dir, error_page.source_path}."
          ExceptionHelper.log_message "Please see #{File.join site.dir, error_page.output_path} for more information"
          return ExceptionHelper.html_error_report e, error_page.source_path
        rescue Exception => e
          error_page = context[:page]
          if error_page[:__is_layout] == true
	    ExceptionHelper.log_message "An error during rendering layout file #{File.join site.dir, error_page.source_path} occurred."
          else
	    ExceptionHelper.log_message "An error during rendering #{File.join site.dir, error_page.source_path} occurred."
          end
          ExceptionHelper.log_message "Please see #{File.join site.dir, error_page.output_path} for more information"
          return ExceptionHelper.html_error_report e, error_page.source_path
        end
      end
    end
  end
end