Merge branch 'lukeshu/reencode-refactor'

1 files changed, 39 insertions, 0 deletions

diff --git a/reencode_nl.go b/reencode_nl.go
new file mode 100644
index 0000000..b7a3cd9
--- /dev/null
+++ b/reencode_nl.go
@@ -0,0 +1,39 @@
+// Copyright (C) 2022-2023  Luke Shumaker <lukeshu@lukeshu.com>
+//
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+package lowmemjson
+
+import (
+	"git.lukeshu.com/go/lowmemjson/internal/jsonparse"
+)
+
+type reEncodeForceNL struct {
+	out reEncoderModule
+
+	// state
+	skipNL bool
+}
+
+var _ reEncoderModule = (*reEncodeForceNL)(nil)
+
+func (enc *reEncodeForceNL) PopWriteBarrier() {
+	enc.out.PopWriteBarrier()
+}
+
+func (enc *reEncodeForceNL) HandleRune(c rune, t jsonparse.RuneType, escape BackslashEscapeMode, stackSize int) error {
+	switch {
+	case t == jsonparse.RuneTypeEOF:
+		if err := enc.out.HandleRune('\n', jsonparse.RuneTypeSpace, 0, stackSize); err != nil {
+			return err
+		}
+		enc.skipNL = true
+		return enc.out.HandleRune(c, t, escape, stackSize)
+	case c == '\n' && t == jsonparse.RuneTypeSpace && enc.skipNL:
+		enc.skipNL = false
+		return nil
+	default:
+		enc.skipNL = false
+		return enc.out.HandleRune(c, t, escape, stackSize)
+	}
+}