StateController.js

export default class StateController {
    _state = {};
    get state() {
      return this._state;
    }
  
    set state(value) {
      this._state = value;
    }
  
    _traversePath(path, current = this._state) {
      const keys = path.split(".");
      for (let i = 0; i < keys.length - 1; i++) {
        const key = keys[i];
        const nextKey = keys[i + 1];
        if (!(key in current) || typeof current[key] !== "object" || current[key] === null) {
          // Create array if nextKey is a number, else object
          current[key] = !isNaN(Number(nextKey)) ? [] : {};
        }
        current = current[key];
      }
      return { current, lastKey: keys[keys.length - 1] };
    }
  
    getKey(key) {
      const { current, lastKey } = this._traversePath(key,  this._state);
      // console.log("getKey", current, lastKey);
      return current && lastKey in current ? current[lastKey] : undefined;
    }
  
    setKey(key, value) {
      const { current, lastKey } = this._traversePath(key, this._state);
      // console.log(current, lastKey);
      current[lastKey] = value;
    }
  
    setState(updates) {
      if (typeof updates !== "object" || updates === null) {
        throw new Error("setState requires an object as argument");
      }
      for (const [key, value] of Object.entries(updates)) {
        this.setKey(key, value);
      }
    }
  
    getState() {
      return this._state;
    }

    // Utility methods
    hasKey(key) {
      const { current, lastKey } = this._traversePath(key);
      return current ? lastKey in current : false;
    }
  
    removeKey(key) {
      const { current, lastKey } = this._traversePath(key);
      if (current && lastKey) {
        delete current[lastKey];
      }
    }
  
    getKeys(key) {
      const { current, lastKey } = this._traversePath(key);
      if (!current || typeof current !== 'object' || current === null) {
        return [];
      }
      return Object.keys(current[lastKey]);
    }
  
    getType(key) {
      const value = this.getKey(key);
      if (value === undefined) return "undefined";
      if (value === null) return "null";
      if (Array.isArray(value)) return "array";
      return typeof value;
    }
  
    // Array operations
    push(key, value) {
      const { current, lastKey } = this._traversePath(key);
      if (current === undefined || lastKey === null) return;
      if (!Array.isArray(current[lastKey])) {
        current[lastKey] = [];
      }
      current[lastKey].push(value);
      return current[lastKey].length;
    }
  
    pop(key) {
      const { current, lastKey } = this._traversePath(key);
      if (current && lastKey && Array.isArray(current[lastKey])) {
        return current[lastKey].pop();
      }
      return undefined;
    }
  
    length(key) {
      const { current, lastKey } = this._traversePath(key);
      if (current && lastKey && Array.isArray(current[lastKey])) {
        return current[lastKey].length;
      }
      return 0;
    }
  
    // Object operations
    merge(key, value) {
      const { current, lastKey } = this._traversePath(key);
      if (current === undefined || lastKey === null) return;
      if (typeof value !== "object" || value === null) {
        current[lastKey] = value;
      } else {
        current[lastKey] = { ...current[lastKey], ...value };
      }
    }
  }
  
export { StateController };