/*
 * This file is part of OpenTTD.
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 */

/** @file script_company.hpp Everything to query a company's financials and statistics or build company related buildings. */

#ifndef SCRIPT_COMPANY_HPP
#define SCRIPT_COMPANY_HPP

#include "script_text.hpp"
#include "../../economy_type.h"
#include "../../livery.h"
#include "../../gfx_type.h"

/**
 * Class that handles all company related functions.
 * @api ai game
 */
class ScriptCompany : public ScriptObject {
public:
	/** The range of possible quarters to get company information of. */
	enum Quarter {
		CURRENT_QUARTER  = 0,                      ///< The current quarter.
		EARLIEST_QUARTER = ::MAX_HISTORY_QUARTERS, ///< The earliest quarter company information is available for.
	};

	/** Different constants related to CompanyID. */
	enum CompanyID {
		/* Note: these values represent part of the in-game Owner enum */
		COMPANY_FIRST     = ::CompanyID::Begin().base(), ///< The first available company.
		COMPANY_LAST      = ::CompanyID::End().base(),   ///< The last available company.

		/* Custom added value, only valid for this API */
		COMPANY_INVALID   = -1,                ///< An invalid company.
		COMPANY_SELF      = 254,               ///< Constant that gets resolved to the correct company index for your company.
		COMPANY_SPECTATOR = 255,               ///< Constant indicating that player is spectating (gets resolved to COMPANY_INVALID)
	};

	/**
	 * Internal helper to convert from the script's company to the game's internal company.
	 * @api none
	 */
	static ::CompanyID FromScriptCompanyID(ScriptCompany::CompanyID company);

	/**
	 * Internal helper to convert from the game's internal company to the script's company.
	 * @api none
	 */
	static ScriptCompany::CompanyID ToScriptCompanyID(::CompanyID company);

	/** Possible genders for company presidents. */
	enum Gender {
		GENDER_MALE,         ///< A male person.
		GENDER_FEMALE,       ///< A female person.
		GENDER_INVALID = -1, ///< An invalid gender.
	};

	/** List of different livery schemes. */
	enum LiveryScheme {
		LS_DEFAULT,                  ///< Default scheme.
		LS_STEAM,                    ///< Steam engines.
		LS_DIESEL,                   ///< Diesel engines.
		LS_ELECTRIC,                 ///< Electric engines.
		LS_MONORAIL,                 ///< Monorail engines.
		LS_MAGLEV,                   ///< Maglev engines.
		LS_DMU,                      ///< DMUs and their passenger wagons.
		LS_EMU,                      ///< EMUs and their passenger wagons.
		LS_PASSENGER_WAGON_STEAM,    ///< Passenger wagons attached to steam engines.
		LS_PASSENGER_WAGON_DIESEL,   ///< Passenger wagons attached to diesel engines.
		LS_PASSENGER_WAGON_ELECTRIC, ///< Passenger wagons attached to electric engines.
		LS_PASSENGER_WAGON_MONORAIL, ///< Passenger wagons attached to monorail engines.
		LS_PASSENGER_WAGON_MAGLEV,   ///< Passenger wagons attached to maglev engines.
		LS_FREIGHT_WAGON,            ///< Freight wagons.
		LS_BUS,                      ///< Buses.
		LS_TRUCK,                    ///< Trucks.
		LS_PASSENGER_SHIP,           ///< Passenger ships.
		LS_FREIGHT_SHIP,             ///< Freight ships.
		LS_HELICOPTER,               ///< Helicopters.
		LS_SMALL_PLANE,              ///< Small aeroplanes.
		LS_LARGE_PLANE,              ///< Large aeroplanes.
		LS_PASSENGER_TRAM,           ///< Passenger trams.
		LS_FREIGHT_TRAM,             ///< Freight trams.
		LS_INVALID = -1,
	};

	/** List of colours. */
	enum Colours {
		COLOUR_DARK_BLUE,
		COLOUR_PALE_GREEN,
		COLOUR_PINK,
		COLOUR_YELLOW,
		COLOUR_RED,
		COLOUR_LIGHT_BLUE,
		COLOUR_GREEN,
		COLOUR_DARK_GREEN,
		COLOUR_BLUE,
		COLOUR_CREAM,
		COLOUR_MAUVE,
		COLOUR_PURPLE,
		COLOUR_ORANGE,
		COLOUR_BROWN,
		COLOUR_GREY,
		COLOUR_WHITE,
		COLOUR_INVALID = ::INVALID_COLOUR
	};

