Sha256: c5f01d418c9cd443e037c2bf088c1f1aa319d99f3359615aa03f63c3036579df

Contents?: true

Size: 1.97 KB

Versions: 21

Compression:

Stored size: 1.97 KB

Contents

#
# Author:: Dylan Page (<dpage@digitalocean.com>)
# Author:: Stafford Brunk (<stafford.brunk@gmail.com>)
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

Ohai.plugin(:DigitalOcean) do
  require_relative "../mixin/do_metadata"
  require_relative "../mixin/http_helper"

  include Ohai::Mixin::DOMetadata
  include Ohai::Mixin::HttpHelper

  provides "digital_ocean"

  depends "dmi"

  # look for digitalocean string in dmi bios data
  def has_do_dmi?
    begin
      # detect a vendor of "DigitalOcean"
      if dmi[:bios][:all_records][0][:Vendor] == "DigitalOcean"
        logger.trace("Plugin DigitalOcean: has_do_dmi? == true")
        return true
      end
    rescue NoMethodError
      # dmi[:bios][:all_records][0][:Vendor] may not exist
    end
    logger.trace("Plugin DigitalOcean: has_do_dmi? == false")
    false
  end

  def looks_like_digital_ocean?
    return true if hint?("digital_ocean")
    return true if has_do_dmi? && can_socket_connect?(Ohai::Mixin::DOMetadata::DO_METADATA_ADDR, 80)

    false
  end

  collect_data do
    if looks_like_digital_ocean?
      logger.trace("Plugin Digitalocean: looks_like_digital_ocean? == true")
      digital_ocean Mash.new
      fetch_metadata.each do |k, v|
        next if k == "vendor_data" # this may have sensitive data we shouldn't store

        digital_ocean[k] = v
      end
    else
      logger.trace("Plugin Digitalocean: No hints present for and doesn't look like digitalocean")
      false
    end
  end
end

Version data entries

21 entries across 21 versions & 1 rubygems

Version Path
ohai-16.6.5 lib/ohai/plugins/digital_ocean.rb
ohai-16.5.6 lib/ohai/plugins/digital_ocean.rb
ohai-16.5.4 lib/ohai/plugins/digital_ocean.rb
ohai-16.5.0 lib/ohai/plugins/digital_ocean.rb
ohai-16.4.12 lib/ohai/plugins/digital_ocean.rb
ohai-16.4.11 lib/ohai/plugins/digital_ocean.rb
ohai-16.3.2 lib/ohai/plugins/digital_ocean.rb
ohai-16.3.0 lib/ohai/plugins/digital_ocean.rb
ohai-16.2.3 lib/ohai/plugins/digital_ocean.rb
ohai-15.12.0 lib/ohai/plugins/digital_ocean.rb
ohai-16.2.1 lib/ohai/plugins/digital_ocean.rb
ohai-16.2.0 lib/ohai/plugins/digital_ocean.rb
ohai-16.1.1 lib/ohai/plugins/digital_ocean.rb
ohai-16.0.20 lib/ohai/plugins/digital_ocean.rb
ohai-15.9.1 lib/ohai/plugins/digital_ocean.rb
ohai-15.8.0 lib/ohai/plugins/digital_ocean.rb
ohai-15.7.4 lib/ohai/plugins/digital_ocean.rb
ohai-15.7.3 lib/ohai/plugins/digital_ocean.rb
ohai-15.6.3 lib/ohai/plugins/digital_ocean.rb
ohai-15.3.1 lib/ohai/plugins/digital_ocean.rb