| 88 | } |
| 89 | |
| 90 | func (f *customHeaderFlag) Set(value string) error { |
| 91 | name, headerValue, ok := strings.Cut(value, ":") |
| 92 | if !ok { |
| 93 | *f.err = errors.Errorf("invalid custom-header, must be in 'Name: value' format") |
| 94 | return nil |
| 95 | } |
| 96 | |
| 97 | name = textproto.CanonicalMIMEHeaderKey(strings.TrimSpace(name)) |
| 98 | if !httpguts.ValidHeaderFieldName(name) { |
| 99 | *f.err = errors.Errorf("invalid custom-header name") |
| 100 | return nil |
| 101 | } |
| 102 | if strings.EqualFold(name, "Authorization") { |
| 103 | *f.err = errors.Errorf("custom-header Authorization is not allowed because bytebase-action manages Bytebase authorization") |
| 104 | return nil |
| 105 | } |
| 106 | if !httpguts.ValidHeaderFieldValue(headerValue) { |
| 107 | *f.err = errors.Errorf("invalid custom-header value") |
| 108 | return nil |
| 109 | } |
| 110 | |
| 111 | f.headers.Add(name, strings.TrimSpace(headerValue)) |
| 112 | return nil |
| 113 | } |
| 114 | |
| 115 | func (f *customHeaderFlag) String() string { |
| 116 | if f == nil || f.headers == nil { |