kuby-crdb 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 452c64d31fbe3da91cf3b12066156004ff8dcc50dcd62ce448b9e5a5e72dbb0a
+  data.tar.gz: 55aca34f100f7cdd53869fb4ce6e675db63a65b9794e9dd001ae35aee82e7aab
+SHA512:
+  metadata.gz: 51883863a1bb078941f88c904d162a00e461d1d7d7d192909c625940cf8f9fe4b8f4dca0294ef5b70e956c8b66df539ed709bb5bfa0dec82005db9f867d33d56
+  data.tar.gz: d2032dd54d710f1bb4ba7db4330b4c187d59c0f4504b65dfbd8337cf023ea3202fc986c0f85f37ba098c616c24b6ffa0fdc8ef6e6613604e4746ea72d84edfb3

data/CHANGELOG.md

@@ -0,0 +1,2 @@
+## 0.1.0
+* Birthday!

data/Gemfile

@@ -0,0 +1,18 @@
+source 'https://rubygems.org'
+
+gemspec
+
+group :development do
+  gem 'dry-inflector'
+end
+
+group :development, :test do
+  gem 'kuby-core'
+  gem 'kuby-kind'
+  gem 'pry-byebug'
+  gem 'rake'
+end
+
+group :test do
+  gem 'rspec', '~> 3.0'
+end

data/LICENSE

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2021 Cameron Dutro
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/Rakefile

@@ -0,0 +1,154 @@
+require 'bundler'
+require 'rspec/core/rake_task'
+require 'rubygems/package_task'
+
+require 'kuby/crdb'
+require 'pry-byebug'
+
+Bundler::GemHelper.install_tasks
+
+task default: :spec
+
+desc 'Run specs'
+RSpec::Core::RakeTask.new do |t|
+  t.pattern = './spec/**/*_spec.rb'
+end
+
+task generate: [:schemagen, :codegen]
+
+task :schemagen do
+  require 'base64'
+  require 'json'
+  require 'kind-rb'
+  require 'net/http'
+  require 'uri'
+  require 'yaml'
+
+  # delete first just in case
+  system("#{KindRb.executable} delete cluster --name kuby-crdb")
+  system("#{KindRb.executable} create cluster --name kuby-crdb")
+
+  system("kubectl apply -f https://raw.githubusercontent.com/cockroachdb/cockroach-operator/v#{Kuby::CRDB::CRDB_OPERATOR_VERSION}/install/crds.yaml")
+
+  cluster_name = 'kind-kuby-crdb'
+  kubeconfig = YAML.load_file(File.join(Dir.home, '.kube', 'config'))
+  cluster = kubeconfig['clusters'].find { |cluster| cluster['name'] == cluster_name }
+  ca_cert_data = Base64.decode64(cluster.dig(*%w(cluster certificate-authority-data)))
+
+  cert_store = OpenSSL::X509::Store.new
+  cert_store.add_cert(OpenSSL::X509::Certificate.new(ca_cert_data))
+  server = URI.parse(cluster.dig(*%w(cluster server)))
+
+  http = Net::HTTP.new(server.host, server.port).tap do |http|
+    http.use_ssl = true
+    http.cert_store = cert_store
+    http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+  end
+
+  token = nil
+  STDOUT.write("Getting API token from cluster...")
+
+  loop do
+    secrets = JSON.parse(`kubectl --context #{cluster_name} get secrets -o json`)
+    secret = secrets['items'].find do |secret|
+      secret.dig(*%w(metadata annotations kubernetes.io/service-account.name)) == 'default'
+    end
+
+    if secret
+      token = Base64.decode64(secret.dig(*%w(data token)))
+      puts ' done'
+      break
+    else
+      STDOUT.write('.')
+      sleep 1
+    end
+  end
+
+  get = Net::HTTP::Get.new('/openapi/v2')
+  get['Authorization'] = "Bearer #{token}"
+
+  crds = []
+  schema = nil
+  STDOUT.write("Requesting OpenAPI v2 schema from cluster...")
+
+  loop do
+    response = http.request(get)
+    schema = JSON.parse(response.body)
+
+    crds = schema['definitions'].each_with_object({}) do |(id, data), memo|
+      memo[id] = data if id.start_with?('com.cockroachlabs')
+    end
+
+    if crds.empty?
+      STDOUT.write('.')
+      sleep 1
+    else
+      puts ' done'
+      break
+    end
+  end
+
+  FileUtils.rm_rf('vendor')
+  FileUtils.mkdir('vendor')
+
+  schema = schema.merge('definitions' => crds)
+  File.write(File.join('vendor', 'open_api.json'), schema.to_json)
+
+  puts 'Pulling openapi2jsonschema Docker image'
+  system('docker pull garethr/openapi2jsonschema')
+
+  puts 'Converting OpenAPI format to JSON schema format'
+  system(<<~END)
+    docker run --rm \
+      -v #{File.expand_path('vendor')}:/usr/local/scratch \
+      garethr/openapi2jsonschema \
+      --kubernetes --output /usr/local/scratch/json_schema /usr/local/scratch/open_api.json
+  END
+
+  puts "Cleaning up"
+  system("#{KindRb.executable} delete cluster --name kuby-crdb")
+end
+
+task :codegen do
+  require 'dry/inflector'
+  require 'fileutils'
+
+  # delete anything that should be re-generated
+  FileUtils.rm_rf('./lib/kuby/crdb/dsl.rb')
+  FileUtils.rm_rf('./lib/kuby/crdb/dsl')
+  FileUtils.mkdir_p('./lib/kuby/crdb/dsl')
+
+  local_json_schema_path = 'vendor/json_schema'
+
+  generator = KubeDSL::Generator.new(
+    schema_dir: local_json_schema_path,
+    output_dir: File.join('lib'),
+    autoload_prefix: File.join('kuby', 'crdb', 'dsl'),
+    dsl_namespace: ['Kuby', 'CRDB', 'DSL'],
+    entrypoint_namespace: ['Kuby', 'CRDB'],
+    inflector: Dry::Inflector.new do |inflections|
+      inflections.acronym('DSL')
+      inflections.acronym('CRDB')
+    end
+  )
+
+  generator.builder.register_resolver('io.k8s') do |ref_str, builder|
+    external_ref = ::KubeDSL::ExternalRef.new(
+      ref_str,
+      ['KubeDSL', 'DSL'],
+      builder.inflector,
+      builder.schema_dir,
+      builder.autoload_prefix,
+      builder.serialize_handlers
+    )
+
+    ns = external_ref.ruby_namespace + [external_ref.kind]
+    exists = ns.inject(Object) { |mod, n| mod.const_get(n, false) } rescue false
+    exists ? external_ref : builder.parse_ref(ref_str)
+  end
+
+  generator.generate_resource_files
+  generator.generate_autoload_files
+
+  FileUtils.rm_rf(File.join('lib', 'kuby.rb'))
+end

