/*
 * 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 (c) 2006, 2010, Oracle and/or its affiliates. All rights reserved.
 *
 */

/* $Id: cancel.c 147 2006-04-25 16:51:06Z 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, "Usage: %s [-u user] (printer|request-id ...)\n", name);
        exit(1);
}

static int32_t
get_job_id_requested(papi_job_t job) {
        int32_t rid = -1;

        papi_attribute_t **list = papiJobGetAttributeList(job);
        papiAttributeListGetInteger(list, NULL,
            "job-id-requested", &rid);

        return (rid);
}

int
cancel_jobs_for_user(char *user, papi_encryption_t encryption, char *pname) {

        papi_status_t status;
        papi_service_t svc = NULL;
        char **printers = NULL;
        int i, exit_code;

        if (pname == NULL) {
                status = papiServiceCreate(&svc, NULL, NULL, NULL,
                    cli_auth_callback, encryption, NULL);
                printers = interest_list(svc);
                papiServiceDestroy(svc);
        } else {
                list_append(&printers, strdup(pname));
        }

        if (printers == NULL)
                exit(0);

        for (i = 0; printers[i] != NULL; i++) {
                char *printer = printers[i];

                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(1);
                }
                exit_code = berkeley_cancel_request(svc, stdout, printer, 1,
                    &user);

                papiServiceDestroy(svc);
                if (exit_code != 0)
                        break;
        }
        free(printers);
        return (exit_code);
}

int
main(int ac, char *av[])
{
        int exit_code = 0;
        char *user = NULL;
        papi_encryption_t encryption = PAPI_ENCRYPT_NEVER;
        int c;
        int32_t rid = -1;
        int first_dest = 0;


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

        if (ac == 1)
                usage(av[0]);

        while ((c = getopt(ac, av, "Eu:")) != EOF)
                switch (c) {
                case 'E':
                        encryption = PAPI_ENCRYPT_REQUIRED;
                        break;
                case 'u':
                        user = optarg;
                        break;
                default:
                        usage(av[0]);
                }

        for (c = optind; c < ac; c++) {
                papi_status_t status;
                papi_service_t svc = NULL;
                papi_job_t *jobs = NULL;
                char *printer = NULL;
                int32_t id = -1;

                status = papiServiceCreate(&svc, av[c], NULL, NULL,
                    cli_auth_callback, encryption, NULL);
                if (status != PAPI_OK) {
                        if (first_dest == 0) {
                                (void) get_printer_id(av[c], &printer, &id);
                                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(1);
                        }
                } else {
                        first_dest = 1;
                        printer = av[c];
                }

#define OUT     ((status == PAPI_OK) ? stdout : stderr)

                if (id != -1) { /* it's a job */
                        char *mesg = gettext("cancelled");

                        /*
                         * Check if the job-id is job-id-requested
                         * or job-id. If it is job-id-requested then find
                         * corresponding job-id and send it to cancel
                         */
                        rid = job_to_be_queried(svc, printer, id);
                        if (rid < 0) {
                                /*
                                 * Either it is a remote job which cannot be
                                 * cancelled based on job-id or job-id is
                                 * not found
                                 */
                                exit_code = 1;
                                fprintf(OUT, "%s-%d: %s\n",
                                    printer, id, gettext("not-found"));
                        } else {
                                status = papiJobCancel(svc, printer, rid);
                                if (status == PAPI_NOT_AUTHORIZED) {
                                        mesg = papiStatusString(status);
                                        exit_code = 1;
                                } else if (status != PAPI_OK) {
                                        mesg = gettext(
                                            verbose_papi_message(
                                            svc, status));
                                        exit_code = 1;
                                }
                                fprintf(OUT, "%s-%d: %s\n", printer, id, mesg);
                        }

                } else {        /* it's a printer */
                        if (user == NULL) {

                                /* Remove first job from printer */

                                status = papiPrinterListJobs(svc, printer,
                                    NULL, 0, 0, &jobs);

                                if (status != PAPI_OK) {
                                        fprintf(stderr, gettext(
                                            "ListJobs %s: %s\n"), printer,
                                            verbose_papi_message(svc, status));
                                        exit_code = 1;
                                }

                                if (jobs != NULL && *jobs != NULL) {
                                        char *mesg = gettext("cancelled");
                                        id = papiJobGetId(*jobs);

                                        status = papiJobCancel(svc,
                                            printer, id);

                                        if (status == PAPI_NOT_AUTHORIZED) {
                                                mesg = papiStatusString(status);
                                                exit_code = 1;
                                        } else if (status != PAPI_OK) {
                                                mesg = gettext(
                                                    verbose_papi_message(
                                                    svc, status));
                                                exit_code = 1;
                                        }
                                        /*
                                         * If job-id-requested exists for this
                                         * job-id then that should be displayed
                                         */
                                        rid = get_job_id_requested(*jobs);
                                        if (rid >= 0)
                                                fprintf(OUT, "%s-%d: %s\n",
                                                    printer, rid, mesg);
                                        else
                                                fprintf(OUT, "%s-%d: %s\n",
                                                    printer, id, mesg);
                                }
                                papiJobListFree(jobs);

                        } else {
                                /* Purging user's print jobs */
                                exit_code = cancel_jobs_for_user(user,
                                    encryption, printer);
                        }
                }
                papiServiceDestroy(svc);
        }

        if (optind == ac)
                exit_code = cancel_jobs_for_user(user, encryption, NULL);

        return (exit_code);
}