httparrot/lib/httparrot.ex

defmodule HTTParrot do
  use Application
  use Supervisor

  def start_link(arg) do
    Supervisor.start_link(__MODULE__, arg)
  end

  def init(_) do
    supervise([], strategy: :simple_one_for_one)
  end

  def start(_type, _args) do
    dispatch = :cowboy_router.compile([
      {:_, [ {'/', :cowboy_static, {:priv_file, :httparrot, "index.html"}},
             {'/ip', HTTParrot.IPHandler, []},
             {'/user-agent', HTTParrot.UserAgentHandler, []},
             {'/headers', HTTParrot.HeadersHandler, []},
             {'/get', HTTParrot.GetHandler, []},
             {'/post', HTTParrot.PHandler, []},
             {'/put', HTTParrot.PHandler, []},
             {'/patch', HTTParrot.PHandler, []},
             {'/delete', HTTParrot.DeleteHandler, []},
             {'/deflate', HTTParrot.DeflateHandler, []},
             {'/gzip', HTTParrot.GzipHandler, []},
             {'/status/:code', [code: :int], HTTParrot.StatusCodeHandler, []},
             {'/redirect/:n', HTTParrot.RedirectHandler, []},
             {'/redirect-to', HTTParrot.RedirectToHandler, []},
             {'/relative-redirect/:n', HTTParrot.RelativeRedirectHandler, []},
             {'/cookies', HTTParrot.CookiesHandler, []},
             {'/cookies/set[/:name/:value]', HTTParrot.SetCookiesHandler, []},
             {'/cookies/delete', HTTParrot.DeleteCookiesHandler, []},
             {'/basic-auth/:user/:passwd', HTTParrot.BasicAuthHandler, []},
             {'/hidden-basic-auth/:user/:passwd', HTTParrot.HiddenBasicAuthHandler, []},
             {'/stream/:n', HTTParrot.StreamHandler, []},
             {'/stream-bytes/:n', HTTParrot.StreamBytesHandler, []},
             {'/delay/:n', HTTParrot.DelayedHandler, []},
             {'/html', :cowboy_static, {:priv_file, :httparrot, "html.html"}},
             {'/deny', HTTParrot.DenyHandler, []},
             {'/cache', HTTParrot.CacheHandler, []},
             {'/robots.txt', HTTParrot.RobotsHandler, []},
             {'/base64/:value', HTTParrot.Base64Handler, []},
             {'/image', HTTParrot.ImageHandler, []},
             {'/websocket', HTTParrot.WebsocketHandler, []},
             {'/response-headers', HTTParrot.ResponseHeadersHandler, []},
             {'/store/:key', HTTParrot.StoreRequestHandler, []},
             {'/retrieve/:key', HTTParrot.RetrieveRequestHandler, []} ] }
    ])

    {:ok, http_port} = Application.fetch_env(:httparrot, :http_port)
    IO.puts "Starting HTTParrot on port #{http_port}"
    {:ok, _} = :cowboy.start_http(:http, 100,
      [port: http_port],
      [env: [dispatch: dispatch],
      onresponse: &prettify_json/4])

    if Application.get_env(:httparrot, :ssl, false) do
      {:ok, https_port} = Application.fetch_env(:httparrot, :https_port)
      priv_dir = :code.priv_dir(:httparrot)
      IO.puts "Starting HTTParrot on port #{https_port} (SSL)"
      {:ok, _} = :cowboy.start_https(:https, 100,
        [port: https_port, cacertfile: priv_dir ++ '/ssl/server-ca.crt',
         certfile: priv_dir ++ '/ssl/server.crt', keyfile: priv_dir ++ '/ssl/server.key'],
        [env: [dispatch: dispatch], onresponse: &prettify_json/4])
    end

    if unix_socket_supported?() && Application.get_env(:httparrot, :unix_socket, false) do
      {:ok, socket_path} = Application.fetch_env(:httparrot, :socket_path)
      if File.exists?(socket_path), do: File.rm(socket_path)
      IO.puts "Starting HTTParrot on unix socket #{socket_path}"
      {:ok, _} = :cowboy.start_http(:http_unix, 100,
        [port: 0, ip: {:local, socket_path}],
        [env: [dispatch: dispatch], onresponse: &prettify_json/4])
    end

    Supervisor.start_link([
        worker(ConCache, [[
            ttl_check: :timer.minutes(5),
            ttl: :timer.hours(12)
          ], [name: :requests_registry]])
      ], strategy: :one_for_one)
  end

  def stop(_State), do: :ok

  def prettify_json(status, headers, body, req) do
    if JSX.is_json? body do
      body = JSX.prettify!(body)
      headers = List.keystore(headers, "content-length", 0, {"content-length", Integer.to_charlist(String.length(body))})
      {:ok, req} = :cowboy_req.reply(status, headers, body, req)
      req
    else
      req
    end
  end

  def unix_socket_supported? do
    case {:os.type, Integer.parse("#{:erlang.system_info(:otp_release)}")} do
      {{:unix, _}, {n, _}} when n >= 19 -> true
      _ -> false
    end
  end
end