22.03.2021 11:44
Tollik 4 480 Источник

Добрый день! Решил поделиться своими небольшими достижениями в использовании популярной библиотеки Passport.js. Задача была такой: использовать две локальные стратегии passport.js при том, что пользователи находятся в двух разных коллекциях Mongo Atlas.

Я искал ответы в интернете и встретил подобные вопросы, но либо ответа нет вообще, либо все не то:

Но из этих двух вопросов мне понравился больше второй, на его основе и рассмотрим как же реализовать локальные стратегии Passport.js если пользователи находятся в разных коллекциях.

И так, Дано: две коллекции пользователей в Mongo Atlas, и две локальные стратеги в Passport.js.

В данной статье я не буду описывать как сделать свою локальную стратегию. Это хорошо описано в этой статье, я как раз опирался на нее, когда писал свою, рекомендую.

Задача, реализовать аутентификацию учителей по логину и паролю, учеников по Фамилии и номеру класса. Тут кстати в вопросе за одно и объясняется, почему мы используем две локальных стратегии, именно потому, что способы идентификации пользователя разные (учителя- логин/пароль, ученики - Фамилия/класс)

Давайте рассмотрим, что мы имеем.

Коллекция учителей:

teachers = {
login: "Zayka-English",
password: "qwerty",
klass: "7Б",
...
}

Коллекция ученики:

students = {
surname: "Иванов",
klass: "7Б",
category: "Ботан",
...
}

Стратегия passport.js для входа учителей:

passport.use('teachers_login', new LocalStrategy({
    // Переписываем переменные из того что пришло из формы
    usernameField: 'login',
    passReqToCallback: true
},
    async function (req, username, password, done) {
        // Поиск пользователя по логину
        await Teachers.findOne({ login: username }, function (err, user) {
            if (err) { return done(err); }
            // проверяем результат поиска
            if (!user) {
                return done(null, false, req.flash('msg', ['Ошибка', ': Пользователь не найден']));
            } else {
                // Проверяем пароль
                if (user.password == password) {
                    console.log("Пользователь выполнил вход");
                    return done(null, user);
                } else {
                    console.log('Неправильно введен пароль!');
                    return done(null, false, req.flash('msg', ['Ошибка', ': Неправильно введен пароль']));
                }
            }
        });
    }
))

Стратегия passport.js для входа учеников:

passport.use('students_login', new LocalStrategy({
    // Переписываем переменные из того что пришло из формы
    usernameField: 'surname',
  	userpasswordField: 'klass',
    passReqToCallback: true
},
    async function (req, username, password, done) {
        // Поиск пользователя по Фамилии
        await Students.findOne({ surname: username }, function (err, user) {
            if (err) { return done(err); }
            // проверяем результат поиска
            if (!user) {
                return done(null, false, req.flash('msg', ['Ошибка', ': Пользователь не найден']));
            } else {
                // Проверяем пароль
                if (user.password == password) {
                    console.log("Пользователь выполнил вход");
                    return done(null, user);
                } else {
                    return done(null, false, req.flash('msg', ['Ошибка', ': Неправильно введен класс']));
                }
            }
        });
    }
))

Ну и конечно же сериализация и десериализация:

// Записывает id пользователя в cookie
passport.serializeUser(function (user, done) {
    done(null, user.id);
});

// Проверяет пользователя по id из cookie
passport.deserializeUser(function(id, done) {
  User.findById(id, function(err, user) {
    done(err, user);
  });
});

Сериализация и десериализация взяты с сайта passport.js и требуют поправки, чем мы сейчас и займемся. Вся проблема нашей задачи состоит в том, что во время входа пользователя сериализация сработает и запишет в cookie id пользователя, но при десериализации нам нужно реализовать поиск в нужной коллекции. В одном из ответов в интернете было предложено поочередно проверять обе коллекции. То есть сначала ищем id в коллекции с учителями, если там нет. то ищем в коллекции с учениками. Но такой подход использует несколько запросов к БД, что увеличивает время на десериализацию. А что если коллекций не две, а больше? Я где-то видел спрашивалось о трех коллекциях, или представим, что коллекций 5! В этом случае десериализация может происходить достаточно долго. Что же делать? А ответ оказался достаточно прост. Нам необходимо дать понять, в какой из коллекций искать id пользователя при десериализации. Давайте сделаем это.

Для того, что бы это реализовать для начала в наших коллекциях добавим к объектам с пользователями некоторые флаги:

Коллекция учителей:

teachers = { 
  login: "Zayka-English", 
  password: "qwerty", 
  klass: "7Б", 
  flag: "teacher", // в значении флага напишем к какой категории относится пользователь
  ... }

Коллекция учеников:

students = {
	surname: "Иванов",
	klass: "7Б",
	category: "Ботан",
	flag: "student", //тут делаем аналогично
	...}

Теперь во время входа пользователя мы запишем в cookie не только id пользователя, но и флаг, который есть в нашем объекте User. Это делается при сериализации

passport.serializeUser(function (user, done) {
// На данный этапе в переменной user находится объект с данными пользователя
// который мы получили при его идентификации в наших стратегиях
    done(null, [user.flag, user.id]);
});

Теперь после того как пользователь "представился" мы внесли в cookie массив, в котором первый элемент - это flag (категория к которой относится пользователь), а второй - это id пользователя.

Теперь при десериализации нам необходимо просто проверить к какой из категорий относится пользователь и проверять его id в конкретной коллекции:

passport.deserializeUser(function(id, done) {
  // На этом этапе переменная id содержит наш массив
  // Если флаг "учитель", ищем id в коллекции "Учителя"
  if (id[0] == "teacher"){
	  Teacher.findById(id[1], function(err, user) {
    	done(err, user);
  	})
  // Если иное, ищем id в коллекции "Ученики"
  }else{
  	Student.findById(id[1], function(err, user) {
    	done(err, user);
  	})
  }
});

При такой реализации десериализация будет происходить именно в той коллекции, к которой относится пользователь, то есть будет происходить только один запрос к БД к конкретной коллекции.

Так как у нас всего две коллекции, то выбор коллекции при десериализации мы осуществили через if/else, если у вас больше коллекций, конечно же будет уместней использовать конструкцию switch

На этом все дорогие друзья, в следующей статье я напишу как же сделать авторизацию учителя, что бы в ответе получить объект с данными учителя и с массивом объектов учеников)

Использовал при написании:

Пожалуй, лучший учебник по JavaSkript

Официальная документация по Passport.js

Статья о том, как создать свою локальную коллекцию в passport.js