opro 0.4.0 → 0.4.1

data/.travis.yml

@@ -1,3 +1,4 @@
 language: ruby
 rvm:
   - 1.9.3
+  - jruby-19mode

data/CHANGELOG.md

@@ -1,3 +1,16 @@
+## master
+
+
+## 0.4.1
+
+- jRuby compatibility
+- [#9] Bugfix: Allow developers to enable refresh tokens by setting `require_refresh_within` even if they were running without it previously. (@robe5)
+- [#10] Bugfix check the correct field when generating unique tokens (@carols10cents)
+- [#12] Bugfix: client app names are now changeable
+- All oPRO related html is wrapped with `opro` class (@carols10cents)
+- Switch from bluecloth to kramdown gem for jRuby compatibility while parsing markdown
+
+
 ## 0.4.0
 
 - Enable configurable authorization headers based on regex.

data/Gemfile

@@ -4,7 +4,7 @@ gem "activesupport" , ">= 3.1.0"
 gem "rails"         , ">= 3.1.0"
 
 
-gem 'bluecloth'
+gem 'kramdown' # pure ruby markdown parser
 
 group :development, :test do
   gem 'mocha'
@@ -14,31 +14,15 @@ group :development, :test do
 
 
   gem "capybara", ">= 0.4.0"
-  gem "sqlite3"
+
   gem "launchy"
-end
 
-group :test do
-  gem 'database_cleaner'
-end
+  gem "sqlite3",                          :platform => [:ruby, :mswin, :mingw]
+  gem "activerecord-jdbcsqlite3-adapter", :platform => :jruby
 
-group :test, :development do
   gem 'devise'
 end
 
-
-platforms :mri_18 do
-  group :development, :test do
-    gem "rcov"
-  end
-end
-
-platforms :mri_19 do
-  group :development, :test do
-    gem "simplecov"
-  end
+group :test do
+  gem 'database_cleaner'
 end
-
-# To use debugger (ruby-debug for Ruby 1.8.7+, ruby-debug19 for Ruby 1.9.2+)
-# gem 'ruby-debug'
-# gem 'ruby-debug19'

data/Gemfile.lock

@@ -31,7 +31,7 @@ GEM
     addressable (2.2.7)
     arel (3.0.2)
     bcrypt-ruby (3.0.1)
-    bluecloth (2.2.0)
+    bcrypt-ruby (3.0.1-java)
     builder (3.0.0)
     capybara (1.1.2)
       mime-types (>= 1.16)
@@ -50,6 +50,7 @@ GEM
       warden (~> 1.1.1)
     erubis (2.7.0)
     ffi (1.0.11)
+    ffi (1.0.11-java)
     git (1.2.5)
     hike (1.2.1)
     i18n (0.6.0)
@@ -59,8 +60,14 @@ GEM
       rake
     journey (1.0.4)
     json (1.7.3)
+    json (1.7.3-java)
+    kramdown (0.14.0)
     launchy (2.1.0)
       addressable (~> 2.2.6)
+    launchy (2.1.0-java)
+      addressable (~> 2.2.6)
+      ffi (~> 1.0.9)
+      spoon (~> 0.0.1)
     mail (2.4.4)
       i18n (>= 0.4.0)
       mime-types (~> 1.16)
@@ -71,6 +78,7 @@ GEM
       metaclass (~> 0.0.1)
     multi_json (1.3.6)
     nokogiri (1.5.2)
+    nokogiri (1.5.2-java)
     orm_adapter (0.0.7)
     polyglot (0.3.3)
     rack (1.4.1)
@@ -96,7 +104,6 @@ GEM
       rdoc (~> 3.4)
       thor (>= 0.14.6, < 2.0)
     rake (0.9.2.2)
-    rcov (1.0.0)
     rdoc (3.12)
       json (~> 1.4)
     rubyzip (0.9.7)
@@ -105,10 +112,7 @@ GEM
       ffi (~> 1.0)
       multi_json (~> 1.0)
       rubyzip
-    simplecov (0.6.1)
-      multi_json (~> 1.0)
-      simplecov-html (~> 0.5.3)
-    simplecov-html (0.5.3)
+    spoon (0.0.1)
     sprockets (2.1.3)
       hike (~> 1.2)
       rack (~> 1.0)
@@ -127,20 +131,20 @@ GEM
       nokogiri (~> 1.3)
 
 PLATFORMS
+  java
   ruby
 
 DEPENDENCIES
+  activerecord-jdbcsqlite3-adapter
   activesupport (>= 3.1.0)
-  bluecloth
   bundler (>= 1.1.3)
   capybara (>= 0.4.0)
   database_cleaner
   devise
   jeweler (~> 1.6.4)
+  kramdown
   launchy
   mocha
   rails (>= 3.1.0)
-  rcov
-  simplecov
   sqlite3
   timecop

data/README.md

@@ -1,15 +1,21 @@
-## Opro
+## oPRO
 
-A production ready Rails Engine that turns your app into an [Oauth2](http://oauth.net/2/) Provider.
+A production ready Rails Engine that turns your app into an [OAuth2](http://oauth.net/2/) Provider.
 
-  * [Demo OAuth Provider app with Opro](http://opro-demo.herokuapp.com/) on Heroku
-  * [Built in Opro docs](http://opro-demo.herokuapp.com/oauth_docs)
+  * [Demo OAuth Provider app with oPRO](http://opro-demo.herokuapp.com/) on Heroku
+  * [Built in oPRO docs](http://opro-demo.herokuapp.com/oauth_docs)
+  * [Example iOS App](https://github.com/opro/opro_iphone_demo)
+  * [Would you like a Mobile app with that (oPRO tutorial)](http://schneems.com/post/33781154129/would-you-like-a-mobile-app-with-that)
+
+oPRO is short for (O)Auth (Pro)vider and is pronounced "oh proh". Not to be confused with [Oprah](http://www.oprah.com/index.html), who does not support or endorse this ruby gem in any way (yet).
+
+[![Build Status](https://secure.travis-ci.org/opro/opro.png)](http://travis-ci.org/opro/opro)
 
 ## Why would I use this?
 
 Lets say you've built a Rails app, awesome. Now you want to build a mobile app on say, the iPhone... cool. You start throwing around `#to_json` like nobody's business, but then you realize you need to authenticate users somehow. "Basic Auth!!", you exclaim, but then you realize that's not the most secure solution. You also realize that some users already signed up with Facebook & Twitter so they don't have a username/password combo. What ever shall you do?
 
-Wouldn't it be great if we could have a token exchange where the user goes to a mobile web view and grants permission, and then we return back an auth token just like the big boys (Facebook, Twitter, *cough* Foursquare *cough*). With Opro, we can add this functionality pretty easily. We'll use your existing authentication strategy and provide some integration end points for your clients to use out of the box.
+Wouldn't it be great if we could have a token exchange where the user goes to a mobile web view and grants permission, and then we return back an auth token just like the big boys (Facebook, Twitter, *cough* Foursquare *cough*). With oPRO, we can add this functionality pretty easily. We'll use your existing authentication strategy and provide some integration end points for your clients to use out of the box.
 
 
 ## Install
@@ -32,7 +38,7 @@ and don't forget
     $ rails g opro:install
 ```
 
-This will put a file in `initializers/opro.rb` and generate some migrations, and add `mount_opro_oauth` to your routes.
+This will put a file in `initializers/opro.rb`, generate some migrations, and add `mount_opro_oauth` to your routes.
 
 
 Now we're ready to migrate the database
@@ -41,11 +47,11 @@ Now we're ready to migrate the database
     $ rake db:migrate
 ````
 
-This will add `Opro::Oauth::AuthGrant` and `Opro::Oauth::ClientApp` to your database. An iPhone app would need to register for a `client_id` and `client_secret` before using Oauth as a ClientApp. Once created they could get authorization from users by going through the oauth flow, thus creating AuthGrants. In other words a ClientApp has many users through AuthGrants.
+This will add `Opro::Oauth::AuthGrant` and `Opro::Oauth::ClientApp` to your database. An iPhone app would need to register for a `client_id` and `client_secret` before using OAuth as a ClientApp. Once created they could get authorization from users by going through the OAuth flow, thus creating AuthGrants. In other words, a ClientApp has many users through AuthGrants.
 
 ## Setup
 
-Go to `initializers/opro.rb` and configure your app for your authentication scheme, if you're not using devise see "Custom Auth" below.
+Go to `initializers/opro.rb` and configure your app for your authentication scheme. If you're not using devise, see "Custom Auth" below.
 
 ```ruby
       Opro.setup do |config|
@@ -72,24 +78,24 @@ You can also disallow OAuth on specific actions. Disallowing will always over-ri
       end
 ```
 
-By default all OAuth access is blacklisted. To whitelist all access, add `allow_oauth!` to your `ApplicationController` (this is not recommended). The best practice is to add `allow_oauth!` or `disallow_oauth` to each and every controller.
+By default, all OAuth access is blacklisted. To whitelist all access, add `allow_oauth!` to your `ApplicationController` (this is not recommended). The best practice is to add `allow_oauth!` or `disallow_oauth` to each and every controller.
 
-That should be all you need to do to get setup, congrats you're now able to authenticate users using OAuth!!
+That should be all you need to do to get set up. Congrats, you're now able to authenticate users using OAuth!!
 
 
 ## Use it
 
-Opro comes with built in documentation, so if you start your server you can view them at [http://localhost:3000/oauth_docs](http://localhost:3000/oauth_docs). Or you can [view the guide](http://opro-demo.herokuapp.com/oauth_docs) on the example app. This guide will walk you through creating your first OAuth client application, giving access to that app as a logged in user, getting an access token for that user, and using that token to access the server as an authenticated user!
+oPRO comes with built in documentation, so if you start your server you can view the docs at [http://localhost:3000/oauth_docs](http://localhost:3000/oauth_docs). Or you can [view the guide](http://opro-demo.herokuapp.com/oauth_docs) on the example app. This guide will walk you through creating your first OAuth client application, giving access to that app as a logged in user, getting an access token for that user, and using that token to access the server as an authenticated user!
 
 
 # Advanced Setup
 
-Opro is simple by default, but easily configurable for a number of common use cases. Check out the options below.
+oPRO is simple by default, but easily configurable for a number of common use cases. Check out the options below.
 
 
 ## Custom Auth
 
-If you're not using devise you can manually configure your own auth strategy. In the future I plan on adding more auth strategies, ping me or submit a pull request for your desired authentication scheme.
+If you're not using devise, you can manually configure your own auth strategy. In the future I plan on adding more auth strategies; ping me or submit a pull request for your desired authentication scheme.
 
 
       Opro.setup do |config|
@@ -100,12 +106,12 @@ If you're not using devise you can manually configure your own auth strategy. In
 
 ## Permissions
 
-When a user auth's with a client they automatically are granting read permission to any action that you `allow_oauth!`. Read only clients are restricted to using GET requests. By default Opro will ask users for write permission on a client by client application. Client apps with `:write` permission can use all HTTP verbs including POST, PATCH, PUT, DESTROY on any url you whitelist using `allow_oauth!`.
+When a user authenticates with a client, they are automatically granting read permission to any action that you `allow_oauth!`. Read-only clients are restricted to using GET requests. By default, oPRO will ask users for write permission on a client by the client application. Client apps with `:write` permission can use all HTTP verbs including POST, PATCH, PUT, DESTROY on any url you whitelist using `allow_oauth!`.
 
 
 ### Custom Permissions
 
-To remove write permissions comment out this line in the Opro initializer:
+To remove write permissions, comment out this line in the oPRO initializer:
 
       config.request_permissions = [:write]
 
@@ -113,11 +119,11 @@ You can add custom permissions by adding to the array:
 
       config.request_permissions = [:write, :email, :picture, :whatever]
 
-You can then restrict access using the custom permissions by calling `require_oauth_permissions` which takes the same arguments as `before_filter`
+You can then restrict access using the custom permissions by calling `require_oauth_permissions`, which takes the same arguments as `before_filter`:
 
       require_oauth_permissions :email, :only => :index
 
-You can also skip permissions using `skip_oauth_permissions`. By default permissions will just check to see if a client has the permission, and will allow the action if it is present. If you want to implement custom permission checks you can write custom methods using the pattern `oauth_client_can_#{permission}?` for example if you were restricting the `:email` permission, you would create a method.
+You can also skip permissions using `skip_oauth_permissions`. By default, permissions will just check to see if a client has the permission and will allow the action if it is present. If you want to implement custom permission checks, you can write custom methods using the pattern `oauth_client_can_#{permission}?`. For example, if you were restricting the `:email` permission, you would create a method:
 
       def oauth_client_can_email?
         # ...
@@ -128,36 +134,44 @@ The result is expected to be true or false.
 
 ## Refresh Tokens
 
-For added security you can require access_tokens be refreshed by client applications. This will help to mitigate risk of a leaked access_token, and enable an all around more secure system. This security comes at a price however, since implementing the `refresh_token` functionality in a client can be more difficult.
+For added security, you can require access_tokens to be refreshed by client applications. This will help to mitigate the risk of a leaked access_token and enable an all around more secure system. This security comes at a price, however, since implementing the `refresh_token` functionality in a client can be more difficult.
 
-By default refresh tokens are enabled, you can disable them in your application and set the timeout period of the tokens by adding this line to your configuration.
+By default, refresh tokens are enabled. You can disable them in your application and set the timeout period of the tokens by adding this line to your configuration:
 
-    config.require_refresh_within = 1.month
+    config.require_refresh_within = false
 
+# Toggling Refresh Tokens
 
+If you disable refresh tokens and then re-enable it you may have authorization grants that do not have a timeout listed, you can keep it like this or you can fix by iterating through all auth grants and setting their `access_token_expires_at` like this:
+
+    Opro::Oauth::AuthGrant.find_each(:conditions => "access_token_expires_at is null") do |grant|
+      grant.access_token_expires_at = Time.now + ::Opro.require_refresh_within
+      grant.save
+    end
 
+You may also need to inform clients that they need to update their credentials and start using refresh tokens.
 
 ## Password Token Exchange
 
-If a client application has a user's password and username/email they can exchange these for a token. This is much safer than storing username and password on a local device, but does not offer the traditional OAuth "Flow". Because of this all available permissions will be granted to the client application. If you want to disable this feature you can set the configuration below to false:
+If a client application has a user's password and username/email, they can exchange these for a token. This is much safer than storing the username and password on a local device, but it does not offer the traditional OAuth "Flow". Because of this, all available permissions will be granted to the client application. If you want to disable this feature you can set the configuration below to false:
 
     config.password_exchange_enabled = true
 
-If you have this feature enabled you can further control what applications can use the feature. Some providers may wish to have "Blessed" client applications that have this ability while restricting all other clients. To accomplish this you can create a method in your ApplicationController called `oauth_valid_password_auth?` that accepts a client_id and client_secret, and returns a true or false based on whether that application can use password auth
+If you have this feature enabled, you can further control what applications can use the feature. Some providers may wish to have "Blessed" client applications that have this ability while restricting all other clients. To accomplish this, you can create a method in your ApplicationController called `oauth_valid_password_auth?` that accepts a client_id and client_secret and returns true or false based on whether that application can use password auth:
 
     def oauth_valid_password_auth?(client_id, client_secret)
       BLESSED_APP_IDS.include?(client_id)
     end
 
 
- If you are using this password functionality without a supported authorization engine (like devise), you will need to add an additional method that supports validating whether or not a user's credentials are valid. The method for this is called `find_user_for_auth` and accepts a controller and the parameters. The output is expected to be a user. Add this to your config like you did to the other required methods in the Custom Auth section.
+ If you are using this password functionality without a supported authorization engine (like devise), you will need to add an additional method that supports validating whether or not a user's credentials are valid. The method for this is called `find_user_for_auth` and accepts a controller and the parameters. The output is expected to be a user. Add this to your config like you did to the other required methods in the "Custom Auth" section:
 
     config.find_user_for_auth do |controller, params|
       # user = User.find(params[:something])
       # return user.valid_password?(params[:password]) ? user : false
     end
 
-If you're authenticating exchanging something other than a password (such as a facebook auth token) client's can still enable this functionality by setting `params[:grant_type] == 'password'` in their initial request. You can then use `find_user_for_auth` method from above and implement your custom behavior. You can call `find_user_for_auth` multiple times and the application will try calling each auth method in order. It is suggested that you return from this block early if the params are missing a vital key like this:
+If you're authenticating by exchanging something other than a password (such as a facebook auth token), clients can still enable this functionality by setting `params[:grant_type] == 'password'` in their initial request. You can then use the `find_user_for_auth` method from above and implement your custom behavior. You can call `find_user_for_auth` multiple times and the application will try calling each auth method in order. It is suggested that you return from this block early if the params are missing a vital key like this:
 
 
     config.find_user_for_auth do |controller, params|
@@ -168,7 +182,7 @@ If you're authenticating exchanging something other than a password (such as a f
 
 ## Rate Limiting
 
-If your API becomes a runaway success and people starte abusing your api, you might chose to limit the rate that client applications can access your API. It is common for popular read only API's to have an hourly, or daily rate limit to help prevent abuse. If you want this type of functionality you can use Opro's built in hooks, one to record the number of times a client application has accessed your api. And another to let the application know if the Client app has gone over it's alloted rate.
+If your API becomes a runaway success and people start abusing your API, you might choqse to limit the rate that client applications can access your API. It is common for popular read-only APIs to have an hourly or daily rate limit to help prevent abuse. If you want this type of functionality, you can use oPRO's built in hooks: one to record the number of times a client application has accessed your API and another to let the application know if the Client app has gone over its allotted rate.
 
 To record the number of times an application has accessed your site add this method to your ApplicationController:
 
@@ -176,26 +190,26 @@ To record the number of times an application has accessed your site add this met
       # implement your rate counting mechanism here
     end
 
-Then to let our server know if a given client has reached add this method, the output is expected to be true if the client has gone over their limit, and false if they have not:
+Then to let our server know if a given client has reached its limit, add the method below. The output is expected to be true if the client has gone over their limit and false if they have not:
 
     def oauth_client_rate_limited?(client_id, params)
       # implement your own custom rate limiting logic here
     end
 
-Rate limited clients will receive an "unsuccessful" response to any query with a message letting them know they've been rate limited. Using redis with a rotating key generator based on (hour, daty, etc.) is one very common way to count rate, and implement the rate limits. Since there are so many different ways to implement this, we decided to give you a blank slate and implement it however you like. The default is that apps are not rate limited, and in general unlimited API access is the way to go, but if you do find abusive behavior you can always easily add in a rate limit.
+Rate limited clients will receive an "unsuccessful" response to any query with a message letting them know they've been rate limited. Using redis with a rotating key generator based on (hour, day, etc.) is one very common way to count accesses and implement the rate limits. Since there are so many different ways to implement this, we decided to give you a blank slate and let you implement it however you would like. The default is that apps are not rate limited, and in general unlimited API access is the way to go, but if you do find abusive behavior you can always easily add in a rate limit.
 
 
 ## Configurable Authorization Header
 
-By default oPRO allows clients to send their authorization token in a header. For example someone using an auth token of `9693accessTokena7ca570bbaf` could set the `Authorization` header in a request like this:
+By default, oPRO allows clients to send their authorization token in a header. For example, someone using an auth token of `9693accessTokena7ca570bbaf` could set the `Authorization` header in a request like this:
 
     $ curl -H "Authorization: Bearer 9693accessTokena7ca570bbaf" "http://localhost:3000/oauth_test/show_me_the_money"
 
-By default oPRO will accept `Bearer` and `token` in the authorization header, but if your client needs to send a custom auth header, you can add a custom extra regular expression to parse parse and return the token. For example if a client was setting the auth header like this:
+By default, oPRO will accept `Bearer` and `token` in the authorization header, but if your client needs to send a custom auth header, you can add a custom extra regular expression to parse and return the token. For example, if a client was setting the auth header like this:
 
     $ curl -H "Authorization: cUsTomAuthHeader 9693accessTokena7ca570bbaf" "http://localhost:3000/oauth_test/show_me_the_money"
 
-You could pull out the auth token using this regular expression `/cUsTomAuthHeader\s(.*)/`. If you're not great with regular expressions I highly recommend using [Rubular](http://rubular.com) to test regex matches. It is very important that we are "capturing" data with in between the `()` characters. The data returned inside of the parens are expected to be the auth token with no spaces or special characters such as new lines or quotes. To set parse this auth header in oPRO, you can specify the `header_auth_regex` in an initializer like this:
+You could pull out the auth token using the regular expression `/cUsTomAuthHeader\s(.*)/`. If you're not great with regular expressions, I highly recommend using [Rubular](http://rubular.com) to test regex matches. It is very important that we are "capturing" data in between the `()` characters. The data returned inside of the parens is expected to be the auth token with no spaces or special characters such as new lines or quotes. To parse this auth header in oPRO, you can specify the `header_auth_regex` in an initializer like this:
 
 
       Opro.setup do |config|
@@ -204,17 +218,17 @@ You could pull out the auth token using this regular expression `/cUsTomAuthHead
         config.header_auth_regex = /cUsTomAuthHeader\s(.*)/
       end
 
-Now when a client sends your custom auth header it will be parsed correctly. Custom authorization headers should not be used for security through obscurity. They may be exposed in the docs or tests in a later iteration of oPRO, if you have strong feelings against this, then please open a pull request or send me a message stating your case.
+Now when a client sends your custom auth header, it will be parsed correctly. Custom authorization headers should not be used for security through obscurity. They may be exposed in the docs or tests in a later iteration of oPRO. If you have strong feelings against this, then please open a pull request or send me a message stating your case.
 
 
 
 ## Assumptions
 
-* You have a user model and that is what your authenticating
-* You're using Active::Record
+* You have a user model and that is what you're authenticating
+* You're using Active Record
 
 ## About
 
-If you have a question file an issue or, find me on the Twitters [@schneems](http://twitter.com/schneems). Another good library for turning your app into an OAuth provider is [Doorkeeper](https://github.com/applicake/doorkeeper), if this project doesn't meet your needs let me know why and use them :)
+If you have a question file an issue or find me on the Twitters [@schneems](http://twitter.com/schneems). Another good library for turning your app into an OAuth provider is [Doorkeeper](https://github.com/applicake/doorkeeper), if this project doesn't meet your needs let me know why and use them :)
 
 This project rocks and uses MIT-LICENSE.

data/Rakefile

@@ -26,7 +26,7 @@ task :default => :test
 
 Rake::RDocTask.new(:rdoc) do |rdoc|
   rdoc.rdoc_dir = 'rdoc'
-  rdoc.title    = 'Opro'
+  rdoc.title    = 'oPRO'
   rdoc.options << '--line-numbers' << '--inline-source'
   rdoc.rdoc_files.include('README.rdoc')
   rdoc.rdoc_files.include('lib/**/*.rb')
@@ -47,7 +47,7 @@ Jeweler::Tasks.new do |gem|
   gem.name        = "opro"
   gem.homepage    = "http://github.com/schneems/opro"
   gem.license     = "MIT"
-  gem.summary     = %Q{ Opro turns your Rails application into an OAuth Provider }
+  gem.summary     = %Q{ oPRO turns your Rails application into an OAuth Provider }
   gem.description = %Q{ Enable OAuth clients (iphone, android, web sites, etc.) to access and use your Rails application, what you do with it is up to you}
   gem.email       = "richard.schneeman@gmail.com"
   gem.authors     = ["schneems"]

data/VERSION

@@ -1 +1 @@
-0.4.0
+0.4.1

data/app/controllers/opro/oauth/client_app_controller.rb

@@ -10,12 +10,30 @@ class Opro::Oauth::ClientAppController < OproController
     @client_apps = Opro::Oauth::ClientApp.where(:user_id => current_user.id)
   end
 
+  def show
+    @client_app = Opro::Oauth::ClientApp.where(:id => params[:id], :user_id => current_user.id).first
+  end
+
+  def edit
+    @client_app = Opro::Oauth::ClientApp.where(:id => params[:id], :user_id => current_user.id).first
+  end
+
+  def update
+    @client_app = Opro::Oauth::ClientApp.where(:id => params[:id], :user_id => current_user.id).first
+    @client_app.name = params[:opro_oauth_client_app][:name]
+    if @client_app.save
+      redirect_to oauth_client_app_path(@client_app)
+    else
+      render :edit
+    end
+  end
+
 
   def create
     @client_app = Opro::Oauth::ClientApp.find_by_user_id_and_name(current_user.id, params[:opro_oauth_client_app][:name])
     @client_app ||= Opro::Oauth::ClientApp.create_with_user_and_name(current_user, params[:opro_oauth_client_app][:name])
     if @client_app.save
-      # do nothing
+      redirect_to oauth_client_app_path(@client_app)
     else
       render :new
     end

data/app/controllers/opro/oauth/docs_controller.rb

@@ -1,19 +1,18 @@
 require 'erb'
-require 'bluecloth'
-
-OPRO_MD_ROOT=File.join(File.dirname(__FILE__), '../../../views/opro/oauth/docs/markdown/')
+require 'kramdown'
 
+OPRO_MD_ROOT = File.join(File.dirname(__FILE__), '../../../views/opro/oauth/docs/markdown/')
 
 class Opro::Oauth::DocsController < OproController
+  before_filter :set_protocol!
   helper_method :render_doc
 
   def index
-    @protocol = protocol
   end
 
   def show
-    @protocol = protocol
-    @doc = params[:id]
+    @doc  = params[:id]
+    render :file => default_404, :status => 404 and return unless md_exists?(@doc)
   end
 
   def render_doc(name)
@@ -25,8 +24,12 @@ class Opro::Oauth::DocsController < OproController
 
   private
 
-  def protocol
-    Rails.env.production? ? "https" : "http"
+  def default_404
+    Rails.root.join("public", "404")
+  end
+
+  def set_protocol!
+    @protocol = Rails.env.production? ? "https" : "http"
   end
 
   def parse_erb(str)
@@ -34,11 +37,18 @@ class Opro::Oauth::DocsController < OproController
   end
 
   def parse_markdown(str)
-    BlueCloth.new(str).to_html
+    Kramdown::Document.new(str).to_html
+  end
+
+  def doc_md_filename(name)
+    OPRO_MD_ROOT + name + '.md.erb'
+  end
+
+  def md_exists?(name)
+    File.exists?(doc_md_filename(name.to_s))
   end
 
   def read_file(name)
-    name = OPRO_MD_ROOT + name
-    File.open(name + '.md.erb' ).read.to_s
+    File.open(doc_md_filename(name)).read.to_s
   end
 end