diff options
Diffstat (limited to 'trunk/funcs/func_groupcount.c')
-rw-r--r-- | trunk/funcs/func_groupcount.c | 231 |
1 files changed, 231 insertions, 0 deletions
diff --git a/trunk/funcs/func_groupcount.c b/trunk/funcs/func_groupcount.c new file mode 100644 index 000000000..f0d7df1b5 --- /dev/null +++ b/trunk/funcs/func_groupcount.c @@ -0,0 +1,231 @@ +/* + * Asterisk -- An open source telephony toolkit. + * + * Copyright (C) 1999 - 2006, Digium, Inc. + * + * See http://www.asterisk.org for more information about + * the Asterisk project. Please do not directly contact + * any of the maintainers of this project for assistance; + * the project provides a web site, mailing lists and IRC + * channels for your use. + * + * This program is free software, distributed under the terms of + * the GNU General Public License Version 2. See the LICENSE file + * at the top of the source tree. + */ + +/*! \file + * + * \brief Channel group related dialplan functions + * + * \ingroup functions + */ + +#include "asterisk.h" + +ASTERISK_FILE_VERSION(__FILE__, "$Revision$") + +#include "asterisk/module.h" +#include "asterisk/channel.h" +#include "asterisk/pbx.h" +#include "asterisk/utils.h" +#include "asterisk/app.h" + +static int group_count_function_read(struct ast_channel *chan, const char *cmd, + char *data, char *buf, size_t len) +{ + int count = -1; + char group[80] = "", category[80] = ""; + + ast_app_group_split_group(data, group, sizeof(group), category, + sizeof(category)); + + /* If no group has been provided let's find one */ + if (ast_strlen_zero(group)) { + struct ast_group_info *gi = NULL; + + ast_app_group_list_rdlock(); + for (gi = ast_app_group_list_head(); gi; gi = AST_LIST_NEXT(gi, list)) { + if (gi->chan != chan) + continue; + if (ast_strlen_zero(category) || (!ast_strlen_zero(gi->category) && !strcasecmp(gi->category, category))) + break; + } + if (gi) { + ast_copy_string(group, gi->group, sizeof(group)); + if (!ast_strlen_zero(gi->category)) + ast_copy_string(category, gi->category, sizeof(category)); + } + ast_app_group_list_unlock(); + } + + if ((count = ast_app_group_get_count(group, category)) == -1) + ast_log(LOG_NOTICE, "No group could be found for channel '%s'\n", chan->name); + else + snprintf(buf, len, "%d", count); + + return 0; +} + +static struct ast_custom_function group_count_function = { + .name = "GROUP_COUNT", + .syntax = "GROUP_COUNT([groupname][@category])", + .synopsis = "Counts the number of channels in the specified group", + .desc = + "Calculates the group count for the specified group, or uses the\n" + "channel's current group if not specifed (and non-empty).\n", + .read = group_count_function_read, +}; + +static int group_match_count_function_read(struct ast_channel *chan, + const char *cmd, char *data, char *buf, + size_t len) +{ + int count; + char group[80] = ""; + char category[80] = ""; + + ast_app_group_split_group(data, group, sizeof(group), category, + sizeof(category)); + + if (!ast_strlen_zero(group)) { + count = ast_app_group_match_get_count(group, category); + snprintf(buf, len, "%d", count); + } + + return 0; +} + +static struct ast_custom_function group_match_count_function = { + .name = "GROUP_MATCH_COUNT", + .syntax = "GROUP_MATCH_COUNT(groupmatch[@category])", + .synopsis = + "Counts the number of channels in the groups matching the specified pattern", + .desc = + "Calculates the group count for all groups that match the specified pattern.\n" + "Uses standard regular expression matching (see regex(7)).\n", + .read = group_match_count_function_read, + .write = NULL, +}; + +static int group_function_read(struct ast_channel *chan, const char *cmd, + char *data, char *buf, size_t len) +{ + struct ast_group_info *gi = NULL; + + ast_app_group_list_rdlock(); + + for (gi = ast_app_group_list_head(); gi; gi = AST_LIST_NEXT(gi, list)) { + if (gi->chan != chan) + continue; + if (ast_strlen_zero(data)) + break; + if (!ast_strlen_zero(gi->category) && !strcasecmp(gi->category, data)) + break; + } + + if (gi) + ast_copy_string(buf, gi->group, len); + + ast_app_group_list_unlock(); + + return 0; +} + +static int group_function_write(struct ast_channel *chan, const char *cmd, + char *data, const char *value) +{ + char grpcat[256]; + + if (!ast_strlen_zero(data)) { + snprintf(grpcat, sizeof(grpcat), "%s@%s", value, data); + } else { + ast_copy_string(grpcat, value, sizeof(grpcat)); + } + + if (ast_app_group_set_channel(chan, grpcat)) + ast_log(LOG_WARNING, + "Setting a group requires an argument (group name)\n"); + + return 0; +} + +static struct ast_custom_function group_function = { + .name = "GROUP", + .syntax = "GROUP([category])", + .synopsis = "Gets or sets the channel group.", + .desc = "Gets or sets the channel group.\n", + .read = group_function_read, + .write = group_function_write, +}; + +static int group_list_function_read(struct ast_channel *chan, const char *cmd, + char *data, char *buf, size_t len) +{ + struct ast_group_info *gi = NULL; + char tmp1[1024] = ""; + char tmp2[1024] = ""; + + if (!chan) + return -1; + + ast_app_group_list_rdlock(); + + for (gi = ast_app_group_list_head(); gi; gi = AST_LIST_NEXT(gi, list)) { + if (gi->chan != chan) + continue; + if (!ast_strlen_zero(tmp1)) { + ast_copy_string(tmp2, tmp1, sizeof(tmp2)); + if (!ast_strlen_zero(gi->category)) + snprintf(tmp1, sizeof(tmp1), "%s %s@%s", tmp2, gi->group, gi->category); + else + snprintf(tmp1, sizeof(tmp1), "%s %s", tmp2, gi->group); + } else { + if (!ast_strlen_zero(gi->category)) + snprintf(tmp1, sizeof(tmp1), "%s@%s", gi->group, gi->category); + else + snprintf(tmp1, sizeof(tmp1), "%s", gi->group); + } + } + + ast_app_group_list_unlock(); + + ast_copy_string(buf, tmp1, len); + + return 0; +} + +static struct ast_custom_function group_list_function = { + .name = "GROUP_LIST", + .syntax = "GROUP_LIST()", + .synopsis = "Gets a list of the groups set on a channel.", + .desc = "Gets a list of the groups set on a channel.\n", + .read = group_list_function_read, + .write = NULL, +}; + +static int unload_module(void) +{ + int res = 0; + + res |= ast_custom_function_unregister(&group_count_function); + res |= ast_custom_function_unregister(&group_match_count_function); + res |= ast_custom_function_unregister(&group_list_function); + res |= ast_custom_function_unregister(&group_function); + + return res; +} + +static int load_module(void) +{ + int res = 0; + + res |= ast_custom_function_register(&group_count_function); + res |= ast_custom_function_register(&group_match_count_function); + res |= ast_custom_function_register(&group_list_function); + res |= ast_custom_function_register(&group_function); + + return res; +} + +AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Channel group dialplan functions"); |