$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[/