summaryrefslogtreecommitdiff
path: root/lib/leap_cli/config
diff options
context:
space:
mode:
Diffstat (limited to 'lib/leap_cli/config')
-rw-r--r--lib/leap_cli/config/secrets.rb21
1 files changed, 16 insertions, 5 deletions
diff --git a/lib/leap_cli/config/secrets.rb b/lib/leap_cli/config/secrets.rb
index e0f3080..184d11d 100644
--- a/lib/leap_cli/config/secrets.rb
+++ b/lib/leap_cli/config/secrets.rb
@@ -14,17 +14,28 @@ module LeapCli; module Config
end
# we can't use fetch() or get(), since those already have special meanings
- def retrieve(key, environment=nil)
- self.fetch(environment||'default', {})[key.to_s]
+ def retrieve(key, environment)
+ self.fetch(environment, {})[key.to_s]
end
- def set(key, value, environment=nil)
- environment ||= 'default'
+ def set(*args, &block)
+ if block_given?
+ set_with_block(*args, &block)
+ else
+ set_without_block(*args)
+ end
+ end
+
+ def set_without_block(key, value, environment)
+ set_with_block(key, environment) {value}
+ end
+
+ def set_with_block(key, environment, &block)
key = key.to_s
@discovered_keys[environment] ||= {}
@discovered_keys[environment][key] = true
self[environment] ||= {}
- self[environment][key] ||= value
+ self[environment][key] ||= yield
end
#