data/kuby-crdb.gemspec

@@ -0,0 +1,19 @@
+$:.unshift File.join(File.dirname(__FILE__), 'lib')
+require 'kuby/crdb/version'
+
+Gem::Specification.new do |s|
+  s.name     = 'kuby-crdb'
+  s.version  = ::Kuby::CRDB::VERSION
+  s.authors  = ['Cameron Dutro']
+  s.email    = ['camertron@gmail.com']
+  s.homepage = 'http://github.com/getkuby/kuby-crdb'
+
+  s.description = s.summary = 'CockroachDB plugin for Kuby.'
+
+  s.platform = Gem::Platform::RUBY
+
+  s.add_dependency 'kube-dsl', '~> 0.7'
+
+  s.require_path = 'lib'
+  s.files = Dir['{lib,spec}/**/*', 'Gemfile', 'LICENSE', 'CHANGELOG.md', 'README.md', 'Rakefile', 'kuby-crdb.gemspec']
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/crdb_cluster.rb

@@ -0,0 +1,27 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class CrdbCluster < ::KubeDSL::DSLObject
+    object_field(:status) { Kuby::CRDB::DSL::CRDB::V1alpha1::Status.new }
+    object_field(:spec) { Kuby::CRDB::DSL::CRDB::V1alpha1::Spec.new }
+    value_field :api_version
+    object_field(:metadata) { KubeDSL::DSL::Meta::V1::ObjectMeta.new }
+
+    validates :status, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::Status }
+    validates :spec, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::Spec }
+    validates :api_version, field: { format: :string }, presence: false
+    validates :metadata, object: { kind_of: KubeDSL::DSL::Meta::V1::ObjectMeta }
+
+    def serialize
+      {}.tap do |result|
+        result[:status] = status.serialize
+        result[:kind] = "CrdbCluster"
+        result[:spec] = spec.serialize
+        result[:apiVersion] = api_version
+        result[:metadata] = metadata.serialize
+      end
+    end
+
+    def kind_sym
+      :crdb_cluster
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/crdb_cluster_list.rb

