From a3ea1ed922be4f9e11c83955612ae17d2d3f7b11 Mon Sep 17 00:00:00 2001
From: Christine Dodrill <xena@yolo-swag.com>
Date: Thu, 13 Aug 2015 20:51:03 -0700
Subject: [PATCH] implement remaining API calls

---
 ponyapi.nim    | 85 +++++++++++++++++++++++++++++++++++++++++++++-----
 ponyapi.nimble |  2 +-
 2 files changed, 79 insertions(+), 8 deletions(-)

diff --git a/ponyapi.nim b/ponyapi.nim
index 1eaefad..4294d9e 100644
--- a/ponyapi.nim
+++ b/ponyapi.nim
@@ -1,7 +1,10 @@
 import asyncdispatch
+import future
 import jester
 import json
+import random
 import strutils
+import times
 
 type
   Episode* = object of RootObj
@@ -46,11 +49,28 @@ proc `%`(ep: Episode): JsonNode =
 proc `%`(eps: seq[Episode]): JsonNode =
   var ret = newJArray()
 
-  for ep in episodes:
+  for ep in eps:
     add ret, %ep
 
   ret
 
+proc `%%`(ep: Episode): JsonNode =
+  %*
+    {
+      "episode": ep,
+    }
+
+proc `%%`(eps: seq[Episode]): JsonNode =
+  %*
+    {
+      "episodes": eps,
+    }
+
+let myHeaders = {
+  "Content-Type": "application/json",
+  "X-Powered-By": "Nim and Jester",
+}
+
 settings:
   port = 5000.Port
   bindAddr = "0.0.0.0"
@@ -60,12 +80,63 @@ routes:
     "http://github.com/Xe/PonyAPI".uri.redirect
 
   get "/all":
-    let headers = {"Content-Type": "application/json"}
-    var rep = %*
-      {
-        "episodes": episodes,
-      }
+    resp Http200, myHeaders, pretty(%%episodes, 4)
 
-    resp Http200, headers, pretty(rep, 4)
+  get "/newest":
+    var
+      now = getTime()
+      ep: Episode
+
+    for episode in episodes:
+      var then = times.fromSeconds(episode.air_date)
+
+      if now < then:
+        ep = episode
+        break
+
+    resp Http200, myHeaders, pretty(%%ep, 4)
+
+  get "/random":
+    resp Http200, myHeaders, pretty(%%episodes.randomChoice(), 4)
+
+  get "/season/@snumber":
+    var
+      season: int = @"snumber".parseInt
+      eps: seq[Episode] = lc[x | (x <- episodes, x.season == season), Episode]
+
+    if eps.len == 0:
+      resp Http404, myHeaders, $ %* { "error": "No episodes found" }
+    else:
+      resp Http200, myHeaders, pretty(%%eps, 4)
+
+  get "/season/@snumber/episode/@epnumber":
+    var
+      season: int = @"snumber".parseInt
+      enumber: int = @"epnumber".parseInt
+      ep: Episode
+
+    for episode in episodes:
+      if episode.season == season:
+        if episode.episode == enumber:
+          ep = episode
+
+    if ep.air_date == 0:
+      resp Http404, myHeaders, $ %* {"error": "No such episode"}
+    else:
+      resp Http200, myHeaders, pretty(%%ep, 4)
+
+  get "/search":
+    var
+      query = @"q".toLower
+      eps: seq[Episode]
+
+    for episode in episodes:
+      if episode.name.toLower.contains query:
+        eps = eps & episode
+
+    if eps.len == 0:
+      resp Http404, myHeaders, $ %* { "error": "No episodes found" }
+    else:
+      resp Http200, myHeaders, pretty(%%eps, 4)
 
 runForever()
diff --git a/ponyapi.nimble b/ponyapi.nimble
index c4ea072..66eb2d7 100644
--- a/ponyapi.nimble
+++ b/ponyapi.nimble
@@ -7,4 +7,4 @@ license       = "MIT"
 bin           = "ponyapi"
 
 [Deps]
-Requires: "nim >= 0.10.0, jester#head"
+Requires: "nim >= 0.10.0, jester#head, random#head"