require 'singleton'
require 'fileutils'
require 'set'
require 'yaml'
require 'apipie/extractor/recorder'
require 'apipie/extractor/writer'
require 'apipie/extractor/collector'

class Apipie::Railtie
  if ENV["APIPIE_RECORD"]
    initializer 'apipie.extractor' do |app|
      ActiveSupport.on_load :action_controller do
        before_filter do |controller|
          Apipie::Extractor.call_recorder.analyse_controller(controller)
        end
      end
      app.middleware.use ::Apipie::Extractor::Recorder::Middleware
      ActionController::TestCase::Behavior.instance_eval do
        include Apipie::Extractor::Recorder::FunctionalTestRecording
      end
    end
  end
end

module Apipie

  module Extractor

    class << self

      def logger
        Rails.logger
      end

      def call_recorder
        Thread.current[:apipie_call_recorder] ||= Recorder.new
      end

      def call_finished
        @collector ||= Collector.new
        if record = call_recorder.record
          @collector.handle_record(record)
        end
      ensure
        Thread.current[:apipie_call_recorder] = nil
      end

      def write_docs
        Writer.new(@collector).write_docs if @collector
      end

      def write_examples
        Writer.new(@collector).write_examples if @collector
      end

      def apis_from_routes
        return @apis_from_routes if @apis_from_routes

        api_prefix = Apipie.configuration.api_base_url.sub(/\/$/,"")
        all_routes = Rails.application.routes.routes.map do |r|
          {
            :verb => case r.verb
                     when Regexp then r.verb.source[/\w+/]
                     else r.verb.to_s
                     end,
            :path => case
                     when r.path.respond_to?(:spec) then r.path.spec.to_s
                     else r.path.to_s
                     end,
            :controller => r.requirements[:controller],
            :action => r.requirements[:action]
          }


        end
        api_routes = all_routes.find_all do |r|
          r[:path].starts_with?(Apipie.configuration.api_base_url)
        end

        @apis_from_routes = Hash.new { |h, k| h[k] = [] }

        api_routes.each do |route|
          controller_path, action = route[:controller], route[:action]
          next unless controller_path && action

          controller = "#{controller_path}_controller".camelize

          path = if /^#{Regexp.escape(api_prefix)}(.*)$/ =~ route[:path]
                   $1.sub!(/\(\.:format\)$/,"")
                 else
                   nil
                 end

          if route[:verb].present?
            @apis_from_routes[[controller, action]] << {:method => route[:verb], :path => path}
          end
        end
        @apis_from_routes

        apis_from_docs = Apipie.method_descriptions.reduce({}) do |h, (method, desc)|
          apis = desc.apis.map do |api|
            {:method => api.http_method, :path => api.api_url, :desc => api.short_description}
          end
          h.update(method => apis)
        end

        @apis_from_routes.each do |(controller, action), new_apis|
          method_key = "#{controller.constantize.controller_name}##{action}"
          old_apis = apis_from_docs[method_key] || []
          new_apis.each do |new_api|
            new_api[:path].sub!(/\(\.:format\)$/,"")
            if old_api = old_apis.find { |api| api[:path] == "#{api_prefix}#{new_api[:path]}" }
              new_api[:desc] = old_api[:desc]
            end
          end
        end
        @apis_from_routes
      end

    end
  end
end

if ENV["APIPIE_RECORD"]
  Apipie.configuration.force_dsl = true
  at_exit do
    record_params, record_examples = false, false
    case ENV["APIPIE_RECORD"]
    when "params"   then record_params = true
    when "examples" then record_examples = true
    when "all"      then record_params = true, record_examples = true
    end

    if record_examples
      puts "Writing examples to a file"
      Apipie::Extractor.write_examples
    end
    if record_params
      puts "Updating auto-generated documentation"
      Apipie::Extractor.write_docs
    end
  end
end