1 files changed, 45 insertions, 4 deletions

diff --git a/src/client/util/LinkManager.ts b/src/client/util/LinkManager.ts
index 8972bf705..82cd791cc 100644
--- a/src/client/util/LinkManager.ts
+++ b/src/client/util/LinkManager.ts
@@ -1,12 +1,16 @@
-import { action, makeObservable, observable, observe, runInAction } from 'mobx';
+import { action, makeObservable, observable, observe } from 'mobx';
 import { computedFn } from 'mobx-utils';
-import { Doc, DocListCast, DocListCastAsync, Field, Opt } from '../../fields/Doc';
+import * as rp from 'request-promise';
+import { ClientUtils } from '../../ClientUtils';
+import { Doc, DocListCast, DocListCastAsync, FieldType, Opt } from '../../fields/Doc';
 import { DirectLinks, DocData } from '../../fields/DocSymbols';
 import { FieldLoader } from '../../fields/FieldLoader';
+import { Id } from '../../fields/FieldSymbols';
 import { List } from '../../fields/List';
 import { ProxyField } from '../../fields/Proxy';
 import { Cast, DocCast, PromiseValue, StrCast } from '../../fields/Types';
 import { DocServer } from '../DocServer';
+import { DocumentType } from '../documents/DocumentTypes';
 import { ScriptingGlobals } from './ScriptingGlobals';
 /*
  * link doc:
@@ -45,6 +49,7 @@ export class LinkManager {
     constructor() {
         makeObservable(this);
         LinkManager._instance = this;
+        Doc.AddLink = this.addLink;
         this.createlink_relationshipLists();
         // since this is an action, not a reaction, we get only one shot to add this link to the Anchor docs
         // Thus make sure all promised values are resolved from link -> link.proto -> link.link_anchor_[1,2] -> link.link_anchor_[1,2].proto
@@ -84,7 +89,7 @@ export class LinkManager {
             );
 
         const watchUserLinkDB = (userLinkDBDoc: Doc) => {
-            const toRealField = (field: Field) => (field instanceof ProxyField ? field.value : field); // see List.ts.  data structure is not a simple list of Docs, but a list of ProxyField/Fields
+            const toRealField = (field: FieldType) => (field instanceof ProxyField ? field.value : field); // see List.ts.  data structure is not a simple list of Docs, but a list of ProxyField/Fields
             if (userLinkDBDoc.data) {
                 observe(
                     userLinkDBDoc.data,
@@ -148,7 +153,7 @@ export class LinkManager {
         Doc.AddDocToList(Doc.UserDoc(), 'links', linkDoc);
         if (!checkExists || !DocListCast(Doc.LinkDBDoc().data).includes(linkDoc)) {
             Doc.AddDocToList(Doc.LinkDBDoc(), 'data', linkDoc);
-            setTimeout(DocServer.UPDATE_SERVER_CACHE, 100);
+            setTimeout(UPDATE_SERVER_CACHE, 100);
         }
     }
     public deleteLink(linkDoc: Doc) {
@@ -240,6 +245,42 @@ export class LinkManager {
     }
 }
 
+let cacheDocumentIds = ''; // ; separate string of all documents ids in the user's working set (cached on the server)
+export function UPDATE_SERVER_CACHE() {
+    const prototypes = Object.values(DocumentType)
+        .filter(type => type !== DocumentType.NONE)
+        .map(type => DocServer._cache[type + 'Proto'])
+        .filter(doc => doc instanceof Doc)
+        .map(doc => doc as Doc);
+    const references = new Set<Doc>(prototypes);
+    Doc.FindReferences(Doc.UserDoc(), references, undefined);
+    DocListCast(DocCast(Doc.UserDoc().myLinkDatabase).data).forEach(link => {
+        if (!references.has(DocCast(link.link_anchor_1)) && !references.has(DocCast(link.link_anchor_2))) {
+            Doc.RemoveDocFromList(DocCast(Doc.UserDoc().myLinkDatabase), 'data', link);
+            Doc.AddDocToList(Doc.MyRecentlyClosed, undefined, link);
+        }
+    });
+    LinkManager.Instance.userLinkDBs.forEach(linkDb => Doc.FindReferences(linkDb, references, undefined));
+    const filtered = Array.from(references);
+
+    const newCacheUpdate = filtered.map(doc => doc[Id]).join(';');
+    if (newCacheUpdate === cacheDocumentIds) return;
+    cacheDocumentIds = newCacheUpdate;
+
+    // print out cached docs
+    Doc.MyDockedBtns.linearView_IsOpen && console.log('Set cached docs = ');
+    const isFiltered = filtered.filter(doc => !Doc.IsSystem(doc));
+    const strings = isFiltered.map(doc => StrCast(doc.title) + ' ' + (Doc.IsDataProto(doc) ? '(data)' : '(embedding)'));
+    Doc.MyDockedBtns.linearView_IsOpen && strings.sort().forEach((str, i) => console.log(i.toString() + ' ' + str));
+
+    rp.post(ClientUtils.prepend('/setCacheDocumentIds'), {
+        body: {
+            cacheDocumentIds,
+        },
+        json: true,
+    });
+}
+
 ScriptingGlobals.add(
     function links(doc: any) {
         return new List(LinkManager.Links(doc));