Add count api

Author: Awea

web/controllers/cats_controller.ex

@@ -8,4 +8,12 @@ defmodule CatuumApi.CatsController do
     |> put_resp_content_type(cat.content_type)
     |> send_file(200, cat.path)
   end
+
+  def count(conn, %{"count" => count}) do
+    cats = CatuumApi.Cat.random(count)
+
+    conn
+    |> assign(:cats, cats)
+    |> render("index.json")
+  end
 end

web/models/cat.ex

@@ -20,6 +20,19 @@ defmodule CatuumApi.Cat do
     end
   end
 
+  def random(count) do
+    filenames = Repo.srandmember([@small_cats_key, count])
+
+
+    filenames |> Enum.map(fn(filename) -> 
+      ext = Path.extname(filename) |> String.replace(".", "")
+      %{
+        path: @uploads_path <> "/" <> filename,
+        content_type: "image/" <> ext
+      }
+    end)
+  end
+
   def random do
     filename = Repo.srandmember(@small_cats_key)
     ext      = Path.extname(filename) |> String.replace(".", "")

web/router.ex

@@ -2,7 +2,7 @@ defmodule CatuumApi.Router do
   use CatuumApi.Web, :router
 
   pipeline :browser do
-    plug :accepts, ["html"]
+    plug :accepts, ["html", "json"]
     plug :fetch_session
     plug :fetch_flash
     plug :protect_from_forgery
@@ -12,6 +12,7 @@ defmodule CatuumApi.Router do
   scope "/", CatuumApi do
     pipe_through :browser
 
+    get "/cats/:count", CatsController, :count
     get "/cats", CatsController, :index
   end
 end

web/views/cats_view.ex

@@ -0,0 +1,7 @@
+defmodule CatuumApi.CatsView do
+  use CatuumApi.Web, :view
+
+  def render("index.json", %{cats: cats}) do
+    cats
+  end
+end

web/views/layout_view.ex

@@ -0,0 +1,5 @@
+defmodule CatuumApi.LayoutView do
+  @moduledoc false
+
+  use CatuumApi.Web, :view
+end