@aguacerowx/javascript-sdk 0.0.15 → 0.0.18

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aguacerowx/javascript-sdk",
-  "version": "0.0.15",
+  "version": "0.0.18",
   "private": false,
   "publishConfig": {
     "access": "public"

package/src/AguaceroCore.js

@@ -69,11 +69,14 @@ export class AguaceroCore extends EventEmitter {
         this.customColormaps = options.customColormaps || {};
 
         const userLayerOptions = options.layerOptions || {};
+        // EDIT: Determine initial mode from options
+        const initialMode = userLayerOptions.mode || 'model';
         const initialVariable = userLayerOptions.variable || null;
 
         this.state = {
             model: userLayerOptions.model || 'gfs',
-            isMRMS: false,
+            // EDIT: Set isMRMS based on the initial mode
+            isMRMS: initialMode === 'mrms',
             mrmsTimestamp: null,
             variable: initialVariable,
             date: null,
@@ -135,15 +138,36 @@ export class AguaceroCore extends EventEmitter {
         await this.fetchModelStatus(true);
         await this.fetchMRMSStatus(true);
         
-        const latestRun = findLatestModelRun(this.modelStatus, this.state.model);
-        let initialState = this.state;
+        let initialState = { ...this.state };
 
-        if (latestRun && !this.state.isMRMS) {
-            initialState = { ...this.state, ...latestRun, forecastHour: 0 };
-
-            const availableVariables = this.getAvailableVariables(initialState.model);
-            if (availableVariables && availableVariables.length > 0) {
-                initialState.variable = availableVariables[0];
+        // ADD: Logic to handle an initial MRMS state
+        if (initialState.isMRMS) {
+            const variable = initialState.variable;
+            if (variable && this.mrmsStatus && this.mrmsStatus[variable]) {
+                const sortedTimestamps = [...(this.mrmsStatus[variable] || [])].sort((a, b) => b - a);
+                initialState.mrmsTimestamp = sortedTimestamps.length > 0 ? sortedTimestamps[0] : null;
+            } else {
+                // Fallback if the provided variable is not valid
+                console.warn(`Initial MRMS variable '${variable}' not found. Using default.`);
+                const availableMRMSVars = this.getAvailableVariables('mrms');
+                if (availableMRMSVars.length > 0) {
+                    const firstVar = availableMRMSVars[0];
+                    initialState.variable = firstVar;
+                    const sortedTimestamps = [...(this.mrmsStatus[firstVar] || [])].sort((a, b) => b - a);
+                    initialState.mrmsTimestamp = sortedTimestamps.length > 0 ? sortedTimestamps[0] : null;
+                }
+            }
+        } else {
+            // EDIT: This is the existing logic, now in an else block
+            const latestRun = findLatestModelRun(this.modelStatus, initialState.model);
+            if (latestRun) {
+                initialState = { ...initialState, ...latestRun, forecastHour: 0 };
+                if (!initialState.variable) {
+                    const availableVariables = this.getAvailableVariables(initialState.model);
+                    if (availableVariables && availableVariables.length > 0) {
+                        initialState.variable = availableVariables[0];
+                    }
+                }
             }
         }