101 lines
2.9 KiB
Ruby
101 lines
2.9 KiB
Ruby
# == Schema Information
|
|
#
|
|
# Table name: captain_custom_tools
|
|
#
|
|
# id :bigint not null, primary key
|
|
# auth_config :jsonb
|
|
# auth_type :string default("none")
|
|
# description :text
|
|
# enabled :boolean default(TRUE), not null
|
|
# endpoint_url :text not null
|
|
# http_method :string default("GET"), not null
|
|
# param_schema :jsonb
|
|
# request_template :text
|
|
# response_template :text
|
|
# slug :string not null
|
|
# title :string not null
|
|
# created_at :datetime not null
|
|
# updated_at :datetime not null
|
|
# account_id :bigint not null
|
|
#
|
|
# Indexes
|
|
#
|
|
# index_captain_custom_tools_on_account_id (account_id)
|
|
# index_captain_custom_tools_on_account_id_and_slug (account_id,slug) UNIQUE
|
|
#
|
|
class Captain::CustomTool < ApplicationRecord
|
|
include Concerns::Toolable
|
|
include Concerns::SafeEndpointValidatable
|
|
|
|
self.table_name = 'captain_custom_tools'
|
|
|
|
NAME_PREFIX = 'custom'.freeze
|
|
NAME_SEPARATOR = '_'.freeze
|
|
PARAM_SCHEMA_VALIDATION = {
|
|
'type': 'array',
|
|
'items': {
|
|
'type': 'object',
|
|
'properties': {
|
|
'name': { 'type': 'string' },
|
|
'type': { 'type': 'string' },
|
|
'description': { 'type': 'string' },
|
|
'required': { 'type': 'boolean' }
|
|
},
|
|
'required': %w[name type description],
|
|
'additionalProperties': false
|
|
}
|
|
}.to_json.freeze
|
|
|
|
belongs_to :account
|
|
|
|
enum :http_method, %w[GET POST].index_by(&:itself), validate: true
|
|
enum :auth_type, %w[none bearer basic api_key].index_by(&:itself), default: :none, validate: true, prefix: :auth
|
|
|
|
before_validation :generate_slug
|
|
|
|
validates :slug, presence: true, uniqueness: { scope: :account_id }
|
|
validates :title, presence: true
|
|
validates :endpoint_url, presence: true
|
|
validates_with JsonSchemaValidator,
|
|
schema: PARAM_SCHEMA_VALIDATION,
|
|
attribute_resolver: ->(record) { record.param_schema }
|
|
|
|
scope :enabled, -> { where(enabled: true) }
|
|
|
|
def to_tool_metadata
|
|
{
|
|
id: slug,
|
|
title: title,
|
|
description: description,
|
|
custom: true
|
|
}
|
|
end
|
|
|
|
private
|
|
|
|
def generate_slug
|
|
return if slug.present?
|
|
return if title.blank?
|
|
|
|
paramterized_title = title.parameterize(separator: NAME_SEPARATOR)
|
|
|
|
base_slug = "#{NAME_PREFIX}#{NAME_SEPARATOR}#{paramterized_title}"
|
|
self.slug = find_unique_slug(base_slug)
|
|
end
|
|
|
|
def find_unique_slug(base_slug)
|
|
return base_slug unless slug_exists?(base_slug)
|
|
|
|
5.times do
|
|
slug_candidate = "#{base_slug}#{NAME_SEPARATOR}#{SecureRandom.alphanumeric(6).downcase}"
|
|
return slug_candidate unless slug_exists?(slug_candidate)
|
|
end
|
|
|
|
raise ActiveRecord::RecordNotUnique, I18n.t('captain.custom_tool.slug_generation_failed')
|
|
end
|
|
|
|
def slug_exists?(candidate)
|
|
self.class.exists?(account_id: account_id, slug: candidate)
|
|
end
|
|
end
|