discourse/lib/faker/discourse_markdown.rb

97 lines
2.4 KiB
Ruby

# frozen_string_literal: true
require 'faker'
require 'net/http'
require 'json'
module Faker
class DiscourseMarkdown < Markdown
class << self
attr_writer(:user_id)
def user_id
@user_id || ::Discourse::SYSTEM_USER_ID
end
def with_user(user_id)
current_user_id = self.user_id
self.user_id = user_id
begin
yield
ensure
self.user_id = current_user_id
end
end
def image
image = next_image
image_file = load_image(image)
upload = ::UploadCreator.new(
image_file,
image[:filename],
origin: image[:url]
).create_for(user_id)
::UploadMarkdown.new(upload).to_markdown if upload.present? && upload.persisted?
rescue => e
STDERR.puts e
STDERR.puts e.backtrace.join("\n")
end
private
def next_image
if @images.blank?
if @stop_loading_images
@images = @all_images.dup
else
@next_page = (@next_page || 0) + 1
url = URI("https://picsum.photos/v2/list?page=#{@next_page}&limit=50")
response = Net::HTTP.get(url)
json = JSON.parse(response)
if json.blank?
@stop_loading_images = true
@images = @all_images.dup
else
@images = json.sort_by { |image| image["id"] }
@all_images = (@all_images || []).concat(@images)
end
end
end
image = @images.pop
{ filename: "#{image['id']}.jpg", url: "#{image['download_url']}.jpg" }
end
def image_cache_dir
@image_cache_dir ||= ::File.join(Rails.root, "tmp", "discourse_dev", "images")
end
def load_image(image)
cache_path = ::File.join(image_cache_dir, image[:filename])
if !::File.exist?(cache_path)
FileUtils.mkdir_p(image_cache_dir)
temp_file = ::FileHelper.download(
image[:url],
max_file_size: [SiteSetting.max_image_size_kb.kilobytes, 10.megabytes].max,
tmp_file_name: "image",
follow_redirect: true
)
FileUtils.cp(temp_file, cache_path)
end
::File.open(cache_path)
end
def available_methods
methods = super
methods << :image if ::DiscourseDev.config.post[:include_images]
methods
end
end
end
end