//===--- XMLUtils.h - Various XML utility routines --------------*- C++ -*-===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

#ifndef SWIFT_MARKUP_XML_UTILS_H
#define SWIFT_MARKUP_XML_UTILS_H

#include "llvm/ADT/StringRef.h"
#include "llvm/Support/raw_ostream.h"

namespace swift {
namespace markup {

// FIXME: copied from Clang's
// CommentASTToXMLConverter::appendToResultWithXMLEscaping
static inline void appendWithXMLEscaping(raw_ostream &OS, StringRef S) {
  auto Start = S.begin(), Cursor = Start, End = S.end();
  for (; Cursor != End; ++Cursor) {
    switch (*Cursor) {
    case '&':
      OS.write(Start, Cursor - Start);
      OS << "&amp;";
      break;
    case '<':
      OS.write(Start, Cursor - Start);
      OS << "&lt;";
      break;
    case '>':
      OS.write(Start, Cursor - Start);
      OS << "&gt;";
      break;
    case '"':
      OS.write(Start, Cursor - Start);
      OS << "&quot;";
      break;
    case '\'':
      OS.write(Start, Cursor - Start);
      OS << "&apos;";
      break;
    default:
      continue;
    }
    Start = Cursor + 1;
  }
  OS.write(Start, Cursor - Start);
}

// FIXME: copied from Clang's
// CommentASTToXMLConverter::appendToResultWithCDATAEscaping
static inline void appendWithCDATAEscaping(raw_ostream &OS, StringRef S) {
  if (S.empty())
    return;

  OS << "<![CDATA[";
  while (!S.empty()) {
    size_t Pos = S.find("]]>");
    if (Pos == 0) {
      OS << "]]]]><![CDATA[>";
      S = S.drop_front(3);
      continue;
    }
    if (Pos == StringRef::npos)
      Pos = S.size();

    OS << S.substr(0, Pos);

    S = S.drop_front(Pos);
  }
  OS << "]]>";
}

} // namespace markup
} // namespace swift

#endif // SWIFT_MARKUP_XML_UTILS_H