#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright (C) 2021 Guillaume Collic
#
#   This program is free software: you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published by
#   the Free Software Foundation, either version 3 of the License, or
#   (at your option) any later version.
#
#   This program 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 for more details.
#
#   You should have received a copy of the GNU General Public License
#   along with this program.  If not, see <http://www.gnu.org/licenses/>.

import re
import argparse
from boxes import Boxes, boolarg

class Keyboard:
    """
    Code to manage Cherry MX compatible switches and Kailh hotswap socket.

    Reference :
    * https://www.cherrymx.de/en/dev.html
    * https://cdn.sparkfun.com/datasheets/Components/Switches/MX%20Series.pdf
    * https://www.kailhswitch.com/uploads/201815927/PG151101S11.pdf
    """

    STANDARD_KEY_SPACING = 19.05
    SWITCH_CASE_SIZE = 15.6
    FRAME_CUTOUT = 14

    def __init__(self):
        pass

    def add_common_keyboard_parameters(
        self,
        add_hotswap_parameter=True,
        add_pcb_mount_parameter=True,
        add_led_parameter=True,
        add_diode_parameter=True,
        add_cutout_type_parameter=True,
        default_columns_definition=None,
    ):
        if add_hotswap_parameter:
            self.argparser.add_argument(
                "--hotswap_enable",
                action="store",
                type=boolarg,
                default=True,
                help=("enlarge switches holes for hotswap pcb sockets"),
            )
        if add_pcb_mount_parameter:
            self.argparser.add_argument(
                "--pcb_mount_enable",
                action="store",
                type=boolarg,
                default=True,
                help=("adds holes for pcb mount switches"),
            )
        if add_led_parameter:
            self.argparser.add_argument(
                "--led_enable",
                action="store",
                type=boolarg,
                default=False,
                help=("adds pin holes under switches for leds"),
            )
        if add_diode_parameter:
            self.argparser.add_argument(
                "--diode_enable",
                action="store",
                type=boolarg,
                default=False,
                help=("adds pin holes under switches for diodes"),
            )
        if add_cutout_type_parameter:
            self.argparser.add_argument(
                "--cutout_type",
                action="store",
                type=str,
                default="castle",
                help=(
                    "Shape of the plate cutout: 'castle' allows for modding, and 'simple' is a tighter and simpler square"
                ),
            )
        if default_columns_definition:
            self.argparser.add_argument(
            "--columns_definition",
                type=self.argparseColumnsDefinition,
                default=default_columns_definition,
                help=(
                    "Each column is separated by '/', and is in the form 'nb_rows @ offset x repeat_count'. "
                    "Nb_rows is the number of rows for this column. "
                    "The offset is in mm and optional. "
                    "Repeat_count is optional and repeats this column multiple times. "
                    "Spaces are not important."
                    "For example '3x2 / 4@11' means we want 3 columns, the two first with "
                    "3 rows without offset, and the last with 4 rows starting at 11mm high."
                ),
            )

    def argparseColumnsDefinition(self, s):
        """
        Parse columns definition parameter

        :param s: string to parse

        Each column is separated by '/', and is in the form 'nb_rows @ offset x repeat_count'.
        Nb_rows is the number of rows for this column.
        The offset is in mm and optional.
        Repeat_count is optional and repeats this column multiple times.
        Spaces are not important.
        For example '3x2 / 4@11' means we want 3 columns, the two first with
        3 rows without offset, and the last with 4 rows starting at 11mm high

        """
        result = []
        try:
            for column_string in s.split("/"):
                m = re.match(r"^\s*(\d+)\s*@?\s*(\d*\.?\d*)(?:\s*x\s*(\d+))?\s*$", column_string)
                keys_count = int(m.group(1))
                offset = float(m.group(2)) if m.group(2) else 0
                n = int(m.group(3)) if m.group(3) else 1
                result.extend([(offset, keys_count)]*n)
        except:
            raise argparse.ArgumentTypeError("Don't understand columns definition string")

        return result

    def pcb_holes(
        self, with_hotswap=True, with_pcb_mount=True, with_led=False, with_diode=False
    ):
        grid_unit = 1.27
        main_hole_size = 4
        pcb_mount_size = 1.7
        led_hole_size = 1
        if with_hotswap:
            pin_hole_size = 2.9
        else:
            pin_hole_size = 1.5

        def grid_hole(x, y, d):
            self.hole(grid_unit * x, grid_unit * y, d=d)

        # main hole
        grid_hole(0, 0, main_hole_size)

        # switch pins
        grid_hole(-3, 2, pin_hole_size)
        grid_hole(2, 4, pin_hole_size)

        if with_pcb_mount:
            grid_hole(-4, 0, pcb_mount_size)
            grid_hole(4, 0, pcb_mount_size)

        if with_led:
            grid_hole(-1, -4, led_hole_size)
            grid_hole(1, -4, led_hole_size)

        if with_diode:
            grid_hole(-3, -4, led_hole_size)
            grid_hole(3, -4, led_hole_size)

    def apply_callback_on_columns(self, cb, columns_definition, spacing=None, reverse=False):
        if spacing == None:
            spacing = self.STANDARD_KEY_SPACING
        if reverse:
            columns_definition = list(reversed(columns_definition))

        for offset, nb_keys in columns_definition:
            self.moveTo(0, offset)
            for _ in range(nb_keys):
                cb()
                self.moveTo(0, spacing)
            self.moveTo(spacing, -nb_keys * spacing)
            self.moveTo(0, -offset)

        total_width = len(columns_definition) * spacing
        self.moveTo(-1 * total_width)

    def outer_hole(self, radius=2, centered=True):
        """
        Draws a rounded square big enough to go around a whole switch (15.6mm)
        """
        half_size = Keyboard.SWITCH_CASE_SIZE / 2
        if centered:
            self.moveTo(-half_size, -half_size)

        # draw clock wise to work with burn correction
        straight_edge = Keyboard.SWITCH_CASE_SIZE - 2 * radius
        polyline = [straight_edge, (-90, radius)] * 4
        self.moveTo(self.burn, radius, 90)
        self.polyline(*polyline)
        self.moveTo(0, 0, 270)
        self.moveTo(0, -radius)
        self.moveTo(-self.burn)

        if centered:
            self.moveTo(half_size, half_size)

    def castle_shaped_plate_cutout(self, centered=True):
        """
        This cutout shaped like a castle enables switch modding and rotation.
        More information (type 4) on https://geekhack.org/index.php?topic=59837.0
        """
        half_size = Keyboard.SWITCH_CASE_SIZE / 2
        if centered:
            self.moveTo(-half_size, -half_size)

        # draw clock wise to work with burn correction
        btn_half_side = [0.98, 90, 0.81, -90, 3.5, -90, 0.81, 90, 2.505]
        btn_full_side = [*btn_half_side, 0, *btn_half_side[::-1]]
        btn = [*btn_full_side, -90] * 4

        self.moveTo(self.burn+0.81, 0.81, 90)
        self.polyline(*btn)
        self.moveTo(0, 0, 270)
        self.moveTo(-self.burn-0.81, -0.81)

        if centered:
            self.moveTo(half_size, half_size)

    def configured_plate_cutout(self, support=False):
        """
        Choose which cutout to use based on configured type.

        support: if true, not the main cutout, but one to glue against the first
        1.5mm cutout to strengthen it, without the clipping part.
        """
        if self.cutout_type.lower() == "castle":
            if support:
                self.outer_hole()
            else:
                self.castle_shaped_plate_cutout()
        else:
            self.simple_plate_cutout(with_notch=support)

    def simple_plate_cutout(self, radius=0.2, with_notch=False):
        """
        A simple plate cutout, a 14mm rectangle, as specified in this reference sheet
        https://cdn.sparkfun.com/datasheets/Components/Switches/MX%20Series.pdf

        With_notch shoul be used for a secondary lower plate, strengthening the first one.
        A notch is added to let the hooks grasp the main upper plate.

        Current position should be switch center.

        Radius should be lower or equal to 0.3 mm
        """
        size = Keyboard.FRAME_CUTOUT
        half_size = size / 2

        if with_notch:
            notch_length = 5
            notch_depth = 1
            straight_part = 0.5 * (size - 2 * radius - 2 * notch_depth - notch_length)

            self.moveTo(-half_size + self.burn, 0, 90)
            polyline_quarter = [
                half_size - radius,
                (-90, radius),
                straight_part,
                (90, notch_depth / 2),
                0,
                (-90, notch_depth / 2),
                notch_length / 2,
            ]
            polyline = (
                polyline_quarter
                + [0]
                + list(reversed(polyline_quarter))
                + [0]
                + polyline_quarter
                + [0]
                + list(reversed(polyline_quarter))
            )
            self.polyline(*polyline)
            self.moveTo(0, 0, -90)
            self.moveTo(half_size - self.burn)
        else:
            self.rectangularHole(0, 0, size, size, r=radius)