38 lines
904 B
Lua
38 lines
904 B
Lua
|
#!/usr/bin/env lua
|
||
|
|
|
||
|
|
require 'luasql.postgres'
|
||
|
|
require 'helper'
|
||
|
|
require 'config'
|
||
|
|
require 'token'
|
||
|
|
|
||
|
|
print "* adding token ..."
|
||
|
|
|
||
|
|
env = luasql.postgres()
|
||
|
|
con = env:connect(config.db)
|
||
|
|
|
||
|
|
print "current users:"
|
||
|
|
for name in rows(con,"select name from users;") do print(name) end
|
||
|
|
|
||
|
|
io.stdout:write("enter name: ")
|
||
|
|
name = io.stdin:read()
|
||
|
|
|
||
|
|
if name == "" then
|
||
|
|
print "* action canceled"
|
||
|
|
else
|
||
|
|
cur = con:execute("insert into users (name, ntc) values ('"..name.."', 1);")
|
||
|
|
if cur == 1 then
|
||
|
|
print("* add user " .. name)
|
||
|
|
else
|
||
|
|
assert(con:execute("update users set ntc=ntc+1 where name='"..name.."';"))
|
||
|
|
end
|
||
|
|
token = generate_token()
|
||
|
|
print "* generate token"
|
||
|
|
ttl = os.time() + config.ttl
|
||
|
|
assert(con:execute("insert into tokens (pid, valid, token, ttl) values (NULL, true, '"..token.."', "..ttl..");"))
|
||
|
|
print "* add token:"
|
||
|
|
print(token)
|
||
|
|
end
|
||
|
|
|
||
|
|
con:close()
|
||
|
|
env:close()
|