From 692076b1356309111c4f2cb69cbdbf4be1a825bd Mon Sep 17 00:00:00 2001
From: eleanor-park <eleanor_park@brown.edu>
Date: Sun, 22 Sep 2024 15:40:19 -0400
Subject: small bug fixes for smart draw

---
 .../collectionFreeForm/CollectionFreeFormView.tsx        | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

(limited to 'src/client/views/collections/collectionFreeForm')

diff --git a/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx b/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx
index 4b4a07757..ad1223097 100644
--- a/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx
+++ b/src/client/views/collections/collectionFreeForm/CollectionFreeFormView.tsx
@@ -43,6 +43,7 @@ import { FocusViewOptions } from '../../nodes/FocusViewOptions';
 import { FormattedTextBox } from '../../nodes/formattedText/FormattedTextBox';
 import { OpenWhere, OpenWhereMod } from '../../nodes/OpenWhere';
 import { PinDocView, PinProps } from '../../PinFuncs';
+import { AnnotationPalette } from '../../smartdraw/AnnotationPalette';
 import { DrawingOptions, SmartDrawHandler } from '../../smartdraw/SmartDrawHandler';
 import { StyleProp } from '../../StyleProp';
 import { CollectionSubView, SubCollectionViewProps } from '../CollectionSubView';
@@ -1269,21 +1270,22 @@ export class CollectionFreeFormView extends CollectionSubView<Partial<collection
             this._drawing.push(inkDoc);
         });
         const collection = MarqueeView.getCollection(this._drawing, undefined, true, { left: 1, top: 1, width: 1, height: 1 });
+        console.log('is error here');
         return collection;
     };
 
     /**
      * Part of regenerating a drawing--deletes the old drawing.
      */
-    removeDrawing = (doc?: Doc) => {
+    removeDrawing = (useLastContainer: boolean, doc?: Doc) => {
         this._batch = UndoManager.StartBatch('regenerateDrawing');
-        if (doc) {
+        if (useLastContainer && this._drawingContainer) {
+            this._props.removeDocument?.(this._drawingContainer);
+        } else if (doc) {
             const docData = doc[DocData];
             const children = DocListCast(docData.data);
             this._props.removeDocument?.(doc);
             this._props.removeDocument?.(children);
-        } else {
-            if (this._drawingContainer) this._props.removeDocument?.(this._drawingContainer);
         }
         this._drawing = [];
     };
@@ -1294,6 +1296,7 @@ export class CollectionFreeFormView extends CollectionSubView<Partial<collection
     addDrawing = (doc: Doc, opts: DrawingOptions, gptRes: string) => {
         const docData = doc[DocData];
         docData.title = opts.text.match(/^(.*?)~~~.*$/)?.[1] || opts.text;
+        docData.width = opts.size;
         docData.drawingInput = opts.text;
         docData.drawingComplexity = opts.complexity;
         docData.drawingColored = opts.autoColor;
@@ -2010,6 +2013,11 @@ export class CollectionFreeFormView extends CollectionSubView<Partial<collection
             }),
             icon: 'pen-to-square',
         });
+        optionItems.push({
+            description: this.Document.savedAsAnno ? 'Saved as Annotation!' : 'Save to Annotation Palette',
+            event: action(undoable(async () => await AnnotationPalette.addToPalette(this.Document), 'save to palette')),
+            icon: this.Document.savedAsAnno ? 'clipboard-check' : 'file-arrow-down',
+        });
         this._props.renderDepth &&
             optionItems.push({
                 description: 'Use Background Color as Default',
-- 
cgit v1.2.3-70-g09d2