add buy slip functionality with initial market maker ... work on /shares

gamemodes/irresistible/cnr/features/stocks/stocks.pwn

@@ -16,6 +16,9 @@
 #define STOCK_REPORTING_PERIODS 	( 30 ) // last 30 periods (days)
 
 #define DIALOG_STOCK_MARKET 		8923
+#define DIALOG_STOCK_MARKET_BUY 	8925
+
+#define STOCK_MM_USER_ID			( 0 )
 
 /* ** Variables ** */
 enum E_STOCK_MARKET_DATA
@@ -23,7 +26,7 @@ enum E_STOCK_MARKET_DATA
 	E_NAME[ 64 ],			E_SYMBOL[ 4 ],			E_MAX_SHARES,
 
 	// market maker
-	E_MM_SHARES,			E_MM_IPO_PRICE
+	Float: E_IPO_SHARES,			Float: E_IPO_PRICE
 };
 
 enum E_STOCK_MARKET_PRICE_DATA
@@ -38,6 +41,11 @@ static stock
 	Iterator: stockmarkets 			< MAX_STOCKS >
 ;
 
+/* ** Forwards / Getters ** */
+stock Float: StockMarket_GetCurrentPrice( stockid ) {
+	return g_stockMarketPriceData[ stockid ] [ 0 ] [ E_PRICE ];
+}
+
 /* ** Hooks ** */
 hook OnScriptInit( )
 {
@@ -45,7 +53,7 @@ hook OnScriptInit( )
 	AddServerVariable( "stock_report_time", "0", GLOBAL_VARTYPE_INT );
 
 	// create markets
-	CreateStockMarket( 0, "The Mining Company", "MC", 50000000, 100 );
+	CreateStockMarket( 0, "The Mining Company", "MC", 1000000, 100 );
 	return 1;
 }
 
@@ -71,6 +79,51 @@ hook OnServerUpdate( )
 	return 1;
 }
 
+hook OnDialogResponse( playerid, dialogid, response, listitem, inputtext[ ] )
+{
+	if ( dialogid == DIALOG_STOCK_MARKET && response )
+	{
+		new
+			x = 0;
+
+		foreach ( new s : stockmarkets )
+		{
+			if ( x == listitem )
+			{
+				SetPVarInt( playerid, "stockmarket_selection", s );
+				StockMarket_ShowBuySlip( playerid, s );
+			}
+			x ++;
+		}
+		return 1;
+	}
+	else if ( dialogid == DIALOG_STOCK_MARKET_BUY )
+	{
+		new
+			stockid = GetPVarInt( playerid, "stockmarket_selection" );
+
+		if ( response )
+		{
+			new
+				Float: shares;
+
+			if ( sscanf( inputtext, "f", shares ) ) SendError( playerid, "You must use a valid value." );
+			else if ( shares <= 10.0 ) SendError( playerid, "The minimum number of shares you can buy is 10." );
+			else
+			{
+				mysql_tquery( dbHandle, sprintf( "SELECT * FROM `STOCK_OWNERS` WHERE `STOCK_ID` = %d AND `USER_ID` = %d", stockid, STOCK_MM_USER_ID ), "StockMarket_OnPurchaseOrder", "ddf", playerid, stockid, shares );
+				return 1;
+			}
+			return StockMarket_ShowBuySlip( playerid, stockid );
+		}
+		else
+		{
+			return cmd_stockmarkets( playerid, "" );
+		}
+	}
+	return 1;
+}
+
 /* ** SQL Thread ** */
 thread Stock_UpdateReportingPeriods( stockid )
 {
@@ -86,6 +139,17 @@ thread Stock_UpdateReportingPeriods( stockid )
 			g_stockMarketPriceData[ stockid ] [ row ] [ E_EARNINGS ] = cache_get_field_content_float( row, "CURRENT_EARNINGS" );
 		}
 	}
