fabian/telegram_backup
1
0
Fork 0
mirror of https://github.com/fabianonline/telegram_backup.git synced 2024-11-22 16:56:16 +00:00
Code Issues Releases Wiki Activity

Committing current state.

Browse Source
This commit is contained in:
Fabian Schlenz 2017-11-24 11:29:26 +01:00
parent f250f40e5e
commit ae95481e54
3 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/main/java/de/fabianonline/telegram_backup/Database.java
View File

@ -47,6 +47,7 @@ import java.text.SimpleDateFormat;
import de.fabianonline.telegram_backup.mediafilemanager.AbstractMediaFileManager; import de.fabianonline.telegram_backup.mediafilemanager.AbstractMediaFileManager;
import de.fabianonline.telegram_backup.mediafilemanager.FileManagerFactory; import de.fabianonline.telegram_backup.mediafilemanager.FileManagerFactory;
import de.fabianonline.telegram_backup.models.Message;
public class Database { public class Database {
private Connection conn; private Connection conn;
@ -396,12 +397,12 @@ public class Database {
} }
} }
public LinkedList<TLMessage> getMessagesWithMedia() { public LinkedList<Message> getMessagesWithMedia() {
try { try {
LinkedList<TLMessage> list = new LinkedList<TLMessage>(); LinkedList<Message> list = new LinkedList<Message>();
ResultSet rs = stmt.executeQuery("SELECT data FROM messages WHERE has_media=1"); ResultSet rs = stmt.executeQuery("SELECT json FROM messages WHERE has_media=1");
while (rs.next()) { while (rs.next()) {
list.add(bytesToTLMessage(rs.getBytes(1))); list.add(new Message(rs.getString(1)));
} }
rs.close(); rs.close();
return list; return list;

2
src/main/java/de/fabianonline/telegram_backup/TelegramUpdateHandler.java
View File

@ -137,7 +137,7 @@ class TelegramUpdateHandler implements UpdateCallback {
private void processUpdate(TLAbsUpdate update, TelegramClient client) { private void processUpdate(TLAbsUpdate update, TelegramClient client) {
if (update instanceof TLUpdateNewMessage) { if (update instanceof TLUpdateNewMessage) {
TLAbsMessage abs_msg = ((TLUpdateNewMessage)update).getMessage(); TLAbsMessage abs_msg = ((TLUpdateNewMessage)update).getMessage();
Message msg = new Message(abs_msg); Message msg = Message.fromObject(abs_msg);
msg.save(); msg.save();
System.out.print('.'); System.out.print('.');
AbstractMediaFileManager fm = FileManagerFactory.getFileManager(msg); AbstractMediaFileManager fm = FileManagerFactory.getFileManager(msg);

5
src/main/java/de/fabianonline/telegram_backup/models/Message.java
View File

@ -1,6 +1,7 @@
package de.fabianonline.telegram_backup.models; package de.fabianonline.telegram_backup.models;
import de.fabianonline.telegram_backup.Database; import de.fabianonline.telegram_backup.Database;
import de.fabianonline.telegram_backup.Utils;
import com.google.gson.JsonParser; import com.google.gson.JsonParser;
import com.google.gson.JsonObject; import com.google.gson.JsonObject;
@ -15,6 +16,10 @@ public class Message {
this.json = new JsonParser().parse(json).getAsJsonObject(); this.json = new JsonParser().parse(json).getAsJsonObject();
} }
public static Message fromObject(TLAbsMessage msg) {
return new Message(Utils.getGson().fromObject(msg));
}
public static Message get(int id) { public static Message get(int id) {
String json = Database.getInstance().queryString("SELECT json FROM " + tableName + " WHERE id=" + id); String json = Database.getInstance().queryString("SELECT json FROM " + tableName + " WHERE id=" + id);
return new Message(json); return new Message(json);