back and forward by N

1 files changed, 11 insertions, 8 deletions

diff --git a/actions.go b/actions.go
index 6e5c7dd..4a51ee6 100644
--- a/actions.go
+++ b/actions.go
@@ -243,19 +243,22 @@ func back(state *BrowserState) error {
 	return nil
 }
 
-func Back(state *BrowserState) error {
-	if err := back(state); err != nil {
-		return err
+func Back(state *BrowserState, num int) error {
+	for i := 0; i < num; i += 1 {
+		if err := back(state); err != nil {
+			return err
+		}
 	}
-
 	return print(state)
 }
 
-func Forward(state *BrowserState) error {
-	if state.Forward == nil {
-		return ErrNoNextHistory
+func Forward(state *BrowserState, num int) error {
+	for i := 0; i < num; i += 1 {
+		if state.Forward == nil {
+			return ErrNoNextHistory
+		}
+		state.History = state.Forward
 	}
-	state.History = state.Forward
 	state.Modal = nil
 
 	return print(state)