	/**
	 * Types of expenses.
	 * @api -ai
	 */
	enum ExpensesType : uint8_t {
		EXPENSES_CONSTRUCTION = ::EXPENSES_CONSTRUCTION,     ///< Construction costs.
		EXPENSES_NEW_VEHICLES = ::EXPENSES_NEW_VEHICLES,     ///< New vehicles.
		EXPENSES_TRAIN_RUN    = ::EXPENSES_TRAIN_RUN,        ///< Running costs trains.
		EXPENSES_ROADVEH_RUN  = ::EXPENSES_ROADVEH_RUN,      ///< Running costs road vehicles.
		EXPENSES_AIRCRAFT_RUN = ::EXPENSES_AIRCRAFT_RUN,     ///< Running costs aircraft.
		EXPENSES_SHIP_RUN     = ::EXPENSES_SHIP_RUN,         ///< Running costs ships.
		EXPENSES_PROPERTY     = ::EXPENSES_PROPERTY,         ///< Property costs.
		EXPENSES_TRAIN_INC    = ::EXPENSES_TRAIN_REVENUE,    ///< Revenue from trains.
		EXPENSES_ROADVEH_INC  = ::EXPENSES_ROADVEH_REVENUE,  ///< Revenue from road vehicles.
		EXPENSES_AIRCRAFT_INC = ::EXPENSES_AIRCRAFT_REVENUE, ///< Revenue from aircraft.
		EXPENSES_SHIP_INC     = ::EXPENSES_SHIP_REVENUE,     ///< Revenue from ships.
		EXPENSES_LOAN_INT     = ::EXPENSES_LOAN_INTEREST,    ///< Interest payments over the loan.
		EXPENSES_OTHER        = ::EXPENSES_OTHER,            ///< Other expenses.
		EXPENSES_INVALID      = ::INVALID_EXPENSES,          ///< Invalid expense type.
	};

	/**
	 * Resolved the given company index to the correct index for the company. If
	 *  the company index was COMPANY_SELF it will be resolved to the index of
	 *  your company. If the company with the given index does not exist it will
	 *  return COMPANY_INVALID.
	 * @param company The company index to resolve.
	 * @return The resolved company index.
	 */
	static ScriptCompany::CompanyID ResolveCompanyID(ScriptCompany::CompanyID company);

	/**
	 * Check if a CompanyID is your CompanyID, to ease up checks.
	 * @param company The company index to check.
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @return True if and only if this company is your CompanyID.
	 */
	static bool IsMine(ScriptCompany::CompanyID company);

	/**
	 * Set the name of your company.
	 * @param name The new name of the company (can be either a raw string, or a ScriptText object).
	 * @pre name != null && len(name) != 0.
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @exception ScriptError::ERR_NAME_IS_NOT_UNIQUE
	 * @return True if the name was changed.
	 */
	static bool SetName(Text *name);

	/**
	 * Get the name of the given company.
	 * @param company The company to get the name for.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @return The name of the given company.
	 */
	static std::optional<std::string> GetName(ScriptCompany::CompanyID company);

	/**
	 * Set the name of your president.
	 * @param name The new name of the president (can be either a raw string, or a ScriptText object).
	 * @pre name != null && len(name) != 0.
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @exception ScriptError::ERR_NAME_IS_NOT_UNIQUE
	 * @return True if the name was changed.
	 */
	static bool SetPresidentName(Text *name);

	/**
	 * Get the name of the president of the given company.
	 * @param company The company to get the president's name for.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @return The name of the president of the given company.
	 */
	static std::optional<std::string> GetPresidentName(ScriptCompany::CompanyID company);

	/**
	 * Set the gender of the president of your company.
	 * @param gender The new gender for your president.
	 * @pre GetPresidentGender(ScriptCompany.COMPANY_SELF) != gender.
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @return True if the gender was changed.
	 * @note When successful a random face will be created.
	 */
	static bool SetPresidentGender(Gender gender);

	/**
	 * Get the gender of the president of the given company.
	 * @param company The company to get the presidents gender off.
	 * @return The gender of the president.
	 */
	static Gender GetPresidentGender(ScriptCompany::CompanyID company);

	/**
	 * Sets the amount to loan.
	 * @param loan The amount to loan (multiplier of GetLoanInterval()).
	 * @pre 'loan' must be non-negative.
	 * @pre GetLoanInterval() must be a multiplier of 'loan'.
	 * @pre 'loan' must be below GetMaxLoanAmount().
	 * @pre 'loan' - GetLoanAmount() + GetBankBalance() must be non-negative.
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @return True if the loan could be set to your requested amount.
	 */
	static bool SetLoanAmount(Money loan);

	/**
	 * Sets the minimum amount to loan, i.e. the given amount of loan rounded up.
	 * @param loan The amount to loan (any positive number).
	 * @pre 'loan' must be non-negative.
	 * @pre 'loan' must be below GetMaxLoanAmount().
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @return True if we could allocate a minimum of 'loan' loan.
	 */
	static bool SetMinimumLoanAmount(Money loan);

