@jupytergis/schema 0.1.7 → 0.2.1

package/lib/model.d.ts

@@ -1,4 +1,3 @@
-import { ICollaborativeDrive } from '@jupyter/collaborative-drive';
 import { MapChange } from '@jupyter/ydoc';
 import { IChangedArgs } from '@jupyterlab/coreutils';
 import { DocumentRegistry } from '@jupyterlab/docregistry';
@@ -6,9 +5,10 @@ import { PartialJSONObject } from '@lumino/coreutils';
 import { ISignal, Signal } from '@lumino/signaling';
 import { GeoJSON } from './_interface/geojsonsource';
 import { IJGISContent, IJGISLayer, IJGISLayerGroup, IJGISLayerTree, IJGISLayers, IJGISOptions, IJGISSource, IJGISSources } from './_interface/jgis';
-import { IJGISLayerDocChange, IJGISLayerTreeDocChange, IJGISSourceDocChange, IJupyterGISClientState, IJupyterGISDoc, IJupyterGISModel, ISelection, IUserData } from './interfaces';
+import { IViewPortState, Pointer, IAnnotationModel, IJGISLayerDocChange, IJGISLayerTreeDocChange, IJGISSourceDocChange, IJupyterGISClientState, IJupyterGISDoc, IJupyterGISModel, ISelection, IUserData, IDict } from './interfaces';
+import { Contents } from '@jupyterlab/services';
 export declare class JupyterGISModel implements IJupyterGISModel {
-    constructor(options: DocumentRegistry.IModelOptions<IJupyterGISDoc>);
+    constructor(options: JupyterGISModel.IOptions);
     private _onSharedModelChanged;
     readonly collaborative: boolean;
     get sharedModel(): IJupyterGISDoc;
@@ -30,6 +30,14 @@ export declare class JupyterGISModel implements IJupyterGISModel {
     get sharedLayerTreeChanged(): ISignal<IJupyterGISDoc, IJGISLayerTreeDocChange>;
     get sharedSourcesChanged(): ISignal<IJupyterGISDoc, IJGISSourceDocChange>;
     get disposed(): ISignal<JupyterGISModel, void>;
+    get sharedMetadataChanged(): ISignal<this, MapChange>;
+    get zoomToAnnotationSignal(): ISignal<this, string>;
+    set isIdentifying(isIdentifying: boolean);
+    get isIdentifying(): boolean;
+    centerOnAnnotation(id: string): void;
+    private _metadataChangedHandler;
+    addMetadata(key: string, value: string): void;
+    removeMetadata(key: string): void;
     dispose(): void;
     toString(): string;
     fromString(data: string): void;
@@ -38,7 +46,7 @@ export declare class JupyterGISModel implements IJupyterGISModel {
     initialize(): void;
     getWorker(): Worker;
     getContent(): IJGISContent;
-    setDrive(value: ICollaborativeDrive, filePath: string): void;
+    setContentsManager(value: Contents.IManager | undefined, filePath: string): void;
     getLayers(): IJGISLayers;
     getSources(): IJGISSources;
     getLayerTree(): IJGISLayerTree;
@@ -89,9 +97,12 @@ export declare class JupyterGISModel implements IJupyterGISModel {
     removeLayer(layer_id: string): void;
     setOptions(value: IJGISOptions): void;
     getOptions(): IJGISOptions;
+    syncViewport(viewport?: IViewPortState, emitter?: string): void;
+    syncPointer(pointer?: Pointer, emitter?: string): void;
     syncSelected(value: {
         [key: string]: ISelection;
     }, emitter?: string): void;
+    syncIdentifiedFeatures(features: IDict<any>, emitter?: string): void;
     setUserToFollow(userId?: number): void;
     getClientId(): number;
     /**
@@ -111,15 +122,18 @@ export declare class JupyterGISModel implements IJupyterGISModel {
     }, group: IJGISLayerGroup): void;
     private _removeLayerTreeLayer;
     private _removeLayerTreeGroup;
+    private _removeLayerTreeItem;
     renameLayerGroup(groupName: string, newName: string): void;
     removeLayerGroup(groupName: string): void;
+    toggleIdentify(): void;
     private _getLayerTreeInfo;
     private _onClientStateChanged;
     readonly defaultKernelName: string;
     readonly defaultKernelLanguage: string;
+    readonly annotationModel?: IAnnotationModel;
     private _sharedModel;
     private _filePath;
-    private _drive?;
+    private _contentsManager?;
     private _dirty;
     private _readOnly;
     private _isDisposed;
@@ -130,6 +144,9 @@ export declare class JupyterGISModel implements IJupyterGISModel {
     private _stateChanged;
     private _themeChanged;
     private _clientStateChanged;
+    private _sharedMetadataChanged;
+    private _zoomToAnnotationSignal;
+    private _isIdentifying;
     static worker: Worker;
 }
 export declare namespace JupyterGISModel {
@@ -137,4 +154,7 @@ export declare namespace JupyterGISModel {
      * Function to get the ordered list of layers according to the tree.
      */
     function getOrderedLayerIds(model: IJupyterGISModel): string[];
+    interface IOptions extends DocumentRegistry.IModelOptions<IJupyterGISDoc> {
+        annotationModel?: IAnnotationModel;
+    }
 }

package/lib/model.js

@@ -16,11 +16,9 @@ export class JupyterGISModel {
         this._onClientStateChanged = (changed) => {
             const clients = this.sharedModel.awareness.getStates();
             this._clientStateChanged.emit(clients);
-            this._sharedModel.awareness.on('change', (update) => {
-                if (update.added.length || update.removed.length) {
-                    this._userChanged.emit(this.users);
-                }
-            });
+            if (changed.added.length || changed.removed.length) {
+                this._userChanged.emit(this.users);
+            }
         };
         this.defaultKernelName = '';
         this.defaultKernelLanguage = '';
@@ -33,7 +31,10 @@ export class JupyterGISModel {
         this._stateChanged = new Signal(this);
         this._themeChanged = new Signal(this);
         this._clientStateChanged = new Signal(this);
-        const { sharedModel } = options;
+        this._sharedMetadataChanged = new Signal(this);
+        this._zoomToAnnotationSignal = new Signal(this);
+        this._isIdentifying = false;
+        const { annotationModel, sharedModel } = options;
         if (sharedModel) {
             this._sharedModel = sharedModel;
         }
@@ -42,6 +43,8 @@ export class JupyterGISModel {
             this._sharedModel.changed.connect(this._onSharedModelChanged);
         }
         this.sharedModel.awareness.on('change', this._onClientStateChanged);
+        this._sharedModel.metadataChanged.connect(this._metadataChangedHandler, this);
+        this.annotationModel = annotationModel;
     }
     get sharedModel() {
         return this._sharedModel;
@@ -109,6 +112,30 @@ export class JupyterGISModel {
     get disposed() {
         return this._disposed;
     }
+    get sharedMetadataChanged() {
+        return this._sharedMetadataChanged;
+    }
+    get zoomToAnnotationSignal() {
+        return this._zoomToAnnotationSignal;
+    }
+    set isIdentifying(isIdentifying) {
+        this._isIdentifying = isIdentifying;
+    }
+    get isIdentifying() {
+        return this._isIdentifying;
+    }
+    centerOnAnnotation(id) {
+        this._zoomToAnnotationSignal.emit(id);
+    }
+    _metadataChangedHandler(_, args) {
+        this._sharedMetadataChanged.emit(args);
+    }
+    addMetadata(key, value) {
+        this.sharedModel.setMetadata(key, value);
+    }
+    removeMetadata(key) {
+        this.sharedModel.removeMetadata(key);
+    }
     dispose() {
         if (this._isDisposed) {
             return;
@@ -134,7 +161,7 @@ export class JupyterGISModel {
             console.warn(errorMsg);
         }
         this.sharedModel.transact(() => {
-            var _a, _b, _c, _d;
+            var _a, _b, _c, _d, _e;
             this.sharedModel.sources = (_a = jsonData.sources) !== null && _a !== void 0 ? _a : {};
             this.sharedModel.layers = (_b = jsonData.layers) !== null && _b !== void 0 ? _b : {};
             this.sharedModel.layerTree = (_c = jsonData.layerTree) !== null && _c !== void 0 ? _c : [];
@@ -146,6 +173,7 @@ export class JupyterGISModel {
                 pitch: 0,
                 projection: 'EPSG:3857'
             };
+            this.sharedModel.metadata = (_e = jsonData.metadata) !== null && _e !== void 0 ? _e : {};
         });
         this.dirty = true;
     }
@@ -166,11 +194,12 @@ export class JupyterGISModel {
             sources: this.sharedModel.sources,
             layers: this.sharedModel.layers,
             layerTree: this.sharedModel.layerTree,
-            options: this.sharedModel.options
+            options: this.sharedModel.options,
+            metadata: this.sharedModel.metadata
         };
     }
-    setDrive(value, filePath) {
-        this._drive = value;
+    setContentsManager(value, filePath) {
+        this._contentsManager = value;
         this._filePath = filePath;
     }
     getLayers() {
@@ -225,22 +254,17 @@ export class JupyterGISModel {
      * @returns a promise to the GeoJSON data.
      */
     async readGeoJSON(filepath) {
-        if (!this._drive) {
+        if (!this._contentsManager) {
             return;
         }
-        let dir = PathExt.dirname(this._filePath);
-        if (dir.includes(':')) {
-            dir = dir.split(':')[1];
-        }
-        const absolutePath = PathExt.join(dir, filepath);
-        return this._drive
-            .get(absolutePath)
-            .then(contentModel => {
-            return JSON.parse(contentModel.content);
-        })
-            .catch(e => {
-            throw e;
+        const absolutePath = PathExt.resolve(PathExt.dirname(this._filePath), filepath);
+        const file = await this._contentsManager.get(absolutePath, {
+            content: true
         });
+        if (typeof file.content === 'string') {
+            return JSON.parse(file.content);
+        }
+        return file.content;
     }
     /**
      * Add a layer group in the layer tree.
@@ -289,10 +313,28 @@ export class JupyterGISModel {
     getOptions() {
         return this._sharedModel.options;
     }
+    syncViewport(viewport, emitter) {
+        this.sharedModel.awareness.setLocalStateField('viewportState', {
+            value: viewport,
+            emitter
+        });
+    }
+    syncPointer(pointer, emitter) {
+        this.sharedModel.awareness.setLocalStateField('pointer', {
+            value: pointer,
+            emitter
+        });
+    }
     syncSelected(value, emitter) {
         this.sharedModel.awareness.setLocalStateField('selected', {
             value,
-            emitter: emitter
+            emitter
+        });
+    }
+    syncIdentifiedFeatures(features, emitter) {
+        this.sharedModel.awareness.setLocalStateField('identifiedFeatures', {
+            value: features,
+            emitter
         });
     }
     setUserToFollow(userId) {
@@ -385,29 +427,22 @@ export class JupyterGISModel {
         this._addLayerTreeItem(group);
     }
     _removeLayerTreeLayer(layerTree, layerIdToRemove) {
-        // Iterate over each item in the layerTree
-        for (let i = 0; i < layerTree.length; i++) {
-            const currentItem = layerTree[i];
-            // Check if the current item is a string and matches the target
-            if (typeof currentItem === 'string' && currentItem === layerIdToRemove) {
-                // Remove the item from the array
-                layerTree.splice(i, 1);
-                // Decrement i to ensure the next iteration processes the remaining items correctly
-                i--;
-            }
-            else if (typeof currentItem !== 'string' && 'layers' in currentItem) {
-                // If the current item is a group, recursively call the function on its layers
-                this._removeLayerTreeLayer(currentItem.layers, layerIdToRemove);
-            }
-        }
+        this._removeLayerTreeItem(layerTree, layerIdToRemove, true);
         this.sharedModel.layerTree = layerTree;
     }
     _removeLayerTreeGroup(layerTree, groupName) {
+        this._removeLayerTreeItem(layerTree, groupName, false);
+        this.sharedModel.layerTree = layerTree;
+    }
+    _removeLayerTreeItem(layerTree, target, isLayer) {
         // Iterate over each item in the layerTree
         for (let i = 0; i < layerTree.length; i++) {
             const currentItem = layerTree[i];
+            const matches = isLayer
+                ? typeof currentItem === 'string' && currentItem === target
+                : typeof currentItem !== 'string' && currentItem.name === target;
             // Check if the current item is a string and matches the target
-            if (typeof currentItem !== 'string' && currentItem.name === groupName) {
+            if (matches) {
                 // Remove the item from the array
                 layerTree.splice(i, 1);
                 // Decrement i to ensure the next iteration processes the remaining items correctly
@@ -415,10 +450,9 @@ export class JupyterGISModel {
             }
             else if (typeof currentItem !== 'string' && 'layers' in currentItem) {
                 // If the current item is a group, recursively call the function on its layers
-                this._removeLayerTreeGroup(currentItem.layers, groupName);
+                this._removeLayerTreeItem(currentItem.layers, target, isLayer);
             }
         }
-        this.sharedModel.layerTree = layerTree;
     }
     renameLayerGroup(groupName, newName) {
         const layerTreeInfo = this._getLayerTreeInfo(groupName);
@@ -451,6 +485,9 @@ export class JupyterGISModel {
             this._sharedModel.updateLayerTreeItem(layerTreeInfo.mainGroupIndex, updatedLayerTree[layerTreeInfo.mainGroupIndex]);
         }
     }
+    toggleIdentify() {
+        this._isIdentifying = !this._isIdentifying;
+    }
     _getLayerTreeInfo(groupName) {
         const layerTree = this.getLayerTree();
         const indexesPath = Private.findItemPath(layerTree, groupName);