mirror of https://github.com/docusealco/docuseal
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
3.3 KiB
102 lines
3.3 KiB
# frozen_string_literal: true
|
|
|
|
module Mcp
|
|
module Tools
|
|
module CreateTemplate
|
|
SCHEMA = {
|
|
name: 'create_template',
|
|
title: 'Create Template',
|
|
description: 'Create a document template. Provide a URL to upload a PDF/DOCX file, or provide only a name ' \
|
|
'to create an empty template and receive an edit URL where the file can be uploaded via the UI.',
|
|
inputSchema: {
|
|
type: 'object',
|
|
properties: {
|
|
name: {
|
|
type: 'string',
|
|
description: 'Template name (used as the template name and required when url is not provided)'
|
|
},
|
|
url: {
|
|
type: 'string',
|
|
description: 'Optional URL of a PDF or DOCX file to upload. If omitted, an empty template is ' \
|
|
'created and the returned edit_url can be used to upload a file via the UI.'
|
|
}
|
|
},
|
|
required: %w[name]
|
|
},
|
|
annotations: {
|
|
readOnlyHint: false,
|
|
destructiveHint: false,
|
|
idempotentHint: false,
|
|
openWorldHint: false
|
|
}
|
|
}.freeze
|
|
|
|
module_function
|
|
|
|
# rubocop:disable Metrics/AbcSize, Metrics/MethodLength
|
|
def call(arguments, current_user, current_ability)
|
|
current_ability.authorize!(:create, Template.new(account_id: current_user.account_id, author: current_user))
|
|
|
|
account = current_user.account
|
|
|
|
template = Template.new(
|
|
account:,
|
|
author: current_user,
|
|
folder: account.default_template_folder,
|
|
name: arguments['name'].to_s.presence || 'New Template',
|
|
fields: [],
|
|
schema: []
|
|
)
|
|
|
|
if arguments['url'].present?
|
|
tempfile = Tempfile.new
|
|
tempfile.binmode
|
|
tempfile.write(DownloadUtils.call(arguments['url'], validate: true).body)
|
|
tempfile.rewind
|
|
|
|
filename = File.basename(URI.decode_www_form_component(arguments['url']))
|
|
|
|
file = ActionDispatch::Http::UploadedFile.new(
|
|
tempfile:,
|
|
filename:,
|
|
type: Marcel::MimeType.for(tempfile)
|
|
)
|
|
|
|
template.name = arguments['name'].presence || File.basename(filename, '.*')
|
|
template.save!
|
|
|
|
documents, = Templates::CreateAttachments.call(template, { files: [file] }, extract_fields: true)
|
|
schema = documents.map { |doc| { attachment_uuid: doc.uuid, name: doc.filename.base } }
|
|
|
|
if template.fields.blank?
|
|
template.fields = Templates::ProcessDocument.normalize_attachment_fields(template, documents)
|
|
end
|
|
|
|
template.update!(schema:)
|
|
else
|
|
template.save!
|
|
end
|
|
|
|
WebhookUrls.enqueue_events(template, 'template.created')
|
|
|
|
SearchEntries.enqueue_reindex(template)
|
|
|
|
{
|
|
content: [
|
|
{
|
|
type: 'text',
|
|
text: {
|
|
id: template.id,
|
|
name: template.name,
|
|
edit_url: Rails.application.routes.url_helpers.edit_template_url(template,
|
|
**Docuseal.default_url_options)
|
|
}.to_json
|
|
}
|
|
]
|
|
}
|
|
end
|
|
# rubocop:enable Metrics/AbcSize, Metrics/MethodLength
|
|
end
|
|
end
|
|
end
|