1 files changed, 24 insertions, 35 deletions
diff --git a/src/client/views/newlightbox/Header/LightboxHeader.tsx b/src/client/views/newlightbox/Header/LightboxHeader.tsx
index 76efe0185..882d28fba 100644
--- a/src/client/views/newlightbox/Header/LightboxHeader.tsx
+++ b/src/client/views/newlightbox/Header/LightboxHeader.tsx
@@ -4,21 +4,21 @@ import { BsBookmark, BsBookmarkFill } from 'react-icons/bs';
 import { MdTravelExplore } from 'react-icons/md';
 import { Doc } from '../../../../fields/Doc';
 import { StrCast } from '../../../../fields/Types';
-import { LightboxView } from '../../LightboxView';
 import { Colors } from '../../global/globalEnums';
+import { DocumentView } from '../../nodes/DocumentView';
 import { NewLightboxView } from '../NewLightboxView';
 import { EditableText } from '../components/EditableText';
 import { getType } from '../utils';
 import './LightboxHeader.scss';
 import { INewLightboxHeader } from './utils';
 
-export const NewLightboxHeader = (props: INewLightboxHeader) => {
+export function NewLightboxHeader(props: INewLightboxHeader) {
     const { height = 100, width } = props;
-    const [doc, setDoc] = React.useState<Doc | undefined>(LightboxView.LightboxDoc);
+    const [doc, setDoc] = React.useState<Doc | undefined>(DocumentView.LightboxDoc());
     const [editing, setEditing] = React.useState<boolean>(false);
     const [title, setTitle] = React.useState<JSX.Element | null>(null);
     React.useEffect(() => {
-        let lbDoc = LightboxView.LightboxDoc;
+        const lbDoc = DocumentView.LightboxDoc();
         setDoc(lbDoc);
         if (lbDoc) {
             setTitle(
@@ -32,39 +32,28 @@ export const NewLightboxHeader = (props: INewLightboxHeader) => {
                 />
             );
         }
-    }, [LightboxView.LightboxDoc]);
+    }, [DocumentView.LightboxDoc()]);
 
     const [saved, setSaved] = React.useState<boolean>(false);
 
     if (!doc) return null;
-    else
-        return (
-            <div className={`newLightboxHeader-container`} onPointerDown={e => e.stopPropagation()} style={{ minHeight: height, height: height, width: width }}>
-                <div className={`title-container`}>
-                    <div className={`lb-label`}>Title</div>
-                    {title}
-                </div>
-                <div className={`type-container`}>
-                    <div className={`lb-label`}>Type</div>
-                    <div className={`type`}>{getType(StrCast(doc.type))}</div>
-                </div>
-                <div style={{ gridColumn: 2, gridRow: 1, height: '100%', display: 'flex', justifyContent: 'flex-end', alignItems: 'center' }}>
-                    <IconButton size={Size.XSMALL} onClick={() => setSaved(!saved)} color={Colors.DARK_GRAY} icon={saved ? <BsBookmarkFill /> : <BsBookmark />} />
-                    <IconButton size={Size.XSMALL} onClick={() => setSaved(!saved)} color={Colors.DARK_GRAY} icon={saved ? <BsBookmarkFill /> : <BsBookmark />} />
-                </div>
-                <div style={{ gridColumn: 2, gridRow: 2, height: '100%', display: 'flex', justifyContent: 'flex-end', alignItems: 'center' }}>
-                    <Button
-                        onClick={() => {
-                            console.log(NewLightboxView.ExploreMode);
-                            NewLightboxView.SetExploreMode(!NewLightboxView.ExploreMode);
-                        }}
-                        size={Size.XSMALL}
-                        color={Colors.DARK_GRAY}
-                        type={Type.SEC}
-                        text={'t-SNE 2D Embeddings'}
-                        icon={<MdTravelExplore />}
-                    />
-                </div>
+    return (
+        <div className="newLightboxHeader-container" onPointerDown={e => e.stopPropagation()} style={{ minHeight: height, height: height, width: width }}>
+            <div className="title-container">
+                <div className="lb-label">Title</div>
+                {title}
             </div>
-        );
-};
+            <div className="type-container">
+                <div className="lb-label">Type</div>
+                <div className="type">{getType(StrCast(doc.type))}</div>
+            </div>
+            <div style={{ gridColumn: 2, gridRow: 1, height: '100%', display: 'flex', justifyContent: 'flex-end', alignItems: 'center' }}>
+                <IconButton size={Size.XSMALL} onClick={() => setSaved(!saved)} color={Colors.DARK_GRAY} icon={saved ? <BsBookmarkFill /> : <BsBookmark />} />
+                <IconButton size={Size.XSMALL} onClick={() => setSaved(!saved)} color={Colors.DARK_GRAY} icon={saved ? <BsBookmarkFill /> : <BsBookmark />} />
+            </div>
+            <div style={{ gridColumn: 2, gridRow: 2, height: '100%', display: 'flex', justifyContent: 'flex-end', alignItems: 'center' }}>
+                <Button onClick={() => NewLightboxView.SetExploreMode(!NewLightboxView.ExploreMode)} size={Size.XSMALL} color={Colors.DARK_GRAY} type={Type.SEC} text="t-SNE 2D Embeddings" icon={<MdTravelExplore />} />
+            </div>
+        </div>
+    );
+}