de.felix_klauke.pegasus.server.user.UserManager.java Source code

Introduction

Here is the source code for de.felix_klauke.pegasus.server.user.UserManager.java

Source

/*
 * Copyright 2016 Felix Klauke
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package de.felix_klauke.pegasus.server.user;

import com.google.common.collect.Lists;
import com.mongodb.MongoCredential;
import com.mongodb.client.FindIterable;
import de.felix_klauke.pegasus.server.database.Database;
import io.netty.channel.Channel;
import org.bson.Document;

import java.util.List;

/**
 * Created by Felix Klauke for project Pegasus on 14.02.2016.
 */
public class UserManager {

    /* ----------------------------------- [ Fields ] ----------------------------------- */

    /*
     * A List of all avtice users
     */
    private List<User> users;

    /**
     * The Database the users are stored in
     */
    private Database userDatabase;

    /* ----------------------------------- [ Constructors ] ----------------------------------- */

    /**
     * Creates the UserManager and connects to the Database
     */
    public UserManager() {
        users = Lists.newArrayList();
        userDatabase = new Database(
                MongoCredential.createScramSha1Credential("user", "imladria", "derKLud".toCharArray()), "imladria",
                "users");
    }

    /* ----------------------------------- [ Methods ] ----------------------------------- */

    /**
     * Get a User by its Channel
     *
     * @param channel
     * @return
     */
    public User getUser(Channel channel) {
        for (User user : users) {
            if (user.getChannel() == channel)
                return user;
        }
        return null;
    }

    /**
     *
     * Authenticate an user
     *
     * @param username the username to auth with
     * @param password the password to auth with
     * @return whether the authentication was successfully
     */
    public boolean authUser(String username, String password) {
        FindIterable<Document> iterable = this.userDatabase.getCollection()
                .find(new Document().append("username", username));
        Document document = iterable.first();
        if (document != null) {
            return document.getString("password").equals(password);
        }
        return false;
    }

    /**
     *
     * Create a new User when he logs in
     *
     * @param username the name of the user
     * @param channel the channel the user logged in in
     * @return the user
     */
    public User createUser(String username, Channel channel) {
        User user = new User(channel, username);
        users.add(user);
        return user;
    }

    public List<User> getUsers() {
        return users;
    }

}