CC-Gestion stock

--[[Gestion base 1.0 : Ce script permet de faire un suivis des stocks d'une base (tank, chest et barrels) avec une station principale et des secondaires. Pour les chest et barrels il faut utiliser "peripheral proxy".--]]
-- Below are values you can change
local isMain = true; -- base principale (qui va recevoire les information sdes autres bases
local myid = os.computerLabel()
local id = myid -- "Base principale"; -- nom de la base
local UpdateIntervalSeconds = 10; -- reduire le temps si server trop charge
local ItemsFullPercentAt = 256; -- stock minimum d'objet (barre %)
local FluidFullPercentAt = 64000; -- stock minimum de liquide (barre %)

local VersionInfo = "Gestion stock"; -- nom de la version

local titleTextColor = colors.blue;
local titleBackColor = colors.white;
local tankTextColor = colors.black;
local tankBackColor = colors.lime;
local chestTextColor = colors.white; -- couleur text (chest)
local chestBackColor = colors.purple; -- couleur fond (chest)
local cacheTextColor = colors.white;
local cacheBackColor = colors.cyan;
local powerTextColor = colors.black;
local powerBackColor = colors.orange;

local blacklist = {"xxxx"}; -- element a exclure de l'affichage ex: "iron" n'affichera pas Iron ore, Iron Block, Iron Helmet, ...
local whitelist = {""}; -- ajouter les therme a afficher. si whitelist = "" pour desactiver

local NameLen = 18;
local mainChannel = 2;

-- Above are values you can change
print("Demmarrage "..VersionInfo);
local peripherals = peripheral.getNames();
local mon;
local wmod;
local x,y;
local CurColumn = 0;
local MaxColumn;
local ColumnWidth;
local CurLine = 2;
local ContentData = {};

function padString (sText, iLen) -- longueur des text
    local iTextLen = string.len(sText);
    -- Too short, pad
    if (iTextLen < iLen) then
        local iDiff = iLen - iTextLen;
        return(sText..string.rep(" ",iDiff));
    end
    -- Too long, trim
    if (iTextLen > iLen) then
        return(string.sub(sText,1,iLen));
    end
    -- Exact length
    return(sText);
end

function prepmonitor(objectmon) -- preparation du moniteur
    mon = peripheral.wrap(objectmon);
    if (mon.isColor() == false) then
        titleTextColor = colors.black;
        titleBackColor = colors.white;
        tankTextColor = colors.black;
        tankBackColor = colors.white;
        chestTextColor = colors.black;
        chestBackColor = colors.white;
        cacheTextColor = colors.black;
        cacheBackColor = colors.white;
        powerTextColor = colors.black;
        powerBackColor = colors.white;
    end
end

function updateTable(strSource,strName,strAmount,timestamp,strLegend) -- gestion des la table et affichage information
    local isWhitelisted = true;
    if (type(whitelist) == "table") then
        isWhitelisted = false;
        for l,filter in pairs(whitelist) do
            if (string.find(string.lower(strName),string.lower(filter)) ~= nil) then
                isWhitelisted = true;
            end
        end
    end
    if (isWhitelisted == false) then
        return;
    end
    if (type(blacklist) == "table") then
        for l,filter in pairs(blacklist) do
            if (string.find(string.lower(strName),string.lower(filter)) ~= nil) then
                return;
            end
        end
    end
    if (ContentData[strSource] == nil) then
        ContentData[strSource] = {};
    end
    if (ContentData[strSource][timestamp] == nil) then
        ContentData[strSource][timestamp] = {};
    end
    if (ContentData[strSource][timestamp][strName] == nil) then
        ContentData[strSource][timestamp][strName] = {};
    end
    if (ContentData[strSource][timestamp][strName]["count"] == nil) then
        ContentData[strSource][timestamp][strName]["count"] = strAmount;
    else
        ContentData[strSource][timestamp][strName]["count"] = ContentData[strSource][timestamp][strName]["count"] + strAmount;
    end
    ContentData[strSource][timestamp][strName]["legend"] = strLegend;
end

function printmon(strName,strAmount,strMax,strLegend)
    local textColor;
    local backColor;
    local FullPercentAt = ItemsFullPercentAt;
    if (strLegend == "#") then
        textColor = chestTextColor;
        backColor = chestBackColor;
        strLegend = "";
    end
    if (strLegend == "+") then
        textColor = tankTextColor;
        backColor = tankBackColor;
        FullPercentAt = FluidFullPercentAt;
        strLegend = "";
    end
    if (strLegend == "*") then
        textColor = powerTextColor;
        backColor = powerBackColor;
        strLegend = "";
    end
    if (strLegend == "$") then
        textColor = cacheTextColor;
        backColor = cacheBackColor;
        strLegend = "";
    end
    local line = string.format("%s  %3i%s",padString(strName,NameLen+1),strAmount,padString(strLegend,1));
    if (strAmount >= 1000000) then
        line = string.format("%s  %3iM%s",padString(strName,NameLen),math.floor(strAmount/1000000),padString(strLegend,1));
    elseif (strAmount >= 1000) then
        line = string.format("%s  %3iK%s",padString(strName,NameLen),math.floor(strAmount/1000),padString(strLegend,1));
    end

    local ColPadding = 0;
    if (CurColumn > 0) then
        ColPadding = 1;
    end
    local CurX = math.floor((CurColumn*ColumnWidth))+math.floor(CurColumn*ColPadding)+1;
    if (CurColumn == 0) then
        --  print("CurX:"..CurX);
    end
    mon.setCursorPos(CurX,CurLine);
    --local percent = strAmount / strMax * 100;
    mon.setBackgroundColor(backColor);
    local percent = strAmount / FullPercentAt * 100;
    if (percent > 100) then percent = 100; end
    local barlength = math.floor(percent / 100 * (string.len(line)-1));
    --if (CurColumn == 0) then
    --  barlength = barlength + 1;
    --end

    if (string.len(line) > barlength) then
        local msg = string.sub(line,1,barlength);
        mon.setTextColor(textColor);
        mon.write(msg);
        --[[if (percent == 0) then
            mon.setBackgroundColor();
        else
            mon.setBackgroundColor(colors.black);
        end--]]
        mon.setBackgroundColor(colors.black);
        mon.setTextColor(backColor);
        mon.write(string.sub(line,barlength+1,-2))
    else
        local spaces = barlength - string.len(line);
        mon.write(line);
        mon.write(string.rep(" ",spaces));
    end

    mon.setTextColor(colors.white);
    CurColumn = CurColumn + 1;
    if (CurColumn > MaxColumn) then
        CurColumn = 0;
        CurLine = CurLine + 1;
    end
    return true;
end

function findMonitor() -- recherche moniteur
    for i,name in pairs(peripherals) do
        for j,method in pairs(peripheral.getMethods(name)) do
            if (method == 'getCursorPos') then
                prepmonitor(name);
            end
        end
    end
end

function findWirelessModem() -- recherche modem
    local foundWireless = false;
    for i,name in pairs(peripherals) do
        for j,method in pairs(peripheral.getMethods(name)) do
            if (method == 'isWireless') then
                wmod = peripheral.wrap(name);
                if (wmod.isWireless()) then
                    wmod.closeAll();
                    foundWireless = true;
                    break;
                else
                    wmod = {};
                end
            end
        end
        if (foundWireless) then
            break;
        end
    end
end

function collectLocalInfo()
    local timestamp = os.clock();
    for i,name in pairs(peripherals) do
        local p = peripheral.wrap(name);
        local displayNames = {};
        if (p.getTankInfo ~= nil) then -- RECHERCHE TANK
            print("Processing Tank");
            local iteminfo = p.getTankInfo();
            if (iteminfo ~= nil) then
                local displayname = "Empty";
                local amount = 0;
                if (iteminfo[1].contents) then
                    displayname = iteminfo[1].contents.rawName;
                    amount = iteminfo[1].contents.amount;
                    amount = math.floor(amount/1000);
                end
                if (amount ~= 0) then
                    updateTable(id,displayname,amount,timestamp,"+");
                end
            end
        end
        if (p.getStoredItems ~= nil) then -- RECHERCHE STOCKAGE ITEM (barrel)
            print("Processing Cache");
            local iteminfo = p.getStoredItems();
            if (iteminfo) then
                local displayname = iteminfo.display_name;
                updateTable(id,displayname,iteminfo.qty,timestamp,"$");
            end
        end
        if (p.getEnergyStored ~= nil) then -- RECHERCHE STOCAGE ENERGY
            print("Processing Energy");
            local energy = p.getEnergyStored();
            if (energy ~= nil) then
                updateTable(id,"Energy",energy,timestamp,"*");
            end
        end
        if (p.getInventorySize ~= nil) then -- RECHERCHE STOCKAGE ITEM (chest)
            print("Processing Chest");
            local chestSize = p.getInventorySize();
            if (chestSize ~= nil) then
                local items = {};
                for j=1,chestSize,1 do
                    local iteminfo = p.getStackInSlot(j);
                    if (iteminfo) then
                        displayname = iteminfo.display_name;
                        if (displayname) then
                            if (not items[displayname]) then
                                items[displayname] = iteminfo.qty;
                            else
                                items[displayname] = items[displayname] + iteminfo.qty;
                            end
                        end
                    end
                end
                local k = 0;
                for key,val in pairs(items) do
                    k = k + 1;
                    updateTable(id,key,val,timestamp,"#");
                end
            end
        end
        if (p.getAvailableItems ~= nil) then -- RECHERCHE DONNEES ME / APPLIED ENERGETIC
            print("Processing ME System");
            local itemarray = p.getAvailableItems();
            if (itemarray ~= nil) then
                for j=1,table.getn(itemarray),1 do
                    if (itemarray[j].size > 0) then
                        local fingerprint = itemarray[j].fingerprint;
                        if ( (displayNames[fingerprint.id] == nil) or (displayNames[fingerprint.id] == fingerprint.id)) then
                            local ItemDetail = p.getItemDetail(fingerprint);
                            if (ItemDetail ~= nil) then
                                displayNames[fingerprint.id] = ItemDetail.basic().display_name;
                            else
                                displayNames[fingerprint.id] = fingerprint.id;
                            end
                        end
                        updateTable(id,displayNames[fingerprint.id],itemarray[j].size,timestamp,"#");
                    end
                end
            end
        end
        if (p.listItems ~= nil) then -- RECHERCHE BRIDGE ME
            print("Processing ME Bridge");
            local itemarray = p.listItems();
            if (itemarray ~= nil) then
                for j=1,table.getn(itemarray),1 do
                    if (itemarray[j].amount > 0) then
                        local fingerprint = itemarray[j].name;
                        local displayName = itemarray[j].displayName;
                        updateTable(id,displayName,itemarray[j].amount,timestamp,"#");
                    end
                end
            end
        end
        print("Done");
    end
end

function updateMonitor() -- mise a jour moniteur
    x,y = mon.getSize();
    ColumnWidth = NameLen + 7;
    MaxColumn = math.floor(x / (ColumnWidth))-1;
    mon.setBackgroundColor(colors.black);
    mon.clear();
    CurColumn = 0;
    CurLine = 1;
    mon.setCursorPos(1,1);
    mon.setTextColor(colors.white);
    mon.setTextScale(0.5);
    mon.write(VersionInfo);
    -- Tri par nom de base
    local sortedSources = {};
    for n in pairs(ContentData) do
        table.insert(sortedSources, n);
    end
    table.sort(sortedSources);
    local name = "";
    for i,source in ipairs(sortedSources) do
        if (name ~= source) then
            name = source;
            CurColumn = 0;
            mon.setTextColor(titleTextColor);
            mon.setBackgroundColor(titleBackColor);
            mon.setCursorPos(1,CurLine+1);
            mon.write(padString("Stock de "..name,x-1));
            CurLine = CurLine + 2;
        end
        sortedTimestamps = {};
        for timestamp in pairs(ContentData[source]) do
            table.insert(sortedTimestamps,timestamp);
        end
        table.sort(sortedTimestamps);
        latest = nil;
        for j=table.maxn(sortedTimestamps),1,-1 do
            if (latest == nil) then
                latest = sortedTimestamps[j];
            else
                ContentData[source][sortedTimestamps[j]] = nil;
            end
        end
        for itemname in pairs(ContentData[source][latest]) do
            displayname = itemname;
            count = ContentData[source][latest][itemname]["count"];
            legend = ContentData[source][latest][itemname]["legend"];
            max = 0;
            printmon(displayname,count,max,legend);
        end
    end
end

-----------------------------------------------PROGRAM---------------------------------------------------
findMonitor();
findWirelessModem();

if (isMain == true) then
    if (type(wmod.isWireless) == "function") then
        wmod.open(mainChannel);
    else
        print("Pas de modem sur le PC principal");
    end
end

ContentData = {};

local timerUpdate = os.startTimer(UpdateIntervalSeconds);
local updateCount = 0;
local wirelessEventCount = 0;
-- Perform Initial Collection and Update the Monitor
collectLocalInfo();
updateMonitor();

-- Main program loop
while true do
    local event, param1, param2, param3, param4, param5 = os.pullEvent();
    print("Event recus :"..event);
    if (event == "timer") then
        if (param1 == timerUpdate) then
            collectLocalInfo();
            if (wmod) then
                if (isMain == false) then
                    wmod.transmit(mainChannel,1,ContentData);
                end
            end
            updateMonitor();
            wirelessEventCount = 0;
            timerUpdate = os.startTimer(UpdateIntervalSeconds);
        end
    end
    if (event == "modem_message") then
        if (isMain == true) then
            wirelessEventCount = wirelessEventCount + 1;
            for source,data in pairs(param4) do
                if (ContentData[source] == nil) then
                    ContentData[source] = {};
                end
                ContentData[source] = param4[source];
            end
            if (wirelessEventCount >= 10) then
                timerUpdate = os.startTimer(1);
            end
        end
    end
    if (event == "monitor_touch") or (event == "monitor_resize") then
        print("Updating the Monitor");
        updateMonitor();
    end
    if (event == "peripheral") or (event == "peripheral_detach") then
        print("Updating the peripheral list");
        peripherals = peripheral.getNames();
    end
end