#!/usr/bin/python
#  -*- coding: utf-8 -*-
""" This module contains the implementations of abstract base class :py:class:`Escpos`.

:author: `Manuel F Martinez <manpaz@bashlinux.com>`_ and others
:organization: Bashlinux and `python-escpos <https://github.com/python-escpos>`_
:copyright: Copyright (c) 2012-2017 Bashlinux and python-escpos
:license: MIT
"""


import os
import socket
import subprocess
import sys

import serial
import usb.core
import usb.util

from .escpos import Escpos
from .exceptions import USBNotFoundError

_WIN32PRINT = False
try:
    import win32print

    _WIN32PRINT = True
except ImportError:
    pass

_CUPSPRINT = False
try:
    import cups
    import tempfile

    _CUPSPRINT = True
except ImportError:
    pass


class Usb(Escpos):
    """USB printer

    This class describes a printer that natively speaks USB.

    inheritance:

    .. inheritance-diagram:: escpos.printer.Usb
        :parts: 1

    """

    def __init__(
        self,
        idVendor,
        idProduct,
        usb_args=None,
        timeout=0,
        in_ep=0x82,
        out_ep=0x01,
        *args,
        **kwargs
    ):  # noqa: N803
        """
        :param idVendor: Vendor ID
        :param idProduct: Product ID
        :param usb_args: Optional USB arguments (e.g. custom_match)
        :param timeout: Is the time limit of the USB operation. Default without timeout.
        :param in_ep: Input end point
        :param out_ep: Output end point
        """
        Escpos.__init__(self, *args, **kwargs)
        self.timeout = timeout
        self.in_ep = in_ep
        self.out_ep = out_ep

        usb_args = usb_args or {}
        if idVendor:
            usb_args["idVendor"] = idVendor
        if idProduct:
            usb_args["idProduct"] = idProduct
        self.open(usb_args)

    def open(self, usb_args):
        """Search device on USB tree and set it as escpos device.

        :param usb_args: USB arguments
        """
        self.device = usb.core.find(**usb_args)
        if self.device is None:
            raise USBNotFoundError("Device not found or cable not plugged in.")

        self.idVendor = self.device.idVendor
        self.idProduct = self.device.idProduct

        # pyusb has three backends: libusb0, libusb1 and openusb but
        # only libusb1 backend implements the methods is_kernel_driver_active()
        # and detach_kernel_driver().
        # This helps enable this library to work on Windows.
        if self.device.backend.__module__.endswith("libusb1"):
            check_driver = None

            try:
                check_driver = self.device.is_kernel_driver_active(0)
            except NotImplementedError:
                pass

            if check_driver is None or check_driver:
                try:
                    self.device.detach_kernel_driver(0)
                except NotImplementedError:
                    pass
                except usb.core.USBError as e:
                    if check_driver is not None:
                        print("Could not detatch kernel driver: {0}".format(str(e)))

        try:
            self.device.set_configuration()
            self.device.reset()
        except usb.core.USBError as e:
            print("Could not set configuration: {0}".format(str(e)))

    def _raw(self, msg):
        """Print any command sent in raw format

        :param msg: arbitrary code to be printed
        :type msg: bytes
        """
        self.device.write(self.out_ep, msg, self.timeout)

    def _read(self):
        """Reads a data buffer and returns it to the caller."""
        return self.device.read(self.in_ep, 16)

    def close(self):
        """Release USB interface"""
        if self.device:
            usb.util.dispose_resources(self.device)
        self.device = None


