fixed loading newly created links. fixed following links to contexts that haven't been loaded yet.

2 files changed, 21 insertions, 20 deletions

diff --git a/src/client/util/LinkFollower.ts b/src/client/util/LinkFollower.ts
index b9cd0c213..807b54cd5 100644
--- a/src/client/util/LinkFollower.ts
+++ b/src/client/util/LinkFollower.ts
@@ -97,7 +97,7 @@ export class LinkFollower {
                     : linkDoc.anchor1
             ) as Doc;
             if (target) {
-                const doFollow = (canToggle?: boolean) => {
+                const doFollow = async (canToggle?: boolean) => {
                     const options: DocFocusOptions = {
                         playAudio: BoolCast(sourceDoc.followLinkAudio),
                         toggleTarget: canToggle && BoolCast(sourceDoc.followLinkToggle),
@@ -119,9 +119,9 @@ export class LinkFollower {
                     } else {
                         const containerAnnoDoc = Cast(target.annotationOn, Doc, null);
                         const containerDoc = containerAnnoDoc || target;
-                        var containerDocContext = containerDoc?.context ? [Cast(containerDoc?.context, Doc, null)] : ([] as Doc[]);
-                        while (containerDocContext.length && !DocumentManager.Instance.getDocumentView(containerDocContext[0]) && containerDocContext[0].context && DocCast(containerDocContext[0].context).viewType !== CollectionViewType.Docking) {
-                            containerDocContext = [Cast(containerDocContext[0].context, Doc, null), ...containerDocContext];
+                        var containerDocContext = containerDoc?.context ? [Cast(await containerDoc?.context, Doc, null)] : ([] as Doc[]);
+                        while (containerDocContext.length && containerDocContext[0]?.context && !DocumentManager.Instance.getDocumentView(containerDocContext[0]) && DocCast(containerDocContext[0].context)?.viewType !== CollectionViewType.Docking) {
+                            containerDocContext = [Cast(await containerDocContext[0].context, Doc, null), ...containerDocContext];
                         }
                         const targetContexts = LightboxView.LightboxDoc ? [containerAnnoDoc || containerDocContext[0]].filter(a => a) : containerDocContext;
                         DocumentManager.Instance.jumpToDocument(target, options, (doc, finished) => createViewFunc(doc, StrCast(linkDoc.followLinkLocation, OpenWhere.inPlace), finished), targetContexts, allFinished);
diff --git a/src/client/util/LinkManager.ts b/src/client/util/LinkManager.ts
index 2b0ce1d3d..4deff3c60 100644
--- a/src/client/util/LinkManager.ts
+++ b/src/client/util/LinkManager.ts
@@ -3,7 +3,7 @@ import { computedFn } from 'mobx-utils';
 import { DirectLinksSym, Doc, DocListCast, DocListCastAsync, Field, Opt } from '../../fields/Doc';
 import { List } from '../../fields/List';
 import { ProxyField } from '../../fields/Proxy';
-import { Cast, StrCast } from '../../fields/Types';
+import { Cast, DocCast, StrCast } from '../../fields/Types';
 /*
  * link doc:
  * - anchor1: doc
@@ -32,21 +32,22 @@ export class LinkManager {
         this.createLinkrelationshipLists();
         LinkManager.userLinkDBs = [];
         const addLinkToDoc = (link: Doc) => {
-            const a1Prom = link?.anchor1;
-            const a2Prom = link?.anchor2;
-            Promise.all([a1Prom, a2Prom]).then(all => {
-                const a1 = all[0];
-                const a2 = all[1];
-                const a1ProtoProm = (link?.anchor1 as Doc)?.proto;
-                const a2ProtoProm = (link?.anchor2 as Doc)?.proto;
-                Promise.all([a1ProtoProm, a2ProtoProm]).then(
-                    action(all => {
-                        if (a1 instanceof Doc && a2 instanceof Doc && ((a1.author !== undefined && a2.author !== undefined) || link.author === Doc.CurrentUserEmail)) {
-                            Doc.GetProto(a1)[DirectLinksSym].add(link);
-                            Doc.GetProto(a2)[DirectLinksSym].add(link);
-                        }
-                    })
-                );
+            Promise.all([link]).then(linkdoc => {
+                const link = DocCast(linkdoc[0]);
+                Promise.all([link.proto]).then(linkproto => {
+                    Promise.all([link.anchor1, link.anchor2]).then(linkdata => {
+                        const a1 = DocCast(linkdata[0]);
+                        const a2 = DocCast(linkdata[1]);
+                        a1 &&
+                            a2 &&
+                            Promise.all([a1.proto, a2.proto]).then(
+                                action(protos => {
+                                    (protos[0] as Doc)?.[DirectLinksSym].add(link);
+                                    (protos[1] as Doc)?.[DirectLinksSym].add(link);
+                                })
+                            );
+                    });
+                });
             });
         };
         const remLinkFromDoc = (link: Doc) => {