package com.juick.components; import com.fasterxml.jackson.core.JsonProcessingException; import com.fasterxml.jackson.databind.ObjectMapper; import com.google.auth.oauth2.GoogleCredentials; import com.google.firebase.FirebaseApp; import com.google.firebase.FirebaseOptions; import com.google.firebase.messaging.FirebaseMessaging; import com.google.firebase.messaging.FirebaseMessagingException; import com.google.firebase.messaging.Message; import com.juick.ExternalToken; import com.juick.User; import com.juick.service.component.*; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Value; import org.springframework.core.io.ClassPathResource; import org.springframework.core.io.Resource; import javax.annotation.PostConstruct; import javax.inject.Inject; import java.io.IOException; import java.io.InputStream; import java.util.List; import java.util.stream.Collectors; public class FirebaseManager implements NotificationListener { private static Logger logger = LoggerFactory.getLogger(FirebaseManager.class); @Inject ObjectMapper jsonMapper; @Value("${gcm_key:}") private String gcmKey; @Value("${fcm_database_url:}") private String fcmDatabaseUrl; @Inject private NotificationsManager notificationsManager; @PostConstruct public void initialize() throws IOException { Resource serviceAccount = new ClassPathResource("serviceAccount.json"); FirebaseOptions options = new FirebaseOptions.Builder() .setCredentials(GoogleCredentials.fromStream(serviceAccount.getInputStream())) .setDatabaseUrl(fcmDatabaseUrl) .build(); FirebaseApp.initializeApp(options); } @Override public void processMessageEvent(MessageEvent messageEvent) { com.juick.Message jmsg = messageEvent.getMessage(); List users = messageEvent.getUsers(); // GCM List regids = users.stream().flatMap(u -> u.getTokens().stream()).filter(d -> d.getType().equals("gcm")) .map(ExternalToken::getToken).collect(Collectors.toList()); if (!regids.isEmpty()) { String json = null; try { json = jsonMapper.writeValueAsString(jmsg); } catch (JsonProcessingException e) { logger.warn("JSON exception", e); } logger.info(json); Message.Builder messageBuilder = Message.builder() .putData("message", json); regids.forEach(token -> { messageBuilder.setToken(token); try { String response = FirebaseMessaging.getInstance().send(messageBuilder.build()); // Response is a message ID string. logger.info("Successfully sent message: " + response); } catch (FirebaseMessagingException e) { logger.warn("Firebase exception", e); /* List results = result.getResults(); for (int i = 0; i < results.size(); i++) { Result currentResult = results.get(i); logger.info("RES {}: {}", i, currentResult); List errorCodes = Arrays.asList(Constants.ERROR_MISMATCH_SENDER_ID, Constants.ERROR_NOT_REGISTERED); if (errorCodes.contains(currentResult.getErrorCodeName())) { // assuming results are in order of regids // http://stackoverflow.com/a/11594531/1097384 String currentId = regids.get(i); logger.info("{} is scheduled to remove", currentId); notificationsManager.addInvalidGCMToken(currentId); } }*/ } }); } else { logger.info("GMS: no recipients"); } } @Override public void processSubscribeEvent(SubscribeEvent subscribeEvent) { } @Override public void processLikeEvent(LikeEvent likeEvent) { } @Override public void processPingEvent(PingEvent pingEvent) { } @Override public void processMessageReadEvent(MessageReadEvent messageReadEvent) { } @Override public void processTopEvent(TopEvent topEvent) { } }