@@ -0,0 +1,24 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class CrdbClusterList < ::KubeDSL::DSLObject
+    array_field(:item) { Kuby::CRDB::DSL::CRDB::V1alpha1::CrdbCluster.new }
+    value_field :api_version
+    object_field(:metadata) { KubeDSL::DSL::Meta::V1::ListMeta.new }
+
+    validates :items, array: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::CrdbCluster }, presence: false
+    validates :api_version, field: { format: :string }, presence: false
+    validates :metadata, object: { kind_of: KubeDSL::DSL::Meta::V1::ListMeta }
+
+    def serialize
+      {}.tap do |result|
+        result[:items] = items.map(&:serialize)
+        result[:kind] = "CrdbClusterList"
+        result[:apiVersion] = api_version
+        result[:metadata] = metadata.serialize
+      end
+    end
+
+    def kind_sym
+      :crdb_cluster_list
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec.rb

@@ -0,0 +1,80 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class Spec < ::KubeDSL::DSLObject
+    object_field(:image) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecImage.new }
+    value_field :additional_args
+    value_field :min_available
+    value_field :max_sql_memory
+    value_field :cache
+    value_field :tls_enabled
+    key_value_field(:node_selector, format: :string)
+    value_field :http_port
+    object_field(:affinity) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecAffinity.new }
+    value_field :tolerations
+    value_field :nodes
+    value_field :cockroach_db_version
+    object_field(:resources) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecResources.new }
+    key_value_field(:additional_annotations, format: :string)
+    value_field :client_tls_secret
+    key_value_field(:additional_labels, format: :string)
+    value_field :sql_port
+    value_field :max_unavailable
+    value_field :node_tls_secret
+    value_field :pod_env_variables
+    object_field(:data_store) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStore.new }
+    value_field :grpc_port
+
+    validates :image, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecImage }
+    validates :additional_args, field: { format: :string }, presence: false
+    validates :min_available, field: { format: :integer }, presence: true
+    validates :max_sql_memory, field: { format: :string }, presence: false
+    validates :cache, field: { format: :string }, presence: false
+    validates :tls_enabled, field: { format: :boolean }, presence: true
+    validates :node_selector, kv: { value_format: :string }, presence: true
+    validates :http_port, field: { format: :integer }, presence: true
+    validates :affinity, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecAffinity }
+    validates :tolerations, field: { format: :string }, presence: false
+    validates :nodes, field: { format: :integer }, presence: true
+    validates :cockroach_db_version, field: { format: :string }, presence: false
+    validates :resources, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecResources }
+    validates :additional_annotations, kv: { value_format: :string }, presence: true
+    validates :client_tls_secret, field: { format: :string }, presence: false
+    validates :additional_labels, kv: { value_format: :string }, presence: true
+    validates :sql_port, field: { format: :integer }, presence: true
+    validates :max_unavailable, field: { format: :integer }, presence: true
+    validates :node_tls_secret, field: { format: :string }, presence: false
+    validates :pod_env_variables, field: { format: :string }, presence: false
+    validates :data_store, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStore }
+    validates :grpc_port, field: { format: :integer }, presence: true
+
+    def serialize
+      {}.tap do |result|
+        result[:image] = image.serialize
+        result[:additionalArgs] = additional_args
+        result[:minAvailable] = min_available
+        result[:maxSQLMemory] = max_sql_memory
+        result[:cache] = cache
+        result[:tlsEnabled] = tls_enabled
+        result[:nodeSelector] = node_selector.serialize
+        result[:httpPort] = http_port
+        result[:affinity] = affinity.serialize
+        result[:tolerations] = tolerations
+        result[:nodes] = nodes
+        result[:cockroachDBVersion] = cockroach_db_version
+        result[:resources] = resources.serialize
+        result[:additionalAnnotations] = additional_annotations.serialize
+        result[:clientTLSSecret] = client_tls_secret
+        result[:additionalLabels] = additional_labels.serialize
+        result[:sqlPort] = sql_port
+        result[:maxUnavailable] = max_unavailable
+        result[:nodeTLSSecret] = node_tls_secret
+        result[:podEnvVariables] = pod_env_variables
+        result[:dataStore] = data_store.serialize
+        result[:grpcPort] = grpc_port
+      end
+    end
+
+    def kind_sym
+      :spec
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_affinity.rb

