2 files changed, 12 insertions, 42 deletions

diff --git a/lua-bindings/diaphragm.lua b/lua-bindings/diaphragm.lua
index ffb3e4b..ad8aac6 100644
--- a/lua-bindings/diaphragm.lua
+++ b/lua-bindings/diaphragm.lua
@@ -399,6 +399,14 @@ function M.constraint.same_width(elems)
 end
 
 M.constrain = M.core.constrain
-M.draw = M.core.draw
+M.draw = function(params)
+        M.core.draw(M.util.tbl_extend(params, {
+                draw = function(figure)
+                        local self = M.shape({ left = 0, top = 0 })
+                        M.constraint.inset(self, figure)
+                        params.draw(self)
+                end,
+        }))
+end
 
 return M
diff --git a/lua-bindings/src/lib.rs b/lua-bindings/src/lib.rs
index 53622de..2cce0f5 100644
--- a/lua-bindings/src/lib.rs
+++ b/lua-bindings/src/lib.rs
@@ -741,25 +741,9 @@ fn draw(lua: &Lua, params: LuaTable) -> LuaResult<()> {
                 }
             }
 
-            let bounds = bounds.unwrap();
-
-            // TODO: it would be nicer to do in Lua
-            let bounds_top = bounds.top(runtime.solver_ctx());
-            let top_constraint = runtime
-                .solver_ctx()
-                .float_eq(bounds_top, CoreFloat::Fixed(0.0));
-            runtime.solver_ctx().constrain(top_constraint);
-
-            let bounds_left = bounds.left(runtime.solver_ctx());
-            let left_constraint = runtime
-                .solver_ctx()
-                .float_eq(bounds_left, CoreFloat::Fixed(0.0));
-            runtime.solver_ctx().constrain(left_constraint);
-
-            runtime.render(bounds);
+            runtime.render(bounds.unwrap());
         });
 
-        let content: LuaValue = params.get("content")?;
         let _output: LuaTable = params.get("output")?;
 
         let figure = complex_shape(lua, ())?;
@@ -776,30 +760,8 @@ fn draw(lua: &Lua, params: LuaTable) -> LuaResult<()> {
                 .unwrap()
         });
 
-        match content {
-            // TODO: this doesn't stop the runtime thread?
-            LuaValue::Table(table) => table.call_method("draw", figure).unwrap(),
-
-            // TODO: switch to enum
-            // LuaValue::UserData(user_data) => {
-            //     if user_data.is::<Text>() {
-            //         let text = user_data
-            //             .borrow::<Text>()
-            //             .expect("Couldn't borrow Text user data");
-            //
-            //         SENDER.with(|f| {
-            //             f.borrow_mut()
-            //                 .as_ref()
-            //                 .expect("Not currently drawing")
-            //                 .send(Message::AddDrawable(CoreShape::Text(text.0.clone())))
-            //                 .expect("Could not send shape");
-            //         })
-            //     } else {
-            //         panic!("Non-drawable passed to draw");
-            //     }
-            // }
-            _ => panic!("Non-drawable passed to draw"),
-        }
+        // TODO: this doesn't stop the runtime thread?
+        let () = params.call_function("draw", figure).unwrap();
 
         SENDER.with(|s| {
             *s.borrow_mut() = None;