github.index.js

const qs = require('querystring');
const fs = require('fs');
const jwt = require('jsonwebtoken');
const cookie = require('cookie');
const jwkToPem = require('jwk-to-pem');
const auth = require('./auth.js');
const axios = require('axios');
var config;

exports.handler = (event, context, callback) => {
  if (typeof config == 'undefined') {
    config = JSON.parse(fs.readFileSync('config.json', 'utf8'));
  }
  mainProcess(event, context, callback);
};

function mainProcess(event, context, callback) {
  // Get request, request headers, and querystring dictionary
  const request = event.Records[0].cf.request;
  const headers = request.headers;
  const queryDict = qs.parse(request.querystring);
  if (event.Records[0].cf.config.hasOwnProperty('test')) {
    config.AUTH_REQUEST.redirect_uri = event.Records[0].cf.config.test + config.CALLBACK_PATH;
    config.TOKEN_REQUEST.redirect_uri = event.Records[0].cf.config.test + config.CALLBACK_PATH;
  }
  if (request.uri.startsWith(config.CALLBACK_PATH)) {
    console.log("Callback from GitHub received");
    /** Verify code is in querystring */
    if (!queryDict.code || !queryDict.state) {
      unauthorized("No code or state found.", callback);
    }
    config.TOKEN_REQUEST.code = queryDict.code;
    config.TOKEN_REQUEST.state = queryDict.state;
    /** Exchange code for authorization token */
    const postData = qs.stringify(config.TOKEN_REQUEST);
    console.log("Requesting access token.");
    axios.post(config.TOKEN_ENDPOINT, postData)
      .then(function (response) {
        console.log(response);
        var responseQueryString = qs.parse(response.data);
        /** Get authenticated user's login */
        if (responseQueryString.error) {
          internalServerError("Error while getting token: " + responseQueryString.error_description, callback);
        } else {
          const authorization = responseQueryString.token_type + ' ' + responseQueryString.access_token;
          axios.get('https://api.github.com/user', { headers: { 'Authorization': authorization } })
            .then(function (response) {
              console.log(response);
              /** Check if authenticated user's login is a member of given org */
              if (!response.data.hasOwnProperty('login')) {
                internalServerError('Unable to find login', callback);
              }
              var username = response.data.login;
              var orgsGet = 'https://api.github.com/orgs/' + config.ORGANIZATION + '/members/' + username;
              console.log("Checking ORG membership.");
              axios.get(orgsGet, { headers: { 'Authorization': authorization } })
                .then(function (response) {
                  console.log(response);
                  /** Set cookie upon verified membership */
                  if (response.status == 204) {
                    console.log("Setting cookie and redirecting.");
                    const nextLocation = {
                      "status": "302",
                      "statusDescription": "Found",
                      "body": "ID token retrieved.",
                      "headers": {
                        "location": [{
                          "key": "Location",
                          "value": event.Records[0].cf.config.hasOwnProperty('test') ? (config.AUTH_REQUEST.redirect_uri + queryDict.state) : queryDict.state
                        }],
                        "set-cookie": [{
                          "key": "Set-Cookie",
                          "value": cookie.serialize('TOKEN', jwt.sign(
                            {},
                            config.PRIVATE_KEY.trim(),
                            {
                              audience: headers.host[0].value,
                              subject: auth.getSubject(username),
                              expiresIn: config.SESSION_DURATION,
                              algorithm: 'RS256'
                            } // Options
                          ), {
                            SameSite: 'None; Secure'
                          })
                        }],
                      },
                    };
                    callback(null, nextLocation);
                  } else {
                    console.log("User not a member of required ORG. Unauthorized.");
                    unauthorized('Unauthorized. User ' + response.login + ' is not a member of required organization.', callback);
                  }
                })
                .catch(function (error) {
                  internalServerError('Error checking membership: ' + error.message, callback);
                });
            })
            .catch(function (error) {
              internalServerError('Error getting user: ' + error.message, callback);
            });
        }
      })
      .catch(function (error) {
        internalServerError('Error getting token: ' + error.message, callback);
      });
  } else if ("cookie" in headers
    && "TOKEN" in cookie.parse(headers["cookie"][0].value)) {
    // Verify the JWT, the payload email, and that the email ends with configured hosted domain
    jwt.verify(cookie.parse(headers["cookie"][0].value).TOKEN, config.PUBLIC_KEY.trim(), { algorithms: ['RS256'] }, function (err, decoded) {
      if (err) {
        switch (err.name) {
          case 'TokenExpiredError':
            console.log("Token expired, redirecting to OIDC provider.");
            redirect(request, headers, callback);
            break;
          case 'JsonWebTokenError':
            console.log("JWT error, unauthorized.");
            unauthorized(err.message, callback);
            break;
          default:
            console.log("Unknown JWT error, unauthorized.");
            unauthorized('Unauthorized. User ' + decoded.sub + ' is not permitted.', callback);
        }
      } else {
        console.log("Authorizing user.");
        auth.isAuthorized(decoded, request, callback, unauthorized, internalServerError, config);
      }
    });
  } else {
    console.log("Redirecting to GitHub.");
    redirect(request, headers, callback);
  }
}

function redirect(request, headers, callback) {
  config.AUTH_REQUEST.state = request.uri;
  // Redirect to Authorization Server
  var querystring = qs.stringify(config.AUTH_REQUEST);

  const response = {
    status: "302",
    statusDescription: "Found",
    body: "Redirecting to OAuth2 provider",
    headers: {
      "location": [{
        "key": "Location",
        "value": config.AUTHORIZATION_ENDPOINT + '?' + querystring
      }],
      "set-cookie": [{
        "key": "Set-Cookie",
        "value": cookie.serialize('TOKEN', '', { path: '/', expires: new Date(1970, 1, 1, 0, 0, 0, 0), SameSite: 'None; Secure' })
      }],
    },
  };
  callback(null, response);
}

function unauthorized(body, callback) {
  const response = {
    "status": "401",
    "statusDescription": "Unauthorized",
    "body": body,
    "headers": {
      "set-cookie": [{
        "key": "Set-Cookie",
        "value": cookie.serialize('TOKEN', '', { path: '/', expires: new Date(1970, 1, 1, 0, 0, 0, 0), SameSite: 'None; Secure' })
      }],
    },
  };
  callback(null, response);
}

function internalServerError(body, callback) {
  const response = {
    "status": "500",
    "statusDescription": "Internal Server Error",
    "body": body,
  };
  callback(null, response);
}