@@ -0,0 +1,23 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecAffinity < ::KubeDSL::DSLObject
+    object_field(:pod_affinity) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecAffinityPodAffinity.new }
+    object_field(:node_affinity) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecAffinityNodeAffinity.new }
+    object_field(:pod_anti_affinity) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecAffinityPodAntiAffinity.new }
+
+    validates :pod_affinity, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecAffinityPodAffinity }
+    validates :node_affinity, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecAffinityNodeAffinity }
+    validates :pod_anti_affinity, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecAffinityPodAntiAffinity }
+
+    def serialize
+      {}.tap do |result|
+        result[:podAffinity] = pod_affinity.serialize
+        result[:nodeAffinity] = node_affinity.serialize
+        result[:podAntiAffinity] = pod_anti_affinity.serialize
+      end
+    end
+
+    def kind_sym
+      :spec_affinity
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_node_affinity.rb

@@ -0,0 +1,20 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecAffinityNodeAffinity < ::KubeDSL::DSLObject
+    object_field(:required_during_scheduling_ignored_during_execution) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecAffinityNodeAffinityRequiredDuringSchedulingIgnoredDuringExecution.new }
+    value_field :preferred_during_scheduling_ignored_during_execution
+
+    validates :required_during_scheduling_ignored_during_execution, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecAffinityNodeAffinityRequiredDuringSchedulingIgnoredDuringExecution }
+    validates :preferred_during_scheduling_ignored_during_execution, field: { format: :string }, presence: false
+
+    def serialize
+      {}.tap do |result|
+        result[:requiredDuringSchedulingIgnoredDuringExecution] = required_during_scheduling_ignored_during_execution.serialize
+        result[:preferredDuringSchedulingIgnoredDuringExecution] = preferred_during_scheduling_ignored_during_execution
+      end
+    end
+
+    def kind_sym
+      :spec_affinity_node_affinity
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_node_affinity_required_during_scheduling_ignored_during_execution.rb

@@ -0,0 +1,17 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecAffinityNodeAffinityRequiredDuringSchedulingIgnoredDuringExecution < ::KubeDSL::DSLObject
+    value_field :node_selector_terms
+
+    validates :node_selector_terms, field: { format: :string }, presence: true
+
+    def serialize
+      {}.tap do |result|
+        result[:nodeSelectorTerms] = node_selector_terms
+      end
+    end
+
+    def kind_sym
+      :spec_affinity_node_affinity_required_during_scheduling_ignored_during_execution
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_pod_affinity.rb

@@ -0,0 +1,20 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecAffinityPodAffinity < ::KubeDSL::DSLObject
+    value_field :required_during_scheduling_ignored_during_execution
+    value_field :preferred_during_scheduling_ignored_during_execution
+
+    validates :required_during_scheduling_ignored_during_execution, field: { format: :string }, presence: false
+    validates :preferred_during_scheduling_ignored_during_execution, field: { format: :string }, presence: false
+
+    def serialize
+      {}.tap do |result|
+        result[:requiredDuringSchedulingIgnoredDuringExecution] = required_during_scheduling_ignored_during_execution
+        result[:preferredDuringSchedulingIgnoredDuringExecution] = preferred_during_scheduling_ignored_during_execution
+      end
+    end
+
+    def kind_sym
+      :spec_affinity_pod_affinity
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_pod_anti_affinity.rb

@@ -0,0 +1,20 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecAffinityPodAntiAffinity < ::KubeDSL::DSLObject
+    value_field :required_during_scheduling_ignored_during_execution
+    value_field :preferred_during_scheduling_ignored_during_execution
+
+    validates :required_during_scheduling_ignored_during_execution, field: { format: :string }, presence: false
+    validates :preferred_during_scheduling_ignored_during_execution, field: { format: :string }, presence: false
+
+    def serialize
+      {}.tap do |result|
+        result[:requiredDuringSchedulingIgnoredDuringExecution] = required_during_scheduling_ignored_during_execution
+        result[:preferredDuringSchedulingIgnoredDuringExecution] = preferred_during_scheduling_ignored_during_execution
+      end
+    end
+
+    def kind_sym
+      :spec_affinity_pod_anti_affinity
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store.rb

@@ -0,0 +1,23 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecDataStore < ::KubeDSL::DSLObject
+    object_field(:pvc) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvc.new }
+    value_field :supports_auto_resize
+    object_field(:host_path) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStoreHostPath.new }
+
+    validates :pvc, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvc }
+    validates :supports_auto_resize, field: { format: :boolean }, presence: true
+    validates :host_path, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStoreHostPath }
+
+    def serialize
+      {}.tap do |result|
+        result[:pvc] = pvc.serialize
+        result[:supportsAutoResize] = supports_auto_resize
+        result[:hostPath] = host_path.serialize
+      end
+    end
+
+    def kind_sym
+      :spec_data_store
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_host_path.rb

