import cadquery2 as cq
from cadquery2 import exporters

from parameters import *

# Small M3x8 screw
screw_l = 10
screw_shaft = 8
screw_head_h = 2.5
screw_head_radius = 3.5
screw_radius = 1.5

from hinge import hinge_profile_1, hinge_profile_2, hinge_t, hinge_gap, hinge_s

reinforcement_height = 1 + 1 + ti_depth - shell_t

rear_mount_thickness = reinforcement_height + screw_head_h + shell_t
rear_mount_width = width / 2 + 2 * shell_t

back_reinforcement_cutout = (
    cq.Sketch()
    .trapezoid(width / 2 + 1, 15 + 1, 90, mode="a")
    .reset()
    .vertices()
    .fillet(2)
)

back_reinforcement_outer = (
    cq.Sketch()
    .trapezoid(rear_mount_width, 15 + 2 * shell_t, 90, mode="a")
    .reset()
    .vertices()
    .fillet(2)
)

# FIXME: use counterbore holes function instead

screw_holes = cq.Sketch().rarray(width / 8, 0, 4, 1).circle(screw_radius, mode="a")
screw_countersinks = (
    cq.Sketch().rarray(width / 8, 0, 4, 1).circle(screw_head_radius, mode="a")
)


rear_mount = (
    # Basic filleted box shape
    cq.Workplane("bottom")
    .placeSketch(back_reinforcement_outer)
    .extrude(rear_mount_thickness)
    .faces(">Y")
    .workplane(centerOption="CenterOfBoundBox")
    .placeSketch(back_reinforcement_cutout)
    .cutBlind(-reinforcement_height)
    .faces("<Y")
    .workplane(centerOption="CenterOfBoundBox")
    .placeSketch(screw_holes)
    .cutBlind(-1000)
    .faces("<Y")
    .workplane(centerOption="CenterOfBoundBox")
    .placeSketch(screw_countersinks)
    .cutBlind(-screw_head_h)
    # Hinge integration
    .faces(">X")
    .workplane(centerOption="CenterOfBoundBox", offset=-(rear_mount_width-20)/2)
    .transformed(rotate=cq.Vector(0, 0, 90))
    .center(-hinge_s / 2, rear_mount_thickness / 2).tag("hingeplane")
    .placeSketch(hinge_profile_1)
    .extrude(-hinge_t)
    .workplaneFromTagged("hingeplane")
    .workplane(offset=-hinge_t - hinge_gap)
    .placeSketch(hinge_profile_2)
    .extrude(-hinge_t))

exporters.export(rear_mount, "rear_mount.stl")