tasks/app/src/main/java/com/todoroo/astrid/dao/Database.java

/**
 * Copyright (c) 2012 Todoroo Inc
 *
 * <p>See the file "LICENSE" for the full license governing this code.
 */
package com.todoroo.astrid.dao;

import androidx.sqlite.db.SupportSQLiteDatabase;
import androidx.room.RoomDatabase;
import android.database.Cursor;
import com.todoroo.astrid.data.Task;
import java.io.IOException;
import org.tasks.analytics.Tracker;
import org.tasks.data.Alarm;
import org.tasks.data.AlarmDao;
import org.tasks.data.CaldavAccount;
import org.tasks.data.CaldavCalendar;
import org.tasks.data.CaldavDao;
import org.tasks.data.CaldavTask;
import org.tasks.data.DeletionDao;
import org.tasks.data.Filter;
import org.tasks.data.FilterDao;
import org.tasks.data.GoogleTask;
import org.tasks.data.GoogleTaskAccount;
import org.tasks.data.GoogleTaskDao;
import org.tasks.data.GoogleTaskList;
import org.tasks.data.GoogleTaskListDao;
import org.tasks.data.Location;
import org.tasks.data.LocationDao;
import org.tasks.data.Tag;
import org.tasks.data.TagDao;
import org.tasks.data.TagData;
import org.tasks.data.TagDataDao;
import org.tasks.data.TaskAttachment;
import org.tasks.data.TaskAttachmentDao;
import org.tasks.data.TaskListMetadata;
import org.tasks.data.TaskListMetadataDao;
import org.tasks.data.UserActivity;
import org.tasks.data.UserActivityDao;
import org.tasks.notifications.Notification;
import org.tasks.notifications.NotificationDao;
import timber.log.Timber;

@androidx.room.Database(
    entities = {
      Notification.class,
      TagData.class,
      UserActivity.class,
      TaskAttachment.class,
      TaskListMetadata.class,
      Task.class,
      Alarm.class,
      Location.class,
      Tag.class,
      GoogleTask.class,
      Filter.class,
      GoogleTaskList.class,
      CaldavCalendar.class,
      CaldavTask.class,
      CaldavAccount.class,
      GoogleTaskAccount.class
    },
    version = 59)
public abstract class Database extends RoomDatabase {

  public static final String NAME = "database";
  private SupportSQLiteDatabase database;
  private Tracker tracker;
  private Runnable onDatabaseUpdated;

  public abstract NotificationDao notificationDao();

  public abstract TagDataDao getTagDataDao();

  public abstract UserActivityDao getUserActivityDao();

  public abstract TaskAttachmentDao getTaskAttachmentDao();

  public abstract TaskListMetadataDao getTaskListMetadataDao();

  public abstract AlarmDao getAlarmDao();

  public abstract LocationDao getLocationDao();

  public abstract TagDao getTagDao();

  public abstract GoogleTaskDao getGoogleTaskDao();

  public abstract FilterDao getFilterDao();

  public abstract GoogleTaskListDao getGoogleTaskListDao();

  public abstract TaskDao getTaskDao();

  public abstract CaldavDao getCaldavDao();

  public abstract DeletionDao getDeletionDao();

  // --- implementation

  public String getName() {
    return NAME;
  }

  public Database init(Tracker tracker, Runnable onDatabaseUpdated) {
    this.tracker = tracker;
    this.onDatabaseUpdated = onDatabaseUpdated;
    return this;
  }

  public void onDatabaseUpdated() {
    if (onDatabaseUpdated != null) {
      onDatabaseUpdated.run();
    }
  }

  /**
   * Open the database for writing. Must be closed afterwards. If user is out of disk space,
   * database may be opened for reading instead
   */
  public final synchronized void openForWriting() {
    if (database != null && !database.isReadOnly() && database.isOpen()) {
      return;
    }

    try {
      database = getOpenHelper().getWritableDatabase();
    } catch (Exception e) {
      tracker.reportException(e);
      throw new IllegalStateException(e);
    }
  }

  /** Open the database for reading. Must be closed afterwards */
  public final synchronized void openForReading() {
    if (database != null && database.isOpen()) {
      return;
    }
    database = getOpenHelper().getReadableDatabase();
  }

  /** Close the database if it has been opened previously */
  @Override
  public final synchronized void close() {
    if (database != null) {
      try {
        database.close();
      } catch (IOException e) {
        Timber.e(e);
      }
    }
    database = null;
  }

  /** @return sql database. opens database if not yet open */
  private synchronized SupportSQLiteDatabase getDatabase() {
    if (database == null) {
      openForWriting();
    }
    return database;
  }

  /** @return human-readable database name for debugging */
  @Override
  public String toString() {
    return "DB:" + getName();
  }

  // --- database wrapper

  public Cursor rawQuery(String sql) {
    return getDatabase().query(sql, null);
  }
}