@@ -18,16 +18,34 @@ package soup.movie.data.database.impl.mapper
1818import soup.movie.data.database.impl.entity.FavoriteMovieEntity
1919import soup.movie.data.database.impl.entity.MovieEntity
2020import soup.movie.data.database.impl.entity.OpenDateAlarmEntity
21+ import soup.movie.datetime.calculateDDay
22+ import soup.movie.datetime.toLocalDate
2123import soup.movie.model.MovieModel
2224import soup.movie.model.OpenDateAlarmModel
2325import soup.movie.model.TheaterRatingsModel
26+ import java.time.LocalDate
2427
25- fun MovieEntity.toMovie () = MovieModel (
26- id, score, title, posterUrl, openDate, isNow, age, nationFilter, genres, boxOffice,
27- TheaterRatingsModel (cgv, lotte, megabox),
28- )
28+ fun MovieEntity.toMovie (today : LocalDate ): MovieModel {
29+ val openLocalDate = openDate.toLocalDate()
30+ return MovieModel (
31+ id = id,
32+ score = score,
33+ title = title,
34+ posterUrl = posterUrl,
35+ openDate = openDate,
36+ isNow = isNow,
37+ age = age,
38+ nationFilter = nationFilter,
39+ genres = genres,
40+ boxOffice = boxOffice,
41+ theater = TheaterRatingsModel (cgv, lotte, megabox),
42+ openLocalDate = openLocalDate,
43+ dDay = if (! isNow) calculateDDay(openDate = openLocalDate, today = today) else null ,
44+ )
45+ }
2946
30- fun FavoriteMovieEntity.toMovie (): MovieModel {
47+ fun FavoriteMovieEntity.toMovie (today : LocalDate ): MovieModel {
48+ val openLocalDate = openDate.toLocalDate()
3149 return MovieModel (
3250 id = id,
3351 score = score,
@@ -40,7 +58,15 @@ fun FavoriteMovieEntity.toMovie(): MovieModel {
4058 genres = genres,
4159 boxOffice = boxOffice,
4260 theater = TheaterRatingsModel (cgv, lotte, megabox),
61+ openLocalDate = openLocalDate,
62+ dDay = if (! isNow) calculateDDay(openDate = openLocalDate, today = today) else null ,
4363 )
4464}
4565
46- fun OpenDateAlarmEntity.toOpenDateAlarm () = OpenDateAlarmModel (movieId, title, openDate)
66+ fun OpenDateAlarmEntity.toOpenDateAlarm (): OpenDateAlarmModel {
67+ return OpenDateAlarmModel (
68+ movieId = movieId,
69+ title = title,
70+ openDate = openDate,
71+ )
72+ }
0 commit comments