Skip to content

Instantly share code, notes, and snippets.

@bubnenkoff
Created May 4, 2016 10:01
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save bubnenkoff/db0632bb14eebb7690b6563c47395831 to your computer and use it in GitHub Desktop.
Save bubnenkoff/db0632bb14eebb7690b6563c47395831 to your computer and use it in GitHub Desktop.
import vibe.d;
import std.stdio;
import std.conv;
import std.algorithm;
import std.datetime;
import std.path;
import std.file;
import parseconfig;
import dbconnect;
import users;
import requests.http;
//DateTime currentdt;
//static this()
//{
// DateTime currentdt = cast(DateTime)(Clock.currTime()); // It's better to declarate globally
// string datestamp = currentdt.toISOExtString;
//}
string roothtml;
static this()
{
roothtml = buildPath(getcwd, "html") ~ "\\";
if(!roothtml.exists)
writeln("[ERROR] HTML dir do not exists");
}
Config config;
void main()
{
config = new Config();
DBConnect dbconnect = new DBConnect(config);
if (dbconnect.checkServerStatus())
{
writeln("Server alive!");
}
else
{
writeln("Could not connect to server");
return;
}
getUsersFromDB();
auto router = new URLRouter;
router.get("/*", serveStaticFiles(roothtml ~ "\\"));
router.get("*", serveStaticFiles(roothtml ~ "static\\"));
router.any("*", &accControl);
router.any("/checkAuthorization", &checkAuthorization);
router.any("/login", &login);
router.post("/logout", &logout);
router.any("/test", &test);
router.any("/questions", &onlinetestdata);
bool isAuthorizated = false;
bool isAdmin = false;
auto settings = new HTTPServerSettings;
settings.port = 8080;
settings.bindAddresses = ["::", "127.0.0.1"];
settings.sessionStore = new MemorySessionStore; // SESSION
writeln(config.allow_voting_for_unauthorized);
writeln("\nHOST: ", config.dbhost);
writeln("--------sending data---------");
listenHTTP(settings, router);
runEventLoop();
}
void accControl(HTTPServerRequest req, HTTPServerResponse res)
{
res.headers["Access-Control-Allow-Origin"] = "*";
}
AuthInfo _auth;
Tuple!(string, "login", string, "password") [] usersInDB;
void getUsersFromDB()
{
// List of documents in collection
string baseURL = "http://localhost:8529";
string url = "http://localhost:8529/_db/onlinetest/_api/document/?collection=users";
import std.experimental.logger;
globalLogLevel(LogLevel.error);
Json mycollection = Json.emptyObject;
auto rq = Request();
auto rs = rq.get(url);
//writeln(rs.responseBody.data!string);
mycollection = parseJsonString(rs.responseBody.data!string);
if (rs.code != 200)
writeln("Can't get list of documents in collection");
if (rs.code == 404)
writeln("URL do not exists: ", url);
string [] listOfDocumentsInUsers;
foreach(Json d;mycollection["documents"])
{
listOfDocumentsInUsers ~= baseURL ~ to!string(d).replace(`"`,``);
}
Tuple!(string, "login", string, "password") user;
Json userJson = Json.emptyObject;
auto rq1 = Request();
foreach(link; listOfDocumentsInUsers)
{
auto rs1 = rq1.get(link);
//userJson = parseJsonString(rs1.responseBody.data!string);
userJson = parseJsonString(rs1.responseBody.data!string);
user.login = to!string(userJson["login"]).replace(`"`,"");
user.password = to!string(userJson["password"]).replace(`"`,"");
usersInDB ~= user;
}
/*
{
"login": "admin",
"password": "123",
"type": "user",
"firstname": "",
"lastname": "",
"organization": "",
"lastvisit": "",
"tests" :
{
"allowed": [],
"passed": []
}
}
*/
//same as string:
// {"login": "admin", "password": "123", "type": "admin", "firstname": "", "lastname": "", "organization": "", "lastvisit": "", "tests" : {"allowed": [], "passed": []} }
}
void checkAuthorization(HTTPServerRequest req, HTTPServerResponse res)
{
logInfo("-----checkAuthorization START-----");
Json responseStatus = Json.emptyObject;
Json responseBody = Json.emptyObject; //should be _in_ responseStatus
//if user already on site
if (req.session)
{
logInfo("user already have active session");
if(_auth.isAuthorized) //only user authorizate
{
responseStatus["status"] = "success";
responseBody["isAuthorized"] = true;
responseBody["isAdmin"] = false;
responseBody["username"] = _auth.user.username;
responseStatus["login"] = responseBody;
if(_auth.isAdmin)
{
responseBody["isAdmin"] = true;
}
res.writeJsonBody(responseStatus);
logInfo(responseStatus.toString); // include responseBody
}
//example: {"login":{"isAuthorized":true,"isAdmin":false,"username":"test"},"status":"success"}
}
// Login info we should check only with /login here
else
{
responseStatus["status"] = "fail"; // unauthorized user
res.writeJsonBody(responseStatus);
}
logInfo("-----checkAuthorization END-------");
}
void test(HTTPServerRequest req, HTTPServerResponse res)
{
if (req.session)
res.writeBody("Hello, World!", "text/plain");
}
void onlinetestdata(HTTPServerRequest req, HTTPServerResponse res)
{
if (req.session)
res.writeBody("Hello, World!", "text/plain");
}
void onlinetestdata(HTTPServerRequest req, HTTPServerResponse res)
{
if (req.session)
res.writeBody("Hello, World!", "text/plain");
}
void login(HTTPServerRequest req, HTTPServerResponse res)
{
Json request = req.json;
//writeln(to!string(request["username"]));
writeln("-------------JSON OBJECT from site:-------------");
writeln(request);
writeln("^-----------------------------------------------^");
//readln;
try
{
// response. responseBody should be nested in "success" or "fail" block. Like {"fail": {...}}
Json responseStatus = Json.emptyObject;
Json responseBody = Json.emptyObject; //should be _in_
foreach(dbuser; usersInDB)
{
if (dbuser.login == request["username"].to!string && dbuser.password != request["password"].to!string)
{
////////USER OK PASSWORD WRONG///////////
responseStatus["status"] = "fail"; // user exists in DB, password NO
responseBody["password"] = "wrongPassword"; // user exists in DB, password NO
responseBody["isAuthorized"] = false;
responseStatus["login"] = responseBody;
logInfo("-------------------------------------------------------------------------------");
logInfo(responseStatus.toString); // include responseBody
logInfo("^-----------------------------------------------------------------------------^");
logWarn("WRONG password for USER: %s", request["username"]); //getting username from request
//output: {"login":{"isAuthorized":false,"password":"wrongPassword"},"status":"fail"}
}
if (dbuser.login == request["username"].to!string && dbuser.password == request["password"].to!string)
{
////////ALL RIGHT///////////
logInfo("DB-User: %s | DB-Password: %s", dbuser.login, dbuser.password);
if (!req.session) //if no session start one
{
req.session = res.startSession();
}
/* we should set this fields:
_auth.isAdmin
_auth.user.username
to get /checkAuthorization work! */
_auth.isAuthorized = true;
if(dbuser.login == "admin") // admin name hardcoded
{
try
{
_auth.isAdmin = true;
_auth.user.username = "admin";
//req.session.set("username", "admin"); //ditto
req.session.set!string("username", "admin");
responseStatus["status"] = "success";
responseBody["isAuthorized"] = true;
responseBody["isAdmin"] = true;
responseBody["username"] = dbuser.login; // admin!
responseStatus["login"] = responseBody;
logInfo("-------------------------------------------------------------------------------");
logInfo(responseStatus.toString); // include responseBody
logInfo("^-----------------------------------------------------------------------------^");
logInfo("Admin session for user: %s started", dbuser.login);
// {"login":{"isAuthorized":true,"isAdmin":true,"username":"admin"},"status":"success"}
}
catch(Exception e)
{
writeln("Error during admin login:");
writeln(e.msg);
}
}
if(dbuser.login != "admin") // start user session
{
try
{
req.session.set("username", dbuser.login); //set current username in parameter of session name
_auth.user.username = dbuser.login; //set field
responseStatus["status"] = "success";
responseBody["isAuthorized"] = true;
responseBody["isAdmin"] = false;
responseBody["username"] = dbuser.login; // user!
responseStatus["login"] = responseBody;
logInfo("-------------------------------------------------------------------------------");
logInfo(responseStatus.toString); // include responseBody
logInfo("^-----------------------------------------------------------------------------^");
logInfo("User session for user: %s started", dbuser.login);
// {"login":{"isAuthorized":true,"isAdmin":false,"username":"test"},"status":"success"}
}
catch (Exception e)
{
writeln("Error during user login:");
writeln(e.msg);
}
}
}
else // userDoNotExists
{
logInfo("User: %s do not exists in DB", dbuser.login);
responseStatus["status"] = "fail"; // user exists in DB, password NO
responseBody["username"] = "userDoNotExists"; // user exists in DB, password NO
responseBody["isAuthorized"] = false;
responseStatus["login"] = responseBody;
logInfo("-------------------------------------------------------------------------------");
logInfo(responseStatus.toString); // include responseBody
logInfo("^-----------------------------------------------------------------------------^");
logWarn("User %s DO NOT exist in DB", request["username"]); //getting username from request
}
res.writeJsonBody(responseStatus); //Final answer to server. Must be at the end
}
}
catch(Exception e)
{
writeln("Can't process select from DB otest.myusers");
writeln(e.msg);
}
}
void logout(HTTPServerRequest req, HTTPServerResponse res)
{
try
{
logInfo("Logout section");
Json request = req.json;
Json responseBody = Json.emptyObject; // function duplicate from login
if (req.session) // if user have active session
{
res.terminateSession();
responseBody["status"] = "success";
responseBody["isAuthorized"] = false;
res.writeJsonBody(responseBody);
logInfo("-------------------------------------------------------------------------------");
logInfo(responseBody.toString);
logInfo("^-----------------------------------------------------------------------------^");
logInfo("User %s logout", request["username"]); //
}
else
{
responseBody["status"] = "fail"; // user do not have active session?
logInfo("User do not have active session");
res.writeJsonBody(responseBody);
}
res.writeJsonBody(responseBody);
}
catch (Exception e)
{
logInfo(e.msg);
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment