package org.ryuslash.Dailies;

import java.text.DateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

import android.content.ContentValues;
import android.content.Context;
import android.database.Cursor;
import android.database.SQLException;
import android.database.sqlite.SQLiteDatabase;

public class DailiesDataSource
{
    private SQLiteDatabase database;
    private SQLiteHelper dbHelper;
    private String[] allColumns = { SQLiteHelper.COLUMN_ID,
                                    SQLiteHelper.COLUMN_TITLE,
                                    SQLiteHelper.COLUMN_ALARM,
                                    SQLiteHelper.COLUMN_HOUR,
                                    SQLiteHelper.COLUMN_MINUTE,
                                    SQLiteHelper.COLUMN_LAST_COMPLETED };

    public DailiesDataSource(Context context)
    {
        dbHelper = new SQLiteHelper(context);
    }

    public void open() throws SQLException
    {
        database = dbHelper.getWritableDatabase();
    }

    public void close()
    {
        dbHelper.close();
    }

    private String dateToString(Date date)
    {
        DateFormat df = DateFormat.getDateInstance();
        return df.format(date);
    }

    private ContentValues createValues(String title, boolean alarm,
                                       int hour, int minute, Date date)
    {
        ContentValues values = new ContentValues();


        values.put(SQLiteHelper.COLUMN_TITLE, title);
        values.put(SQLiteHelper.COLUMN_ALARM, alarm ? 1 : 0);
        values.put(SQLiteHelper.COLUMN_HOUR, hour);
        values.put(SQLiteHelper.COLUMN_MINUTE, minute);

        if (date != null)
            values.put(SQLiteHelper.COLUMN_LAST_COMPLETED,
                       dateToString(date));

        return values;
    }

    private ContentValues createValues(String title, boolean alarm,
                                       int  hour, int minute)
    {
        return createValues(title, alarm, hour, minute, null);
    }

    public Daily createDaily(String title, boolean alarm, int hour,
                             int minute)
    {
        ContentValues values = createValues(title, alarm, hour, minute);
        long insertId = database.insert(SQLiteHelper.TABLE_DAILY, null,
                                        values);
        Daily daily = new Daily(insertId, title, alarm, hour, minute);

        return daily;
    }

    public void updateDaily(long id, String title, boolean alarm,
                            int hour, int minute, Date last_completed)
    {
        ContentValues values = createValues(title, alarm, hour, minute,
                                            last_completed);
        String whereArgs[] = { "" + id };

        database.update(SQLiteHelper.TABLE_DAILY, values, "id = ?",
                        whereArgs);
    }

    public void deleteDaily(Daily daily)
    {
        deleteDaily(daily.getId());
    }

    public void deleteDaily(long id)
    {
        database.delete(SQLiteHelper.TABLE_DAILY,
                        SQLiteHelper.COLUMN_ID + " = " + id, null);
    }

    public List<Daily> getDailiesList()
    {
        List<Daily> dailies = new ArrayList<Daily>();
        Cursor cursor = database.query(
            SQLiteHelper.TABLE_DAILY, allColumns, null, null, null,
            null, "last_completed DESC, hour, minute"
        );

        cursor.moveToFirst();

        while(!cursor.isAfterLast()) {
            Daily daily = cursorToDaily(cursor);

            dailies.add(daily);
            cursor.moveToNext();
        }

        cursor.close();

        return dailies;
    }

    public Daily getById(long id)
    {
        Cursor cursor = database.query(SQLiteHelper.TABLE_DAILY,
                                       allColumns,
                                       SQLiteHelper.COLUMN_ID + " = ?",
                                       new String[] { "" + id },
                                       null, null, null);
        cursor.moveToFirst();
        Daily daily = cursorToDaily(cursor);
        cursor.close();

        return daily;
    }

    private Daily cursorToDaily(Cursor cursor)
    {
        Daily daily = new Daily();

        daily.setId(cursor.getLong(0));
        daily.setTitle(cursor.getString(1));
        daily.setAlarm(cursor.getInt(2) == 1);
        daily.setHour(cursor.getInt(3));
        daily.setMinute(cursor.getInt(4));
        daily.setLastCompleted(cursor.getString(5));

        return daily;
    }
}