import { action, makeObservable, observable, observe, runInAction } from 'mobx';
import { computedFn } from 'mobx-utils';
import { Doc, DocListCast, DocListCastAsync, Field, Opt } from '../../fields/Doc';
import { DirectLinks } from '../../fields/DocSymbols';
import { FieldLoader } from '../../fields/FieldLoader';
import { List } from '../../fields/List';
import { ProxyField } from '../../fields/Proxy';
import { Cast, DocCast, PromiseValue, StrCast } from '../../fields/Types';
import { DocServer } from '../DocServer';
import { ScriptingGlobals } from './ScriptingGlobals';
/*
 * link doc:
 * - link_anchor_1: doc
 * - link_anchor_2: doc
 *
 * group doc:
 * - type: string representing the group type/name/category
 * - metadata: doc representing the metadata kvps
 *
 * metadata doc:
 * - user defined kvps
 */
export class LinkManager {
    @observable static _instance: LinkManager;
    @observable.shallow userLinkDBs: Doc[] = [];
    @observable public static currentLink: Opt<Doc>;
    @observable public static currentLinkAnchor: Opt<Doc>;
    public static get Instance() {
        return LinkManager._instance;
    }

    public static Links(doc: Doc | undefined) {
        return doc ? LinkManager.Instance.getAllRelatedLinks(doc) : [];
    }
    public addLinkDB = async (linkDb: any) => {
        await Promise.all(
            ((await DocListCastAsync(linkDb.data)) ?? []).map(link =>
                // makes sure link anchors are loaded to avoid incremental updates to computedFns in LinkManager
                [PromiseValue(link?.link_anchor_1), PromiseValue(link?.link_anchor_2)]
            )
        );
        this.userLinkDBs.push(linkDb);
    };
    public static AutoKeywords = 'keywords:Usages';
    static _links: Doc[] = [];
    constructor() {
        makeObservable(this);
        LinkManager._instance = this;
        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
        // Then add the link to the anchor protos.
        const addLinkToDoc = (lprom: Doc) =>
            PromiseValue(lprom).then((link: Opt<Doc>) =>
                PromiseValue(link?.proto as Doc).then((lproto: Opt<Doc>) =>
                    Promise.all([lproto?.link_anchor_1 as Doc, lproto?.link_anchor_2 as Doc].map(PromiseValue)).then((lAnchs: Opt<Doc>[]) =>
                        Promise.all(lAnchs.map(lAnch => PromiseValue(lAnch?.proto as Doc))).then((lAnchProtos: Opt<Doc>[]) =>
                            Promise.all(lAnchProtos.map(lAnchProto => PromiseValue(lAnchProto?.proto as Doc))).then(
                                link &&
                                    action(lAnchProtoProtos => {
                                        Doc.AddDocToList(Doc.UserDoc(), 'links', link);
                                        lAnchs[0] && Doc.GetProto(lAnchs[0])[DirectLinks].add(link);
                                        lAnchs[1] && Doc.GetProto(lAnchs[1])[DirectLinks].add(link);
                                    })
                            )
                        )
                    )
                )
            );

        const remLinkFromDoc = (lprom: Doc) =>
            PromiseValue(lprom).then((link: Opt<Doc>) =>
                PromiseValue(link?.proto as Doc).then((lproto: Opt<Doc>) =>
                    Promise.all([lproto?.link_anchor_1 as Doc, lproto?.link_anchor_2 as Doc].map(PromiseValue)).then((lAnchs: Opt<Doc>[]) =>
                        Promise.all(lAnchs.map(lAnch => PromiseValue(lAnch?.proto as Doc))).then((lAnchProtos: Opt<Doc>[]) =>
                            Promise.all(lAnchProtos.map(lAnchProto => PromiseValue(lAnchProto?.proto as Doc))).then(
                                action(lAnchProtoProtos => {
                                    link && lAnchs[0] && Doc.GetProto(lAnchs[0])[DirectLinks].delete(link);
                                    link && lAnchs[1] && Doc.GetProto(lAnchs[1])[DirectLinks].delete(link);
                                })
                            )
                        )
                    )
                )
            );

        const watchUserLinkDB = (userLinkDBDoc: Doc) => {
            LinkManager._links.push(...DocListCast(userLinkDBDoc.data));
            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
            if (userLinkDBDoc.data) {
                observe(
                    userLinkDBDoc.data,
                    change => {
                        // observe pushes/splices on a user link DB 'data' field (should only happen for local changes)
                        switch (change.type as any) {
                            case 'splice':
                                (change as any).added.forEach((link: any) => addLinkToDoc(toRealField(link)));
                                (change as any).removed.forEach((link: any) => remLinkFromDoc(toRealField(link)));
                                break;
                            case 'update': //let oldValue = change.oldValue;
                        }
                    },
                    true
                );
                observe(
                    userLinkDBDoc,
                    'data', // obsever when a new array of links is assigned as the link DB 'data' field (should happen whenever a remote user adds/removes a link)
                    change => {
                        switch (change.type as any) {
                            case 'update':
                                Promise.all([...((change.oldValue as any as Doc[]) || []), ...((change.newValue as any as Doc[]) || [])]).then(doclist => {
                                    const oldDocs = doclist.slice(0, ((change.oldValue as any as Doc[]) || []).length);
                                    const newDocs = doclist.slice(((change.oldValue as any as Doc[]) || []).length, doclist.length);

                                    const added = newDocs?.filter(link => !(oldDocs || []).includes(link));
                                    const removed = oldDocs?.filter(link => !(newDocs || []).includes(link));
                                    added?.forEach((link: any) => addLinkToDoc(toRealField(link)));
                                    removed?.forEach((link: any) => remLinkFromDoc(toRealField(link)));
                                });
                        }
                    },
                    true
                );
            }
        };
        observe(
            this.userLinkDBs,
            change => {
                switch (change.type as any) {
                    case 'splice':
                        (change as any).added.forEach(watchUserLinkDB);
                        break;
                    case 'update': //let oldValue = change.oldValue;
                }
            },
            true
        );
        runInAction(() => (FieldLoader.ServerLoadStatus.message = 'links'));
        this.addLinkDB(Doc.LinkDBDoc());
    }

