Class: PromptScrub::Rehydrator

Inherits:
Object
  • Object
show all
Defined in:
lib/promptscrub/rehydrator.rb

Constant Summary collapse

TOKEN_PATTERN = 
/<[A-Z]+_\d{3}>/

Instance Method Summary collapse

Constructor Details

#initialize(vault) ⇒ Rehydrator

Returns a new instance of Rehydrator.



7
8
9
 
# File 'lib/promptscrub/rehydrator.rb', line 7

def initialize(vault)
  @vault = vault
end

Instance Method Details

#rehydrate(text) ⇒ Object 



11
12
13
14
15
16
17
18
 
# File 'lib/promptscrub/rehydrator.rb', line 11

def rehydrate(text)
  return text if text.nil? || text.empty?
  return text if @vault.empty?

  text.gsub(TOKEN_PATTERN) do |token|
    @vault.rehydrate(token) || token
  end
end