# frozen_string_literal: true

require 'browser/browser'
require 'wisper'
require 'oga'
require 'sinatra/base'
require 'sinatra/flash'
require 'sinatra/param'
require 'sinatra/respond_with'
require 'ditty/helpers/views'
require 'ditty/helpers/pundit'
require 'ditty/helpers/authentication'
require 'ditty/services/logger'
require 'active_support'
require 'active_support/inflector'
require 'rack/contrib'
require 'rack/csrf'

module Ditty
  class ApplicationController < Sinatra::Base
    include ActiveSupport::Inflector

    set :root, ENV['APP_ROOT'] || ::File.expand_path("#{::File.dirname(__FILE__)}/../../../")
    set :map_path, nil
    set :view_location, nil
    set :view_folder, nil
    set :base_path, nil
    set :model_class, nil
    set :raise_sinatra_param_exceptions, true
    set track_actions: false

    # The order here is important, since Wisper has a deprecated method respond_with method
    helpers Wisper::Publisher
    helpers Helpers::Pundit, Helpers::Views, Helpers::Authentication
    helpers Sinatra::Param

    register Sinatra::Flash, Sinatra::RespondWith

    use Rack::Csrf, raise: ENV['APP_ENV'] == 'development' unless ENV['APP_ENV'] == 'test'
    use Rack::JSONBodyParser
    use Rack::MethodOverride
    use Rack::NestedParams

    helpers do
      def logger
        ::Ditty::Services::Logger
      end

      def base_path
        settings.base_path || "#{settings.map_path}/#{dasherize(view_location)}"
      end

      def view_location
        return settings.view_location if settings.view_location

        loc = demodulize(settings.model_class || self.class)
        pluralize(underscore(loc.gsub(/Controller\Z/, '')))
      end

      def browser
        Browser.new(request.user_agent, accept_language: request.env['HTTP_ACCEPT_LANGUAGE'])
      end

      def config(name, default = '')
        ::Ditty::Services::Settings[name] || default
      end
    end

    def view_folders
      folders = ['./views']
      folders << settings.view_folder if settings.view_folder
      folders << ::Ditty::Ditty.view_folder
    end

    def find_template(views, name, engine, &block)
      # Backwards compatability
      if settings.view_folder.nil? && self.class.name.split('::').first != 'Ditty'
        return super(views, name, engine, &block)
      end

      view_folders.each do |folder|
        super(folder, name, engine, &block) # Root
      end
      raise ::Ditty::TemplateNotFoundError, "Could not find template `#{name}`"
    end

    configure :production do
      disable :show_exceptions
      set :dump_errors, false
    end

    configure :development do
      set :show_exceptions, :after_handler
    end

    configure :production, :development do
      disable :logging
      use Rack::CommonLogger, ::Ditty::Services::Logger
    end

    not_found do
      respond_to do |format|
        status 404
        format.html do
          haml :'404', locals: { title: '4 oh 4' }, layout: layout
        end
        format.json do
          if response.body.empty?
            json code: 404, errors: ['Not Found']
          else
            [404, response.body]
          end
        end
      end
    end

    error Helpers::NotAuthenticated, ::Pundit::NotAuthorizedError do
      # TODO: Check if this is logged / tracked
      if authenticated?
        respond_to do |format|
          status 403
          format.html do
            flash.now[:danger] = 'Cannot perform that action at the moment.'
            haml :'403', locals: { title: 'Forbidden' }, layout: layout
          end
          format.json do
            json code: 403, errors: ['Forbidden']
          end
        end
      else
        respond_to do |format|
          format.html do
            flash[:warning] = 'Please log in first.'
            redirect with_layout("#{settings.map_path}/auth/login")
          end
          format.json do
            status 401
            json code: 401, errors: ['Not Authenticated']
          end
        end
      end
    end

    error ::Sinatra::Param::InvalidParameterError do
      respond_to do |format|
        status 400
        format.html do
          flash.now[:danger] = env['sinatra.error'].message
          haml :'400', locals: { title: '4 oh oh' }, layout: layout
        end
        format.json do
          json code: 400, errors: { env['sinatra.error'].param => env['sinatra.error'].message },
               full_errors: [env['sinatra.error'].message]
        end
      end
    end

    error ::Sequel::NoMatchingRow do
      respond_to do |format|
        status 404
        format.html do
          haml :'404', locals: { title: '4 oh 4' }, layout: layout
        end
        format.json do
          if response.body.empty?
            json code: 404, errors: ['Not Found']
          else
            [404, response.body]
          end
        end
      end
    end

    error ::Sequel::ValidationFailed do
      respond_to do |format|
        entity = env['sinatra.error'].model
        errors = env['sinatra.error'].errors
        status 400
        format.html do
          action = entity.id ? :edit : :new
          haml :"#{view_location}/#{action}", locals: { entity: entity, title: heading(action) }, layout: layout
        end
        format.json do
          json code: 400, errors: errors, full_errors: errors.full_messages
        end
      end
    end

    error ::Sequel::ForeignKeyConstraintViolation do
      error = env['sinatra.error']
      broadcast(:application_error, error)
      logger.error error
      respond_to do |format|
        status 400
        format.html do
          haml :error, locals: { title: 'Something went wrong', error: error }, layout: layout
        end
        format.json do
          json code: 400, errors: ['Invalid Relation Specified']
        end
      end
    end

    error ::Ditty::TemplateNotFoundError do
      # TODO: Display a better error message
      error = env['sinatra.error']
      broadcast(:application_error, error)
      logger.error error
      respond_to do |format|
        status 500
        format.html do
          haml :error, locals: { title: 'Template not found', error: error }, layout: layout
        end
      end
    end

    error do
      error = env['sinatra.error']
      broadcast(:application_error, error)
      logger.error error
      respond_to do |format|
        status 500
        format.html do
          haml :error, locals: { title: 'Something went wrong', error: error }, layout: layout
        end
        format.json do
          json code: 500, errors: ['Something went wrong']
        end
      end
    end

    before(/.*/) do
      logger.info "Running with #{self.class} - #{request.path_info}"
      if /.*\.json\Z/.match?(request.path)
        content_type :json
        request.path_info = request.path_info.gsub(/.json$/, '')
      elsif /.*\.csv\Z/.match?(request.path)
        content_type :csv
        request.path_info = request.path_info.gsub(/.csv$/, '')
      elsif request.env['ACCEPT']
        content_type request.env['ACCEPT']
      elsif request.accept.count.eql?(1) && request.accept.first.to_s.eql?('*/*')
        content_type(:json)
      end
    end

    after do
      return if params[:layout].nil?
      return unless response.body.respond_to?(:map)

      begin
        orig = response.body
        response.body = response.body.map do |resp|
          document = Oga.parse_html(resp)
          document.css('a').each do |elm|
            unless (href = elm.get('href')).nil?
              elm.set 'href', with_layout(href)
            end
          end
          document.to_xml
        end
      rescue StandardError => _e
        orig
      end
    end
  end
end