From 7ef83dc508484f0818e0b7eb4cc3f50d134a1de0 Mon Sep 17 00:00:00 2001
From: Steve Ruiz <steveruizok@gmail.com>
Date: Thu, 27 May 2021 18:59:40 +0100
Subject: [PATCH] Adds drawing

---
 components/panel.tsx                    |  32 ++---
 components/style-panel/color-picker.tsx |   4 +-
 components/style-panel/style-panel.tsx  |   7 +-
 components/toolbar.tsx                  |   7 +
 hooks/useKeyboardEvents.ts              |   4 +
 lib/shape-utils/draw.tsx                | 162 ++++++++++++++++++++++++
 lib/shape-utils/index.tsx               |   9 ++
 state/commands/index.ts                 |   2 +
 state/commands/points.ts                |  28 ++++
 state/sessions/draw-session.ts          |  57 +++++++++
 state/sessions/index.ts                 |  16 ++-
 state/state.ts                          |  44 +++++++
 types.ts                                |   8 ++
 13 files changed, 353 insertions(+), 27 deletions(-)
 create mode 100644 lib/shape-utils/draw.tsx
 create mode 100644 state/commands/points.ts
 create mode 100644 state/sessions/draw-session.ts

diff --git a/components/panel.tsx b/components/panel.tsx
index 3c74c9989..7078d3ce5 100644
--- a/components/panel.tsx
+++ b/components/panel.tsx
@@ -31,7 +31,7 @@ export const Layout = styled("div", {
   gridTemplateRows: "auto 1fr",
   gridAutoRows: "28px",
   height: "100%",
-  width: "100%",
+  width: "auto",
   minWidth: "100%",
   maxWidth: 560,
   overflow: "hidden",
@@ -41,30 +41,32 @@ export const Layout = styled("div", {
 
 export const Header = styled("div", {
   pointerEvents: "all",
-  display: "grid",
-  gridTemplateColumns: "auto 1fr auto",
+  display: "flex",
+  width: "100%",
   alignItems: "center",
-  justifyContent: "center",
+  justifyContent: "space-between",
   borderBottom: "1px solid $border",
-
-  "& button": {
-    gridColumn: "1",
-    gridRow: "1",
-  },
+  position: "relative",
 
   "& h3": {
-    gridColumn: "1 / span 3",
-    gridRow: "1",
+    position: "absolute",
+    top: 0,
+    left: 0,
+    width: "100%",
+    height: "100%",
     textAlign: "center",
-    margin: "0",
-    padding: "0",
+    padding: 0,
+    margin: 0,
+    display: "flex",
+    justifyContent: "center",
+    alignItems: "center",
     fontSize: "13px",
+    pointerEvents: "none",
+    userSelect: "none",
   },
 })
 
 export const ButtonsGroup = styled("div", {
-  gridRow: "1",
-  gridColumn: "3",
   display: "flex",
 })
 
diff --git a/components/style-panel/color-picker.tsx b/components/style-panel/color-picker.tsx
index 680663125..7d74e0d44 100644
--- a/components/style-panel/color-picker.tsx
+++ b/components/style-panel/color-picker.tsx
@@ -98,8 +98,8 @@ const CurrentColor = styled(DropdownMenu.Trigger, {
     content: "''",
     position: "absolute",
     top: 0,
-    left: 4,
-    right: 4,
+    left: 0,
+    right: 0,
     bottom: 0,
     pointerEvents: "none",
     zIndex: -1,
diff --git a/components/style-panel/style-panel.tsx b/components/style-panel/style-panel.tsx
index f7ed3ccca..52e64455f 100644
--- a/components/style-panel/style-panel.tsx
+++ b/components/style-panel/style-panel.tsx
@@ -73,9 +73,6 @@ function SelectedShapeStyles({}: {}) {
   return (
     <Panel.Layout>
       <Panel.Header>
-        <IconButton onClick={() => state.send("TOGGLED_STYLE_PANEL_OPEN")}>
-          <X />
-        </IconButton>
         <h3>Style</h3>
         <Panel.ButtonsGroup>
           <IconButton
@@ -85,6 +82,9 @@ function SelectedShapeStyles({}: {}) {
             <Trash />
           </IconButton>
         </Panel.ButtonsGroup>
+        <IconButton onClick={() => state.send("TOGGLED_STYLE_PANEL_OPEN")}>
+          <X />
+        </IconButton>
       </Panel.Header>
       <Content>
         <ColorPicker
@@ -112,6 +112,7 @@ const StylePanelRoot = styled(Panel.Root, {
   minWidth: 1,
   width: 184,
   maxWidth: 184,
+  overflow: "hidden",
   position: "relative",
 
   variants: {
diff --git a/components/toolbar.tsx b/components/toolbar.tsx
index e0e7c5ee6..3c37501b3 100644
--- a/components/toolbar.tsx
+++ b/components/toolbar.tsx
@@ -13,6 +13,7 @@ export default function Toolbar() {
       line: "line",
       polyline: "polyline",
       rectangle: "rectangle",
+      draw: "draw",
     })
   )
 
@@ -28,6 +29,12 @@ export default function Toolbar() {
         >
           Select
         </Button>
+        <Button
+          isSelected={activeTool === "draw"}
+          onClick={() => state.send("SELECTED_DRAW_TOOL")}
+        >
+          Draw
+        </Button>
         <Button
           isSelected={activeTool === "dot"}
           onClick={() => state.send("SELECTED_DOT_TOOL")}
diff --git a/hooks/useKeyboardEvents.ts b/hooks/useKeyboardEvents.ts
index 2208a06d4..280198de3 100644
--- a/hooks/useKeyboardEvents.ts
+++ b/hooks/useKeyboardEvents.ts
@@ -115,6 +115,10 @@ export default function useKeyboardEvents() {
           break
         }
         case "d": {
+          state.send("SELECTED_DRAW_TOOL", getKeyboardEventInfo(e))
+          break
+        }
+        case "t": {
           if (metaKey(e)) {
             state.send("DUPLICATED", getKeyboardEventInfo(e))
           } else {
diff --git a/lib/shape-utils/draw.tsx b/lib/shape-utils/draw.tsx
new file mode 100644
index 000000000..496f03afb
--- /dev/null
+++ b/lib/shape-utils/draw.tsx
@@ -0,0 +1,162 @@
+import { v4 as uuid } from "uuid"
+import * as vec from "utils/vec"
+import { DrawShape, ShapeType } from "types"
+import { registerShapeUtils } from "./index"
+import { intersectPolylineBounds } from "utils/intersections"
+import { boundsContainPolygon } from "utils/bounds"
+import { getBoundsFromPoints, translateBounds } from "utils/utils"
+
+const draw = registerShapeUtils<DrawShape>({
+  boundsCache: new WeakMap([]),
+
+  create(props) {
+    return {
+      id: uuid(),
+      type: ShapeType.Draw,
+      isGenerated: false,
+      name: "Draw",
+      parentId: "page0",
+      childIndex: 0,
+      point: [0, 0],
+      points: [[0, 0]],
+      rotation: 0,
+      ...props,
+      style: {
+        strokeWidth: 2,
+        strokeLinecap: "round",
+        strokeLinejoin: "round",
+        ...props.style,
+        fill: "transparent",
+      },
+    }
+  },
+
+  render({ id, points }) {
+    return <polyline id={id} points={points.toString()} />
+  },
+
+  applyStyles(shape, style) {
+    Object.assign(shape.style, style)
+    shape.style.fill = "transparent"
+    return this
+  },
+
+  getBounds(shape) {
+    if (!this.boundsCache.has(shape)) {
+      const bounds = getBoundsFromPoints(shape.points)
+      this.boundsCache.set(shape, bounds)
+    }
+
+    return translateBounds(this.boundsCache.get(shape), shape.point)
+  },
+
+  getRotatedBounds(shape) {
+    return this.getBounds(shape)
+  },
+
+  getCenter(shape) {
+    const bounds = this.getBounds(shape)
+    return [bounds.minX + bounds.width / 2, bounds.minY + bounds.height / 2]
+  },
+
+  hitTest(shape, point) {
+    let pt = vec.sub(point, shape.point)
+    let prev = shape.points[0]
+
+    for (let i = 1; i < shape.points.length; i++) {
+      let curr = shape.points[i]
+      if (vec.distanceToLineSegment(prev, curr, pt) < 4) {
+        return true
+      }
+      prev = curr
+    }
+
+    return false
+  },
+
+  hitTestBounds(this, shape, brushBounds) {
+    const b = this.getBounds(shape)
+    const center = [b.minX + b.width / 2, b.minY + b.height / 2]
+
+    const rotatedCorners = [
+      [b.minX, b.minY],
+      [b.maxX, b.minY],
+      [b.maxX, b.maxY],
+      [b.minX, b.maxY],
+    ].map((point) => vec.rotWith(point, center, shape.rotation))
+
+    return (
+      boundsContainPolygon(brushBounds, rotatedCorners) ||
+      intersectPolylineBounds(
+        shape.points.map((point) => vec.add(point, shape.point)),
+        brushBounds
+      ).length > 0
+    )
+  },
+
+  rotateTo(shape, rotation) {
+    shape.rotation = rotation
+    return this
+  },
+
+  translateTo(shape, point) {
+    shape.point = point
+    return this
+  },
+
+  transform(shape, bounds, { initialShape, scaleX, scaleY }) {
+    const initialShapeBounds = this.boundsCache.get(initialShape)
+    shape.points = initialShape.points.map(([x, y]) => {
+      return [
+        bounds.width *
+          (scaleX < 0
+            ? 1 - x / initialShapeBounds.width
+            : x / initialShapeBounds.width),
+        bounds.height *
+          (scaleY < 0
+            ? 1 - y / initialShapeBounds.height
+            : y / initialShapeBounds.height),
+      ]
+    })
+
+    const newBounds = getBoundsFromPoints(shape.points)
+
+    shape.point = vec.sub(
+      [bounds.minX, bounds.minY],
+      [newBounds.minX, newBounds.minY]
+    )
+    return this
+  },
+
+  transformSingle(shape, bounds, info) {
+    this.transform(shape, bounds, info)
+    return this
+  },
+
+  setParent(shape, parentId) {
+    shape.parentId = parentId
+    return this
+  },
+
+  setChildIndex(shape, childIndex) {
+    shape.childIndex = childIndex
+    return this
+  },
+
+  setPoints(shape, points) {
+    // const bounds = getBoundsFromPoints(points)
+    // const corner = [bounds.minX, bounds.minY]
+    // const nudged = points.map((point) => vec.sub(point, corner))
+    // this.boundsCache.set(shape, translategetBoundsFromPoints(nudged))
+    // shape.point = vec.add(shape.point, corner)
+
+    shape.points = points
+
+    return this
+  },
+
+  canTransform: true,
+  canChangeAspectRatio: true,
+})
+
+export default draw
diff --git a/lib/shape-utils/index.tsx b/lib/shape-utils/index.tsx
index 054b407d7..5a8b93d78 100644
--- a/lib/shape-utils/index.tsx
+++ b/lib/shape-utils/index.tsx
@@ -16,6 +16,7 @@ import rectangle from "./rectangle"
 import ellipse from "./ellipse"
 import line from "./line"
 import ray from "./ray"
+import draw from "./draw"
 
 /*
 Shape Utiliies
@@ -91,6 +92,13 @@ export interface ShapeUtility<K extends Readonly<Shape>> {
     childIndex: number
   ): ShapeUtility<K>
 
+  // Add a point
+  setPoints?(
+    this: ShapeUtility<K>,
+    shape: K,
+    points: number[][]
+  ): ShapeUtility<K>
+
   // Render a shape to JSX.
   render(this: ShapeUtility<K>, shape: K): JSX.Element
 
@@ -119,6 +127,7 @@ const shapeUtilityMap: Record<ShapeType, ShapeUtility<Shape>> = {
   [ShapeType.Ellipse]: ellipse,
   [ShapeType.Line]: line,
   [ShapeType.Ray]: ray,
+  [ShapeType.Draw]: draw,
 }
 
 /**
diff --git a/state/commands/index.ts b/state/commands/index.ts
index a3ed8444c..76244946d 100644
--- a/state/commands/index.ts
+++ b/state/commands/index.ts
@@ -4,6 +4,7 @@ import direct from "./direct"
 import distribute from "./distribute"
 import generate from "./generate"
 import move from "./move"
+import points from "./points"
 import rotate from "./rotate"
 import stretch from "./stretch"
 import style from "./style"
@@ -18,6 +19,7 @@ const commands = {
   distribute,
   generate,
   move,
+  points,
   rotate,
   stretch,
   style,
diff --git a/state/commands/points.ts b/state/commands/points.ts
new file mode 100644
index 000000000..75aeff6f5
--- /dev/null
+++ b/state/commands/points.ts
@@ -0,0 +1,28 @@
+import Command from "./command"
+import history from "../history"
+import { Data } from "types"
+import { getPage } from "utils/utils"
+import { getShapeUtils } from "lib/shape-utils"
+
+export default function pointsCommand(
+  data: Data,
+  id: string,
+  before: number[][],
+  after: number[][]
+) {
+  history.execute(
+    data,
+    new Command({
+      name: "set_points",
+      category: "canvas",
+      do(data) {
+        const shape = getPage(data).shapes[id]
+        getShapeUtils(shape).setPoints!(shape, after)
+      },
+      undo(data) {
+        const shape = getPage(data).shapes[id]
+        getShapeUtils(shape).setPoints!(shape, before)
+      },
+    })
+  )
+}
diff --git a/state/sessions/draw-session.ts b/state/sessions/draw-session.ts
new file mode 100644
index 000000000..23eab8510
--- /dev/null
+++ b/state/sessions/draw-session.ts
@@ -0,0 +1,57 @@
+import { current } from "immer"
+import { Data, DrawShape } from "types"
+import BaseSession from "./base-session"
+import { getShapeUtils } from "lib/shape-utils"
+import { getPage } from "utils/utils"
+import * as vec from "utils/vec"
+import commands from "state/commands"
+
+export default class BrushSession extends BaseSession {
+  origin: number[]
+  points: number[][]
+  snapshot: DrawSnapshot
+  shapeId: string
+
+  constructor(data: Data, id: string, point: number[]) {
+    super(data)
+    this.shapeId = id
+    this.origin = point
+    this.points = [[0, 0]]
+    this.snapshot = getDrawSnapshot(data, id)
+
+    const page = getPage(data)
+    const shape = page.shapes[id]
+    getShapeUtils(shape).translateTo(shape, point)
+  }
+
+  update = (data: Data, point: number[]) => {
+    const { shapeId } = this
+
+    this.points.push(vec.sub(point, this.origin))
+
+    const page = getPage(data)
+    const shape = page.shapes[shapeId]
+    getShapeUtils(shape).setPoints!(shape, [...this.points])
+  }
+
+  cancel = (data: Data) => {
+    const { shapeId, snapshot } = this
+    const page = getPage(data)
+    const shape = page.shapes[shapeId]
+    getShapeUtils(shape).setPoints!(shape, snapshot.points)
+  }
+
+  complete = (data: Data) => {
+    commands.points(data, this.shapeId, this.snapshot.points, this.points)
+  }
+}
+
+export function getDrawSnapshot(data: Data, shapeId: string) {
+  const page = getPage(current(data))
+  const { points } = page.shapes[shapeId] as DrawShape
+  return {
+    points,
+  }
+}
+
+export type DrawSnapshot = ReturnType<typeof getDrawSnapshot>
diff --git a/state/sessions/index.ts b/state/sessions/index.ts
index ad62e56d0..da65aa21c 100644
--- a/state/sessions/index.ts
+++ b/state/sessions/index.ts
@@ -1,17 +1,19 @@
 import BaseSession from "./base-session"
 import BrushSession from "./brush-session"
-import TranslateSession from "./translate-session"
+import DirectionSession from "./direction-session"
+import DrawSession from "./draw-session"
+import RotateSession from "./rotate-session"
 import TransformSession from "./transform-session"
 import TransformSingleSession from "./transform-single-session"
-import DirectionSession from "./direction-session"
-import RotateSession from "./rotate-session"
+import TranslateSession from "./translate-session"
 
 export {
-  BrushSession,
   BaseSession,
-  TranslateSession,
+  BrushSession,
+  DirectionSession,
+  DrawSession,
+  RotateSession,
   TransformSession,
   TransformSingleSession,
-  DirectionSession,
-  RotateSession,
+  TranslateSession,
 }
diff --git a/state/state.ts b/state/state.ts
index ade91d85a..f98b24b06 100644
--- a/state/state.ts
+++ b/state/state.ts
@@ -31,6 +31,7 @@ import {
   DistributeType,
   AlignType,
   StretchType,
+  DrawShape,
 } from "types"
 
 const initialData: Data = {
@@ -70,6 +71,7 @@ const state = createState({
       do: "panCamera",
     },
     SELECTED_SELECT_TOOL: { to: "selecting" },
+    SELECTED_DRAW_TOOL: { unless: "isReadOnly", to: "draw" },
     SELECTED_DOT_TOOL: { unless: "isReadOnly", to: "dot" },
     SELECTED_CIRCLE_TOOL: { unless: "isReadOnly", to: "circle" },
     SELECTED_ELLIPSE_TOOL: { unless: "isReadOnly", to: "ellipse" },
@@ -246,6 +248,32 @@ const state = createState({
             },
           },
         },
+        draw: {
+          initial: "creating",
+          states: {
+            creating: {
+              on: {
+                POINTED_CANVAS: {
+                  get: "newDraw",
+                  do: "createShape",
+                  to: "draw.editing",
+                },
+              },
+            },
+            editing: {
+              onEnter: "startDrawSession",
+              on: {
+                STOPPED_POINTING: { do: "completeSession", to: "selecting" },
+                CANCELLED: {
+                  do: ["cancelSession", "deleteSelectedIds"],
+                  to: "selecting",
+                },
+                MOVED_POINTER: "updateDrawSession",
+                PANNED_CAMERA: "updateDrawSession",
+              },
+            },
+          },
+        },
         dot: {
           initial: "creating",
           states: {
@@ -451,6 +479,9 @@ const state = createState({
     },
   },
   results: {
+    newDraw() {
+      return ShapeType.Draw
+    },
     newDot() {
       return ShapeType.Dot
     },
@@ -646,6 +677,19 @@ const state = createState({
       session.update(data, screenToWorld(payload.point, data))
     },
 
+    // Drawing
+    startDrawSession(data) {
+      const id = Array.from(data.selectedIds.values())[0]
+      session = new Sessions.DrawSession(
+        data,
+        id,
+        screenToWorld(inputs.pointer.origin, data)
+      )
+    },
+    updateDrawSession(data, payload: PointerInfo) {
+      session.update(data, screenToWorld(payload.point, data))
+    },
+
     /* -------------------- Selection ------------------- */
 
     selectAll(data) {
diff --git a/types.ts b/types.ts
index bffd1daac..5b7ab9973 100644
--- a/types.ts
+++ b/types.ts
@@ -53,6 +53,7 @@ export enum ShapeType {
   Ray = "ray",
   Polyline = "polyline",
   Rectangle = "rectangle",
+  Draw = "draw",
 }
 
 // Consider:
@@ -111,6 +112,11 @@ export interface RectangleShape extends BaseShape {
   radius: number
 }
 
+export interface DrawShape extends BaseShape {
+  type: ShapeType.Draw
+  points: number[][]
+}
+
 export type MutableShape =
   | DotShape
   | CircleShape
@@ -118,6 +124,7 @@ export type MutableShape =
   | LineShape
   | RayShape
   | PolylineShape
+  | DrawShape
   | RectangleShape
 
 export type Shape = Readonly<MutableShape>
@@ -129,6 +136,7 @@ export interface Shapes {
   [ShapeType.Line]: Readonly<LineShape>
   [ShapeType.Ray]: Readonly<RayShape>
   [ShapeType.Polyline]: Readonly<PolylineShape>
+  [ShapeType.Draw]: Readonly<DrawShape>
   [ShapeType.Rectangle]: Readonly<RectangleShape>
 }