    public createlink_relationshipLists = () => {
        //create new lists for link relations and their associated colors if the lists don't already exist
        !Doc.UserDoc().link_relationshipList && (Doc.UserDoc().link_relationshipList = new List<string>());
        !Doc.UserDoc().link_ColorList && (Doc.UserDoc().link_ColorList = new List<string>());
        !Doc.UserDoc().link_relationshipSizes && (Doc.UserDoc().link_relationshipSizes = new List<number>());
    };

    public addLink(linkDoc: Doc, checkExists = false) {
        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);
        }
    }
    public deleteLink(linkDoc: Doc) {
        return Doc.RemoveDocFromList(Doc.LinkDBDoc(), 'data', linkDoc);
    }
    public deleteAllLinksOnAnchor(anchor: Doc) {
        LinkManager.Instance.relatedLinker(anchor).forEach(linkDoc => LinkManager.Instance.deleteLink(linkDoc));
    }

    public getAllRelatedLinks(anchor: Doc) {
        return this.relatedLinker(anchor);
    } // finds all links that contain the given anchor
    public getAllDirectLinks(anchor?: Doc): Doc[] {
        return anchor ? Array.from(Doc.GetProto(anchor)[DirectLinks]) : [];
    } // finds all links that contain the given anchor

    relatedLinker = computedFn(function relatedLinker(this: any, anchor: Doc): Doc[] {
        if (!anchor || anchor instanceof Promise || Doc.GetProto(anchor) instanceof Promise) {
            console.log('WAITING FOR DOC/PROTO IN LINKMANAGER');
            return [];
        }
        const dirLinks = Doc.GetProto(anchor)[DirectLinks];
        const annos = DocListCast(anchor[Doc.LayoutFieldKey(anchor) + '_annotations']);
        if (!annos) debugger;
        return annos.reduce((list, anno) => [...list, ...LinkManager.Instance.relatedLinker(anno)], Array.from(dirLinks).slice());
    }, true);

    // returns map of group type to anchor's links in that group type
    public getRelatedGroupedLinks(anchor: Doc): Map<string, Array<Doc>> {
        const anchorGroups = new Map<string, Array<Doc>>();
        this.relatedLinker(anchor).forEach(link => {
            if (link.link_relationship && link.link_relationship !== '-ungrouped-') {
                const relation = StrCast(link.link_relationship);
                const anchorRelation = relation.indexOf(':') !== -1 ? relation.split(':')[Doc.AreProtosEqual(Cast(link.link_anchor_1, Doc, null), anchor) ? 0 : 1] : relation;
                const group = anchorGroups.get(anchorRelation);
                anchorGroups.set(anchorRelation, group ? [...group, link] : [link]);
            } else {
                // if link is in no groups then put it in default group
                const group = anchorGroups.get('*');
                anchorGroups.set('*', group ? [...group, link] : [link]);
            }
        });
        return anchorGroups;
    }

    // finds the opposite anchor of a given anchor in a link
    public static getOppositeAnchor(linkDoc: Doc, anchor: Doc): Doc | undefined {
        const id = LinkManager.anchorIndex(linkDoc, anchor);
        const a1 = DocCast(linkDoc.link_anchor_1);
        const a2 = DocCast(linkDoc.link_anchor_2);
        return id === '1' ? a2 : id === '2' ? a1 : id === '0' ? linkDoc : undefined;
        // if (Doc.AreProtosEqual(DocCast(anchor.annotationOn, anchor), DocCast(a1?.annotationOn, a1))) return a2;
        // if (Doc.AreProtosEqual(DocCast(anchor.annotationOn, anchor), DocCast(a2?.annotationOn, a2))) return a1;
        // if (Doc.AreProtosEqual(anchor, linkDoc)) return linkDoc;
    }
    public static anchorIndex(linkDoc: Doc, anchor: Doc) {
        const a1 = DocCast(linkDoc.link_anchor_1);
        const a2 = DocCast(linkDoc.link_anchor_2);
        if (linkDoc.link_matchEmbeddings) {
            return [a2, a2.annotationOn].includes(anchor) ? '2' : '1';
        }
        if (Doc.AreProtosEqual(DocCast(anchor.annotationOn, anchor), DocCast(a1?.annotationOn, a1))) return '1';
        if (Doc.AreProtosEqual(DocCast(anchor.annotationOn, anchor), DocCast(a2?.annotationOn, a2))) return '2';
        if (Doc.AreProtosEqual(anchor, linkDoc)) return '0';

        // const a1 = DocCast(linkDoc.link_anchor_1);
        // const a2 = DocCast(linkDoc.link_anchor_2);
        // if (linkDoc.link_matchEmbeddings) {
        //     return [a2, a2.annotationOn].includes(anchor) ? '2' : '1';
        // }
        // if (Doc.AreProtosEqual(a2, anchor) || Doc.AreProtosEqual(a2.annotationOn as Doc, anchor)) return '2';
        // return Doc.AreProtosEqual(a1, anchor) || Doc.AreProtosEqual(a1.annotationOn as Doc, anchor) ? '1' : '2';
    }
}

ScriptingGlobals.add(
    function links(doc: any) {
        return new List(LinkManager.Links(doc));
    },
    'returns all the links to the document or its annotations',
    '(doc: any)'
);