browser-client.cpp

#include "browser-client.hpp"
#include "base64/base64.hpp"
#include "json11/json11.hpp"

#include <QApplication>
#include <QThread>
#include <QToolTip>
#include "GrpcBrowser.h"
#include "JavascriptApi.h"
#include "SlBrowser.h"
#include "WindowsFunctions.h"

#include <json11/json11.hpp>

using namespace json11;

inline bool BrowserClient::valid() const
{
	return true;
}

CefRefPtr<CefLoadHandler> BrowserClient::GetLoadHandler()
{
	return this;
}

CefRefPtr<CefRenderHandler> BrowserClient::GetRenderHandler()
{
	return this;
}

CefRefPtr<CefDisplayHandler> BrowserClient::GetDisplayHandler()
{
	return this;
}

CefRefPtr<CefLifeSpanHandler> BrowserClient::GetLifeSpanHandler()
{
	return this;
}

CefRefPtr<CefContextMenuHandler> BrowserClient::GetContextMenuHandler()
{
	return this;
}

CefRefPtr<CefAudioHandler> BrowserClient::GetAudioHandler()
{
	return m_reroute_audio ? this : nullptr;
}

#if CHROME_VERSION_BUILD >= 4638
CefRefPtr<CefRequestHandler> BrowserClient::GetRequestHandler()
{
	return this;
}

CefRefPtr<CefResourceRequestHandler> BrowserClient::GetResourceRequestHandler(CefRefPtr<CefBrowser>, CefRefPtr<CefFrame>, CefRefPtr<CefRequest> request, bool, bool, const CefString &, bool &)
{
	if (request->GetHeaderByName("origin") == "null")
	{
		return this;
	}

	return nullptr;
}

CefResourceRequestHandler::ReturnValue BrowserClient::OnBeforeResourceLoad(CefRefPtr<CefBrowser>, CefRefPtr<CefFrame>, CefRefPtr<CefRequest>, CefRefPtr<CefCallback>)
{
	return RV_CONTINUE;
}
#endif

bool BrowserClient::OnBeforePopup(CefRefPtr<CefBrowser>, CefRefPtr<CefFrame>, const CefString &, const CefString &, cef_window_open_disposition_t, bool, const CefPopupFeatures &, CefWindowInfo &, CefRefPtr<CefClient> &, CefBrowserSettings &, CefRefPtr<CefDictionaryValue> &, bool *)
{
	/* block popups */
	return true;
}

void BrowserClient::OnBeforeContextMenu(CefRefPtr<CefBrowser>, CefRefPtr<CefFrame>, CefRefPtr<CefContextMenuParams>, CefRefPtr<CefMenuModel> model)
{
	/* remove all context menu contributions */
	model->Clear();
}

/*static*/
json11::Json convertCefValueToJSON(CefRefPtr<CefValue> value)
{
	switch (value->GetType())
	{
	case VTYPE_NULL:
		return nullptr;

	case VTYPE_BOOL:
		return value->GetBool();

	case VTYPE_INT:
		return value->GetInt();

	case VTYPE_DOUBLE:
		return value->GetDouble();

	case VTYPE_STRING:
		return value->GetString().ToString();

	case VTYPE_LIST: {
		const auto &list = value->GetList();
		std::vector<json11::Json> jsonList;
		for (size_t i = 0; i < list->GetSize(); ++i)
		{
			jsonList.push_back(convertCefValueToJSON(list->GetValue(i)));
		}
		return jsonList;
	}

	case VTYPE_DICTIONARY: {
		const auto &dict = value->GetDictionary();
		std::map<std::string, json11::Json> jsonMap;
		CefDictionaryValue::KeyList keys;
		dict->GetKeys(keys);
		for (const auto &key : keys)
		{
			jsonMap[key] = convertCefValueToJSON(dict->GetValue(key));
		}
		return jsonMap;
	}

	default:
		return nullptr;
	}
}

/*static*/
std::string BrowserClient::cefListValueToJSONString(CefRefPtr<CefListValue> listValue)
{
	std::map<std::string, json11::Json> jsonMap;
	for (size_t i = 0; i < listValue->GetSize(); ++i)
	{
		// Convert index to string key like "param1", "param2", ...
		std::string key = "param" + std::to_string(i + 1);
		jsonMap[key] = convertCefValueToJSON(listValue->GetValue(i));
	}

	std::string json_str;
	json_str = json11::Json(jsonMap).dump();
	return json_str;
}

