Signal-Android/app/src/main/java/org/thoughtcrime/securesms/devicetransfer/olddevice/OldDeviceTransferFragment.java

package org.thoughtcrime.securesms.devicetransfer.olddevice;

import android.os.Bundle;
import android.view.View;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.navigation.fragment.NavHostFragment;

import org.greenrobot.eventbus.EventBus;
import org.greenrobot.eventbus.Subscribe;
import org.greenrobot.eventbus.ThreadMode;
import org.signal.devicetransfer.DeviceToDeviceTransferService;
import org.signal.devicetransfer.TransferStatus;
import org.thoughtcrime.securesms.R;
import org.thoughtcrime.securesms.devicetransfer.DeviceTransferFragment;

/**
 * Shows transfer progress on the old device. Most logic is in {@link DeviceTransferFragment}
 * and it delegates to this class for strings, navigation, and updating progress.
 */
public final class OldDeviceTransferFragment extends DeviceTransferFragment {

  private final ClientTaskListener clientTaskListener = new ClientTaskListener();

  @Override
  public void onViewCreated(@NonNull View view, @Nullable Bundle savedInstanceState) {
    super.onViewCreated(view, savedInstanceState);
    EventBus.getDefault().register(clientTaskListener);
  }

  @Override
  public void onDestroyView() {
    EventBus.getDefault().unregister(clientTaskListener);
    super.onDestroyView();
  }

  @Override
  protected void navigateToRestartTransfer() {
    NavHostFragment.findNavController(this).navigate(R.id.action_directly_to_oldDeviceTransferInstructions);
  }

  @Override
  protected void navigateAwayFromTransfer() {
    EventBus.getDefault().unregister(clientTaskListener);
    requireActivity().finish();
  }

  @Override
  protected void navigateToTransferComplete() {
    NavHostFragment.findNavController(this).navigate(R.id.action_oldDeviceTransfer_to_oldDeviceTransferComplete);
  }

  private class ClientTaskListener {
    @Subscribe(threadMode = ThreadMode.MAIN)
    public void onEventMainThread(@NonNull OldDeviceClientTask.Status event) {
      if (event.isDone()) {
        transferFinished = true;
        ignoreTransferStatusEvents();
        EventBus.getDefault().removeStickyEvent(TransferStatus.class);
        DeviceToDeviceTransferService.stop(requireContext());
        NavHostFragment.findNavController(OldDeviceTransferFragment.this).navigate(R.id.action_oldDeviceTransfer_to_oldDeviceTransferComplete);
      } else {
        status.setText(getString(R.string.DeviceTransfer__d_messages_so_far, event.getMessageCount()));
      }
    }
  }
}