Push/pop state, full json examplev0.0.2

1 files changed, 14 insertions, 0 deletions

diff --git a/statefunc.go b/statefunc.go
index 5980ecc..734fe57 100644
--- a/statefunc.go
+++ b/statefunc.go
@@ -2,3 +2,17 @@ package unilex
 
 // StateFunc represents function that scans lexems and returns new state function or nil if lexing completed.
 type StateFunc func(*Lexer) StateFunc
+
+type stateStack []StateFunc
+
+func (ss *stateStack) Push(s StateFunc) {
+	*ss = append(*ss, s)
+}
+
+func (ss *stateStack) Pop() (s StateFunc) {
+	if len(*ss) == 0 {
+		return nil
+	}
+	*ss, s = (*ss)[:len(*ss)-1], (*ss)[len(*ss)-1]
+	return s
+}