	/**
	 * Gets the amount your company have loaned.
	 * @return The amount loaned money.
	 * @post GetLoanInterval() is always a multiplier of the return value.
	 */
	static Money GetLoanAmount();

	/**
	 * Gets the maximum amount your company can loan. In deity mode returns the global max loan.
	 * @return The maximum amount your company can loan.
	 * @post GetLoanInterval() is always a multiplier of the return value.
	 */
	static Money GetMaxLoanAmount();

	/**
	 * Sets the max amount of money company can loan.
	 * @param company The company ID.
	 * @param amount Max loan amount. Will be rounded down to a multiple of GetLoanInterval().
	 * @return True, if the max loan was changed.
	 * @pre ScriptCompanyMode::IsDeity().
	 * @pre amount >= 0.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @note You need to create your own news message to inform about max loan change.
	 * @note Max loan value set with this method is not affected by inflation.
	 * @api -ai
	 */
	static bool SetMaxLoanAmountForCompany(ScriptCompany::CompanyID company, Money amount);

	/**
	 * Makes the max amount of money company can loan follow the global max loan setting.
	 * @param company The company ID.
	 * @return True, if the max loan was reset.
	 * @pre ScriptCompanyMode::IsDeity().
	 * @pre amount >= 0 && amount <= MAX_LOAN_LIMIT.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @note You need to create your own news message to inform about max loan change.
	 * @api -ai
	 */
	static bool ResetMaxLoanAmountForCompany(ScriptCompany::CompanyID company);

	/**
	 * Gets the interval/loan step.
	 * @return The loan step.
	 * @post Return value is always positive.
	 */
	static Money GetLoanInterval();

	/**
	 * Gets the bank balance. In other words, the amount of money the given company can spent.
	 * If infinite money is enabled, it returns INT32_MAX.
	 * @param company The company to get the bank balance of.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @return The actual bank balance or INT32_MAX.
	 */
	static Money GetBankBalance(ScriptCompany::CompanyID company);

	/**
	 * Changes the bank balance by a delta value. This method does not affect the loan but instead
	 * allows a GS to give or take money from a company.
	 * @param company The company to change the bank balance of.
	 * @param delta Amount of money to give or take from the bank balance. A positive value adds money to the bank balance.
	 * @param expenses_type The account in the finances window that will register the cost.
	 * @param tile The tile to show text effect on or ScriptMap::TILE_INVALID
	 * @return True, if the bank balance was changed.
	 * @game @pre ScriptCompanyMode::IsDeity().
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @note You need to create your own news message to inform about costs/gifts that you create using this command.
	 * @api -ai
	 */
	static bool ChangeBankBalance(ScriptCompany::CompanyID company, Money delta, ExpensesType expenses_type, TileIndex tile);

	/**
	 * Get the income of the company in the given economy-quarter.
	 * Note that this function only considers recurring income from vehicles;
	 * it does not include one-time income from selling stuff.
	 * @param company The company to get the quarterly income of.
	 * @param quarter The economy-quarter to get the income of.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @pre quarter <= EARLIEST_QUARTER.
	 * @return The gross income of the company in the given economy-quarter.
	 * @see \ref ScriptEconomyTime
	 */
	static Money GetQuarterlyIncome(ScriptCompany::CompanyID company, SQInteger quarter);

	/**
	 * Get the expenses of the company in the given economy-quarter.
	 * Note that this function only considers recurring expenses from vehicle
	 * running cost, maintenance and interests; it does not include one-time
	 * expenses from construction and buying stuff.
	 * @param company The company to get the quarterly expenses of.
	 * @param quarter The economy-quarter to get the expenses of.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @pre quarter <= EARLIEST_QUARTER.
	 * @return The expenses of the company in the given economy-quarter.
	 * @see \ref ScriptEconomyTime
	 */
	static Money GetQuarterlyExpenses(ScriptCompany::CompanyID company, SQInteger quarter);

	/**
	 * Get the amount of cargo delivered by the given company in the given economy-quarter.
	 * @param company The company to get the amount of delivered cargo of.
	 * @param quarter The economy-quarter to get the amount of delivered cargo of.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @pre quarter <= EARLIEST_QUARTER.
	 * @return The amount of cargo delivered by the given company in the given economy-quarter.
	 * @see \ref ScriptEconomyTime
	 */
	static SQInteger GetQuarterlyCargoDelivered(ScriptCompany::CompanyID company, SQInteger quarter);

	/**
	 * Get the performance rating of the given company in the given economy-quarter.
	 * @param company The company to get the performance rating of.
	 * @param quarter The economy-quarter to get the performance rating of.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @pre quarter <= EARLIEST_QUARTER.
	 * @pre quarter != CURRENT_QUARTER.
	 * @note The performance rating is calculated after every economy-quarter, so the value for CURRENT_QUARTER is undefined.
	 * @return The performance rating of the given company in the given economy-quarter.
	 * @see \ref ScriptEconomyTime
	 */
	static SQInteger GetQuarterlyPerformanceRating(ScriptCompany::CompanyID company, SQInteger quarter);

