#141 feat: 전체적인 검색 로직 작성

boostcampwm-2022 · Dec 6, 2022 · 02aef9c · 02aef9c
1 parent 32746c3
commit 02aef9c
Show file tree

Hide file tree

Showing 3 changed files with 75 additions and 6 deletions.
diff --git a/Segno/Segno/Data/Repository/MusicRepository.swift b/Segno/Segno/Data/Repository/MusicRepository.swift
@@ -16,20 +16,39 @@ protocol MusicRepository {
 }
 
 final class MusicRepositoryImpl: MusicRepository {
-    private let shazamSession = ShazamSession()
-    private let musicSession = MusicSession()
+    private let shazamSession: ShazamSession
+    private let musicSession: MusicSession
+    private let disposeBag = DisposeBag()
 
     var shazamSearchResult = PublishSubject<ShazamSearchResult>()
 
-    func startSearchingMusic() {
+    init(shazamSession: ShazamSession = ShazamSession(),
+         musicSession: MusicSession = MusicSession()) {
+        self.shazamSession = shazamSession
+        self.musicSession = musicSession
 
+        subscribeSearchresult()
+    }
+
+    func startSearchingMusic() {
+        shazamSession.start()
     }
 
     func stopSearchingMusic() {
-
+        shazamSession.stop()
     }
 
     func playMusic() {
 
     }
 }
+
+extension MusicRepositoryImpl {
+    private func subscribeSearchresult() {
+        shazamSession.result
+            .subscribe(onNext: {
+                self.shazamSearchResult.onNext($0)
+            })
+            .disposed(by: disposeBag)
+    }
+}
diff --git a/Segno/Segno/Domain/UseCase/SearchMusicUseCase.swift b/Segno/Segno/Domain/UseCase/SearchMusicUseCase.swift
@@ -17,6 +17,8 @@ protocol SearchMusicUseCase {
 }
 
 final class SearchMusicUseCaseImpl: SearchMusicUseCase {
+    private let disposeBag = DisposeBag()
+
     let musicRepository: MusicRepository
     var musicInfoResult = PublishSubject<MusicInfoResult>()
 
@@ -25,10 +27,26 @@ final class SearchMusicUseCaseImpl: SearchMusicUseCase {
     }
 
     func startSearching() {
-
+        musicRepository.startSearchingMusic()
     }
 
     func stopSearching() {
-
+        musicRepository.stopSearchingMusic()
+    }
+}
+
+extension SearchMusicUseCaseImpl {
+    private func subscribeShazamResult() {
+        musicRepository.shazamSearchResult
+            .subscribe(onNext: {
+                switch $0 {
+                case .success(let shazamSong):
+                    let musicInfo = MusicInfo(shazamSong: shazamSong)
+                    self.musicInfoResult.onNext(.success(musicInfo))
+                case .failure(let error):
+                    self.musicInfoResult.onNext(.failure(error))
+                }
+            })
+            .disposed(by: disposeBag)
     }
 }
diff --git a/Segno/Segno/Presentation/ViewModel/DiaryEditViewModel.swift b/Segno/Segno/Presentation/ViewModel/DiaryEditViewModel.swift
@@ -8,6 +8,7 @@
 import RxSwift
 
 final class DiaryEditViewModel {
+    private let disposeBag = DisposeBag()
     var diaryDetail: DiaryDetail?
     // 에딧 화면에 들어갈 여러 요소들
 
@@ -16,19 +17,50 @@ final class DiaryEditViewModel {
     // 위치 검색 유즈케이스
 
     var isSearching = BehaviorSubject(value: false)
+    var musicInfo = PublishSubject<MusicInfoResult>()
 
     init(diaryDetailUseCase: DiaryDetailUseCase = DiaryDetailUseCaseImpl(),
          searchMusicUseCase: SearchMusicUseCase = SearchMusicUseCaseImpl()) {
         self.diaryDetailUseCase = diaryDetailUseCase
         self.searchMusicUseCase = searchMusicUseCase
+
+        subscribeSearchingStatus()
     }
 
     func addTags() {
 
     }
 
     func toggleSearchMusic() {
+        guard let value = try? isSearching.value() else {
+            return
+        }
 
+        value ? isSearching.onNext(false) : isSearching.onNext(true)
+    }
+
+    func subscribeSearchingStatus() {
+        isSearching
+            .subscribe(onNext: {
+                $0 ? self.startSearchingMusic() : self.stopSearchingMusic()
+            })
+            .disposed(by: disposeBag)
+    }
+
+    func startSearchingMusic() {
+        searchMusicUseCase.startSearching()
+    }
+
+    func stopSearchingMusic() {
+        searchMusicUseCase.stopSearching()
+    }
+
+    func subscribeSearchResult() {
+        searchMusicUseCase.musicInfoResult
+            .subscribe(onNext: {
+                self.musicInfo.onNext($0)
+            })
+            .disposed(by: disposeBag)
     }
 
     func setLocation() {