CC-Tweaked/src/main/java/dan200/computercraft/shared/media/items/ItemPrintout.java

/**
 * This file is part of ComputerCraft - http://www.computercraft.info
 * Copyright Daniel Ratcliffe, 2011-2016. Do not distribute without permission.
 * Send enquiries to dratcliffe@gmail.com
 */

package dan200.computercraft.shared.media.items;

import dan200.computercraft.ComputerCraft;
import net.minecraft.creativetab.CreativeTabs;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.util.ActionResult;
import net.minecraft.util.EnumActionResult;
import net.minecraft.util.EnumHand;
import net.minecraft.world.World;

import java.util.List;

public class ItemPrintout extends Item
{
    public static final int LINES_PER_PAGE = 21;
    public static final int LINE_MAX_LENGTH = 25;
    public static final int MAX_PAGES = 16;

    public enum Type
    {
        Single,
        Multiple,
        Book
    }

    public ItemPrintout()
    {
        setMaxStackSize( 1 );
        setHasSubtypes( true );
        setUnlocalizedName( "computercraft:page" );
        setCreativeTab( ComputerCraft.mainCreativeTab );
    }

    @Override
    public void getSubItems( Item itemID, CreativeTabs tabs, List list )
    {
        list.add( createSingleFromTitleAndText( null, new String[ LINES_PER_PAGE ], new String[ LINES_PER_PAGE ] ) );
        list.add( createMultipleFromTitleAndText( null, new String[ 2*LINES_PER_PAGE ], new String[ 2*LINES_PER_PAGE ] ) );
        list.add( createBookFromTitleAndText( null, new String[ 2*LINES_PER_PAGE ], new String[ 2*LINES_PER_PAGE ] ) );
    }

    @Override
    public void addInformation( ItemStack itemstack, EntityPlayer par2EntityPlayer, List list, boolean flag )
    {
        String title = getTitle( itemstack );
        if( title != null && title.length() > 0 )
        {
            list.add( title );
        }
    }

    @Override
    public String getUnlocalizedName( ItemStack stack )
    {
        Type type = getType( stack );
        switch( type )
        {
            case Single:
            default:
            {
                return "item.computercraft:page";
            }
            case Multiple:
            {
                return "item.computercraft:pages";
            }
            case Book:
            {
                return "item.computercraft:book";
            }
        }
    }

    @Override
    public ActionResult<ItemStack> onItemRightClick( ItemStack stack, World world, EntityPlayer player, EnumHand hand )
    {
        if( !world.isRemote )
        {
            ComputerCraft.openPrintoutGUI( player, hand );
        }
        return new ActionResult<ItemStack>( EnumActionResult.SUCCESS, stack );
    }

    private static ItemStack createFromTitleAndText( Type type, String title, String[] text, String[] colours )
    {
        // Calculate damage
        int damage;
        switch( type )
        {
            case Single:
            default:
            {
                damage = 0;
                break;
            }
            case Multiple:
            {
                damage = 1;
                break;
            }
            case Book:
            {
                damage = 2;
                break;
            }
        }

        // Create stack
        ItemStack stack = new ItemStack( ComputerCraft.Items.printout, 1, damage );

        // Build NBT
        NBTTagCompound nbt = new NBTTagCompound();
        if( title != null )
        {
            nbt.setString( "title", title );
        }
        if( text != null )
        {
            nbt.setInteger( "pages", text.length / LINES_PER_PAGE );
            for(int i=0; i<text.length; ++i)
            {
                if( text[i] != null )
                {
                    nbt.setString( "line"+i, text[i] );
                }
            }
        }
        if( colours != null )
        {
            for(int i=0; i<colours.length; ++i)
            {
                if( colours[i] != null )
                {
                    nbt.setString( "colour"+i, colours[i] );
                }
            }
        }
        stack.setTagCompound( nbt );

        // Return stack
        return stack;
    }

    public static ItemStack createSingleFromTitleAndText( String title, String[] text, String[] colours )
    {
        return createFromTitleAndText( Type.Single, title, text, colours );
    }

    public static ItemStack createMultipleFromTitleAndText( String title, String[] text, String[] colours )
    {
        return createFromTitleAndText( Type.Multiple, title, text, colours );
    }

    public static ItemStack createBookFromTitleAndText( String title, String[] text, String[] colours )
    {
        return createFromTitleAndText( Type.Book, title, text, colours );
    }

    public static Type getType( ItemStack stack )
    {
        int damage = stack.getItemDamage();
        switch( damage )
        {
            case 0:
            default:
            {
                return Type.Single;
            }
            case 1:
            {
                return Type.Multiple;
            }
            case 2:
            {
                return Type.Book;
            }
        }
    }

    public static String getTitle( ItemStack stack )
    {
        NBTTagCompound nbt = stack.getTagCompound();
        if( nbt != null && nbt.hasKey( "title" ) )
        {
            return nbt.getString( "title" );
        }
        return null;
    }

    public static int getPageCount( ItemStack stack )
    {
        NBTTagCompound nbt = stack.getTagCompound();
        if( nbt != null && nbt.hasKey( "pages" ) )
        {
            return nbt.getInteger( "pages" );
        }
        return 1;
    }

    public static String[] getText( ItemStack stack )
    {
        NBTTagCompound nbt = stack.getTagCompound();
        int numLines = getPageCount( stack ) * LINES_PER_PAGE;
        String[] lines = new String[numLines];
        for( int i=0; i<lines.length; ++i )
        {
            if( nbt != null )
            {
                lines[i] = nbt.getString( "line"+i );
            }
            else
            {
                lines[i] = "";
            }
        }
        return lines;
    }

    public static String[] getColours( ItemStack stack )
    {
        NBTTagCompound nbt = stack.getTagCompound();
        int numLines = getPageCount( stack ) * LINES_PER_PAGE;
        String[] lines = new String[numLines];
        for( int i=0; i<lines.length; ++i )
        {
            if( nbt != null )
            {
                lines[i] = nbt.getString( "colour"+i );
            }
            else
            {
                lines[i] = "";
            }
        }
        return lines;
    }
}