src/main/java/com/mavlonerkinboev/animarfo/database/Database.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

package com.mavlonerkinboev.animarfo.database;

import com.mavlonerkinboev.animarfo.telegram.bot.Bot;
import com.mavlonerkinboev.animarfo.telegram.user.TelegramUser;
import com.mongodb.MongoClient;
import com.mongodb.MongoClientSettings;
import com.mongodb.MongoClientURI;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.Filters;

import org.bson.Document;
import org.bson.codecs.configuration.CodecRegistries;
import org.bson.codecs.pojo.PojoCodecProvider;

import static org.fusesource.jansi.Ansi.*;
import static org.fusesource.jansi.Ansi.Color.*;

public class Database {
    private static final String MONGO_URI = System.getenv().get("DATABASE_URL");
    private static final MongoDatabase database = new MongoClient(new MongoClientURI(MONGO_URI))
            .getDatabase("animarfo")
            .withCodecRegistry(
                    CodecRegistries.fromRegistries(
                    MongoClientSettings.getDefaultCodecRegistry(),
                    CodecRegistries.fromProviders(PojoCodecProvider.builder().automatic(true).build()))
            );

    public static void saveUser(TelegramUser telegramUser) {
        System.out.println(ansi().fg(YELLOW).a("Trying to save user to database..."));
        MongoCollection<TelegramUser> collection = database.getCollection("users", TelegramUser.class);
        System.out.println(ansi().fg(GREEN).a("Connected to database and got collection!"));
        TelegramUser searchedTelegramUser = collection.find(Filters.eq("_id", telegramUser.getId())).first();

        if (searchedTelegramUser == null) {
            collection.insertOne(telegramUser);
            System.out.println(ansi().fg(GREEN).a("User is added to the database!"));
        } else {
            collection.updateOne(Filters.eq("_id", telegramUser.getId()), new Document("$set", telegramUser));
            System.out.println(ansi().fg(GREEN).a("User is updated"));
        }
    }
}