Make record exists before call methods on it.

Author: st0012

server.gb

@@ -57,28 +57,47 @@ server.put('/items/{id:[0-9]+}') do |req, res|
   if params.nil? || params[:title].nil?
     res.status = 400
   else
-    title = params[:title]
-    ListItem.find(req.params[:id]).update_title(title)
-    res.status = 200
+    record = ListItem.find(req.params[:id])
+    if record
+      record.update_title(params[:title])
+      res.status = 200
+    else
+      res.status = 404
+    end
   end
 end
 
 # PUT Check Action
 server.put('/items/{id:[0-9]+}/check') do |req, res|
-  ListItem.find(req.params[:id]).check
-  res.status = 200
+  record = ListItem.find(req.params["id"])
+  if record
+    record.check
+    res.status = 200
+  else
+    res.status = 404
+  end
 end
 
 # PUT Check Action
 server.put('/items/{id:[0-9]+}/uncheck') do |req, res|
-  ListItem.find(req.params[:id]).uncheck
-  res.status = 200
+  record = ListItem.find(req.params["id"])
+  if record
+    record.uncheck
+    res.status = 200
+  else
+    res.status = 404
+  end
 end
 
 # DELETE Delete Action
 server.delete('/items/{id:[0-9]+}') do |req, res|
-  ListItem.find(req.params["id"]).destroy
-  res.status = 200
+  record = ListItem.find(req.params["id"])
+  if record
+    record.destroy
+    res.status = 200
+  else
+    res.status = 404
+  end
 end
 
 server.start