Convert app_db to function variables (bug #4201)
[asterisk/asterisk.git] / funcs / func_db.c
1 /*
2  * Asterisk -- A telephony toolkit for Linux.
3  *
4  * Functions for interaction with the Asterisk database
5  * 
6  * Copyright (C) 2005, Russell Bryant <russelb@clemson.edu> 
7  *
8  * func_db.c adapted from the old app_db.c, copyright by the following people 
9  * Copyright (C) 2005, Mark Spencer <markster@digium.com>
10  * Copyright (C) 2003, Jefferson Noxon <jeff@debian.org>
11  *
12  * This program is free software, distributed under the terms of
13  * the GNU General Public License
14  */
15
16 #include <stdlib.h>
17 #include <string.h>
18 #include <sys/types.h>
19 #include <regex.h>
20
21 #include "asterisk/channel.h"
22 #include "asterisk/pbx.h"
23 #include "asterisk/logger.h"
24 #include "asterisk/options.h"
25 #include "asterisk/utils.h"
26 #include "asterisk/app.h"
27 #include "asterisk/astdb.h"
28
29 static char *function_db_read(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len)
30 {
31         int argc;       
32         char *args;
33         char *argv[2];
34         char *family;
35         char *key;
36
37         if (!data || ast_strlen_zero(data)) {
38                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)\n");
39                 return buf;
40         }
41
42         args = ast_strdupa(data);
43         argc = ast_separate_app_args(args, '/', argv, sizeof(argv) / sizeof(argv[0]));
44         
45         if (argc > 1) {
46                 family = argv[0];
47                 key = argv[1];
48         } else {
49                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)\n");
50                 return buf;
51         }
52
53         if (ast_db_get(family, key, buf, len-1)) {
54                 ast_log(LOG_WARNING, "DB: %s/%s not found in database.\n", family, key);
55         }
56         
57         return buf;
58 }
59
60 static void function_db_write(struct ast_channel *chan, char *cmd, char *data, const char *value) 
61 {
62         int argc;       
63         char *args;
64         char *argv[2];
65         char *family;
66         char *key;
67
68         if (!data || ast_strlen_zero(data)) {
69                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)=<value>\n");
70                 return;
71         }
72
73         args = ast_strdupa(data);
74         argc = ast_separate_app_args(args, '/', argv, sizeof(argv) / sizeof(argv[0]));
75         
76         if (argc > 1) {
77                 family = argv[0];
78                 key = argv[1];
79         } else {
80                 ast_log(LOG_WARNING, "DB requires an argument, DB(<family>/<key>)=value\n");
81                 return;
82         }
83
84         if (ast_db_put(family, key, (char*)value)) {
85                 ast_log(LOG_WARNING, "DB: Error writing value to database.\n");
86         }
87 }
88
89 #ifndef BUILTIN_FUNC
90 static
91 #endif
92 struct ast_custom_function db_function = {
93         .name = "DB",
94         .synopsis = "Read or Write from/to the Asterisk database",
95         .syntax = "DB(<family>/<key>)",
96         .desc = "This function will read or write a value from/to the Asterisk database."
97                 "DB(...) will read a value from the database, while DB(...)=value"
98                 "will write a value to the database.  On a read, this function"
99                 "returns the value from the datase, or NULL if it does not exist."
100                 "On a write, this function will always return NULL.",
101         .read = function_db_read,
102         .write = function_db_write,
103 };
104