2012-10-03 18:08:49 +04:00
|
|
|
/*============================================================================
|
|
|
|
CMake - Cross Platform Makefile Generator
|
|
|
|
Copyright 2012 Kitware, Inc.
|
|
|
|
|
|
|
|
Distributed under the OSI-approved BSD License (the "License");
|
|
|
|
see accompanying file Copyright.txt for details.
|
|
|
|
|
|
|
|
This software is distributed WITHOUT ANY WARRANTY; without even the
|
|
|
|
implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
|
|
|
See the License for more information.
|
|
|
|
============================================================================*/
|
|
|
|
|
|
|
|
#include "cmWIXSourceWriter.h"
|
|
|
|
|
|
|
|
#include <CPack/cmCPackGenerator.h>
|
|
|
|
|
|
|
|
#include <windows.h>
|
|
|
|
|
|
|
|
cmWIXSourceWriter::cmWIXSourceWriter(cmCPackLog* logger,
|
|
|
|
const std::string& filename,
|
|
|
|
bool isIncludeFile):
|
|
|
|
Logger(logger),
|
2013-12-20 21:12:01 +04:00
|
|
|
File(filename.c_str()),
|
|
|
|
State(DEFAULT),
|
|
|
|
SourceFilename(filename)
|
2012-10-03 18:08:49 +04:00
|
|
|
{
|
|
|
|
WriteXMLDeclaration();
|
|
|
|
|
|
|
|
if(isIncludeFile)
|
|
|
|
{
|
|
|
|
BeginElement("Include");
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
BeginElement("Wix");
|
|
|
|
}
|
|
|
|
|
|
|
|
AddAttribute("xmlns", "http://schemas.microsoft.com/wix/2006/wi");
|
|
|
|
}
|
|
|
|
|
|
|
|
cmWIXSourceWriter::~cmWIXSourceWriter()
|
|
|
|
{
|
2013-12-20 21:12:01 +04:00
|
|
|
if(Elements.size() > 1)
|
2012-10-03 18:08:49 +04:00
|
|
|
{
|
2013-10-23 14:14:39 +04:00
|
|
|
cmCPackLogger(cmCPackLog::LOG_ERROR,
|
2013-12-20 21:12:01 +04:00
|
|
|
Elements.size() - 1 << " WiX elements were still open when closing '" <<
|
|
|
|
SourceFilename << "'" << std::endl);
|
2013-10-23 14:14:39 +04:00
|
|
|
return;
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
2013-10-23 14:14:39 +04:00
|
|
|
|
2013-12-20 21:12:01 +04:00
|
|
|
EndElement(Elements.back());
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
void cmWIXSourceWriter::BeginElement(const std::string& name)
|
|
|
|
{
|
2013-12-20 21:12:01 +04:00
|
|
|
if(State == BEGIN)
|
2012-10-03 18:08:49 +04:00
|
|
|
{
|
2013-12-20 21:12:01 +04:00
|
|
|
File << ">";
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
|
|
|
|
2013-12-20 21:12:01 +04:00
|
|
|
File << "\n";
|
|
|
|
Indent(Elements.size());
|
|
|
|
File << "<" << name;
|
2012-10-03 18:08:49 +04:00
|
|
|
|
2013-12-20 21:12:01 +04:00
|
|
|
Elements.push_back(name);
|
|
|
|
State = BEGIN;
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
|
|
|
|
2013-10-23 14:14:39 +04:00
|
|
|
void cmWIXSourceWriter::EndElement(std::string const& name)
|
2012-10-03 18:08:49 +04:00
|
|
|
{
|
2013-12-20 21:12:01 +04:00
|
|
|
if(Elements.empty())
|
2012-10-03 18:08:49 +04:00
|
|
|
{
|
|
|
|
cmCPackLogger(cmCPackLog::LOG_ERROR,
|
2013-10-23 14:14:39 +04:00
|
|
|
"can not end WiX element with no open elements in '" <<
|
2013-12-20 21:12:01 +04:00
|
|
|
SourceFilename << "'" << std::endl);
|
2013-10-23 14:14:39 +04:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2013-12-20 21:12:01 +04:00
|
|
|
if(Elements.back() != name)
|
2013-10-23 14:14:39 +04:00
|
|
|
{
|
|
|
|
cmCPackLogger(cmCPackLog::LOG_ERROR,
|
2013-12-20 21:12:01 +04:00
|
|
|
"WiX element <" << Elements.back() <<
|
2013-10-23 14:14:39 +04:00
|
|
|
"> can not be closed by </" << name << "> in '" <<
|
2013-12-20 21:12:01 +04:00
|
|
|
SourceFilename << "'" << std::endl);
|
2012-10-03 18:08:49 +04:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2013-12-20 21:12:01 +04:00
|
|
|
if(State == DEFAULT)
|
2012-10-03 18:08:49 +04:00
|
|
|
{
|
2013-12-20 21:12:01 +04:00
|
|
|
File << "\n";
|
|
|
|
Indent(Elements.size()-1);
|
|
|
|
File << "</" << Elements.back() << ">";
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2013-12-20 21:12:01 +04:00
|
|
|
File << "/>";
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
|
|
|
|
2013-12-20 21:12:01 +04:00
|
|
|
Elements.pop_back();
|
|
|
|
State = DEFAULT;
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
void cmWIXSourceWriter::AddProcessingInstruction(
|
|
|
|
const std::string& target, const std::string& content)
|
|
|
|
{
|
2013-12-20 21:12:01 +04:00
|
|
|
if(State == BEGIN)
|
2012-10-03 18:08:49 +04:00
|
|
|
{
|
2013-12-20 21:12:01 +04:00
|
|
|
File << ">";
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
|
|
|
|
2013-12-20 21:12:01 +04:00
|
|
|
File << "\n";
|
|
|
|
Indent(Elements.size());
|
|
|
|
File << "<?" << target << " " << content << "?>";
|
2012-10-03 18:08:49 +04:00
|
|
|
|
2013-12-20 21:12:01 +04:00
|
|
|
State = DEFAULT;
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
void cmWIXSourceWriter::AddAttribute(
|
|
|
|
const std::string& key, const std::string& value)
|
|
|
|
{
|
|
|
|
std::string utf8 = WindowsCodepageToUtf8(value);
|
|
|
|
|
2013-12-20 21:12:01 +04:00
|
|
|
File << " " << key << "=\"" << EscapeAttributeValue(utf8) << '"';
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
|
|
|
|
2013-11-20 00:38:09 +04:00
|
|
|
void cmWIXSourceWriter::AddAttributeUnlessEmpty(
|
|
|
|
const std::string& key, const std::string& value)
|
|
|
|
{
|
|
|
|
if(value.size())
|
|
|
|
{
|
|
|
|
AddAttribute(key, value);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-10-03 18:08:49 +04:00
|
|
|
std::string cmWIXSourceWriter::WindowsCodepageToUtf8(const std::string& value)
|
|
|
|
{
|
|
|
|
if(value.empty())
|
|
|
|
{
|
|
|
|
return std::string();
|
|
|
|
}
|
|
|
|
|
|
|
|
int characterCount = MultiByteToWideChar(
|
2012-12-03 22:35:29 +04:00
|
|
|
CP_ACP, 0, value.c_str(), static_cast<int>(value.size()), 0, 0);
|
2012-10-03 18:08:49 +04:00
|
|
|
|
|
|
|
if(characterCount == 0)
|
|
|
|
{
|
|
|
|
return std::string();
|
|
|
|
}
|
|
|
|
|
|
|
|
std::vector<wchar_t> utf16(characterCount);
|
|
|
|
|
|
|
|
MultiByteToWideChar(
|
2012-12-03 22:35:29 +04:00
|
|
|
CP_ACP, 0, value.c_str(), static_cast<int>(value.size()),
|
|
|
|
&utf16[0], static_cast<int>(utf16.size()));
|
2012-10-03 18:08:49 +04:00
|
|
|
|
2012-12-03 22:35:29 +04:00
|
|
|
int utf8ByteCount = WideCharToMultiByte(
|
|
|
|
CP_UTF8, 0, &utf16[0], static_cast<int>(utf16.size()), 0, 0, 0, 0);
|
2012-10-03 18:08:49 +04:00
|
|
|
|
|
|
|
if(utf8ByteCount == 0)
|
|
|
|
{
|
|
|
|
return std::string();
|
|
|
|
}
|
|
|
|
|
|
|
|
std::vector<char> utf8(utf8ByteCount);
|
|
|
|
|
2012-12-03 22:35:29 +04:00
|
|
|
WideCharToMultiByte(CP_UTF8, 0, &utf16[0], static_cast<int>(utf16.size()),
|
|
|
|
&utf8[0], static_cast<int>(utf8.size()), 0, 0);
|
2012-10-03 18:08:49 +04:00
|
|
|
|
|
|
|
return std::string(&utf8[0], utf8.size());
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void cmWIXSourceWriter::WriteXMLDeclaration()
|
|
|
|
{
|
2013-12-20 21:12:01 +04:00
|
|
|
File << "<?xml version=\"1.0\" encoding=\"UTF-8\"?>" << std::endl;
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
|
|
|
|
2012-12-04 23:37:41 +04:00
|
|
|
void cmWIXSourceWriter::Indent(size_t count)
|
2012-10-03 18:08:49 +04:00
|
|
|
{
|
2012-12-04 23:37:41 +04:00
|
|
|
for(size_t i = 0; i < count; ++i)
|
2012-10-03 18:08:49 +04:00
|
|
|
{
|
2013-12-20 21:12:01 +04:00
|
|
|
File << " ";
|
2012-10-03 18:08:49 +04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string cmWIXSourceWriter::EscapeAttributeValue(
|
|
|
|
const std::string& value)
|
|
|
|
{
|
|
|
|
std::string result;
|
|
|
|
result.reserve(value.size());
|
|
|
|
|
|
|
|
char c = 0;
|
2012-12-04 23:37:41 +04:00
|
|
|
for(size_t i = 0 ; i < value.size(); ++i)
|
2012-10-03 18:08:49 +04:00
|
|
|
{
|
|
|
|
c = value[i];
|
|
|
|
switch(c)
|
|
|
|
{
|
|
|
|
case '<':
|
|
|
|
result += "<";
|
|
|
|
break;
|
2013-10-23 14:14:39 +04:00
|
|
|
case '>':
|
|
|
|
result += ">";
|
|
|
|
break;
|
2012-10-03 18:08:49 +04:00
|
|
|
case '&':
|
|
|
|
result +="&";
|
|
|
|
break;
|
|
|
|
case '"':
|
|
|
|
result += """;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
result += c;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|