class Serial(Escpos):
    """Serial printer

    This class describes a printer that is connected by serial interface.

    inheritance:

    .. inheritance-diagram:: escpos.printer.Serial
        :parts: 1

    """

    def __init__(
        self,
        devfile="/dev/ttyS0",
        baudrate=9600,
        bytesize=8,
        timeout=1,
        parity=serial.PARITY_NONE,
        stopbits=serial.STOPBITS_ONE,
        xonxoff=False,
        dsrdtr=True,
        *args,
        **kwargs
    ):
        """

        :param devfile:  Device file under dev filesystem
        :param baudrate: Baud rate for serial transmission
        :param bytesize: Serial buffer size
        :param timeout:  Read/Write timeout
        :param parity:   Parity checking
        :param stopbits: Number of stop bits
        :param xonxoff:  Software flow control
        :param dsrdtr:   Hardware flow control (False to enable RTS/CTS)
        """
        Escpos.__init__(self, *args, **kwargs)
        self.devfile = devfile
        self.baudrate = baudrate
        self.bytesize = bytesize
        self.timeout = timeout
        self.parity = parity
        self.stopbits = stopbits
        self.xonxoff = xonxoff
        self.dsrdtr = dsrdtr

        self.open()

    def open(self):
        """Setup serial port and set is as escpos device"""
        if self.device is not None and self.device.is_open:
            self.close()
        self.device = serial.Serial(
            port=self.devfile,
            baudrate=self.baudrate,
            bytesize=self.bytesize,
            parity=self.parity,
            stopbits=self.stopbits,
            timeout=self.timeout,
            xonxoff=self.xonxoff,
            dsrdtr=self.dsrdtr,
        )

        if self.device is not None:
            print("Serial printer enabled")
        else:
            print("Unable to open serial printer on: {0}".format(str(self.devfile)))

    def _raw(self, msg):
        """Print any command sent in raw format

        :param msg: arbitrary code to be printed
        :type msg: bytes
        """
        self.device.write(msg)

    def _read(self):
        """Reads a data buffer and returns it to the caller."""
        return self.device.read(16)

    def close(self):
        """Close Serial interface"""
        if self.device is not None and self.device.is_open:
            self.device.flush()
            self.device.close()


class Network(Escpos):
    """Network printer

    This class is used to attach to a networked printer. You can also use this in order to attach to a printer that
    is forwarded with ``socat``.

    If you have a local printer on parallel port ``/dev/usb/lp0`` then you could start ``socat`` with:

    .. code-block:: none

        socat -u TCP4-LISTEN:4242,reuseaddr,fork OPEN:/dev/usb/lp0

    Then you should be able to attach to port ``4242`` with this class.
    Otherwise the normal usecase would be to have a printer with ethernet interface. This type of printer should
    work the same with this class. For the address of the printer check its manuals.

    inheritance:

    .. inheritance-diagram:: escpos.printer.Network
        :parts: 1

    """

    def __init__(self, host, port=9100, timeout=60, *args, **kwargs):
        """

        :param host:    Printer's hostname or IP address
        :param port:    Port to write to
        :param timeout: timeout in seconds for the socket-library
        """
        Escpos.__init__(self, *args, **kwargs)
        self.host = host
        self.port = port
        self.timeout = timeout
        self.open()

    def open(self):
        """Open TCP socket with ``socket``-library and set it as escpos device"""
        self.device = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.device.settimeout(self.timeout)
        self.device.connect((self.host, self.port))

        if self.device is None:
            print("Could not open socket for {0}".format(self.host))

    def _raw(self, msg):
        """Print any command sent in raw format

        :param msg: arbitrary code to be printed
        :type msg: bytes
        """
        self.device.sendall(msg)

    def _read(self):
        """Read data from the TCP socket"""

        return self.device.recv(16)

    def close(self):
        """Close TCP connection"""
        if self.device is not None:
            try:
                self.device.shutdown(socket.SHUT_RDWR)
            except socket.error:
                pass
            self.device.close()