CefRefPtr<CefBrowser> BrowserClient::GetMostRecentRenderKnown()
{
	std::lock_guard<std::recursive_mutex> grd(m_recursiveMutex);
	return m_MostRecentRenderKnowOf;
}

void BrowserClient::RegisterCallback(const int functionId, CefRefPtr<CefBrowser> browser)
{
	std::lock_guard<std::recursive_mutex> grd(m_recursiveMutex);
	m_callbackDictionary[functionId] = browser;
	m_MostRecentRenderKnowOf = browser;
}

CefRefPtr<CefBrowser> BrowserClient::PopCallback(const int functionId)
{
	std::lock_guard<std::recursive_mutex> grd(m_recursiveMutex);

	auto itr = m_callbackDictionary.find(functionId);

	if (itr != m_callbackDictionary.end())
	{
		CefRefPtr<CefBrowser> browser = itr->second;
		m_callbackDictionary.erase(itr);
		return browser;
	}

	return nullptr;
}

bool BrowserClient::OnProcessMessageReceived(CefRefPtr<CefBrowser> browser, CefRefPtr<CefFrame>, CefProcessId processId, CefRefPtr<CefProcessMessage> message)
{
	const std::string &name = message->GetName();
	CefRefPtr<CefListValue> input_args = message->GetArgumentList();

	if (!valid())
		return false;

	int funcid = input_args->GetInt(0);

	if (JavascriptApi::isBrowserFunctionName(name))
	{
		std::string jsonOutput = "{}";

		std::vector <CefRefPtr<CefValue>> argsWithoutFunc;

		for (u_long l = 1; l < input_args->GetSize(); l++)
			argsWithoutFunc.push_back(input_args->GetValue(l));

		// Stuff done right here and now to the browser
		// Put this into a sub function if it gets bigger
		switch (JavascriptApi::getFunctionId(name))
		{
		case JavascriptApi::JS_BROWSER_RESIZE_BROWSER:
		{
			if (argsWithoutFunc.size() < 2)
			{
				jsonOutput = Json(Json::object({{"error", "Invalid parameters"}})).dump();
				break;
			}

			int w = argsWithoutFunc[0]->GetInt();
			int h = argsWithoutFunc[1]->GetInt();

			if (w < 200 || h < 200 || w > 8096 || h > 8096)
			{
				jsonOutput = Json(Json::object({{"error", "Invalid parameters"}})).dump();
				break;
			}

			SlBrowser::instance().m_widget->resize(w, h);
			break;
		}
		case JavascriptApi::JS_BROWSER_BRING_FRONT:
		{
			HWND hwnd = HWND(SlBrowser::instance().m_widget->winId());

			if (::IsIconic(hwnd))
				::ShowWindow(hwnd, SW_RESTORE);

			WindowsFunctions::ForceForegroundWindow(hwnd);
			break;
		}
		case JavascriptApi::JS_BROWSER_SET_WINDOW_POSITION:
		{
			if (argsWithoutFunc.size() < 2)
			{
				jsonOutput = Json(Json::object({{"error", "Invalid parameters"}})).dump();
				break;
			}

			int x = argsWithoutFunc[0]->GetInt();
			int y = argsWithoutFunc[1]->GetInt();

			SlBrowser::instance().m_widget->move(x, y);
			break;
		}
		case JavascriptApi::JS_BROWSER_SET_ALLOW_HIDE_BROWSER:
		{
			if (argsWithoutFunc.size() < 1)
			{
				jsonOutput = Json(Json::object({{"error", "Invalid parameters"}})).dump();
				break;
			}

			SlBrowser::instance().m_allowHideBrowser = argsWithoutFunc[0]->GetBool();
			break;
		}
		case JavascriptApi::JS_BROWSER_SET_HIDDEN_STATE:
		{
			if (argsWithoutFunc.size() < 1)
			{
				jsonOutput = Json(Json::object({{"error", "Invalid parameters"}})).dump();
				break;
			}

			SlBrowser::instance().m_widget->setHidden(argsWithoutFunc[0]->GetBool());
			SlBrowser::instance().saveHiddenState(SlBrowser::instance().m_widget->isHidden());

			if (!SlBrowser::instance().m_widget->isHidden())
			{
				HWND hwnd = HWND(SlBrowser::instance().m_widget->winId());
				WindowsFunctions::ForceForegroundWindow(hwnd);
			}

			break;
		}
		}

		CefRefPtr<CefProcessMessage> msg = CefProcessMessage::Create("executeCallback");
		CefRefPtr<CefListValue> execute_args = msg->GetArgumentList();
		execute_args->SetInt(0, funcid);
		execute_args->SetString(1, jsonOutput);

		SendBrowserProcessMessage(browser, PID_RENDERER, msg);
	}
	else
	{
		RegisterCallback(funcid, browser);

		if (!GrpcBrowser::instance().getClient()->send_js_api(name, cefListValueToJSONString(input_args)))
		{
			// todo; handle
			abort();
			return false;
		}
	}


	return true;
}

