Halyde/halyde/kernel/modules/ipc.lua

local module = {}
module.dependencies = { "tsched" }

function module.check()
  return true -- IPC should always be loaded
end

local checkProcess
function module.init()
  _G.ipc = {}
  _G.ipc.shared = {}
  _PUBLIC.ipc = {}

  function _PUBLIC.ipc.shareWithAll()
    local shareTable = {}
    setmetatable(shareTable, {
      ["__newindex"] = function(_, key, value)
        local currentPID = _PUBLIC.tsched.getCurrentTask().id
        if not _G.ipc.shared[currentPID] then
          _G.ipc.shared[currentPID] = {}
        end
        local globalTable
        for _, tab in pairs(_G.ipc.shared[currentPID]) do
          if tab.sharedWith == "all" then
            globalTable = tab
          end
        end
        if not globalTable then
          globalTable = { ["sharedWith"] = "all" }
          table.insert(_G.ipc.shared[currentPID], globalTable)
        end
        if not globalTable.vars then
          globalTable.vars = {}
        end
        globalTable.vars[key] = value
      end,
      ["__index"] = function(_, key)
        local currentPID = _PUBLIC.tsched.getCurrentTask().id
        if not _G.ipc.shared[currentPID] then
          return nil
        end
        local globalTable
        for _, tab in pairs(_G.ipc.shared[currentPID]) do
          if tab.sharedWith == "all" then
            globalTable = tab
          end
        end
        if not globalTable then
          return nil
        end
        if not globalTable.vars then
          return nil
        end
        return globalTable.vars[key]
      end,
      ["__pairs"] = function()
        if not _G.ipc.shared[currentPID] then
          return pairs({})
        end
        local globalTable
        for _, tab in pairs(_G.ipc.shared[currentPID]) do
          if tab.sharedWith == pid then
            globalTable = tab
          end
        end
        if not globalTable then
          return pairs({})
        end
        if not globalTable.vars then
          return pairs({})
        end

        return pairs(table.copy(globalTable.vars))
      end,
    })
    return shareTable
  end

  function _PUBLIC.ipc.shareWith(pid)
    checkArg(1, pid, "number")
    local shareTable = {}
    setmetatable(shareTable, {
      ["__newindex"] = function(_, key, value)
        local currentPID = _PUBLIC.tsched.getCurrentTask().id
        if not _G.ipc.shared[currentPID] then
          _G.ipc.shared[currentPID] = {}
        end
        local globalTable
        for _, tab in pairs(_G.ipc.shared[currentPID]) do
          if tab.sharedWith == "all" then
            globalTable = tab
          end
        end
        if not globalTable then
          globalTable = { ["sharedWith"] = pid }
          table.insert(_G.ipc.shared[currentPID], globalTable)
        end
        if not globalTable.vars then
          globalTable.vars = {}
        end
        globalTable.vars[key] = value
      end,
      ["__index"] = function(_, key)
        print(_G.ipc.shared)

        local currentPID = _PUBLIC.tsched.getCurrentTask().id
        if not _G.ipc.shared[currentPID] then
          return nil
        end
        local globalTable
        for _, tab in pairs(_G.ipc.shared[currentPID]) do
          if tab.sharedWith == pid then
            globalTable = tab
          end
        end
        if not globalTable then
          return nil
        end
        if not globalTable.vars then
          return nil
        end
        return globalTable.vars[key]
      end,
      ["__pairs"] = function()
        if not _G.ipc.shared[currentPID] then
          return pairs({})
        end
        local globalTable
        for _, tab in pairs(_G.ipc.shared[currentPID]) do
          if tab.sharedWith == pid then
            globalTable = tab
          end
        end
        if not globalTable then
          return pairs({})
        end
        if not globalTable.vars then
          return pairs({})
        end

        return pairs(table.copy(globalTable.vars))
      end,
    })

    -- check if the reverse is also available
    --[[ if not _G.ipc.shared[pid] then
      _G.ipc.shared[pid]={}
    end
    for _, tab in pairs(_G.ipc.shared[pid]) do
      if tab.sharedWith == currentPID then
        return -- it's already added
      end
    end
    local reverseTable = {}
    reverseTable.vars = globalTable.vars
    reverseTable.sharedWith = currentPID
    table.insert(_G.ipc.shared[pid],reverseTable) ]]

    return shareTable
  end

  _PUBLIC.ipc.shared = {}
  setmetatable(_PUBLIC.ipc.shared, {
    ["__index"] = function(_, pid)
      local currentPID = _PUBLIC.tsched.getCurrentTask().id
      local returnTable = {}
      for _, shareTable in pairs(ipc.shared[pid] or {}) do
        if shareTable.sharedWith == currentPID then
          for key, value in pairs(shareTable.vars) do
            returnTable[key] = table.copy(value)
          end
        elseif shareTable.sharedWith == "all" then
          for key, value in pairs(shareTable.vars) do
            if not returnTable[key] then
              returnTable[key] = table.copy(value)
            end
          end
        end
      end
      return returnTable
    end,
    ["__pairs"] = function()
      local ftbl = {}
      for i in pairs(_G.ipc.shared) do
        ftbl[i] = _PUBLIC.ipc.shared[i]
      end
      return pairs(ftbl)
    end,
  })

  _, checkProcess = _PUBLIC.tsched.addTask(function()
    while true do
      -- get all PIDs that exists
      local tasks = _PUBLIC.tsched.getTasks()
      local pids = {}
      for _, v in pairs(tasks) do
        table.insert(pids, v.id)
      end

      -- get all shares from unexistant processes and delete them
      for i in pairs(_G.ipc.shared) do
        if not table.find(pids, i) then
          _G.ipc.shared[i] = nil
        end
      end

      -- let the other processes run
      coroutine.yield()
    end
  end, "ipc")
end

function module.exit()
  _G.ipc = nil
  _PUBLIC.ipc = nil
  _PUBLIC.tsched.removeTask(checkProcess.id)
end

return module