fix ya.make

1 files changed, 55 insertions, 0 deletions

diff --git a/library/cpp/neh/http_headers.h b/library/cpp/neh/http_headers.h
new file mode 100644
index 0000000000..70cf3a9fbe
--- /dev/null
+++ b/library/cpp/neh/http_headers.h
@@ -0,0 +1,55 @@
+#pragma once
+
+#include <util/generic/strbuf.h>
+#include <util/stream/output.h>
+#include <util/string/ascii.h>
+
+namespace NNeh {
+    namespace NHttp {
+        template <typename Port>
+        void WriteHostHeader(IOutputStream& out, TStringBuf host, Port port) {
+            out << TStringBuf("Host: ") << host;
+            if (port) {
+                out << TStringBuf(":") << port;
+            }
+            out << TStringBuf("\r\n");
+        }
+
+        class THeaderSplitter {
+        public:
+            THeaderSplitter(TStringBuf headers)
+                : Headers_(headers)
+            {
+            }
+
+            bool Next(TStringBuf& header) {
+                while (Headers_.ReadLine(header)) {
+                    if (!header.Empty()) {
+                        return true;
+                    }
+                }
+                return false;
+            }
+        private:
+            TStringBuf Headers_;
+        };
+
+        inline bool HasHostHeader(TStringBuf headers) {
+            THeaderSplitter splitter(headers);
+            TStringBuf header;
+            while (splitter.Next(header)) {
+                if (AsciiHasPrefixIgnoreCase(header, "Host:")) {
+                    return true;
+                }
+            }
+            return false;
+        }
+
+        template <typename Port>
+        void WriteHostHeaderIfNot(IOutputStream& out, TStringBuf host, Port port, TStringBuf headers) {
+            if (!NNeh::NHttp::HasHostHeader(headers)) {
+                NNeh::NHttp::WriteHostHeader(out, host, port);
+            }
+        }
+    }
+}