esp8266_milight_hub/lib/WebServer/PatternHandler.cpp

#include <PatternHandler.h>

PatternHandler::PatternHandler(
    const String& pattern,
    const HTTPMethod method,
    const PatternHandler::TPatternHandlerFn fn
  ) : method(method), fn(fn), tokenPositions(NULL) {
  Vector<StringToken>* tokenPositions = new Vector<StringToken>();
  tokenize(pattern, tokenPositions);

  numPatternTokens = tokenPositions->size();
  patternTokens = new String[numPatternTokens];

  for (int i = 0; i < tokenPositions->size(); i++) {
    patternTokens[i] = (*tokenPositions)[i].extract(pattern);
  }

  delete tokenPositions;
}

bool PatternHandler::canHandle(HTTPMethod requestMethod, String requestUri) {
  if (this->method != HTTP_ANY && requestMethod != this->method) {
    return false;
  }

  if (tokenPositions) {
    delete tokenPositions;
  }

  bool canHandle = true;

  tokenPositions = new Vector<StringToken>();
  tokenize(requestUri, tokenPositions);

  if (numPatternTokens == tokenPositions->size()) {
    for (int i = 0; i < numPatternTokens; i++) {
      const StringToken urlTokenP = (*tokenPositions)[i];

      if (!patternTokens[i].startsWith(":")
        && patternTokens[i] != urlTokenP.extract(requestUri)) {
        canHandle = false;
        break;
      }
    }
  } else {
    canHandle = false;
  }

  return canHandle;
}

bool PatternHandler::handle(ESP8266WebServer& server, HTTPMethod requestMethod, String requestUri) {
  if (! canHandle(requestMethod, requestUri)) {
    return false;
  }

  UrlTokenBindings* bindings = new UrlTokenBindings(patternTokens, tokenPositions, requestUri);
  fn(bindings);

  delete bindings;
}

void PatternHandler::tokenize(const String& path, Vector<StringToken>* tokenPositions) {
  int lastStart = 0;
  int currentPosition = 0;

  for (int i = 0; i < path.length(); i++) {
    if (path.charAt(i) == '/' || i == path.length()-1) {
      // If we're in the last position, include the last character if it isn't
      // a '/'
      if (path.charAt(i) != '/') {
        currentPosition++;
      }

      if (lastStart > 0 && currentPosition > lastStart) {
        StringToken token(lastStart, currentPosition);
        tokenPositions->push_back(token);
      }

      lastStart = i+1;
    }

    currentPosition++;
  }
}