class ActionController::RequestForgeryProtection::CookieStore

def fetch(request)
  contents = request.cookie_jar.encrypted[@cookie_name]
  return nil if contents.nil?
  value = JSON.parse(contents)
  return nil unless value.dig("session_id", "public_id") == request.session.id_was&.public_id
  value["token"]
rescue JSON::ParserError
  nil
end

def initialize(cookie = :csrf_token)
  @cookie_name = cookie
end

def reset(request)
  request.cookie_jar.delete(@cookie_name)
end

def store(request, csrf_token)
  request.cookie_jar.encrypted.permanent[@cookie_name] = {
    value: {
      token: csrf_token,
      session_id: request.session.id,
    }.to_json,
    httponly: true,
    same_site: :lax,
  }
end