/**
 * @file XTPMarkupXtpScript.cpp
 *
 * @copyright
 * (c) 1998-2025 Codejock Software, All Rights Reserved.
 *
 * This source file is the property of Codejock Software and must not be
 * redistributed by any means without the explicit written permission of
 * Codejock Software.
 *
 * The use of this source code is governed by the terms and conditions specified
 * in the Toolkit Pro license agreement. Codejock Software grants you, as a
 * single software developer, the limited right to use this software on one
 * computer only.
 *
 * Contact Information:
 * support@codejock.com
 * http://www.codejock.com
 *
 */

#include "StdAfx.h"

#include "Common/XTPTypeId.h"
#include "Common/XTPCasting.h"
#include "Common/XTPFramework.h"
#include "Common/XTPSystemHelpers.h"
#include "Common/XTPSynchro.h"
#include "Common/XTPApplication.h"
#include "Common/XTPSingleton.h"
#include "Common/XTPResourceManager.h"

#include <atlbase.h>
#include "Markup/XTPMarkupObject.h"
#include "Markup/XTPMarkupString.h"
#include "Markup/XTPMarkupContext.h"
#include "Markup/Extensions/XTPMarkupExtension.h"
#include "Markup/XTPMarkupBuilder.h"
#include "Markup/Extensions/XTPMarkupXtpScript.h"

#include "Common/Base/Diagnostic/XTPDisableNoisyWarnings.h"

#ifdef _DEBUG
#	undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#	define new DEBUG_NEW
#endif

CXTPMarkupDependencyProperty* CXTPMarkupXtpScript::m_pPropertyLanguage = NULL;
CXTPMarkupDependencyProperty* CXTPMarkupXtpScript::m_pPropertySource   = NULL;
CXTPMarkupDependencyProperty* CXTPMarkupXtpScript::m_pPropertyRequires = NULL;

IMPLEMENT_MARKUPCLASS(L"xtp:Script", CXTPMarkupXtpScript, CXTPMarkupExtension);

void CXTPMarkupXtpScript::RegisterMarkupClass()
{
	m_pPropertyLanguage = CXTPMarkupDependencyProperty::Register(L"Language",
																 MARKUP_TYPE(CXTPMarkupString),
																 MARKUP_TYPE(CXTPMarkupXtpScript));

	m_pPropertySource = CXTPMarkupDependencyProperty::Register(L"Source",
															   MARKUP_TYPE(CXTPMarkupString),
															   MARKUP_TYPE(CXTPMarkupXtpScript));

	m_pPropertyRequires = CXTPMarkupDependencyProperty::Register(L"Requires",
																 MARKUP_TYPE(CXTPMarkupString),
																 MARKUP_TYPE(CXTPMarkupXtpScript));
}

LPCWSTR CXTPMarkupXtpScript::GetLanguage() const
{
	CXTPMarkupString* pLanguage = MARKUP_DYNAMICCAST(CXTPMarkupString,
													 GetValue(m_pPropertyLanguage));
	LPCWSTR lpszLanguage		= (NULL != pLanguage ? (LPCWSTR)*pLanguage : L"JScript");
	return lpszLanguage;
}

LPCWSTR CXTPMarkupXtpScript::GetSource() const
{
	CXTPMarkupString* pSource = MARKUP_DYNAMICCAST(CXTPMarkupString, GetValue(m_pPropertySource));
	LPCWSTR lpszSource		  = NULL;
	if (NULL != pSource)
		lpszSource = (LPCWSTR)*pSource;

	return lpszSource;
}

LPCWSTR CXTPMarkupXtpScript::GetRequires() const
{
	CXTPMarkupString* pList = MARKUP_DYNAMICCAST(CXTPMarkupString, GetValue(m_pPropertyRequires));
	LPCWSTR lpszList		= NULL;
	if (NULL != pList)
		lpszList = (LPCWSTR)*pList;

	return lpszList;
}

void CXTPMarkupXtpScript::SetContentObject(CXTPMarkupBuilder* pBuilder, CXTPMarkupObject* pContent)
{
	_ASSERTE(NULL != pBuilder);
	_ASSERTE(NULL != pContent);

	if (IsStringObject(pContent))
	{
		if (NULL == GetSource())
		{
			CXTPMarkupString* pCode = static_cast<CXTPMarkupString*>(pContent);
			m_strCode.Append((LPCWSTR)*pCode, pCode->GetLength());
		}
		else
		{
			pBuilder->ThrowBuilderException(
				_T("xtp:Script does not allow combining script source and content"));
		}
	}
	else
	{
		CXTPMarkupExtension::SetContentObject(pBuilder, pContent);
	}
}

void CXTPMarkupXtpScript::OnLoaded(CXTPMarkupBuilder* pBuilder)
{
	_ASSERTE(NULL != pBuilder);

	LPCWSTR pLangauge = GetLanguage();
	LPCWSTR pSource   = GetSource();
	LPCWSTR pRequires = GetRequires();

	if (NULL == pSource)
	{
		m_pMarkupContext->LoadScriptCodeChunk(pBuilder, pLangauge, m_strCode, pRequires);
		return;
	}

	CString strSource = XTP_CW2CT(pSource);
	XTP_URI uri;
	HRESULT hr = XTPParseUri(strSource, &uri);
	if (FAILED(hr))
		pBuilder->ThrowBuilderException(hr);

	CComPtr<IStream> pStream;
	hr = XTPCreateReadOnlyStreamOnUri(&uri, &pStream);
	if (FAILED(hr))
		pBuilder->ThrowBuilderException(hr);

	LPVOID pCodeData  = NULL;
	SIZE_T cbCodeData = 0;
	hr				  = XTPReadStreamData(pStream, pCodeData, cbCodeData);
	if (FAILED(hr))
		pBuilder->ThrowBuilderException(hr);

	LPWSTR pCode = reinterpret_cast<LPWSTR>(malloc(cbCodeData + sizeof(WCHAR)));
	if (NULL == pCode)
	{
		XTPReleaseReadStreamData(pCodeData);
		return;
	}

	UINT cchCode = 0;
	if (!XTPLoadUnicodeText(pCodeData, XTPToUIntChecked(cbCodeData), pCode, &cchCode)
		|| 0 == cchCode)
	{
		XTPReleaseReadStreamData(pCodeData);
		free(pCode);
		return;
	}

	m_pMarkupContext->LoadScriptCodeChunk(pBuilder, pLangauge, pCode);

	XTPReleaseReadStreamData(pCodeData);
	free(pCode);
}