Shop_Server

debug = true
authTime = 100 --This might work better if it's changed
mSide = "top" -- modem side
scrx,scry =term.getSize()
cur = term.current()
local oPull = os.pullEvent
os.pullEvent = os.pullEventRaw
doExits = true --Maybe change this.
tChan = 1337 --Channel for turtle communication, should probably be changed
cChan = 31337 --Channel for client communication, also should be changed.
sAddr = "kvse3lmart" --Address to send to here
toAddr = "klukereedk" --Profits go here
iWindowX = 0
iWindowY = 0
iWindowZ = 0

if not fs.exists('dateApi') then
    if fs.exists("dateApi") then
        fs.delete("dateApi")
    end
    print("Installing Date API")
    shell.run("pastebin","get","21kyJN5A","dateApi")
    os.loadAPI("dateApi")
else
    os.loadAPI("dateApi")
end

if fs.exists("/authKey") then
    if debug then
        print("Found auth file")
    end
    f = fs.open("/authKey","r")
    authKey = f.readAll()
    f.close()
else
    error("authKey file does not exist.")
end

if fs.exists("/pkey") then
    f = fs.open("/pkey","r")
    pKey = f.readAll()
    f.close()
else
    error("No private key file, put the master/private key (The hash ending with -000) for the store address in '/pkey'")
end

if not fs.isDir("/coupons") then
    if fs.exists("/coupons") then
        fs.delete("/coupons")
    end
    fs.makeDir("/coupons")
end

if not fs.exists("/stock") then
    f = fs.open("/stock","w")
    f.write(textutils.serialize({}))
    f.close()
end

local MOD = 2^32
local MODM = MOD-1

local function memoize(f)
        local mt = {}
        local t = setmetatable({}, mt)
        function mt:__index(k)
                local v = f(k)
                t[k] = v
                return v
        end
        return t
end

local function make_bitop_uncached(t, m)
        local function bitop(a, b)
                local res,p = 0,1
                while a ~= 0 and b ~= 0 do
                        local am, bm = a % m, b % m
                        res = res + t[am][bm] * p
                        a = (a - am) / m
                        b = (b - bm) / m
                        p = p*m
                end
                res = res + (a + b) * p
                return res
        end
        return bitop
end

local function make_bitop(t)
        local op1 = make_bitop_uncached(t,2^1)
        local op2 = memoize(function(a) return memoize(function(b) return op1(a, b) end) end)
        return make_bitop_uncached(op2, 2 ^ (t.n or 1))
end

local bxor1 = make_bitop({[0] = {[0] = 0,[1] = 1}, [1] = {[0] = 1, [1] = 0}, n = 4})

local function bxor(a, b, c, ...)
        local z = nil
        if b then
                a = a % MOD
                b = b % MOD
                z = bxor1(a, b)
                if c then z = bxor(z, c, ...) end
                return z
        elseif a then return a % MOD
        else return 0 end
end

local function band(a, b, c, ...)
        local z
        if b then
                a = a % MOD
                b = b % MOD
                z = ((a + b) - bxor1(a,b)) / 2
                if c then z = bit32_band(z, c, ...) end
                return z
        elseif a then return a % MOD
        else return MODM end
end

local function bnot(x) return (-1 - x) % MOD end

local function rshift1(a, disp)
        if disp < 0 then return lshift(a,-disp) end
        return math.floor(a % 2 ^ 32 / 2 ^ disp)
end

local function rshift(x, disp)
        if disp > 31 or disp < -31 then return 0 end
        return rshift1(x % MOD, disp)
end

local function lshift(a, disp)
        if disp < 0 then return rshift(a,-disp) end
        return (a * 2 ^ disp) % 2 ^ 32
end

local function rrotate(x, disp)
    x = x % MOD
    disp = disp % 32
    local low = band(x, 2 ^ disp - 1)
    return rshift(x, disp) + lshift(low, 32 - disp)
end

local k = {
        0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5,
        0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5,
        0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3,
        0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174,
        0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc,
        0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da,
        0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7,
        0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967,
        0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13,
        0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85,
        0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3,
        0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070,
        0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5,
        0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
        0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208,
        0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2,
}

local function str2hexa(s)
        return (string.gsub(s, ".", function(c) return string.format("%02x", string.byte(c)) end))
end

local function num2s(l, n)
        local s = ""
        for i = 1, n do
                local rem = l % 256
                s = string.char(rem) .. s
                l = (l - rem) / 256
        end
        return s
end

local function s232num(s, i)
        local n = 0
        for i = i, i + 3 do n = n*256 + string.byte(s, i) end
        return n
end

