org.isoron.uhabits.helpers.DatabaseHelper.java Source code

Introduction

Here is the source code for org.isoron.uhabits.helpers.DatabaseHelper.java

Source

/*
 * Copyright (C) 2016 ?linson Santos Xavier <isoron@gmail.com>
 *
 * This file is part of Loop Habit Tracker.
 *
 * Loop Habit Tracker is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by the
 * Free Software Foundation, either version 3 of the License, or (at your
 * option) any later version.
 *
 * Loop Habit Tracker is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program. If not, see <http://www.gnu.org/licenses/>.
 */

package org.isoron.uhabits.helpers;

import android.content.Context;
import android.database.Cursor;
import android.os.Environment;
import android.support.annotation.NonNull;
import android.support.annotation.Nullable;
import android.support.v4.content.ContextCompat;
import android.util.Log;

import com.activeandroid.ActiveAndroid;
import com.activeandroid.Cache;
import com.activeandroid.Configuration;

import org.isoron.uhabits.BuildConfig;
import org.isoron.uhabits.HabitsApplication;
import org.isoron.uhabits.models.Checkmark;
import org.isoron.uhabits.models.Habit;
import org.isoron.uhabits.models.Repetition;
import org.isoron.uhabits.models.Score;
import org.isoron.uhabits.models.Streak;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.text.SimpleDateFormat;

public class DatabaseHelper {
    public static void copy(File src, File dst) throws IOException {
        FileInputStream inStream = new FileInputStream(src);
        FileOutputStream outStream = new FileOutputStream(dst);
        copy(inStream, outStream);
    }

    public static void copy(InputStream inStream, File dst) throws IOException {
        FileOutputStream outStream = new FileOutputStream(dst);
        copy(inStream, outStream);
    }

    public static void copy(InputStream in, OutputStream out) throws IOException {
        int numBytes;
        byte[] buffer = new byte[1024];

        while ((numBytes = in.read(buffer)) != -1)
            out.write(buffer, 0, numBytes);
    }

    public interface Command {
        void execute();
    }

    public static void executeAsTransaction(Command command) {
        ActiveAndroid.beginTransaction();
        try {
            command.execute();
            ActiveAndroid.setTransactionSuccessful();
        } finally {
            ActiveAndroid.endTransaction();
        }
    }

    @SuppressWarnings("ResultOfMethodCallIgnored")
    public static String saveDatabaseCopy(File dir) throws IOException {
        File db = getDatabaseFile();

        SimpleDateFormat dateFormat = DateHelper.getBackupDateFormat();
        String date = dateFormat.format(DateHelper.getLocalTime());
        File dbCopy = new File(String.format("%s/Loop Habits Backup %s.db", dir.getAbsolutePath(), date));

        copy(db, dbCopy);

        return dbCopy.getAbsolutePath();
    }

    @NonNull
    public static File getDatabaseFile() {
        Context context = HabitsApplication.getContext();
        if (context == null)
            throw new RuntimeException("No application context found");

        String databaseFilename = getDatabaseFilename();

        return new File(String.format("%s/../databases/%s", context.getApplicationContext().getFilesDir().getPath(),
                databaseFilename));
    }

    @NonNull
    public static String getDatabaseFilename() {
        String databaseFilename = BuildConfig.databaseFilename;

        if (HabitsApplication.isTestMode())
            databaseFilename = "test.db";

        return databaseFilename;
    }

    @Nullable
    public static File getSDCardDir(@Nullable String relativePath) {
        File parents[] = new File[] { Environment.getExternalStorageDirectory() };
        return getDir(parents, relativePath);
    }

    @Nullable
    public static File getFilesDir(@Nullable String relativePath) {
        Context context = HabitsApplication.getContext();
        if (context == null) {
            Log.e("DatabaseHelper", "getFilesDir: no application context available");
            return null;
        }

        File externalFilesDirs[] = ContextCompat.getExternalFilesDirs(context, null);

        if (externalFilesDirs == null) {
            Log.e("DatabaseHelper", "getFilesDir: getExternalFilesDirs returned null");
            return null;
        }

        return getDir(externalFilesDirs, relativePath);
    }

    @Nullable
    private static File getDir(@NonNull File potentialParentDirs[], @Nullable String relativePath) {
        if (relativePath == null)
            relativePath = "";

        File chosenDir = null;
        for (File dir : potentialParentDirs) {
            if (dir == null || !dir.canWrite())
                continue;
            chosenDir = dir;
            break;
        }

        if (chosenDir == null) {
            Log.e("DatabaseHelper", "getDir: all potential parents are null or non-writable");
            return null;
        }

        File dir = new File(String.format("%s/%s/", chosenDir.getAbsolutePath(), relativePath));
        if (!dir.exists() && !dir.mkdirs()) {
            Log.e("DatabaseHelper", "getDir: chosen dir does not exist and cannot be created");
            return null;
        }

        return dir;
    }

    @SuppressWarnings("unchecked")
    public static void initializeActiveAndroid() {
        Context context = HabitsApplication.getContext();
        if (context == null)
            throw new RuntimeException("application context should not be null");

        Configuration dbConfig = new Configuration.Builder(context).setDatabaseName(getDatabaseFilename())
                .setDatabaseVersion(BuildConfig.databaseVersion)
                .addModelClasses(Checkmark.class, Habit.class, Repetition.class, Score.class, Streak.class)
                .create();

        ActiveAndroid.initialize(dbConfig);
    }

    public static long longQuery(String query, String args[]) {
        Cursor c = null;

        try {
            c = Cache.openDatabase().rawQuery(query, args);
            if (!c.moveToFirst())
                return 0;
            return c.getLong(0);
        } finally {
            if (c != null)
                c.close();
        }
    }
}