Signal-Android/app/src/main/java/org/thoughtcrime/securesms/util/ObservingLiveData.java

package org.thoughtcrime.securesms.util;

import android.database.ContentObserver;

import androidx.annotation.NonNull;
import androidx.lifecycle.MutableLiveData;

import org.signal.core.util.StreamUtil;
import org.thoughtcrime.securesms.database.ObservableContent;

import java.io.Closeable;

/**
 * Implementation of {@link androidx.lifecycle.LiveData} that will handle closing the contained
 * {@link Closeable} when the value changes.
 */
public class ObservingLiveData<E extends ObservableContent> extends MutableLiveData<E> {

  private ContentObserver observer;

  @Override
  public void setValue(E value) {
    E previous = getValue();

    if (previous != null) {
      previous.unregisterContentObserver(observer);
      StreamUtil.close(previous);
    }

    value.registerContentObserver(observer);

    super.setValue(value);
  }

  public void close() {
    E value = getValue();

    if (value != null) {
      value.unregisterContentObserver(observer);
      StreamUtil.close(value);
    }
  }

  public void registerContentObserver(@NonNull ContentObserver observer) {
    this.observer = observer;
  }
}