forked from nalc/global_exchange
1339 lines
28 KiB
Lua
1339 lines
28 KiB
Lua
|
|
local insecure_env = ...
|
|
local sql = insecure_env.require("lsqlite3")
|
|
local exports = {}
|
|
|
|
local order_book_cache = (function(cache)
|
|
return function(ex_name)
|
|
local maybe_cache = cache[ex_name]
|
|
if not maybe_cache then
|
|
maybe_cache = {}
|
|
cache[ex_name] = maybe_cache
|
|
end
|
|
return maybe_cache
|
|
end
|
|
end)({})
|
|
|
|
local init_query = [=[
|
|
BEGIN TRANSACTION;
|
|
CREATE TABLE if not exists Credit
|
|
(
|
|
Owner TEXT PRIMARY KEY NOT NULL,
|
|
Balance INTEGER NOT NULL
|
|
);
|
|
|
|
CREATE TABLE if not exists Log
|
|
(
|
|
Id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
Recipient TEXT NOT NULL,
|
|
Time INTEGER NOT NULL,
|
|
Message TEXT NOT NULL
|
|
);
|
|
|
|
CREATE TABLE if not exists Orders
|
|
(
|
|
Id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
Poster TEXT NOT NULL,
|
|
Exchange TEXT NOT NULL,
|
|
Type TEXT NOT NULL CHECK(Type IN ("buy", "sell")),
|
|
Time INTEGER NOT NULL,
|
|
Item TEXT NOT NULL,
|
|
Wear INTEGER NOT NULL CHECK(Wear >= 0 AND Wear <= 65535),
|
|
Amount INTEGER NOT NULL CHECK(Amount > 0),
|
|
Rate INTEGER NOT NULL CHECK(Rate > 0)
|
|
);
|
|
|
|
CREATE TABLE if not exists Inbox
|
|
(
|
|
Id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
Recipient TEXT NOT NULL,
|
|
Item TEXT NOT NULL,
|
|
Wear INTEGER NOT NULL CHECK(Wear >= 0 AND Wear <= 65535),
|
|
Amount INTEGER NOT NULL CHECK(Amount > 0)
|
|
);
|
|
|
|
CREATE INDEX if not exists index_log
|
|
ON Log (Recipient, Time);
|
|
|
|
CREATE INDEX if not exists index_orders
|
|
ON Orders (Exchange, Type, Item, Rate, Wear, Time);
|
|
|
|
CREATE INDEX if not exists index_own_orders
|
|
ON Orders (Poster, Time);
|
|
|
|
CREATE INDEX if not exists index_inbox
|
|
ON Inbox (Recipient, Item, Wear);
|
|
|
|
CREATE VIEW if not exists distinct_items AS
|
|
SELECT DISTINCT Item FROM Orders;
|
|
|
|
CREATE VIEW if not exists market_summary AS
|
|
SELECT
|
|
distinct_items.Item,
|
|
(
|
|
SELECT sum(Orders.Amount) FROM Orders
|
|
WHERE Orders.Item = distinct_items.Item
|
|
AND Orders.Type = "buy"
|
|
),
|
|
(
|
|
SELECT max(Orders.Rate) FROM Orders
|
|
WHERE Orders.Item = distinct_items.Item
|
|
AND Orders.Type = "buy"
|
|
),
|
|
(
|
|
SELECT sum(Orders.Amount) FROM Orders
|
|
WHERE Orders.Item = distinct_items.Item
|
|
AND Orders.Type = "sell"
|
|
),
|
|
(
|
|
SELECT min(Orders.Rate) FROM Orders
|
|
WHERE Orders.Item = distinct_items.Item
|
|
AND Orders.Type = "sell"
|
|
)
|
|
FROM distinct_items;
|
|
|
|
|
|
END TRANSACTION;
|
|
]=]
|
|
|
|
local new_act_query = [=[
|
|
INSERT INTO Credit (Owner, Balance)
|
|
VALUES (:owner, :start_balance);
|
|
]=]
|
|
|
|
local get_balance_query = [[
|
|
SELECT Balance FROM Credit
|
|
WHERE Owner = ?;
|
|
]]
|
|
|
|
local set_balance_query = [[
|
|
UPDATE Credit
|
|
SET Balance = :new_balance
|
|
WHERE Owner = :p_name;
|
|
]]
|
|
|
|
local log_query = [[
|
|
INSERT INTO Log (Recipient, Time, Message)
|
|
VALUES(:recipient, :time, :message);
|
|
]]
|
|
|
|
local add_order_query = [=[
|
|
INSERT INTO Orders (Poster, Exchange, Type, Time, Item, Wear, Amount, Rate)
|
|
VALUES (:p_name, :ex_name, :order_type, :time, :item_name, :wear, :amount, :rate);
|
|
]=]
|
|
|
|
local del_order_query = [=[
|
|
DELETE FROM Orders
|
|
WHERE Id = ?;
|
|
]=]
|
|
|
|
local reduce_order_query = [=[
|
|
UPDATE Orders
|
|
SET Amount = Amount - :delta
|
|
WHERE Id = :id;
|
|
]=]
|
|
|
|
local get_order_query = [=[
|
|
SELECT * FROM Orders
|
|
WHERE Id = ?
|
|
]=]
|
|
|
|
local cancel_order_query = [=[
|
|
DELETE FROM Orders
|
|
WHERE Id = :id
|
|
AND Poster = :p_name
|
|
]=]
|
|
|
|
local search_bids_query = [=[
|
|
SELECT * FROM Orders
|
|
WHERE Exchange = :ex_name
|
|
AND Type = "buy"
|
|
AND Item = :item_name
|
|
ORDER BY Rate ASC, Wear DESC;
|
|
]=]
|
|
|
|
local search_asks_query = [=[
|
|
SELECT * FROM Orders
|
|
WHERE Exchange = :ex_name
|
|
AND Type = "sell"
|
|
AND Item = :item_name
|
|
ORDER BY Rate DESC, Wear ASC;
|
|
]=]
|
|
|
|
local qual_bids_query = [=[
|
|
SELECT * FROM Orders
|
|
WHERE Exchange = :ex_name
|
|
AND Type = "buy"
|
|
AND Item = :item_name
|
|
AND Rate >= :rate_min
|
|
AND Wear >= :wear_min
|
|
ORDER BY Rate DESC, Time ASC;
|
|
]=]
|
|
|
|
local qual_asks_query = [=[
|
|
SELECT * FROM Orders
|
|
WHERE Exchange = :ex_name
|
|
AND Type = "sell"
|
|
AND Item = :item_name
|
|
AND Rate <= :rate_max
|
|
AND Wear <= :wear_max
|
|
ORDER BY Rate ASC, Time ASC;
|
|
]=]
|
|
|
|
local search_own_query = [=[
|
|
SELECT * FROM Orders
|
|
WHERE Poster = ?
|
|
ORDER BY Time ASC;
|
|
]=]
|
|
|
|
local order_book_asks_query = [=[
|
|
SELECT Type, Rate, Wear, SUM(Amount) AS Amount FROM Orders
|
|
GROUP BY Rate, Wear
|
|
HAVING Exchange = :ex_name
|
|
AND Type = "sell"
|
|
AND Item = :item_name
|
|
ORDER BY Rate ASC, Wear ASC
|
|
LIMIT 3;
|
|
]=]
|
|
|
|
local order_book_bids_query = [=[
|
|
SELECT Type, Rate, Wear, SUM(Amount) AS Amount FROM Orders
|
|
GROUP BY Rate, Wear
|
|
HAVING Exchange = :ex_name
|
|
AND Type = "buy"
|
|
AND Item = :item_name
|
|
ORDER BY Rate DESC, Wear DESC
|
|
LIMIT 3;
|
|
]=]
|
|
|
|
local insert_inbox_query = [=[
|
|
INSERT INTO Inbox(Recipient, Item, Wear, Amount)
|
|
VALUES(:p_name, :item_name, :wear, :amount);
|
|
]=]
|
|
|
|
local add_inbox_query = [=[
|
|
UPDATE Inbox
|
|
SET Amount = Amount + :change
|
|
WHERE Id = :id;
|
|
]=]
|
|
|
|
local view_inbox_query = [=[
|
|
SELECT * FROM Inbox
|
|
WHERE Recipient = ?
|
|
ORDER BY Item ASC, Wear ASC;
|
|
]=]
|
|
|
|
local search_inbox_query = [=[
|
|
SELECT * FROM Inbox
|
|
WHERE Recipient = :p_name
|
|
AND Item = :item_name
|
|
AND Wear = :wear;
|
|
]=]
|
|
|
|
local get_inbox_query = [=[
|
|
SELECT Amount, Wear FROM Inbox
|
|
WHERE Id = :id;
|
|
]=]
|
|
|
|
local red_inbox_query = [=[
|
|
UPDATE Inbox
|
|
SET Amount = Amount - :change
|
|
WHERE Id = :id;
|
|
]=]
|
|
|
|
local del_inbox_query = [=[
|
|
DELETE FROM Inbox
|
|
WHERE Id = :id;
|
|
]=]
|
|
|
|
local summary_query = [=[
|
|
SELECT * FROM market_summary;
|
|
]=]
|
|
|
|
local transaction_log_query = [=[
|
|
SELECT Time, Message FROM Log
|
|
WHERE Recipient = ?
|
|
ORDER BY Time DESC;
|
|
]=]
|
|
|
|
|
|
local ex_methods = {}
|
|
local ex_meta = { __index = ex_methods }
|
|
|
|
|
|
local function sql_error(err)
|
|
error("SQL error: " .. err)
|
|
end
|
|
|
|
|
|
local function is_integer(x)
|
|
local num = tonumber(x)
|
|
return num and math.floor(num) == num
|
|
end
|
|
|
|
|
|
local function exec_stmt(db, stmt, names)
|
|
stmt:bind_names(names)
|
|
|
|
local res = stmt:step()
|
|
stmt:reset()
|
|
|
|
if res == sqlite3.BUSY then
|
|
return false, "Database Busy."
|
|
elseif res ~= sqlite3.DONE then
|
|
sql_error(db:errmsg())
|
|
else
|
|
return true
|
|
end
|
|
end
|
|
|
|
|
|
function exports.open_exchange(path)
|
|
local db = assert(sqlite3.open(path))
|
|
|
|
local res = db:exec(init_query)
|
|
|
|
if res ~= sqlite3.OK then
|
|
sql_error(db:errmsg())
|
|
end
|
|
|
|
local stmts = {
|
|
new_act_stmt = assert(db:prepare(new_act_query)),
|
|
get_balance_stmt = assert(db:prepare(get_balance_query)),
|
|
set_balance_stmt = assert(db:prepare(set_balance_query)),
|
|
log_stmt = assert(db:prepare(log_query)),
|
|
search_asks_stmt = assert(db:prepare(search_asks_query)),
|
|
search_bids_stmt = assert(db:prepare(search_bids_query)),
|
|
qual_asks_stmt = assert(db:prepare(qual_asks_query)),
|
|
qual_bids_stmt = assert(db:prepare(qual_bids_query)),
|
|
search_own_stmt = assert(db:prepare(search_own_query)),
|
|
add_order_stmt = assert(db:prepare(add_order_query)),
|
|
get_order_stmt = assert(db:prepare(get_order_query)),
|
|
del_order_stmt = assert(db:prepare(del_order_query)),
|
|
reduce_order_stmt = assert(db:prepare(reduce_order_query)),
|
|
cancel_order_stmt = assert(db:prepare(cancel_order_query)),
|
|
insert_inbox_stmt = assert(db:prepare(insert_inbox_query)),
|
|
add_inbox_stmt = assert(db:prepare(add_inbox_query)),
|
|
order_book_bids_stmt = assert(db:prepare(order_book_bids_query)),
|
|
order_book_asks_stmt = assert(db:prepare(order_book_asks_query)),
|
|
view_inbox_stmt = assert(db:prepare(view_inbox_query)),
|
|
search_inbox_stmt = assert(db:prepare(search_inbox_query)),
|
|
get_inbox_stmt = assert(db:prepare(get_inbox_query)),
|
|
red_inbox_stmt = assert(db:prepare(red_inbox_query)),
|
|
del_inbox_stmt = assert(db:prepare(del_inbox_query)),
|
|
summary_stmt = assert(db:prepare(summary_query)),
|
|
transaction_log_stmt = assert(db:prepare(transaction_log_query)),
|
|
}
|
|
|
|
|
|
local ret = {
|
|
db = db,
|
|
stmts = stmts,
|
|
}
|
|
setmetatable(ret, ex_meta)
|
|
|
|
return ret
|
|
end
|
|
|
|
|
|
function ex_methods.close(self)
|
|
for k, v in pairs(self.stmts) do
|
|
v:finalize()
|
|
end
|
|
|
|
self.db:close()
|
|
end
|
|
|
|
-- Returns success boolean
|
|
function ex_methods.log(self, message, recipient)
|
|
recipient = recipient or ""
|
|
|
|
local db = self.db
|
|
local stmt = self.stmts.log_stmt
|
|
|
|
stmt:bind_names({
|
|
recipient = recipient,
|
|
time = os.time(),
|
|
message = message,
|
|
})
|
|
|
|
local res = stmt:step()
|
|
stmt:reset()
|
|
|
|
if res == sqlite3.ERROR then
|
|
sql_error(db:errmsg())
|
|
elseif res == sqlite3.MISUSE then
|
|
error("Programmer error.")
|
|
elseif res == sqlite3.BUSY then
|
|
return false, "Failed to log message."
|
|
else
|
|
return true
|
|
end
|
|
end
|
|
|
|
|
|
-- Returns success boolean and error.
|
|
function ex_methods.new_account(self, p_name, amt)
|
|
local db = self.db
|
|
amt = amt or 0
|
|
|
|
local exists = self:get_balance(p_name)
|
|
|
|
if exists then
|
|
return false, "Account already exists."
|
|
end
|
|
|
|
db:exec("BEGIN TRANSACTION;")
|
|
|
|
local stmt = self.stmts.new_act_stmt
|
|
|
|
stmt:bind_names({
|
|
owner = p_name,
|
|
start_balance = amt,
|
|
time = os.time(),
|
|
})
|
|
|
|
local res = stmt:step()
|
|
|
|
if res == sqlite3.MISUSE then
|
|
error("Programmer error.")
|
|
elseif res == sqlite3.BUSY then
|
|
stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, "Database Busy."
|
|
elseif res ~= sqlite3.DONE then
|
|
sql_error(db:errmsg())
|
|
end
|
|
|
|
stmt:reset()
|
|
|
|
local log_succ1, log_err1 =
|
|
self:log("Account opened with balance " .. amt, p_name)
|
|
local log_succ2, log_err2 =
|
|
self:log(p_name .. " opened an account with balance " .. amt)
|
|
|
|
if not log_succ1 then
|
|
db:exec("ROLLBACK;")
|
|
return false, log_err1
|
|
end
|
|
|
|
if not log_succ2 then
|
|
db:exec("ROLLBACK;")
|
|
return false, log_err2
|
|
end
|
|
|
|
db:exec("COMMIT;")
|
|
|
|
return true
|
|
end
|
|
|
|
|
|
-- Returns nil if no balance.
|
|
function ex_methods.get_balance(self, p_name)
|
|
local db = self.db
|
|
local stmt = self.stmts.get_balance_stmt
|
|
|
|
stmt:bind_values(p_name)
|
|
local res = stmt:step()
|
|
|
|
if res == sqlite3.ERROR then
|
|
sql_error(db:errmsg())
|
|
elseif res == sqlite3.MISUSE then
|
|
error("Programmer error.")
|
|
elseif res == sqlite3.ROW then
|
|
local balance = stmt:get_value(0)
|
|
stmt:reset()
|
|
|
|
return balance
|
|
end
|
|
|
|
stmt:reset()
|
|
return nil
|
|
end
|
|
|
|
|
|
-- Returns success boolean, and error message if false.
|
|
function ex_methods.set_balance(self, p_name, new_bal)
|
|
local db = self.db
|
|
local set_stmt = self.stmts.set_balance_stmt
|
|
|
|
local bal = self:get_balance(p_name)
|
|
|
|
if not bal then
|
|
return false, p_name .. " does not have an account."
|
|
end
|
|
|
|
set_stmt:bind_names({
|
|
p_name = p_name,
|
|
new_balance = new_bal,
|
|
})
|
|
|
|
local res = set_stmt:step()
|
|
|
|
if res == sqlite3.ERROR then
|
|
sql_error(db:errmsg())
|
|
elseif res == sqlite3.MISUSE then
|
|
error("Programmer error.")
|
|
elseif res == sqlite3.BUSY then
|
|
set_stmt:reset()
|
|
return false, "Database busy"
|
|
else
|
|
set_stmt:reset()
|
|
return true
|
|
end
|
|
end
|
|
|
|
|
|
-- Change balance by the given amount. Returns a success boolean, and error
|
|
-- message on fail.
|
|
function ex_methods.change_balance(self, p_name, delta)
|
|
if not is_integer(delta) then
|
|
error("Non-integer credit delta")
|
|
end
|
|
|
|
local bal = self:get_balance(p_name)
|
|
|
|
if not bal then
|
|
return false, p_name .. " does not have an account."
|
|
end
|
|
|
|
if bal + delta < 0 then
|
|
return false, p_name .. " does not have enough money."
|
|
end
|
|
|
|
return self:set_balance(p_name, bal + delta)
|
|
end
|
|
|
|
|
|
-- Sends credits from one user to another. Returns a success boolean, and error
|
|
-- message on fail.
|
|
function ex_methods.transfer_credits(self, sender, receiver, amt)
|
|
local db = self.db
|
|
|
|
if not is_integer(amt) then
|
|
return false, "Non-integer credit amount"
|
|
end
|
|
|
|
db:exec("BEGIN TRANSACTION;")
|
|
|
|
local succ_minus, err = self:change_balance(sender, -amt)
|
|
|
|
if not succ_minus then
|
|
db:exec("ROLLBACK")
|
|
return false, err
|
|
end
|
|
|
|
local succ_plus, err = self:change_balance(receiver, amt)
|
|
|
|
if not succ_plus then
|
|
db:exec("ROLLBACK")
|
|
return false, err
|
|
end
|
|
|
|
local succ_log1 = self:log("Sent " .. amt .. " credits to " .. receiver, sender)
|
|
|
|
if not succ_log1 then
|
|
db:exec("ROLLBACK")
|
|
return false, "Failed to log sender message"
|
|
end
|
|
|
|
local succ_log2 = self:log("Received " .. amt .. " credits from " .. sender, receiver)
|
|
|
|
if not succ_log2 then
|
|
db:exec("ROLLBACK")
|
|
return false, "Failed to log receiver message"
|
|
end
|
|
|
|
db:exec("COMMIT;")
|
|
|
|
return true
|
|
end
|
|
|
|
|
|
function ex_methods.give_credits(self, p_name, amt, msg)
|
|
local db = self.db
|
|
|
|
db:exec("BEGIN TRANSACTION;")
|
|
|
|
local succ_change, err = self:change_balance(p_name, amt)
|
|
|
|
if not succ_change then
|
|
db:exec("ROLLBACK;")
|
|
return false, err
|
|
end
|
|
|
|
local succ_log, err = self:log(msg, p_name)
|
|
|
|
if not succ_log then
|
|
db:exec("ROLLBACK;")
|
|
return false, er
|
|
end
|
|
|
|
db:exec("COMMIT;")
|
|
|
|
return true
|
|
end
|
|
|
|
|
|
-- The best asks & bids for an item, grouped and sorted by rate and wear.
|
|
-- Result fields: Type, Rate, Wear, Amount.
|
|
function ex_methods.order_book(self, ex_name, item_name)
|
|
if not ex_name or not item_name then return {} end
|
|
|
|
local ex_cache = order_book_cache(ex_name)
|
|
if ex_cache[item_name] then
|
|
return ex_cache[item_name]
|
|
end
|
|
|
|
local res = {}
|
|
|
|
local stmt = self.stmts.order_book_asks_stmt
|
|
stmt:bind_names({
|
|
ex_name = ex_name,
|
|
item_name = item_name,
|
|
})
|
|
|
|
-- Insert asks in reverse order
|
|
for row in stmt:nrows() do
|
|
table.insert(res, 1, row)
|
|
end
|
|
|
|
stmt:reset()
|
|
|
|
local stmt = self.stmts.order_book_bids_stmt
|
|
stmt:bind_names({
|
|
ex_name = ex_name,
|
|
item_name = item_name,
|
|
})
|
|
|
|
-- Append bids in normal order
|
|
for row in stmt:nrows() do
|
|
table.insert(res, row)
|
|
end
|
|
|
|
stmt:reset()
|
|
|
|
ex_cache[item_name] = res
|
|
|
|
return res
|
|
end
|
|
|
|
|
|
-- Returns a list of orders, sorted by price.
|
|
function ex_methods.search_orders(self, ex_name, order_type, item_name)
|
|
local stmt
|
|
if order_type == "buy" then
|
|
stmt = self.stmts.search_bids_stmt
|
|
else
|
|
stmt = self.stmts.search_asks_stmt
|
|
end
|
|
|
|
stmt:bind_names({
|
|
ex_name = ex_name,
|
|
item_name = item_name,
|
|
})
|
|
|
|
local orders = {}
|
|
|
|
for tab in stmt:nrows() do
|
|
table.insert(orders, tab)
|
|
end
|
|
|
|
stmt:reset()
|
|
return orders
|
|
end
|
|
|
|
|
|
-- Same as above, except not sorted in any particular order.
|
|
function ex_methods.search_player_orders(self, p_name)
|
|
local stmt = self.stmts.search_own_stmt
|
|
|
|
stmt:bind_values(p_name)
|
|
|
|
local orders = {}
|
|
|
|
for tab in stmt:nrows() do
|
|
table.insert(orders, tab)
|
|
end
|
|
|
|
stmt:reset()
|
|
return orders
|
|
end
|
|
|
|
|
|
-- Adds a new order. Returns success, and an error string if failed.
|
|
function ex_methods.add_order(self, p_name, ex_name, order_type, item_name, wear, amount, rate)
|
|
if not is_integer(amount) then
|
|
return false, "Noninteger quantity"
|
|
elseif amount <= 0 then
|
|
return false, "Nonpositive quantity"
|
|
elseif not is_integer(rate) then
|
|
return false, "Noninteger rate"
|
|
elseif rate <= 0 then
|
|
return false, "Nonpositive rate"
|
|
elseif not is_integer(wear) then
|
|
return false, "Noninteger wear"
|
|
elseif wear < 0 or wear > 65535 then
|
|
return false, "Invalid wear"
|
|
end
|
|
|
|
order_book_cache(ex_name)[item_name] = nil
|
|
|
|
local db = self.db
|
|
local stmt = self.stmts.add_order_stmt
|
|
|
|
stmt:bind_names({
|
|
p_name = p_name,
|
|
ex_name = ex_name,
|
|
order_type = order_type,
|
|
time = os.time(),
|
|
item_name = item_name,
|
|
wear = wear,
|
|
amount = amount,
|
|
rate = rate,
|
|
})
|
|
|
|
local res = stmt:step()
|
|
|
|
if res == sqlite3.BUSY then
|
|
stmt:reset()
|
|
return false, "Database Busy"
|
|
elseif res ~= sqlite3.DONE then
|
|
sql_error(db:errmsg())
|
|
end
|
|
|
|
stmt:reset()
|
|
return true
|
|
end
|
|
|
|
|
|
-- Returns true, or false and an error message.
|
|
function ex_methods.cancel_order(self, p_name, id)
|
|
local params = {
|
|
p_name = p_name,
|
|
id = id,
|
|
}
|
|
|
|
local db = self.db
|
|
db:exec("BEGIN TRANSACTION;")
|
|
|
|
local get_stmt = self.stmts.get_order_stmt
|
|
get_stmt:bind_values(id)
|
|
local res = get_stmt:step()
|
|
local order
|
|
|
|
if res == sqlite3.ERROR then
|
|
sql_error(db:errmsg())
|
|
elseif res == sqlite3.MISUSE then
|
|
error("Programmer error.")
|
|
elseif res == sqlite3.ROW then
|
|
order = get_stmt:get_named_values()
|
|
get_stmt:reset()
|
|
else
|
|
db:exec("ROLLBACK;")
|
|
return false, "No such order."
|
|
end
|
|
|
|
order_book_cache(order.Exchange)[order.Item] = nil
|
|
|
|
local cancel_stmt = self.stmts.cancel_order_stmt
|
|
local canc_succ, canc_err = exec_stmt(db, cancel_stmt, params)
|
|
if not canc_succ then
|
|
db:exec("ROLLBACK;")
|
|
return false, canc_err
|
|
end
|
|
|
|
local message = "Cancelled an order to " ..
|
|
order.Type .. " " .. order.Amount .. " " .. order.Item .. "."
|
|
|
|
local succ, err = self:log(message, p_name)
|
|
if not succ then
|
|
db:exec("ROLLBACK")
|
|
return false, err
|
|
end
|
|
|
|
db:exec("COMMIT;")
|
|
|
|
return true, order
|
|
end
|
|
|
|
|
|
-- Puts things in a player's item inbox. Returns success, and also returns an
|
|
-- error message if failed.
|
|
function ex_methods.put_in_inbox(self, p_name, item_name, wear, amount)
|
|
local db = self.db
|
|
local search_stmt = self.stmts.search_inbox_stmt
|
|
|
|
db:exec("BEGIN TRANSACTION;")
|
|
|
|
search_stmt:bind_names({
|
|
p_name = p_name,
|
|
item_name = item_name,
|
|
wear = wear,
|
|
})
|
|
|
|
local res = search_stmt:step()
|
|
local row = nil
|
|
|
|
if res == sqlite3.BUSY then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, "Database Busy."
|
|
elseif res == sqlite3.ROW then
|
|
row = search_stmt:get_named_values()
|
|
elseif res ~= sqlite3.DONE then
|
|
sql_error(db:errmsg())
|
|
end
|
|
|
|
search_stmt:reset()
|
|
|
|
local stmt
|
|
|
|
if row then
|
|
stmt = self.stmts.add_inbox_stmt
|
|
|
|
stmt:bind_names({
|
|
id = row.Id,
|
|
change = amount,
|
|
})
|
|
else
|
|
stmt = self.stmts.insert_inbox_stmt
|
|
|
|
stmt:bind_names({
|
|
p_name = p_name,
|
|
item_name = item_name,
|
|
wear = wear,
|
|
amount = amount,
|
|
})
|
|
end
|
|
|
|
local res = stmt:step()
|
|
|
|
if res == sqlite3.BUSY then
|
|
stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, "Database Busy."
|
|
elseif res ~= sqlite3.DONE then
|
|
sql_error(db:errmsg())
|
|
end
|
|
|
|
stmt:reset()
|
|
|
|
db:exec("COMMIT;")
|
|
|
|
return true
|
|
end
|
|
|
|
|
|
-- Tries to buy from orders at the provided rate, and posts an offer with any
|
|
-- remaining desired amount. Returns success. If succeeded, also returns amount
|
|
-- bought. If failed, returns an error message
|
|
function ex_methods.buy(self, p_name, ex_name, item_name, wear, amount, rate)
|
|
if not is_integer(amount) then
|
|
return false, "Noninteger quantity"
|
|
elseif amount <= 0 then
|
|
return false, "Nonpositive quantity"
|
|
elseif not is_integer(rate) then
|
|
return false, "Noninteger rate"
|
|
elseif rate <= 0 then
|
|
return false, "Nonpositive rate"
|
|
elseif not is_integer(wear) then
|
|
return false, "Noninteger wear"
|
|
elseif wear < 0 or wear > 65535 then
|
|
return false, "Invalid wear"
|
|
end
|
|
|
|
local db = self.db
|
|
|
|
db:exec("BEGIN TRANSACTION");
|
|
|
|
local balance = self:get_balance(p_name)
|
|
|
|
if not balance then
|
|
db:exec("ROLLBACK;")
|
|
return false, p_name .. " does not have an account."
|
|
end
|
|
|
|
local bought = {}
|
|
local remaining = amount
|
|
|
|
local del_stmt = self.stmts.del_order_stmt
|
|
local red_stmt = self.stmts.reduce_order_stmt
|
|
local search_stmt = self.stmts.qual_asks_stmt
|
|
|
|
search_stmt:bind_names({
|
|
ex_name = ex_name,
|
|
item_name = item_name,
|
|
rate_max = rate,
|
|
wear_max = wear,
|
|
})
|
|
|
|
for row in search_stmt:nrows() do
|
|
local poster = row.Poster
|
|
local row_wear = row.Wear
|
|
local row_amount = row.Amount
|
|
local row_rate = row.Rate
|
|
local row_bought = math.min(row_amount, remaining)
|
|
|
|
if poster ~= p_name then
|
|
local can_afford = math.floor(balance / row_rate)
|
|
row_bought = math.min(row_bought, can_afford)
|
|
-- asking prices can only increase from here
|
|
if row_bought == 0 then break end
|
|
end
|
|
|
|
local red_del_stmt
|
|
|
|
if row_bought < row_amount then
|
|
red_stmt:bind_names({
|
|
id = row.Id,
|
|
delta = row_bought,
|
|
})
|
|
red_del_stmt = red_stmt
|
|
else -- row_bought == row_amount
|
|
del_stmt:bind_values(row.Id)
|
|
red_del_stmt = del_stmt
|
|
end
|
|
|
|
local red_del_res = red_del_stmt:step()
|
|
if red_del_res == sqlite3.BUSY then
|
|
red_del_stmt:reset()
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, "Database Busy."
|
|
elseif red_del_res ~= sqlite3.DONE then
|
|
red_del_stmt:reset()
|
|
search_stmt:reset()
|
|
sql_error(db:errmsg())
|
|
end
|
|
red_del_stmt:reset()
|
|
|
|
if poster ~= p_name then
|
|
local cost = row_rate * row_bought
|
|
|
|
local ch_succ, ch_err = self:change_balance(poster, cost)
|
|
if not ch_succ then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, ch_err
|
|
end
|
|
|
|
local ch_succ, ch_err = self:change_balance(p_name, -cost)
|
|
if not ch_succ then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, ch_err
|
|
end
|
|
|
|
balance = balance - cost
|
|
|
|
local log_succ, log_err =
|
|
self:log(p_name .. " bought " .. row_amount .. " " ..
|
|
item_name .. " from you. (+" .. cost .. ")",
|
|
poster)
|
|
if not log_succ then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, log_err
|
|
end
|
|
|
|
local log_succ, log_err =
|
|
self:log("Bought " .. row_amount .. " " .. item_name ..
|
|
" from " .. poster .. ". (-" .. cost .. ")",
|
|
p_name)
|
|
if not log_succ then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, log_err
|
|
end
|
|
else
|
|
local log_succ, log_err =
|
|
self:log("Bought " .. row_amount .. " " ..
|
|
item_name .. " from yourself.",
|
|
p_name)
|
|
if not log_succ then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, log_err
|
|
end
|
|
end
|
|
|
|
order_book_cache(ex_name)[item_name] = nil
|
|
|
|
table.insert(bought, { amount = row_bought, wear = row_wear })
|
|
|
|
remaining = remaining - row_bought
|
|
|
|
if remaining == 0 then break end
|
|
end
|
|
|
|
search_stmt:reset()
|
|
|
|
if remaining > 0 then
|
|
local add_succ, add_err =
|
|
self:add_order(p_name, ex_name, "buy", item_name, wear, remaining, rate)
|
|
|
|
if not add_succ then
|
|
db:exec("ROLLBACK;")
|
|
return false, add_err
|
|
end
|
|
|
|
local log_succ, log_err =
|
|
self:log("Posted buy offer for " .. remaining .. " " ..
|
|
item_name .. " at " .. rate .. "/ea.",
|
|
p_name)
|
|
if not log_succ then
|
|
db:exec("ROLLBACK;")
|
|
return false, log_err
|
|
end
|
|
end
|
|
|
|
db:exec("COMMIT;")
|
|
|
|
return true, bought
|
|
end
|
|
|
|
|
|
-- Tries to sell to orders at the provided rate, and posts an offer with any
|
|
-- remaining desired amount. Returns success. If failed, returns an error message.
|
|
function ex_methods.sell(self, p_name, ex_name, item_name, wear, amount, rate)
|
|
if not is_integer(amount) then
|
|
return false, "Noninteger quantity"
|
|
elseif amount <= 0 then
|
|
return false, "Nonpositive quantity"
|
|
elseif not is_integer(rate) then
|
|
return false, "Noninteger rate"
|
|
elseif rate <= 0 then
|
|
return false, "Nonpositive rate"
|
|
elseif not is_integer(wear) then
|
|
return false, "Noninteger wear"
|
|
elseif wear < 0 or wear > 65535 then
|
|
return false, "Invalid wear"
|
|
end
|
|
|
|
local db = self.db
|
|
|
|
db:exec("BEGIN TRANSACTION");
|
|
|
|
local remaining = amount
|
|
|
|
local del_stmt = self.stmts.del_order_stmt
|
|
local red_stmt = self.stmts.reduce_order_stmt
|
|
local search_stmt = self.stmts.qual_bids_stmt
|
|
|
|
search_stmt:bind_names({
|
|
ex_name = ex_name,
|
|
item_name = item_name,
|
|
rate_min = rate,
|
|
wear_min = wear,
|
|
})
|
|
|
|
for row in search_stmt:nrows() do
|
|
local poster = row.Poster
|
|
local row_amount = row.Amount
|
|
local row_rate = row.Rate
|
|
local row_sold = math.min(row_amount, remaining)
|
|
|
|
if poster ~= p_name then
|
|
local bal = self:get_balance(poster)
|
|
|
|
if not bal then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, poster .. " does not have an account."
|
|
end
|
|
|
|
local can_afford = math.floor(bal / row_rate)
|
|
row_sold = math.min(row_sold, can_afford)
|
|
end
|
|
|
|
if row_sold > 0 then
|
|
local red_del_stmt
|
|
|
|
if row_sold < row_amount then
|
|
red_stmt:bind_names({
|
|
id = row.Id,
|
|
delta = row_sold,
|
|
})
|
|
red_del_stmt = red_stmt
|
|
else -- row_sold == row_amount
|
|
del_stmt:bind_values(row.Id)
|
|
red_del_stmt = del_stmt
|
|
end
|
|
|
|
local red_del_res = red_del_stmt:step()
|
|
red_del_stmt:reset()
|
|
if red_del_res == sqlite3.BUSY then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, "Database Busy."
|
|
elseif red_del_res ~= sqlite3.DONE then
|
|
search_stmt:reset()
|
|
sql_error(db:errmsg())
|
|
end
|
|
|
|
local in_succ, in_err =
|
|
self:put_in_inbox(poster, item_name, wear, row_sold)
|
|
if not in_succ then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, in_err
|
|
end
|
|
|
|
if poster ~= p_name then
|
|
local revenue = row_sold * row_rate
|
|
|
|
local ch_succ, ch_err = self:change_balance(poster, -revenue)
|
|
if not ch_succ then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, ch_err
|
|
end
|
|
|
|
local ch_succ, ch_err = self:change_balance(p_name, revenue)
|
|
if not ch_succ then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, ch_err
|
|
end
|
|
|
|
local log_succ, log_err =
|
|
self:log(p_name .. " sold " .. row_amount .. " " ..
|
|
item_name .. " to you. (-" .. revenue .. ")",
|
|
poster)
|
|
if not log_succ then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, log_err
|
|
end
|
|
|
|
local log_succ, log_err =
|
|
self:log("Sold " .. row_amount .. " " .. item_name ..
|
|
" to " .. poster .. "(+" .. revenue .. ")",
|
|
p_name)
|
|
if not log_succ then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, log_err
|
|
end
|
|
else
|
|
local log_succ, log_err =
|
|
self:log("Sold " .. row_amount .. " " ..
|
|
item_name .. " to yourself.",
|
|
p_name)
|
|
if not log_succ then
|
|
search_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, log_err
|
|
end
|
|
end
|
|
|
|
order_book_cache(ex_name)[item_name] = nil
|
|
|
|
remaining = remaining - row_sold
|
|
|
|
if remaining == 0 then break end
|
|
end
|
|
end
|
|
|
|
search_stmt:reset()
|
|
|
|
if remaining > 0 then
|
|
local add_succ, add_err =
|
|
self:add_order(p_name, ex_name, "sell", item_name, wear, remaining, rate)
|
|
|
|
if not add_succ then
|
|
db:exec("ROLLBACK;")
|
|
return false, add_err
|
|
end
|
|
end
|
|
|
|
db:exec("COMMIT;")
|
|
|
|
return true
|
|
end
|
|
|
|
|
|
-- On success, returns true and a list of inbox entries.
|
|
-- TODO: On failure, return false and an error message.
|
|
function ex_methods.view_inbox(self, p_name)
|
|
local stmt = self.stmts.view_inbox_stmt
|
|
|
|
stmt:bind_values(p_name)
|
|
|
|
local res = {}
|
|
|
|
for row in stmt:nrows() do
|
|
table.insert(res, row)
|
|
end
|
|
|
|
stmt:reset()
|
|
|
|
return true, res
|
|
end
|
|
|
|
|
|
-- Returns success boolean. On success, also returns the number actually
|
|
-- taken. On failure, also returns an error message
|
|
function ex_methods.take_inbox(self, id, amount)
|
|
local db = self.db
|
|
local get_stmt = self.stmts.get_inbox_stmt
|
|
local red_stmt = self.stmts.red_inbox_stmt
|
|
local del_stmt = self.stmts.del_inbox_stmt
|
|
|
|
get_stmt:bind_names({ id = id })
|
|
|
|
local res = get_stmt:step()
|
|
|
|
if res == sqlite3.BUSY then
|
|
get_stmt:reset()
|
|
return false, "Database Busy."
|
|
elseif res == sqlite3.DONE then
|
|
get_stmt:reset()
|
|
return false, "Order does not exist."
|
|
elseif res ~= sqlite3.ROW then
|
|
sql_error(db:errmsg())
|
|
end
|
|
|
|
local available = get_stmt:get_value(0)
|
|
get_stmt:reset()
|
|
|
|
db:exec("BEGIN TRANSACTION;")
|
|
|
|
local red_del_stmt
|
|
|
|
if available > amount then
|
|
red_stmt:bind_names({
|
|
id = id,
|
|
change = amount
|
|
})
|
|
|
|
red_del_stmt = red_stmt
|
|
else
|
|
del_stmt:bind_values(id)
|
|
red_del_stmt = del_stmt
|
|
end
|
|
|
|
local red_del_res = red_del_stmt:step()
|
|
|
|
if red_del_res == sqlite3.BUSY then
|
|
red_del_stmt:reset()
|
|
db:exec("ROLLBACK;")
|
|
return false, "Database Busy."
|
|
elseif red_del_res ~= sqlite3.DONE then
|
|
sql_error(db:errmsg())
|
|
end
|
|
|
|
red_del_stmt:reset()
|
|
|
|
db:exec("COMMIT;")
|
|
return true, math.min(amount, available)
|
|
end
|
|
|
|
|
|
-- Returns a list of tables with fields:
|
|
-- item_name: Name of the item
|
|
-- buy_volume: Number of items sought
|
|
-- buy_max: Maximum buy rate
|
|
-- sell_volume: Number of items for sale
|
|
-- sell_min: Minimum sell rate
|
|
function ex_methods.market_summary(self)
|
|
local stmt = self.stmts.summary_stmt
|
|
|
|
local res = {}
|
|
for a in stmt:rows() do
|
|
table.insert(res, {
|
|
item_name = a[1],
|
|
buy_volume = a[2],
|
|
buy_max = a[3],
|
|
sell_volume = a[4],
|
|
sell_min = a[5],
|
|
})
|
|
end
|
|
stmt:reset()
|
|
|
|
return res
|
|
end
|
|
|
|
|
|
-- Returns a list of log entries, sorted by time.
|
|
function ex_methods.player_log(self, p_name)
|
|
local stmt = self.stmts.transaction_log_stmt
|
|
stmt:bind_values(p_name)
|
|
|
|
local res = {}
|
|
|
|
for row in stmt:nrows() do
|
|
table.insert(res, row)
|
|
end
|
|
|
|
stmt:reset()
|
|
|
|
return res
|
|
end
|
|
|
|
function exports.test()
|
|
local ex = exports.open_exchange("test.db")
|
|
|
|
local alice_bal = ex:get_balance("Alice")
|
|
local bob_bal = ex:get_balance("Bob")
|
|
|
|
|
|
local function print_balances()
|
|
print("Alice: ", ex:get_balance("Alice"))
|
|
print("Bob: ", ex:get_balance("Bob"))
|
|
end
|
|
|
|
|
|
-- Initialize balances
|
|
if alice_bal then
|
|
ex:set_balance("Alice", 420)
|
|
else
|
|
ex:new_account("Alice", 420)
|
|
end
|
|
|
|
if bob_bal then
|
|
ex:set_balance("Bob", 2015)
|
|
else
|
|
ex:new_account("Bob", 2015)
|
|
end
|
|
|
|
print_balances()
|
|
|
|
|
|
-- Transfer a valid amount
|
|
print("Transfering 1000 credits from Bob to Alice")
|
|
|
|
local succ, err = ex:transfer_credits("Bob", "Alice", 1000)
|
|
|
|
print("Success: ", succ, " ", err)
|
|
print_balances()
|
|
|
|
|
|
-- Transfer an invalid amount
|
|
print("Transfering 3000 credits from Alice to Bob")
|
|
|
|
local succ, err = ex:transfer_credits("Alice", "Bob", 3000)
|
|
|
|
print("Success: ", succ, " ", err)
|
|
print_balances()
|
|
|
|
|
|
-- Simulate a transaction
|
|
print("Alice posting an offer to buy 10 cobble at 2 credits each")
|
|
local succ, err = ex:buy("Alice", "", "default:cobble", 32767, 10, 2)
|
|
print("Success: ", succ, " ", err)
|
|
print_balances()
|
|
|
|
print("Bob posting an offer to sell 20 cobble at 1 credits each")
|
|
local succ, err = ex:sell("Bob", "", "default:cobble", 32767, 20, 1)
|
|
print("Success: ", succ, " ", err)
|
|
print_balances()
|
|
|
|
ex:close()
|
|
end
|
|
|
|
|
|
return exports
|
|
-- vim:set ts=4 sw=4 noet:
|