undo creating new drawings

1 files changed, 3 insertions, 1 deletions

diff --git a/main.lua b/main.lua
index 7be15cf..53751bb 100644
--- a/main.lua
+++ b/main.lua
@@ -219,11 +219,13 @@ function App.draw()
         button('draw', {x=4,y=y+4, w=12,h=12, color={1,1,0},
           icon = icon.insert_drawing,
           onpress1 = function()
-                       Drawing.before = snapshot()
+                       Drawing.before = snapshot(line_index-1, line_index)
                        table.insert(Lines, line_index, {mode='drawing', y=y, h=256/2, points={}, shapes={}, pending={}})
                        if Cursor1.line >= line_index then
                          Cursor1.line = Cursor1.line+1
                        end
+                       save_to_disk(Lines, Filename)
+                       record_undo_event({before=Drawing.before, after=snapshot(line_index-1, line_index+1)})
                      end})
           if Search_term == nil then
             if line_index == Cursor1.line then