2024-11-18 17:22:39 -05:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
|
|
module DiscourseAi
|
|
|
|
|
module AiBot
|
|
|
|
|
class ArtifactsController < ApplicationController
|
|
|
|
|
requires_plugin DiscourseAi::PLUGIN_NAME
|
|
|
|
|
before_action :require_site_settings!
|
|
|
|
|
|
|
|
|
|
skip_before_action :preload_json, :check_xhr, only: %i[show]
|
|
|
|
|
|
|
|
|
|
def show
|
|
|
|
|
artifact = AiArtifact.find(params[:id])
|
|
|
|
|
|
|
|
|
|
post = Post.find_by(id: artifact.post_id)
|
|
|
|
|
if artifact.metadata&.dig("public")
|
|
|
|
|
# no guardian needed
|
|
|
|
|
else
|
|
|
|
|
raise Discourse::NotFound if !post&.topic&.private_message?
|
|
|
|
|
raise Discourse::NotFound if !guardian.can_see?(post)
|
|
|
|
|
end
|
|
|
|
|
|
2024-11-19 06:44:17 -05:00
|
|
|
# Prepare the inner (untrusted) HTML document
|
|
|
|
|
untrusted_html = <<~HTML
|
2024-11-18 17:22:39 -05:00
|
|
|
<!DOCTYPE html>
|
|
|
|
|
<html>
|
|
|
|
|
<head>
|
|
|
|
|
<meta charset="UTF-8">
|
|
|
|
|
<title>#{ERB::Util.html_escape(artifact.name)}</title>
|
|
|
|
|
<style>
|
|
|
|
|
#{artifact.css}
|
|
|
|
|
</style>
|
|
|
|
|
</head>
|
|
|
|
|
<body>
|
|
|
|
|
#{artifact.html}
|
|
|
|
|
<script>
|
|
|
|
|
#{artifact.js}
|
|
|
|
|
</script>
|
|
|
|
|
</body>
|
|
|
|
|
</html>
|
|
|
|
|
HTML
|
|
|
|
|
|
2024-11-19 06:44:17 -05:00
|
|
|
# Prepare the outer (trusted) HTML document
|
|
|
|
|
trusted_html = <<~HTML
|
|
|
|
|
<!DOCTYPE html>
|
|
|
|
|
<html>
|
|
|
|
|
<head>
|
|
|
|
|
<meta charset="UTF-8">
|
|
|
|
|
<title>#{ERB::Util.html_escape(artifact.name)}</title>
|
|
|
|
|
<style>
|
|
|
|
|
html, body, iframe {
|
|
|
|
|
margin: 0;
|
|
|
|
|
padding: 0;
|
|
|
|
|
width: 100%;
|
|
|
|
|
height: 100%;
|
2024-11-19 21:13:03 -05:00
|
|
|
border: 0;
|
|
|
|
|
overflow: hidden;
|
|
|
|
|
}
|
|
|
|
|
iframe {
|
|
|
|
|
overflow: auto;
|
2024-11-19 06:44:17 -05:00
|
|
|
}
|
|
|
|
|
</style>
|
|
|
|
|
</head>
|
|
|
|
|
<body>
|
|
|
|
|
<iframe sandbox="allow-scripts allow-forms" height="100%" width="100%" srcdoc="#{ERB::Util.html_escape(untrusted_html)}" frameborder="0"></iframe>
|
|
|
|
|
</body>
|
|
|
|
|
</html>
|
|
|
|
|
HTML
|
|
|
|
|
|
2024-11-18 17:22:39 -05:00
|
|
|
response.headers.delete("X-Frame-Options")
|
2024-11-19 04:25:07 -05:00
|
|
|
response.headers["Content-Security-Policy"] = "script-src 'unsafe-inline';"
|
2024-11-18 17:22:39 -05:00
|
|
|
|
|
|
|
|
# Render the content
|
2024-11-19 06:44:17 -05:00
|
|
|
render html: trusted_html.html_safe, layout: false, content_type: "text/html"
|
2024-11-18 17:22:39 -05:00
|
|
|
end
|
|
|
|
|
|
|
|
|
|
private
|
|
|
|
|
|
|
|
|
|
def require_site_settings!
|
|
|
|
|
if !SiteSetting.discourse_ai_enabled ||
|
|
|
|
|
!SiteSetting.ai_artifact_security.in?(%w[lax strict])
|
|
|
|
|
raise Discourse::NotFound
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
end
|