class File(Escpos):
    """Generic file printer

    This class is used for parallel port printer or other printers that are directly attached to the filesystem.
    Note that you should stay away from using USB-to-Parallel-Adapter since they are unreliable
    and produce arbitrary errors.

    inheritance:

    .. inheritance-diagram:: escpos.printer.File
        :parts: 1

    """

    def __init__(self, devfile="/dev/usb/lp0", auto_flush=True, *args, **kwargs):
        """

        :param devfile: Device file under dev filesystem
        :param auto_flush: automatically call flush after every call of _raw()
        """
        Escpos.__init__(self, *args, **kwargs)
        self.devfile = devfile
        self.auto_flush = auto_flush
        self.open()

    def open(self):
        """Open system file"""
        self.device = open(self.devfile, "wb")

        if self.device is None:
            print("Could not open the specified file {0}".format(self.devfile))

    def flush(self):
        """Flush printing content"""
        self.device.flush()

    def _raw(self, msg):
        """Print any command sent in raw format

        :param msg: arbitrary code to be printed
        :type msg: bytes
        """
        self.device.write(msg)
        if self.auto_flush:
            self.flush()

    def close(self):
        """Close system file"""
        if self.device is not None:
            self.device.flush()
            self.device.close()


class Dummy(Escpos):
    """Dummy printer

    This class is used for saving commands to a variable, for use in situations where
    there is no need to send commands to an actual printer. This includes
    generating print jobs for later use, or testing output.

    inheritance:

    .. inheritance-diagram:: escpos.printer.Dummy
        :parts: 1

    """

    def __init__(self, *args, **kwargs):
        """ """
        Escpos.__init__(self, *args, **kwargs)
        self._output_list = []

    def _raw(self, msg):
        """Print any command sent in raw format

        :param msg: arbitrary code to be printed
        :type msg: bytes
        """
        self._output_list.append(msg)

    @property
    def output(self):
        """Get the data that was sent to this printer"""
        return b"".join(self._output_list)

    def clear(self):
        """Clear the buffer of the printer

        This method can be called if you send the contents to a physical printer
        and want to use the Dummy printer for new output.
        """
        del self._output_list[:]

    def close(self):
        pass


if _WIN32PRINT:

    class Win32Raw(Escpos):
        def __init__(self, printer_name=None, *args, **kwargs):
            Escpos.__init__(self, *args, **kwargs)
            if printer_name is not None:
                self.printer_name = printer_name
            else:
                self.printer_name = win32print.GetDefaultPrinter()
            self.hPrinter = None
            self.open()

        def open(self, job_name="python-escpos"):
            if self.printer_name is None:
                raise Exception("Printer not found")
            self.hPrinter = win32print.OpenPrinter(self.printer_name)
            self.current_job = win32print.StartDocPrinter(
                self.hPrinter, 1, (job_name, None, "RAW")
            )
            win32print.StartPagePrinter(self.hPrinter)

        def close(self):
            if not self.hPrinter:
                return
            win32print.EndPagePrinter(self.hPrinter)
            win32print.EndDocPrinter(self.hPrinter)
            win32print.ClosePrinter(self.hPrinter)
            self.hPrinter = None

        def _raw(self, msg):
            """Print any command sent in raw format

            :param msg: arbitrary code to be printed
            :type msg: bytes
            """
            if self.printer_name is None:
                raise Exception("Printer not found")
            if self.hPrinter is None:
                raise Exception("Printer job not opened")
            win32print.WritePrinter(self.hPrinter, msg)


