/*

 * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/*

 * @test

 * @bug      6921664

 * @summary  Verifies number of copies and the job name are passed to a

 *           3rd party PrintService.

 * @run      main DummyPrintTest

 */

import java.awt.Graphics;

import java.awt.print.PageFormat;

import java.awt.print.Printable;

import java.awt.print.PrinterException;

import java.awt.print.PrinterJob;

import java.util.HashSet;

import java.util.Set;

import javax.print.Doc;

import javax.print.DocFlavor;

import javax.print.DocPrintJob;

import javax.print.PrintException;

import javax.print.PrintService;

import javax.print.PrintServiceLookup;

import javax.print.ServiceUIFactory;

import javax.print.attribute.Attribute;

import javax.print.attribute.AttributeSet;

import javax.print.attribute.HashPrintJobAttributeSet;

import javax.print.attribute.HashPrintServiceAttributeSet;

import javax.print.attribute.PrintJobAttributeSet;

import javax.print.attribute.PrintRequestAttributeSet;

import javax.print.attribute.PrintServiceAttribute;

import javax.print.attribute.PrintServiceAttributeSet;

import javax.print.attribute.standard.Copies;

import javax.print.attribute.standard.JobName;

import javax.print.attribute.standard.PrinterName;

import javax.print.attribute.standard.PrinterState;

import javax.print.event.PrintJobAttributeListener;

import javax.print.event.PrintJobListener;

import javax.print.event.PrintServiceAttributeListener;

public class DummyPrintTest {

    public static void main(String[] args) throws Exception {

        // register custom print service implementation

        String printerName = "myDummyPrintService";

        PrintServiceLookup.registerService(new DummyPrintService(printerName));

        // calling third party print logic

        thirdPartyPrintLogic(printerName);

    }

    static void thirdPartyPrintLogic(String printerName) throws Exception {

        PrinterJob printerjob = PrinterJob.getPrinterJob();

        printerjob.setCopies(2);

        printerjob.setJobName("myJobName");

        printerjob.setPrintable(new DummyPrintable());

        for (PrintService printService : PrinterJob.lookupPrintServices()) {

            System.out.println("check printer name of service " + printService);

            if (printerName.equals(printService.getName())) {

                System.out.println("correct printer service do print...");

                printerjob.setPrintService(printService);

                printerjob.print();

                break;

            }

        }

    }

}

class DummyPrintService implements PrintService {

    private final String _name;

    private final Set<DocFlavor> _supportedFlavors;

    private final PrintServiceAttributeSet _printServiceAttributeSet;

    public DummyPrintService(String name) {

        _name = name;

        _supportedFlavors = new HashSet<DocFlavor>();

        _supportedFlavors.add(DocFlavor.SERVICE_FORMATTED.PAGEABLE);

        _supportedFlavors.add(DocFlavor.SERVICE_FORMATTED.PRINTABLE);

        _printServiceAttributeSet = new HashPrintServiceAttributeSet();

        _printServiceAttributeSet.add(new PrinterName(name, null));

        _printServiceAttributeSet.add(PrinterState.IDLE);

    }

    @Override

    public String toString() {

        return "Dummy Printer : " + getName();

    }

    @Override

    public String getName() {

        return _name;

    }

    @Override

    public DocPrintJob createPrintJob() {

        return new DummyDocPrintJob(this);

    }

    @Override

    public boolean isDocFlavorSupported(DocFlavor flavor) {

        return _supportedFlavors.contains(flavor);

    }

    @Override

    public <T extends PrintServiceAttribute> T getAttribute(Class<T> category) {

        return category.cast(_printServiceAttributeSet.get(category));

    }

    @Override

    public PrintServiceAttributeSet getAttributes() {

        return _printServiceAttributeSet;

    }

    @Override

    public DocFlavor[] getSupportedDocFlavors() {

        return _supportedFlavors.toArray(new DocFlavor[_supportedFlavors.size()]);

    }

    @Override

    public Object getDefaultAttributeValue(Class<? extends Attribute> category) {

        return null;

    }

    @Override

    public ServiceUIFactory getServiceUIFactory() {

        return null;

    }

    @Override

    public Class<?>[] getSupportedAttributeCategories() {

        return null;

    }

    @Override

    public Object getSupportedAttributeValues(Class<? extends Attribute> category,

                                    DocFlavor flavor, AttributeSet attributes) {

        return null;

    }

    @Override

    public AttributeSet getUnsupportedAttributes(DocFlavor flavor,

                                                 AttributeSet attributes) {

        return null;

    }

    @Override

    public boolean isAttributeCategorySupported(Class<? extends Attribute> category) {

        return false;

    }

    @Override

    public boolean isAttributeValueSupported(Attribute attrval,

                                             DocFlavor flavor,

                                             AttributeSet attributes) {

        return false;

    }

    @Override

    public void addPrintServiceAttributeListener(PrintServiceAttributeListener listener) {

    }

    @Override

    public void removePrintServiceAttributeListener(PrintServiceAttributeListener listener) {

    }

}

class DummyDocPrintJob implements DocPrintJob {

    private static int _counter;

    private final PrintService _printService;

    private final PrintJobAttributeSet _printJobAttributeSet;

    public DummyDocPrintJob(PrintService printService) {

        _counter++;

        _printService = printService;

        _printJobAttributeSet = new HashPrintJobAttributeSet();

    }

    @Override

    public PrintService getPrintService() {

        return _printService;

    }

    @Override

    public PrintJobAttributeSet getAttributes() {

        return _printJobAttributeSet;

    }

    @Override

    public void addPrintJobAttributeListener(PrintJobAttributeListener listener,

                                    PrintJobAttributeSet printJobAttributeSet) {

    }

    @Override

    public void removePrintJobAttributeListener(PrintJobAttributeListener listener) {

    }

    @Override

    public void addPrintJobListener(PrintJobListener listener) {

    }

    @Override

    public void removePrintJobListener(PrintJobListener listener) {

    }

    @Override

    public void print(Doc doc,

                      PrintRequestAttributeSet printRequestAttributeSet)

          throws PrintException {

        System.out.println("job name: " + printRequestAttributeSet.get(JobName.class));

        System.out.println("copies: " + printRequestAttributeSet.get(Copies.class));

        if(printRequestAttributeSet.get(JobName.class) == null ||

            printRequestAttributeSet.get(Copies.class) == null) {

            throw new RuntimeException("Copies and JobName is not passed correctly");

        }

    }

}

class DummyPrintable implements Printable {

    @Override

    public int print(Graphics graphics, PageFormat pageFormat, int pageIndex)

            throws PrinterException {

        if (pageIndex == 0) {

            return Printable.PAGE_EXISTS;

        } else {

            return Printable.NO_SUCH_PAGE;

        }

    }

}