dumpctx/ctx.go at master · go-x-pkg/dumpctx · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package dumpctx

import (
	"fmt"
	"io"
)

type Ctx struct {
	indent string
	isList bool
}

func (ctx *Ctx) SetIsList()   { ctx.isList = true }
func (ctx *Ctx) UnsetIsList() { ctx.isList = false }

func (ctx *Ctx) Indent() string { return ctx.indent }
func (ctx *Ctx) Enter()         { ctx.indent += IndentToken }
func (ctx *Ctx) EnterList()     { ctx.Enter(); ctx.isList = true }

func (ctx *Ctx) Leave() {
	if len(ctx.indent) == 0 {
		return
	}

	ctx.indent = ctx.indent[:len(ctx.indent)-len(IndentToken)]
}
func (ctx *Ctx) LeaveList() { ctx.Leave(); ctx.isList = false }

func (ctx *Ctx) NoList(cb func()) {
	isList := ctx.isList

	// clear is list if ctx is set
	if isList {
		ctx.isList = false
	}

	cb()

	// restore
	if isList {
		ctx.isList = true
	}
}

func (ctx *Ctx) EmitPrefix(w io.Writer) {
	fmt.Fprint(w, ctx.indent)

	if ctx.isList {
		fmt.Fprintf(w, "%c ", ListSeparator)
	} else {
		fmt.Fprint(w, IndentToken)
	}
}

func (ctx *Ctx) WrapList(cb func()) {
	ctx.EnterList()
	cb()
	ctx.LeaveList()
}

func (ctx *Ctx) Wrap(cb func()) {
	ctx.Enter()
	cb()
	ctx.Leave()
}

func (ctx *Ctx) Copy(other *Ctx) {
	ctx.indent = other.indent
	ctx.isList = other.isList
}

func (ctx *Ctx) Init() {
	ctx.indent = ""
	ctx.isList = false
}