User's age handled

Author: aqerd

src/main/java/org/oopproject/Database.java

@@ -60,7 +60,6 @@ public Integer getUserAge(long chatId) {
             e.printStackTrace();
         }
         return null;
-
     }
 
     public void updateGenreIndexesJson(long chatId, String genreIndexesJson) {

src/main/java/org/oopproject/TelegramBot.java

@@ -14,6 +14,8 @@
 import org.oopproject.parameters.MovieParameters;
 import org.oopproject.parameters.ParametersBuilder;
 import org.oopproject.deserializers.FilmDeserializer;
+
+import static org.oopproject.utils.AgeRating.getRatingForAge;
 import static org.oopproject.utils.CommandWaiter.*;
 import static org.oopproject.utils.Config.tmdbService;
 import static org.oopproject.utils.Validators.isCommand;
@@ -251,16 +253,15 @@ protected String handleYear(String messageText, long chatId) {
         try {
             int userYear = Integer.parseInt(messageText);
             int currentYear = java.time.Year.now().getValue();
-
+            String userRating = getRatingForAge(getUserAge(chatId));
             if (userYear < 1900 || userYear > currentYear) {
                 responseMessage = "Пожалуйста, введите год в диапазоне от 1900 до " + currentYear;
                 return responseMessage;
             }
 
             MovieParameters params = new ParametersBuilder()
                     .withYear(userYear)
-                    .withCertificationLte("PG-13")
-                    .withCertificationCountry("US")
+                    .withCertificationLte(userRating)
                     .build();
             ListDeserializer moviesByYear = tmdbService.findMovie(params);
 
@@ -441,9 +442,6 @@ public void startBroadcasting() {
         }
     }
 
-
-
-
     private void sendMessage(long chatId, String text) {
         SendMessage message=SendMessage.builder()
                 .chatId(chatId)
@@ -481,5 +479,4 @@ public String handleSubscription(String messageText, long chatId) {
         }
         return responseMessage;
     }
-
 }

src/main/java/org/oopproject/utils/AgeRating.java

@@ -0,0 +1,32 @@
+package org.oopproject.utils;
+
+public enum AgeRating {
+    GENERAL("G", 0, 5),
+    PARENTAL_GUIDANCE("PG", 6, 11),
+    WITH_PARENTS_STRONGLY("PG-13", 12, 15),
+    RESTRICTED("R", 16, 17),
+    ADULTS("NC-17", 18, 120);
+
+    private final String rating;
+    private final int minAge;
+    private final int maxAge;
+
+    AgeRating(String rating, int minAge, int maxAge) {
+        this.rating = rating;
+        this.minAge = minAge;
+        this.maxAge = maxAge;
+    }
+
+    public static String getRatingForAge(int age) {
+        for (AgeRating rating : values()) {
+            if (age >= rating.minAge && age <= rating.maxAge) {
+                return rating.rating;
+            }
+        }
+        return null;
+    }
+
+    public String getRating() {
+        return rating;
+    }
+}