warc/header.go

   1 /*
   2 tofuproxy -- flexible HTTP/WARC proxy with TLS certificates management
   3 Copyright (C) 2021 Sergey Matveev <stargrave@stargrave.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, version 3 of the License.
   8
   9 This program is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 GNU General Public License for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 */
  17
  18 package warc
  19
  20 import "strings"
  21
  22 type Header map[string]string
  23
  24 func splitKeyValue(line string) (string, string) {
  25         parts := strings.SplitN(line, ":", 2)
  26         if len(parts) != 2 {
  27                 return "", ""
  28         }
  29         return parts[0], strings.TrimSpace(parts[1])
  30 }
  31
  32 func NewHeader() Header {
  33         return make(map[string]string)
  34 }
  35
  36 func (h Header) Set(key, value string) {
  37         h[strings.ToLower(key)] = value
  38 }
  39
  40 func (h Header) Get(key string) string {
  41         return h[strings.ToLower(key)]
  42 }
  43
  44 func (h Header) Del(key string) {
  45         delete(h, strings.ToLower(key))
  46 }
  47
  48 func (h Header) AddLine(line string) {
  49         parts := strings.SplitN(line, ":", 2)
  50         if len(parts) != 2 {
  51                 return
  52         }
  53         h.Set(parts[0], strings.TrimSpace(parts[1]))
  54 }