config_o_mat 0.1.0

data/lib/config_o_mat/flip_flopper/op/start_activating_instance.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'lifecycle_vm/op_base'
+
+module ConfigOMat
+  module Op
+    class StartActivatingInstance < LifecycleVM::OpBase
+      reads :service, :activating_instance, :runtime_directory
+
+      def call
+        instance_name = "#{service}#{activating_instance}"
+        file_path = File.join(runtime_directory,  "#{instance_name}.start")
+
+        logger&.notice(:service_start, name: instance_name)
+
+        FileUtils.touch(file_path)
+      end
+    end
+  end
+end

data/lib/config_o_mat/flip_flopper/op/stop_activating_instance.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'lifecycle_vm/op_base'
+
+module ConfigOMat
+  module Op
+    class StopActivatingInstance < LifecycleVM::OpBase
+      reads :service, :activating_instance, :runtime_directory
+
+      def call
+        instance_name = "#{service}#{activating_instance}"
+        file_path = File.join(runtime_directory,  "#{instance_name}.stop")
+
+        logger&.notice(:service_stop, name: instance_name)
+
+        FileUtils.touch(file_path)
+      end
+    end
+  end
+end

data/lib/config_o_mat/flip_flopper/op/stop_initial_instance.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'lifecycle_vm/op_base'
+
+module ConfigOMat
+  module Op
+    class StopInitialInstance < LifecycleVM::OpBase
+      reads :service, :running_instance, :runtime_directory
+
+      def call
+        instance_name = "#{service}#{running_instance}"
+        file_path = File.join(runtime_directory,  "#{instance_name}.stop")
+
+        logger&.notice(:service_stop, name: instance_name)
+
+        FileUtils.touch(file_path)
+      end
+    end
+  end
+end

data/lib/config_o_mat/flip_flopper/op.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+Dir[File.join(__dir__, 'op', '*')].sort.each { |file| require file }

data/lib/config_o_mat/flip_flopper.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'lifecycle_vm'
+
+require 'config_o_mat/flip_flopper/op'
+require 'config_o_mat/flip_flopper/cond'
+require 'config_o_mat/flip_flopper/memory'
+
+module ConfigOMat
+  module FlipFlopper
+    class VM < LifecycleVM::VM
+      memory_class ConfigOMat::FlipFlopper::Memory
+
+      on :start, do: Op::DetermineRunningInstance, then: :start_activating_instance
+      on :start_activating_instance, do: Op::StartActivatingInstance, then: :wait_for_service_start
+      on :wait_for_service_start, do: Op::CheckServiceStatus, then: {
+        case: Cond::ServiceStatus,
+        when: {
+          starting: :wait_for_service_start,
+          started: :stop_initial_service,
+          failed: :fail,
+          timed_out: :abort
+        }
+      }
+
+      on :stop_initial_service, do: Op::StopInitialInstance, then: :exit
+      on :fail, do: Op::ReportFailure, then: :exit
+
+      on :abort, do: Op::StopActivatingInstance, then: :fail
+    end
+  end
+end

data/lib/config_o_mat/meta_configurator/memory.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'lifecycle_vm/memory'
+
+module ConfigOMat
+  module MetaConfigurator
+    class Memory < LifecycleVM::Memory
+      attr_accessor :argv, :env, :early_exit, :configuration_directory, :runtime_directory,
+                    :systemd_directory, :logs_directory, :profile_defs,
+                    :template_defs, :service_defs, :dependencies, :refresh_interval,
+                    :client_id, :retry_count, :retries_left, :retry_wait,
+                    :region, :systemd_interface
+
+      def initialize(
+        argv: [],
+        env: {},
+        early_exit: false,
+        configuration_directory: nil,
+        runtime_directory: nil,
+        logs_directory: nil,
+        systemd_directory: nil,
+        profile_defs: {},
+        template_defs: {},
+        service_defs: {},
+        dependencies: {},
+        refresh_interval: 5,
+        client_id: '',
+        logger: nil,
+        retry_count: 3,
+        retries_left: 3,
+        retry_wait: 2,
+        region: nil,
+        systemd_interface: nil
+      )
+        super()
+
+        @argv = argv
+        @env = env
+        @early_exit = early_exit
+        @configuration_directory = configuration_directory
+        @runtime_directory = runtime_directory
+        @logs_directory = logs_directory
+        @systemd_directory = systemd_directory
+        @profile_defs = profile_defs
+        @template_defs = template_defs
+        @service_defs = service_defs
+        @dependencies = dependencies
+        @refresh_interval = refresh_interval
+        @client_id = client_id
+        @logger = logger
+        @retry_count = retry_count
+        @retries_left = retries_left
+        @retry_wait = retry_wait
+        @region = region
+        @systemd_interface = systemd_interface
+      end
+    end
+  end
+end

data/lib/config_o_mat/meta_configurator/op/generate_systemd_config.rb

