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.fluids.FluidStack;
import net.minecraftforge.fml.relauncher.Side;
import net.minecraftforge.fml.relauncher.SideOnly;

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<IBookletChapter>();
    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;
    }

    @SideOnly(Side.CLIENT)
    private static boolean fitsFilter(IBookletPage page, String searchBarText){
        Minecraft mc = Minecraft.getMinecraft();

        List<ItemStack> items = new ArrayList<ItemStack>();
        page.getItemStacksForPage(items);
        if(!items.isEmpty()){
            for(ItemStack stack : items){
                if(StackUtil.isValid(stack)){
                    List<String> tooltip = stack.getTooltip(mc.player, mc.gameSettings.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<FluidStack>();
        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
    @SideOnly(Side.CLIENT)
    public String getLocalizedName(){
        return StringUtil.localize("booklet."+ActuallyAdditions.MODID+".indexEntry."+this.getIdentifier()+".name");
    }

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

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

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

            List<IBookletChapter> fittingChapters = new ArrayList<IBookletChapter>();
            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
    @SideOnly(Side.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;
    }

}