module AnyCable::WithConnectionState

def cstate
  env.cstate
end

def cstate=(val)
  env.cstate = val
end

def initialize(session: nil, **other)
  if session
    other[:cstate] ||= {}
    other[:cstate][SESSION_KEY] = session
  end
  super(**other)
end

def istate
  env.istate
end

def istate=(val)
  env.istate = val
end

def session
  state_ = cstate
  if state_
    state_[SESSION_KEY]
  end
end

def session=(val)
  self.cstate = {} unless cstate
  state_ = cstate
  if state_
    state_[SESSION_KEY] = val
  end
end