cloudfoundry/cloud_controller_ng

module VCAP::CloudController
  class SetDefaultIsolationSegment
    class Error < StandardError; end

    def set(org, isolation_segment, message)
      iso_seg_guid = message.default_isolation_segment_guid
      if iso_seg_guid
        invalid_relationship!(iso_seg_guid) unless isolation_segment

        entitled_iso_segs = org.isolation_segment_guids
        invalid_relationship!(iso_seg_guid) unless entitled_iso_segs.include?(iso_seg_guid)
      end

      org.db.transaction do
        org.lock!

        org.default_isolation_segment_guid = iso_seg_guid if message.requested?(:data)

        org.save
      end
    rescue Sequel::ValidationFailed => e
      raise Error.new(e.message)
    end

    private

    def invalid_relationship!(isolation_segment_guid)
      raise Error.new "Unable to assign isolation segment with guid '#{isolation_segment_guid}'. Ensure it has been entitled to the organization."
    end
  end
end