unLisp.h

Go to the documentation of this file.

/*
 * This is a part of the Uniot project.
 * Copyright (C) 2016-2023 Uniot <contact@uniot.io>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */


 
#pragma once
 
#include <Bytes.h>
#include <CBORObject.h>
#include <Common.h>
#include <EventListener.h>
#include <LimitedQueue.h>
#include <LispEvents.h>
#include <LispHelper.h>
#include <Logger.h>
#include <PrimitiveExpeditor.h>
#include <Singleton.h>
#include <TaskScheduler.h>
#include <libminilisp.h>
 
#ifndef UNIOT_LISP_HEAP
#if defined(ESP32)
#define UNIOT_LISP_HEAP 20480  // 20 KB for ESP32
#elif defined(ESP8266)
#define UNIOT_LISP_HEAP 8192  // 8 KB for ESP8266
#else
#define UNIOT_LISP_HEAP 4096  // 4 KB for other platforms
#endif
#endif
 
namespace uniot {
using namespace lisp;
 
class IncomingEventManager {
 public:
  struct IncomingEvent {
    int32_t value;
    int8_t errorCode;
 
    IncomingEvent() : value(0), errorCode(0) {}
    IncomingEvent(int32_t v, int8_t err = 0) : value(v), errorCode(err) {}
  };
 
  void pushEvent(const Bytes &eventData) {
    CBORObject eventObj(eventData);
    auto eventID = eventObj.getString("eventID");
    auto valueStr = eventObj.getValueAsString("value");
 
    if (eventID.isEmpty() || valueStr.isEmpty()) {
      return;  // Invalid event
    }
 
    auto value = valueStr.toInt();
    auto isNumber = value || valueStr == "0";
    if (!isNumber) {
      return;  // Invalid event
    }
 
    bool isNewEvent = !mIncomingEvents.exist(eventID);
    if (isNewEvent) {
      auto eventQueue = std::make_shared<EventQueue>();
      mIncomingEvents.put(eventID, eventQueue);
      UNIOT_LOG_DEBUG("created new event queue for '%s'", eventID.c_str());
    }
 
    auto eventQueue = mIncomingEvents.get(eventID);
    eventQueue->lastAccessed = millis();
 
    size_t queueSizeBefore = eventQueue->queue.size();
    eventQueue->queue.pushLimited(IncomingEvent(value));
    size_t queueSizeAfter = eventQueue->queue.size();
 
    if (queueSizeAfter > queueSizeBefore) {
      UNIOT_LOG_TRACE("pushed event '%s' with value '%d', queue size: %d", eventID.c_str(), value, queueSizeAfter);
    } else {
      UNIOT_LOG_WARN("event queue for '%s' is full (limit: %d), oldest event was dropped", eventID.c_str(), EVENTS_LIMIT);
    }
  }
 
  bool isEventAvailable(const String &eventID) {
    if (!mIncomingEvents.exist(eventID)) {
      return false;
    }
 
    auto eventQueue = mIncomingEvents.get(eventID);
    eventQueue->isUsedInScript = true;
    eventQueue->lastAccessed = millis();
 
    return eventQueue->queue.size() > 0;
  }
 
  IncomingEvent popEvent(const String &eventID) {
    const IncomingEvent emptyEvent(0, -1);
 
    if (!mIncomingEvents.exist(eventID)) {
      UNIOT_LOG_WARN("attempted to pop non-existent event '%s'", eventID.c_str());
      return emptyEvent;
    }
 
    auto eventQueue = mIncomingEvents.get(eventID);
    eventQueue->isUsedInScript = true;
    eventQueue->lastAccessed = millis();
 
    if (eventQueue->queue.size() == 0) {
      UNIOT_LOG_WARN("attempted to pop from empty event queue '%s'", eventID.c_str());
      return emptyEvent;
    }
 
    return eventQueue->queue.popLimited(emptyEvent);
  }
 
