PromptManager

require 'prompt_manager'

# Configure storage adapter
PromptManager::Prompt.storage_adapter =
  PromptManager::Storage::FileSystemAdapter.config do |config|
    config.prompts_dir = '~/.prompts'
  end.new

# Load and use a prompt
prompt = PromptManager::Prompt.new(id: 'greeting')
prompt.parameters = {
  "[NAME]" => "Alice",
  "[LANGUAGE]" => "English"
}

# Get the processed prompt text
result = prompt.to_s

prompt_text = "Hello [NAME], please translate this to [LANGUAGE]"

# Use {{mustache}} style
PromptManager::Prompt.parameter_regex = /(\{\{[A-Za-z_]+\}\})/

# Use :colon style
PromptManager::Prompt.parameter_regex = /(:[a-z_]+)/

prompt = PromptManager::Prompt.new(id: 'example')

# Get all keywords found in the prompt
keywords = prompt.keywords  #=> ["[NAME]", "[LANGUAGE]"]

# Set parameter values
prompt.parameters = {
  "[NAME]" => "Alice",
  "[LANGUAGE]" => "French"
}

# Get processed text with substitutions
final_text = prompt.to_s

# Save changes
prompt.save

class MyDirectiveProcessor < PromptManager::DirectiveProcessor
  def process_directive(directive, prompt)
    case directive
    when /^\/\/model (.+)$/
      set_model($1)
    when /^\/\/temperature (.+)$/
      set_temperature($1.to_f)
    else
      super
    end
  end
end

prompt = PromptManager::Prompt.new(
  id: 'example',
  directives_processor: MyDirectiveProcessor.new
)

prompt = PromptManager::Prompt.new(
  id: 'dynamic',
  erb_flag: true
)

prompt = PromptManager::Prompt.new(
  id: 'with_env',
  envar_flag: true
)

# Parameters are stored as arrays
prompt.parameters = {
  "[NAME]" => ["Alice", "Bob", "Charlie"]  # Charlie is most recent
}

# The last value is always the most recent
current_name = prompt.parameters["[NAME]"].last

# Useful for:
# - Implementing value history in UIs
# - Providing dropdown selections
# - Tracking parameter usage over time

begin
  prompt = PromptManager::Prompt.new(id: 'missing')
rescue PromptManager::StorageError => e
  # Handle storage-related errors
  puts "Storage error: #{e.message}"
rescue PromptManager::ParameterError => e
  # Handle parameter substitution errors
  puts "Parameter error: #{e.message}"
rescue PromptManager::ConfigurationError => e
  # Handle configuration errors
  puts "Configuration error: #{e.message}"
end

PromptManager::Storage::FileSystemAdapter.config do |config|
  config.prompts_dir       = "~/.prompts"      # Required
  config.search_proc       = nil               # Optional custom search
  config.prompt_extension  = '.txt'            # Default
  config.params_extension  = '.json'           # Default
end

config.search_proc = ->(query) {
  # Use ripgrep for fast searching
  `rg -l "#{query}" #{config.prompts_dir}`.split("\n")
}

PromptManager::Storage::ActiveRecordAdapter.config do |config|
  config.model              = PromptModel      # Your AR model
  config.id_column          = :prompt_id       # Column for ID
  config.text_column        = :content         # Column for text
  config.parameters_column  = :params          # Column for parameters
end

class CreatePrompts < ActiveRecord::Migration[7.0]
  def change
    create_table :prompts do |t|
      t.string :prompt_id, null: false, index: { unique: true }
      t.text :content
      t.json :params
      t.timestamps
    end
  end
end

class RedisAdapter
  def initialize(redis_client)
    @redis = redis_client
  end

  def get(id)
    prompt_text = @redis.get("prompt:#{id}:text")
    parameters = JSON.parse(@redis.get("prompt:#{id}:params") || '{}')
    [prompt_text, parameters]
  end

  def save(id, text, parameters)
    @redis.set("prompt:#{id}:text", text)
    @redis.set("prompt:#{id}:params", parameters.to_json)
  end

  def delete(id)
    @redis.del("prompt:#{id}:text", "prompt:#{id}:params")
  end

  def list
    @redis.keys("prompt:*:text").map { |k| k.split(':')[1] }
  end
end

prompt = PromptManager::Prompt.new(
  id: 'example',
  context: ['additional', 'context'],
  directives_processor: CustomProcessor.new,
  external_binding: binding,
  erb_flag: true,
  envar_flag: true
)

PromptManager::Prompt.storage_adapter = adapter_instance

# Handlebars style: {{name}}
PromptManager::Prompt.parameter_regex = /(\{\{[a-z_]+\}\})/

# Colon prefix: :name
PromptManager::Prompt.parameter_regex = /(:[a-z_]+)/

# Dollar sign: $NAME
PromptManager::Prompt.parameter_regex = /(\$[A-Z_]+)/

# Percentage: %name%
PromptManager::Prompt.parameter_regex = /(%[a-z_]+%)/

# With FileSystemAdapter
adapter.search_proc = ->(query) {
  # Custom search implementation
  results = []
  Dir.glob("#{prompts_dir}/**/*.txt").each do |file|
    content = File.read(file)
    if content.include?(query)
      results << File.basename(file, '.txt')
    end
  end
  results
}

# With ActiveRecordAdapter
PromptModel.where("content LIKE ?", "%#{query}%").pluck(:prompt_id)

# examples/simple.rb
require 'prompt_manager'

# Setup
PromptManager::Prompt.storage_adapter =
  PromptManager::Storage::FileSystemAdapter.config do |c|
    c.prompts_dir = '~/.prompts'
  end.new

# Create and use a prompt
prompt = PromptManager::Prompt.new(id: 'story')
prompt.parameters = {
  "[GENRE]" => "fantasy",
  "[CHARACTER]" => "wizard"
}

puts prompt.to_s

# examples/redis_storage.rb
class RedisStorage
  # ... implementation
end

PromptManager::Prompt.storage_adapter = RedisStorage.new(Redis.new)