CC-Tweaked/src/main/java/dan200/computercraft/shared/peripheral/commandblock/CommandBlockPeripheral.java

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

package dan200.computercraft.shared.peripheral.commandblock;

import dan200.computercraft.api.lua.ILuaContext;
import dan200.computercraft.api.lua.LuaException;
import dan200.computercraft.api.peripheral.IComputerAccess;
import dan200.computercraft.api.peripheral.IPeripheral;
import net.minecraft.tileentity.TileEntityCommandBlock;

import javax.annotation.Nonnull;

import static dan200.computercraft.core.apis.ArgumentHelper.getString;

public class CommandBlockPeripheral implements IPeripheral
{
    private final TileEntityCommandBlock m_commandBlock;

    public CommandBlockPeripheral( TileEntityCommandBlock commandBlock )
    {
        m_commandBlock = commandBlock;
    }

    // IPeripheral methods

    @Nonnull
    @Override
    public String getType()
    {
        return "command";
    }

    @Nonnull
    @Override
    public String[] getMethodNames()
    {
        return new String[] {
            "getCommand",
            "setCommand",
            "runCommand",
        };
    }

    @Override
    public Object[] callMethod( @Nonnull IComputerAccess computer, @Nonnull ILuaContext context, int method, @Nonnull final Object[] arguments ) throws LuaException, InterruptedException
    {
        switch( method )
        {
            case 0: // getCommand
                return context.executeMainThreadTask( () -> new Object[] {
                    m_commandBlock.getCommandBlockLogic().getCommand(),
                } );
            case 1:
            {
                // setCommand
                final String command = getString( arguments, 0 );
                context.issueMainThreadTask( () ->
                {
                    m_commandBlock.getCommandBlockLogic().setCommand( command );
                    m_commandBlock.getCommandBlockLogic().updateCommand();
                    return null;
                } );
                return null;
            }
            case 2: // runCommand
                return context.executeMainThreadTask( () ->
                {
                    m_commandBlock.getCommandBlockLogic().trigger( m_commandBlock.getWorld() );
                    int result = m_commandBlock.getCommandBlockLogic().getSuccessCount();
                    if( result > 0 )
                    {
                        return new Object[] { true };
                    }
                    else
                    {
                        return new Object[] { false, "Command failed" };
                    }
                } );
        }
        return null;
    }

    @Override
    public boolean equals( IPeripheral other )
    {
        return other != null && other.getClass() == getClass();
    }
}