1 files changed, 80 insertions, 0 deletions
diff --git a/chef-helpers/lib/chef-helpers/introspection.rb b/chef-helpers/lib/chef-helpers/introspection.rb
new file mode 100644
index 0000000000..85941bc878
--- /dev/null
+++ b/chef-helpers/lib/chef-helpers/introspection.rb
@@ -0,0 +1,80 @@
+#
+# Copyright:: Copyright 2018-2018, Chef Software Inc.
+# 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.
+#
+
+
+module ChefHelpers
+  # This is for "introspection" helpers in the sense that we are inspecting the
+  # actual server or image under management to determine running state (duck-typing the system).
+  # The helpers here may use the node object state from ohai, but typically not the big 5:  platform,
+  # platform_family, platform_version, arch, os.  The helpers here should infer somewhat
+  # higher level facts about the system.
+  #
+  module Introspection
+    # Returns whether the node is a docker container.
+    #
+    # @param [Chef::Node] node
+    #
+    # @return [Boolean]
+    #
+    def docker?(node = Internal.getnode)
+      # Using "File.exist?('/.dockerinit') || File.exist?('/.dockerenv')" makes Travis sad,
+      # and that makes us sad too.
+      !!(node && node.read("virtualization", "systems", "docker") == "guest")
+    end
+
+    # @param [Chef::Node] node
+    #
+    # @return [Boolean]
+    #
+    def systemd?(node = Internal.getnode)
+      ::File.exist?("/proc/1/comm") && ::File.open("/proc/1/comm").gets.chomp == "systemd"
+    end
+
+    # @param [Chef::Node] node
+    #
+    # @return [Boolean]
+    #
+    def kitchen?(node = Internal.getnode)
+      ENV.key?('TEST_KITCHEN')
+    end
+
+    # @param [Chef::Node] node
+    #
+    # @return [Boolean]
+    #
+    def ci?(node = Internal.getnode)
+      ENV.key?('CI')
+    end
+
+    class << self
+      def has_systemd_service_unit?(svc_name)
+        %w{ /etc /usr/lib /lib /run }.any? do |load_path|
+          ::File.exist?(
+            "#{load_path}/systemd/system/#{svc_name.gsub(/@.*$/, '@')}.service"
+          )
+        end
+      end
+
+      def has_systemd_unit?(svc_name)
+        # TODO: stop supporting non-service units with service resource
+        %w{ /etc /usr/lib /lib /run }.any? do |load_path|
+          ::File.exist?("#{load_path}/systemd/system/#{svc_name}")
+        end
+      end
+    end
+  end
+end