fix sequence state

lib.go

@@ -145,6 +145,10 @@ func (w *Writer) flush() error {
 
 	r := w.editor.ReleaseState(w.state)
 	w.state = nil
+	if initializer, ok := w.editor.(interface{initialize() any}); ok {
+		w.state = initializer.initialize()
+	}
+
 	if _, err := w.out.Write([]byte(string(r))); err != nil {
 		w.err = err
 		return w.err

sequence.go

@@ -5,11 +5,7 @@ func sequenceEdit(e []Editor) func(rune, []any) ([]rune, []any) {
 		var sr []any
 		rr := []rune{r}
 		for i, ei := range e {
-			var si any
-			if len(s) > i {
-				si = s[i]
-			}
-
+			si := s[i]
 			rin := rr
 			rr = nil
 			for _, ri := range rin {
@@ -29,26 +25,11 @@ func sequenceRelease(e []Editor) func([]any) []rune {
 	return func(s []any) []rune {
 		var rr []rune
 		for i, ei := range e {
-			var r []rune
-			if i < len(s) {
-				r = ei.ReleaseState(s[i])
-			}
-
+			r := ei.ReleaseState(s[i])
 			for _, ri := range r {
-				var (
-					rri []rune
-					ssi []any
-				)
-
-				if i < len(s) {
-					ssi = s[i:]
-				}
-
-				rri, ssi = sequenceEdit(e[i+1:])(ri, ssi)
+				rri, ssi := sequenceEdit(e[i+1:])(ri, s[i + 1:])
 				rr = append(rr, rri...)
-				if i < len(s) {
-					s = append(s[:i], ssi...)
-				}
+				s = append(s[:i + 1], ssi...)
 			}
 		}
 

wrap.go

@@ -117,8 +117,7 @@ func wrapIndentRelease(first, rest []rune, firstWidth, restWidth int) func(wrapI
 			ret = append(ret, ' ')
 		}
 
-		ret = append(ret, state.currentWord...)
-		return ret
+		return append(ret, state.currentWord...)
 	}
 }