package com.mavlonerkinboev.animarfo.database;

import com.mavlonerkinboev.animarfo.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 = Bot.PROPERTIES.getProperty("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").reset());
        }
    }
}