@@ -0,0 +1,20 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecDataStoreHostPath < ::KubeDSL::DSLObject
+    value_field :path
+    value_field :type
+
+    validates :path, field: { format: :string }, presence: true
+    validates :type, field: { format: :string }, presence: false
+
+    def serialize
+      {}.tap do |result|
+        result[:path] = path
+        result[:type] = type
+      end
+    end
+
+    def kind_sym
+      :spec_data_store_host_path
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc.rb

@@ -0,0 +1,20 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecDataStorePvc < ::KubeDSL::DSLObject
+    object_field(:source) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvcSource.new }
+    object_field(:spec) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvcSpec.new }
+
+    validates :source, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvcSource }
+    validates :spec, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvcSpec }
+
+    def serialize
+      {}.tap do |result|
+        result[:source] = source.serialize
+        result[:spec] = spec.serialize
+      end
+    end
+
+    def kind_sym
+      :spec_data_store_pvc
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_source.rb

@@ -0,0 +1,20 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecDataStorePvcSource < ::KubeDSL::DSLObject
+    value_field :read_only
+    value_field :claim_name
+
+    validates :read_only, field: { format: :boolean }, presence: true
+    validates :claim_name, field: { format: :string }, presence: true
+
+    def serialize
+      {}.tap do |result|
+        result[:readOnly] = read_only
+        result[:claimName] = claim_name
+      end
+    end
+
+    def kind_sym
+      :spec_data_store_pvc_source
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec.rb

@@ -0,0 +1,35 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecDataStorePvcSpec < ::KubeDSL::DSLObject
+    value_field :storage_class_name
+    value_field :volume_mode
+    object_field(:selector) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvcSpecSelector.new }
+    value_field :access_modes
+    object_field(:data_source) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvcSpecDataSource.new }
+    value_field :volume_name
+    object_field(:resources) { Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvcSpecResources.new }
+
+    validates :storage_class_name, field: { format: :string }, presence: false
+    validates :volume_mode, field: { format: :string }, presence: false
+    validates :selector, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvcSpecSelector }
+    validates :access_modes, field: { format: :string }, presence: false
+    validates :data_source, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvcSpecDataSource }
+    validates :volume_name, field: { format: :string }, presence: false
+    validates :resources, object: { kind_of: Kuby::CRDB::DSL::CRDB::V1alpha1::SpecDataStorePvcSpecResources }
+
+    def serialize
+      {}.tap do |result|
+        result[:storageClassName] = storage_class_name
+        result[:volumeMode] = volume_mode
+        result[:selector] = selector.serialize
+        result[:accessModes] = access_modes
+        result[:dataSource] = data_source.serialize
+        result[:volumeName] = volume_name
+        result[:resources] = resources.serialize
+      end
+    end
+
+    def kind_sym
+      :spec_data_store_pvc_spec
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec_data_source.rb

@@ -0,0 +1,23 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecDataStorePvcSpecDataSource < ::KubeDSL::DSLObject
+    value_field :api_group
+    value_field :kind
+    value_field :name
+
+    validates :api_group, field: { format: :string }, presence: false
+    validates :kind, field: { format: :string }, presence: true
+    validates :name, field: { format: :string }, presence: true
+
+    def serialize
+      {}.tap do |result|
+        result[:apiGroup] = api_group
+        result[:kind] = kind
+        result[:name] = name
+      end
+    end
+
+    def kind_sym
+      :spec_data_store_pvc_spec_data_source
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec_resources.rb

@@ -0,0 +1,20 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecDataStorePvcSpecResources < ::KubeDSL::DSLObject
+    key_value_field(:requests, format: :string)
+    key_value_field(:limits, format: :string)
+
+    validates :requests, kv: { value_format: :string }, presence: true
+    validates :limits, kv: { value_format: :string }, presence: true
+
+    def serialize
+      {}.tap do |result|
+        result[:requests] = requests.serialize
+        result[:limits] = limits.serialize
+      end
+    end
+
+    def kind_sym
+      :spec_data_store_pvc_spec_resources
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec_selector.rb

