ActuallyAdditions/src/main/java/de/ellpeck/actuallyadditions/mod/booklet/entry/BookletEntry.java

/*
 * This file ("BookletEntry.java") is part of the Actually Additions mod for Minecraft.
 * It is created and owned by Ellpeck and distributed
 * under the Actually Additions License to be found at
 * http://ellpeck.de/actaddlicense
 * View the source code at https://github.com/Ellpeck/ActuallyAdditions
 *
 * © 2015-2017 Ellpeck
 */

package de.ellpeck.actuallyadditions.mod.booklet.entry;

import de.ellpeck.actuallyadditions.api.ActuallyAdditionsAPI;
import de.ellpeck.actuallyadditions.api.booklet.IBookletChapter;
import de.ellpeck.actuallyadditions.api.booklet.IBookletEntry;
import de.ellpeck.actuallyadditions.api.booklet.IBookletPage;
import de.ellpeck.actuallyadditions.mod.ActuallyAdditions;
import de.ellpeck.actuallyadditions.mod.util.StackUtil;
import de.ellpeck.actuallyadditions.mod.util.StringUtil;
import net.minecraft.client.Minecraft;
import net.minecraft.client.util.ITooltipFlag;
import net.minecraft.item.ItemStack;
import net.minecraft.util.text.TextFormatting;
import net.minecraftforge.api.distmarker.Dist;
import net.minecraftforge.api.distmarker.OnlyIn;
import net.minecraftforge.fluids.FluidStack;

import java.util.ArrayList;
import java.util.List;
import java.util.Locale;

public class BookletEntry implements IBookletEntry {

    private final String identifier;
    private final int priority;
    private final List<IBookletChapter> chapters = new ArrayList<>();
    private TextFormatting color;

    public BookletEntry(String identifier) {
        this(identifier, 0);
    }

    public BookletEntry(String identifier, int prio) {
        this.identifier = identifier;
        this.priority = prio;
        ActuallyAdditionsAPI.addBookletEntry(this);

        this.color = TextFormatting.RESET;
    }

    @OnlyIn(Dist.CLIENT)
    private static boolean fitsFilter(IBookletPage page, String searchBarText) {
        Minecraft mc = Minecraft.getInstance();

        List<ItemStack> items = new ArrayList<>();
        page.getItemStacksForPage(items);
        if (!items.isEmpty()) {
            for (ItemStack stack : items) {
                if (StackUtil.isValid(stack)) {
                    List<String> tooltip = stack.getTooltipLines(mc.player, mc.options.advancedItemTooltips
                        ? ITooltipFlag.TooltipFlags.ADVANCED
                        : ITooltipFlag.TooltipFlags.NORMAL);
                    for (String strg : tooltip) {
                        if (strg != null && strg.toLowerCase(Locale.ROOT).contains(searchBarText)) {
                            return true;
                        }
                    }
                }
            }
        }

        List<FluidStack> fluids = new ArrayList<>();
        page.getFluidStacksForPage(fluids);
        if (!fluids.isEmpty()) {
            for (FluidStack stack : fluids) {
                if (stack != null) {
                    String strg = stack.getLocalizedName();
                    if (strg != null && strg.toLowerCase(Locale.ROOT).contains(searchBarText)) {
                        return true;
                    }
                }
            }
        }

        return false;
    }

    @Override
    public List<IBookletChapter> getAllChapters() {
        return this.chapters;
    }

    @Override
    public String getIdentifier() {
        return this.identifier;
    }

    @Override
    @OnlyIn(Dist.CLIENT)
    public String getLocalizedName() {
        return StringUtil.localize("booklet." + ActuallyAdditions.MODID + ".indexEntry." + this.getIdentifier() + ".name");
    }

    @Override
    @OnlyIn(Dist.CLIENT)
    public String getLocalizedNameWithFormatting() {
        return this.color + this.getLocalizedName();
    }

    @Override
    public void addChapter(IBookletChapter chapter) {
        this.chapters.add(chapter);
    }

    @Override
    @OnlyIn(Dist.CLIENT)
    public List<IBookletChapter> getChaptersForDisplay(String searchBarText) {
        if (searchBarText != null && !searchBarText.isEmpty()) {
            String search = searchBarText.toLowerCase(Locale.ROOT);

            List<IBookletChapter> fittingChapters = new ArrayList<>();
            for (IBookletChapter chapter : this.getAllChapters()) {
                if (chapter.getLocalizedName().toLowerCase(Locale.ROOT).contains(search)) {
                    fittingChapters.add(chapter);
                } else {
                    for (IBookletPage page : chapter.getAllPages()) {
                        if (fitsFilter(page, search)) {
                            fittingChapters.add(chapter);
                            break;
                        }
                    }
                }
            }

            return fittingChapters;
        } else {
            return this.getAllChapters();
        }
    }

    @Override
    public int getSortingPriority() {
        return this.priority;
    }

    @Override
    @OnlyIn(Dist.CLIENT)
    public boolean visibleOnFrontPage() {
        return true;
    }

    public BookletEntry setImportant() {
        this.color = TextFormatting.DARK_GREEN;
        return this;
    }

    public BookletEntry setSpecial() {
        this.color = TextFormatting.DARK_PURPLE;
        return this;
    }

}