# Python imports
import html, time

import pyscreenshot as capture

# Lib imports
import gi, cairo
gi.require_version('Gdk', '3.0')

from gi.repository import Gdk
from gi.repository import GLib

# Application imports




class MouseButtons:
    LEFT_BUTTON  = 1
    RIGHT_BUTTON = 3


class DrawingArea:
    def __init__(self):
        self.builder       = settings.get_builder()
        self.main_window   = self.builder.get_object('Main_Window')
        self.region_window = self.builder.get_object('Region_Window')
        self.region_menu   = self.builder.get_object('regionMenu')
        self.message_label = self.builder.get_object('messageLabel')
        MONITOR            = settings.get_monitor_data()

        settings.set_window_data(self.region_window)
        self.region_window.set_default_size(MONITOR[0].width, MONITOR[0].height)
        self.region_window.set_size_request(MONITOR[0].width, MONITOR[0].height)
        self.region_window.set_keep_above(True)

        self.DRAW_AREA = self.builder.get_object("selectionArea")
        self.DRAW_AREA.add_events(Gdk.EventMask.BUTTON_PRESS_MASK)
        self.DRAW_AREA.add_events(Gdk.EventMask.BUTTON_RELEASE_MASK)
        self.DRAW_AREA.add_events(Gdk.EventMask.BUTTON1_MOTION_MASK)
        self.DRAW_AREA.connect("button-press-event", self.on_button_press)
        self.DRAW_AREA.connect("button-release-event", self.on_button_release)
        self.DRAW_AREA.connect("motion-notify-event", self.on_mouse_move)
        self.DRAW_AREA.connect("draw", self.on_draw)

        area                 = settings.get_monitor_data()[0]
        self.SCREENSHOTS_DIR = settings.get_screenshots_dir()
        self.WIN_REC         = [area.x, area.y, area.width, area.height]
        self.coords          = [[0.0, 0.0], [0.0, 0.0]] # point1 and point2
        self.BORDER_COLOR    = [255, 0, 0, 0.84]
        self.TRANS_COLOR     = [0, 0, 0, 0.0]
        self.BG_COLOR        = [0, 0, 0, 0.4]
        self.success         = "#88cc27"
        self.warning         = "#ffa800"
        self.error           = "#ff0000"
        self.rec             = None
        self.cr              = None

        self.region_window.set_default_size(area.width, area.height)
        self.region_window.set_size_request(area.width, area.height)
        self.region_window.move(area.x, area.y)
        self.region_window.set_resizable(False)
        self.region_window.set_keep_above(True)


    def on_button_press(self, w, e):
        self.message_label.set_markup("")
        if e.type == Gdk.EventType.BUTTON_PRESS and e.button == MouseButtons.LEFT_BUTTON:
            self.coords[0] = [e.x, e.y]
            self.region_menu.hide()

            # This will reset draw area initially. No further use
            if self.cr:
                self.draw(self.cr, self.WIN_REC, self.BG_COLOR)
                self.DRAW_AREA.queue_draw()
        if e.type == Gdk.EventType.BUTTON_PRESS and e.button == MouseButtons.RIGHT_BUTTON:
            self.region_menu.show()

    # Update second set of coords.
    def on_mouse_move(self, w, e):
        self.coords[1] = [e.x, e.y]
        self.DRAW_AREA.queue_draw()

    def on_button_release(self, w, e):
        if e.type == Gdk.EventType.BUTTON_RELEASE and e.button == MouseButtons.LEFT_BUTTON:
            self.region_menu.show()

    @daemon_threaded
    def grab_region(self, widget):
        GLib.idle_add(self.main_window.hide)
        GLib.idle_add(self.region_menu.hide)

        while self.main_window.is_visible():
            ...
        while self.region_menu.is_visible():
            ...

        time.sleep(0.05)
        self.grab_region_idle()

    def go_to_main_window(self, widget):
        self.go_to_main_window_idle()

    def grab_region_idle(self):
        self.do_bounding_box_grab(self.rec[0], self.rec[1], self.rec[2], self.rec[3])
        GLib.idle_add(utils.referesh_directory_list)
        GLib.idle_add(self.main_window.show)
        GLib.idle_add(self.region_menu.show)

    def go_to_main_window_idle(self):
        self.region_window.hide()
        self.region_menu.hide()
        self.main_window.show()


    def on_draw(self, wid, cr):
        if not self.cr: self.cr = cr
        # Reset the screen with transparent view
        self.draw(cr, self.WIN_REC, self.BG_COLOR)

        point1   = self.coords[0]
        point2   = self.coords[1]
        x1       = point1[0]
        y1       = point1[1]
        x2       = point2[0]
        y2       = point2[1]
        w        = x2 - x1
        h        = y2 - y1

        # Rectangle information for region and screen grab
        self.rec = [int(x1), int(y1), int(x2), int(y2)]
        # Draw the new selection region
        self.selection_draw(cr, [x1, y1, w, h], self.BORDER_COLOR, self.TRANS_COLOR)


    def draw(self, cr, x1y1wh, rgba):
        cr.set_source_rgba(rgba[0], rgba[1], rgba[2], rgba[3])
        cr.rectangle(x1y1wh[0], x1y1wh[1], x1y1wh[2], x1y1wh[3])
        cr.set_operator(1);
        cr.fill()

    def selection_draw(self, cr, x1y1wh, brdrcol, transclr):
        # Clear the region
        cr.set_source_rgba(transclr[0], transclr[1], transclr[2], transclr[3])
        cr.rectangle(x1y1wh[0], x1y1wh[1], x1y1wh[2], x1y1wh[3])
        cr.set_operator(0);
        cr.fill()

        # Draw a border
        cr.set_source_rgba(brdrcol[0], brdrcol[1], brdrcol[2], brdrcol[3])
        cr.rectangle(x1y1wh[0] - 2, x1y1wh[1] - 2, x1y1wh[2] + 4, x1y1wh[3] + 4)
        cr.set_operator(1);
        cr.stroke()

    # Actual region grab
    def do_bounding_box_grab(self, x1, y1, x2, y2):
        try:
            temp = 0;
            if x2 < x1:
                temp = x1
                x1   = x2
                x2   = temp

            if y2 < y1:
                temp = y1
                y1   = y2
                y2   = temp


            utils.do_bounding_box_grab(x1, y1, x2, y2)
            markup = "<span foreground='" + self.success + "'>Grabbed region successfully...</span>"
            GLib.idle_add(self.message_label.set_markup, markup)
        except Exception as e:
            print(e)
            markup = "<span foreground='" + self.warning + "' >Oops...</span>" + \
                    "\n<span foreground='" + self.error + "'>" + html.escape( str(e) ) + "</span>"
            GLib.idle_add(self.message_label.set_markup, markup)