WIP: Added Daemon-mode to keep the app running and save new messages in real time.

src/main/java/de/fabianonline/telegram_backup/TelegramUpdateHandler.java

@@ -0,0 +1,138 @@
+/* Telegram_Backup
+ * Copyright (C) 2016 Fabian Schlenz
+ * 
+ * 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 3 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, see <http://www.gnu.org/licenses/>. */
+
+package de.fabianonline.telegram_backup;
+
+import com.github.badoualy.telegram.api.UpdateCallback;
+import com.github.badoualy.telegram.api.TelegramClient;
+import com.github.badoualy.telegram.tl.api.*;
+import com.github.badoualy.telegram.tl.core.TLVector;
+
+import de.fabianonline.telegram_backup.Database;
+import de.fabianonline.telegram_backup.UserManager;
+
+class TelegramUpdateHandler implements UpdateCallback {
+	private UserManager user = null;
+	private Database db = null;
+	
+	public void setUser(UserManager user) { this.user = user; this.db = new Database(user, false);}
+	
+	public void onUpdates(TelegramClient c, TLUpdates u) {
+		System.out.println("onUpdates - " + u.getUpdates().size() + " Updates, " + u.getUsers().size() + " Users, " + u.getChats().size() + " Chats");
+		for(TLAbsUpdate update : u.getUpdates()) {
+			processUpdate(update, c);
+			System.out.println("  " + update.getClass().getName());
+		}
+		db.saveUsers(u.getUsers());
+		db.saveChats(u.getChats());
+	}
+	
+	public void onUpdatesCombined(TelegramClient c, TLUpdatesCombined u) {
+		System.out.println("onUpdatesCombined");
+		for(TLAbsUpdate update : u.getUpdates()) {
+			processUpdate(update, c);
+		}
+		db.saveUsers(u.getUsers());
+		db.saveChats(u.getChats());
+	}
+		
+	public void onUpdateShort(TelegramClient c, TLUpdateShort u) {
+		System.out.println("onUpdateShort");
+		processUpdate(u.getUpdate(), c);
+		System.out.println("  " + u.getUpdate().getClass().getName());
+	}
+	
+	public void onShortChatMessage(TelegramClient c, TLUpdateShortChatMessage m) {
+		System.out.println("onShortChatMessage - " + m.getMessage());
+		TLMessage msg = new TLMessage(
+			m.getUnread(),
+			m.getOut(),
+			m.getMentioned(),
+			m.getMediaUnread(),
+			m.getId(),
+			m.getFromId(),
+			new TLPeerChat(m.getChatId()),
+			m.getFwdFromId(),
+			m.getFwdDate(),
+			m.getViaBotId(),
+			m.getReplyToMsgId(),
+			m.getDate(),
+			m.getMessage(),
+			null,
+			null,
+			m.getEntities(),
+			null);
+		TLVector<TLAbsMessage> vector = new TLVector<TLAbsMessage>(TLAbsMessage.class);
+		vector.add(msg);
+		db.saveMessages(vector);
+	}
+	
+	public void onShortMessage(TelegramClient c, TLUpdateShortMessage m) {
+		System.out.println("onShortMessage - " + m.getOut() + " - " + m.getUserId() + " - " + m.getMessage());
+		int from_id, to_id;
+		if (m.getOut()==true) {
+			from_id = user.getUser().getId();
+			to_id = m.getUserId();
+		} else {
+			to_id = user.getUser().getId();
+			from_id = m.getUserId();
+		}
+		TLMessage msg = new TLMessage(
+			m.getUnread(),
+			m.getOut(),
+			m.getMentioned(),
+			m.getMediaUnread(),
+			m.getId(),
+			from_id,
+			new TLPeerUser(to_id),
+			m.getFwdFromId(),
+			m.getFwdDate(),
+			m.getViaBotId(),
+			m.getReplyToMsgId(),
+			m.getDate(),
+			m.getMessage(),
+			null,
+			null,
+			m.getEntities(),
+			null);
+		TLVector<TLAbsMessage> vector = new TLVector<TLAbsMessage>(TLAbsMessage.class);
+		vector.add(msg);
+		db.saveMessages(vector);
+	}
+	
+	public void onShortSentMessage(TelegramClient c, TLUpdateShortSentMessage m) { System.out.println("onShortSentMessage"); }
+	public void onUpdateTooLong(TelegramClient c) { System.out.println("onUpdateTooLong"); }
+	
+	private void processUpdate(TLAbsUpdate update, TelegramClient client) {
+		if (update instanceof TLUpdateNewMessage) {
+			TLAbsMessage abs_msg = ((TLUpdateNewMessage)update).getMessage();
+			TLVector<TLAbsMessage> vector = new TLVector<TLAbsMessage>(TLAbsMessage.class);
+			vector.add(abs_msg);
+			db.saveMessages(vector);
+			if (abs_msg instanceof TLMessage && ((TLMessage)abs_msg).getMedia()!=null) {
+				try {
+					new DownloadManager(user, client, new CommandLineDownloadProgress()).downloadSingleMessageMedia((TLMessage)abs_msg, ((TLMessage)abs_msg).getMedia());
+				} catch (Exception e) {
+					System.out.println("We got an exception while downloading media, but we're going to ignore it.");
+					System.out.println("Here it is anyway:");
+					e.printStackTrace();
+				}
+			}
+		} else {
+			// ignore everything else...
+		}
+	}
+}