wbrawner/thunderbird-android
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add migration for new pending commands

Browse source
This commit is contained in:
Vincent Breitmoser 2016-08-06 12:19:18 +02:00
parent 3eb949e674
commit fb1dfd5786
1 changed files with 181 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

182
k9mail/src/main/java/com/fsck/k9/mailstore/migrations/MigrationTo56.java
View file

@ -1,14 +1,194 @@
package com.fsck.k9.mailstore.migrations; package com.fsck.k9.mailstore.migrations;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import android.content.ContentValues;
import android.database.Cursor;
import android.database.sqlite.SQLiteDatabase; import android.database.sqlite.SQLiteDatabase;
import android.support.annotation.VisibleForTesting;
import com.fsck.k9.controller.MessagingControllerCommands;
import com.fsck.k9.controller.MessagingControllerCommands.PendingCommand;
import com.fsck.k9.controller.PendingCommandSerializer;
import com.fsck.k9.helper.Utility;
import com.fsck.k9.mail.Flag;
public class MigrationTo56 { public class MigrationTo56 {
private static final String PENDING_COMMAND_MOVE_OR_COPY = "com.fsck.k9.MessagingController.moveOrCopy";
private static final String PENDING_COMMAND_MOVE_OR_COPY_BULK = "com.fsck.k9.MessagingController.moveOrCopyBulk";
private static final String PENDING_COMMAND_MOVE_OR_COPY_BULK_NEW = "com.fsck.k9.MessagingController.moveOrCopyBulkNew";
private static final String PENDING_COMMAND_EMPTY_TRASH = "com.fsck.k9.MessagingController.emptyTrash";
private static final String PENDING_COMMAND_SET_FLAG_BULK = "com.fsck.k9.MessagingController.setFlagBulk";
private static final String PENDING_COMMAND_SET_FLAG = "com.fsck.k9.MessagingController.setFlag";
private static final String PENDING_COMMAND_APPEND = "com.fsck.k9.MessagingController.append";
private static final String PENDING_COMMAND_MARK_ALL_AS_READ = "com.fsck.k9.MessagingController.markAllAsRead";
private static final String PENDING_COMMAND_EXPUNGE = "com.fsck.k9.MessagingController.expunge";
public static void migratePendingCommands(SQLiteDatabase db) { public static void migratePendingCommands(SQLiteDatabase db) {
// TODO actually migrate List<PendingCommand> pendingCommmands = new ArrayList<>();
for (OldPendingCommand oldPendingCommand : getPendingCommands(db)) {
PendingCommand newPendingCommand = migratePendingCommand(oldPendingCommand);
pendingCommmands.add(newPendingCommand);
}
db.execSQL("DROP TABLE IF EXISTS pending_commands"); db.execSQL("DROP TABLE IF EXISTS pending_commands");
db.execSQL("CREATE TABLE pending_commands " + db.execSQL("CREATE TABLE pending_commands " +
"(id INTEGER PRIMARY KEY, command TEXT, data TEXT)"); "(id INTEGER PRIMARY KEY, command TEXT, data TEXT)");
PendingCommandSerializer pendingCommandSerializer = PendingCommandSerializer.getInstance();
for (PendingCommand pendingCommand : pendingCommmands) {
final ContentValues cv = new ContentValues();
cv.put("command", pendingCommand.getCommandName());
cv.put("data", pendingCommandSerializer.serialize(pendingCommand));
db.insert("pending_commands", "command", cv);
}
}
@VisibleForTesting
static PendingCommand migratePendingCommand(OldPendingCommand oldPendingCommand) {
switch (oldPendingCommand.command) {
case PENDING_COMMAND_APPEND:
return migrateCommandAppend(oldPendingCommand);
case PENDING_COMMAND_SET_FLAG_BULK:
return migrateCommandSetFlagBulk(oldPendingCommand);
case PENDING_COMMAND_SET_FLAG:
return migrateCommandSetFlag(oldPendingCommand);
case PENDING_COMMAND_MARK_ALL_AS_READ:
return migrateCommandMarkAllAsRead(oldPendingCommand);
case PENDING_COMMAND_MOVE_OR_COPY_BULK:
return migrateCommandMoveOrCopyBulk(oldPendingCommand);
case PENDING_COMMAND_MOVE_OR_COPY_BULK_NEW:
return migrateCommandMoveOrCopyBulkNew(oldPendingCommand);
case PENDING_COMMAND_MOVE_OR_COPY:
return migrateCommandMoveOrCopy(oldPendingCommand);
case PENDING_COMMAND_EMPTY_TRASH:
return migrateCommandEmptyTrash();
case PENDING_COMMAND_EXPUNGE:
return migrateCommandExpunge(oldPendingCommand);
default:
throw new IllegalArgumentException("Tried to migrate unknown pending command!");
}
}
private static PendingCommand migrateCommandExpunge(OldPendingCommand command) {
String folder = command.arguments[0];
return MessagingControllerCommands.createExpunge(folder);
}
private static PendingCommand migrateCommandEmptyTrash() {
return MessagingControllerCommands.createEmptyTrash();
}
private static PendingCommand migrateCommandMoveOrCopy(OldPendingCommand command) {
String srcFolder = command.arguments[0];
String uid = command.arguments[1];
String destFolder = command.arguments[2];
boolean isCopy = Boolean.parseBoolean(command.arguments[3]);
return MessagingControllerCommands.createMoveOrCopyBulk(srcFolder, destFolder, isCopy,
Collections.singletonList(uid));
}
private static PendingCommand migrateCommandMoveOrCopyBulkNew(OldPendingCommand command) {
String srcFolder = command.arguments[0];
String destFolder = command.arguments[1];
boolean isCopy = Boolean.parseBoolean(command.arguments[2]);
boolean hasNewUids = Boolean.parseBoolean(command.arguments[3]);
if (hasNewUids) {
Map<String, String> uidMap = new HashMap<>();
int offset = (command.arguments.length - 4) / 2;
for (int i = 4; i < 4 + offset; i++) {
uidMap.put(command.arguments[i], command.arguments[i + offset]);
}
return MessagingControllerCommands.createMoveOrCopyBulk(srcFolder, destFolder, isCopy, uidMap);
} else {
List<String> uids = new ArrayList<>(command.arguments.length -4);
uids.addAll(Arrays.asList(command.arguments).subList(4, command.arguments.length));
return MessagingControllerCommands.createMoveOrCopyBulk(srcFolder, destFolder, isCopy, uids);
}
}
private static PendingCommand migrateCommandMoveOrCopyBulk(OldPendingCommand command) {
int len = command.arguments.length;
OldPendingCommand newCommand = new OldPendingCommand();
newCommand.command = PENDING_COMMAND_MOVE_OR_COPY_BULK_NEW;
newCommand.arguments = new String[len + 1];
newCommand.arguments[0] = command.arguments[0];
newCommand.arguments[1] = command.arguments[1];
newCommand.arguments[2] = command.arguments[2];
newCommand.arguments[3] = Boolean.toString(false);
System.arraycopy(command.arguments, 3, newCommand.arguments, 4, len - 3);
return migratePendingCommand(newCommand);
}
private static PendingCommand migrateCommandMarkAllAsRead(OldPendingCommand command) {
return MessagingControllerCommands.createMarkAllAsRead(command.arguments[0]);
}
private static PendingCommand migrateCommandSetFlag(OldPendingCommand command) {
String folder = command.arguments[0];
String uid = command.arguments[1];
boolean newState = Boolean.parseBoolean(command.arguments[2]);
Flag flag = Flag.valueOf(command.arguments[3]);
return MessagingControllerCommands.createSetFlag(folder, newState, flag, Collections.singletonList(uid));
}
private static PendingCommand migrateCommandSetFlagBulk(OldPendingCommand command) {
String folder = command.arguments[0];
boolean newState = Boolean.parseBoolean(command.arguments[1]);
Flag flag = Flag.valueOf(command.arguments[2]);
List<String> uids = new ArrayList<>(command.arguments.length -3);
uids.addAll(Arrays.asList(command.arguments).subList(3, command.arguments.length));
return MessagingControllerCommands.createSetFlag(folder, newState, flag, uids);
}
private static PendingCommand migrateCommandAppend(OldPendingCommand command) {
String folder = command.arguments[0];
String uid = command.arguments[1];
return MessagingControllerCommands.createAppend(folder, uid);
}
private static List<OldPendingCommand> getPendingCommands(SQLiteDatabase db) {
Cursor cursor = null;
try {
cursor = db.query("pending_commands",
new String[] { "id", "command", "arguments" }, null, null, null, null, "id ASC");
List<OldPendingCommand> commands = new ArrayList<>();
while (cursor.moveToNext()) {
OldPendingCommand command = new OldPendingCommand();
command.command = cursor.getString(1);
String arguments = cursor.getString(2);
command.arguments = arguments.split(",");
for (int i = 0; i < command.arguments.length; i++) {
command.arguments[i] = Utility.fastUrlDecode(command.arguments[i]);
}
commands.add(command);
}
return commands;
} finally {
Utility.closeQuietly(cursor);
}
}
static class OldPendingCommand {
public String command;
public String[] arguments;
} }
} }