Add SET function (bug #4335)
[asterisk/asterisk.git] / funcs / func_logic.c
1 /*
2  * Asterisk -- A telephony toolkit for Linux.
3  *
4  * Conditional logic dialplan functions
5  * 
6  * Copyright (C) 2005, Digium, Inc.
7  * Portions Copyright (C) 2005, Anthony Minessale II
8  *
9  * This program is free software, distributed under the terms of
10  * the GNU General Public License
11  */
12
13 #include <stdlib.h>
14 #include <string.h>
15 #include <sys/types.h>
16
17 #include "asterisk/channel.h"
18 #include "asterisk/pbx.h"
19 #include "asterisk/logger.h"
20 #include "asterisk/utils.h"
21 #include "asterisk/app.h"
22 #include "asterisk/config.h"            /* for ast_true */
23
24 static char *builtin_function_isnull(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len) 
25 {
26         char *ret_true = "1", *ret_false = "0";
27
28         return data && *data ? ret_false : ret_true;
29 }
30
31 static char *builtin_function_exists(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len) 
32 {
33         char *ret_true = "1", *ret_false = "0";
34
35         return data && *data ? ret_true : ret_false;
36 }
37
38 static char *builtin_function_if(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len) 
39 {
40         char *ret = NULL;
41         char *mydata = NULL;
42         char *expr = NULL;
43         char *iftrue = NULL;
44         char *iffalse = NULL;
45
46         if((mydata = ast_strdupa(data))) {
47                 expr = mydata;
48                 if ((iftrue = strchr(mydata, '?'))) {
49                         *iftrue = '\0';
50                         iftrue++;
51                         if ((iffalse = strchr(iftrue, ':'))) {
52                                 *iffalse = '\0';
53                                 iffalse++;
54                         }
55                 } 
56
57                 if (expr && iftrue) {
58                         ret = ast_true(expr) ? iftrue : iffalse;
59                         if (ret) {
60                                 ast_copy_string(buf, ret, len);
61                                 ret = buf;
62                         }
63                 } else {
64                         ast_log(LOG_WARNING, "Syntax $(if <expr>?[<truecond>][:<falsecond>])\n");
65                         ret = NULL;
66                 }
67         } else {
68                 ast_log(LOG_WARNING, "Memory Error!\n");
69                 ret = NULL;
70         }
71
72         return ret;
73 }
74
75 static char *builtin_function_set(struct ast_channel *chan, char *cmd, char *data, char *buf, size_t len) 
76 {
77         char *ret = NULL, *varname, *val;
78
79         if ((varname = ast_strdupa(data))) {
80                 if ((val = strchr(varname, '='))) {
81                         *val = '\0';
82                         val++;
83                         varname = ast_strip(varname);
84                         val = ast_strip(val);
85                         pbx_builtin_setvar_helper(chan, varname, val);
86                         ast_copy_string(buf, val, len);
87                 } else {
88                         ast_log(LOG_WARNING, "Syntax Error!\n");
89                 }
90                 
91         } else {
92         ast_log(LOG_WARNING, "Memory Error!\n");
93     }
94
95         return ret;
96 }
97
98 #ifndef BUILTIN_FUNC
99 static
100 #endif
101 struct ast_custom_function isnull_function = {
102         .name = "ISNULL",
103         .synopsis = "NULL Test: Returns 1 if NULL or 0 otherwise",
104         .syntax = "ISNULL(<data>)",
105         .read = builtin_function_isnull,
106 };
107
108 #ifndef BUILTIN_FUNC
109 static
110 #endif
111 struct ast_custom_function set_function = {
112         .name = "SET",
113         .synopsis = "SET assigns a value to a function call.",
114         .syntax = "SET(<varname>=<value>)",
115         .read = builtin_function_set,
116 };
117
118 #ifndef BUILTIN_FUNC
119 static
120 #endif
121 struct ast_custom_function exists_function = {
122         .name = "EXISTS",
123         .synopsis = "Existence Test: Returns 1 if exists, 0 otherwise",
124         .syntax = "EXISTS(<data>)",
125         .read = builtin_function_exists,
126 };
127
128 #ifndef BUILTIN_FUNC
129 static
130 #endif
131 struct ast_custom_function if_function = {
132         .name = "IF",
133         .synopsis = "Conditional: Returns the data following '?' if true else the data following ':'",
134         .syntax = "IF(<expr>?<true>:<false>)",
135         .read = builtin_function_if,
136 };