001/*******************************************************************************
002 * Copyright 2017 The MIT Internet Trust Consortium
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *   http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 *******************************************************************************/
016
017package org.mitre.openid.connect.service.impl;
018
019import org.mitre.openid.connect.model.UserInfo;
020import org.mitre.openid.connect.service.LoginHintExtracter;
021import org.mitre.openid.connect.service.UserInfoService;
022import org.springframework.beans.factory.annotation.Autowired;
023
024import com.google.common.base.Strings;
025
026/**
027 * Checks the login hint against the User Info collection, only populates it if a user is found.
028 * @author jricher
029 *
030 */
031public class MatchLoginHintsAgainstUsers implements LoginHintExtracter {
032
033        @Autowired
034        private UserInfoService userInfoService;
035
036        /* (non-Javadoc)
037         * @see org.mitre.openid.connect.service.LoginHintTester#useHint(java.lang.String)
038         */
039        @Override
040        public String extractHint(String loginHint) {
041                if (Strings.isNullOrEmpty(loginHint)) {
042                        return null;
043                } else {
044                        UserInfo user = userInfoService.getByEmailAddress(loginHint);
045                        if (user == null) {
046                                user = userInfoService.getByUsername(loginHint);
047                                if (user == null) {
048                                        return null;
049                                } else {
050                                        return user.getPreferredUsername();
051                                }
052                        } else {
053                                return user.getPreferredUsername();
054                        }
055                }
056        }
057
058}