UNPKG

@pixelygroup/keycloak-koa-connect

Version:
94 lines (84 loc) 2.16 kB
/** * Created by zhangsong on 2018/8/9. */ // class Admin { // public _keycloak; // public _adminRequest; // public _url; // // constructor(keycloak, url) { // this._keycloak = keycloak; // if (url[url.length - 1] !== '/') { // url += '/;'; // } // this._url = url + 'k_logout'; // } // // public getFunction() { // return this._adminRequest.bind(this); // } // // } function adminLogout(ctx, keycloak) { let data = ''; ctx.req.on('data', (d) => { data += d.toString(); }); ctx.req.on('end', () => { const parts = data.split('.'); const payload = JSON.parse(new Buffer(parts[1], 'base64').toString()); if (payload.action === 'LOGOUT') { const sessionIDs = payload.adapterSessionIds; if (!sessionIDs) { keycloak.grantManager.notBefore = payload.notBefore; return ctx.response.body('ok'); } if (sessionIDs && sessionIDs.length > 0) { let seen = 0; sessionIDs.forEach((id) => { keycloak.unstoreGrant(id); ++seen; if (seen === sessionIDs.length) { return ctx.response.body('ok'); } }); } else { return ctx.response.body('ok'); } } }); } function adminNotBefore(ctx, keycloak) { let data = ''; ctx.req.on('data', (d) => { data += d.toString(); }); ctx.req.on('end', () => { const parts = data.split('.'); const payload = JSON.parse(new Buffer(parts[1], 'base64').toString()); if (payload.action === 'PUSH_NOT_BEFORE') { keycloak.grantManager.notBefore = payload.notBefore; ctx.response.body('ok'); } }); } export default function(keycloak, adminUrl) { let url = adminUrl; if (url[url.length - 1] !== '/') { url = url + '/'; } const urlLogout = url + 'k_logout'; const urlNotBefore = url + 'k_push_not_before'; return async function adminRequest(ctx, next) { switch (ctx.req.url) { case urlLogout: adminLogout(ctx, keycloak); break; case urlNotBefore: adminNotBefore(ctx, keycloak); break; default: return await next(); } }; }