diff --git a/lib/consumer/koa.js b/lib/consumer/koa.js index 23e66d0d..16d6b9e6 100644 --- a/lib/consumer/koa.js +++ b/lib/consumer/koa.js @@ -1,7 +1,7 @@ 'use strict' var koa = require('koa') - , router = require('koa-router') + , route = require('koa-route') , bodyParser = require('koa-bodyparser') var thunkify = require('thunkify') @@ -22,31 +22,30 @@ flows.getpocket.step3 = thunkify(flows.getpocket.step3) function Grant (_config) { var app = koa() app.use(bodyParser()) - app.use(router(app)) app.config = config.init(_config) - app.get('/connect/:provider/:override?', function *() { - if (this.params.override == 'callback') return yield callback + app.use(route.get('/connect/:provider/:override?', function *(provider, override) { + if (override == 'callback') return yield callback this.session.grant = { - provider:this.params.provider, - override:this.params.override, + provider:provider, + override:override, dynamic:this.request.query } yield connect - }) + })) - app.post('/connect/:provider/:override?', function *() { + app.use(route.post('/connect/:provider/:override?', function *(provider, override) { this.session.grant = { - provider:this.params.provider, - override:this.params.override, + provider:provider, + override:override, dynamic:this.request.body } yield connect - }) + })) function* connect () { var grant = this.session.grant