Plugins » 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