@@ -0,0 +1,20 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecDataStorePvcSpecSelector < ::KubeDSL::DSLObject
+    key_value_field(:match_labels, format: :string)
+    value_field :match_expressions
+
+    validates :match_labels, kv: { value_format: :string }, presence: true
+    validates :match_expressions, field: { format: :string }, presence: false
+
+    def serialize
+      {}.tap do |result|
+        result[:matchLabels] = match_labels.serialize
+        result[:matchExpressions] = match_expressions
+      end
+    end
+
+    def kind_sym
+      :spec_data_store_pvc_spec_selector
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_image.rb

@@ -0,0 +1,23 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecImage < ::KubeDSL::DSLObject
+    value_field :pull_policy
+    value_field :name
+    value_field :pull_secret
+
+    validates :pull_policy, field: { format: :string }, presence: false
+    validates :name, field: { format: :string }, presence: true
+    validates :pull_secret, field: { format: :string }, presence: false
+
+    def serialize
+      {}.tap do |result|
+        result[:pullPolicy] = pull_policy
+        result[:name] = name
+        result[:pullSecret] = pull_secret
+      end
+    end
+
+    def kind_sym
+      :spec_image
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/spec_resources.rb

@@ -0,0 +1,20 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class SpecResources < ::KubeDSL::DSLObject
+    key_value_field(:requests, format: :string)
+    key_value_field(:limits, format: :string)
+
+    validates :requests, kv: { value_format: :string }, presence: true
+    validates :limits, kv: { value_format: :string }, presence: true
+
+    def serialize
+      {}.tap do |result|
+        result[:requests] = requests.serialize
+        result[:limits] = limits.serialize
+      end
+    end
+
+    def kind_sym
+      :spec_resources
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1/status.rb

@@ -0,0 +1,29 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  class Status < ::KubeDSL::DSLObject
+    value_field :cluster_status
+    value_field :version
+    value_field :conditions
+    value_field :operator_actions
+    value_field :crdbcontainerimage
+
+    validates :cluster_status, field: { format: :string }, presence: false
+    validates :version, field: { format: :string }, presence: false
+    validates :conditions, field: { format: :string }, presence: false
+    validates :operator_actions, field: { format: :string }, presence: false
+    validates :crdbcontainerimage, field: { format: :string }, presence: false
+
+    def serialize
+      {}.tap do |result|
+        result[:clusterStatus] = cluster_status
+        result[:version] = version
+        result[:conditions] = conditions
+        result[:operatorActions] = operator_actions
+        result[:crdbcontainerimage] = crdbcontainerimage
+      end
+    end
+
+    def kind_sym
+      :status
+    end
+  end
+end

data/lib/kuby/crdb/dsl/crdb/v1alpha1.rb

@@ -0,0 +1,21 @@
+module Kuby::CRDB::DSL::CRDB::V1alpha1
+  autoload :Status, 'kuby/crdb/dsl/crdb/v1alpha1/status'
+  autoload :SpecImage, 'kuby/crdb/dsl/crdb/v1alpha1/spec_image'
+  autoload :SpecAffinityPodAffinity, 'kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_pod_affinity'
+  autoload :SpecAffinityNodeAffinityRequiredDuringSchedulingIgnoredDuringExecution, 'kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_node_affinity_required_during_scheduling_ignored_during_execution'
+  autoload :SpecAffinityNodeAffinity, 'kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_node_affinity'
+  autoload :SpecAffinityPodAntiAffinity, 'kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_pod_anti_affinity'
+  autoload :SpecAffinity, 'kuby/crdb/dsl/crdb/v1alpha1/spec_affinity'
+  autoload :SpecResources, 'kuby/crdb/dsl/crdb/v1alpha1/spec_resources'
+  autoload :SpecDataStorePvcSource, 'kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_source'
+  autoload :SpecDataStorePvcSpecSelector, 'kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec_selector'
+  autoload :SpecDataStorePvcSpecDataSource, 'kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec_data_source'
+  autoload :SpecDataStorePvcSpecResources, 'kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec_resources'
+  autoload :SpecDataStorePvcSpec, 'kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec'
+  autoload :SpecDataStorePvc, 'kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc'
+  autoload :SpecDataStoreHostPath, 'kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_host_path'
+  autoload :SpecDataStore, 'kuby/crdb/dsl/crdb/v1alpha1/spec_data_store'
+  autoload :Spec, 'kuby/crdb/dsl/crdb/v1alpha1/spec'
+  autoload :CrdbClusterList, 'kuby/crdb/dsl/crdb/v1alpha1/crdb_cluster_list'
+  autoload :CrdbCluster, 'kuby/crdb/dsl/crdb/v1alpha1/crdb_cluster'
+end

