Skip to content
Snippets Groups Projects
app_queue.c 96.4 KiB
Newer Older
  • Learn to ignore specific revisions
  • 	case RES_OUTOFMEMORY:
    		astman_send_error(s, m, "Out of memory");
    		break;
    	}
    	return 0;
    }
    
    static int manager_remove_queue_member(struct mansession *s, struct message *m)
    {
    	char *queuename, *interface;
    
    	queuename = astman_get_header(m, "Queue");
    	interface = astman_get_header(m, "Interface");
    
    	if (ast_strlen_zero(queuename) || ast_strlen_zero(interface)) {
    		astman_send_error(s, m, "Need 'Queue' and 'Interface' parameters.");
    		return 0;
    	}
    
    	switch (remove_from_queue(queuename, interface)) {
    	case RES_OKAY:
    		astman_send_ack(s, m, "Removed interface from queue");
    		break;
    	case RES_EXISTS:
    		astman_send_error(s, m, "Unable to remove interface: Not there");
    		break;
    	case RES_NOSUCHQUEUE:
    		astman_send_error(s, m, "Unable to remove interface from queue: No such queue");
    		break;
    	case RES_OUTOFMEMORY:
    		astman_send_error(s, m, "Out of memory");
    		break;
    	}
    	return 0;
    }
    
    
    static int manager_pause_queue_member(struct mansession *s, struct message *m)
    {
    	char *queuename, *interface, *paused_s;
    	int paused;
    
    	interface = astman_get_header(m, "Interface");
    	paused_s = astman_get_header(m, "Paused");
    	queuename = astman_get_header(m, "Queue");	/* Optional - if not supplied, pause the given Interface in all queues */
    
    	if (ast_strlen_zero(interface) || ast_strlen_zero(paused_s)) {
    		astman_send_error(s, m, "Need 'Interface' and 'Paused' parameters.");
    		return 0;
    	}
    
    	paused = abs(ast_true(paused_s));
    
    	if (set_member_paused(queuename, interface, paused))
    		astman_send_error(s, m, "Interface not found");
    	else
    		if (paused)
    			astman_send_ack(s, m, "Interface paused successfully");
    		else
    			astman_send_ack(s, m, "Interface unpaused successfully");
    
    	return 0;
    }
    
    
    static int handle_add_queue_member(int fd, int argc, char *argv[])
    {
    	char *queuename, *interface;
    	int penalty;
    
    	if ((argc != 6) && (argc != 8)) {
    		return RESULT_SHOWUSAGE;
    	} else if (strcmp(argv[4], "to")) {
    		return RESULT_SHOWUSAGE;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	} else if ((argc == 8) && strcmp(argv[6], "penalty")) {
    
    		return RESULT_SHOWUSAGE;
    	}
    
    	queuename = argv[5];
    	interface = argv[3];
    	if (argc == 8) {
    		if (sscanf(argv[7], "%d", &penalty) == 1) {
    			if (penalty < 0) {
    				ast_cli(fd, "Penalty must be >= 0\n");
    				penalty = 0;
    			}
    		} else {
    			ast_cli(fd, "Penalty must be an integer >= 0\n");
    			penalty = 0;
    		}
    	} else {
    		penalty = 0;
    	}
    
    
    	switch (add_to_queue(queuename, interface, penalty, 0, queue_persistent_members)) {
    
    	case RES_OKAY:
    		ast_cli(fd, "Added interface '%s' to queue '%s'\n", interface, queuename);
    		return RESULT_SUCCESS;
    	case RES_EXISTS:
    		ast_cli(fd, "Unable to add interface '%s' to queue '%s': Already there\n", interface, queuename);
    		return RESULT_FAILURE;
    	case RES_NOSUCHQUEUE:
    		ast_cli(fd, "Unable to add interface to queue '%s': No such queue\n", queuename);
    		return RESULT_FAILURE;
    	case RES_OUTOFMEMORY:
    		ast_cli(fd, "Out of memory\n");
    		return RESULT_FAILURE;
    	default:
    		return RESULT_FAILURE;
    	}
    }
    
    static char *complete_add_queue_member(char *line, char *word, int pos, int state)
    {
    	/* 0 - add; 1 - queue; 2 - member; 3 - <member>; 4 - to; 5 - <queue>; 6 - penalty; 7 - <penalty> */
    	switch (pos) {
    	case 3:
    		/* Don't attempt to complete name of member (infinite possibilities) */
    		return NULL;
    	case 4:
    		if (state == 0) {
    			return strdup("to");
    		} else {
    			return NULL;
    		}
    	case 5:
    		/* No need to duplicate code */
    		return complete_queue(line, word, pos, state);
    	case 6:
    		if (state == 0) {
    			return strdup("penalty");
    		} else {
    			return NULL;
    		}
    	case 7:
    		if (state < 100) {	/* 0-99 */
    			char *num = malloc(3);
    			if (num) {
    				sprintf(num, "%d", state);
    			}
    			return num;
    		} else {
    			return NULL;
    		}
    	default:
    		return NULL;
    	}
    }
    
    static int handle_remove_queue_member(int fd, int argc, char *argv[])
    {
    	char *queuename, *interface;
    
    	if (argc != 6) {
    		return RESULT_SHOWUSAGE;
    	} else if (strcmp(argv[4], "from")) {
    		return RESULT_SHOWUSAGE;
    	}
    
    	queuename = argv[5];
    	interface = argv[3];
    
    	switch (remove_from_queue(queuename, interface)) {
    	case RES_OKAY:
    		ast_cli(fd, "Removed interface '%s' from queue '%s'\n", interface, queuename);
    		return RESULT_SUCCESS;
    	case RES_EXISTS:
    		ast_cli(fd, "Unable to remove interface '%s' from queue '%s': Not there\n", interface, queuename);
    		return RESULT_FAILURE;
    	case RES_NOSUCHQUEUE:
    		ast_cli(fd, "Unable to remove interface from queue '%s': No such queue\n", queuename);
    		return RESULT_FAILURE;
    	case RES_OUTOFMEMORY:
    		ast_cli(fd, "Out of memory\n");
    		return RESULT_FAILURE;
    	default:
    		return RESULT_FAILURE;
    	}
    }
    
    static char *complete_remove_queue_member(char *line, char *word, int pos, int state)
    {
    	int which = 0;
    	struct ast_call_queue *q;
    	struct member *m;
    
    	/* 0 - add; 1 - queue; 2 - member; 3 - <member>; 4 - to; 5 - <queue> */
    	if ((pos > 5) || (pos < 3)) {
    		return NULL;
    	}
    	if (pos == 4) {
    		if (state == 0) {
    			return strdup("from");
    		} else {
    			return NULL;
    		}
    	}
    
    	if (pos == 5) {
    		/* No need to duplicate code */
    		return complete_queue(line, word, pos, state);
    	}
    
    	if (queues != NULL) {
    		for (q = queues ; q ; q = q->next) {
    			ast_mutex_lock(&q->lock);
    			for (m = q->members ; m ; m = m->next) {
    				if (++which > state) {
    					ast_mutex_unlock(&q->lock);
    
    					return strdup(m->interface);
    
    Mark Spencer's avatar
    Mark Spencer committed
    static char show_queues_usage[] = 
    "Usage: show queues\n"
    "       Provides summary information on call queues.\n";
    
    static struct ast_cli_entry cli_show_queues = {
    	{ "show", "queues", NULL }, queues_show, 
    	"Show status of queues", show_queues_usage, NULL };
    
    
    static char show_queue_usage[] = 
    "Usage: show queue\n"
    "       Provides summary information on a specified queue.\n";
    
    static struct ast_cli_entry cli_show_queue = {
    	{ "show", "queue", NULL }, queue_show, 
    	"Show status of a specified queue", show_queue_usage, complete_queue };
    
    
    static char aqm_cmd_usage[] =
    "Usage: add queue member <channel> to <queue> [penalty <penalty>]\n";
    
    static struct ast_cli_entry cli_add_queue_member = {
    	{ "add", "queue", "member", NULL }, handle_add_queue_member,
    	"Add a channel to a specified queue", aqm_cmd_usage, complete_add_queue_member };
    
    static char rqm_cmd_usage[] =
    "Usage: remove queue member <channel> from <queue>\n";
    
    static struct ast_cli_entry cli_remove_queue_member = {
    	{ "remove", "queue", "member", NULL }, handle_remove_queue_member,
    	"Removes a channel from a specified queue", rqm_cmd_usage, complete_remove_queue_member };
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    int unload_module(void)
    {
    	STANDARD_HANGUP_LOCALUSERS;
    
    	ast_cli_unregister(&cli_show_queue);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	ast_cli_unregister(&cli_show_queues);
    
    	ast_cli_unregister(&cli_add_queue_member);
    	ast_cli_unregister(&cli_remove_queue_member);
    	ast_manager_unregister("Queues");
    	ast_manager_unregister("QueueStatus");
    	ast_manager_unregister("QueueAdd");
    	ast_manager_unregister("QueueRemove");
    
    	ast_manager_unregister("QueuePause");
    
    Mark Spencer's avatar
    Mark Spencer committed
    	ast_devstate_del(statechange_queue, NULL);
    
    	ast_unregister_application(app_aqm);
    	ast_unregister_application(app_rqm);
    
    	ast_unregister_application(app_pqm);
    	ast_unregister_application(app_upqm);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	return ast_unregister_application(app);
    }
    
    int load_module(void)
    {
    	int res;
    	res = ast_register_application(app, queue_exec, synopsis, descrip);
    
    Mark Spencer's avatar
    Mark Spencer committed
    	if (!res) {
    
    		ast_cli_register(&cli_show_queue);
    
    Mark Spencer's avatar
    Mark Spencer committed
    		ast_cli_register(&cli_show_queues);
    
    		ast_cli_register(&cli_add_queue_member);
    		ast_cli_register(&cli_remove_queue_member);
    
    Mark Spencer's avatar
    Mark Spencer committed
    		ast_devstate_add(statechange_queue, NULL);
    
    Mark Spencer's avatar
    Mark Spencer committed
    		ast_manager_register( "Queues", 0, manager_queues_show, "Queues" );
    
    		ast_manager_register( "QueueStatus", 0, manager_queues_status, "Queue Status" );
    
    		ast_manager_register( "QueueAdd", EVENT_FLAG_AGENT, manager_add_queue_member, "Add interface to queue." );
    		ast_manager_register( "QueueRemove", EVENT_FLAG_AGENT, manager_remove_queue_member, "Remove interface from queue." );
    
    		ast_manager_register( "QueuePause", EVENT_FLAG_AGENT, manager_pause_queue_member, "Makes a queue member temporarily unavailable" );
    
    		ast_register_application(app_aqm, aqm_exec, app_aqm_synopsis, app_aqm_descrip) ;
    		ast_register_application(app_rqm, rqm_exec, app_rqm_synopsis, app_rqm_descrip) ;
    
    		ast_register_application(app_pqm, pqm_exec, app_pqm_synopsis, app_pqm_descrip) ;
    		ast_register_application(app_upqm, upqm_exec, app_upqm_synopsis, app_upqm_descrip) ;
    
    Mark Spencer's avatar
    Mark Spencer committed
    	}
    
    Mark Spencer's avatar
    Mark Spencer committed
    	reload_queues();
    
    	
    	if (queue_persistent_members)
    	    reload_queue_members();
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    	return res;
    }
    
    
    Mark Spencer's avatar
    Mark Spencer committed
    int reload(void)
    {
    	reload_queues();
    	return 0;
    }
    
    char *description(void)
    {
    	return tdesc;
    }
    
    int usecount(void)
    {
    	int res;
    	STANDARD_USECOUNT(res);
    	return res;
    }
    
    char *key()
    {
    	return ASTERISK_GPL_KEY;
    }