#!/usr/bin/env lem
--
-- This file is part of lem-postgres.
-- Copyright 2011 Emil Renner Berthing
-- Copyright 2013 Asbjørn Sloth Tønnesen
--
-- lem-postgres is free software: you can redistribute it and/or
-- modify it under the terms of the GNU General Public License as
-- published by the Free Software Foundation, either version 3 of
-- the License, or (at your option) any later version.
--
-- lem-postgres is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with lem-postgres. If not, see <http://www.gnu.org/licenses/>.
--

package.path = '?.lua;'..package.path
package.cpath = '?.so;'..package.cpath

print("Entering " .. arg[0])

local prettyprint
do
	local write, format, tostring = io.write, string.format, tostring

	function prettyprint(t)
		local widths, columns = {}, #t[0]
		for i = 1, columns do
			widths[i] = 0
		end

		for i = 0, #t do
			local row = t[i]
			for j = 1, columns do
				local value = row[j]
				if value and #value > widths[j] then
					widths[j] = #value
				end
			end
		end

		for i = 1, #widths do
			widths[i] = '%-' .. tostring(widths[i] + 1) .. 's';
		end

		for i = 0, #t do
			local row = t[i]
			for j = 1, columns do
				write(format(widths[j], row[j] or 'NULL'))
			end
			write('\n')
		end
	end
end

local utils    = require 'lem.utils'
--local postgres = require 'lem.postgres'
local postgres = require 'lem.postgres.queued'

local dbconnstr = [[
host=localhost
user=myuser
password=mypasswd
dbname=mydb
]]

local db = assert(postgres.connect(dbconnstr))

assert(db:reset())

assert(db:exec(
'CREATE TEMPORARY TABLE mytable (id serial PRIMARY KEY, name TEXT, foo integer)'))

assert(db:exec("COPY mytable (name, foo) FROM STDIN (delimiter ',')"))
assert(db:put('alpha,1\n'))
assert(db:put('beta,2\n'))
assert(db:put('gamma,4\n'))
assert(db:put('delta,8\n'))
assert(db:put('epsilon,\\N\n'))
assert(db:done())

local res = assert(db:exec('SELECT * FROM mytable WHERE id = $1', '1'))
prettyprint(res)

local res1, res2 = assert(db:exec([[
SELECT count(id) FROM mytable;
SELECT * FROM mytable WHERE id = 2;
]]));
prettyprint(res2)

assert(db:prepare('mystmt', 'SELECT * FROM mytable WHERE id = $1'))
local res = assert(db:run('mystmt', '3'))
prettyprint(res)

assert(db:prepare('insert', 'INSERT INTO mytable (name, foo) VALUES ($1, $2)'))
assert(db:run('insert', 'zeta', 32))
assert(db:run('insert', 'eta', nil))

assert(db:exec("COPY mytable TO STDIN (format csv, null '\\N')"))

print("\nDumping CSV:")
while true do
	local row = assert(db:get())
	if type(row) ~= 'string' then break end

	io.write(row)
end

assert(db:exec('DROP TABLE mytable'))

assert(db:exec('LISTEN mytest;'))

assert(db:exec('NOTIFY mytest;'))
assert(db:exec("NOTIFY mytest, 'foo';"))

db:set_notifier(function(ev, payload, pid)
	print('notification', ev, payload, pid)
end)

assert(db:exec('SELECT pg_notify($1, $2);', 'mytest', 'bar'))
assert(db:exec('SELECT pg_notify($1, $2), pg_notify($1, $3);', 'mytest', 'foo', 'bar'))

db:set_notifier(function(ev, payload, pid)
	print('\nnotification', ev, payload, pid)
end)

utils.spawn(function()
	local sleeper = utils.newsleeper()
	for i=1,19 do
		sleeper:sleep(0.20)
		io.write('.')
	end
	print('')
end)


utils.spawn(function()
	local sleeper = utils.newsleeper()
	local db2 = assert(postgres.connect(dbconnstr))
	for i=1,3 do
		sleeper:sleep(1)
		assert(db2:exec("NOTIFY mytest, 'foo from secondary connection';"))
	end
end)

local sleeper = utils.newsleeper()
print('\ngoing to sleep')
sleeper:sleep(4)
print('awake again\n')

print("Exiting " .. arg[0])

-- vim: syntax=lua ts=2 sw=2 noet: