2 * Asterisk -- An open source telephony toolkit.
4 * Copyright (c) 2003-2006 Tilghman Lesher. All rights reserved.
6 * Tilghman Lesher <app_cut__v003@the-tilghman.com>
8 * This code is released by the author with no restrictions on usage.
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.
22 * \author Tilghman Lesher <app_cut__v003@the-tilghman.com>
29 ASTERISK_FILE_VERSION(__FILE__
, "$Revision$")
36 #include "asterisk/file.h"
37 #include "asterisk/logger.h"
38 #include "asterisk/options.h"
39 #include "asterisk/channel.h"
40 #include "asterisk/pbx.h"
41 #include "asterisk/module.h"
42 #include "asterisk/app.h"
44 /* Maximum length of any variable */
45 #define MAXRESULT 1024
50 struct sortable_keys
{
55 static int sort_subroutine(const void *arg1
, const void *arg2
)
57 const struct sortable_keys
*one
=arg1
, *two
=arg2
;
58 if (one
->value
< two
->value
)
60 else if (one
->value
== two
->value
)
66 #define ERROR_NOARG (-1)
67 #define ERROR_NOMEM (-2)
68 #define ERROR_USAGE (-3)
70 static int sort_internal(struct ast_channel
*chan
, char *data
, char *buffer
, size_t buflen
)
72 char *strings
, *ptrkey
, *ptrvalue
;
73 int count
=1, count2
, element_count
=0;
74 struct sortable_keys
*sortable_keys
;
76 memset(buffer
, 0, buflen
);
81 strings
= ast_strdupa(data
);
83 for (ptrkey
= strings
; *ptrkey
; ptrkey
++) {
88 sortable_keys
= alloca(count
* sizeof(struct sortable_keys
));
90 memset(sortable_keys
, 0, count
* sizeof(struct sortable_keys
));
92 /* Parse each into a struct */
94 while ((ptrkey
= strsep(&strings
, "|"))) {
95 ptrvalue
= index(ptrkey
, ':');
101 sortable_keys
[count2
].key
= ptrkey
;
102 sscanf(ptrvalue
, "%f", &sortable_keys
[count2
].value
);
106 /* Sort the structs */
107 qsort(sortable_keys
, count
, sizeof(struct sortable_keys
), sort_subroutine
);
109 for (count2
= 0; count2
< count
; count2
++) {
110 int blen
= strlen(buffer
);
111 if (element_count
++) {
112 strncat(buffer
+ blen
, ",", buflen
- blen
- 1);
115 strncat(buffer
+ blen
, sortable_keys
[count2
].key
, buflen
- blen
- 1);
121 static int cut_internal(struct ast_channel
*chan
, char *data
, char *buffer
, size_t buflen
)
124 AST_DECLARE_APP_ARGS(args
,
125 AST_APP_ARG(varname
);
126 AST_APP_ARG(delimiter
);
130 memset(buffer
, 0, buflen
);
132 parse
= ast_strdupa(data
);
134 AST_STANDARD_APP_ARGS(args
, parse
);
136 /* Check and parse arguments */
141 char *tmp
= alloca(strlen(args
.varname
) + 4);
142 char varvalue
[MAXRESULT
], *tmp2
=varvalue
;
145 snprintf(tmp
, strlen(args
.varname
) + 4, "${%s}", args
.varname
);
146 memset(varvalue
, 0, sizeof(varvalue
));
151 d
= args
.delimiter
[0] ? args
.delimiter
[0] : '-';
153 /* String form of the delimiter, for use with strsep(3) */
154 snprintf(ds
, sizeof(ds
), "%c", d
);
156 pbx_substitute_variables_helper(chan
, tmp
, tmp2
, MAXRESULT
- 1);
160 while (tmp2
!= NULL
&& args
.field
!= NULL
) {
161 char *nextgroup
= strsep(&(args
.field
), "&");
162 int num1
= 0, num2
= MAXRESULT
;
165 if (sscanf(nextgroup
, "%d-%d", &num1
, &num2
) == 2) {
166 /* range with both start and end */
167 } else if (sscanf(nextgroup
, "-%d", &num2
) == 1) {
170 } else if ((sscanf(nextgroup
, "%d%c", &num1
, &trashchar
) == 2) && (trashchar
== '-')) {
171 /* range with start */
173 } else if (sscanf(nextgroup
, "%d", &num1
) == 1) {
180 /* Get to start, if any */
182 while (tmp2
!= (char *)NULL
+ 1 && curfieldnum
< num1
) {
183 tmp2
= index(tmp2
, d
) + 1;
188 /* Most frequent problem is the expectation of reordering fields */
189 if ((num1
> 0) && (curfieldnum
> num1
))
190 ast_log(LOG_WARNING
, "We're already past the field you wanted?\n");
192 /* Re-null tmp2 if we added 1 to NULL */
193 if (tmp2
== (char *)NULL
+ 1)
196 /* Output fields until we either run out of fields or num2 is reached */
197 while (tmp2
!= NULL
&& curfieldnum
<= num2
) {
198 char *tmp3
= strsep(&tmp2
, ds
);
199 int curlen
= strlen(buffer
);
202 snprintf(buffer
+ curlen
, buflen
- curlen
, "%c%s", d
, tmp3
);
204 snprintf(buffer
, buflen
, "%s", tmp3
);
214 static int acf_sort_exec(struct ast_channel
*chan
, char *cmd
, char *data
, char *buf
, size_t len
)
221 switch (sort_internal(chan
, data
, buf
, len
)) {
223 ast_log(LOG_ERROR
, "SORT() requires an argument\n");
226 ast_log(LOG_ERROR
, "Out of memory\n");
232 ast_log(LOG_ERROR
, "Unknown internal error\n");
234 LOCAL_USER_REMOVE(u
);
239 static int acf_cut_exec(struct ast_channel
*chan
, char *cmd
, char *data
, char *buf
, size_t len
)
246 switch (cut_internal(chan
, data
, buf
, len
)) {
248 ast_log(LOG_ERROR
, "Syntax: CUT(<varname>,<char-delim>,<range-spec>) - missing argument!\n");
251 ast_log(LOG_ERROR
, "Out of memory\n");
254 ast_log(LOG_ERROR
, "Usage: CUT(<varname>,<char-delim>,<range-spec>)\n");
260 ast_log(LOG_ERROR
, "Unknown internal error\n");
262 LOCAL_USER_REMOVE(u
);
267 struct ast_custom_function acf_sort
= {
269 .synopsis
= "Sorts a list of key/vals into a list of keys, based upon the vals",
270 .syntax
= "SORT(key1:val1[...][,keyN:valN])",
272 "Takes a comma-separated list of keys and values, each separated by a colon, and returns a\n"
273 "comma-separated list of the keys, sorted by their values. Values will be evaluated as\n"
274 "floating-point numbers.\n",
275 .read
= acf_sort_exec
,
278 struct ast_custom_function acf_cut
= {
280 .synopsis
= "Slices and dices strings, based upon a named delimiter.",
281 .syntax
= "CUT(<varname>,<char-delim>,<range-spec>)",
283 " varname - variable you want cut\n"
284 " char-delim - defaults to '-'\n"
285 " range-spec - number of the field you want (1-based offset)\n"
286 " may also be specified as a range (with -)\n"
287 " or group of ranges and fields (with &)\n",
288 .read
= acf_cut_exec
,
291 static int unload_module(void *mod
)
295 res
|= ast_custom_function_unregister(&acf_cut
);
296 res
|= ast_custom_function_unregister(&acf_sort
);
298 STANDARD_HANGUP_LOCALUSERS
;
303 static int load_module(void *mod
)
307 res
|= ast_custom_function_register(&acf_cut
);
308 res
|= ast_custom_function_register(&acf_sort
);
313 static const char *description(void)
315 return "Cut out information from a string";
318 static const char *key(void)
320 return ASTERISK_GPL_KEY
;
323 STD_MOD(MOD_1
, NULL
, NULL
, NULL
);