/* * Copyright (C) 2008-2023, 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 . */ package com.juick.www.api.webhooks; import com.fasterxml.jackson.databind.JsonNode; import com.fasterxml.jackson.databind.ObjectMapper; import com.juick.service.UserService; import com.juick.service.VKService; import com.juick.util.HttpBadRequestException; import org.apache.commons.io.IOUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Value; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RestController; import javax.inject.Inject; import java.io.IOException; import java.io.InputStream; import java.nio.charset.StandardCharsets; @RestController public class VkWebhook { private static final Logger logger = LoggerFactory.getLogger("VK"); @Value("${vk_webhook_secret:}") private String secretKey; @Value("${vk_webhook_confirmation:}") private String confirmationCode; @Inject private ObjectMapper jsonMapper; @Inject private UserService userService; @Inject private VKService vkService; @PostMapping(value = "/api/_vk") public String processUpdate(InputStream body) throws IOException { String data = IOUtils.toString(body, StandardCharsets.UTF_8); logger.debug("Data: {}", data); JsonNode json = jsonMapper.readTree(data); var type = json.get("type").asText(); var secret = json.get("secret").asText(""); logger.info("Event received: {}, secret: {}", type, secret); if (type.equals("confirmation")) { return confirmationCode; } else { if (secretKey.equals(secret)) { if (type.startsWith("donut_")) { var vkId = json.get("object").get("user_id").asLong(0); var userId = userService.getUIDbyVKID(vkId); if (userId > 0) { vkService.updatePremiumStatus(userId); } } return "ok"; } else { throw new HttpBadRequestException(); } } } }