Piped-Backend/src/main/java/me/kavin/piped/utils/PubSubHelper.java

package me.kavin.piped.utils;

import me.kavin.piped.consts.Constants;
import me.kavin.piped.utils.obj.db.PubSub;
import okhttp3.FormBody;
import okhttp3.Request;
import org.hibernate.StatelessSession;

import java.io.IOException;
import java.util.Objects;
import java.util.concurrent.TimeUnit;

public class PubSubHelper {
    public static void subscribePubSub(String channelId) throws IOException {

        PubSub pubsub = DatabaseHelper.getPubSubFromId(channelId);

        if (pubsub == null || System.currentTimeMillis() - pubsub.getSubbedAt() > TimeUnit.DAYS.toMillis(4)) {

            String callback = Constants.PUBSUB_URL + "/webhooks/pubsub";
            String topic = "https://www.youtube.com/xml/feeds/videos.xml?channel_id=" + channelId;

            var builder = new Request.Builder()
                    .url(Constants.PUBSUB_HUB_URL);

            var formBuilder = new FormBody.Builder();

            formBuilder.add("hub.callback", callback);
            formBuilder.add("hub.topic", topic);
            formBuilder.add("hub.verify", "async");
            formBuilder.add("hub.mode", "subscribe");
            formBuilder.add("hub.lease_seconds", "432000");

            try (var resp = Constants.h2client
                    .newCall(builder.post(formBuilder.build())
                            .build()).execute()) {

                if (resp.code() != 202)
                    System.out.println("Failed to subscribe: " + resp.code() + "\n" + Objects.requireNonNull(resp.body()).string());

            }
        }

    }

    public static void updatePubSub(String channelId) {
        var pubsub = DatabaseHelper.getPubSubFromId(channelId);
        try (StatelessSession s = DatabaseSessionFactory.createStatelessSession()) {
            s.beginTransaction();
            if (pubsub == null) {
                pubsub = new PubSub(channelId, System.currentTimeMillis());
                s.insert(pubsub);
            } else {
                pubsub.setSubbedAt(System.currentTimeMillis());
                s.update(pubsub);
            }
            s.getTransaction().commit();
        }
    }
}