/*
* 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.models.sqlite;
import android.support.annotation.*;
import android.support.annotation.Nullable;
import com.activeandroid.query.*;
import org.isoron.uhabits.models.*;
import org.isoron.uhabits.models.sqlite.records.*;
import org.isoron.uhabits.utils.*;
import org.jetbrains.annotations.*;
import java.util.*;
/**
* Implementation of a StreakList that is backed by SQLite.
*/
public class SQLiteStreakList extends StreakList
{
private HabitRecord habitRecord;
@NonNull
private final SQLiteUtils<StreakRecord> sqlite;
public SQLiteStreakList(Habit habit)
{
super(habit);
sqlite = new SQLiteUtils<>(StreakRecord.class);
}
@Override
public List<Streak> getAll()
{
check(habit.getId());
rebuild();
String query = StreakRecord.SELECT + "where habit = ? " +
"order by end desc";
String params[] = { Long.toString(habit.getId())};
List<StreakRecord> records = sqlite.query(query, params);
return recordsToStreaks(records);
}
@Override
public Streak getNewestComputed()
{
StreakRecord newestRecord = getNewestRecord();
if (newestRecord == null) return null;
return newestRecord.toStreak();
}
@Override
public void invalidateNewerThan(long timestamp)
{
new Delete()
.from(StreakRecord.class)
.where("habit = ?", habit.getId())
.and("end >= ?", timestamp - DateUtils.millisecondsInOneDay)
.execute();
observable.notifyListeners();
}
@Override
protected void add(@NonNull List<Streak> streaks)
{
check(habit.getId());
DatabaseUtils.executeAsTransaction(() -> {
for (Streak streak : streaks)
{
StreakRecord record = new StreakRecord();
record.copyFrom(streak);
record.habit = habitRecord;
record.save();
}
});
}
@Override
protected void removeNewestComputed()
{
StreakRecord newestStreak = getNewestRecord();
if (newestStreak != null) newestStreak.delete();
}
@Nullable
private StreakRecord getNewestRecord()
{
check(habit.getId());
String query = StreakRecord.SELECT + "where habit = ? " +
"order by end desc " +
"limit 1 ";
String params[] = { habit.getId().toString() };
StreakRecord record = sqlite.querySingle(query, params);
if (record != null) record.habit = habitRecord;
return record;
}
@NonNull
private List<Streak> recordsToStreaks(List<StreakRecord> records)
{
LinkedList<Streak> streaks = new LinkedList<>();
for (StreakRecord record : records)
{
record.habit = habitRecord;
streaks.add(record.toStreak());
}
return streaks;
}
@Contract("null -> fail")
private void check(Long id)
{
if (id == null) throw new RuntimeException("habit is not saved");
if(habitRecord != null) return;
habitRecord = HabitRecord.get(id);
if (habitRecord == null) throw new RuntimeException("habit not found");
}
}