package qz.printer.info;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import qz.utils.SystemUtilities;

import javax.print.PrintService;
import javax.print.attribute.standard.Media;
import javax.print.attribute.standard.PrinterName;
import javax.print.attribute.standard.PrinterResolution;
import java.util.Arrays;
import java.util.List;

public class NativePrinter {
    private static final Logger log = LogManager.getLogger(NativePrinter.class);
    /**
     * Simple object wrapper allowing lazy fetching of values
     * @param <T>
     */
    public class PrinterProperty<T> {
        T value;
        boolean set;

        public PrinterProperty() {
            this.set = false;
        }

        @Override
        public String toString() {
            if (value == null) {
                return null;
            } else if (value instanceof String) {
                return (String)value;
            } return value.toString();
        }

        public void set() {
            this.set = true;
        }

        public void set(T content) {
            this.value = content;
            this.set = true;
        }

        public T value() {
            return value;
        }

        public boolean isSet() {
            return set;
        }

        public boolean isNull() {
            return value == null;
        }

        @Override
        public boolean equals(Object o) {
            if (value != null) {
                return value.equals(o);
            }
            return false;
        }
    }

    private final String printerId;

    private boolean outdated;
    private PrinterProperty<String> description;
    private PrinterProperty<PrintService> printService;
    private PrinterProperty<String> driver;
    private PrinterProperty<String> connection;
    private PrinterProperty<PrinterResolution> resolution;
    private PrinterProperty<String> driverFile;

    public NativePrinter(String printerId) {
        this.printerId = printerId;
        this.description = new PrinterProperty<>();
        this.printService = new PrinterProperty<>();
        this.driverFile = new PrinterProperty<>();
        this.driver = new PrinterProperty<>();
        this.connection = new PrinterProperty<>();
        this.resolution = new PrinterProperty<>();
        this.outdated = false;
    }

    public PrinterProperty<String> getDescription() {
        return description;
    }

    public void setDescription(String description) {
        this.description.set(description);
    }

    public PrinterProperty<String> getDriverFile() {
        return driverFile;
    }

    public void setDriverFile(String driverFile) {
        this.driverFile.set(driverFile);
    }

    public PrinterProperty<String> getDriver() {
        if (!driver.isSet()) {
            getDriverAttributes(this);
        }
        return driver;
    }

    public String getName() {
        if (printService != null && printService.value() != null) {
            return printService.value().getName();
        }
        return null;
    }

    public PrinterName getLegacyName() {
        if (printService != null && printService.value() != null) {
            return printService.value().getAttribute(PrinterName.class);
        }
        return null;
    }

    public void setDriver(String driver) {
        this.driver.set(driver);
    }

    public void setConnection(String connection) {
        this.connection.set(connection);
    }

    public String getConnection() {
        if (!connection.isSet()) {
            getDriverAttributes(this);
        }
        return connection.value();
    }

    public PrinterProperty<PrintService> getPrintService() {
        return printService;
    }

    public void setPrintService(PrintService printService) {
        this.printService.set(printService);
    }

    public String getPrinterId() {
        return printerId;
    }

    public void setResolution(PrinterResolution resolution) {
        this.resolution.set(resolution);
    }

    public PrinterProperty<PrinterResolution> getResolution() {
        if (!resolution.isSet()) {
            // Fetch resolution, if available
            try {
                Object resolution = printService.value().getDefaultAttributeValue(PrinterResolution.class);
                if (resolution != null) {
                    this.resolution.set((PrinterResolution)resolution);
                }
            }
            catch(IllegalArgumentException e) {
                log.warn("Unable to obtain PrinterResolution from {}", printService.value().getName(), e);
            }
        }

        return resolution;
    }

    public List<PrinterResolution> getResolutions() {
        PrintService ps = getPrintService().value();
        PrinterResolution[] resSupport = (PrinterResolution[])ps.getSupportedAttributeValues(PrinterResolution.class, ps.getSupportedDocFlavors()[0], null);
        if (resSupport == null || resSupport.length == 0) {
            NativePrinterMap printerMap = NativePrinterMap.getInstance();
            // CUPS doesn't report resolutions properly, instead return the values scraped from console
            if(printerMap instanceof CupsPrinterMap) {
                return ((CupsPrinterMap)printerMap).getResolutions(this);
            }
            resSupport = new PrinterResolution[]{ getResolution().value() };
        }

        return Arrays.asList(resSupport);
    }

    public static void getDriverAttributes(NativePrinter printer) {
        // First, perform slow JDK operations, see issues #940, #932
        printer.getPrintService().value().getSupportedAttributeValues(Media.class, null, null); // cached by JDK
        printer.getResolution();

        // Mark properties as "found" so we don't attempt to gather them again
        printer.driver.set();
        printer.resolution.set();
        printer.connection.set();

        // Gather properties not exposed by the JDK
        NativePrinterMap.getInstance().fillAttributes(printer);
    }

    public boolean isOutdated() {
        return outdated;
    }

    public void setOutdated(boolean outdated) {
        this.outdated = outdated;
    }
}