  void cleanupUnusedEvents() {
    unsigned long currentTime = millis();
    ClearQueue<String> keysToRemove;
    size_t totalEvents = 0;
    size_t unusedEvents = 0;
    size_t expiredEvents = 0;
    size_t totalQueuedItems = 0;
 
    mIncomingEvents.forEach([&](const Pair<String, SharedPointer<EventQueue>> &entry) {
      totalEvents++;
      auto eventQueue = entry.second;
      totalQueuedItems += eventQueue->queue.size();
      unsigned long timeSinceLastAccess = currentTime - eventQueue->lastAccessed;
 
      if (!eventQueue->isUsedInScript) {
        unusedEvents++;
        if (timeSinceLastAccess > EVENT_TTL_MS) {
          expiredEvents++;
          keysToRemove.push(entry.first);
          UNIOT_LOG_DEBUG("marking event '%s' for removal (unused, last accessed %lu ms ago)",
            entry.first.c_str(), timeSinceLastAccess);
        }
      }
    });
 
    // Remove expired events
    keysToRemove.forEach([this](const String &key) {
      mIncomingEvents.remove(key);
    });
 
    if (expiredEvents > 0) {
      UNIOT_LOG_INFO("cleaned up %d expired events (total: %d, unused: %d, queued items: %d)",
        expiredEvents, totalEvents, unusedEvents, totalQueuedItems);
    } else if (totalEvents > 0) {
      UNIOT_LOG_TRACE("no events to cleanup (total: %d, unused: %d, queued items: %d)", totalEvents, unusedEvents, totalQueuedItems);
    }
  }
 
  void clean() {
    mIncomingEvents.clean();
    UNIOT_LOG_DEBUG("cleared all incoming events");
  }
 
 private:
  struct EventQueue {
    LimitedQueue<IncomingEvent> queue;
    unsigned long lastAccessed;
    bool isUsedInScript;
 
    EventQueue() : lastAccessed(millis()), isUsedInScript(false) {
      queue.limit(EVENTS_LIMIT);
    }
  };
 
  Map<String, SharedPointer<EventQueue>> mIncomingEvents;
  static constexpr size_t EVENTS_LIMIT = 2;
  static constexpr unsigned long EVENT_TTL_MS = 30000;  // 30 seconds TTL
};

class unLisp : public CoreEventListener, public Singleton<unLisp> {
  friend class Singleton<unLisp>;
 
 public:
  TaskScheduler::TaskPtr getTask() {
    return mTaskLispEval;
  }
 
  TaskScheduler::TaskPtr getCleanupTask() {
    return mTaskEventCleanup;
  }

  bool isCreated() {
    return lisp_is_created();
  }

  bool taskIsRunning() {
    return mTaskLispEval->isAttached();
  }

  size_t memoryUsed() {
    return lisp_mem_used();
  }

  void runCode(const Bytes &data) {
    if (!data.size())
      return;
 
    mLastCode = data;
 
    mTaskLispEval->detach();
    _destroyMachine();
    _createMachine();
 
    auto code = mLastCode.terminate().c_str();
    UNIOT_LOG_DEBUG("eval: %s", code);
 
    _refreshIncomingEvents();
    lisp_eval(mLispRoot, mLispEnv, code);
    if (!mTaskLispEval->isAttached()) {
      _destroyMachine();
    }
  }

  unLisp *pushPrimitive(Primitive *primitive) {
    auto description = PrimitiveExpeditor::extractDescription(primitive);
    mUserPrimitives.push(MakePair(description.name, primitive));
    UNIOT_LOG_TRACE("primitive added: %s", description.name.c_str());
    UNIOT_LOG_TRACE("args count: %d", description.argsCount);
    UNIOT_LOG_TRACE("return type: %d", description.returnType);
    for (auto i = 0; i < description.argsCount; i++) {
      UNIOT_LOG_TRACE("arg %d: %d", i, description.argsTypes[i]);
    }
    return this;
  }

  void serializePrimitives(CBORObject &obj) {
    mUserPrimitives.forEach([&](Pair<const String &, Primitive *> holder) {
      auto description = PrimitiveExpeditor::extractDescription(holder.second);
      obj.putArray(holder.first.c_str())
        .append(description.returnType)
        .appendArray()
        .append(description.argsCount, reinterpret_cast<const uint8_t *>(description.argsTypes));
    });
  }

  const Bytes &getLastCode() {
    UNIOT_LOG_WARN_IF(!mLastCode.size(), "there is no last saved code");
    return mLastCode;
  }

  void cleanLastCode() {
    mLastCode.clean();
  }

  virtual void onEventReceived(unsigned int topic, int msg) override {
    if (topic == events::lisp::Topic::IN_LISP_EVENT) {
      if (msg == events::lisp::Msg::IN_NEW_EVENT) {
        CoreEventListener::receiveDataFromChannel(events::lisp::Channel::IN_EVENT, [&](unsigned int id, bool empty, Bytes data) {
          _pushIncomingEvent(data);
        });
        return;
      }
      return;
    }
  }
 
