Commenting, fixed Ctrl-C not working, Fixed a segfault in the server,
[tpg/opendispense2.git] / src / server / dispense.c
1 /**
2  */
3 #include "common.h"
4 #include <stdlib.h>
5 #include <limits.h>
6 #include <string.h>
7
8  int    _GetMinBalance(int Account);
9  int    _CanTransfer(int Source, int Destination, int Ammount);
10  int    _Transfer(int Source, int Destination, int Ammount, const char *Reason);
11
12 // === CODE ===
13 /**
14  * \brief Dispense an item for a user
15  * 
16  * The core of the dispense system, I kinda like it :)
17  */
18 int DispenseItem(int ActualUser, int User, tItem *Item)
19 {
20          int    ret, salesAcct;
21         tHandler        *handler;
22         char    *username, *actualUsername;
23         
24         salesAcct = Bank_GetAcctByName(COKEBANK_SALES_ACCT);
25
26         // Check if the user can afford it
27         if( Item->Price && !_CanTransfer(User, salesAcct, Item->Price) )
28         {
29                 return 2;       // 2: No balance
30         }
31         
32         handler = Item->Handler;
33         
34         // Check if the dispense is possible
35         if( handler->CanDispense ) {
36                 ret = handler->CanDispense( User, Item->ID );
37                 if(ret) return 1;       // 1: Unable to dispense
38         }
39         
40         // Get username for debugging
41         username = Bank_GetAcctName(User);
42         
43         // Actually do the dispense
44         if( handler->DoDispense ) {
45                 ret = handler->DoDispense( User, Item->ID );
46                 if(ret) {
47                         Log_Error("Dispense failed (%s dispensing '%s' - %ic)",
48                                 username, Item->Name, Item->Price);
49                         free( username );
50                         return -1;      // 1: Unknown Error again
51                 }
52         }
53         
54         // Take away money
55         if( Item->Price )
56         {
57                 char    *reason;
58                 reason = mkstr("Dispense - %s:%i %s", handler->Name, Item->ID, Item->Name);
59                 _Transfer( User, salesAcct, Item->Price, reason );
60                 free(reason);
61         }
62         
63         actualUsername = Bank_GetAcctName(ActualUser);
64         
65         // And log that it happened
66         Log_Info("dispense '%s' (%s:%i) for %s by %s [cost %i, balance %i]",
67                 Item->Name, handler->Name, Item->ID,
68                 username, actualUsername, Item->Price, Bank_GetBalance(User)
69                 );
70         
71         free( username );
72         free( actualUsername );
73         return 0;       // 0: EOK
74 }
75
76 /**
77  * \brief Give money from one user to another
78  */
79 int DispenseGive(int ActualUser, int SrcUser, int DestUser, int Ammount, const char *ReasonGiven)
80 {
81          int    ret;
82         char    *actualUsername;
83         char    *srcName, *dstName;
84         
85         if( Ammount < 0 )       return 1;       // Um... negative give? Not on my watch!
86         
87         ret = _Transfer( SrcUser, DestUser, Ammount, ReasonGiven );
88         if(ret) return 2;       // No Balance
89         
90         
91         actualUsername = Bank_GetAcctName(ActualUser);
92         srcName = Bank_GetAcctName(SrcUser);
93         dstName = Bank_GetAcctName(DestUser);
94         
95         Log_Info("give %i to %s from %s by %s [balances %i, %i] - %s",
96                 Ammount, dstName, srcName, actualUsername,
97                 Bank_GetBalance(SrcUser), Bank_GetBalance(DestUser),
98                 ReasonGiven
99                 );
100         
101         free(srcName);
102         free(dstName);
103         free(actualUsername);
104         
105         return 0;
106 }
107
108 /**
109  * \brief Move money from one user to another (Admin Only)
110  */
111 int DispenseTransfer(int ActualUser, int SrcUser, int DestUser, int Ammount, const char *ReasonGiven)
112 {
113          int    ret;
114         char    *actualUsername;
115         char    *srcName, *dstName;
116
117         // Make sure the user is an admin
118         if( !(Bank_GetFlags(ActualUser) & USER_FLAG_ADMIN) )
119                 return 1;
120         
121         ret = _Transfer( SrcUser, DestUser, Ammount, ReasonGiven );
122         if(ret) return 2;       // No Balance
123         
124         
125         actualUsername = Bank_GetAcctName(ActualUser);
126         srcName = Bank_GetAcctName(SrcUser);
127         dstName = Bank_GetAcctName(DestUser);
128         
129         Log_Info("move %i to %s from %s by %s [balances %i, %i] - %s",
130                 Ammount, dstName, srcName, actualUsername,
131                 Bank_GetBalance(SrcUser), Bank_GetBalance(DestUser),
132                 ReasonGiven
133                 );
134         
135         free(srcName);
136         free(dstName);
137         free(actualUsername);
138         
139         return 0;
140 }
141 /**
142  * \brief Add money to an account
143  */
144 int DispenseAdd(int ActualUser, int User, int Ammount, const char *ReasonGiven)
145 {
146          int    ret;
147         char    *dstName, *byName;
148         
149         ret = _Transfer( Bank_GetAcctByName(COKEBANK_DEBT_ACCT), User, Ammount, ReasonGiven );
150         if(ret) return 2;
151         
152         byName = Bank_GetAcctName(ActualUser);
153         dstName = Bank_GetAcctName(User);
154         
155         Log_Info("add %i to %s by %s [balance %i] - %s",
156                 Ammount, dstName, byName, Bank_GetBalance(User), ReasonGiven
157                 );
158         
159         free(byName);
160         free(dstName);
161         
162         return 0;
163 }
164
165 int DispenseSet(int ActualUser, int User, int Balance, const char *ReasonGiven)
166 {
167          int    curBal = Bank_GetBalance(User);
168         char    *byName, *dstName;
169         
170         _Transfer( Bank_GetAcctByName(COKEBANK_DEBT_ACCT), User, Balance-curBal, ReasonGiven );
171         
172         byName = Bank_GetAcctName(ActualUser);
173         dstName = Bank_GetAcctName(User);
174         
175         Log_Info("set balance of %s to %i by %s [balance %i] - %s",
176                 dstName, Balance, byName, Bank_GetBalance(User), ReasonGiven
177                 );
178         
179         free(byName);
180         free(dstName);
181         
182         return 0;
183 }
184
185 /**
186  * \brief Donate money to the club
187  */
188 int DispenseDonate(int ActualUser, int User, int Ammount, const char *ReasonGiven)
189 {
190          int    ret;
191         char    *srcName, *byName;
192         
193         if( Ammount < 0 )       return 2;
194         
195         ret = _Transfer( User, Bank_GetAcctByName(COKEBANK_DEBT_ACCT), Ammount, ReasonGiven );
196         if(ret) return 2;
197         
198         byName = Bank_GetAcctName(ActualUser);
199         srcName = Bank_GetAcctName(User);
200         
201         Log_Info("donate %i from %s by %s [balance %i] - %s",
202                 Ammount, srcName, byName, Bank_GetBalance(User), ReasonGiven
203                 );
204         
205         free(byName);
206         free(srcName);
207         
208         return 0;
209 }
210
211 // --- Internal Functions ---
212 int _GetMinBalance(int Account)
213 {
214          int    flags = Bank_GetFlags(Account);
215         
216         // Evil little piece of HACK:
217         // root's balance cannot be changed by any of the above functions
218         // - Stops dispenses as root by returning insufficent balance.
219         {
220                 char    *username = Bank_GetAcctName(Account);
221                 if( strcmp(username, "root") == 0 )
222                 {
223                         free(username);
224                         return INT_MAX;
225                 }
226                 free(username);
227         }
228         
229         // - Internal accounts have no lower bound
230         if( flags & USER_FLAG_INTERNAL )        return INT_MIN;
231         
232         // Admin to -$10
233         //if( flags & USER_FLAG_ADMIN ) return -1000;
234         
235         // Coke to -$5
236         //if( flags & USER_FLAG_COKE )  return -500;
237         
238         // Anyone else, non-negative
239         return 0;
240 }
241
242 /**
243  * \brief Check if a transfer is possible
244  */
245 int _CanTransfer(int Source, int Destination, int Ammount)
246 {
247         if( Ammount > 0 )
248         {
249                 if( Bank_GetBalance(Source) + Ammount < _GetMinBalance(Source) )
250                         return 0;
251         }
252         else
253         {
254                 if( Bank_GetBalance(Destination) - Ammount < _GetMinBalance(Destination) )
255                         return 0;
256         }
257         return 1;
258 }
259
260 int _Transfer(int Source, int Destination, int Ammount, const char *Reason)
261 {
262         if( !_CanTransfer(Source, Destination, Ammount) )
263                 return 1;
264         return Bank_Transfer(Source, Destination, Ammount, Reason);
265 }

UCC git Repository :: git.ucc.asn.au