45ebcc63a86a7fef224f3c6236cc49b0b456f306
[asterisk/asterisk.git] / res / res_realtime.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 1999 - 2005, Digium, Inc.
5  *
6  * Anthony Minessale <anthmct@yahoo.com>
7  * Mark Spencer <markster@digium.com>
8  *
9  * See http://www.asterisk.org for more information about
10  * the Asterisk project. Please do not directly contact
11  * any of the maintainers of this project for assistance;
12  * the project provides a web site, mailing lists and IRC
13  * channels for your use.
14  *
15  * This program is free software, distributed under the terms of
16  * the GNU General Public License Version 2. See the LICENSE file
17  * at the top of the source tree.
18  */
19
20 /*! \file
21  *
22  * \brief RealTime CLI
23  *
24  * \author Anthony Minessale <anthmct@yahoo.com>
25  * \author Mark Spencer <markster@digium.com>
26  * 
27  * \ingroup applications
28  */
29
30 #include "asterisk.h"
31
32 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
33
34 #include "asterisk/file.h"
35 #include "asterisk/channel.h"
36 #include "asterisk/pbx.h"
37 #include "asterisk/config.h"
38 #include "asterisk/module.h"
39 #include "asterisk/lock.h"
40 #include "asterisk/cli.h"
41
42
43 static char *cli_realtime_load(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a) 
44 {
45 #define CRL_HEADER_FORMAT "%30s  %-30s\n"
46         struct ast_variable *var=NULL;
47
48         switch (cmd) {
49         case CLI_INIT:
50                 e->command = "realtime load";
51                 e->usage =
52                         "Usage: realtime load <family> <colmatch> <value>\n"
53                         "       Prints out a list of variables using the RealTime driver.\n"
54                         "       You must supply a family name, a column to match on, and a value to match to.\n";
55                 return NULL;
56         case CLI_GENERATE:
57                 return NULL;
58         }
59
60
61         if (a->argc < 5) 
62                 return CLI_SHOWUSAGE;
63
64         var = ast_load_realtime_all(a->argv[2], a->argv[3], a->argv[4], SENTINEL);
65
66         if (var) {
67                 ast_cli(a->fd, CRL_HEADER_FORMAT, "Column Name", "Column Value");
68                 ast_cli(a->fd, CRL_HEADER_FORMAT, "--------------------", "--------------------");
69                 while (var) {
70                         ast_cli(a->fd, CRL_HEADER_FORMAT, var->name, var->value);
71                         var = var->next;
72                 }
73         } else {
74                 ast_cli(a->fd, "No rows found matching search criteria.\n");
75         }
76         return CLI_SUCCESS;
77 }
78
79 static char *cli_realtime_update(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a)
80 {
81         int res = 0;
82
83         switch (cmd) {
84         case CLI_INIT:
85                 e->command = "realtime update";
86                 e->usage =
87                         "Usage: realtime update <family> <colmatch> <valuematch> <colupdate> <newvalue>\n"
88                         "       Update a single variable using the RealTime driver.\n"
89                         "       You must supply a family name, a column to update on, a new value, column to match, and value to match.\n"
90                         "       Ex: realtime update sipfriends name bobsphone port 4343\n"
91                         "       will execute SQL as UPDATE sipfriends SET port = 4343 WHERE name = bobsphone\n";
92                 return NULL;
93         case CLI_GENERATE:
94                 return NULL;
95         }
96
97         if (a->argc < 7) 
98                 return CLI_SHOWUSAGE;
99
100         res = ast_update_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], SENTINEL);
101
102         if (res < 0) {
103                 ast_cli(a->fd, "Failed to update. Check the debug log for possible SQL related entries.\n");
104                 return CLI_FAILURE;
105         }
106
107         ast_cli(a->fd, "Updated %d RealTime record%s.\n", res, ESS(res));
108
109         return CLI_SUCCESS;
110 }
111
112 static char *cli_realtime_update2(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a)
113 {
114         int res = -1;
115
116         switch (cmd) {
117         case CLI_INIT:
118                 e->command = "realtime update2";
119                 e->usage =
120                         "Usage: realtime update2 <family> <colmatch> <valuematch> [... <colmatch5> <valuematch5>] NULL <colupdate> <newvalue>\n"
121                         "       Update a single variable using the RealTime driver.\n"
122                         "       You must supply a family name, a column to update on, a new value, column to match, and value to match.\n"
123                         "       Ex: realtime update sipfriends name bobsphone port 4343\n"
124                         "       will execute SQL as UPDATE sipfriends SET port = 4343 WHERE name = bobsphone\n";
125                 return NULL;
126         case CLI_GENERATE:
127                 return NULL;
128         }
129
130         if (a->argc < 7) 
131                 return CLI_SHOWUSAGE;
132
133         if (a->argc == 7) {
134                 res = ast_update2_realtime(a->argv[2], a->argv[3], a->argv[4], SENTINEL, a->argv[5], a->argv[6], SENTINEL);
135         } else if (a->argc == 9) {
136                 res = ast_update2_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], SENTINEL, a->argv[7], a->argv[8], SENTINEL);
137         } else if (a->argc == 11) {
138                 res = ast_update2_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], a->argv[7], a->argv[8], SENTINEL, a->argv[9], a->argv[10], SENTINEL);
139         } else if (a->argc == 13) {
140                 res = ast_update2_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], a->argv[7], a->argv[8], a->argv[9], a->argv[10], SENTINEL, a->argv[11], a->argv[12], SENTINEL);
141         } else if (a->argc == 15) {
142                 res = ast_update2_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], a->argv[7], a->argv[8], a->argv[9], a->argv[10], a->argv[11], a->argv[12], SENTINEL, a->argv[13], a->argv[14], SENTINEL);
143         } else {
144                 return CLI_SHOWUSAGE;
145         }
146
147         if (res < 0) {
148                 ast_cli(a->fd, "Failed to update. Check the debug log for possible SQL related entries.\n");
149                 return CLI_FAILURE;
150         }
151
152         ast_cli(a->fd, "Updated %d RealTime record%s.\n", res, ESS(res));
153
154         return CLI_SUCCESS;
155 }
156
157 static char *cli_realtime_store(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a)
158 {
159         int res = -1;
160
161         switch (cmd) {
162         case CLI_INIT:
163                 e->command = "realtime store";
164                 e->usage =
165                         "Usage: realtime store <family> <colname1> <value1> [<colname2> <value2> [... <colname5> <value5>]]\n"
166                         "       Create a stored row using the RealTime driver.\n"
167                         "       You must supply a family name and name/value pairs (up to 5).  If\n"
168                         "       you need to store more than 5 key/value pairs, start with the first\n"
169                         "       five, then use 'realtime update' or 'realtime update2' to add\n"
170                         "       additional columns.\n";
171                 return NULL;
172         case CLI_GENERATE:
173                 return NULL;
174         }
175
176         if (a->argc < 5) {
177                 return CLI_SHOWUSAGE;
178         } else if (a->argc == 5) {
179                 res = ast_store_realtime(a->argv[2], a->argv[3], a->argv[4], SENTINEL);
180         } else if (a->argc == 7) {
181                 res = ast_store_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], SENTINEL);
182         } else if (a->argc == 9) {
183                 res = ast_store_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], a->argv[7], a->argv[8], SENTINEL);
184         } else if (a->argc == 11) {
185                 res = ast_store_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], a->argv[7], a->argv[8], a->argv[9], a->argv[10], SENTINEL);
186         } else if (a->argc == 13) {
187                 res = ast_store_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], a->argv[7], a->argv[8], a->argv[9], a->argv[10], a->argv[11], a->argv[12], SENTINEL);
188         } else {
189                 return CLI_SHOWUSAGE;
190         }
191
192         if (res < 0) {
193                 ast_cli(a->fd, "Failed to store record. Check the debug log for possible SQL related entries.\n");
194                 return CLI_FAILURE;
195         }
196
197         ast_cli(a->fd, "Stored RealTime record.\n");
198
199         return CLI_SUCCESS;
200 }
201
202 static char *cli_realtime_destroy(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a)
203 {
204         int res = -1;
205
206         switch (cmd) {
207         case CLI_INIT:
208                 e->command = "realtime destroy";
209                 e->usage =
210                         "Usage: realtime destroy <family> <colmatch1> <valuematch1> [<colmatch2> <valuematch2> [... <colmatch5> <valuematch5>]]\n"
211                         "       Remove a stored row using the RealTime driver.\n"
212                         "       You must supply a family name and name/value pairs (up to 5).\n";
213                 return NULL;
214         case CLI_GENERATE:
215                 return NULL;
216         }
217
218         if (a->argc < 5) {
219                 return CLI_SHOWUSAGE;
220         } else if (a->argc == 5) {
221                 res = ast_destroy_realtime(a->argv[2], a->argv[3], a->argv[4], SENTINEL);
222         } else if (a->argc == 7) {
223                 res = ast_destroy_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], SENTINEL);
224         } else if (a->argc == 9) {
225                 res = ast_destroy_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], a->argv[7], a->argv[8], SENTINEL);
226         } else if (a->argc == 11) {
227                 res = ast_destroy_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], a->argv[7], a->argv[8], a->argv[9], a->argv[10], SENTINEL);
228         } else if (a->argc == 13) {
229                 res = ast_destroy_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], a->argv[7], a->argv[8], a->argv[9], a->argv[10], a->argv[11], a->argv[12], SENTINEL);
230         } else {
231                 return CLI_SHOWUSAGE;
232         }
233
234         if (res < 0) {
235                 ast_cli(a->fd, "Failed to remove record. Check the debug log for possible SQL related entries.\n");
236                 return CLI_FAILURE;
237         }
238
239         ast_cli(a->fd, "Removed %d RealTime record%s.\n", res, ESS(res));
240
241         return CLI_SUCCESS;
242 }
243
244 static struct ast_cli_entry cli_realtime[] = {
245         AST_CLI_DEFINE(cli_realtime_load, "Used to print out RealTime variables."),
246         AST_CLI_DEFINE(cli_realtime_update, "Used to update RealTime variables."),
247         AST_CLI_DEFINE(cli_realtime_update2, "Used to test the RealTime update2 method"),
248         AST_CLI_DEFINE(cli_realtime_store, "Store a new row into a RealTime database"),
249         AST_CLI_DEFINE(cli_realtime_destroy, "Delete a row from a RealTime database"),
250 };
251
252 static int unload_module(void)
253 {
254         ast_cli_unregister_multiple(cli_realtime, sizeof(cli_realtime) / sizeof(struct ast_cli_entry));
255         return 0;
256 }
257
258 static int load_module(void)
259 {
260         ast_cli_register_multiple(cli_realtime, sizeof(cli_realtime) / sizeof(struct ast_cli_entry));
261         return AST_MODULE_LOAD_SUCCESS;
262 }
263
264 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Realtime Data Lookup/Rewrite");