55 lines
1.4 KiB
C++
55 lines
1.4 KiB
C++
/*
|
|
* ws_httpd.cpp
|
|
* Author: Benjamin Sergeant
|
|
* Copyright (c) 2018 Machine Zone, Inc. All rights reserved.
|
|
*/
|
|
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <ixwebsocket/IXHttpServer.h>
|
|
#include <spdlog/spdlog.h>
|
|
#include <sstream>
|
|
#include <vector>
|
|
|
|
namespace ix
|
|
{
|
|
int ws_httpd_main(int port,
|
|
const std::string& hostname,
|
|
bool redirect,
|
|
const std::string& redirectUrl)
|
|
{
|
|
spdlog::info("Listening on {}:{}", hostname, port);
|
|
|
|
ix::HttpServer server(port, hostname);
|
|
|
|
if (redirect)
|
|
{
|
|
//
|
|
// See https://developer.mozilla.org/en-US/docs/Web/HTTP/Redirections
|
|
//
|
|
server.setOnConnectionCallback(
|
|
[redirectUrl](
|
|
HttpRequestPtr request,
|
|
std::shared_ptr<ConnectionState> /*connectionState*/) -> HttpResponsePtr {
|
|
WebSocketHttpHeaders headers;
|
|
headers["Location"] = redirectUrl;
|
|
|
|
return std::make_shared<HttpResponse>(
|
|
301, "OK", HttpErrorCode::Ok, headers, std::string());
|
|
});
|
|
}
|
|
|
|
auto res = server.listen();
|
|
if (!res.first)
|
|
{
|
|
std::cerr << res.second << std::endl;
|
|
return 1;
|
|
}
|
|
|
|
server.start();
|
|
server.wait();
|
|
|
|
return 0;
|
|
}
|
|
} // namespace ix
|