refactor: 로그인 시 UserDataStorage에 id 넣어주는 로직 추가 #82

YAPP-Github · Jul 19, 2022 · 6e599cd · 6e599cd
1 parent 2c88ba8
commit 6e599cd
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/Weekand/Weekand/Domain/UseCase/Sign/SignInUseCase.swift b/Weekand/Weekand/Domain/UseCase/Sign/SignInUseCase.swift
@@ -29,4 +29,16 @@ final class SignInUseCase {
  }
  .asSingle()
  }
+
+ func userID() -> Single<String> {
+ return NetWork.shared.fetch(query: UserSummaryQuery())
+ .map {
+ if let user = $0.user {
+ return user.id
+ } else {
+ return UserSummary.defaultData.name
+ }
+ }
+ .asSingle()
+ }
 }
diff --git a/Weekand/Weekand/Presentation/SignScene/SignIn/ViewModel/SignInViewModel.swift b/Weekand/Weekand/Presentation/SignScene/SignIn/ViewModel/SignInViewModel.swift
@@ -75,6 +75,16 @@ extension SignInViewModel {
 
  self.signInUseCase.login(email: emailText, password: passwordText).subscribe(onSuccess: { tokenData in
  UserDataStorage.shared.setAccessToken(token: tokenData.accessToken)
+ self.userID()
+ }, onFailure: { _ in
+ self.coordinator?.showToastMessage()
+ }, onDisposed: nil)
+ .disposed(by: disposeBag)
+ }
+
+ private func userID() {
+ self.signInUseCase.userID().subscribe(onSuccess: { id in
+ UserDataStorage.shared.setUserID(id: id)
  self.coordinator?.showMainScene()
  }, onFailure: { _ in
  self.coordinator?.showToastMessage()