mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-02-12 14:34:50 +00:00
117c06220e
Add support for versioned artifacts with improved diff handling * Add versioned artifacts support allowing artifacts to be updated and tracked - New `ai_artifact_versions` table to store version history - Support for updating artifacts through a new `UpdateArtifact` tool - Add version-aware artifact rendering in posts - Include change descriptions for version tracking * Enhance artifact rendering and security - Add support for module-type scripts and external JS dependencies - Expand CSP to allow trusted CDN sources (unpkg, cdnjs, jsdelivr, googleapis) - Improve JavaScript handling in artifacts * Implement robust diff handling system (this is dormant but ready to use once LLMs catch up) - Add new DiffUtils module for applying changes to artifacts - Support for unified diff format with multiple hunks - Intelligent handling of whitespace and line endings - Comprehensive error handling for diff operations * Update routes and UI components - Add versioned artifact routes - Update markdown processing for versioned artifacts Also - Tweaks summary prompt - Improves upload support in custom tool to also provide urls
90 lines
2.6 KiB
Ruby
90 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class AiArtifact < ActiveRecord::Base
|
|
has_many :versions, class_name: "AiArtifactVersion", dependent: :destroy
|
|
belongs_to :user
|
|
belongs_to :post
|
|
validates :html, length: { maximum: 65_535 }
|
|
validates :css, length: { maximum: 65_535 }
|
|
validates :js, length: { maximum: 65_535 }
|
|
|
|
def self.iframe_for(id, version = nil)
|
|
<<~HTML
|
|
<div class='ai-artifact'>
|
|
<iframe src='#{url(id, version)}' frameborder="0" height="100%" width="100%"></iframe>
|
|
<a href='#{url(id, version)}' target='_blank'>#{I18n.t("discourse_ai.ai_artifact.link")}</a>
|
|
</div>
|
|
HTML
|
|
end
|
|
|
|
def self.url(id, version = nil)
|
|
url = Discourse.base_url + "/discourse-ai/ai-bot/artifacts/#{id}"
|
|
if version
|
|
"#{url}/#{version}"
|
|
else
|
|
url
|
|
end
|
|
end
|
|
|
|
def self.share_publicly(id:, post:)
|
|
artifact = AiArtifact.find_by(id: id)
|
|
artifact.update!(metadata: { public: true }) if artifact&.post&.topic&.id == post.topic.id
|
|
end
|
|
|
|
def self.unshare_publicly(id:)
|
|
artifact = AiArtifact.find_by(id: id)
|
|
artifact&.update!(metadata: { public: false })
|
|
end
|
|
|
|
def url
|
|
self.class.url(id)
|
|
end
|
|
|
|
def apply_diff(html_diff: nil, css_diff: nil, js_diff: nil, change_description: nil)
|
|
differ = DiscourseAi::Utils::DiffUtils
|
|
|
|
html = html_diff ? differ.apply_hunk(self.html, html_diff) : self.html
|
|
css = css_diff ? differ.apply_hunk(self.css, css_diff) : self.css
|
|
js = js_diff ? differ.apply_hunk(self.js, js_diff) : self.js
|
|
|
|
create_new_version(html: html, css: css, js: js, change_description: change_description)
|
|
end
|
|
|
|
def create_new_version(html: nil, css: nil, js: nil, change_description: nil)
|
|
latest_version = versions.order(version_number: :desc).first
|
|
new_version_number = latest_version ? latest_version.version_number + 1 : 1
|
|
version = nil
|
|
|
|
transaction do
|
|
# Create the version record
|
|
version =
|
|
versions.create!(
|
|
version_number: new_version_number,
|
|
html: html || self.html,
|
|
css: css || self.css,
|
|
js: js || self.js,
|
|
change_description: change_description,
|
|
)
|
|
save!
|
|
end
|
|
|
|
version
|
|
end
|
|
end
|
|
|
|
# == Schema Information
|
|
#
|
|
# Table name: ai_artifacts
|
|
#
|
|
# id :bigint not null, primary key
|
|
# user_id :integer not null
|
|
# post_id :integer not null
|
|
# name :string(255) not null
|
|
# html :string(65535)
|
|
# css :string(65535)
|
|
# js :string(65535)
|
|
# metadata :jsonb
|
|
# created_at :datetime not null
|
|
# updated_at :datetime not null
|
|
#
|