]> git.stg.codes - stg.git/blob - stargazer/corps_impl.cpp
3fe3f57204a665ebc1fc19d4fb06094e02ac3a0c
[stg.git] / stargazer / corps_impl.cpp
1 /*
2  *    This program is free software; you can redistribute it and/or modify
3  *    it under the terms of the GNU General Public License as published by
4  *    the Free Software Foundation; either version 2 of the License, or
5  *    (at your option) any later version.
6  *
7  *    This program is distributed in the hope that it will be useful,
8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  *    GNU General Public License for more details.
11  *
12  *    You should have received a copy of the GNU General Public License
13  *    along with this program; if not, write to the Free Software
14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
15  */
16
17 /*
18  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
19  */
20
21 #include "corps_impl.h"
22
23 #include "stg/admin.h"
24 #include "stg/common.h"
25
26 #include <cerrno>
27 #include <cassert>
28 #include <algorithm>
29
30 //-----------------------------------------------------------------------------
31 CORPORATIONS_IMPL::CORPORATIONS_IMPL(STORE * st)
32     : CORPORATIONS(),
33       data(),
34       store(st),
35       WriteServLog(GetStgLogger()),
36       searchDescriptors(),
37       handle(0),
38       mutex(),
39       strError()
40 {
41 pthread_mutex_init(&mutex, NULL);
42 Read();
43 }
44 //-----------------------------------------------------------------------------
45 int CORPORATIONS_IMPL::Add(const CORP_CONF & corp, const ADMIN * admin)
46 {
47 STG_LOCKER lock(&mutex);
48 const PRIV * priv = admin->GetPriv();
49
50 if (!priv->corpChg)
51     {
52     std::string s = admin->GetLogStr() + " Add corporation \'" + corp.name + "\'. Access denied.";
53     strError = "Access denied.";
54     WriteServLog(s.c_str());
55     return -1;
56     }
57
58 crp_iter si(find(data.begin(), data.end(), corp));
59
60 if (si != data.end())
61     {
62     strError = "Corporation \'" + corp.name + "\' cannot not be added. Corporation already exist.";
63     WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str());
64
65     return -1;
66     }
67
68 data.push_back(corp);
69
70 if (store->AddCorp(corp.name) == 0)
71     {
72     WriteServLog("%s Corporation \'%s\' added.",
73                  admin->GetLogStr().c_str(), corp.name.c_str());
74     return 0;
75     }
76
77 strError = "Corporation \'" + corp.name + "\' was not added. Error: " + store->GetStrError();
78 WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str());
79
80 return -1;
81 }
82 //-----------------------------------------------------------------------------
83 int CORPORATIONS_IMPL::Del(const std::string & name, const ADMIN * admin)
84 {
85 STG_LOCKER lock(&mutex);
86 const PRIV * priv = admin->GetPriv();
87
88 if (!priv->corpChg)
89     {
90     std::string s = admin->GetLogStr() + " Delete corporation \'" + name + "\'. Access denied.";
91     strError = "Access denied.";
92     WriteServLog(s.c_str());
93     return -1;
94     }
95
96 crp_iter si(find(data.begin(), data.end(), CORP_CONF(name)));
97
98 if (si == data.end())
99     {
100     strError = "Corporation \'" + name + "\' cannot be deleted. Corporation does not exist.";
101     WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str());
102     return -1;
103     }
104
105 std::map<int, const_crp_iter>::iterator csi;
106 csi = searchDescriptors.begin();
107 while (csi != searchDescriptors.end())
108     {
109     if (csi->second == si)
110         (csi->second)++;
111     ++csi;
112     }
113
114 data.erase(si);
115 if (store->DelCorp(name) < 0)
116     {
117     strError = "Corporation \'" + name + "\' was not deleted. Error: " + store->GetStrError();
118     WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str());
119
120     return -1;
121     }
122
123 WriteServLog("%s Corporation \'%s\' deleted.", admin->GetLogStr().c_str(), name.c_str());
124 return 0;
125 }
126 //-----------------------------------------------------------------------------
127 int CORPORATIONS_IMPL::Change(const CORP_CONF & corp, const ADMIN * admin)
128 {
129 STG_LOCKER lock(&mutex);
130 const PRIV * priv = admin->GetPriv();
131
132 if (!priv->corpChg)
133     {
134     std::string s = admin->GetLogStr() + " Change corporation \'" + corp.name + "\'. Access denied.";
135     strError = "Access denied.";
136     WriteServLog(s.c_str());
137     return -1;
138     }
139
140 crp_iter si(find(data.begin(), data.end(), corp));
141
142 if (si == data.end())
143     {
144     strError = "Corporation \'" + corp.name + "\' cannot be changed " + ". Corporation does not exist.";
145     WriteServLog("%s %s", admin->GetLogStr().c_str(), strError.c_str());
146     return -1;
147     }
148
149 *si = corp;
150 if (store->SaveCorp(corp))
151     {
152     WriteServLog("Cannot write corporation %s.", corp.name.c_str());
153     WriteServLog("%s", store->GetStrError().c_str());
154     return -1;
155     }
156
157 WriteServLog("%s Corporation \'%s\' changed.",
158              admin->GetLogStr().c_str(), corp.name.c_str());
159
160 return 0;
161 }
162 //-----------------------------------------------------------------------------
163 bool CORPORATIONS_IMPL::Read()
164 {
165 STG_LOCKER lock(&mutex);
166 std::vector<std::string> corpsList;
167 if (store->GetCorpsList(&corpsList) < 0)
168     {
169     WriteServLog(store->GetStrError().c_str());
170     return true;
171     }
172
173 for (size_t i = 0; i < corpsList.size(); i++)
174     {
175     CORP_CONF corp;
176
177     if (store->RestoreCorp(&corp, corpsList[i]))
178         {
179         WriteServLog(store->GetStrError().c_str());
180         return true;
181         }
182
183     data.push_back(corp);
184     }
185 return false;
186 }
187 //-----------------------------------------------------------------------------
188 bool CORPORATIONS_IMPL::Find(const std::string & name, CORP_CONF * corp)
189 {
190 assert(corp != NULL && "Pointer to corporation is not null");
191
192 STG_LOCKER lock(&mutex);
193 if (data.empty())
194     return false;
195
196 crp_iter si(find(data.begin(), data.end(), CORP_CONF(name)));
197
198 if (si != data.end())
199     {
200     *corp = *si;
201     return false;
202     }
203
204 return true;
205 }
206 //-----------------------------------------------------------------------------
207 bool CORPORATIONS_IMPL::Exists(const std::string & name) const
208 {
209 STG_LOCKER lock(&mutex);
210 if (data.empty())
211     {
212     printfd(__FILE__, "no admin in system!\n");
213     return true;
214     }
215
216 const_crp_iter si(find(data.begin(), data.end(), CORP_CONF(name)));
217
218 if (si != data.end())
219     return true;
220
221 return false;
222 }
223 //-----------------------------------------------------------------------------
224 int CORPORATIONS_IMPL::OpenSearch() const
225 {
226 STG_LOCKER lock(&mutex);
227 handle++;
228 searchDescriptors[handle] = data.begin();
229 return handle;
230 }
231 //-----------------------------------------------------------------------------
232 int CORPORATIONS_IMPL::SearchNext(int h, CORP_CONF * corp) const
233 {
234 STG_LOCKER lock(&mutex);
235 if (searchDescriptors.find(h) == searchDescriptors.end())
236     {
237     WriteServLog("CORPORATIONS. Incorrect search handle.");
238     return -1;
239     }
240
241 if (searchDescriptors[h] == data.end())
242     return -1;
243
244 *corp = *searchDescriptors[h]++;
245
246 return 0;
247 }
248 //-----------------------------------------------------------------------------
249 int CORPORATIONS_IMPL::CloseSearch(int h) const
250 {
251 STG_LOCKER lock(&mutex);
252 if (searchDescriptors.find(h) != searchDescriptors.end())
253     {
254     searchDescriptors.erase(searchDescriptors.find(h));
255     return 0;
256     }
257
258 WriteServLog("CORPORATIONS. Incorrect search handle.");
259 return -1;
260 }
261 //-----------------------------------------------------------------------------