Arcturus-Community/src/main/java/com/eu/habbo/habbohotel/pets/HorsePet.java

package com.eu.habbo.habbohotel.pets;

import com.eu.habbo.Emulator;
import com.eu.habbo.habbohotel.users.Habbo;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

public class HorsePet extends Pet
{
    private int hairColor;
    private int hairStyle;

    private boolean hasSaddle;
    private boolean anyoneCanRide;

    private Habbo rider;

    public HorsePet(ResultSet set) throws SQLException
    {
        super(set);
        this.hairColor = set.getInt("hair_color");
        this.hairStyle = set.getInt("hair_style");
        this.hasSaddle = set.getString("saddle").equalsIgnoreCase("1");
        this.anyoneCanRide = set.getString("ride").equalsIgnoreCase("1");
        this.rider = null;
    }

    public HorsePet(int type, int race, String color, String name, int userId)
    {
        super(type, race, color, name, userId);
        this.hairColor = 0;
        this.hairStyle = -1;
        this.hasSaddle = false;
        this.anyoneCanRide = false;
        this.rider = null;
    }

    @Override
    public void run()
    {
        if(this.needsUpdate)
        {
            try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("UPDATE users_pets SET hair_style = ?, hair_color = ?, saddle = ?, ride = ? WHERE id = ?"))
            {
                statement.setInt(1, this.hairStyle);
                statement.setInt(2, this.hairColor);
                statement.setString(3, this.hasSaddle ? "1" : "0");
                statement.setString(4, this.anyoneCanRide ? "1" : "0");
                statement.setInt(5, super.getId());
                statement.execute();
            }
            catch (SQLException e)
            {
                Emulator.getLogging().logSQLException(e);
            }

            super.run();
        }
    }

    public int getHairColor()
    {
        return this.hairColor;
    }

    public void setHairColor(int hairColor)
    {
        this.hairColor = hairColor;
    }

    public int getHairStyle()
    {
        return this.hairStyle;
    }

    public void setHairStyle(int hairStyle)
    {
        this.hairStyle = hairStyle;
    }

    public boolean hasSaddle()
    {
        return this.hasSaddle;
    }

    public void hasSaddle(boolean hasSaddle)
    {
        this.hasSaddle = hasSaddle;
    }

    public boolean anyoneCanRide()
    {
        return this.anyoneCanRide;
    }

    public void setAnyoneCanRide(boolean anyoneCanRide)
    {
        this.anyoneCanRide = anyoneCanRide;
    }

    public Habbo getRider()
    {
        return this.rider;
    }

    public void setRider(Habbo rider)
    {
        this.rider = rider;
    }
}