Correctly use defined return values in (some) load_module functions.
[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 <stdlib.h>
35 #include <stdio.h>
36 #include <string.h>
37 #include <unistd.h>
38
39 #include "asterisk/file.h"
40 #include "asterisk/logger.h"
41 #include "asterisk/channel.h"
42 #include "asterisk/options.h"
43 #include "asterisk/pbx.h"
44 #include "asterisk/config.h"
45 #include "asterisk/module.h"
46 #include "asterisk/lock.h"
47 #include "asterisk/cli.h"
48
49
50 static char *cli_realtime_load(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a) 
51 {
52         char *header_format = "%30s  %-30s\n";
53         struct ast_variable *var=NULL;
54
55         switch (cmd) {
56         case CLI_INIT:
57                 e->command = "realtime load";
58                 e->usage =
59                         "Usage: realtime load <family> <colmatch> <value>\n"
60                         "       Prints out a list of variables using the RealTime driver.\n"
61                         "       You must supply a family name, a column to match on, and a value to match to.\n";
62                 return NULL;
63         case CLI_GENERATE:
64                 return NULL;
65         }
66
67
68         if (a->argc < 5) 
69                 return CLI_SHOWUSAGE;
70
71         var = ast_load_realtime_all(a->argv[2], a->argv[3], a->argv[4], NULL);
72
73         if (var) {
74                 ast_cli(a->fd, header_format, "Column Name", "Column Value");
75                 ast_cli(a->fd, header_format, "--------------------", "--------------------");
76                 while (var) {
77                         ast_cli(a->fd, header_format, var->name, var->value);
78                         var = var->next;
79                 }
80         } else {
81                 ast_cli(a->fd, "No rows found matching search criteria.\n");
82         }
83         return CLI_SUCCESS;
84 }
85
86 static char *cli_realtime_update(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a) {
87         int res = 0;
88
89         switch (cmd) {
90         case CLI_INIT:
91                 e->command = "realtime update";
92                 e->usage =
93                         "Usage: realtime update <family> <colupdate> <newvalue> <colmatch> <valuematch>\n"
94                         "       Update a single variable using the RealTime driver.\n"
95                         "       You must supply a family name, a column to update on, a new value, column to match, and value to match.\n"
96                         "       Ex: realtime update sipfriends name bobsphone port 4343\n"
97                         "       will execute SQL as UPDATE sipfriends SET port = 4343 WHERE name = bobsphone\n";
98                 return NULL;
99         case CLI_GENERATE:
100                 return NULL;
101         }
102
103
104         if (a->argc < 7) 
105                 return CLI_SHOWUSAGE;
106
107         res = ast_update_realtime(a->argv[2], a->argv[3], a->argv[4], a->argv[5], a->argv[6], NULL);
108
109         if(res < 0) {
110                 ast_cli(a->fd, "Failed to update. Check the debug log for possible SQL related entries.\n");
111                 return CLI_FAILURE;
112         }
113
114        ast_cli(a->fd, "Updated %d RealTime record%s.\n", res, ESS(res));
115
116         return CLI_SUCCESS;
117 }
118
119 static struct ast_cli_entry cli_realtime[] = {
120         AST_CLI_DEFINE(cli_realtime_load, "Used to print out RealTime variables."),
121         AST_CLI_DEFINE(cli_realtime_update, "Used to update RealTime variables."),
122 };
123
124 static int unload_module(void)
125 {
126         ast_cli_unregister_multiple(cli_realtime, sizeof(cli_realtime) / sizeof(struct ast_cli_entry));
127         return 0;
128 }
129
130 static int load_module(void)
131 {
132         ast_cli_register_multiple(cli_realtime, sizeof(cli_realtime) / sizeof(struct ast_cli_entry));
133         return AST_MODULE_LOAD_SUCCESS;
134 }
135
136 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Realtime Data Lookup/Rewrite");