	/**
	 * Get the value of the company in the given economy-quarter.
	 * @param company The company to get the value of.
	 * @param quarter The economy-quarter to get the value of.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @pre quarter <= EARLIEST_QUARTER.
	 * @return The value of the company in the given economy-quarter.
	 * @see \ref ScriptEconomyTime
	 */
	static Money GetQuarterlyCompanyValue(ScriptCompany::CompanyID company, SQInteger quarter);

	/**
	 * Build your company's HQ on the given tile.
	 * @param tile The tile to build your HQ on, this tile is the most northern tile of your HQ.
	 * @pre ScriptMap::IsValidTile(tile).
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @exception ScriptError::ERR_AREA_NOT_CLEAR
	 * @exception ScriptError::ERR_FLAT_LAND_REQUIRED
	 * @return True if the HQ could be build.
	 * @note An HQ can not be removed, only by water or rebuilding; If an HQ is
	 *  build again, the old one is removed.
	 */
	static bool BuildCompanyHQ(TileIndex tile);

	/**
	 * Return the location of a company's HQ.
	 * @param company The company the get the HQ of.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @return The tile of the company's HQ, this tile is the most northern tile
	 *  of that HQ, or ScriptMap::TILE_INVALID if there is no HQ yet.
	 */
	static TileIndex GetCompanyHQ(ScriptCompany::CompanyID company);

	/**
	 * Set whether autorenew is enabled for your company.
	 * @param autorenew The new autorenew status.
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @return True if autorenew status has been modified.
	 */
	static bool SetAutoRenewStatus(bool autorenew);

	/**
	 * Return whether autorenew is enabled for a company.
	 * @param company The company to get the autorenew status of.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @return True if autorenew is enabled.
	 */
	static bool GetAutoRenewStatus(ScriptCompany::CompanyID company);

	/**
	 * Set the number of months before/after max age to autorenew an engine for your company.
	 * @param months The number of calendar-months before/after max age of engine.
	 *               The value will be clamped to MIN(int16_t) .. MAX(int16_t).
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @return True if autorenew months has been modified.
	 * @see \ref ScriptCalendarTime
	 */
	static bool SetAutoRenewMonths(SQInteger months);

	/**
	 * Return the number of months before/after max age to autorenew an engine for a company.
	 * @param company The company to get the autorenew months of.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @return The number of calendar-months before/after max age of engine.
	 * @see \ref ScriptCalendarTime
	 */
	static SQInteger GetAutoRenewMonths(ScriptCompany::CompanyID company);

	/**
	 * Set the minimum money needed to autorenew an engine for your company.
	 * @param money The new minimum required money for autorenew to work.
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @return True if autorenew money has been modified.
	 * @pre money >= 0
	 * @pre money <  2**32
	 */
	static bool SetAutoRenewMoney(Money money);

	/**
	 * Return the minimum money needed to autorenew an engine for a company.
	 * @param company The company to get the autorenew money of.
	 * @pre ResolveCompanyID(company) != COMPANY_INVALID.
	 * @return The minimum required money for autorenew to work.
	 */
	static Money GetAutoRenewMoney(ScriptCompany::CompanyID company);

	/**
	 * Set primary colour for your company.
	 * @param scheme Livery scheme to set.
	 * @param colour Colour to set.
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @return False if unable to set primary colour of the livery scheme (e.g. colour in use).
	 */
	static bool SetPrimaryLiveryColour(LiveryScheme scheme, Colours colour);

	/**
	 * Set secondary colour for your company.
	 * @param scheme Livery scheme to set.
	 * @param colour Colour to set.
	 * @game @pre ScriptCompanyMode::IsValid().
	 * @return False if unable to set secondary colour of the livery scheme.
	 */
	static bool SetSecondaryLiveryColour(LiveryScheme scheme, Colours colour);

	/**
	 * Get primary colour of a livery for your company.
	 * @param scheme Livery scheme to get.
	 * @return Primary colour of livery.
	 */
	static ScriptCompany::Colours GetPrimaryLiveryColour(LiveryScheme scheme);

	/**
	 * Get secondary colour of a livery for your company.
	 * @param scheme Livery scheme to get.
	 * @return Secondary colour of livery.
	 */
	static ScriptCompany::Colours GetSecondaryLiveryColour(LiveryScheme scheme);
};

DECLARE_INCREMENT_DECREMENT_OPERATORS(ScriptCompany::CompanyID)

#endif /* SCRIPT_COMPANY_HPP */
