require 'net/http'
require 'uri'

module AhcHelper

  def render_ahc(template)
    if params[:format] == 'json'
      render json: @data
    else
      render text: Ahc.render(template, @data)
    end
  end

end

module AhcInheritedResourcesHelper
  include AhcHelper

  def respond_with(*resources, &block)
    controller = params[:controller].split('/').last
    action = params[:action]
    @data ||= {}
    @data['resources'] = collection
    render_ahc("#{controller}/#{action}")
  end

end

class AhcProxy

  def initialize(app)
    @app = app
  end

  def call(env)
    request = Rack::Request.new(env)
    path = request.fullpath
    if path.start_with?('/s/') || path.start_with?('/static/') || path.start_with?('/plovr/')
      begin
        response = Net::HTTP.get(URI.parse("http://#{Ahc.host || Ahc.default_host}:#{Ahc.port || Ahc.default_port}" + path))
        return [200, {}, [response]]
      rescue
        return [200, {}, ['Ahc server is not running']]
      end
    end
    @app.call(env)
  end

end