/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 *
 */

/* $Id: disable.c 146 2006-03-24 00:26:54Z njacobs $ */


#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <locale.h>
#include <libintl.h>
#include <papi.h>
#include "common.h"

static void
usage(char *program)
{
	char *name;

	if ((name = strrchr(program, '/')) == NULL)
		name = program;
	else
		name++;

	fprintf(stdout,
	    gettext("Usage: %s [-c] [-W] [-r reason] destination ...\n"),
	    name);
	exit(1);
}

static void
cancel_active_job(papi_service_t svc, char *dest)
{
	papi_status_t status;
	papi_job_t *j = NULL;
	char *req_attrs[] = { "job-state", "job-id", NULL };

	status = papiPrinterListJobs(svc, dest, req_attrs, 0, 0, &j);
	if ((status == PAPI_OK) && (j != NULL)) {
		int i;

		for (i = 0; j[i] != NULL; j++) {
			papi_attribute_t **a = papiJobGetAttributeList(j[i]);
			int state = 0;

			if (a == NULL)
				continue;

			(void) papiAttributeListGetInteger(a, NULL,
			    "job-state", &state);
			if (state & 0x082A) { /* If state is RS_ACTIVE */
				int32_t id = papiJobGetId(j[i]);

				(void) papiJobCancel(svc, dest, id);
			}
		}
		papiJobListFree(j);
	}
}

int
main(int ac, char *av[])
{
	papi_status_t status;
	papi_service_t svc = NULL;
	papi_encryption_t encryption = PAPI_ENCRYPT_NEVER;
	int exit_status = 0;
	int cancel = 0;
	int pending = 0;	/* not implemented */
	char *reason = NULL;
	int c;

	(void) setlocale(LC_ALL, "");
	(void) textdomain("SUNW_OST_OSCMD");

	while ((c = getopt(ac, av, "EcWr:")) != EOF)
		switch (c) {
		case 'c':	/* cancel active job first */
			cancel = 1;
			break;
		case 'W':	/* wait for active request, not implemented */
			pending = 1;
			break;
		case 'r':	/* reason */
			reason = optarg;
			break;
		case 'E':
			encryption = PAPI_ENCRYPT_NEVER;
			break;
		default:
			usage(av[0]);
		}

	if (ac <= optind)
		usage(av[0]);

	while (optind < ac) {
		char *printer = av[optind++];

		status = papiServiceCreate(&svc, printer, NULL, NULL,
		    cli_auth_callback, encryption, NULL);
		if (status != PAPI_OK) {
			fprintf(stderr, gettext(
			    "Failed to contact service for %s: %s\n"),
			    printer, verbose_papi_message(svc, status));
			exit_status = 1;
		}

		status = papiPrinterDisable(svc, printer, reason);
		if (status == PAPI_OK) {
			printf(gettext("printer \"%s\" now disabled\n"),
			    printer);
		} else if (status == PAPI_NOT_ACCEPTING) {
			fprintf(stderr, gettext(
			    "Destination \"%s\" was already disabled.\n"),
			    printer);
			exit_status = 1;
		} else {
			/* The operation is not supported in lpd protocol */
			if (status == PAPI_OPERATION_NOT_SUPPORTED) {
				fprintf(stderr,
				    verbose_papi_message(svc, status));
			} else {
				fprintf(stderr, gettext("disable: %s: %s\n"),
				    printer, verbose_papi_message(svc, status));
			}
			exit_status = 1;
		}

		if (cancel != 0)
			cancel_active_job(svc, printer);

		papiServiceDestroy(svc);
	}

	return (exit_status);
}