Support file upload in server side.

webcc/common.h

@@ -0,0 +1,136 @@
+#ifndef WEBCC_COMMON_H_
+#define WEBCC_COMMON_H_
+
+#include <cassert>
+#include <string>
+#include <utility>
+#include <vector>
+
+namespace webcc {
+
+// -----------------------------------------------------------------------------
+
+// Split a string to two parts by the given token.
+bool Split2(const std::string& str, char token, std::string* part1,
+            std::string* part2);
+
+// -----------------------------------------------------------------------------
+
+typedef std::pair<std::string, std::string> HttpHeader;
+
+class HttpHeaders {
+public:
+  std::size_t size() const {
+    return headers_.size();
+  }
+
+  const std::vector<HttpHeader>& data() const {
+    return headers_;
+  }
+
+  void Set(const std::string& key, const std::string& value);
+
+  void Set(std::string&& key, std::string&& value);
+
+  bool Has(const std::string& key) const;
+
+  // Get header by index.
+  const HttpHeader& Get(std::size_t index) const {
+    assert(index < size());
+    return headers_[index];
+  }
+
+  // Get header value by key.
+  // If there's no such header with the given key, besides return empty, the
+  // optional |existed| parameter will be set to false.
+  const std::string& Get(const std::string& key, bool* existed = nullptr) const;
+
+  void Clear() {
+    headers_.clear();
+  }
+
+private:
+  std::vector<HttpHeader>::iterator Find(const std::string& key);
+
+  std::vector<HttpHeader> headers_;
+};
+
+// -----------------------------------------------------------------------------
+
+// Content-Type header.
+// Syntax:
+//   Content-Type: text/html; charset=utf-8
+//   Content-Type: multipart/form-data; boundary=something
+class ContentType {
+public:
+  explicit ContentType(const std::string& str = "");
+
+  void Parse(const std::string& str);
+
+  bool Valid() const;
+
+  bool multipart() const {
+    return multipart_;
+  }
+
+  const std::string& media_type() const {
+    return media_type_;
+  }
+
+  const std::string& charset() const {
+    assert(!multipart_);
+    return additional_;
+  }
+
+  const std::string& boundary() const {
+    assert(multipart_);
+    return additional_;
+  }
+
+private:
+  void Init(const std::string& str);
+
+private:
+  std::string media_type_;
+  std::string additional_;
+  bool multipart_ = false;
+};
+
+// -----------------------------------------------------------------------------
+
+// Content-Disposition header.
+// Syntax:
+//   Content-Disposition: form-data
+//   Content-Disposition: form-data; name="fieldName"
+//   Content-Disposition: form-data; name="fieldName"; filename="filename.jpg"
+// https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Content-Disposition
+class ContentDisposition {
+public:
+  explicit ContentDisposition(const std::string& str) {
+    valid_ = Init(str);
+  }
+
+  bool valid() const {
+    return valid_;
+  }
+
+  const std::string& name() const {
+    return name_;
+  }
+
+  const std::string& file_name() const {
+    return file_name_;
+  }
+
+private:
+  bool Init(const std::string& str);
+
+private:
+  std::string name_;
+  std::string file_name_;
+  bool valid_ = false;
+};
+
+}  // namespace webcc
+
+#endif  // WEBCC_COMMON_H_