local function preproc(msg, len)
        local extra = 64 - ((len + 9) % 64)
        len = num2s(8 * len, 8)
        msg = msg .. "\128" .. string.rep("\0", extra) .. len
        assert(#msg % 64 == 0)
        return msg
end

local function initH256(H)
        H[1] = 0x6a09e667
        H[2] = 0xbb67ae85
        H[3] = 0x3c6ef372
        H[4] = 0xa54ff53a
        H[5] = 0x510e527f
        H[6] = 0x9b05688c
        H[7] = 0x1f83d9ab
        H[8] = 0x5be0cd19
        return H
end

local function digestblock(msg, i, H)
        local w = {}
        for j = 1, 16 do w[j] = s232num(msg, i + (j - 1)*4) end
        for j = 17, 64 do
                local v = w[j - 15]
                local s0 = bxor(rrotate(v, 7), rrotate(v, 18), rshift(v, 3))
                v = w[j - 2]
                w[j] = w[j - 16] + s0 + w[j - 7] + bxor(rrotate(v, 17), rrotate(v, 19), rshift(v, 10))
        end

        local a, b, c, d, e, f, g, h = H[1], H[2], H[3], H[4], H[5], H[6], H[7], H[8]
        for i = 1, 64 do
                local s0 = bxor(rrotate(a, 2), rrotate(a, 13), rrotate(a, 22))
                local maj = bxor(band(a, b), band(a, c), band(b, c))
                local t2 = s0 + maj
                local s1 = bxor(rrotate(e, 6), rrotate(e, 11), rrotate(e, 25))
                local ch = bxor (band(e, f), band(bnot(e), g))
                local t1 = h + s1 + ch + k[i] + w[i]
                h, g, f, e, d, c, b, a = g, f, e, d + t1, c, b, a, t1 + t2
        end

        H[1] = band(H[1] + a)
        H[2] = band(H[2] + b)
        H[3] = band(H[3] + c)
        H[4] = band(H[4] + d)
        H[5] = band(H[5] + e)
        H[6] = band(H[6] + f)
        H[7] = band(H[7] + g)
        H[8] = band(H[8] + h)
end

local function sha256(msg)
        msg = preproc(msg, #msg)
        local H = initH256({})
        for i = 1, #msg, 64 do digestblock(msg, i, H) end
        return str2hexa(num2s(H[1], 4) .. num2s(H[2], 4) .. num2s(H[3], 4) .. num2s(H[4], 4) ..
                num2s(H[5], 4) .. num2s(H[6], 4) .. num2s(H[7], 4) .. num2s(H[8], 4))
end

function readStock()
    f = fs.open("/stock","r")
    fc = textutils.unserialize(f.readAll())
    f.close()
    return fc
end

function updateStock(nFile)
    f = fs.open("/stock","w")
    f.write(textutils.serialize(nFile))
    f.close()
    return true
end


function getData(id,name)
    sFile = readStock()
    if not sFile[id] then
        return false
    end
    if not sFile[id][name] then
        return false
    end
    return sFile[id][name]
end

function setData(id,name,value)
    sFile = readStock()
    if not sFile[id] then
        return false --The entry needs to exist
    end
    sFile[id][name] = value
    updateStock(sFile)
    return true
end

function getEntry(id)
    sFile = readStock()
    if not sFile[id] then
        return false
    end
    return sFile[id]
end

function addEntry(tValues)
    sFile = readStock()
    table.insert(sFile,tValues)
    updateStock(sFile)
    return #sFile
end

function addLog(toAdd)
    term.redirect(logs)
    cTime = textutils.formatTime(os.time(),false)
    d,m,y = dateApi.genDate(os.day())
    date = m.."/"..d.."/"..y
    print("["..cTime.." "..date.."] "..toAdd)
    return true
end

function runCommand(cmd)
    cArgs = {}
    for w in cmd:gmatch("[^ ]+") do
        table.insert(cArgs,w)
    end
    if string.lower(cArgs[1]) == "echo" then
        if #cmd > 5 then
            addLog(cmd:sub(6,#cmd))
            return true
        else
            addLog("Usage: "..cArgs[1].." <What to echo>")
            return false
        end
    elseif string.lower(cArgs[1]) == "exit" then
        addLog("Exiting program.")
        term.redirect(cur)
        os.pullEvent = oPull
        runServ = false
    end
end

function doCheck(rec)
    --Got an OK/COMPLETE message
    curWait = httpWait
    if rec == "OK" then
        checkOK = "COMPLETE"
        oCheck = toCheck
        toCheck = nil --Wait for a complete
    elseif rec == "COMPLETE" then
        if oCheck[1] == "get_chest" then
            checkOK = "OK"
            toCheck = {"deliver_item",iWindowX,iWindowY,iWindowZ}
        else
            checkOK = false
        end
    end
end

function genValid(time) --Generate something to detect if the turtle is legit.
    if not time then
        time = os.time()
    end
    tHash = authKey.." "..tostring(math.floor(math.floor(time() * authTime) / (authTime / 10)))
    vKey = sha256(tHash)
    return vKey:sub(1,10) --Just because
end

modem = peripheral.wrap(mSide)
modem.open(tChan)
modem.open(cChan)

ce = ""
sce = 1
txtLog = {}
tls = 0
TID = os.startTimer(2.5) --Delay between checking recent transactions
stime = os.time() * 1000 --Measure time in ticks
hst = os.time() * 1000
cvs = {}
checkOK = false
nBal = 0
httpWait = 2.5
curWait = 2.5
logs = window.create(cur, 1, 1,scrx,scry-1,true)
rTxt = window.create(cur,1,scry,scrx,1,true)
term.redirect(rTxt)
term.setCursorBlink(true)
runServ = true

while runServ do
    --Event 1, Side 2, Channel 3, Reply Channel 4, Message 5, Distance 6
    evnt = {os.pullEvent()}
    etime = os.time() * 1000
    te = etime - stime
    tes = te / 20
    htes = (((os.time() * 1000) - hst) / 20)
    if htes >= httpWait then
        suc = http.request("http://65.26.252.225/krist/?getbalance="..sAddr)
        hst = os.time() * 1000
    end
    if tes >= curWait then --Check if curWait seconds have passed.
        if checkOK then
            if toCheck then
                toSend = genValid(os.time()).." "
                for i = 1, #toCheck do
                    toSend = toSend..toCheck[i].." "
                end
                toSend = toSend:sub(1,#toSend-1)
                table.insert(cvs,1,toSend:sub(1,10))
                modem.transmit(tChan, 10000, toSend)
            end
        end
        TID = os.startTimer(curWait) --Reboot the timer,so we get events
        stime = os.time() * 1000 --Reboot the measuring system
    end
    if evnt[1] == "http_success" then
        cBal = tonumber(evnt[3].readAll())
        if cBal > 0 then
            if nBal then
                if cBal == nBal then
                    --Give items
                end
            else
                --Refund money
            end
        end
    elseif evnt[1] == "modem_message" then
        curTime = os.time()
        cKey = genValid(curTime)
        msg = evnt[5]
        if evnt[5]:sub(1,10) == cKey then
            --Received a message that may, or may not be a checkOK
            args = {}
            for w in msg:gmatch("[^ ]+") do
                table.insert(args,w)
            end
            if args[2] == checkOK then
                doCheck(args[2])
            end
        elseif checkOK then
            for i = 1, #cvs do
                if cvs[1] == evnt[5]:sub(1,10) then
                    args = {}
                    for w in msg:gmatch("[^ ]+") do
                        table.insert(args,w)
                    end
                    if args[2] == checkOK then
                        doCheck(args[2])
                    end
                end
            end
        elseif evnt[3] == cChan then
            --Client
        end
    elseif evnt[1] == "key" then
        if evnt[2] == keys.left then
            if sce > 1 then
                sce = sce-1
            end
            term.setCursorPos(sce,1)
        elseif evnt[2] == keys.right then
            if sce < #ce +1 then
                sce = sce+1
            end
            term.setCursorPos(sce,1)
        elseif evnt[2] == 199 then
            sce = 1
            term.setCursorPos(sce,1)
        elseif evnt[2] == 207 then
            sce = #ce +1
            term.setCursorPos(scr,1)
        elseif evnt[2] == keys.backspace then
            ce = ce:sub(0,sce-2) .. ce:sub(sce,#ce)
            sce = sce-1
            if sce < 1 then
                sce = 1
            end
            term.redirect(rTxt)
            term.clearLine()
            term.setCursorPos(1,1)
            write(ce)
            term.setCursorPos(sce,1)
        elseif evnt[2] == keys.enter then
            runCommand(ce)
            table.insert(txtLog,1,ce)
            sce = 1
            ce = ""
            tls = 0
            term.redirect(rTxt)
            term.clearLine()
            term.setCursorPos(1,1)
            write(ce)
            term.setCursorPos(sce,1)
        elseif evnt[2] == keys.up then
            if txtLog[tls+1] then
                tls = tls+1
                ce = txtLog[tls]
                sce = #ce +1
                term.redirect(rTxt)
                term.clearLine()
                term.setCursorPos(1,1)
                write(ce)
                term.setCursorPos(sce,1)
            end
        elseif evnt[2] == keys.down then
            if txtLog[tls-1] then
                tls = tls-1
                ce = txtLog[tls]
                sce = #ce +1
                term.redirect(rTxt)
                term.clearLine()
                term.setCursorPos(1,1)
                write(ce)
                term.setCursorPos(sce,1)
            else
                tls = 0
                ce = ""
                sce = 1
                term.redirect(rTxt)
                term.clearLine()
                term.setCursorPos(1,1)
                write(ce)
                term.setCursorPos(sce,1)
            end
        end
    elseif evnt[1] == "char" then
        ce = ce:sub(0,sce-1) .. evnt[2] .. ce:sub(sce,#ce)
        sce = sce+1
        term.redirect(rTxt)
        term.clearLine()
        term.setCursorPos(1,1)
        write(ce)
        term.setCursorPos(sce,1)
    elseif evnt[1] == "terminate" then
        if doExits then
            term.redirect(cur)
            os.pullEvent = oPull
            break
        end
    end
end