Plex/src/main/java/dev/plex/storage/MongoConnection.java

package dev.plex.storage;

import com.mongodb.*;
import com.mongodb.client.MongoClient;
import com.mongodb.client.MongoClients;
import dev.morphia.Datastore;
import dev.morphia.Morphia;
import dev.morphia.mapping.MapperOptions;
import dev.plex.PlexBase;
import dev.plex.player.PlexPlayer;
import dev.plex.util.PlexLog;

public class MongoConnection extends PlexBase
{
    // USE MORPHIA API FOR MONGO <3

    public Datastore getDatastore()
    {
        if (!plugin.config.getString("data.central.storage").equalsIgnoreCase("mongodb"))
        {
            return null;
        }
        String host = plugin.config.getString("data.central.hostname");
        int port = plugin.config.getInt("data.central.port");
        String username = plugin.config.getString("data.central.user");
        String password = plugin.config.getString("data.central.password");
        String database = plugin.config.getString("data.central.db");

        String connectionString;
        if (username != null && password != null && !username.isEmpty() && !password.isEmpty())
        {
            if (database != null && !database.isEmpty())
            {
                connectionString = "mongodb://" + username + ":" + password + "@" + host + ":" + port + "/?authSource=" + database;
            } else {
                connectionString = "mongodb://" + username + ":" + password + "@" + host + ":" + port + "/";
            }
        } else {
            connectionString = "mongodb://" + host + ":" + port + "/";
        }
        connectionString += "?uuidRepresentation=STANDARD";
        PlexLog.debug("Using mongo connection string: " + connectionString);
        MongoClient client = MongoClients.create(connectionString);
        Datastore datastore = Morphia.createDatastore(client, database == null ? "admin" : database, MapperOptions.DEFAULT);
        datastore.getMapper().map(PlexPlayer.class);
        datastore.ensureIndexes();
        plugin.setStorageType(StorageType.MONGODB);
        return datastore;
    }
}