 private:
  unLisp() {
    CoreEventListener::listenToEvent(events::lisp::Topic::IN_LISP_EVENT);
 
    auto fnPrintOut = [](const char *msg, int size) {
      if (size > 0) {
        auto &instance = unLisp::getInstance();
        instance.sendDataToChannel(events::lisp::Channel::OUT_LISP, Bytes((uint8_t *)msg, size).terminate());
        instance.emitEvent(events::lisp::Topic::OUT_LISP_MSG, events::lisp::Msg::OUT_MSG_ADDED);
      }
      yield();
    };
 
    auto fnPrintLog = [](const char *msg, int size) {
      if (size > 0) {
        auto &instance = unLisp::getInstance();
        instance.sendDataToChannel(events::lisp::Channel::OUT_LISP_LOG, Bytes((uint8_t *)msg, size).terminate());
        instance.emitEvent(events::lisp::Topic::OUT_LISP_MSG, events::lisp::Msg::OUT_MSG_LOG);
      }
      yield();
    };
 
    auto fnPrintErr = [](const char *msg, int size) {
      auto &instance = unLisp::getInstance();
      instance.sendDataToChannel(events::lisp::Channel::OUT_LISP_ERR, Bytes((uint8_t *)msg, size).terminate());
      instance.emitEvent(events::lisp::Topic::OUT_LISP_MSG, events::lisp::Msg::OUT_MSG_ERROR);
 
      instance.mTaskLispEval->detach();
      instance._destroyMachine();
    };
 
    lisp_set_cycle_yield(yield);
    lisp_set_printers(fnPrintOut, fnPrintLog, fnPrintErr);
 
    _constructLispEnv();
 
    mTaskLispEval = TaskScheduler::make([this](SchedulerTask &self, short t) {
      auto root = mLispRoot;
      auto env = mLispEnv;
 
      DEFINE2(t_obj, t_pass);
      *t_pass = get_variable(root, env, "#t_pass");
      (*t_pass)->cdr->value = t;
      *t_obj = get_variable(root, env, "#t_obj")->cdr;
      safe_eval(root, env, t_obj);
 
      if (!t) {
        _destroyMachine();
      }
 
      // UNIOT_LOG_DEBUG("lisp machine running, mem used: %d", lisp_mem_used());
    });
 
    mTaskEventCleanup = TaskScheduler::make([this](SchedulerTask &self, short t) {
      mEventManager.cleanupUnusedEvents();
      yield();
    });
  }

  void _constructLispEnv() {
    mLispEnvConstructor[0] = NULL;
    mLispEnvConstructor[1] = NULL;
    mLispEnvConstructor[2] = ROOT_END;
    mLispRoot = mLispEnvConstructor;
    mLispEnv = (Obj **)(mLispEnvConstructor + 1);
  }

  void _createMachine() {
    lisp_create(UNIOT_LISP_HEAP);
 
    *mLispEnv = make_env(mLispRoot, &Nil, &Nil);
    define_constants(mLispRoot, mLispEnv);
    define_primitives(mLispRoot, mLispEnv);
 
    // Register special variables and built-in primitives
    add_constant(mLispRoot, mLispEnv, "#t_obj", &Nil);
    add_constant_int(mLispRoot, mLispEnv, "#t_pass", 0);
    add_primitive(mLispRoot, mLispEnv, "task", mPrimitiveTask);
    add_primitive(mLispRoot, mLispEnv, "is_event", mPrimitiveIsEventAvailable);
    add_primitive(mLispRoot, mLispEnv, "pop_event", mPrimitivePopEvent);
    add_primitive(mLispRoot, mLispEnv, "push_event", mPrimitivePushEvent);
 
    // Register all user-defined primitives
    mUserPrimitives.forEach([this](Pair<const String &, Primitive *> holder) {
      add_primitive(mLispRoot, mLispEnv, holder.first.c_str(), holder.second);
    });
 
    UNIOT_LOG_DEBUG("lisp machine created, mem used: %d", lisp_mem_used());
  }

  void _destroyMachine() {
    lisp_destroy();
  }

