theforeman/foreman-docker

module Service
  class Containers
    def errors
      @errors ||= []
    end

    def start_container!(wizard_state)
      ActiveRecord::Base.transaction do
        container = create_container_object(wizard_state)
        container.save!
        run_container(container)
        destroy_wizard_state(wizard_state)
        container
      end
    end

    def create_container!(wizard_state)
      ActiveRecord::Base.transaction do
        container = create_container_object(wizard_state)
        container.save!
        destroy_wizard_state(wizard_state)
        container
      end
    end

    def create_container_object(wizard_state)
      container = Container.new do |r|
        r.attributes = wizard_state.container_attributes
        # eagerly load environment variables and exposed ports configuration
        state = DockerContainerWizardState.includes(
          :environment => [:environment_variables, :exposed_ports]).find(wizard_state.id)

        load_environment_variables(state, r)
        load_exposed_ports(state, r)
        load_dns(state, r)
      end

      Taxonomy.enabled_taxonomies.each do |taxonomy|
        container.send(:"#{taxonomy}=", wizard_state.preliminary.send(:"#{taxonomy}"))
      end

      pull_image(container)
      start_container(container)
      unless container.valid?
        @errors = errors + container.errors.full_messages
      end

      if @errors.present?
        @errors = @errors.flatten.uniq
        fail ActiveRecord::Rollback
      end

      container.name = container.in_fog.name[1..-1] unless container.name.present?

      container
    end

    def pull_image(container)
      success = container.compute_resource.
        create_image(:fromImage => container.repository_pull_url)
      return true if success
      @errors = errors + container.compute_resource.errors.full_messages
    end

    def start_container(container)
      started = container.compute_resource.create_container(container.parametrize)
      if started
        container.uuid = started.id
      else
        @errors = errors + container.compute_resource.errors.full_messages
      end
      started
    end

    def destroy_wizard_state(wizard_state)
      wizard_state.destroy
      DockerContainerWizardState.where(["updated_at < ?", (Time.now.utc - 24.hours)]).destroy_all
    end

    def load_environment_variables(state, r)
      state.environment_variables.each do |environment_variable|
        var = r.environment_variables.build
        var.key = environment_variable.key
        var.value = environment_variable.value
      end
    end

    def load_exposed_ports(state, r)
      state.exposed_ports.each do |e|
        port = r.exposed_ports.build
        port.key = e.key
        port.value = e.value
      end
    end

    def load_dns(state, r)
      state.dns.each do |e|
        dns = r.dns.build
        dns.key = e.key
      end
    end

    def full_messages
      return errors.full_messages if errors.respond_to?(:full_messages)
      @errors
    end

    def run_container(container)
      docker_container = container.compute_resource.find_vm_by_uuid(container.uuid)
      error(_('Could not start container')) unless docker_container.send(:start)
    end
  end
end