1 files changed, 94 insertions, 0 deletions
diff --git a/juick-server/src/main/java/com/juick/service/TelegramServiceImpl.java b/juick-server/src/main/java/com/juick/service/TelegramServiceImpl.java
new file mode 100644
index 00000000..4e37bfac
--- /dev/null
+++ b/juick-server/src/main/java/com/juick/service/TelegramServiceImpl.java
@@ -0,0 +1,94 @@
+/*
+ * Copyright (C) 2008-2017, Juick
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero 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 Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package com.juick.service;
+
+import com.juick.User;
+import org.springframework.dao.DuplicateKeyException;
+import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
+import org.springframework.stereotype.Repository;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.Collections;
+import java.util.List;
+import java.util.UUID;
+import java.util.stream.Collectors;
+
+/**
+ * Created by vt on 24/11/2016.
+ */
+@Repository
+public class TelegramServiceImpl extends BaseJdbcService implements TelegramService {
+
+    @Transactional
+    @Override
+    public boolean addChat(final Long id) {
+        try {
+            return getJdbcTemplate().update("INSERT INTO telegram_chats(chat_id) VALUES(?)", id) > 0;
+        } catch (DuplicateKeyException e) {
+            return true;
+        }
+    }
+
+    @Transactional
+    @Override
+    public boolean deleteChat(Long id) {
+        return getJdbcTemplate().update("DELETE FROM telegram_chats WHERE chat_id=?", id) > 0;
+    }
+
+    @Transactional(readOnly = true)
+    @Override
+    public List<Long> getChats() {
+        return getJdbcTemplate().queryForList("SELECT chat_id FROM telegram_chats", Long.class);
+    }
+
+    @Transactional(readOnly = true)
+    @Override
+    public int getUser(final long tgId) {
+        List<Integer> list = getJdbcTemplate().queryForList(
+                "SELECT id FROM users INNER JOIN telegram " +
+                        "ON telegram.user_id = users.id WHERE telegram.tg_id=?", Integer.class, tgId);
+
+        return list.isEmpty() ? 0 : list.get(0);
+    }
+
+    @Transactional
+    @Override
+    public boolean createTelegramUser(final long tgID, final String tgName) {
+        return getJdbcTemplate().update(
+                "INSERT INTO telegram(tg_id, tg_name, loginhash) VALUES(?,?,?)",
+                tgID, tgName, UUID.randomUUID().toString()) > 0;
+    }
+
+    @Transactional
+    @Override
+    public boolean deleteTelegramUser(Integer uid) {
+        return getJdbcTemplate().update("DELETE FROM telegram WHERE user_id=?", uid) > 0;
+    }
+
+    @Transactional(readOnly = true)
+    @Override
+    public List<Long> getTelegramIdentifiers(List<User> users) {
+        List<Integer> uids = users.stream().map(User::getUid).collect(Collectors.toList());
+        if (uids.isEmpty()) {
+            return Collections.emptyList();
+        }
+        return getNamedParameterJdbcTemplate().queryForList("" +
+                "SELECT tg_id FROM telegram WHERE user_id IN(:uids)", new MapSqlParameterSource()
+                .addValue("uids", uids), Long.class);
+    }
+}