strapi/packages/strapi-admin/ee/controllers/authentication/middlewares.js

'use strict';

const passport = require('koa-passport');

const constants = require('./constants');
const utils = require('./utils');

const isProduction = process.env.NODE_ENV === 'production';

const authenticate = async (ctx, next) => {
  const {
    params: { provider },
  } = ctx;

  return passport.authenticate(provider, null, async (error, profile) => {
    if (error || !profile) {
      if (error) {
        strapi.log.error(error);
      }

      return ctx.redirect(constants.PROVIDER_REDIRECT_ERROR);
    }

    const { email, firstname, lastname, username } = profile;
    const user = await strapi.admin.services.user.findOne({ email });

    // If a profile exists with this email, login with this user
    if (user) {
      ctx.state.user = user;
      return next();
    }

    const adminStore = await utils.getAdminStore();
    const { providers } = await adminStore.get({ key: 'auth' });

    // We need at least the username or the firstname/lastname combination to register a new user
    const isMissingRegisterFields = !username && (!firstname || !lastname);

    if (!providers.autoRegister || !providers.defaultRole || isMissingRegisterFields) {
      return ctx.redirect(constants.PROVIDER_REDIRECT_ERROR);
    }

    const defaultRole = await strapi.admin.services.role.findOne({ id: providers.defaultRole });

    // If the default role has been misconfigured, redirect with an error
    if (!defaultRole) {
      return ctx.redirect(constants.PROVIDER_REDIRECT_ERROR);
    }

    // Register a new user with the information given by the provider and login with it
    ctx.state.user = await strapi.admin.services.user.create({
      email,
      username,
      firstname,
      lastname,
      roles: [defaultRole.id],
      isActive: true,
      registrationToken: null,
    });

    return next();
  })(ctx, next);
};

const redirectWithAuth = ctx => {
  const { user } = ctx.state;
  const jwt = strapi.admin.services.token.createJwtToken(user);
  const cookiesOptions = { httpOnly: false, secure: isProduction, overwrite: true };

  ctx.cookies.set('jwtToken', jwt, cookiesOptions);
  ctx.redirect(constants.PROVIDER_REDIRECT_SUCCESS);
};

module.exports = {
  authenticate,
  redirectWithAuth,
};