void BrowserClient::GetViewRect(CefRefPtr<CefBrowser>, CefRect &rect)
{
	if (!valid())
	{
		rect.Set(0, 0, 16, 16);
		return;
	}

	rect.Set(0, 0, 1920, 1080);
}

bool BrowserClient::OnTooltip(CefRefPtr<CefBrowser>, CefString &text)
{
	std::string str_text = text;
	QMetaObject::invokeMethod(QCoreApplication::instance()->thread(), [str_text]() { QToolTip::showText(QCursor::pos(), str_text.c_str()); });
	return true;
}

void BrowserClient::OnPaint(CefRefPtr<CefBrowser>, PaintElementType type, const RectList &, const void *buffer, int width, int height)
{
	if (type != PET_VIEW)
	{
		// TODO Overlay texture on top of bs->texture
		return;
	}

	if (!valid())
	{
		return;
	}
}

void BrowserClient::OnAudioStreamStarted(CefRefPtr<CefBrowser> browser, const CefAudioParameters &params_, int channels_)
{
	m_channels = channels_;
	m_channel_layout = (ChannelLayout)params_.channel_layout;
	m_sample_rate = params_.sample_rate;
	m_frames_per_buffer = params_.frames_per_buffer;
}

void BrowserClient::OnAudioStreamPacket(CefRefPtr<CefBrowser> browser, const float **data, int frames, int64_t pts)
{
	if (!valid())
	{
		return;
	}
}

void BrowserClient::OnAudioStreamStopped(CefRefPtr<CefBrowser> browser) {}

void BrowserClient::OnAudioStreamError(CefRefPtr<CefBrowser> browser, const CefString &message) {}

static CefAudioHandler::ChannelLayout Convert2CEFSpeakerLayout(int channels)
{
	switch (channels)
	{
	case 1:
		return CEF_CHANNEL_LAYOUT_MONO;
	case 2:
		return CEF_CHANNEL_LAYOUT_STEREO;
	case 3:
		return CEF_CHANNEL_LAYOUT_2_1;
	case 4:
		return CEF_CHANNEL_LAYOUT_4_0;
	case 5:
		return CEF_CHANNEL_LAYOUT_4_1;
	case 6:
		return CEF_CHANNEL_LAYOUT_5_1;
	case 8:
		return CEF_CHANNEL_LAYOUT_7_1;
	default:
		return CEF_CHANNEL_LAYOUT_UNSUPPORTED;
	}
}

bool BrowserClient::GetAudioParameters(CefRefPtr<CefBrowser> browser, CefAudioParameters &params)
{
	int channels = 2;
	params.channel_layout = CEF_CHANNEL_LAYOUT_MONO;
	params.sample_rate = 48000;
	params.frames_per_buffer = kFramesPerBuffer;
	return true;
}

void BrowserClient::OnLoadStart(CefRefPtr<CefBrowser> browser, CefRefPtr<CefFrame> frame, TransitionType transition_type)
{
	SlBrowser::instance().setMainLoadingInProgress(true);
}

void BrowserClient::OnLoadEnd(CefRefPtr<CefBrowser>, CefRefPtr<CefFrame> frame, int httpStatusCode)
{
	if (httpStatusCode == 200)
		SlBrowser::instance().setMainPageSuccess(true);

	SlBrowser::instance().setMainLoadingInProgress(false);
}

void BrowserClient::OnLoadError(CefRefPtr<CefBrowser> browser, CefRefPtr<CefFrame> frame, ErrorCode errorCode, const CefString &errorText, const CefString &failedUrl)
{

}

bool BrowserClient::OnConsoleMessage(CefRefPtr<CefBrowser>, cef_log_severity_t level, const CefString &message, const CefString &source, int line)
{
	return false;
}