class TestLab
  class Node

    module Doctor

      # Node Doctor
      #
      # Attempts to analyze the current node and report any issues.
      #
      # @return [Boolean] True if everything is OK; false otherwise.
      def doctor
        if self.dead?
          @ui.stderr.puts(format_message("The node #{self.id.inspect} is dead! (Did you forget to up or build the node?)".red.bold))
          return false
        end

        if !self.lxc.installed?
          @ui.stderr.puts(format_message("LXC does not appear to be installed on your TestLab node!  (Did you forget to provision or build the node?)".red.bold))
          return false
        end

        result = true

        # make sure the node has some free space
        free_space_percent = (self.exec(%(df -P /), :ignore_exit_status => true).output.split("\n")[1].split[-2].to_i rescue nil)
        if free_space_percent.nil?
          @ui.stderr.puts(format_message("ERROR: We could not determine how much free space node #{self.id.inspect} has!".red.bold))
          result = false
        elsif (free_space_percent >= 90)
          @ui.stderr.puts(format_message("WARNING: Your TestLab node #{self.id.inspect} is using #{free_space_percent}% of its available disk space!".red.bold))
          result = false
        end

        # get the names of all of the defined containers
        my_container_names = self.containers.map(&:id)

        # ephemeral containers parent containers have a "-master" suffix; we need to remove these from the results or we will complain about them
        node_container_names = self.lxc.containers.map(&:name).delete_if do |node_container_name|
          my_container_names.any?{ |my_container_name| "#{my_container_name}-master" == node_container_name }
        end

        unknown_container_names = (node_container_names - my_container_names)
        unknown_running_container_names = self.lxc.containers.select{ |c| (unknown_container_names.include?(c.name) && (c.state == :running)) }.map(&:name)

        if unknown_container_names.count > 0
          if unknown_running_container_names.count > 0
            @ui.stderr.puts(format_message("WARNING: You have *running* containers on your TestLab node #{self.id.inspect} which are not defined in your Labfile!".red.bold))
            @ui.stderr.puts(format_message(">>> You may need to manually stop the following containers: #{unknown_running_container_names.join(', ')}".red.bold))
            result = false
          end

          @ui.stderr.puts(format_message("WARNING: You have containers on your TestLab node #{self.id.inspect} which are not defined in your Labfile!".red.bold))
          @ui.stderr.puts(format_message(">>> You may need to manually remove the following containers: #{unknown_container_names.join(', ')}".red.bold))
        end

        result
      end

    end

  end
end