fluid

---------------------------------------------
--  Tank module for caftNanny
--  by demethan
--  www.breakfastcraft.com
--  www.craftnanny.org
--  2015 08 12  demethan:
--      -fixed modem support
--      -did some error magement
--      -added visual bar
---------------------------------------------

-- variables

local containers={}
local version = 2

local installer = "Q8ah3K9S"
local token = '0'
local module_name = ''
local username = ''
local type = ''

-- write text to the terminal screen
function draw_text_term(x, y, text, text_color, bg_color)
  term.setTextColor(text_color)
  term.setBackgroundColor(bg_color)
  term.setCursorPos(x,y)
  write(text)
end

-- draw a line on the terminal screen
function draw_line_term(x, y, length, color)
    term.setBackgroundColor(color)
    term.setCursorPos(x,y)
    term.write(string.rep(" ", length))
end

function bars()
    draw_line_term(1, 1, 51, colors.lime)
    draw_line_term(1, 19, 51, colors.lime)
    draw_text_term(15, 1, 'CraftNanny Fluid Module', colors.gray, colors.lime)
    draw_text_term(10, 19, 'www.craftnanny.org', colors.gray, colors.lime)
end

function terminal_screen()
    term.clear()

    bars()
    draw_text_term(1, 2, 'Module: ', colors.lime, colors.black)
    draw_text_term(10, 2, module_name, colors.white, colors.black)
    draw_text_term(1, 3, 'Owner: ', colors.lime, colors.black)
    draw_text_term(8, 3, username, colors.white, colors.black)
    draw_text_term(1, 4 , string.rep("-", 51), colors.lime, colors.black)

    --draw_text_term(2, 8, "I dont know what to put here...", colors.white, colors.black)
end

-- retrieves token from local text file
function load_config()
  sr = fs.open("config.txt", "r")
    token = sr.readLine()
    module_name = sr.readLine()
    username = sr.readLine()
    type = sr.readLine()
  sr.close()
end

-- called for new installations and when the scanner needs to be updated
function run_installer()
    if fs.exists("install") then
        fs.delete("install")
      end
      shell.run("pastebin get "..installer.." install")
      sleep(1)
      shell.run("install")
end


------  Start module specific code ---------


function phone_home(tank_name, fluid_type, percent)
    response = http.post("https://craftnanny.org/code/fluid.php",
                "token="..token.."&id="..os.getComputerID().."&tank_name="..tank_name.."&fluid_type="..fluid_type.."&percent="..percent)
    return_string = response.readAll()

    if tonumber(return_string) > version then
            run_installer()
    end
end


-- function findPeripheralSide()
--     -- Define the list of faces to check
--     local faces = {"left", "right", "bottom", "top", "back"}

--     -- Iterate through each face
--     for _, face in ipairs(faces) do
--         if peripheral.isPresent(face) then
--             return true, face  -- Return if a peripheral is found on this face
--         end
--     end

--     return false, ""  -- Return false if no peripheral is found
-- end

function round(num, decimals)
    local mult = 10^(decimals or 0)
    return math.floor(num * mult + 0.5) / mult
end

function capitalizeAndRemoveUnderscores(str)
    return str:gsub("([^_]+)([^_]*)", function(first, rest)
        first = first:gsub("^%l", string.upper) -- Capitalize the first letter of the first word
        rest = rest:gsub("^%l", string.upper)   -- Capitalize the first letter of the second word
        return first:gsub("_", "") .. rest:gsub("_", "") -- Concatenate the words after removing underscores
    end)
end



function getTankInformation(t, tankName)
    local tnk = peripheral.wrap(t)
    local okLiquid, msg = pcall(tnk.getInfo)

    if okLiquid then
        local amount = msg.amount
        local capacity = msg.capacity
        local fluid = tostring(msg.fluid):gsub("^[^:]*:", "")  -- Remove anything before the first colon
        fluid = capitalizeAndRemoveUnderscores(fluid) -- Capitalize the first letter
        local percent = round((amount / capacity * 100), 2)


        phone_home(tankName, fluid, percent)

        print(fluid, ":")
        local graphBar = round(((term.getSize() * percent) / 100), 0)
        if graphBar < 50 then
            draw_line_term(6, 7, graphBar , colors.green)
            draw_line_term(6 + graphBar, 7, term.getSize() - graphBar - 6, colors.red)
            draw_text_term(1, 7, percent .. " % ", colors.lime, colors.black)
            term.setBackgroundColor(colors.black)
        else
            draw_line_term(6, 7, graphBar - 6 , colors.green)
            draw_text_term(1, 7, percent .. " % ", colors.lime, colors.black)
            term.setBackgroundColor(colors.black)
        end
        return true
    else
        return false
    end
end


function start_loop()
    local side
    local tanks = peripheral.getNames()
    local faces = {"left", "right", "bottom", "top", "back"}

    -- Find the side where a peripheral is present
    for _, face in ipairs(faces) do
        if peripheral.isPresent(face) then
            side = face
            break
        end
    end

    -- -- If no peripheral is found, exit the loop
    -- if not side then
    --     print("No tank storage found")
    --     print("Please check your modems")
    --     return
    -- end

    while true do
        terminal_screen()

        if #tanks > 2 then
            print("Only one device is supported")
            break
        elseif #tanks == 2 then
            for _, tank in pairs(tanks) do
                if tank ~= side then
                    ok = getTankInformation(tank, tank)
                end
            end
        else
            ok = getTankInformation(side, "Tank"..os.getComputerID())
        end

        if not ok then
            print("No tank storage found")
            print("Do you have the right module?")
            print("Please check your modems")
            break
        end

        -- Main active status with server
        sleep(30)
    end
end


function start()
    term.clear()
    term.setCursorPos(1,1)

  if fs.exists("config.txt") then
      load_config()
      start_loop()
  else
      run_installer()
  end
end

start()