class MockRbVmomiSnapshot
  attr_accessor :name, :rootSnapshotList, :childSnapshotList

  def initialize
    @name = nil
    @rootSnapshotList = []
    @childSnapshotList = []
  end

  def print_nested_array(arg)
    str = '[ '
    arg.each do |arry|
      if arry.is_a?(Array)
        str += print_nested_array(arry)
      elsif arry.is_a?(MockRbVmomiSnapshot)
        str += arry.to_s + ', '
      end
    end
    str + ' ]'
  end

  def to_s
    "Snapshot(name: #{@name}, rootSnapshotlist: #{print_nested_array(@rootSnapshotList)}, childSnapshotList: #{print_nested_array(@childSnapshotList)})"
  end

  def snapshot
    self
  end
end

class MockRbVmomiVM
  attr_accessor :snapshot, :name, :state

  def info
    self
  end

  def process_snaphash(snaphash)
    shotlist = []
    snaphash.each do |name, subsnaps|
      new_snap = MockRbVmomiSnapshot.new
      new_snap.name = name
      new_snap.childSnapshotList = process_snaphash(subsnaps) if subsnaps.is_a?(Hash)
      shotlist << new_snap
    end
    shotlist
  end

  def get_snapshot(name, snaplist = @snapshot.rootSnapshotList)
    snapshot = nil
    snaplist.each do |snap|
      if snap.is_a?(Array)
        snapshot = get_snapshot(snap, name)
      elsif snap.name == name
        snapshot = snap.snapshot
      end
    end
    snapshot
  end

  def initialize(name, snaphash)
    @name = name
    @snapshot = MockRbVmomiSnapshot.new
    @snapshot.name = name
    @snapshot.rootSnapshotList = process_snaphash(snaphash)
  end
end

class MockRbVmomiConnection
  class CustomizationSpecManager
    class CustomizationSpec
      def spec
        true
      end
    end

    def initialize
      @customizationspec = CustomizationSpec.new
    end

    def GetCustomizationSpec(_arg)
      @customizationspec
    end
  end

  class PropertyCollector
    class Result
      def initialize(name, object)
        @name = name
        @object = object
      end

      def val
        @name
      end

      def obj
        @object
      end

      def propSet
        [self]
      end
    end

    class ResultContainer
      attr_accessor :token

      def initialize
        @results = []
      end

      def objects
        @results
      end

      def add_object(obj)
        @results << obj
      end
    end

    def initialize
      @results = ResultContainer.new
    end

    def RetrievePropertiesEx(_hash)
      @results.token = true
      @results
    end

    def ContinueRetrievePropertiesEx(_token)
      @results.token = false
      @results
    end

    def add_result(name, object)
      @results.add_object(Result.new(name, object))
    end

    def WaitForUpdates(_arg)
      result = OpenStruct.new
      result.version = 'version'
      result
    end

    def CreateFilter(_arg)
      filter = OpenStruct.new
      filter.DestroyPropertyFilter = true
      filter
    end
  end

  class ServiceInstance
    class Datacenter
      attr_accessor :vmFolder, :hostFolder

      def initialize
        @vmFolder = MockRbVmomi::VIM::Folder.new
        @vmFolder.name = '/root'
      end

      def find_datastore(_arg)
        true
      end
    end

    def initialize
      @datacenter = Datacenter.new
    end

    def find_datacenter(_dc)
      @datacenter
    end
  end

  class ServiceManager
    class ViewManager
      def CreateContainerView(hash)
        @view = hash
      end
    end

    def initialize
      @customizationspecmanager = CustomizationSpecManager.new
      @viewmanager = ViewManager.new
    end

    def customizationSpecManager
      @customizationspecmanager
    end

    def viewManager
      @viewmanager
    end

    def rootFolder
      '/root'
    end
  end

  def initialize(opts)
    @host = opts[:host]
    @user = opts[:user]
    @password = opts[:password]
    @insecure = opts[:insecure]
    @serviceinstance = ServiceInstance.new
    @servicemanager = ServiceManager.new
    @propertycollector = PropertyCollector.new
  end

  def serviceInstance
    @serviceinstance
  end

  def serviceContent
    @servicemanager
  end

  def propertyCollector
    @propertycollector
  end

  def set_info(vms)
    vms.each do |vm|
      @propertycollector.add_result(vm.name, vm)
    end
  end
end

class MockRbVmomi
  class VIM
    class Folder
      attr_accessor :name

      def find
        self
      end

      def childEntity
        self
      end

      def resourcePool
        self
      end

      def traverse(_path, _type = Object, _create = false)
        self
      end
    end

    class ResourcePool
      attr_accessor :name

      def find
        self
      end

      def resourcePool
        self
      end
    end

    class ClusterComputeResource
      attr_accessor :name

      def find
        self
      end

      def resourcePool
        self
      end
    end

    class TraversalSpec
      def initialize(hash); end
    end

    def self.connect(opts)
      MockRbVmomiConnection.new(opts)
    end
  end
end