module VagrantPlugins
  module CommandServe
    module Client
      class Machine

        attr_reader :client
        attr_reader :resource_id

        # Create a new instance
        def initialize(name:)
          @client = ServiceInfo.client_for(SDK::MachineService)
          if name.nil?
            @resource_id = ServiceInfo.info.machine
          else
            load_machine(name)
          end
        end

        # Seed this instance with the resource id for the
        # requested machine. If the machine doesn't already
        # exist, create it
        def load_machine(name)
          c = ServiceInfo.client_for(SRV::Vagrant)
          machine = SRV::Machine.new(
            name: name,
            project: SRV::Ref::Project.new(
              resource_id: ServiceInfo.info.project
            ),
            state: SDK::Args::MachineState.new(
              id: :not_created,
              short_description: "not created",
              long_description: "Machine not currently created"
            )
          )

          begin
            resp = c.find_machine(SRV::FindMachineRequest.new(
              machine: machine))
            @resource_id = resp.machine.resource_id
            return
          rescue GRPC::NotFound
            # Let this fall through so we create the machine
          end

          resp = c.upsert_machine(SRV::UpsertMachineRequest.new(
            machine: machine))
          @resource_id = resp.machine.resource_id
        end

        def ref
          SDK::Ref::Machine.new(resource_id: resource_id)
        end

        # @return [String] machine name
        def get_name
          req = SDK::Machine::GetNameRequest.new(
            machine: ref
          )
          client.get_name(req).name
        end

        def set_name(name)
          req = SDK::Machine::SetNameRequest.new(
            machine: ref,
            name: name
          )
          client.set_name(req)
        end

        def get_id
          req = SDK::Machine::GetIDRequest.new(
            machine: ref
          )
          result = client.get_id(req).id
          if result.nil?
            raise "Failed to get machine ID. REF: #{ref.inspect} - ID WAS NIL"
          end
          result
        end

        def set_id(new_id)
          req = SDK::Machine::SetIDRequest.new(
            machine: ref,
            id: new_id
          )
          client.set_id(req)
        end

        def get_box
          req = SDK::Machine::BoxRequest.new(
            machine: ref
          )
          resp = client.box(req)
          Vagrant::Box.new(
            resp.box.name,
            resp.box.provider.to_sym,
            resp.box.version,
            Pathname.new(resp.box.directory),
          )
        end

        def get_data_dir
          req = SDK::Machine::DatadirRequest.new(
            machine: ref
          )
          client.datadir(req).data_dir
        end

        def get_local_data_path
          req = SDK::Machine::LocalDataPathRequest.new(
            machine: ref
          )
          client.localdatapath(req).path
        end

        def get_provider
          req = SDK::Machine::ProviderRequest.new(
            machine: ref
          )
          client.provider(req)
        end

        def get_vagrantfile_name
          req = SDK::Machine::VagrantfileNameRequest.new(
            machine: ref
          )
          resp = client.vagrantfile_name(req)
          resp.name
        end

        def get_vagrantfile_path
          req = SDK::Machine::VagrantfilePathRequest.new(
            machine: ref
          )
          resp = client.vagrantfile_path(req)
          Pathname.new(resp.path)
        end

        def updated_at
          req = SDK::Machine::UpdatedAtRequest.new(
            machine: ref
          )
          resp = client.updated_at(req)
          resp.updated_at
        end

        def get_state
          req = SDK::Machine::GetStateRequest.new(
            machine: ref
          )
          resp = client.get_state(req)
          Vagrant::MachineState.new(
            resp.state.id.to_sym,
            resp.state.short_description,
            resp.state.long_description
          )
        end

        def set_state(state)
          req = SDK::Machine::SetStateRequest.new(
            machine: ref,
            state: SDK::Args::MachineState.new(
              id: state.id.to_s,
              short_description: state.short_description.to_s,
              long_description: state.long_description.to_s
            )
          )
          client.set_state(req)
        end

        def get_uuid
          req = SDK::Machine::GetUUIDRequest.new(
            machine: ref
          )
          client.get_uuid(req).uuid
        end

      end
    end

  end
end