src/main/java/com/juick/www/api/webhooks/VkWebhook.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

/*
 * 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 <http://www.gnu.org/licenses/>.
 */

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 jakarta.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 user = userService.getUserByVKID(vkId);
                    if (user.isPresent()) {
                        vkService.updatePremiumStatus(user.get().getUid());
                    }
                }
                return "ok";
            } else {
                throw new HttpBadRequestException();
            }
        }
    }
}