1 files changed, 22 insertions, 0 deletions
diff --git a/src/guff/handler.cr b/src/guff/handler.cr
index 357ed1f..52269a3 100644
--- a/src/guff/handler.cr
+++ b/src/guff/handler.cr
@@ -12,6 +12,28 @@ module Guff
       call_next(context)
     end
 
+    def with_session(context, &block : Hash(String, String) -> Bool)
+      sid, data = nil, nil
+      if sid = context.request.headers["x-guff-session-id"]?
+        data = Hash(String, String).from_json(@models.session.get(sid) || "{}")
+      else
+        sid = @models.session.add
+        context.request.headers["x-guff-session-id"] = sid
+        data = {} of String => String
+      end
+
+      save = block.call(data)
+
+      # save data
+      @models.session.set(sid, data.to_json) if save
+    end
+
+    private def get_session(context) : Hash(String, String)
+    end
+
+    private def save_session(sid, data)
+    end
+
     private def redirect(context, url : String)
       context.response.status_code = 302
       context.response.headers["location"] = url