frc971/output/CameraServer.cc

#include <string.h>

#include "aos/atom_code/output/HTTPServer.h"
#include "aos/atom_code/output/evhttp_ctemplate_emitter.h"
#include "aos/atom_code/output/ctemplate_cache.h"
#include "aos/common/Configuration.h"
#include "aos/common/messages/RobotState.q.h"
#include "ctemplate/template.h"
#include "aos/atom_code/init.h"
#include "aos/common/logging/logging.h"

#include "frc971/constants.h"

RegisterTemplateFilename(ROBOT_HTML, "robot.html.tpl");

namespace frc971 {

class CameraServer : public aos::http::HTTPServer {
 public:
  CameraServer() : HTTPServer(aos::configuration::GetRootDirectory(), 8080),
      buf_(NULL) {
    AddPage<CameraServer>("/robot.html", &CameraServer::RobotHTML, this);
  }

 private:
  evbuffer *buf_;
  bool Setup(evhttp_request *request, const char *content_type) {
    if (evhttp_add_header(evhttp_request_get_output_headers(request),
                          "Content-Type", content_type) == -1) {
      LOG(WARNING, "adding Content-Type failed\n");
      evhttp_send_error(request, HTTP_INTERNAL, NULL);
      return false;
    }
    if (buf_ == NULL) buf_ = evbuffer_new();
    if (buf_ == NULL) {
      LOG(WARNING, "evbuffer_new() failed\n");
      evhttp_send_error(request, HTTP_INTERNAL, NULL);
      return false;
    }
    return true;
  }
  void RobotHTML(evhttp_request *request) {
    if (!Setup(request, "text/html")) return;

    ctemplate::TemplateDictionary dict(ROBOT_HTML);
    const char *host = evhttp_find_header(
        evhttp_request_get_input_headers(request), "Host");
    if (host == NULL) {
      evhttp_send_error(request, HTTP_BADREQUEST, "no Host header");
      return;
    }
    const char *separator = strchrnul(host, ':');
    size_t length = separator - host;
    // Don't include the last ':' (or the terminating '\0') or anything else
    // after it.
    dict.SetValue("HOST", ctemplate::TemplateString(host, length));

    if (!aos::robot_state.FetchLatest()) {
      LOG(WARNING, "getting a RobotState message failed\n");
      evhttp_send_error(request, HTTP_INTERNAL, NULL);
      return;
    }
    int center;
    if (!constants::camera_center(&center)) {
      evhttp_send_error(request, HTTP_INTERNAL, NULL);
      return;
    }
    dict.SetIntValue("CENTER", center);

    aos::http::EvhttpCtemplateEmitter emitter(buf_);
    if (!aos::http::get_template_cache()->
        ExpandWithData(ROBOT_HTML, ctemplate::STRIP_WHITESPACE,
                       &dict, NULL, &emitter)) {
      LOG(ERROR, "expanding the template failed\n");
      evhttp_send_error(request, HTTP_INTERNAL, NULL);
      return;
    }
    if (emitter.error()) {
      evhttp_send_error(request, HTTP_INTERNAL, NULL);
      return;
    }
    evhttp_send_reply(request, HTTP_OK, NULL, buf_);
  }
};

}  // namespace frc971

int main() {
  ::aos::InitNRT();
  ::frc971::CameraServer server;
  server.Run();
  ::aos::Cleanup();
}