summaryrefslogtreecommitdiffstats
path: root/lib/puppet/indirector/rest.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/puppet/indirector/rest.rb')
-rw-r--r--lib/puppet/indirector/rest.rb54
1 files changed, 51 insertions, 3 deletions
diff --git a/lib/puppet/indirector/rest.rb b/lib/puppet/indirector/rest.rb
index 7b7c932c4..d33150fc2 100644
--- a/lib/puppet/indirector/rest.rb
+++ b/lib/puppet/indirector/rest.rb
@@ -1,8 +1,56 @@
-require 'puppet/indirector/rest'
+require 'net/http'
+require 'uri'
# Access objects via REST
class Puppet::Indirector::REST < Puppet::Indirector::Terminus
- def find(name, options = {})
- indirection.model.new(name)
+
+ def rest_connection_details
+ { :host => Puppet[:server], :port => Puppet[:masterport].to_i }
+ end
+
+ def network_fetch(path)
+ network {|conn| conn.get("/#{path}").body }
+ end
+
+ def network_delete(path)
+ network {|conn| conn.delete("/#{path}").body }
+ end
+
+ def network_put(path, data)
+ network {|conn| conn.put("/#{path}", data).body }
+ end
+
+ def find(request)
+ network_result = network_fetch("#{indirection.name}/#{request.key}")
+ raise YAML.load(network_result) if exception?(network_result)
+ indirection.model.from_yaml(network_result)
+ end
+
+ def search(request)
+ network_results = network_fetch("#{indirection.name}s/#{request.key}")
+ raise YAML.load(network_results) if exception?(network_results)
+ YAML.load(network_results.to_s).collect {|result| indirection.model.from_yaml(result) }
+ end
+
+ def destroy(request)
+ network_result = network_delete("#{indirection.name}/#{request.key}")
+ raise YAML.load(network_result) if exception?(network_result)
+ YAML.load(network_result.to_s)
+ end
+
+ def save(request)
+ network_result = network_put("#{indirection.name}/", request.instance.to_yaml)
+ raise YAML.load(network_result) if exception?(network_result)
+ indirection.model.from_yaml(network_result)
+ end
+
+ private
+
+ def network(&block)
+ Net::HTTP.start(rest_connection_details[:host], rest_connection_details[:port]) {|conn| yield(conn) }
+ end
+
+ def exception?(yaml_string)
+ yaml_string =~ %r{--- !ruby/exception}
end
end