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-2016 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.api.booklet.internal.GuiBookletBase;
import de.ellpeck.actuallyadditions.mod.util.ModUtil;
import de.ellpeck.actuallyadditions.mod.util.StringUtil;
import net.minecraft.client.Minecraft;
import net.minecraft.item.ItemStack;
import net.minecraft.util.text.TextFormatting;
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 List<IBookletChapter> chapters = new ArrayList<IBookletChapter>();
    private TextFormatting color;

    public BookletEntry(String identifier){
        this.identifier = identifier;
        ActuallyAdditionsAPI.addBookletEntry(this);

        this.color = TextFormatting.RESET;
    }

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

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

    @Override
    public String getLocalizedName(){
        return StringUtil.localize("booklet."+ModUtil.MOD_ID+".indexEntry."+this.getIdentifier()+".name");
    }

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

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

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

    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(stack != null){
                    List<String> tooltip = stack.getTooltip(mc.thePlayer, mc.gameSettings.advancedItemTooltips);
                    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;
    }

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

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

}