Merge pull request #279 from ripienaar/278

(#278) add the choria_kv_bucket resource

Author: ripienaar

lib/puppet/provider/choria_kv_bucket/choria_kv_bucket.rb

@@ -0,0 +1,66 @@
+require "puppet/resource_api/simple_provider"
+
+class Puppet::Provider::ChoriaKvBucket::ChoriaKvBucket < Puppet::ResourceApi::SimpleProvider
+  def get(context)
+    run("list", "present")
+  end
+
+  def delete(context, name)
+    run("delete", "absent", "name" => name)
+  end
+
+  def create(context, name, should)
+    run("ensure", "present", should)
+  end
+
+  def update(context, name, should)
+    create(context, name, should)
+  end
+
+  def run(action, ens, args={})
+    args.delete(:ensure)
+    parse_result(Puppet::Util::Execution.execute(make_cmd(action, args), :failonfail => false, :custom_environment => environment), ens)
+  end
+
+  def make_cmd(action, args={})
+    choria = Puppet::Util.which("choria")
+    raise("cannot find choria executable") if choria == ""
+
+    cmd = [choria, "kv", "api", "--%s" % action]
+    args.each {|k, v|
+      cmd << "--%s" % k
+      cmd << v unless v.is_a?(TrueClass)
+    }
+
+    cmd.join(" ")
+  end
+
+  def parse_result(result, ens)
+    if result == ""
+      return ""
+    end
+
+    parsed = JSON.parse(result, :symbolize_names => true)
+
+    if parsed.is_a?(Hash)
+      raise(parsed[:error]) if parsed[:error]
+
+      parsed[:ensure] = ens
+    end
+
+    if parsed.is_a?(Array)
+      parsed.each {|g| g[:ensure] = ens}
+    end
+
+    parsed
+  end
+
+  def environment
+    user = Etc.getpwuid(Process.euid)
+
+    {
+      "USER" => user.name,
+      "HOME" => user.dir
+    }
+  end
+end

lib/puppet/type/choria_kv_bucket.rb

@@ -0,0 +1,60 @@
+require "puppet/resource_api"
+
+Puppet::ResourceApi.register_type(
+  name: "choria_kv_bucket",
+  desc: <<-EOS,
+  Manage Choria Key-Value store Buckets hosted in Choria Broker
+
+  See https://choria.io/docs/streams/key-value/ for more information
+  EOS
+
+  attributes: {
+    name: {
+      type: 'Pattern[/[\Aa-zA-Z0-9_-]+\Z/]',
+      desc: "Unique name for this Bucket",
+      behaviour: :namevar,
+    },
+
+    history: {
+      type: "Integer[1,100]",
+      desc: "How many historic values to keep for any key in the bucket",
+      default: 1
+    },
+
+    expire: {
+      type: "Integer",
+      desc: "How long before a value expires from the bucket, in seconds",
+      default: -1
+    },
+
+    replicas: {
+      type: "Integer[1,5]",
+      desc: "In clustered environments this is how many active Replicas of the data and configuration is kept, odd number is best",
+    },
+
+    max_value_size: {
+      type: "Integer",
+      desc: "The maximum size for any value that will be accepted by the bucket",
+      default: -1
+    },
+
+    max_bucket_size: {
+      type: "Integer",
+      desc: "The maximum size of all values in the bucket, including history, before writes will fail",
+      default: -1,
+    },
+
+    force: {
+      type: "Boolean",
+      default: false,
+      desc: "When updating replicas the Bucket has to be removed and recreated, this can only happen if force is true",
+      behaviour: :parameter,
+    },
+
+    ensure: {
+      type: "Enum[present, absent]",
+      desc: "Create or remove the Bucket",
+      default: "present"
+    }
+  }
+)

spec/classes/init_spec.rb

@@ -146,7 +146,7 @@
             is_expected.to contain_class("choria::repo").with_nightly(false)
             is_expected.to contain_class("choria::repo").with_ensure("present")
             is_expected.to contain_file("/etc/apt/sources.list.d/choria-release.list")
-            is_expected.to contain_apt__source("choria-release").with(release: "bionic")
+            is_expected.to contain_apt__source("choria-release").with(release: "ubuntu")
                              .with(location: "mirror://mirrorlists.choria.io/apt/release/ubuntu/bionic/mirrors.txt")
           end
         end
@@ -162,7 +162,7 @@
           it "should manage the main repo" do
             is_expected.to contain_file("/etc/apt/sources.list.d/choria-release.list")
             is_expected.to contain_apt__source("choria-release")
-                             .with(release: "xenial")
+                             .with(release: "ubuntu")
                              .with(location: "https://apt.eu.choria.io/release/ubuntu/xenial")
           end
         end