download playlist

Author: audiz

README.MD

@@ -11,7 +11,8 @@ _*Этот бот только для персонального использ
 1. Любой текст отправляемый боту выполняет поиск по яндекс музыке.
 2. Отправив команду "/daily" можно получить свой персональный плейлист.
 3. Отправив команду "/playlists" можно получить список персональных плейлистов.
-4. Отправив команду "/login с параметрам" происходит обновление сессии яндекс музыки
+4. Отправив команду "/login" можно обновить сессию яндекс музыки без рестарта бота.
+5. Отправив команду "/download" можно скачать предыдущий найденный плейлист по частям.
 
 ### Соберите проект командой, используйте Java 8 и выше.
 

src/main/java/bot/common/UtilClasses.kt

@@ -1,6 +1,7 @@
 package bot.common
 
 import bot.telegram.callback.Callback
+import bot.telegram.repository.model.UserStorageData
 import bot.yandex.dto.domain.CaptchaInfo
 import java.net.URLDecoder
 import java.net.URLEncoder
@@ -30,4 +31,14 @@ inline fun <reified T> ResultOf<T>.returnNok(block: (ResultOf<Nothing>) -> Unit)
 
 
 fun String.urlEncode(): String = URLEncoder.encode(this, java.nio.charset.StandardCharsets.UTF_8.toString())
-fun String.urlDecode(): String = URLDecoder.decode(this, java.nio.charset.StandardCharsets.UTF_8.toString())
+fun String.urlDecode(): String = URLDecoder.decode(this, java.nio.charset.StandardCharsets.UTF_8.toString())
+
+inline fun UserStorageData?.isDownloadProcess(block: () -> Unit) {
+    if (this == null) {
+        return
+    }
+    val inProgress: Boolean = !((this.loaded.find{ it == false }) ?: true)
+    if (inProgress) {
+        return block()
+    }
+}

src/main/java/bot/common/Utils.kt

@@ -1,17 +1,8 @@
 package bot.common
 
-import org.telegram.telegrambots.meta.api.objects.Message
 import org.telegram.telegrambots.meta.api.objects.User
 
 object Utils {
-    /**
-     * Формирование имени пользователя
-     * @param msg сообщение
-     */
-    fun getUserName(msg: Message): String {
-        return getUserName(msg.from)
-    }
-
     /**
      * Формирование имени пользователя. Если заполнен никнейм, используем его. Если нет - используем фамилию и имя
      * @param user пользователь

src/main/java/bot/telegram/callback/PlaylistActionsImpl.kt

@@ -130,6 +130,13 @@ class PlaylistActionsImpl(val yandexMusic: YandexMusic, val messageSource: Messa
     }
 
     override fun playlist(userId: Long, chatId: String, callback: PlaylistCallback): ResultOf<SendMessage> {
+        userStorage.getUserStorageData(userId).isDownloadProcess {
+            return ResultOf.Failure(
+                ErrorBuilder.newBuilder(ErrorKind.APP_INTERNAL)
+                    .withCode(403)
+                    .withDescription(messageSource.getMessage("errors.latest.playlists.position.using", null, Locale.getDefault())))
+        }
+
         val playlistResult = yandexMusic.getPlaylist(callback.kind, callback.owner)
         playlistResult.returnNok { return it }
         val playlist = (playlistResult as ResultOf.Success).value
@@ -151,6 +158,13 @@ class PlaylistActionsImpl(val yandexMusic: YandexMusic, val messageSource: Messa
      * Daily playlist for current yandex user
      * */
     override fun dailyPlaylist(userId: Long, chatId: String): ResultOf<SendMessage> {
+        userStorage.getUserStorageData(userId).isDownloadProcess {
+            return ResultOf.Failure(
+                ErrorBuilder.newBuilder(ErrorKind.APP_INTERNAL)
+                    .withCode(403)
+                    .withDescription(messageSource.getMessage("errors.latest.playlists.position.using", null, Locale.getDefault())))
+        }
+
         val dailyIdResult = yandexMusic.getPlaylists()
         dailyIdResult.returnNok { return it }
         val dailyPlaylists = (dailyIdResult as ResultOf.Success).value
@@ -180,6 +194,13 @@ class PlaylistActionsImpl(val yandexMusic: YandexMusic, val messageSource: Messa
      * Show search tracks to user
      * */
     override fun searchByString(userId: Long, chatId: String, text: String): ResultOf<SendMessage> {
+        userStorage.getUserStorageData(userId).isDownloadProcess {
+            return ResultOf.Failure(
+                ErrorBuilder.newBuilder(ErrorKind.APP_INTERNAL)
+                    .withCode(403)
+                    .withDescription(messageSource.getMessage("errors.latest.playlists.position.using", null, Locale.getDefault())))
+        }
+
         // TODO cut string to limit
         val searchText = URLEncoder.encode(text.trimIndent(), "utf-8")
         val searchResult = yandexMusic.search(searchText)
@@ -209,6 +230,13 @@ class PlaylistActionsImpl(val yandexMusic: YandexMusic, val messageSource: Messa
      * Limit for json, when more than 150 we should request by ids
      * */
     override fun artistWithPagesMsg(userId: Long, chatId: String, callback: ArtistTrackWithPagesCallback): ResultOf<SendMessage> {
+        userStorage.getUserStorageData(userId).isDownloadProcess {
+            return ResultOf.Failure(
+                ErrorBuilder.newBuilder(ErrorKind.APP_INTERNAL)
+                    .withCode(403)
+                    .withDescription(messageSource.getMessage("errors.latest.playlists.position.using", null, Locale.getDefault())))
+        }
+
         val page = callback.page
         val searchResult = yandexMusic.searchTrack(callback.artistId)
         searchResult.returnNok { return it }
@@ -237,6 +265,13 @@ class PlaylistActionsImpl(val yandexMusic: YandexMusic, val messageSource: Messa
      * Pagination for searched tracks
      * */
     override fun searchWithPagesMsg(userId: Long, chatId: String, callback: SearchTrackWithPagesCallback): ResultOf<SendMessage> {
+        userStorage.getUserStorageData(userId).isDownloadProcess {
+            return ResultOf.Failure(
+                ErrorBuilder.newBuilder(ErrorKind.APP_INTERNAL)
+                    .withCode(403)
+                    .withDescription(messageSource.getMessage("errors.latest.playlists.position.using", null, Locale.getDefault())))
+        }
+
         //logger.info("searchWithPagesMsg callback = {}", callback)
         val page = callback.page
         val realPageForRequest = ((page - 1) * TRACKS_PER_PAGE) / 100