cloudfoundry/cloud_controller_ng

module VCAP
  class VarsBuilder
    def initialize(process,
                   memory_limit: nil,
                   staging_disk_in_mb: nil,
                   space: nil,
                   file_descriptors: nil,
                   version: nil)
      @process = process
      @staging_disk_in_mb = staging_disk_in_mb
      @memory_limit = memory_limit
      @space = space
      @file_descriptors = file_descriptors
      @version = version
    end

    def to_hash
      app_name = @process.name

      if @process.is_a?(VCAP::CloudController::AppModel)
        uris = @process.routes.map(&:uri)
      else
        @staging_disk_in_mb ||= @process.disk_quota
        @memory_limit ||= @process.memory
        @file_descriptors ||= @process.file_descriptors
        @version = @process.version
        uris = @process.uris
      end

      @space = @process.space if @space.nil?

      my_uri        = URI::HTTP.build(host: VCAP::CloudController::Config.config.get(:external_domain))
      my_uri.scheme = VCAP::CloudController::Config.config.get(:external_protocol)

      env_hash = {
        cf_api: my_uri.to_s,
        limits: {},
        application_name: app_name,
        application_uris: uris,
        name: @process.name,
        space_name: @space.name,
        space_id: @space.guid,
        organization_id: @space.organization_guid,
        organization_name: @space.organization.name,
        uris: uris,
        users: nil
      }

      if @process.is_a?(VCAP::CloudController::ProcessModel)
        env_hash[:process_id] = @process.guid
        env_hash[:process_type] = @process.type
        env_hash[:application_id] = @process.app_guid
      else # process is an AppModel
        env_hash[:application_id] = @process.guid
      end

      env_hash[:limits][:fds] = @file_descriptors if @file_descriptors
      env_hash[:limits][:mem] = @memory_limit if @memory_limit
      env_hash[:limits][:disk] = @staging_disk_in_mb if @staging_disk_in_mb

      unless @version.nil?
        env_hash[:version] = @version
        env_hash[:application_version] = @version
      end

      env_hash
    end
  end
end