GoogleDriveUploadHelper.java

/*
Copyright (C) 2012 Haowen Ning

This program 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 2
of the License, or (at your option) any later version.

This program 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, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

*/
package org.liberty.android.fantastischmemo.downloader.google;

import android.content.Context;

import org.liberty.android.fantastischmemo.common.AnyMemoDBOpenHelper;
import org.liberty.android.fantastischmemo.common.AnyMemoDBOpenHelperManager;
import org.liberty.android.fantastischmemo.dao.CardDao;
import org.liberty.android.fantastischmemo.dao.CategoryDao;
import org.liberty.android.fantastischmemo.dao.LearningDataDao;
import org.liberty.android.fantastischmemo.entity.Card;
import org.liberty.android.fantastischmemo.entity.LearningData;

import java.text.SimpleDateFormat;
import java.util.Iterator;
import java.util.List;
import java.util.concurrent.Callable;

public class GoogleDriveUploadHelper {
    private Context mContext;

    private final String authToken;

    private static SimpleDateFormat ISO8601_FORMATTER = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSS");

    public GoogleDriveUploadHelper(Context context, String authToken) {
        this.authToken = authToken;
        mContext = context;
    }

    public Spreadsheet createSpreadsheet(String title, String dbPath) throws Exception {


        // First read card because if it failed we don't even bother uploading.
        AnyMemoDBOpenHelper helper = AnyMemoDBOpenHelperManager.getHelper(mContext, dbPath);
        List<Card> cardList = null;
        try {
            final CardDao cardDao = helper.getCardDao();
            final CategoryDao categoryDao = helper.getCategoryDao();
            final LearningDataDao learningDataDao = helper.getLearningDataDao();
            cardList = cardDao.callBatchTasks(new Callable<List<Card>>() {
                public List<Card> call() throws Exception {
                    List<Card> cards = cardDao.queryForAll();
                    for (Card c: cards) {
                        categoryDao.refresh(c.getCategory());
                        learningDataDao.refresh(c.getLearningData());
                    }
                    return cards;
                }
            });
        } finally {
            AnyMemoDBOpenHelperManager.releaseHelper(helper);
        }

        // Find the spreadsheets to delete after the process is done
        List<Document> spreadsheetsToDelete = DocumentFactory.findDocuments(title, authToken);

        // Create the AnyMemo folder if needed
        Folder folder = FolderFactory.createOrReturnFolder("AnyMemo", authToken);

        // Create new spreadsheet
        Document newSpreadsheetDocument = DocumentFactory.createSpreadsheet(title, authToken);
        List<Spreadsheet> spreadsheetList = SpreadsheetFactory.getSpreadsheets(authToken);
        Spreadsheet newSpreadsheet = spreadsheetList.get(0);

        // Create worksheets
        List<Worksheet> worksheetsToDelete = WorksheetFactory.getWorksheets(newSpreadsheet, authToken);
        // Delete the worksheets with the duplicated name
        // First create a dummy worksheet so we have at least one worksheet
        boolean needsToCreateDummy = true;
        for (Worksheet ws: worksheetsToDelete) {
            if (ws.getTitle().equals("dummy")) {
                needsToCreateDummy = false;
                break;
            }
        }

        // Dummy worksheet will be deleted at the last step
        if (needsToCreateDummy) {
            Worksheet dummyWorksheet = WorksheetFactory.createWorksheet(newSpreadsheet, "dummy", 1, 1, authToken);
            worksheetsToDelete.add(dummyWorksheet);
        }

        Iterator<Worksheet> worksheetToDeleteIterator = worksheetsToDelete.iterator();
        while (worksheetToDeleteIterator.hasNext()) {
            Worksheet ws = worksheetToDeleteIterator.next();
            if (ws.getTitle().equals("cards")) {
                WorksheetFactory.deleteWorksheet(newSpreadsheet, ws, authToken);
                worksheetToDeleteIterator.remove();
            }
            if (ws.getTitle().equals("learning_data")) {
                WorksheetFactory.deleteWorksheet(newSpreadsheet, ws, authToken);
                worksheetToDeleteIterator.remove();
            }
        }

        // setting up the worksheet size is critical.
        Worksheet cardsWorksheet = WorksheetFactory.createWorksheet(newSpreadsheet, "cards", cardList.size() + 1, 4, authToken);

        Cells cardCells = new Cells();

        // Add the header for cards first
        cardCells.addCell(1, 1, "question");
        cardCells.addCell(1, 2, "answer");
        cardCells.addCell(1, 3, "category");
        cardCells.addCell(1, 4, "note");
        for (int i = 0; i < cardList.size(); i++) {
            Card c = cardList.get(i);
            // THe first row is the header.
            cardCells.addCell(i + 2, 1, c.getQuestion());
            cardCells.addCell(i + 2, 2, c.getAnswer());
            cardCells.addCell(i + 2, 3, c.getCategory().getName());
            cardCells.addCell(i + 2, 4, c.getNote());
        }

        // upload card's rows into worksheet
        CellsFactory.uploadCells(newSpreadsheet, cardsWorksheet, cardCells, authToken);

        // Let GC free up memory
        cardCells = null;

        // Now deal with learning data
        Worksheet learningDataWorksheet =
            WorksheetFactory.createWorksheet(newSpreadsheet, "learning_data", cardList.size() + 1, 9, authToken);
        Cells learningDataCells = new Cells();

        // The first row is the header.
        learningDataCells.addCell(1, 1, "acqReps");
        learningDataCells.addCell(1, 2, "acqRepsSinceLapse");
        learningDataCells.addCell(1, 3, "easiness");
        learningDataCells.addCell(1, 4, "grade");
        learningDataCells.addCell(1, 5, "lapses");
        learningDataCells.addCell(1, 6, "lastLearnDate");
        learningDataCells.addCell(1, 7, "nextLearnDate");
        learningDataCells.addCell(1, 8, "retReps");
        learningDataCells.addCell(1, 9, "retRepsSinceLapse");
        for (int i = 0; i < cardList.size(); i++) {
            LearningData ld = cardList.get(i).getLearningData();
            learningDataCells.addCell(i + 2, 1, Integer.toString(ld.getAcqReps()));
            learningDataCells.addCell(i + 2, 2, Integer.toString(ld.getAcqRepsSinceLapse()));
            learningDataCells.addCell(i + 2, 3, Float.toString(ld.getEasiness()));
            learningDataCells.addCell(i + 2, 4, Integer.toString(ld.getGrade()));
            learningDataCells.addCell(i + 2, 5, Integer.toString(ld.getLapses()));
            learningDataCells.addCell(i + 2, 6, ISO8601_FORMATTER.format(ld.getLastLearnDate()));
            learningDataCells.addCell(i + 2, 7, ISO8601_FORMATTER.format(ld.getNextLearnDate()));
            learningDataCells.addCell(i + 2, 8, Integer.toString(ld.getRetReps()));
            learningDataCells.addCell(i + 2, 9, Integer.toString(ld.getRetRepsSinceLapse()));
        }

        // upload learning data rows into worksheet
        CellsFactory.uploadCells(newSpreadsheet, learningDataWorksheet, learningDataCells, authToken);
        learningDataCells = null;


        // Put new spreadsheet into the folder
        FolderFactory.addDocumentToFolder(newSpreadsheetDocument, folder, authToken);

        // Finally delete the unneeded worksheets ...
        for (Worksheet ws : worksheetsToDelete) {
            WorksheetFactory.deleteWorksheet(newSpreadsheet, ws, authToken);
        }

        // ... And spreadsheets with duplicated names.
        for (Document ss : spreadsheetsToDelete) {
            DocumentFactory.deleteDocument(ss, authToken);
        }
        return null;
    }
}