data/lib/kuby/crdb/dsl/crdb.rb

@@ -0,0 +1,3 @@
+module Kuby::CRDB::DSL::CRDB
+  autoload :V1alpha1, 'kuby/crdb/dsl/crdb/v1alpha1'
+end

data/lib/kuby/crdb/dsl.rb

@@ -0,0 +1,3 @@
+module Kuby::CRDB::DSL
+  autoload :CRDB, 'kuby/crdb/dsl/crdb'
+end

data/lib/kuby/crdb/entrypoint.rb

@@ -0,0 +1,15 @@
+module Kuby::CRDB::Entrypoint
+  def crdb_cluster_list(&block)
+    ::Kuby::CRDB::DSL::CRDB::V1alpha1::CrdbClusterList.new do
+      api_version 'crdb.cockroachlabs.com/v1alpha1'
+      instance_eval(&block)
+    end
+  end
+
+  def crdb_cluster(&block)
+    ::Kuby::CRDB::DSL::CRDB::V1alpha1::CrdbCluster.new do
+      api_version 'crdb.cockroachlabs.com/v1alpha1'
+      instance_eval(&block)
+    end
+  end
+end

data/lib/kuby/crdb/plugin.rb

@@ -0,0 +1,160 @@
+require 'kuby'
+require 'kuby/crdb/version'
+
+module Kuby
+  module CRDB
+    class CRDBError < StandardError; end
+    class OperatorDeployError < CRDBError; end
+    class APIResourcesError < CRDBError; end
+
+    class Plugin < ::Kuby::Plugin
+      CRD_URL = "https://raw.githubusercontent.com/cockroachdb/cockroach-operator/v#{Kuby::CRDB::CRDB_OPERATOR_VERSION}/install/crds.yaml".freeze
+      OPERATOR_URL = "https://raw.githubusercontent.com/cockroachdb/cockroach-operator/v#{Kuby::CRDB::CRDB_OPERATOR_VERSION}/install/operator.yaml".freeze
+      NAMESPACE = 'cockroach-operator-system'.freeze
+      OPERATOR_DEPLOYMENT_NAME = 'cockroach-operator-manager'.freeze
+      WAIT_INTERVAL = 5  # seconds
+      WAIT_MAX = 120     # seconds
+
+      REPLICA_SET_MATCH_LABELS = {
+        'app' => 'cockroach-operator'
+      }
+
+      REPLICA_SET_MATCH_LABELS.freeze
+
+      def setup
+        Kuby.logger.info('Deploying CockroachDB operator')
+        operator_deployed? ? upgrade_operator : install_operator
+
+        wait_for_operator do
+          Kuby.logger.info('Waiting for CockroachDB operator deployment')
+        end
+
+        wait_for_api_resources do
+          Kuby.logger.info('Waiting for API resources to become available')
+        end
+
+        Kuby.logger.info('CockroachDB setup finished')
+      end
+
+      private
+
+      def install_operator
+        kubernetes_cli.apply_uri(CRD_URL)
+        kubernetes_cli.apply_uri(OPERATOR_URL)
+      end
+
+      def upgrade_operator
+        # TODO: add upgrade logic
+      end
+
+      def wait_for_operator
+        time_elapsed = 0
+
+        loop do
+          break if operator_ready?
+
+          if time_elapsed >= WAIT_MAX
+            raise OperatorDeployError, 'timeout waiting for operator to start. '\
+              "Waited #{time_elapsed}s."
+          end
+
+          yield
+
+          sleep WAIT_INTERVAL
+          time_elapsed += WAIT_INTERVAL
+        end
+      end
+
+      def wait_for_api_resources
+        time_elapsed = 0
+
+        loop do
+          begin
+            if time_elapsed >= WAIT_MAX
+              raise APIResourcesError, 'timeout waiting for API resources to '\
+                "become available. Waited #{time_elapsed}s."
+            end
+
+            kubernetes_cli.api_resources
+            break
+          rescue KubernetesCLI::KubernetesError
+            yield
+            sleep WAIT_INTERVAL
+            time_elapsed += WAIT_INTERVAL
+          end
+        end
+      end
+
+      def operator_ready?
+        depl = operator_deployment
+        rs = find_operator_rs(depl)
+        return false unless depl && rs
+
+        observed_generation = depl.dig('status', 'observedGeneration')
+        current_generation = depl.dig('metadata', 'generation')
+        return false unless observed_generation == current_generation
+
+        rollout_data = { 'replicas' => 0 }.merge(
+          depl['status'].slice(
+            *%w(replicas updatedReplicas availableReplicas unavailableReplicas)
+          )
+        )
+
+        desired_replicas = depl['spec']['replicas'].to_i
+        rs_desired_replicas = rs['spec']['replicas'].to_i
+
+        rs_ready?(rs) &&
+          rs_desired_replicas == desired_replicas &&
+          rollout_data['updatedReplicas'].to_i == desired_replicas &&
+          rollout_data['updatedReplicas'].to_i == rollout_data['availableReplicas'].to_i
+      end
+
+      def rs_ready?(rs)
+        observed_generation = rs.dig('status', 'observedGeneration')
+        current_generation = rs.dig('metadata', 'generation')
+        return false if observed_generation != current_generation
+
+        rollout_data = { "replicas" => 0 }.merge(
+          rs['status'].slice('replicas', 'availableReplicas', 'readyReplicas')
+        )
+
+        desired_replicas = rs['spec']['replicas'].to_i
+        desired_replicas == rollout_data['availableReplicas'].to_i &&
+          desired_replicas == rollout_data['readyReplicas'].to_i
+      end
+
+      def operator_deployment
+        kubernetes_cli.get_object(
+          'Deployment', NAMESPACE, OPERATOR_DEPLOYMENT_NAME
+        )
+      rescue ::KubernetesCLI::GetResourceError
+        nil
+      end
+
+      def find_operator_rs(depl)
+        all_rs_data = kubernetes_cli.get_objects('ReplicaSet', NAMESPACE, REPLICA_SET_MATCH_LABELS)
+        current_revision = depl.dig('metadata', 'annotations', 'deployment.kubernetes.io/revision')
+
+        all_rs_data.find do |rs|
+          rs.dig('metadata', 'ownerReferences').any? { |ref| ref['uid'] == depl.dig('metadata', 'uid') } &&
+            rs.dig('metadata', 'annotations', 'deployment.kubernetes.io/revision') == current_revision
+        end
+      rescue ::KubernetesCLI::GetResourceError
+        nil
+      end
+
+      def operator_deployed?
+        # TODO
+        false
+      end
+
+      def kubernetes_cli
+        provider.kubernetes_cli
+      end
+
+      def provider
+        environment.kubernetes.provider
+      end
+    end
+  end
+end

