module VagrantPlugins
  module CommandServe
    class Mappers
      # Build a machine client from a FuncSpec value
      class MachineFromSpec < Mapper
        def initialize
          inputs = [].tap do |i|
            i << Input.new(type: SDK::FuncSpec::Value) { |arg|
              arg.type == "hashicorp.vagrant.sdk.Args.Machine" &&
                !arg&.value&.value.nil?
            }
            i << Input.new(type: Broker)
          end
          super(inputs: inputs, output: Client::Machine, func: method(:converter))
        end

        def converter(proto, broker)
          Client::Machine.load(proto.value.value, broker: broker)
        end
      end

      # Build a machine client from a proto instance
      class MachineFromProto < Mapper
        def initialize
          inputs = [].tap do |i|
            i << Input.new(type: SDK::Args::Target::Machine)
            i << Input.new(type: Broker)
          end
          super(inputs: inputs, output: Client::Machine, func: method(:converter))
        end

        def converter(proto, broker)
          Client::Machine.load(proto, broker: broker)
        end
      end

      # Build a machine client from a serialized proto string
      class MachineFromString < Mapper
        def initialize
          inputs = [].tap do |i|
            i << Input.new(type: String)
            i << Input.new(type: Broker)
          end
          super(inputs: inputs, output: Client::Machine, func: method(:converter))
        end

        def converter(proto, broker)
          Client::Machine.load(proto, broker: broker)
        end
      end

      # Build a machine from a target
      class MachineFromTarget < Mapper
        def initialize
          inputs = [].tap do |i|
            i << Input.new(type: Vagrant::Environment)
            i << Input.new(type: Client::Target)
          end
          super(inputs: inputs, output: Vagrant::Machine, func: method(:converter))
        end

        def converter(env, target)
          env.machine(
            target.name.to_sym,
            target.provider_name.to_sym,
          )
        end
      end

      # Build a machine from a machine client
      class MachineFromMachineClient < Mapper
        def initialize
          inputs = [].tap do |i|
            i << Input.new(type: Vagrant::Environment)
            i << Input.new(type: Client::Machine)
          end
          super(inputs: inputs, output: Vagrant::Machine, func: method(:converter))
        end

        def converter(env, machine)
          env.machine(
            machine.name.to_sym,
            machine.provider_name.to_sym,
          )
        end
      end
    end
  end
end