1 files changed, 41 insertions, 0 deletions
diff --git a/src/main/java/files/Database.java b/src/main/java/files/Database.java
new file mode 100644
index 0000000..2d35c2d
--- /dev/null
+++ b/src/main/java/files/Database.java
@@ -0,0 +1,41 @@
+package files;
+
+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"));
+        }
+    }
+}