X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=src%2Fserver%2Fdispense.c;h=c631bed7686234a000621274680a7fcd18a3cade;hb=ff5f048b2b302b569fb2011ebff11274e968a2e9;hp=3c18b588b3b497e4c0ff2dacc24663b290a90b66;hpb=3b0d5c732ceb17236bd93ebc2fe718ec1b2b3080;p=tpg%2Fopendispense2.git diff --git a/src/server/dispense.c b/src/server/dispense.c index 3c18b58..c631bed 100644 --- a/src/server/dispense.c +++ b/src/server/dispense.c @@ -2,6 +2,12 @@ */ #include "common.h" #include +#include +#include + + int _GetMinBalance(int Account); + int _CanTransfer(int Source, int Destination, int Ammount); + int _Transfer(int Source, int Destination, int Ammount, const char *Reason); // === CODE === /** @@ -11,10 +17,17 @@ */ int DispenseItem(int ActualUser, int User, tItem *Item) { - int ret; + int ret, salesAcct; tHandler *handler; char *username, *actualUsername; - char *reason; + + salesAcct = Bank_GetAcctByName(COKEBANK_SALES_ACCT); + + // Check if the user can afford it + if( Item->Price && !_CanTransfer(User, salesAcct, Item->Price) ) + { + return 2; // 2: No balance + } handler = Item->Handler; @@ -23,35 +36,36 @@ int DispenseItem(int ActualUser, int User, tItem *Item) ret = handler->CanDispense( User, Item->ID ); if(ret) return 1; // 1: Unable to dispense } - - // Subtract the balance - reason = mkstr("Dispense - %s:%i %s", handler->Name, Item->ID, Item->Name); - if( !reason ) reason = Item->Name; // TODO: Should I instead return an error? - ret = Transfer( User, GetUserID(COKEBANK_SALES_ACCT), Item->Price, reason); - free(reason); - if(ret) return 2; // 2: No balance // Get username for debugging - username = GetUserName(User); + username = Bank_GetAcctName(User); // Actually do the dispense if( handler->DoDispense ) { ret = handler->DoDispense( User, Item->ID ); if(ret) { - Log_Error("Dispense failed after deducting cost (%s dispensing %s - %ic)", + Log_Error("Dispense failed (%s dispensing '%s' - %ic)", username, Item->Name, Item->Price); - Transfer( GetUserID(COKEBANK_SALES_ACCT), User, Item->Price, "rollback" ); free( username ); - return -1; // 1: Unkown Error again + return -1; // 1: Unknown Error again } } - actualUsername = GetUserName(ActualUser); + // Take away money + if( Item->Price ) + { + char *reason; + reason = mkstr("Dispense - %s:%i %s", handler->Name, Item->ID, Item->Name); + _Transfer( User, salesAcct, Item->Price, reason ); + free(reason); + } + + actualUsername = Bank_GetAcctName(ActualUser); // And log that it happened - Log_Info("dispense '%s' (%s:%i) for %s by %s [cost %i, balance %i cents]", + Log_Info("dispense '%s' (%s:%i) for %s by %s [cost %i, balance %i]", Item->Name, handler->Name, Item->ID, - username, actualUsername, Item->Price, GetBalance(User) + username, actualUsername, Item->Price, Bank_GetBalance(User) ); free( username ); @@ -70,16 +84,18 @@ int DispenseGive(int ActualUser, int SrcUser, int DestUser, int Ammount, const c if( Ammount < 0 ) return 1; // Um... negative give? Not on my watch! - ret = Transfer( SrcUser, DestUser, Ammount, ReasonGiven ); + ret = _Transfer( SrcUser, DestUser, Ammount, ReasonGiven ); if(ret) return 2; // No Balance - srcName = GetUserName(SrcUser); - dstName = GetUserName(DestUser); - actualUsername = GetUserName(ActualUser); + actualUsername = Bank_GetAcctName(ActualUser); + srcName = Bank_GetAcctName(SrcUser); + dstName = Bank_GetAcctName(DestUser); - Log_Info("give %i to %s from %s by %s (%s)", - Ammount, dstName, srcName, actualUsername, ReasonGiven + Log_Info("give %i to %s from %s by %s [balances %i, %i] - %s", + Ammount, dstName, srcName, actualUsername, + Bank_GetBalance(SrcUser), Bank_GetBalance(DestUser), + ReasonGiven ); free(srcName); @@ -92,19 +108,19 @@ int DispenseGive(int ActualUser, int SrcUser, int DestUser, int Ammount, const c /** * \brief Add money to an account */ -int DispenseAdd(int User, int ByUser, int Ammount, const char *ReasonGiven) +int DispenseAdd(int ActualUser, int User, int Ammount, const char *ReasonGiven) { int ret; char *dstName, *byName; - ret = Transfer( GetUserID(COKEBANK_DEBT_ACCT), User, Ammount, ReasonGiven ); + ret = _Transfer( Bank_GetAcctByName(COKEBANK_DEBT_ACCT), User, Ammount, ReasonGiven ); if(ret) return 2; - byName = GetUserName(ByUser); - dstName = GetUserName(User); + byName = Bank_GetAcctName(ActualUser); + dstName = Bank_GetAcctName(User); - Log_Info("add %i to %s by %s (%s)", - Ammount, dstName, byName, ReasonGiven + Log_Info("add %i to %s by %s [balance %i] - %s", + Ammount, dstName, byName, Bank_GetBalance(User), ReasonGiven ); free(byName); @@ -112,3 +128,105 @@ int DispenseAdd(int User, int ByUser, int Ammount, const char *ReasonGiven) return 0; } + +int DispenseSet(int ActualUser, int User, int Balance, const char *ReasonGiven) +{ + int curBal = Bank_GetBalance(User); + char *byName, *dstName; + + _Transfer( Bank_GetAcctByName(COKEBANK_DEBT_ACCT), User, Balance-curBal, ReasonGiven ); + + byName = Bank_GetAcctName(ActualUser); + dstName = Bank_GetAcctName(User); + + Log_Info("set balance of %s to %i by %s [balance %i] - %s", + dstName, Balance, byName, Bank_GetBalance(User), ReasonGiven + ); + + free(byName); + free(dstName); + + return 0; +} + +/** + * \brief Donate money to the club + */ +int DispenseDonate(int ActualUser, int User, int Ammount, const char *ReasonGiven) +{ + int ret; + char *srcName, *byName; + + if( Ammount < 0 ) return 2; + + ret = _Transfer( User, Bank_GetAcctByName(COKEBANK_DEBT_ACCT), Ammount, ReasonGiven ); + if(ret) return 2; + + byName = Bank_GetAcctName(ActualUser); + srcName = Bank_GetAcctName(User); + + Log_Info("donate %i from %s by %s [balance %i] - %s", + Ammount, srcName, byName, Bank_GetBalance(User), ReasonGiven + ); + + free(byName); + free(srcName); + + return 0; +} + +// --- Internal Functions --- +int _GetMinBalance(int Account) +{ + int flags = Bank_GetFlags(Account); + + // Evil little piece of HACK: + // root's balance cannot be changed by any of the above functions + // - Stops dispenses as root by returning insufficent balance. + { + char *username = Bank_GetAcctName(Account); + if( strcmp(username, "root") == 0 ) + { + free(username); + return INT_MAX; + } + free(username); + } + + // - Internal accounts have no lower bound + if( flags & USER_FLAG_INTERNAL ) return INT_MIN; + + // Admin to -$10 + if( flags & USER_FLAG_ADMIN ) return -1000; + + // Coke to -$5 + if( flags & USER_FLAG_COKE ) return -500; + + // Anyone else, non-negative + return 0; +} + +/** + * \brief Check if a transfer is possible + */ +int _CanTransfer(int Source, int Destination, int Ammount) +{ + if( Ammount > 0 ) + { + if( Bank_GetBalance(Source) + Ammount < _GetMinBalance(Source) ) + return 0; + } + else + { + if( Bank_GetBalance(Destination) - Ammount < _GetMinBalance(Destination) ) + return 0; + } + return 1; +} + +int _Transfer(int Source, int Destination, int Ammount, const char *Reason) +{ + if( !_CanTransfer(Source, Destination, Ammount) ) + return 1; + return Bank_Transfer(Source, Destination, Ammount, Reason); +}