New: Reset Quality Definitions to default

Closes #1719

(cherry picked from commit d5fff15f32fdb49768dcadd94c760678e650c884)

frontend/src/App/AppRoutes.js

@@ -22,7 +22,7 @@ import MediaManagementConnector from 'Settings/MediaManagement/MediaManagementCo
 import MetadataSettings from 'Settings/Metadata/MetadataSettings';
 import NotificationSettings from 'Settings/Notifications/NotificationSettings';
 import Profiles from 'Settings/Profiles/Profiles';
-import Quality from 'Settings/Quality/Quality';
+import QualityConnector from 'Settings/Quality/QualityConnector';
 import Settings from 'Settings/Settings';
 import TagSettings from 'Settings/Tags/TagSettings';
 import UISettingsConnector from 'Settings/UI/UISettingsConnector';
@@ -172,7 +172,7 @@ function AppRoutes(props) {
 
       <Route
         path="/settings/quality"
-        component={Quality}
+        component={QualityConnector}
       />
 
       <Route

frontend/src/Commands/commandNames.js

@@ -21,6 +21,7 @@ export const RESCAN_FOLDERS = 'RescanFolders';
 export const RETAG_FILES = 'RetagFiles';
 export const RETAG_AUTHOR = 'RetagAuthor';
 export const RESET_API_KEY = 'ResetApiKey';
+export const RESET_QUALITY_DEFINITIONS = 'ResetQualityDefinitions';
 export const RSS_SYNC = 'RssSync';
 export const SEASON_SEARCH = 'BookSearch';
 export const AUTHOR_SEARCH = 'AuthorSearch';

frontend/src/Components/SignalRConnector.js

@@ -9,7 +9,7 @@ import { removeItem, update, updateItem } from 'Store/Actions/baseActions';
 import { deleteAuthorBooks } from 'Store/Actions/bookActions';
 import { fetchCommands, finishCommand, updateCommand } from 'Store/Actions/commandActions';
 import { fetchQueue, fetchQueueDetails } from 'Store/Actions/queueActions';
-import { fetchRootFolders } from 'Store/Actions/settingsActions';
+import { fetchQualityDefinitions, fetchRootFolders } from 'Store/Actions/settingsActions';
 import { fetchHealth } from 'Store/Actions/systemActions';
 import { fetchTagDetails, fetchTags } from 'Store/Actions/tagActions';
 import { repopulatePage } from 'Utilities/pagePopulator';
@@ -49,6 +49,7 @@ const mapDispatchToProps = {
   dispatchDeleteAuthorBooks: deleteAuthorBooks,
   dispatchFetchAuthor: fetchAuthor,
   dispatchFetchHealth: fetchHealth,
+  dispatchFetchQualityDefinitions: fetchQualityDefinitions,
   dispatchFetchQueue: fetchQueue,
   dispatchFetchQueueDetails: fetchQueueDetails,
   dispatchFetchRootFolders: fetchRootFolders,
@@ -223,6 +224,10 @@ class SignalRConnector extends Component {
     }
   }
 
+  handleQualitydefinition = () => {
+    this.props.dispatchFetchQualityDefinitions();
+  };
+
   handleQueue = () => {
     if (this.props.isQueuePopulated) {
       this.props.dispatchFetchQueue();
@@ -370,6 +375,7 @@ SignalRConnector.propTypes = {
   dispatchDeleteAuthorBooks: PropTypes.func.isRequired,
   dispatchFetchAuthor: PropTypes.func.isRequired,
   dispatchFetchHealth: PropTypes.func.isRequired,
+  dispatchFetchQualityDefinitions: PropTypes.func.isRequired,
   dispatchFetchQueue: PropTypes.func.isRequired,
   dispatchFetchQueueDetails: PropTypes.func.isRequired,
   dispatchFetchRootFolders: PropTypes.func.isRequired,

frontend/src/Settings/Quality/Quality.js

@@ -1,9 +1,14 @@
-import React, { Component } from 'react';
+import PropTypes from 'prop-types';
+import React, { Component, Fragment } from 'react';
 import PageContent from 'Components/Page/PageContent';
 import PageContentBody from 'Components/Page/PageContentBody';
+import PageToolbarButton from 'Components/Page/Toolbar/PageToolbarButton';
+import PageToolbarSeparator from 'Components/Page/Toolbar/PageToolbarSeparator';
+import { icons } from 'Helpers/Props';
 import SettingsToolbarConnector from 'Settings/SettingsToolbarConnector';
 import translate from 'Utilities/String/translate';
 import QualityDefinitionsConnector from './Definition/QualityDefinitionsConnector';
+import ResetQualityDefinitionsModal from './Reset/ResetQualityDefinitionsModal';
 
 class Quality extends Component {
 
@@ -17,7 +22,8 @@ class Quality extends Component {
 
     this.state = {
       isSaving: false,
-      hasPendingChanges: false
+      hasPendingChanges: false,
+      isConfirmQualityDefinitionResetModalOpen: false
     };
   }
 
@@ -32,6 +38,14 @@ class Quality extends Component {
     this.setState(payload);
   }
 
+  onResetQualityDefinitionsPress = () => {
+    this.setState({ isConfirmQualityDefinitionResetModalOpen: true });
+  };
+
+  onCloseResetQualityDefinitionsModal = () => {
+    this.setState({ isConfirmQualityDefinitionResetModalOpen: false });
+  };
+
   onSavePress = () => {
     if (this._saveCallback) {
       this._saveCallback();
@@ -44,6 +58,7 @@ class Quality extends Component {
   render() {
     const {
       isSaving,
+      isResettingQualityDefinitions,
       hasPendingChanges
     } = this.state;
 
@@ -52,6 +67,18 @@ class Quality extends Component {
         <SettingsToolbarConnector
           isSaving={isSaving}
           hasPendingChanges={hasPendingChanges}
+          additionalButtons={
+            <Fragment>
+              <PageToolbarSeparator />
+
+              <PageToolbarButton
+                label="Reset Definitions"
+                iconName={icons.REFRESH}
+                isSpinning={isResettingQualityDefinitions}
+                onPress={this.onResetQualityDefinitionsPress}
+              />
+            </Fragment>
+          }
           onSavePress={this.onSavePress}
         />
 
@@ -61,9 +88,18 @@ class Quality extends Component {
             onChildStateChange={this.onChildStateChange}
           />
         </PageContentBody>
+
+        <ResetQualityDefinitionsModal
+          isOpen={this.state.isConfirmQualityDefinitionResetModalOpen}
+          onModalClose={this.onCloseResetQualityDefinitionsModal}
+        />
       </PageContent>
     );
   }
 }
 
+Quality.propTypes = {
+  isResettingQualityDefinitions: PropTypes.bool.isRequired
+};
+
 export default Quality;

frontend/src/Settings/Quality/QualityConnector.js

@@ -0,0 +1,38 @@
+import PropTypes from 'prop-types';
+import React, { Component } from 'react';
+import { connect } from 'react-redux';
+import { createSelector } from 'reselect';
+import * as commandNames from 'Commands/commandNames';
+import createCommandExecutingSelector from 'Store/Selectors/createCommandExecutingSelector';
+import Quality from './Quality';
+
+function createMapStateToProps() {
+  return createSelector(
+    createCommandExecutingSelector(commandNames.RESET_QUALITY_DEFINITIONS),
+    (isResettingQualityDefinitions) => {
+      return {
+        isResettingQualityDefinitions
+      };
+    }
+  );
+}
+
+class QualityConnector extends Component {
+
+  //
+  // Render
+
+  render() {
+    return (
+      <Quality
+        {...this.props}
+      />
+    );
+  }
+}
+
+QualityConnector.propTypes = {
+  isResettingQualityDefinitions: PropTypes.bool.isRequired
+};
+
+export default connect(createMapStateToProps)(QualityConnector);

frontend/src/Settings/Quality/Reset/ResetQualityDefinitionsModal.js

@@ -0,0 +1,33 @@
+import PropTypes from 'prop-types';
+import React from 'react';
+import Modal from 'Components/Modal/Modal';
+import { sizes } from 'Helpers/Props';
+import ResetQualityDefinitionsModalContentConnector from './ResetQualityDefinitionsModalContentConnector';
+
+function ResetQualityDefinitionsModal(props) {
+  const {
+    isOpen,
+    onModalClose,
+    ...otherProps
+  } = props;
+
+  return (
+    <Modal
+      isOpen={isOpen}
+      size={sizes.MEDIUM}
+      onModalClose={onModalClose}
+    >
+      <ResetQualityDefinitionsModalContentConnector
+        {...otherProps}
+        onModalClose={onModalClose}
+      />
+    </Modal>
+  );
+}
+
+ResetQualityDefinitionsModal.propTypes = {
+  isOpen: PropTypes.bool.isRequired,
+  onModalClose: PropTypes.func.isRequired
+};
+
+export default ResetQualityDefinitionsModal;

frontend/src/Settings/Quality/Reset/ResetQualityDefinitionsModalContent.css

@@ -0,0 +1,3 @@
+.messageContainer {
+  margin-bottom: 20px;
+}

frontend/src/Settings/Quality/Reset/ResetQualityDefinitionsModalContent.js

@@ -0,0 +1,103 @@
+import PropTypes from 'prop-types';
+import React, { Component } from 'react';
+import FormGroup from 'Components/Form/FormGroup';
+import FormInputGroup from 'Components/Form/FormInputGroup';
+import FormLabel from 'Components/Form/FormLabel';
+import Button from 'Components/Link/Button';
+import ModalBody from 'Components/Modal/ModalBody';
+import ModalContent from 'Components/Modal/ModalContent';
+import ModalFooter from 'Components/Modal/ModalFooter';
+import ModalHeader from 'Components/Modal/ModalHeader';
+import { inputTypes, kinds } from 'Helpers/Props';
+import styles from './ResetQualityDefinitionsModalContent.css';
+
+class ResetQualityDefinitionsModalContent extends Component {
+
+  //
+  // Lifecycle
+
+  constructor(props, context) {
+    super(props, context);
+
+    this.state = {
+      resetDefinitionTitles: false
+    };
+  }
+
+  //
+  // Listeners
+
+  onResetDefinitionTitlesChange = ({ value }) => {
+    this.setState({ resetDefinitionTitles: value });
+  };
+
+  onResetQualityDefinitionsConfirmed = () => {
+    const resetDefinitionTitles = this.state.resetDefinitionTitles;
+
+    this.setState({ resetDefinitionTitles: false });
+    this.props.onResetQualityDefinitions(resetDefinitionTitles);
+  };
+
+  //
+  // Render
+
+  render() {
+    const {
+      onModalClose,
+      isResettingQualityDefinitions
+    } = this.props;
+
+    const resetDefinitionTitles = this.state.resetDefinitionTitles;
+
+    return (
+      <ModalContent
+        onModalClose={onModalClose}
+      >
+        <ModalHeader>
+          Reset Quality Definitions
+        </ModalHeader>
+
+        <ModalBody>
+          <div className={styles.messageContainer}>
+            Are you sure you want to reset quality definitions?
+          </div>
+
+          <FormGroup>
+            <FormLabel>Reset Titles</FormLabel>
+
+            <FormInputGroup
+              type={inputTypes.CHECK}
+              name="resetDefinitionTitles"
+              value={resetDefinitionTitles}
+              helpText="Reset definition titles as well as values"
+              onChange={this.onResetDefinitionTitlesChange}
+            />
+          </FormGroup>
+
+        </ModalBody>
+
+        <ModalFooter>
+          <Button onPress={onModalClose}>
+            Cancel
+          </Button>
+
+          <Button
+            kind={kinds.DANGER}
+            onPress={this.onResetQualityDefinitionsConfirmed}
+            isDisabled={isResettingQualityDefinitions}
+          >
+            Reset
+          </Button>
+        </ModalFooter>
+      </ModalContent>
+    );
+  }
+}
+
+ResetQualityDefinitionsModalContent.propTypes = {
+  onResetQualityDefinitions: PropTypes.func.isRequired,
+  isResettingQualityDefinitions: PropTypes.bool.isRequired,
+  onModalClose: PropTypes.func.isRequired
+};
+
+export default ResetQualityDefinitionsModalContent;

frontend/src/Settings/Quality/Reset/ResetQualityDefinitionsModalContentConnector.js

@@ -0,0 +1,54 @@
+import PropTypes from 'prop-types';
+import React, { Component } from 'react';
+import { connect } from 'react-redux';
+import { createSelector } from 'reselect';
+import * as commandNames from 'Commands/commandNames';
+import { executeCommand } from 'Store/Actions/commandActions';
+import createCommandExecutingSelector from 'Store/Selectors/createCommandExecutingSelector';
+import ResetQualityDefinitionsModalContent from './ResetQualityDefinitionsModalContent';
+
+function createMapStateToProps() {
+  return createSelector(
+    createCommandExecutingSelector(commandNames.RESET_QUALITY_DEFINITIONS),
+    (isResettingQualityDefinitions) => {
+      return {
+        isResettingQualityDefinitions
+      };
+    }
+  );
+}
+
+const mapDispatchToProps = {
+  executeCommand
+};
+
+class ResetQualityDefinitionsModalContentConnector extends Component {
+
+  //
+  // Listeners
+
+  onResetQualityDefinitions = (resetTitles) => {
+    this.props.executeCommand({ name: commandNames.RESET_QUALITY_DEFINITIONS, resetTitles });
+    this.props.onModalClose(true);
+  };
+
+  //
+  // Render
+
+  render() {
+    return (
+      <ResetQualityDefinitionsModalContent
+        {...this.props}
+        onResetQualityDefinitions={this.onResetQualityDefinitions}
+      />
+    );
+  }
+}
+
+ResetQualityDefinitionsModalContentConnector.propTypes = {
+  onModalClose: PropTypes.func.isRequired,
+  isResettingQualityDefinitions: PropTypes.bool.isRequired,
+  executeCommand: PropTypes.func.isRequired
+};
+
+export default connect(createMapStateToProps, mapDispatchToProps)(ResetQualityDefinitionsModalContentConnector);

src/NzbDrone.Core/Qualities/Commands/ResetQualityDefinitionsCommand.cs

@@ -0,0 +1,14 @@
+using NzbDrone.Core.Messaging.Commands;
+
+namespace NzbDrone.Core.Qualities.Commands
+{
+    public class ResetQualityDefinitionsCommand : Command
+    {
+        public bool ResetTitles { get; set; }
+
+        public ResetQualityDefinitionsCommand(bool resetTitles = false)
+        {
+            ResetTitles = resetTitles;
+        }
+    }
+}

src/NzbDrone.Core/Qualities/QualityDefinitionService.cs

@@ -4,7 +4,9 @@ using System.Linq;
 using NLog;
 using NzbDrone.Common.Cache;
 using NzbDrone.Core.Lifecycle;
+using NzbDrone.Core.Messaging.Commands;
 using NzbDrone.Core.Messaging.Events;
+using NzbDrone.Core.Qualities.Commands;
 
 namespace NzbDrone.Core.Qualities
 {
@@ -17,7 +19,7 @@ namespace NzbDrone.Core.Qualities
         QualityDefinition Get(Quality quality);
     }
 
-    public class QualityDefinitionService : IQualityDefinitionService, IHandle<ApplicationStartedEvent>
+    public class QualityDefinitionService : IQualityDefinitionService, IExecute<ResetQualityDefinitionsCommand>, IHandle<ApplicationStartedEvent>
     {
         private readonly IQualityDefinitionRepository _repo;
         private readonly ICached<Dictionary<Quality, QualityDefinition>> _cache;
@@ -105,5 +107,28 @@ namespace NzbDrone.Core.Qualities
 
             InsertMissingDefinitions();
         }
+
+        public void Execute(ResetQualityDefinitionsCommand message)
+        {
+            List<QualityDefinition> updateList = new List<QualityDefinition>();
+
+            var allDefinitions = Quality.DefaultQualityDefinitions.OrderBy(d => d.Weight).ToList();
+            var existingDefinitions = _repo.All().ToList();
+
+            foreach (var definition in allDefinitions)
+            {
+                var existing = existingDefinitions.SingleOrDefault(d => d.Quality == definition.Quality);
+
+                existing.MinSize = definition.MinSize;
+                existing.MaxSize = definition.MaxSize;
+                existing.Title = message.ResetTitles ? definition.Title : existing.Title;
+
+                updateList.Add(existing);
+            }
+
+            _repo.UpdateMany(updateList);
+
+            _cache.Clear();
+        }
     }
 }

src/Readarr.Api.V1/Qualities/QualityDefinitionController.cs

@@ -1,19 +1,23 @@
 using System.Collections.Generic;
 using System.Linq;
 using Microsoft.AspNetCore.Mvc;
+using NzbDrone.Core.Datastore.Events;
+using NzbDrone.Core.Messaging.Events;
 using NzbDrone.Core.Qualities;
 using NzbDrone.Http.REST.Attributes;
+using NzbDrone.SignalR;
 using Readarr.Http;
 using Readarr.Http.REST;
 
 namespace Readarr.Api.V1.Qualities
 {
     [V1ApiController]
-    public class QualityDefinitionController : RestController<QualityDefinitionResource>
+    public class QualityDefinitionController : RestControllerWithSignalR<QualityDefinitionResource, QualityDefinition>, IHandle<CommandExecutedEvent>
     {
         private readonly IQualityDefinitionService _qualityDefinitionService;
 
-        public QualityDefinitionController(IQualityDefinitionService qualityDefinitionService)
+        public QualityDefinitionController(IQualityDefinitionService qualityDefinitionService, IBroadcastSignalRMessage signalRBroadcaster)
+            : base(signalRBroadcaster)
         {
             _qualityDefinitionService = qualityDefinitionService;
         }
@@ -50,5 +54,14 @@ namespace Readarr.Api.V1.Qualities
             return Accepted(_qualityDefinitionService.All()
                 .ToResource());
         }
+
+        [NonAction]
+        public void Handle(CommandExecutedEvent message)
+        {
+            if (message.Command.Name == "ResetQualityDefinitions")
+            {
+                BroadcastResourceChange(ModelAction.Sync);
+            }
+        }
     }
 }