From 35cafff877fbf5ac73c082755ebf0df2fbd4fcfe Mon Sep 17 00:00:00 2001
From: Ritchie <skawful@gmail.com>
Date: Thu, 25 Jul 2013 16:24:00 -0700
Subject: [PATCH] Swagger integration

---
 README.md              | 15 +++++++++++++++
 lib/application.js     | 32 ++++++++++++++++++++++++++++++--
 lib/middleware/rest.js | 10 +---------
 3 files changed, 46 insertions(+), 11 deletions(-)

diff --git a/README.md b/README.md
index ee2bf258..56275b52 100644
--- a/README.md
+++ b/README.md
@@ -67,6 +67,21 @@ Get the app's exposed models.
       console.log(Model.modelName); // color
     });
     
+#### app.docs(options)
+
+Enable swagger REST api documentation.
+
+**Options**
+
+ - `basePath` The basepath for your API - eg. 'http://localhost:3000'.
+
+**Example**
+   
+    // enable docs
+    app.docs({basePath: 'http://localhost:3000'});
+    
+Run your app then navigate to [the api explorer](http://petstore.swagger.wordnik.com/). Enter your API basepath to view your generated docs.
+    
 ### Model
 
 A Loopback `Model` is a vanilla JavaScript class constructor with an attached set of properties and options. A `Model` instance is created by passing a data object containing properties to the `Model` constructor. A `Model` constructor will clean the object passed to it and only set the values matching the properties you define.
diff --git a/lib/application.js b/lib/application.js
index cf72ff2a..56ead479 100644
--- a/lib/application.js
+++ b/lib/application.js
@@ -5,7 +5,8 @@
 var DataSource = require('loopback-data').DataSource
   , ModelBuilder = require('loopback-data').ModelBuilder
   , assert = require('assert')
-  , RemoteObjects = require('strong-remoting');
+  , RemoteObjects = require('strong-remoting')
+  , swagger = require('strong-remoting/ext/swagger');
 
 /**
  * Export the app prototype.
@@ -52,6 +53,7 @@ app._models = [];
  */
 
 app.model = function (Model) {
+  this.remotes().exports[Model.pluralModelName] = Model;
   this._models.push(Model);
   Model.shared = true;
   Model.app = this;
@@ -91,4 +93,30 @@ app.remoteObjects = function () {
  
 app.remotes = function () {
   return this._remotes || (this._remotes = RemoteObjects.create());
-}
\ No newline at end of file
+}
+
+/**
+ * Enable documentation
+ */
+ 
+app.docs = function (options) {
+  var remotes = this.remotes();
+  swagger(remotes, options);
+}
+
+
+/*!
+ * Get a handler of the specified type from the handler cache.
+ */
+ 
+app.handler = function (type) {
+  var handlers = this._handlers || (this._handlers = {});
+  if(handlers[type]) {
+    return handlers[type];
+  }
+  
+  var remotes = this.remotes();
+  var handler = this._handlers[type] = remotes.handler(type);
+  return handler;
+}
+
diff --git a/lib/middleware/rest.js b/lib/middleware/rest.js
index bd116cad..080add68 100644
--- a/lib/middleware/rest.js
+++ b/lib/middleware/rest.js
@@ -18,15 +18,7 @@ module.exports = rest;
 function rest() {
   return function (req, res, next) {
     var app = req.app;
-    var remotes = app.remotes();
-    
-    // get all remote objects
-    var objs = app.remoteObjects();
-    
-    // export remote objects
-    remotes.exports = objs;
-    
-    var handler = remotes.handler('rest');
+    var handler = app.handler('rest');
     
     if(req.url === '/routes') {
       res.send(handler.adapter.allRoutes());