diff options
Diffstat (limited to 'vendor/gitlab.com/golang-commonmark/markdown/writer.go')
-rw-r--r-- | vendor/gitlab.com/golang-commonmark/markdown/writer.go | 69 |
1 files changed, 69 insertions, 0 deletions
diff --git a/vendor/gitlab.com/golang-commonmark/markdown/writer.go b/vendor/gitlab.com/golang-commonmark/markdown/writer.go new file mode 100644 index 00000000..886a1de4 --- /dev/null +++ b/vendor/gitlab.com/golang-commonmark/markdown/writer.go @@ -0,0 +1,69 @@ +// Copyright 2015 The Authors. All rights reserved. +// Use of this source code is governed by a BSD-style +// license that can be found in the LICENSE file. + +package markdown + +import ( + "bufio" + "io" +) + +type writer interface { + Write([]byte) (int, error) + WriteByte(byte) error + WriteString(string) (int, error) + Flush() error +} + +type monadicWriter struct { + writer + err error +} + +func newMonadicWriter(w io.Writer) *monadicWriter { + if w, ok := w.(writer); ok { + return &monadicWriter{writer: w} + } + return &monadicWriter{writer: bufio.NewWriter(w)} +} + +func (w *monadicWriter) Write(p []byte) (n int, err error) { + if w.err != nil { + return + } + + n, err = w.writer.Write(p) + w.err = err + return +} + +func (w *monadicWriter) WriteByte(b byte) (err error) { + if w.err != nil { + return + } + + err = w.writer.WriteByte(b) + w.err = err + return +} + +func (w *monadicWriter) WriteString(s string) (n int, err error) { + if w.err != nil { + return + } + + n, err = w.writer.WriteString(s) + w.err = err + return +} + +func (w *monadicWriter) Flush() (err error) { + if w.err != nil { + return + } + + err = w.writer.Flush() + w.err = err + return +} |