$LOAD_PATH.unshift(File.expand_path('lib', __dir__)) require 'prism' require 'net/http' require 'zlib' require 'stringio' require 'csv' require_relative 'examples/blog/app' server = Prism::Server.new(Blog, port: 0).start port = server.port puts "blog demo server on port #{port}" puts "database file: #{File.join(__dir__, 'examples/blog/db/development.pstore')}" log_path = File.join(__dir__, 'examples/blog/log/development.log') log_offset_before_requests = File.exist?(log_path) ? File.size(log_path) : 0 puts '-' * 40 def cookie_from(response) lines = response.get_fields('set-cookie') || [] lines.map { |line| line.split(';').first }.join('; ') end def cookie_value_from(response, name) lines = response.get_fields('set-cookie') || [] line = lines.find { |l| l.start_with?("#{name}=") } line&.split(';')&.first end def csrf_token_from(html) html[/name="csrf-token" content="([^"]+)"/, 1] end def multipart_body(boundary, fields:, files: {}) parts = fields.map do |name, value| "--#{boundary}\r\nContent-Disposition: form-data; name=\"#{name}\"\r\n\r\n#{value}\r\n" end parts += files.map do |name, (filename, content_type, content)| "--#{boundary}\r\nContent-Disposition: form-data; name=\"#{name}\"; filename=\"#{filename}\"\r\n" \ "Content-Type: #{content_type}\r\n\r\n#{content}\r\n" end parts.join + "--#{boundary}--\r\n" end # Signs in as the writer from a brand new, entirely independent cookie jar # (its own session cookie, its own CSRF token fetched fresh from the # sign-in page) - a second real HTTP client would look exactly like this, # unlike reusing the caller's own `cookie`/`token`. Returns the # remember_session_token cookie this "browser" got back, proving # Prism::Authenticatable's remember_sessions: hands out one independent # LoginSession row per device instead of one shared column. def sign_in_new_device(http, label, user_agent:) new_session_page = http.get('/session/new', 'User-Agent' => user_agent) device_cookie = cookie_from(new_session_page) device_token = csrf_token_from(new_session_page.body) body = URI.encode_www_form( 'user[email]' => 'writer@example.com', 'user[password]' => 'correct horse battery staple', 'user[remember_me]' => '1', 'authenticity_token' => device_token ) signed_in = http.post('/session.json', body, 'Content-Type' => 'application/x-www-form-urlencoded', 'Cookie' => device_cookie, 'User-Agent' => user_agent) raise "expected #{label} to sign in, got #{signed_in.code}" unless signed_in.code == '200' remember_cookie = cookie_value_from(signed_in, 'remember_session_token') raise "expected #{label}'s sign-in to set its own remember_session_token cookie" unless remember_cookie puts " #{label} signed in from its own independent cookie jar -> #{signed_in.code}, own remember-me cookie set" remember_cookie end Net::HTTP.start('127.0.0.1', port) do |http| home = http.get('/') cookie = cookie_from(home) token = csrf_token_from(home.body) puts "GET / -> #{home.code} (session cookie set, csrf token captured)" css_href = home.body[//, 1] js_src = home.body[/