XMLParserPrint.h

Go to the documentation of this file.

#pragma once
 
#include "XMLParser.h"
#include "basic/StrPrint.h"
 
namespace tiny_dlna {
 
class XMLParserPrint : public Print {
 public:
  XMLParserPrint() {
    p.setCallback(&XMLParserPrint::wrapperCallback);
    p.setReference(&cbref);
    p.setReportTextOnly(false);
  }
 
  size_t write(uint8_t ch) override { return buffer.write(ch); }
 
  size_t write(const uint8_t* data, size_t size) override {
    return buffer.write(data, size);
  }
 
  void setExpandEncoded(bool flag) { buffer.setExpandEncoded(flag); }
 
  bool parse(Str& outNodeName, Vector<Str>& outPath, Str& outText,
             Str& outAttributes) {
    const char* data = buffer.c_str();
    size_t total_len = buffer.length();
    if (data == nullptr || total_len == 0) return false;
 
    cbref.outNode = &outNodeName;
    cbref.outPath = &outPath;
    cbref.outText = &outText;
    cbref.outAttrs = &outAttributes;
    // clear previous outputs so callers get an empty result on failure
    if (cbref.outNode) cbref.outNode->clear();
    if (cbref.outPath) cbref.outPath->resize(0);
    if (cbref.outText) cbref.outText->clear();
    if (cbref.outAttrs) cbref.outAttrs->clear();
 
    p.setXml(data);
    // mark as not-yet-found
    cbref.start = -1;
    cbref.len = 0;
    bool parsed = p.parseSingle();
    buffer.consume(p.getParsePos());
    return parsed;
  }
 
  void end() {
    p.end();
    buffer.reset();
    if (cbref.outNode) cbref.outNode->clear();
    if (cbref.outPath) cbref.outPath->resize(0);
    if (cbref.outText) cbref.outText->clear();
    if (cbref.outAttrs) cbref.outAttrs->clear();
  }
 
  const char* c_str() { return buffer.c_str(); }
 
  size_t length() { return buffer.length(); }
 
 protected:
  StrPrint buffer;
 
  XMLParser p;
 
  struct CBRef {
    int start = 0;
    int len = 0;
    Str* outNode = nullptr;
    Vector<Str>* outPath = nullptr;
    Str* outText = nullptr;
    Str* outAttrs = nullptr;
  } cbref;
 
  static void wrapperCallback(Str& nodeName, Vector<Str>& path, Str& text,
                              Str& attributes, int start, int len, void* ref) {
    CBRef* r = static_cast<CBRef*>(ref);
    if (!r) return;
    r->start = start;
    r->len = len;
    if (r->outNode) *(r->outNode) = nodeName;
    if (r->outText) *(r->outText) = text;
    if (r->outAttrs) {
      if (attributes.isEmpty())
        r->outAttrs->reset();
      else
        *(r->outAttrs) = attributes;
    }
    if (r->outPath) {
      r->outPath->resize(0);
      for (int i = 0; i < path.size(); ++i) {
        r->outPath->push_back(path[i]);
      }
    }
  }
};
 
}  // namespace tiny_dlna