feat: Search for better works now on the my episodes screen

pct-org · Mar 6, 2019 · 6b4e08c · 6b4e08c
1 parent 1773ba8
commit 6b4e08c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/app/screens/MyEpisodes/MyEpisodesContainer.js b/app/screens/MyEpisodes/MyEpisodesContainer.js
@@ -6,7 +6,7 @@ import * as HomeSelectors from '../Home/HomeSelectors'
 import MyEpisodes from './MyEpisodesScreen'
 
 export const mapStateToProps = state => ({
-  modes: HomeSelectors.getModes(state),
+  myEpisodes: HomeSelectors.getModes(state).myEpisodes,
 })
 
 export default connect(mapStateToProps, HomeActions)(MyEpisodes)
diff --git a/app/screens/MyEpisodes/MyEpisodesScreen.js b/app/screens/MyEpisodes/MyEpisodesScreen.js
@@ -62,7 +62,7 @@ const styles = StyleSheet.create({
 
 export class MyEpisodes extends React.PureComponent {
 
-  static getDerivedStateFromProps(props) {
+  static getDerivedStateFromProps(props, state) {
     // If we are not focused cancel the quality selector
     if (!props.isFocused) {
       return {
@@ -71,6 +71,18 @@ export class MyEpisodes extends React.PureComponent {
       }
     }
 
+    const { episodeToPlay, selectFromTorrents } = state
+
+    if (episodeToPlay && selectFromTorrents) {
+      const { myEpisodes: { items } } = props
+
+      const newEpisode = items.find(item => item.id === episodeToPlay.id)
+
+      return {
+        selectFromTorrents: newEpisode ? newEpisode.torrents : selectFromTorrents,
+      }
+    }
+
     return {}
   }
 
@@ -186,7 +198,7 @@ export class MyEpisodes extends React.PureComponent {
   )
 
   render() {
-    const { modes: { myEpisodes: { items, refreshing, fetching } } } = this.props
+    const { myEpisodes: { items, refreshing, fetching } } = this.props
     const { selectFromTorrents, episodeToPlay } = this.state
 
     return (