+	else // no historical reporting data, restock the market maker
+	{
+		// set current stock market prices to IPO
+		g_stockMarketPriceData[ stockid ] [ 1 ] [ E_PRICE ] = g_stockMarketData[ stockid ] [ E_IPO_PRICE ];
+
+		// create report for the company using the IPO price
+		StockMarket_CreateReport( stockid );
+
+		// give market maker shares
+		StockMarket_GiveShares( stockid, STOCK_MM_USER_ID, g_stockMarketData[ stockid ] [ E_IPO_SHARES ] ); // , g_stockMarketData[ stockid ] [ E_IPO_PRICE ]
+	}
 	return 1;
 }
 
@@ -95,6 +159,101 @@ thread StockMarket_InsertReport( stockid )
 	return 1;
 }
 
+thread StockMarket_OnPurchaseOrder( playerid, stockid, Float: shares )
+{
+	new
+		rows = cache_get_row_count( );
+
+	if ( ! rows ) {
+		return SendError( playerid, "This stock has no available shares for sale." );
+	}
+
+	// check if the player has the money for the purchase
+	new Float: ask_price = g_stockMarketPriceData[ stockid ] [ 1 ] [ E_PRICE ];
+	new purchase_cost = floatround( ask_price * shares );
+
+	if ( GetPlayerCash( playerid ) < purchase_cost ) {
+		return SendError( playerid, "You need at least %s to purchase this many shares.", cash_format( purchase_cost ) ), StockMarket_ShowBuySlip( playerid, stockid ), 1;
+	}
+
+	// check if quantity is valid
+	new Float: available_quantity = cache_get_field_content_float( 0, "SHARES" );
+
+	if ( shares > available_quantity ) {
+		return SendError( playerid, "There are not that many shares available for sale." ), StockMarket_ShowBuySlip( playerid, stockid ), 1;
+	}
+
+	// reduce the market makers shares
+	StockMarket_GiveShares( stockid, STOCK_MM_USER_ID, -shares );
+
+	// increment the players shares
+	StockMarket_GiveShares( stockid, GetPlayerAccountID( playerid ), shares );
+
+	// reduce player balance and alert
+	GivePlayerCash( playerid, -purchase_cost );
+	SendServerMessage( playerid, "You have successfully purchased %s shares of %s (@ %s/ea) for %s.", number_format( shares, .decimals = 3 ), g_stockMarketData[ stockid ] [ E_NAME ], cash_format( ask_price, .decimals = 2 ), cash_format( purchase_cost ) );
+	return 1;
+}
+
+thread StockMarket_OnShowBuySlip( playerid, stockid )
+{
+	new
+		rows = cache_get_row_count( );
+
+	if ( ! rows ) {
+		return SendError( playerid, "This stock does now have any shares available to buy." );
+	}
+
+	new
+		Float: available_quantity = cache_get_field_content_float( 0, "SHARES" );
+
+	format(
+		szBigString, sizeof ( szBigString ),
+		""COL_WHITE"You can buy shares of %s for "COL_GREEN"%s"COL_WHITE" each.\n\nThere are %s available shares to buy.",
+		g_stockMarketData[ stockid ] [ E_NAME ],
+		cash_format( g_stockMarketPriceData[ stockid ] [ 1 ] [ E_PRICE ], .decimals = 2 ),
+		number_format( available_quantity, .decimals = 3 )
+	);
+	ShowPlayerDialog( playerid, DIALOG_STOCK_MARKET_BUY, DIALOG_STYLE_INPUT, ""COL_WHITE"Stock Market", szBigString, "Buy", "Close" );
+	return 1;
+}
+
+thread StockMarket_OnShowShares( playerid )
+{
+	new
+		rows = cache_get_row_count( );
+
+	if ( ! rows ) {
+		return SendError( playerid, "This stock does now have any shares available to buy." );
+	}
+
+	szLargeString = ""COL_WHITE"Stock\t"COL_WHITE"Total Shares\t"COL_WHITE"Current Price ($)\t"COL_GREEN"Value ($)\n";
+
+	for ( new row = 0; row < rows; row ++ )
+	{
+		new
+			stockid = cache_get_field_content_int( row, "ID" );
+
+		if ( Iter_Contains( stockmarkets, stockid ) )
+		{
+			new Float: current_price = StockMarket_GetCurrentPrice( stockid );
+			new Float: shares = cache_get_field_content_float( row, "SHARES" );
+
+			format(
+				szLargeString, sizeof( szLargeString ),
+				"%s%s (%s)\t%s\t%s\t"COL_GREEN"%s\n",
+				szLargeString,
+				g_stockMarketData[ stockid ] [ E_NAME ],
+				g_stockMarketData[ stockid ] [ E_SYMBOL ],
+				number_format( shares, .decimals = 3 ),
+				cash_format( current_price, .decimals = 2 ),
+				cash_format( floatround( shares * current_price ) )
+			);
+		}
+	}
+	return ShowPlayerDialog( playerid, DIALOG_STOCK_MARKET, DIALOG_STYLE_TABLIST_HEADERS, ""COL_WHITE"Stock Market", szLargeString, "Sell", "Close" ), 1;
+}
+
 /* ** Command ** */
 CMD:increase( playerid, params[ ] ) {
 	StockMarket_UpdateEarnings( 0, strval( params ) );
@@ -106,9 +265,10 @@ CMD:newreport( playerid, params[ ] ) {
 	return 1;
 }
 
+CMD:stocks( playerid, params[ ] ) return cmd_stockmarkets( playerid, params );
 CMD:stockmarkets( playerid, params[ ] )
 {
-	szLargeString = ""COL_WHITE"Stock\t"COL_WHITE"Current Price\t"COL_WHITE"Market Capitalization\t"COL_WHITE"Price Change (24H)\n";
+	szLargeString = ""COL_WHITE"Stock\t"COL_WHITE"Current Price\t"COL_WHITE"Price Change (24H)\n";
 
 	foreach ( new s : stockmarkets )
 	{
@@ -129,24 +289,30 @@ CMD:stockmarkets( playerid, params[ ] )
 	}
 
 	SendServerMessage( playerid, "The stock market will close in %s.", secondstotime( GetServerVariableInt( "stock_report_time" ) - GetServerTime( ) ) );
-	return ShowPlayerDialog( playerid, DIALOG_STOCK_MARKET, DIALOG_STYLE_TABLIST_HEADERS, ""COL_WHITE"Stock Market", szLargeString, "Select", "Close" );
+	return ShowPlayerDialog( playerid, DIALOG_STOCK_MARKET, DIALOG_STYLE_TABLIST_HEADERS, ""COL_WHITE"Stock Market", szLargeString, "Buy", "Close" );
+}
+
+CMD:shares( playerid, params[ ] )
+{
+	mysql_tquery( dbHandle, sprintf( "SELECT * FROM `STOCK_OWNERS` WHERE `USER_ID` = %d", GetPlayerAccountID( playerid ) ), "StockMarket_OnShowShares", "d", playerid );
+	return 1;
 }
 
 /* ** Functions ** */
-stock CreateStockMarket( stockid, const name[ 64 ], const symbol[ 4 ], max_shares, ipo_price )
+stock CreateStockMarket( stockid, const name[ 64 ], const symbol[ 4 ], Float: ipo_shares, Float: ipo_price )
 {
 	if ( ! Iter_Contains( stockmarkets, stockid ) )
 	{
 		strcpy( g_stockMarketData[ stockid ] [ E_NAME ], name );
 		strcpy( g_stockMarketData[ stockid ] [ E_SYMBOL ], symbol );
 
-		g_stockMarketData[ stockid ] [ E_MM_SHARES ] = max_shares;
+		g_stockMarketData[ stockid ] [ E_IPO_SHARES ] = ipo_shares;
-		g_stockMarketData[ stockid ] [ E_MM_IPO_PRICE ] = ipo_price;
+		g_stockMarketData[ stockid ] [ E_IPO_PRICE ] = ipo_price;
 
 		// reset stock price information
 		for ( new r = 0; r < sizeof( g_stockMarketPriceData[ ] ); r ++ ) {
 			g_stockMarketPriceData[ stockid ] [ r ] [ E_PRICE ] = 0.0;
-			g_stockMarketPriceData[ stockid ] [ r ] [ E_EARNINGS ] = 0.0;
+			g_stockMarketPriceData[ stockid ] [ r ] [ E_EARNINGS ] = 1.0;
 		}
 
 		// load price information if there is
@@ -189,7 +355,7 @@ stock StockMarket_CreateReport( stockid )
 	g_stockMarketPriceData[ stockid ] [ 0 ] [ E_PRICE ] = g_stockMarketPriceData[ stockid ] [ 1 ] [ E_PRICE ];
 
 	// reset earnings
-	g_stockMarketPriceData[ stockid ] [ 0 ] [ E_EARNINGS ] = 0.0; // set to 0
+	g_stockMarketPriceData[ stockid ] [ 0 ] [ E_EARNINGS ] = 1.0; // set to 1 instead of 0 to prevent errors
 
 	// insert to database the old information
 	mysql_format(
@@ -200,6 +366,8 @@ stock StockMarket_CreateReport( stockid )
 		g_stockMarketPriceData[ stockid ] [ 0 ] [ E_EARNINGS ]
 	);
 
+	// todo: adjust sell orders
+
 	mysql_tquery( dbHandle, szBigString, "StockMarket_InsertReport", "d", stockid );
 	return 1;
 }
@@ -219,18 +387,23 @@ stock Float: StockMarket_GetEarnings( stockid ) {
 	return g_stockMarketPriceData[ stockid ] [ 0 ] [ E_EARNINGS ];
 }
 
-stock Float: StockMarket_GetCurrentPrice( stockid ) {
+stock StockMarket_GiveShares( stockid, accountid, Float: shares )
-	return g_stockMarketPriceData[ stockid ] [ 0 ] [ E_PRICE ];
+{
+	mysql_format(
+		dbHandle, szBigString, sizeof ( szBigString ),
+		"INSERT INTO `STOCK_OWNERS` (`USER_ID`, `STOCK_ID`, `SHARES`) VALUES (%d, %d, %f) ON DUPLICATE KEY UPDATE `SHARES` = `SHARES` + %f",
+		accountid, stockid, shares, shares
+	);
+	mysql_single_query( szBigString );
 }
 
-stock StockMarket_MakeSellOrder( stockid, accountid, Float: shares, Float: price )
+static stock StockMarket_ShowBuySlip( playerid, stockid )
 {
-	//INSERT INTO `STOCK_ORDERS` (`USER_ID`, `STOCK_ID`, ``)
+	mysql_tquery( dbHandle, sprintf( "SELECT * FROM `STOCK_OWNERS` WHERE `STOCK_ID` = %d AND `USER_ID` = %d", stockid, STOCK_MM_USER_ID ), "StockMarket_OnShowBuySlip", "dd", playerid, stockid );
 	return 1;
 }
 
 /*
-
 	CREATE TABLE IF NOT EXISTS `STOCK_REPORTS` (
 		`ID` int(11) primary key auto_increment,
 		`STOCK_ID` int(11),
@@ -240,18 +413,16 @@ stock StockMarket_MakeSellOrder( stockid, accountid, Float: shares, Float: price
 	);
 
 	CREATE TABLE IF NOT EXISTS `STOCK_OWNERS` (
-		`ID` int(11) primary key auto_increment,
 		`USER_ID` int(11),
 		`STOCK_ID` int(11),
 		`SHARES` float,
-		`PRICE` float
+		PRIMARY KEY (USER_ID, STOCK_ID)
 	);
 
-	CREATE TABLE IF NOT EXISTS `STOCK_ORDERS` (
+	CREATE TABLE IF NOT EXISTS `STOCK_TRADE_LOG` (
 		`ID` int(11) primary key auto_increment,
 		`USER_ID` int(11),
 		`STOCK_ID` int(11),
 		`SHARES` float,
-		`PRICE` float,
+	)
-	);
  */