Bat/src/main/java/cc/fascinated/bat/model/BatUser.java

package cc.fascinated.bat.model;

import cc.fascinated.bat.BatApplication;
import cc.fascinated.bat.common.ProfileHolder;
import cc.fascinated.bat.common.Serializable;
import cc.fascinated.bat.common.UserUtils;
import cc.fascinated.bat.features.namehistory.profile.user.NameHistoryProfile;
import cc.fascinated.bat.features.scoresaber.profile.user.ScoreSaberProfile;
import cc.fascinated.bat.service.DiscordService;
import cc.fascinated.bat.service.MongoService;
import com.mongodb.client.model.ReplaceOptions;
import lombok.Getter;
import lombok.NonNull;
import lombok.RequiredArgsConstructor;
import lombok.Setter;
import net.dv8tion.jda.api.entities.User;
import org.springframework.data.annotation.Id;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;

/**
 * @author Fascinated (fascinated7)
 */
@RequiredArgsConstructor
@Getter
@Setter
public class BatUser extends ProfileHolder {
    /**
     * The document that belongs to this user
     */
    private final org.bson.Document document;

    /**
     * The ID of the user
     */
    @NonNull
    @Id
    private final String id;

    /**
     * The global name of the user
     */
    private String globalName;

    /**
     * The time this user was created
     */
    private Date createdAt;

    /**
     * The discord user associated with this user
     */
    private User user;

    public BatUser(@NonNull String id, User user, @NonNull org.bson.Document document) {
        this.id = id;
        this.document = document;
        boolean newAccount = this.document.isEmpty();
        this.createdAt = newAccount ? new Date() : document.getDate("createdAt");

        // User was not passed through
        if (user == null) {
            user = UserUtils.getUser(id);
        }
        this.user = user;
        this.globalName = user.getGlobalName();
    }

    /**
     * The name of the user
     */
    public String getName() {
        return this.getGlobalName();
    }

    /**
     * Gets the guild as the JDA Guild
     *
     * @return the guild
     */
    public User getDiscordUser() {
        if (user == null) {
            user = DiscordService.JDA.getUserById(id);
        }
        return user;
    }

    /**
     * Gets the user's ScoreSaber profile
     *
     * @return the user's ScoreSaber profile
     */
    public ScoreSaberProfile getScoreSaberProfile() {
        return getProfile(ScoreSaberProfile.class);
    }

    /**
     * Gets the user's name history profile
     *
     * @return the user's name history profile
     */
    public NameHistoryProfile getNameHistoryProfile() {
        return getProfile(NameHistoryProfile.class);
    }

    /**
     * Saves the user
     */
    public void save() {
        org.bson.Document document = new org.bson.Document();
        document.put("_id", id);
        document.put("createdAt", createdAt);

        Map<String, org.bson.Document> profileDocuments = this.document.get("profiles", new HashMap<>());
        for (Serializable profile : getProfiles().values()) {
            profileDocuments.put(profile.getClass().getSimpleName(), profile.serialize(BatApplication.GSON));
        }
        document.put("profiles", profileDocuments);

        MongoService.INSTANCE.getUsersCollection().replaceOne(
                new org.bson.Document("_id", id),
                document,
                new ReplaceOptions().upsert(true)
        );
    }

    @Override
    public <T extends Serializable> T getProfile(Class<T> clazz) {
        return getProfileFromDocument(clazz, document);
    }
}