session-android/src/org/thoughtcrime/securesms/avatar/AvatarSelection.java

package org.thoughtcrime.securesms.avatar;

import android.Manifest;
import android.app.Activity;
import android.content.Context;
import android.content.Intent;
import android.content.pm.PackageManager;
import android.net.Uri;
import android.os.Build;
import android.provider.MediaStore;

import androidx.annotation.Nullable;
import androidx.annotation.StringRes;
import androidx.core.content.ContextCompat;

import com.theartofdev.edmodo.cropper.CropImage;
import com.theartofdev.edmodo.cropper.CropImageView;

import org.thoughtcrime.securesms.database.NoExternalStorageException;
import org.thoughtcrime.securesms.logging.Log;
import org.thoughtcrime.securesms.util.ExternalStorageUtil;
import org.thoughtcrime.securesms.util.FileProviderUtil;
import org.thoughtcrime.securesms.util.IntentUtils;

import java.io.File;
import java.io.IOException;
import java.util.LinkedList;
import java.util.List;

import network.loki.messenger.R;

import static android.provider.MediaStore.EXTRA_OUTPUT;

public final class AvatarSelection {

  private static final String TAG = AvatarSelection.class.getSimpleName();

  public static final int REQUEST_CODE_CROP_IMAGE = CropImage.CROP_IMAGE_ACTIVITY_REQUEST_CODE;
  public static final int REQUEST_CODE_AVATAR = REQUEST_CODE_CROP_IMAGE + 1;

  private AvatarSelection() {
  }

  /**
   * Returns result on {@link #REQUEST_CODE_CROP_IMAGE}
   */
  public static void circularCropImage(Activity activity, Uri inputFile, Uri outputFile, @StringRes int title) {
    CropImage.activity(inputFile)
             .setGuidelines(CropImageView.Guidelines.ON)
             .setAspectRatio(1, 1)
             .setCropShape(Build.VERSION.SDK_INT >= Build.VERSION_CODES.P ? CropImageView.CropShape.RECTANGLE : CropImageView.CropShape.OVAL)
             .setOutputUri(outputFile)
             .setAllowRotation(true)
             .setAllowFlipping(true)
             .setBackgroundColor(ContextCompat.getColor(activity, R.color.avatar_background))
             .setActivityTitle(activity.getString(title))
             .start(activity);
  }

  public static Uri getResultUri(Intent data) {
    return CropImage.getActivityResult(data).getUri();
  }

  /**
   * Returns result on {@link #REQUEST_CODE_AVATAR}
   *
   * @return Temporary capture file if created.
   */
  public static File startAvatarSelection(Activity activity, boolean includeClear, boolean attemptToIncludeCamera) {
    File captureFile = null;
    boolean hasCameraPermission = ContextCompat
            .checkSelfPermission(activity, Manifest.permission.CAMERA) == PackageManager.PERMISSION_GRANTED;
    if (attemptToIncludeCamera && hasCameraPermission) {
      try {
        captureFile = File.createTempFile("avatar-capture", ".jpg", ExternalStorageUtil.getImageDir(activity));
      } catch (IOException | NoExternalStorageException e) {
        Log.e("Cannot reserve a temporary avatar capture file.", e);
      }
    }

    Intent chooserIntent = createAvatarSelectionIntent(activity, captureFile, includeClear);
    activity.startActivityForResult(chooserIntent, REQUEST_CODE_AVATAR);
    return captureFile;
  }

  private static Intent createAvatarSelectionIntent(Context context, @Nullable File tempCaptureFile, boolean includeClear) {
    List<Intent> extraIntents  = new LinkedList<>();
    Intent galleryIntent = new Intent(Intent.ACTION_PICK);
    galleryIntent.setDataAndType(android.provider.MediaStore.Images.Media.INTERNAL_CONTENT_URI, "image/*");

    if (!IntentUtils.isResolvable(context, galleryIntent)) {
      galleryIntent = new Intent(Intent.ACTION_GET_CONTENT);
      galleryIntent.setType("image/*");
    }

    if (tempCaptureFile != null) {
      Uri uri = FileProviderUtil.getUriFor(context, tempCaptureFile);
      Intent cameraIntent = new Intent(MediaStore.ACTION_IMAGE_CAPTURE);
      cameraIntent.putExtra(EXTRA_OUTPUT, uri);
      cameraIntent.setFlags(Intent.FLAG_GRANT_WRITE_URI_PERMISSION);
      extraIntents.add(cameraIntent);
    }

    if (includeClear) {
      extraIntents.add(new Intent("network.loki.securesms.action.CLEAR_PROFILE_PHOTO"));
    }

    Intent chooserIntent = Intent.createChooser(galleryIntent, context.getString(R.string.CreateProfileActivity_profile_photo));

    if (!extraIntents.isEmpty()) {
      chooserIntent.putExtra(Intent.EXTRA_INITIAL_INTENTS, extraIntents.toArray(new Intent[0]));
    }

    return chooserIntent;
  }
}