Bat/src/main/java/cc/fascinated/bat/model/BatGuild.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.features.FeatureProfile;
import cc.fascinated.bat.features.birthday.profile.BirthdayProfile;
import cc.fascinated.bat.features.counter.CounterProfile;
import cc.fascinated.bat.features.leveling.LevelingProfile;
import cc.fascinated.bat.features.logging.LogProfile;
import cc.fascinated.bat.features.minecraft.MinecraftProfile;
import cc.fascinated.bat.features.moderation.punish.PunishmentProfile;
import cc.fascinated.bat.features.namehistory.profile.guild.NameHistoryProfile;
import cc.fascinated.bat.features.reminder.ReminderProfile;
import cc.fascinated.bat.features.statschannel.StatsChannelProfile;
import cc.fascinated.bat.features.welcomer.WelcomerProfile;
import cc.fascinated.bat.premium.PremiumProfile;
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.Setter;
import net.dv8tion.jda.api.entities.Guild;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.data.annotation.Id;

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

/**
 * @author Fascinated (fascinated7)
 */
@Getter
@Setter
public class BatGuild extends ProfileHolder {
    private static final Logger log = LoggerFactory.getLogger(BatGuild.class);
    /**
     * The document that belongs to this guild
     */
    private final org.bson.Document document;

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

    /**
     * The time this guild was joined
     */
    private Date createdAt;

    /**
     * The guild as the JDA Guild
     */
    private Guild guild;

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

        Guild guild = DiscordService.JDA.getGuildById(id);
        if (guild != null) {
            this.guild = guild;
        }
    }

    /**
     * Gets the name of the guild
     *
     * @return the name
     */
    public String getName() {
        return getDiscordGuild().getName();
    }

    /**
     * Gets the guild as the JDA Guild
     *
     * @return the guild
     */
    public Guild getDiscordGuild() {
        if (guild == null) {
            guild = DiscordService.JDA.getGuildById(id);
        }
        return guild;
    }

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

    /**
     * Gets the feature profile
     *
     * @return the feature profile
     */
    public FeatureProfile getFeatureProfile() {
        return getProfile(FeatureProfile.class);
    }

    /**
     * Gets the premium profile
     *
     * @return the premium profile
     */
    public PremiumProfile getPremiumProfile() {
        return getProfile(PremiumProfile.class);
    }

    /**
     * Gets the birthday profile
     *
     * @return the birthday profile
     */
    public BirthdayProfile getBirthdayProfile() {
        return getProfile(BirthdayProfile.class);
    }

    /**
     * Gets the log profile
     *
     * @return the log profile
     */
    public LogProfile getLogProfile() {
        return getProfile(LogProfile.class);
    }

    /**
     * Gets the reminder profile
     *
     * @return the reminder profile
     */
    public ReminderProfile getReminderProfile() {
        return getProfile(ReminderProfile.class);
    }

    /**
     * Gets the welcomer profile
     *
     * @return the welcomer profile
     */
    public WelcomerProfile getWelcomerProfile() {
        return getProfile(WelcomerProfile.class);
    }

    /**
     * Gets the counter profile
     *
     * @return the counter profile
     */
    public CounterProfile getCounterProfile() {
        return getProfile(CounterProfile.class);
    }

    /**
     * Gets the leveling profile
     *
     * @return the leveling profile
     */
    public LevelingProfile getLevelingProfile() {
        return getProfile(LevelingProfile.class);
    }

    /**
     * Gets the minecraft profile
     *
     * @return the minecraft profile
     */
    public MinecraftProfile getMinecraftProfile() {
        return getProfile(MinecraftProfile.class);
    }

    /**
     * Gets the stats channel profile
     *
     * @return the stats channel profile
     */
    public StatsChannelProfile getStatsChannelProfile() {
        return getProfile(StatsChannelProfile.class);
    }

    /**
     * Gets the punishment profile
     *
     * @return the punishment profile
     */
    public PunishmentProfile getPunishmentProfile() {
        return getProfile(PunishmentProfile.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.getGuildsCollection().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);
    }
}