ManageIQ/manageiq

class MiqWorker
  module ServiceWorker
    extend ActiveSupport::Concern

    HEALTH_PORT  = 4000
    SERVICE_PORT = 3000

    def create_container_objects
      orchestrator = ContainerOrchestrator.new

      orchestrator.create_deployment(worker_deployment_name) do |definition|
        configure_worker_deployment(definition)
        configure_service_worker_deployment(definition)

        add_liveness_probe(definition[:spec][:template][:spec][:containers].first)
        add_readiness_probe(definition[:spec][:template][:spec][:containers].first)
      end

      scale_deployment
    end

    def delete_container_objects
      orch = ContainerOrchestrator.new
      orch.delete_deployment(worker_deployment_name)
    end

    def stop_container
      scale_deployment
    end

    def add_liveness_probe(container_definition)
      container_definition[:livenessProbe] = container_definition[:livenessProbe].except(:exec).merge(:httpGet => {:path => "/ping", :port => 4000})
    end

    def add_readiness_probe(container_definition)
      container_definition[:readinessProbe] = {
        :httpGet             => {:path => "/ping", :port => HEALTH_PORT},
        :initialDelaySeconds => 60,
        :timeoutSeconds      => 3
      }
    end

    def configure_service_worker_deployment(definition)
      definition[:spec][:template][:metadata][:labels].merge!(service_label)

      container = definition[:spec][:template][:spec][:containers].first
      container[:ports] = [{:containerPort => SERVICE_PORT}, {:containerPort => HEALTH_PORT}]
      container[:volumeMounts] ||= []
      definition[:spec][:template][:spec][:volumes] ||= []
    end

    def service_label
      {:service => worker_deployment_name.delete_prefix(deployment_prefix)}
    end

    # Can be overriden by including classes
    def container_port
      SERVICE_PORT
    end

    # Can be overriden by including classes
    def container_image_name
      "manageiq-webserver-worker"
    end

    def container_image
      ENV["WEBSERVER_WORKER_IMAGE"] || default_image
    end

    def container_environment_variables
      super.merge("PORT" => container_port.to_s, "BINDING_ADDRESS" => "0.0.0.0")
    end
  end
end