/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * FreeRDP Proxy Server
 *
 * Copyright 2019 Kobi Mizrachi <kmizrachi18@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <freerdp/server/disp.h>

#include "pf_disp.h"
#include "pf_log.h"

#define TAG PROXY_TAG("disp")

BOOL pf_server_disp_init(pServerContext* ps)
{
	DispServerContext* disp;
	disp = ps->disp = disp_server_context_new(ps->vcm);

	if (!disp)
	{
		return FALSE;
	}

	disp->rdpcontext = (rdpContext*)ps;
	return TRUE;
}

static UINT pf_disp_monitor_layout(DispServerContext* context,
                                   const DISPLAY_CONTROL_MONITOR_LAYOUT_PDU* pdu)
{
	proxyData* pdata = (proxyData*) context->custom;
	DispClientContext* client = (DispClientContext*) pdata->pc->disp;
	WLog_DBG(TAG, __FUNCTION__);
	return client->SendMonitorLayout(client, pdu->NumMonitors, pdu->Monitors);
}

static UINT pf_disp_on_caps_control(DispClientContext* context, UINT32 MaxNumMonitors,
                                    UINT32 MaxMonitorAreaFactorA,
                                    UINT32 MaxMonitorAreaFactorB)
{
	proxyData* pdata = (proxyData*) context->custom;
	DispServerContext* server = (DispServerContext*) pdata->ps->disp;
	WLog_DBG(TAG, __FUNCTION__);
	/* Update caps of proxy's disp server */
	server->MaxMonitorAreaFactorA = MaxMonitorAreaFactorA;
	server->MaxMonitorAreaFactorB = MaxMonitorAreaFactorB;
	server->MaxNumMonitors = MaxNumMonitors;
	/* Send CapsControl to client */
	return server->DisplayControlCaps(server);
}

void pf_disp_register_callbacks(DispClientContext* client, DispServerContext* server,
                           proxyData* pdata)
{
	client->custom = (void*) pdata;
	server->custom = (void*) pdata;
	/* client receives from server, forward using disp server to original client */
	client->DisplayControlCaps = pf_disp_on_caps_control;
	/* server receives from client, forward to target server using disp client */
	server->DispMonitorLayout = pf_disp_monitor_layout;
}