data/lib/kuby/crdb/version.rb

@@ -0,0 +1,6 @@
+module Kuby
+  module CRDB
+    VERSION = '0.1.0'.freeze
+    CRDB_OPERATOR_VERSION = '2.4.0'
+  end
+end

data/lib/kuby/crdb.rb

@@ -0,0 +1,13 @@
+require 'kube-dsl'
+
+module Kuby
+  module CRDB
+    autoload :DSL,        'kuby/crdb/dsl'
+    autoload :Entrypoint, 'kuby/crdb/entrypoint'
+    autoload :Plugin,     'kuby/crdb/plugin'
+
+    extend Entrypoint
+  end
+end
+
+Kuby.register_plugin(:crdb, ::Kuby::CRDB::Plugin)

metadata

@@ -0,0 +1,87 @@
+--- !ruby/object:Gem::Specification
+name: kuby-crdb
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Cameron Dutro
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2022-05-05 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: kube-dsl
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.7'
+description: CockroachDB plugin for Kuby.
+email:
+- camertron@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- CHANGELOG.md
+- Gemfile
+- LICENSE
+- Rakefile
+- kuby-crdb.gemspec
+- lib/kuby/crdb.rb
+- lib/kuby/crdb/dsl.rb
+- lib/kuby/crdb/dsl/crdb.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/crdb_cluster.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/crdb_cluster_list.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_affinity.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_node_affinity.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_node_affinity_required_during_scheduling_ignored_during_execution.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_pod_affinity.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_affinity_pod_anti_affinity.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_host_path.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_source.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec_data_source.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec_resources.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_data_store_pvc_spec_selector.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_image.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/spec_resources.rb
+- lib/kuby/crdb/dsl/crdb/v1alpha1/status.rb
+- lib/kuby/crdb/entrypoint.rb
+- lib/kuby/crdb/plugin.rb
+- lib/kuby/crdb/version.rb
+homepage: http://github.com/getkuby/kuby-crdb
+licenses: []
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.22
+signing_key:
+specification_version: 4
+summary: CockroachDB plugin for Kuby.
+test_files: []