add subtitle hard link generate in media dir (#194)

* fix special char dir name generate issue and add subtitle hard link generate in media dir

* update changelog for pr #194

Author: chivehao

CHANGELOG.MD

@@ -5,6 +5,7 @@
 ## Features
 - 添加剧集URL更新事件，当剧集URL字段更新时，给用户发送邮件通知用户番剧更新了
 - 添加应用URL前缀环境变量
+- 媒体目录会生成与剧集同名的已经上传的ass字幕文件，jellyfin可自动识别
 
 ## Pages
 - 用户信息更新页面，支持用户邮箱更新
@@ -16,6 +17,7 @@
 - 邮件URL的斜杠重复，导致无法访问资源 #188
 - 邮件剧集文件名称不是视频类型的也进行更新了 #188
 - 剧集和文件批量匹配失效 #191
+- 特殊的番剧名称生成的媒体目录不正确问题，如 `Cyberpunk: Edgerunners` 生成的媒体目录成了 `_J1OWX~4` 原因是目录名不能有英文冒号
 
 ## Improvements
 - 优化剧集URL更新时通知的代码逻辑，目前在拉取qbittorrent已完成下载的文件任务里，创建服务端对应的文件后，当匹配到对应的剧集URL时，会给用户发送通知邮件

src/main/java/run/ikaros/server/core/repository/FileRepository.java

@@ -38,4 +38,6 @@ public interface FileRepository extends BaseRepository<FileEntity, Long> {
     FileEntity findFileEntityByNameAndTypeAndPlace(String name, FileType type, FilePlace place);
 
     Optional<FileEntity> findByUrlAndStatus(String url, Boolean status);
+
+    List<FileEntity> findByNameLike(String name);
 }

src/main/java/run/ikaros/server/core/service/FileService.java

@@ -119,4 +119,7 @@ default Long getEpisodeSeqFromName(@Nonnull String name) {
 
     @Nonnull
     Optional<FileEntity> findByUrl(@Nonnull String url);
+
+    @Nonnull
+    List<FileEntity> findListByNameLike(@Nonnull String keyword);
 }

src/main/java/run/ikaros/server/service/FileServiceImpl.java

@@ -445,6 +445,13 @@ public Optional<FileEntity> findByUrl(@Nonnull String url) {
         return fileRepository.findByUrlAndStatus(url, true);
     }
 
+    @Nonnull
+    @Override
+    public List<FileEntity> findListByNameLike(@Nonnull String keyword) {
+        AssertUtils.notBlank(keyword, "keyword");
+        return new ArrayList<>(fileRepository.findByNameLike(StringUtils.addLikeChar(keyword)));
+    }
+
     private String meringTempChunkFile(String unique, String postfix) throws IOException {
         LOGGER.debug("All chunks upload has finish, will start merging files");
 

src/main/java/run/ikaros/server/service/MediaServiceImpl.java

@@ -29,6 +29,7 @@
 import java.util.Date;
 import java.util.List;
 import java.util.Optional;
+import java.util.stream.Collectors;
 
 @Service
 public class MediaServiceImpl implements MediaService {
@@ -75,6 +76,19 @@ private void generateMediaDirBySingleAnimeEntity(@Nonnull AnimeEntity animeEntit
             animeDirName += " (" + localDateTime.getYear() + "-" + localDateTime.getMonthValue()
                 + "-" + localDateTime.getDayOfMonth() + ")";
         }
+
+        // remove char \ / : * ? " < > |
+        animeDirName = animeDirName
+            .replace("\\", "")
+            .replace("/", "")
+            .replace(":", "")
+            .replace("*", "")
+            .replace("?", "")
+            .replace("\"", "")
+            .replace("<", "")
+            .replace(">", "")
+            .replace("|", "");
+
         final String animeDirPath = SystemVarUtils.getCurrentAppMediaDirPath()
             + File.separator + animeDirName;
         File animeDir = new File(animeDirPath);
@@ -181,7 +195,7 @@ private void generateMediaDirBySingleSeason(@Nonnull String animeDirPath,
                     uploadEpFilePath, seasonEntity.getTitle(), episodeEntity.getTitle());
                 continue;
             }
-            Optional<FileEntity> fileEntityOptional = fileService.findByUrl(url);
+            Optional<FileEntity> fileEntityOptional = fileService.findByUrl(episodeEntity.getUrl());
             String fileName;
             if (fileEntityOptional.isPresent()) {
                 fileName = fileEntityOptional.get().getName();
@@ -220,6 +234,46 @@ private void generateMediaDirBySingleSeason(@Nonnull String animeDirPath,
                     episodeEntity.getOverview(), episodePlot, String.valueOf(seasonNum),
                     String.valueOf(seq), String.valueOf(bgmtvId));
             }
+
+            // link episode subtitle file
+            List<FileEntity> existsFileEntityList =
+                fileService.findListByNameLike(fileName.replaceAll(RegexConst.FILE_POSTFIX, ""))
+                    .stream()
+                    .filter(fileEntity -> !fileName.equalsIgnoreCase(fileEntity.getName()))
+                    .toList();
+            if (!existsFileEntityList.isEmpty()) {
+                for (FileEntity fileEntity : existsFileEntityList) {
+                    String name = fileEntity.getName();
+                    String postfix = FileUtils.parseFilePostfix(name);
+                    if ("ass".equalsIgnoreCase(postfix)
+                        && name.contains(
+                        fileName.replaceAll(RegexConst.FILE_POSTFIX, ""))) {
+                        File mediaSubtitleFile =
+                            new File(seasonDirPath + File.separator + name);
+                        File uploadSubtitleFile = new File(
+                            SystemVarUtils.getCurrentAppDirPath()
+                                + File.separator + fileEntity.getUrl());
+                        if (mediaSubtitleFile.exists()) {
+                            continue;
+                        }
+                        try {
+                            Files.createLink(mediaSubtitleFile.toPath(),
+                                uploadSubtitleFile.toPath());
+                            LOGGER.info(
+                                "create media episode subtitle hard link success, "
+                                    + "link={}, existing={}",
+                                mediaSubtitleFile.getAbsolutePath(),
+                                uploadSubtitleFile.getAbsolutePath());
+                        } catch (IOException e) {
+                            LOGGER.error(
+                                "create media episode subtitle hard link fail, link={}, "
+                                    + "existing={}",
+                                mediaSubtitleFile.getAbsolutePath(),
+                                uploadSubtitleFile.getAbsolutePath());
+                        }
+                    }
+                }
+            }
         }
     }