warc/header.go

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