mirror of
https://github.com/go-gitea/gitea.git
synced 2024-11-21 01:14:31 +08:00
31 lines
743 B
Go
31 lines
743 B
Go
|
// Copyright 2024 The Gitea Authors. All rights reserved.
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
|
||
|
|
package internal
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
type finalProcessor struct {
|
||
|
|
renderInternal *RenderInternal
|
||
|
|
|
||
|
|
output io.Writer
|
||
|
|
buf bytes.Buffer
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *finalProcessor) Write(data []byte) (int, error) {
|
||
|
|
p.buf.Write(data)
|
||
|
|
return len(data), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *finalProcessor) Close() error {
|
||
|
|
// TODO: reading the whole markdown isn't a problem at the moment,
|
||
|
|
// because "postProcess" already does so. In the future we could optimize the code to process data on the fly.
|
||
|
|
buf := p.buf.Bytes()
|
||
|
|
buf = bytes.ReplaceAll(buf, []byte(` data-attr-class="`+p.renderInternal.secureIDPrefix), []byte(` class="`))
|
||
|
|
_, err := p.output.Write(buf)
|
||
|
|
return err
|
||
|
|
}
|