simplify autoconfig include mechanism (make tholo happy he can use lint again :-)
[asterisk/asterisk.git] / funcs / func_db.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2005-2006, Russell Bryant <russelb@clemson.edu> 
5  *
6  * func_db.c adapted from the old app_db.c, copyright by the following people 
7  * Copyright (C) 2005, Mark Spencer <markster@digium.com>
8  * Copyright (C) 2003, Jefferson Noxon <jeff@debian.org>
9  *
10  * See http://www.asterisk.org for more information about
11  * the Asterisk project. Please do not directly contact
12  * any of the maintainers of this project for assistance;
13  * the project provides a web site, mailing lists and IRC
14  * channels for your use.
15  *
16  * This program is free software, distributed under the terms of
17  * the GNU General Public License Version 2. See the LICENSE file
18  * at the top of the source tree.
19  */
20
21 /*! \file
22  *
23  * \brief Functions for interaction with the Asterisk database
24  *
25  * \author Russell Bryant <russelb@clemson.edu>
26  */
27
28 #include "asterisk.h"
29
30 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
31
32 #include <stdlib.h>
33 #include <string.h>
34 #include <sys/types.h>
35 #include <regex.h>
36
37 #include "asterisk/module.h"
38 #include "asterisk/channel.h"
39 #include "asterisk/pbx.h"
40 #include "asterisk/logger.h"
41 #include "asterisk/options.h"
42 #include "asterisk/utils.h"
43 #include "asterisk/app.h"
44 #include "asterisk/astdb.h"
45
46 static int function_db_read(struct ast_channel *chan, char *cmd,
47                             char *parse, char *buf, size_t len)
48 {
49         AST_DECLARE_APP_ARGS(args,
50                              AST_APP_ARG(family);
51                              AST_APP_ARG(key);
52         );
53
54         buf[0] = '\0';
55
56         if (ast_strlen_zero(parse)) {
57                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)\n");
58                 return -1;
59         }
60
61         AST_NONSTANDARD_APP_ARGS(args, parse, '/');
62
63         if (args.argc < 2) {
64                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)\n");
65                 return -1;
66         }
67
68         if (ast_db_get(args.family, args.key, buf, len - 1)) {
69                 ast_log(LOG_DEBUG, "DB: %s/%s not found in database.\n", args.family,
70                                 args.key);
71         } else
72                 pbx_builtin_setvar_helper(chan, "DB_RESULT", buf);
73
74         return 0;
75 }
76
77 static int function_db_write(struct ast_channel *chan, char *cmd, char *parse,
78                              const char *value)
79 {
80         AST_DECLARE_APP_ARGS(args,
81                              AST_APP_ARG(family);
82                              AST_APP_ARG(key);
83         );
84
85         if (ast_strlen_zero(parse)) {
86                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)=<value>\n");
87                 return -1;
88         }
89
90         AST_NONSTANDARD_APP_ARGS(args, parse, '/');
91
92         if (args.argc < 2) {
93                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)=value\n");
94                 return -1;
95         }
96
97         if (ast_db_put(args.family, args.key, (char *) value))
98                 ast_log(LOG_WARNING, "DB: Error writing value to database.\n");
99
100         return 0;
101 }
102
103 static struct ast_custom_function db_function = {
104         .name = "DB",
105         .synopsis = "Read or Write from/to the Asterisk database",
106         .syntax = "DB(<family>/<key>)",
107         .desc =
108                 "This function will read or write a value from/to the Asterisk database.\n"
109                 "DB(...) will read a value from the database, while DB(...)=value\n"
110                 "will write a value to the database.  On a read, this function\n"
111                 "returns the value from the datase, or NULL if it does not exist.\n"
112                 "On a write, this function will always return NULL.  Reading a database value\n"
113                 "will also set the variable DB_RESULT.\n",
114         .read = function_db_read,
115         .write = function_db_write,
116 };
117
118 static int function_db_exists(struct ast_channel *chan, char *cmd,
119                               char *parse, char *buf, size_t len)
120 {
121         AST_DECLARE_APP_ARGS(args,
122                              AST_APP_ARG(family);
123                              AST_APP_ARG(key);
124         );
125
126         buf[0] = '\0';
127
128         if (ast_strlen_zero(parse)) {
129                 ast_log(LOG_WARNING, "DB_EXISTS requires an argument, DB(<family>/<key>)\n");
130                 return -1;
131         }
132
133         AST_NONSTANDARD_APP_ARGS(args, parse, '/');
134
135         if (args.argc < 2) {
136                 ast_log(LOG_WARNING, "DB_EXISTS requires an argument, DB(<family>/<key>)\n");
137                 return -1;
138         }
139
140         if (ast_db_get(args.family, args.key, buf, len - 1))
141                 strcpy(buf, "0");
142         else {
143                 pbx_builtin_setvar_helper(chan, "DB_RESULT", buf);
144                 strcpy(buf, "1");
145         }
146
147         return 0;
148 }
149
150 static struct ast_custom_function db_exists_function = {
151         .name = "DB_EXISTS",
152         .synopsis = "Check to see if a key exists in the Asterisk database",
153         .syntax = "DB_EXISTS(<family>/<key>)",
154         .desc =
155                 "This function will check to see if a key exists in the Asterisk\n"
156                 "database. If it exists, the function will return \"1\". If not,\n"
157                 "it will return \"0\".  Checking for existence of a database key will\n"
158                 "also set the variable DB_RESULT to the key's value if it exists.\n",
159         .read = function_db_exists,
160 };
161
162 static int function_db_delete(struct ast_channel *chan, char* cmd,
163                               char *parse, char *buf, size_t len)
164 {
165         AST_DECLARE_APP_ARGS(args,
166                              AST_APP_ARG(family);
167                              AST_APP_ARG(key);
168         );
169
170         buf[0] = '\0';
171
172         if (ast_strlen_zero(parse)) {
173                 ast_log(LOG_WARNING, "DB_DELETE requires an argument, DB_DELETE(<family>/<key>)\n");
174                 return -1;
175         }
176
177         AST_NONSTANDARD_APP_ARGS(args, parse, '/');
178
179         if (args.argc < 2) {
180                 ast_log(LOG_WARNING, "DB_DELETE requires an argument, DB_DELETE(<family>/<key>)\n");
181                 return -1;
182         }
183
184         if (ast_db_get(args.family, args.key, buf, len - 1)) {
185                 ast_log(LOG_DEBUG, "DB_DELETE: %s/%s not found in database.\n", args.family, args.key);
186         } else {
187                 if (ast_db_del(args.family, args.key)) {
188                         ast_log(LOG_DEBUG, "DB_DELETE: %s/%s could not be deleted from the database\n", 
189                                 args.family, args.key);
190                 }
191         }
192         pbx_builtin_setvar_helper(chan, "DB_RESULT", buf);
193
194         return 0;
195 }
196
197
198 static struct ast_custom_function db_delete_function = {
199         .name = "DB_DELETE",
200         .synopsis = "Return a value from the database and delete it",
201         .syntax = "DB_DELETE(<family>/<key>)",
202         .desc =
203                 "This function will retrieve a value from the Asterisk database\n"
204                 " and then remove that key from the database.  DB_RESULT\n"
205                 "will be set to the key's value if it exists.\n",
206         .read = function_db_delete,
207 };
208
209 static char *tdesc = "Database (astdb) related dialplan functions";
210
211 static int unload_module(void *mod)
212 {
213         int res = 0;
214
215         res |= ast_custom_function_unregister(&db_function);
216         res |= ast_custom_function_unregister(&db_exists_function);
217         res |= ast_custom_function_unregister(&db_delete_function);
218
219         return res;
220 }
221
222 static int load_module(void *mod)
223 {
224         int res = 0;
225
226         res |= ast_custom_function_register(&db_function);
227         res |= ast_custom_function_register(&db_exists_function);
228         res |= ast_custom_function_register(&db_delete_function);
229
230         return res;
231 }
232
233 static const char *description(void)
234 {
235         return tdesc;
236 }
237
238 static const char *key(void)
239 {
240         return ASTERISK_GPL_KEY;
241 }
242
243 STD_MOD(MOD_1 | NO_USECOUNT, NULL, NULL, NULL);