// Buy buys a given number of stocks for the user func Buy(userID string, quantity int, symbol string, ag *accessors.AccessorGroup) string { if !MarketOpen() { return fmt.Sprintf("The Stalk Market is currently closed.") } stock := models.CheckStock(symbol) user := ag.GetUser(userID) if stock.Name == "N/A" || stock.Price == 0 { return fmt.Sprintf("%s does not appear to be a valid stock...\n", strings.ToUpper(symbol)) // Return the price through the API endpoint } // Make sure they have enough turnips to buy if user.Turnips < stock.Price*quantity { return fmt.Sprintf("%s shares of %s costs %s turnips and you have %s turnips.\n", Comma(quantity), strings.ToUpper(symbol), Comma(stock.Price*quantity), Comma(user.Turnips)) // Return information about a user's portfolio } ag.SubtractTurnips(userID, stock.Price*quantity) if quantity == 0 { quantity = 1 } ag.AddShares(userID, symbol, quantity) Webhook(fmt.Sprintf("<@%s|%s> purchased %s share(s) of %s for %s turnips.", user.UserID, user.Username, Comma(quantity), strings.ToUpper(symbol), Comma(stock.Price*quantity))) return fmt.Sprintf("%s turnips were spent to add %s share(s) of %s to your portfolio.\n", Comma(stock.Price*quantity), Comma(quantity), strings.ToUpper(symbol)) // Return information about a user's portfolio }
// MakeUser creates the given user and puts them into the database func MakeUser(userID string, username string, ag *accessors.AccessorGroup) string { user := ag.GetUser(userID) if len(user.Username) > 0 { return fmt.Sprintf("Your account already exists. You have %s turnips.\n", Comma(user.Turnips)) } ag.MakeUser(userID, username) user = ag.GetUser(userID) return fmt.Sprintf("Your account has been created and supplied with %s turnips. Welcome to the Stalk Exchange!\n", Comma(user.Turnips)) }
// Sell sells a given number of the user's holdings in the symbol func Sell(userID string, quantity int, symbol string, ag *accessors.AccessorGroup) string { if !MarketOpen() { return fmt.Sprintf("The Stalk Market is currently closed.") } stock := models.CheckStock(symbol) user := ag.GetUser(userID) if stock.Name == "N/A" || stock.Price == 0 { return fmt.Sprintf("%s does not appear to be a valid stock...\n", strings.ToUpper(symbol)) // Return the price through the API endpoint } holdings := ag.GetShare(userID, symbol) if holdings >= quantity { // If we successfully sell ag.SubtractShares(userID, symbol, quantity) ag.AddTurnips(userID, stock.Price*quantity) // Add turnips to our wallet } else { // Else return a human-readable error return fmt.Sprintf("You do not have enough shares of %s to sell %s. You have %s shares.\n", strings.ToUpper(symbol), Comma(quantity), Comma(holdings)) // Return information about a user's portfolio } Webhook(fmt.Sprintf("<@%s|%s> sold %s share(s) of %s for %s turnips.", user.UserID, user.Username, Comma(quantity), strings.ToUpper(symbol), Comma(stock.Price*quantity))) return fmt.Sprintf("%s share(s) of %s have been sold for a total of %s turnips.\n", Comma(quantity), strings.ToUpper(symbol), Comma(quantity*stock.Price)) // Return information about a user's portfolio }