module Scrivito

class WorkspaceSelectionMiddleware
  def initialize(app)
    @app = app
  end

  def call(env)
    Workspace.current = proc do
      env[EditingContextMiddleware::ENVKEY].visible_workspace
    end
    @app.call(env)
  end

  private

  def session(env)
    env[Rack::Session::Abstract::ENV_SESSION_KEY]
  end
end

end # module Scrivito