if _CUPSPRINT:

    class CupsPrinter(Escpos):
        """Simple CUPS printer connector.

        .. note::
                Requires _pycups_ which in turn needs the cups development library package:
                - Ubuntu/Debian: _libcups2-dev_
                - OpenSuse/Fedora: _cups-devel_
        """

        def __init__(self, printer_name=None, *args, **kwargs):
            """CupsPrinter class constructor.

            :param printer_name: CUPS printer name (Optional)
            :type printer_name: str
            :param host: CUPS server host/ip (Optional)
            :type host: str
            :param port: CUPS server port (Optional)
            :type port: int
            """
            Escpos.__init__(self, *args, **kwargs)
            host, port = args or (
                kwargs.get("host", cups.getServer()),
                kwargs.get("port", cups.getPort()),
            )
            cups.setServer(host)
            cups.setPort(port)
            self.conn = cups.Connection()
            self.tmpfile = None
            self.printer_name = printer_name
            self.job_name = ""
            self.pending_job = False
            self.open()

        @property
        def printers(self):
            """Available CUPS printers."""
            return self.conn.getPrinters()

        def open(self, job_name="python-escpos"):
            """Setup a new print job and target printer.

            A call to this method is required to send new jobs to
            the same CUPS connection.

            Defaults to default CUPS printer.
            Creates a new temporary file buffer.
            """
            self.job_name = job_name
            if self.printer_name not in self.printers:
                self.printer_name = self.conn.getDefault()
            self.tmpfile = tempfile.NamedTemporaryFile(delete=True)

        def _raw(self, msg):
            """Append any command sent in raw format to temporary file

            :param msg: arbitrary code to be printed
            :type msg: bytes
            """
            self.pending_job = True
            try:
                self.tmpfile.write(msg)
            except ValueError:
                self.pending_job = False
                raise ValueError("Printer job not opened")

        def send(self):
            """Send the print job to the printer."""
            if self.pending_job:
                # Rewind tempfile
                self.tmpfile.seek(0)
                # Print temporary file via CUPS printer.
                self.conn.printFile(
                    self.printer_name,
                    self.tmpfile.name,
                    self.job_name,
                    {"document-format": cups.CUPS_FORMAT_RAW},
                )
            self._clear()

        def _clear(self):
            """Finish the print job.

            Remove temporary file.
            """
            self.tmpfile.close()
            self.pending_job = False

        def _read(self):
            """Return a single-item array with the accepting state of the print queue.

            states: idle = [3], printing a job = [4], stopped = [5]
            """
            printer = self.printers.get(self.printer_name, {})
            state = printer.get("printer-state")
            if not state:
                return []
            return [state]

        def close(self):
            """Close CUPS connection.

            Send pending job to the printer if needed.
            """
            if self.pending_job:
                self.send()
            if self.conn:
                print("Closing CUPS connection to printer {}".format(self.printer_name))
                self.conn = None


if not sys.platform.startswith("win"):

    class LP(Escpos):
        """Simple UNIX lp command raw printing.

        Thanks to `Oyami-Srk comment <https://github.com/python-escpos/python-escpos/pull/348#issuecomment-549558316>`_.
        """

        def __init__(self, printer_name: str, *args, **kwargs):
            """LP class constructor.

            :param printer_name: CUPS printer name (Optional)
            :type printer_name: str
            :param auto_flush: Automatic flush after every _raw() (Optional)
            :type auto_flush: bool
            """
            Escpos.__init__(self, *args, **kwargs)
            self.printer_name = printer_name
            self.auto_flush = kwargs.get("auto_flush", True)
            self.open()

        def open(self):
            """Invoke _lp_ in a new subprocess and wait for commands."""
            self.lp = subprocess.Popen(
                ["lp", "-d", self.printer_name, "-o", "raw"],
                stdin=subprocess.PIPE,
                stdout=open(os.devnull, "w"),
            )

        def close(self):
            """Stop the subprocess."""
            self.lp.terminate()

        def flush(self):
            """End line and wait for new commands"""
            if self.lp.stdin.writable():
                self.lp.stdin.write(b"\n")
            if self.lp.stdin.closed is False:
                self.lp.stdin.close()
            self.lp.wait()
            self.open()

        def _raw(self, msg):
            """Write raw command(s) to the printer.

            :param msg: arbitrary code to be printed
            :type msg: bytes
            """
            if self.lp.stdin.writable():
                self.lp.stdin.write(msg)
            else:
                raise Exception("Not a valid pipe for lp process")
            if self.auto_flush:
                self.flush()