CustomNPC-Plus/src/main/java/noppes/npcs/scripted/roles/ScriptRoleFollower.java

package noppes.npcs.scripted.roles;

import net.minecraft.entity.player.EntityPlayer;
import noppes.npcs.api.entity.IPlayer;
import noppes.npcs.api.roles.IRoleFollower;
import noppes.npcs.entity.EntityNPCInterface;
import noppes.npcs.roles.RoleFollower;
import noppes.npcs.scripted.NpcAPI;
import noppes.npcs.scripted.constants.RoleType;

public class ScriptRoleFollower extends ScriptRoleInterface implements IRoleFollower {
    private RoleFollower role;

    public ScriptRoleFollower(EntityNPCInterface npc) {
        super(npc);
        role = (RoleFollower) npc.roleInterface;
    }

    public void setOwner(IPlayer player) {
        if (player == null || player.getMCEntity() == null) {
            role.setOwner(null);
            return;
        }
        EntityPlayer mcplayer = (EntityPlayer) player.getMCEntity();
        role.setOwner(mcplayer);
    }

    public IPlayer getOwner() {
        if (role.owner == null)
            return null;

        return (IPlayer) NpcAPI.Instance().getIEntity(role.owner);
    }

    public boolean hasOwner() {
        return role.owner != null;
    }

    public boolean isFollowing() {
        return role.isFollowing();
    }

    public void setIsFollowing(boolean following) {
        if (role.owner != null && role.getDaysLeft() > 0) role.isFollowing = following;
    }

    public int getDaysLeft() {
        return role.getDaysLeft();
    }

    public void addDaysLeft(int days) {
        role.addDays(days);
    }

    public boolean getInfiniteDays() {
        return role.infiniteDays;
    }

    public void setInfiniteDays(boolean infinite) {
        role.infiniteDays = infinite;
    }

    public boolean getGuiDisabled() {
        return role.disableGui;
    }

    public void setGuiDisabled(boolean disabled) {
        role.disableGui = disabled;
    }


    /**
     * @param index  Index of Rate [0 - 2]
     * @param amount Amount hired for
     */
    public void setRate(int index, int amount) {
        role.setRate(index, amount);
    }

    /**
     * @param index Index of Rate [0 - 2]
     * @return amount fired for
     */
    public int getRate(int index) {
        return role.getRate(index);
    }

    /**
     * @param dialogHire New dialog hire string
     */
    public void setDialogHire(String dialogHire) {
        role.setDialogHire(dialogHire);
    }

    /**
     * @return dialog hire string
     */
    public String getDialogHire() {
        return role.getDialogHire();
    }

    /**
     * @param dialogFarewell New dialog farewell string
     */
    public void setDialogFarewell(String dialogFarewell) {
        role.setDialogFarewell(dialogFarewell);
    }

    /**
     * @return dialog farewell string
     */
    public String getDialogFarewell() {
        return role.getDialogFarewell();
    }

    @Override
    public int getType() {
        return RoleType.FOLLOWER;
    }
}