@@ -0,0 +1,80 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'lifecycle_vm/op_base'
+
+require 'fileutils'
+
+module ConfigOMat
+  module Op
+    class GenerateSystemdConfig < LifecycleVM::OpBase
+      DROPIN_FILE_NAME = '99_teak_configurator.conf'
+
+      reads :template_defs, :service_defs, :runtime_directory, :systemd_directory, :systemd_interface
+
+      def call
+        grouped_restart_modes = service_defs.values.group_by(&:restart_mode)
+        restarts = grouped_restart_modes[:restart]
+        flip_flops = grouped_restart_modes[:flip_flop]
+
+        enable_restarts(restarts) if restarts
+        enable_flip_flops(flip_flops) if flip_flops
+
+        systemd_interface.daemon_reload
+      end
+
+    private
+
+      def write_dropin(unit, service)
+        dropin = %([Service]\n)
+        service.templates.each do |tmpl|
+          template = template_defs[tmpl.to_sym]
+          dropin += %(LoadCredential=#{template.dst}:#{File.join(runtime_directory, template.dst)}\n)
+        end
+
+        dropin_dir = File.join(systemd_directory, "#{unit}.service.d")
+
+        FileUtils.mkdir_p(dropin_dir)
+        File.open(File.join(dropin_dir, DROPIN_FILE_NAME), 'w') { |f| f.write(dropin) }
+      end
+
+      def enable_restarts(services)
+        units = []
+        services.each do |service|
+          unit = service.systemd_unit
+          units << unit
+
+          write_dropin(unit, service)
+        end
+
+        systemd_interface.enable_restart_paths(units)
+      end
+
+      def enable_flip_flops(services)
+        units = []
+        services.each do |service|
+          unit = service.systemd_unit
+          units << "#{unit}1"
+          units << "#{unit}2"
+
+          write_dropin(unit, service)
+        end
+
+        systemd_interface.enable_start_stop_paths(units)
+      end
+    end
+  end
+end

data/lib/config_o_mat/meta_configurator/op/parse_meta_cli.rb

@@ -0,0 +1,103 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'lifecycle_vm/op_base'
+
+require 'optparse'
+
+module ConfigOMat
+  module Op
+    class ParseMetaCli < LifecycleVM::OpBase
+      DEFAULT_SYSTEMD_DIRECTORY = '/run/systemd/system'
+
+      reads :argv, :env
+      writes :configuration_directory, :runtime_directory, :logs_directory, :systemd_directory, :early_exit
+
+      # Add expand_path to string because the safe accessor is nice to use.
+      module CoreExt
+        refine String do
+          def expand_path
+            File.expand_path(self)
+          end
+        end
+      end
+
+      using CoreExt
+
+      def call
+        self.early_exit = false
+
+        parser = OptionParser.new do |opts|
+          opts.on('-v', '--version', 'Display the version') do
+            self.early_exit = true
+          end
+
+          opts.on('-h', '--help', 'Prints this help') do
+            puts opts
+            self.early_exit = true
+          end
+
+          opts.on(
+            '-c', '--configuration-directory directory',
+            'Read configuration from the given directory instead of from $CONFIGURATION_DIRECTORY'
+          ) do |dir|
+            self.configuration_directory = dir
+          end
+
+          opts.on(
+            '-r', '--runtime-directory directory',
+            'Use the given directory for writing templates instead of $RUNTIME_DIRECTORY'
+          ) do |dir|
+            self.runtime_directory = dir
+          end
+
+          opts.on(
+            '-l', '--logs-directory directory',
+            'Use the given directory for writing log files instead of $LOGS_DIRECTORY'
+          ) do |dir|
+            self.logs_directory = dir
+          end
+
+          opts.on(
+            '-s', '--systemd-directory directory',
+            "Write out systemd configuration to the given directory instead of #{DEFAULT_SYSTEMD_DIRECTORY}"
+          ) do |dir|
+            self.systemd_directory = dir
+          end
+        end
+
+        parser.parse(argv)
+
+        self.configuration_directory ||= env['CONFIGURATION_DIRECTORY']
+        self.runtime_directory ||= env['RUNTIME_DIRECTORY']
+        self.logs_directory ||= env['LOGS_DIRECTORY']
+        self.systemd_directory ||= DEFAULT_SYSTEMD_DIRECTORY
+
+        self.configuration_directory = configuration_directory&.expand_path
+      end
+
+      def validate
+        return if early_exit
+
+        if configuration_directory.nil? || configuration_directory.empty?
+          error :configuration_directory, 'must be present'
+        end
+
+        error :runtime_directory, 'must be present' if runtime_directory.nil? || runtime_directory.empty?
+      end
+    end
+  end
+end

data/lib/config_o_mat/meta_configurator/op.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+Dir[File.join(__dir__, 'op', '*')].sort.each { |file| require file }

data/lib/config_o_mat/meta_configurator.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'lifecycle_vm'
+
+require 'config_o_mat/shared/op'
+require 'config_o_mat/shared/cond'
+require 'config_o_mat/meta_configurator/op'
+require 'config_o_mat/meta_configurator/memory'
+
+module ConfigOMat
+  module MetaConfigurator
+    class VM < LifecycleVM::VM
+      VERSION = "0.0.1"
+
+      memory_class ConfigOMat::MetaConfigurator::Memory
+
+      on :start, do: Op::ParseMetaCli, then: {
+        case: Cond::EarlyExit,
+        when: {
+          true => :exit,
+          false => :reading_meta_config
+        }
+      }
+
+      on :reading_meta_config, do: Op::LoadMetaConfig, then: :generating_systemd_config
+
+      on :generating_systemd_config, do: Op::GenerateSystemdConfig, then: :exit
+    end
+  end
+end

data/lib/config_o_mat/shared/cond/early_exit.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+require 'lifecycle_vm/cond_base'
+
+module ConfigOMat
+  module Cond
+    class EarlyExit < LifecycleVM::CondBase
+      reads :early_exit
+
+      def call
+        early_exit
+      end
+    end
+  end
+end

data/lib/config_o_mat/shared/cond.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+# Copyright 2021 Teak.io, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+Dir[File.join(__dir__, 'cond', '*')].sort.each { |file| require file }