penpot/src/uxbox/data/workspace.cljs
2016-02-03 17:07:47 +02:00

456 lines
13 KiB
Clojure

(ns uxbox.data.workspace
(:require [bouncer.validators :as v]
[beicon.core :as rx]
[uxbox.shapes :as sh]
[uxbox.rstore :as rs]
[uxbox.router :as r]
[uxbox.state :as st]
[uxbox.state.shapes :as stsh]
[uxbox.schema :as sc]
[uxbox.time :as time]
[uxbox.xforms :as xf]
[uxbox.shapes :as sh]
[uxbox.util.data :refer (index-of)]))
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Schemas
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
(def ^:static +shape-schema+
{:x [sc/integer]
:y [sc/integer]
:width [sc/integer]
:height [sc/integer]
:type [sc/required sc/shape-type]})
(def ^:static +shape-size-schema+
{:width [sc/integer]
:height [sc/integer]
:lock [sc/boolean]})
(def ^:static +shape-fill-attrs-schema+
{:color [sc/color]
:opacity [sc/number]})
(def ^:static +shape-stroke-attrs-schema+
{:color [sc/color]
:width [sc/integer]
:type [sc/keyword]
:opacity [sc/number]})
(def ^:static +shape-line-attrs-schema+
{:x1 [sc/integer]
:y1 [sc/integer]
:x2 [sc/integer]
:y2 [sc/integer]})
(def ^:static +shape-radius-attrs-schema+
{:rx [sc/integer]
:ry [sc/integer]})
(def ^:static +shape-position-schema+
{:x [sc/integer]
:y [sc/integer]})
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Events (explicit)
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
(defn initialize
"Initialize the workspace state."
[projectid pageid]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(let [s {:project projectid
:toolboxes #{:layers}
:flags #{}
:drawing nil
:selected #{}
:page pageid}]
(assoc state :workspace s)))))
(defn toggle-tool
"Toggle the enabled flag of the specified tool."
[key]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(let [flags (get-in state [:workspace :flags])]
(if (contains? flags key)
(assoc-in state [:workspace :flags] (disj flags key))
(assoc-in state [:workspace :flags] (conj flags key)))))))
(defn toggle-toolbox
"Toggle the visibility flag of the specified toolbox."
[toolname]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(let [toolboxes (get-in state [:workspace :toolboxes])]
(assoc-in state [:workspace :toolboxes]
(if (contains? toolboxes toolname)
(disj toolboxes toolname)
(conj toolboxes toolname)))))))
(defn select-for-drawing
"Mark a shape selected for drawing in the canvas."
[shape]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(if shape
(assoc-in state [:workspace :drawing] shape)
(update-in state [:workspace] dissoc :drawing)))))
(defn select-shape
"Mark a shape selected for drawing in the canvas."
[id]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(let [selected (get-in state [:workspace :selected])]
(if (contains? selected id)
(update-in state [:workspace :selected] disj id)
(update-in state [:workspace :selected] conj id))))))
(defn select-shapes
"Select shapes that matches the select rect."
[selrect]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(let [pageid (get-in state [:workspace :page])
xf (comp
(filter #(= (:page %) pageid))
(remove :hidden)
(remove :blocked)
(map sh/-outer-rect)
(filter #(sh/contained-in? % selrect))
(map :id))]
(->> (into #{} xf (vals (:shapes-by-id state)))
(assoc-in state [:workspace :selected]))))))
(defn add-shape
"Create and add shape to the current selected page."
[shape]
(sc/validate! +shape-schema+ shape)
(reify
rs/UpdateEvent
(-apply-update [_ state]
(let [sid (random-uuid)
pid (get-in state [:workspace :page])
shape (merge shape {:id sid :page pid})]
(as-> state $
(update-in $ [:pages-by-id pid :shapes] conj sid)
(assoc-in $ [:shapes-by-id sid] shape))))))
(defn delete-shape
"Remove the shape using its id."
[id]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(let [shape (get-in state [:shapes-by-id id])]
(stsh/dissoc-shape state shape)))))
(defn move-shape
"Mark a shape selected for drawing in the canvas."
[sid [dx dy :as delta]]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(let [shape (get-in state [:shapes-by-id sid])]
(update-in state [:shapes-by-id sid] sh/-move delta)))))
(defn update-line-attrs
[sid {:keys [x1 y1 x2 y2] :as opts}]
(sc/validate! +shape-line-attrs-schema+ opts)
(reify
rs/UpdateEvent
(-apply-update [_ state]
(let [shape (get-in state [:shapes-by-id sid])
props (select-keys opts [:x1 :y1 :x2 :y2])
props' (select-keys shape [:x1 :y1 :x2 :y2])]
(update-in state [:shapes-by-id sid] sh/-initialize
(merge props' props))))))
(defn update-rotation
[sid rotation]
{:pre [(number? rotation)
(>= rotation 0)
(>= 360 rotation)]}
(reify
rs/UpdateEvent
(-apply-update [_ state]
(update-in state [:shapes-by-id sid]
sh/-rotate rotation))))
(defn update-size
"A helper event just for update the position
of the shape using the width and heigt attrs
instread final point of coordinates.
WARN: only works with shapes that works
with height and width such are"
[sid {:keys [width height] :as opts}]
(sc/validate! +shape-size-schema+ opts)
(reify
rs/UpdateEvent
(-apply-update [_ state]
(let [size {:width width :height height}]
(update-in state [:shapes-by-id sid] sh/-resize' size)))))
(defn update-position
"Update the start position coordenate of the shape."
[sid {:keys [x y] :as opts}]
(sc/validate! +shape-position-schema+ opts)
(reify
rs/UpdateEvent
(-apply-update [_ state]
(update-in state [:shapes-by-id sid] sh/-move' [x y]))))
(defn update-fill-attrs
[sid {:keys [color opacity] :as opts}]
(sc/validate! +shape-fill-attrs-schema+ opts)
(reify
rs/UpdateEvent
(-apply-update [_ state]
(update-in state [:shapes-by-id sid]
merge
(when color {:fill color})
(when opacity {:opacity opacity})))))
(defn update-stroke-attrs
[sid {:keys [color opacity type width] :as opts}]
(sc/validate! +shape-stroke-attrs-schema+ opts)
(reify
rs/UpdateEvent
(-apply-update [_ state]
(update-in state [:shapes-by-id sid]
merge
{:stroke-type type}
(when width {:stroke-width width})
(when color {:stroke color})
(when opacity {:stroke-opacity opacity})))))
(defn update-radius-attrs
[sid {:keys [rx ry] :as opts}]
(sc/validate! +shape-radius-attrs-schema+ opts)
(reify
rs/UpdateEvent
(-apply-update [_ state]
(update-in state [:shapes-by-id sid]
merge
(when rx {:rx rx})
(when ry {:ry ry})))))
(defn hide-shape
[sid]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(assoc-in state [:shapes-by-id sid :hidden] true))
rs/WatchEvent
(-apply-watch [_ state]
(let [shape (get-in state [:shapes-by-id sid])]
(if-not (= (:type shape) :builtin/group)
(rx/empty)
(rx/from-coll
(map hide-shape (:items shape))))))))
(defn show-shape
[sid]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(assoc-in state [:shapes-by-id sid :hidden] false))
rs/WatchEvent
(-apply-watch [_ state]
(let [shape (get-in state [:shapes-by-id sid])]
(if-not (= (:type shape) :builtin/group)
(rx/empty)
(rx/from-coll
(map show-shape (:items shape))))))))
(defn block-shape
[sid]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(assoc-in state [:shapes-by-id sid :blocked] true))
rs/WatchEvent
(-apply-watch [_ state]
(let [shape (get-in state [:shapes-by-id sid])]
(if-not (= (:type shape) :builtin/group)
(rx/empty)
(rx/from-coll
(map block-shape (:items shape))))))))
(defn unblock-shape
[sid]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(assoc-in state [:shapes-by-id sid :blocked] false))
rs/WatchEvent
(-apply-watch [_ state]
(let [shape (get-in state [:shapes-by-id sid])]
(if-not (= (:type shape) :builtin/group)
(rx/empty)
(rx/from-coll
(map unblock-shape (:items shape))))))))
(defn lock-shape
[sid]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(assoc-in state [:shapes-by-id sid :locked] true))
rs/WatchEvent
(-apply-watch [_ state]
(let [shape (get-in state [:shapes-by-id sid])]
(if-not (= (:type shape) :builtin/group)
(rx/empty)
(rx/from-coll
(map lock-shape (:items shape))))))))
(defn unlock-shape
[sid]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(assoc-in state [:shapes-by-id sid :locked] false))
rs/WatchEvent
(-apply-watch [_ state]
(let [shape (get-in state [:shapes-by-id sid])]
(if-not (= (:type shape) :builtin/group)
(rx/empty)
(rx/from-coll
(map unlock-shape (:items shape))))))))
(defn drop-shape
"Event used in drag and drop for transfer shape
from one position to an other."
[sid tid loc]
{:pre [(not (nil? tid))
(not (nil? sid))]}
(reify
rs/UpdateEvent
(-apply-update [_ state]
(stsh/drop-shape state tid sid loc))))
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Events (for selected)
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
(defn deselect-all
"Mark a shape selected for drawing in the canvas."
[]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(assoc-in state [:workspace :selected] #{}))))
;; FIXME
;; (defn copy-selected
;; "Copy the selected shapes."
;; []
;; (letfn [(valid-selection? [shapes]
;; (let [groups (into #{} (map :group shapes))]
;; (= 1 (count groups))))]
;; (reify
;; rs/WatchEvent
;; (-apply-watch [_ state]
;; (let [selected (get-in state [:workspace :selected])
;; selected (map #(get-in state [:shapes-by-id %]) selected)]
;; (if (valid-selection? selected)
;; (as-> selected $
;; (map #(assoc % :id (random-uuid)) $)
;; (map #(add-shape % %) $)
;; (rx/from-coll $))))))
(defn group-selected
[]
(letfn [(update-shapes-on-page [state pid selected group]
(as-> (get-in state [:pages-by-id pid :shapes]) $
(remove selected $)
(into [group] $)
(assoc-in state [:pages-by-id pid :shapes] $)))
(update-shapes-on-index [state shapes group]
(reduce (fn [state {:keys [id] :as shape}]
(as-> shape $
(assoc $ :group group)
(assoc-in state [:shapes-by-id id] $)))
state
shapes))
(valid-selection? [shapes]
(let [groups (into #{} (map :group shapes))]
(= 1 (count groups))))]
(reify
rs/UpdateEvent
(-apply-update [_ state]
(let [shapes-by-id (get state :shapes-by-id)
sid (random-uuid)
pid (get-in state [:workspace :page])
selected (get-in state [:workspace :selected])
selected' (map #(get shapes-by-id %) selected)
group {:type :builtin/group
:name (str "Group " (rand-int 1000))
:items (into [] selected)
:id sid
:page pid}]
(if (valid-selection? selected')
(as-> state $
(update-shapes-on-index $ selected' sid)
(update-shapes-on-page $ pid selected sid)
(update $ :shapes-by-id assoc sid group)
(update $ :workspace assoc :selected #{}))
state))))))
(defn delete-selected
"Deselect all and remove all selected shapes."
[]
(reify
rs/WatchEvent
(-apply-watch [_ state]
(let [selected (get-in state [:workspace :selected])]
(rx/from-coll
(into [(deselect-all)] (map #(delete-shape %) selected)))))))
(defn move-selected
"Move a minimal position unit the selected shapes."
[dir]
{:pre [(contains? #{:up :down :right :left} dir)]}
(reify
rs/WatchEvent
(-apply-watch [_ state]
(let [selected (get-in state [:workspace :selected])
delta (case dir
:up [0 -1]
:down [0 +1]
:right [+1 0]
:left [-1 0])]
(rx/from-coll
(map #(move-shape % delta) selected))))))
(defn update-selected-shapes-fill
"Update the fill related attributed on
selected shapes."
[opts]
(sc/validate! +shape-fill-attrs-schema+ opts)
(reify
rs/WatchEvent
(-apply-watch [_ state]
(rx/from-coll
(->> (get-in state [:workspace :selected])
(map #(update-fill-attrs % opts)))))))