##############################################################################
# Python From Scratch
# Autor: Nilo Ney Coutinho Menezes
# Editora Novatec (c) 2010-2024
# Site: https://pythonfromscratch.com
#
# File: listing\chapter 13\13.07 - Draw.py
# Description: Draw
##############################################################################

import tkinter as tk
import tkinter.ttk as ttk
from tkinter.colorchooser import askcolor

class App(tk.Tk):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.background_color = ""
        self.foreground_color = "black"
        self.frame = tk.Frame(self)
        self.create_toolbar()
        self.create_drawing_area()
        self.title("Drawing")
        self.geometry("800x600")
        self.cross = []
        self.cross.append(self.canvas.create_line((0, 0, 0, 0), dash=[2, 4]))
        self.cross.append(self.canvas.create_line((0, 0, 0, 0), dash=[2, 4]))
        self.state = 0
        self.xi = None
        self.yi = None
        self.curr_id = 0
        self.frame.pack(expand=True, fill=tk.BOTH)
        self.tool = self.canvas.create_line

    def create_drawing_area(self):
        self.work = tk.Frame(self.frame, height=600)
        self.work.grid(column=1, row=0, sticky=tk.NSEW)
        self.frame.grid_columnconfigure(1, weight=1)
        self.frame.grid_rowconfigure(0, weight=1)
        self.canvas = tk.Canvas(self.work, background="white")
        self.canvas.pack(fill=tk.BOTH, expand=True)
        self.canvas.bind("<Motion>", self.mouse_move)
        self.canvas.bind("<Button-1>", self.mouse_click)
        self.canvas.bind("<ButtonRelease-1>", self.mouse_release)
        self.coordinates = tk.Label(self.work, text="Move the mouse")
        self.coordinates.pack(ipadx=10, ipady=10)

    def create_toolbar(self):
        self.toolbar = tk.Frame(self.frame, width=100, height=600)
        self.bline = ttk.Button(
            self.toolbar, text="Line", padding="10", command=self.line_tool)
        self.bline.pack()
        self.boval = ttk.Button(
            self.toolbar, text="Circle", padding="10", command=self.oval_tool)
        self.boval.pack()
        self.brectangle = ttk.Button(
            self.toolbar,
            text="Rectangle",
            padding="10",
            command=self.rectangle_tool)
        self.brectangle.pack()
        bundo = ttk.Button(
            self.toolbar, text="Undo", padding="10", command=self.undo)
        bundo.pack()
        bclear = ttk.Button(self.toolbar, text="Clear", padding="10", command=self.clear)
        bclear.pack()
        self.lforeground = ttk.Label(self.toolbar, text="Foreground Color")
        self.lforeground.pack()
        self.bforeground = tk.Button(
            self.toolbar, text="color", command=self.set_foreground_color, bg=self.foreground_color)
        self.bforeground.pack(fill="x")
        self.lbackground = ttk.Label(self.toolbar, text="Background color")
        self.lbackground.pack()
        self.bbackground = tk.Button(
            self.toolbar, text="transparent", command=self.set_background_color, bg=None)
        self.bbackground.pack(fill="x")
        self.toolbar.grid(column=0, row=0, sticky=tk.NS)

    def undo(self):
        if items := self.canvas.find_withtag("drawing"):
            self.canvas.delete(items[-1])

    def clear(self):
        self.canvas.delete("drawing")

    def set_background_color(self):
        color = askcolor(title="Background color")
        self.background_color = color[1] or ""
        self.bbackground.config(
            text="Transparent" if self.background_color == "" else "",
            background=self.background_color or "SystemButtonFace")

    def set_foreground_color(self):
        color = askcolor(title="Foreground color")
        if color[1]:
            self.foreground_color = color[1]
            self.bforeground.config(background=self.foreground_color)

    def line_tool(self):
        self.tool = self.canvas.create_line

    def oval_tool(self):
        self.tool = self.canvas.create_oval

    def rectangle_tool(self):
        self.tool = self.canvas.create_rectangle

    def mouse_move(self, event):
        self.coordinates["text"] = f"Mouse x={event.x} y={event.y}"
        self.canvas.coords(
            self.cross[0], event.x, 0, event.x, self.canvas.winfo_height()
        )
        self.canvas.coords(self.cross[1], 0, event.y, self.canvas.winfo_width(), event.y)
        if self.state == 1:
            self.canvas.coords(self.curr_id, self.xi, self.yi, event.x, event.y)

    def mouse_click(self, event):
        if self.state == 0:
            self.xi = event.x
            self.yi = event.y
            self.curr_id = self.tool(
                (self.xi, self.yi, event.x, event.y),
                fill=self.foreground_color,
                tags=["drawing"])
            type = self.canvas.type(self.curr_id)
            if type in ["rectangle", "oval"]:
                self.canvas.itemconfig(
                    self.curr_id,
                    {
                        "outline": self.foreground_color,
                        "fill": self.background_color,
                    })
            self.state = 1

    def mouse_release(self, event):
        if self.state == 1:
            self.state = 0

App().mainloop()