  void _refreshIncomingEvents() {
    mEventManager.clean();
    this->emitEvent(events::lisp::Topic::OUT_LISP_REQUEST, events::lisp::Msg::OUT_REFRESH_EVENTS);
  }

  void _pushIncomingEvent(const Bytes &eventData) {
    // mEventManager.cleanupUnusedEvents();
    mEventManager.pushEvent(eventData);
  }

  bool _isIncomingEventAvailable(const String &eventID) {
    return mEventManager.isEventAvailable(eventID);
  }

  IncomingEventManager::IncomingEvent _popIncomingEvent(const String &eventID) {
    return mEventManager.popEvent(eventID);
  }

  bool _pushOutgoingEvent(String eventID, int value) {
    CBORObject event;
    event.put("eventID", eventID.c_str());
    event.put("value", value);
 
    auto sent = this->sendDataToChannel(events::lisp::Channel::OUT_EVENT, event.build());
    this->emitEvent(events::lisp::Topic::OUT_LISP_EVENT, events::lisp::Msg::OUT_NEW_EVENT);
    return sent;
  }

  inline Object _primTask(Root root, VarObject env, VarObject list) {
    auto expeditor = PrimitiveExpeditor::describe("task", Lisp::Bool, 3, Lisp::Int, Lisp::Int, Lisp::Cell)
                       .init(root, env, list);
    expeditor.assertDescribedArgs();
 
    auto times = expeditor.getArgInt(0);
    auto ms = expeditor.getArgInt(1);
    auto obj = expeditor.getArg(2);
 
    DEFINE1(t_obj);
    *t_obj = get_variable(root, env, "#t_obj");
    (*t_obj)->cdr = obj;
 
    mTaskLispEval->attach(ms, times);
 
    return expeditor.makeBool(true);
  }

  inline Object _primIsEventAvailable(Root root, VarObject env, VarObject list) {
    auto expeditor = PrimitiveExpeditor::describe("is_event", Lisp::Bool, 1, Lisp::Symbol)
                       .init(root, env, list);
    expeditor.assertDescribedArgs();
 
    auto eventId = expeditor.getArgSymbol(0);
    auto available = _isIncomingEventAvailable(eventId);
 
    return expeditor.makeBool(available);
  }

  inline Object _primPopEvent(Root root, VarObject env, VarObject list) {
    auto expeditor = PrimitiveExpeditor::describe("pop_event", Lisp::Bool, 1, Lisp::Symbol)
                       .init(root, env, list);
    expeditor.assertDescribedArgs();
 
    auto eventId = expeditor.getArgSymbol(0);
    auto event = _popIncomingEvent(eventId);
 
    UNIOT_LOG_WARN_IF(event.errorCode, "error popping event '%s': %d", event.errorCode);
 
    return expeditor.makeInt(event.value);
  }

  inline Object _primPushEvent(Root root, VarObject env, VarObject list) {
    auto expeditor = PrimitiveExpeditor::describe("push_event", Lisp::Bool, 2, Lisp::Symbol, Lisp::BoolInt)
                       .init(root, env, list);
    expeditor.assertDescribedArgs();
 
    auto eventId = expeditor.getArgSymbol(0);
    auto value = expeditor.getArgInt(1);
 
    auto sent = _pushOutgoingEvent(eventId, value);
 
    return expeditor.makeBool(sent);
  }
 
  // Private member variables
  Bytes mLastCode;                       
  TaskScheduler::TaskPtr mTaskLispEval;  
  TaskScheduler::TaskPtr mTaskEventCleanup;
  ClearQueue<Pair<String, Primitive *>> mUserPrimitives;  

  const Primitive *mPrimitiveTask = [](Root root, VarObject env, VarObject list) { return getInstance()._primTask(root, env, list); };
  const Primitive *mPrimitiveIsEventAvailable = [](Root root, VarObject env, VarObject list) { return getInstance()._primIsEventAvailable(root, env, list); };
  const Primitive *mPrimitivePopEvent = [](Root root, VarObject env, VarObject list) { return getInstance()._primPopEvent(root, env, list); };
  const Primitive *mPrimitivePushEvent = [](Root root, VarObject env, VarObject list) { return getInstance()._primPushEvent(root, env, list); };
 
  void *mLispEnvConstructor[3];        
  Root mLispRoot;                      
  VarObject mLispEnv;                  
  IncomingEventManager mEventManager;  
};

}  // namespace uniot