move data transformation logic into a service so it can be reused

awx/ui/client/src/shared/smart-search/queryset.service.js

@@ -1,5 +1,11 @@
-export default ['$q', 'Rest', 'ProcessErrors', '$rootScope', 'Wait', 'DjangoSearchModel', 'SmartSearchService',
+function searchWithoutKey (term, singleSearchParam = null) {
-    function($q, Rest, ProcessErrors, $rootScope, Wait, DjangoSearchModel, SmartSearchService) {
+    if (singleSearchParam) {
+        return { [singleSearchParam]: `search=${encodeURIComponent(term)}` };
+    }
+    return { search: encodeURIComponent(term) };
+}
+
+function QuerysetService ($q, Rest, ProcessErrors, $rootScope, Wait, DjangoSearchModel, SmartSearchService) {
     return {
         // kick off building a model for a specific endpoint
         // this is usually a list's basePath
@@ -9,7 +15,6 @@ export default ['$q', 'Rest', 'ProcessErrors', '$rootScope', 'Wait', 'DjangoSear
             defer.resolve(this.getCommonModelOptions(path, name));
             return defer.promise;
         },
-
         getCommonModelOptions(path, name) {
             let resolve, base,
                 defer = $q.defer();
@@ -28,18 +33,15 @@ export default ['$q', 'Rest', 'ProcessErrors', '$rootScope', 'Wait', 'DjangoSear
                 });
             return defer.promise;
         },
-
         replaceDefaultFlags (value) {
             value = value.toString().replace(/__icontains_DEFAULT/g, "__icontains");
             value = value.toString().replace(/__search_DEFAULT/g, "__search");
 
             return value;
         },
-
         replaceEncodedTokens(value) {
             return decodeURIComponent(value).replace(/"|'/g, "");
         },
-
         encodeTerms (values, key) {
             key = this.replaceDefaultFlags(key);
 
@@ -204,14 +206,6 @@ export default ['$q', 'Rest', 'ProcessErrors', '$rootScope', 'Wait', 'DjangoSear
                 return decodeParamString(value);
             }
         },
-            convertToSearchTags(obj) {
-                const tags = [];
-                for (let key in obj) {
-                    const value = obj[key];
-                    tags.push(this.decodeParam(value, key));
-                }
-                return tags;
-            },
         // encodes a django queryset for ui-router's URLMatcherFactory
         // {operand__key__comparator: value, } => 'operand:key:comparator:value;...'
         // value.isArray expands to:
@@ -238,11 +232,15 @@ export default ['$q', 'Rest', 'ProcessErrors', '$rootScope', 'Wait', 'DjangoSear
                 }
             }
         },
-
         // decodes a django queryset for ui-router's URLMatcherFactory
         // 'operand:key:comparator:value,...' => {operand__key__comparator: value, }
         decodeArr(arr) {
             let params = {};
+
+            if (!arr) {
+                return params;
+            }
+
             _.forEach(arr.split(';'), (item) => {
                 let key = item.split(':')[0],
                     value = item.split(':')[1];
@@ -303,19 +301,22 @@ export default ['$q', 'Rest', 'ProcessErrors', '$rootScope', 'Wait', 'DjangoSear
             });
         },
         // Removes state definition defaults and pagination terms
-            stripDefaultParams(params, defaults) {
+        stripDefaultParams(params, defaultParams) {
-                if(defaults) {
+            if (!params) {
+                return [];
+            }
+            if(defaultParams) {
                 let stripped =_.pick(params, (value, key) => {
                     // setting the default value of a term to null in a state definition is a very explicit way to ensure it will NEVER generate a search tag, even with a non-default value
-                        return defaults[key] !== value && key !== 'order_by' && key !== 'page' && key !== 'page_size' && defaults[key] !== null;
+                    return defaultParams[key] !== value && key !== 'order_by' && key !== 'page' && key !== 'page_size' && defaultParams[key] !== null;
                 });
                 let strippedCopy = _.cloneDeep(stripped);
-                    if(_.keys(_.pick(defaults, _.keys(strippedCopy))).length > 0){
+                if(_.keys(_.pick(defaultParams, _.keys(strippedCopy))).length > 0){
                     for (var key in strippedCopy) {
                         if (strippedCopy.hasOwnProperty(key)) {
                             let value = strippedCopy[key];
                             if(_.isArray(value)){
-                                    let index = _.indexOf(value, defaults[key]);
+                                let index = _.indexOf(value, defaultParams[key]);
                                 value = value.splice(index, 1)[0];
                             }
                         }
@@ -327,7 +328,180 @@ export default ['$q', 'Rest', 'ProcessErrors', '$rootScope', 'Wait', 'DjangoSear
             else {
                 return _(params).map(this.decodeParam).flatten().value();
             }
+        },
+        mergeQueryset (queryset, additional, singleSearchParam) {
+            const space = '%20and%20';
+
+            const merged = _.merge({}, queryset, additional, (objectValue, sourceValue, key, object) => {
+                if (!(object[key] && object[key] !== sourceValue)) {
+                    // // https://lodash.com/docs/3.10.1#each
+                    // If this returns undefined merging is handled by default _.merge algorithm
+                    return undefined;
+                }
+
+                if (_.isArray(object[key])) {
+                    object[key].push(sourceValue);
+                    return object[key];
+                }
+
+                if (singleSearchParam) {
+                    if (!object[key]) {
+                        return sourceValue;
+                    }
+
+                    const singleSearchParamKeys = object[key].split(space);
+
+                    if (_.includes(singleSearchParamKeys, sourceValue)) {
+                        return object[key];
+                    }
+
+                    return `${object[key]}${space}${sourceValue}`;
+                }
+
+                // Start the array of keys
+                return [object[key], sourceValue];
+            });
+
+            return merged;
+        },
+        getSearchInputQueryset (searchInput, isRelatedField = null, isAnsibleFactField = null, singleSearchParam = null) {
+            // XXX Should find a better approach than passing in the two 'is...Field' callbacks XXX
+            const space = '%20and%20';
+            let params = {};
+
+            // Remove leading/trailing whitespace if there is any
+            const terms = (searchInput) ? searchInput.trim() : '';
+
+            if (!(terms && terms !== '')) {
+                return;
+            }
+
+            let splitTerms;
+
+            if (singleSearchParam === 'host_filter') {
+                splitTerms = SmartSearchService.splitFilterIntoTerms(terms);
+            } else {
+                splitTerms = SmartSearchService.splitSearchIntoTerms(terms);
+            }
+
+            const combineSameSearches = (a, b) => {
+                if (!a) {
+                    return undefined;
+                }
+
+                if (_.isArray(a)) {
+                    return a.concat(b);
+                }
+
+                if (singleSearchParam) {
+                    return `${a}${space}${b}`;
+                }
+
+                return [a, b];
+            };
+
+            _.each(splitTerms, term => {
+                const termParts = SmartSearchService.splitTermIntoParts(term);
+                let termParams;
+
+                if (termParts.length === 1) {
+                    termParams = searchWithoutKey(term, singleSearchParam);
+                } else if (isAnsibleFactField && isAnsibleFactField(termParts)) {
+                    termParams = this.encodeParam({ term, singleSearchParam });
+                } else if (isRelatedField && isRelatedField(termParts)) {
+                    termParams = this.encodeParam({ term, singleSearchParam, related: true });
+                } else {
+                    termParams = this.encodeParam({ term, singleSearchParam });
+                }
+
+                params = _.merge(params, termParams, combineSameSearches);
+            });
+
+            return params;
+        },
+        removeTermsFromQueryset(queryset, term, isRelatedField = null, singleSearchParam = null) {
+            const modifiedQueryset = _.cloneDeep(queryset);
+
+            const removeSingleTermFromQueryset = (value, key) => {
+                const space = '%20and%20';
+
+                if (Array.isArray(modifiedQueryset[key])) {
+                    modifiedQueryset[key] = modifiedQueryset[key].filter(item => item !== value);
+                    if (modifiedQueryset[key].length < 1) {
+                        delete modifiedQueryset[key];
+                    }
+                } else if (singleSearchParam && _.get(modifiedQueryset, singleSearchParam, []).includes(space)) {
+                    const searchParamParts = modifiedQueryset[singleSearchParam].split(space);
+                    // The value side of each paramPart might have been encoded in
+                    // SmartSearch.splitFilterIntoTerms
+                    _.each(searchParamParts, (paramPart, paramPartIndex) => {
+                        searchParamParts[paramPartIndex] = decodeURIComponent(paramPart);
+                    });
+
+                    const paramPartIndex = searchParamParts.indexOf(value);
+
+                    if (paramPartIndex !== -1) {
+                        searchParamParts.splice(paramPartIndex, 1);
+                    }
+
+                    modifiedQueryset[singleSearchParam] = searchParamParts.join(space);
+
+                } else {
+                    delete modifiedQueryset[key];
+                }
+            };
+
+            const termParts = SmartSearchService.splitTermIntoParts(term);
+
+            let removed;
+
+            if (termParts.length === 1) {
+                removed = searchWithoutKey(term, singleSearchParam);
+            } else if (isRelatedField && isRelatedField(termParts)) {
+                removed = this.encodeParam({ term, singleSearchParam, related: true });
+            } else {
+                removed = this.encodeParam({ term, singleSearchParam });
+            }
+
+            if (!removed) {
+                removed = searchWithoutKey(termParts[termParts.length - 1], singleSearchParam);
+            }
+
+            _.each(removed, removeSingleTermFromQueryset);
+
+            return modifiedQueryset;
+        },
+        createSearchTagsFromQueryset(queryset, defaultParams = null, singleSearchParam = null) {
+            const space = '%20and%20';
+            const modifiedQueryset = angular.copy(queryset);
+
+            let searchTags = [];
+
+            if (singleSearchParam && modifiedQueryset[singleSearchParam]) {
+                const searchParam = modifiedQueryset[singleSearchParam].split(space);
+                delete modifiedQueryset[singleSearchParam];
+
+                $.each(searchParam, (index, param) => {
+                    const paramParts = decodeURIComponent(param).split(/=(.+)/);
+                    const reconstructedSearchString = this.decodeParam(paramParts[1], paramParts[0]);
+
+                    searchTags.push(reconstructedSearchString);
+                });
+            }
+
+            return searchTags.concat(this.stripDefaultParams(modifiedQueryset, defaultParams));
         }
     };
 }
+
+QuerysetService.$inject = [
+    '$q',
+    'Rest',
+    'ProcessErrors',
+    '$rootScope',
+    'Wait',
+    'DjangoSearchModel',
+    'SmartSearchService',
 ];
+
+export default QuerysetService;

awx/ui/client/src/shared/smart-search/smart-search.controller.js

@@ -7,7 +7,6 @@ function SmartSearchController (
     GetBasePath,
     i18n,
     qs,
-    SmartSearchService
 ) {
     const searchKey = `${$scope.iterator}_search`;
     const optionsKey = `${$scope.list.iterator}_options`;
@@ -106,22 +105,8 @@ function SmartSearchController (
     }
 
     function generateSearchTags () {
-        $scope.searchTags = [];
+        const { singleSearchParam } = $scope;
-
+        $scope.searchTags = qs.createSearchTagsFromQueryset(queryset, defaults, singleSearchParam);
-        const querysetCopy = angular.copy(queryset);
-
-        if ($scope.singleSearchParam && querysetCopy[$scope.singleSearchParam]) {
-            const searchParam = querysetCopy[$scope.singleSearchParam].split('%20and%20');
-            delete querysetCopy[$scope.singleSearchParam];
-
-            $.each(searchParam, (index, param) => {
-                const paramParts = decodeURIComponent(param).split(/=(.+)/);
-                const reconstructedSearchString = qs.decodeParam(paramParts[1], paramParts[0]);
-                $scope.searchTags.push(reconstructedSearchString);
-            });
-        }
-
-        $scope.searchTags = $scope.searchTags.concat(qs.stripDefaultParams(querysetCopy, defaults));
     }
 
     function revertSearch (queryToBeRestored) {
@@ -149,14 +134,7 @@ function SmartSearchController (
         $scope.showKeyPane = !$scope.showKeyPane;
     };
 
-    function searchWithoutKey (term, singleSearchParam = null) {
+    function isAnsibleFactField (termParts) {
-        if (singleSearchParam) {
-            return { [singleSearchParam]: `search=${encodeURIComponent(term)}` };
-        }
-        return { search: encodeURIComponent(term) };
-    }
-
-    function isAnsibleFactSearchTerm (termParts) {
         const rootField = termParts[0].split('.')[0].replace(/^-/, '');
         return rootField === 'ansible_facts';
     }
@@ -172,111 +150,21 @@ function SmartSearchController (
         return (isRelatedSearchTermField || isBaseModelRelatedSearchTermField);
     }
 
-    function getSearchInputQueryset ({ terms, singleSearchParam }) {
-        let params = {};
-
-        // remove leading/trailing whitespace
-        terms = (terms) ? terms.trim() : '';
-        let splitTerms;
-
-        if (singleSearchParam === 'host_filter') {
-            splitTerms = SmartSearchService.splitFilterIntoTerms(terms);
-        } else {
-            splitTerms = SmartSearchService.splitSearchIntoTerms(terms);
-        }
-
-        const combineSameSearches = (a, b) => {
-            if (!a) {
-                return undefined;
-            }
-
-            if (_.isArray(a)) {
-                return a.concat(b);
-            }
-
-            if (singleSearchParam) {
-                return `${a}%20and%20${b}`;
-            }
-
-            return [a, b];
-        };
-
-        _.each(splitTerms, term => {
-            const termParts = SmartSearchService.splitTermIntoParts(term);
-            let termParams;
-
-            if (termParts.length === 1) {
-                termParams = searchWithoutKey(term, singleSearchParam);
-            } else if (isAnsibleFactSearchTerm(termParts)) {
-                termParams = qs.encodeParam({ term, singleSearchParam });
-            } else if (isRelatedField(termParts)) {
-                termParams = qs.encodeParam({ term, singleSearchParam, related: true });
-            } else {
-                termParams = qs.encodeParam({ term, singleSearchParam });
-            }
-
-            params = _.merge(params, termParams, combineSameSearches);
-        });
-
-        return params;
-    }
-
-    function mergeQueryset (qset, additional, singleSearchParam) {
-        const merged = _.merge({}, qset, additional, (objectValue, sourceValue, key, object) => {
-            if (!(object[key] && object[key] !== sourceValue)) {
-                // // https://lodash.com/docs/3.10.1#each
-                // If this returns undefined merging is handled by default _.merge algorithm
-                return undefined;
-            }
-
-            if (_.isArray(object[key])) {
-                object[key].push(sourceValue);
-                return object[key];
-            }
-
-            if (singleSearchParam) {
-                if (!object[key]) {
-                    return sourceValue;
-                }
-
-                const singleSearchParamKeys = object[key].split('%20and%20');
-
-                if (_.includes(singleSearchParamKeys, sourceValue)) {
-                    return object[key];
-                }
-
-                return `${object[key]}%20and%20${sourceValue}`;
-            }
-
-            // Start the array of keys
-            return [object[key], sourceValue];
-        });
-
-        return merged;
-    }
-
     $scope.addTerms = terms => {
         const { singleSearchParam } = $scope;
-        const origQueryset = _.clone(queryset);
+        const unmodifiedQueryset = _.clone(queryset);
 
-        // Remove leading/trailing whitespace if there is any
+        const searchInputQueryset = qs.getSearchInputQueryset(terms, isRelatedField, isAnsibleFactField, singleSearchParam);
-        terms = (terms) ? terms.trim() : '';
+        const modifiedQueryset = qs.mergeQueryset(queryset, searchInputQueryset, singleSearchParam);
-
-        if (!(terms && terms !== '')) {
-            return;
-        }
-
-        const searchInputQueryset = getSearchInputQueryset({ terms, singleSearchParam });
-        queryset = mergeQueryset(queryset, searchInputQueryset, singleSearchParam);
 
         // Go back to the first page after a new search
-        delete queryset.page;
+        delete modifiedQueryset.page;
 
         // https://ui-router.github.io/docs/latest/interfaces/params.paramdeclaration.html#dynamic
         // This transition will not reload controllers/resolves/views but will register new
         // $stateParams[searchKey] terms.
         if (!$scope.querySet) {
-            $state.go('.', { [searchKey]: queryset })
+            $state.go('.', { [searchKey]: modifiedQueryset })
                 .then(() => {
                     // same as above in $scope.remove.  For some reason deleting the page
                     // from the queryset works for all lists except lists in modals.
@@ -284,80 +172,26 @@ function SmartSearchController (
                 });
         }
 
-        qs.search(path, queryset)
+        qs.search(path, modifiedQueryset)
             .then(({ data }) => {
                 if ($scope.querySet) {
-                    $scope.querySet = queryset;
+                    $scope.querySet = modifiedQueryset;
                 }
                 $scope.dataset = data;
                 $scope.collection = data.results;
             })
-            .catch(() => revertSearch(origQueryset));
+            .catch(() => revertSearch(unmodifiedQueryset));
 
         $scope.searchTerm = null;
 
         generateSearchTags();
     };
-
-    function removeTermsFromQueryset(qset, term, singleSearchParam = null) {
-        const returnedQueryset = _.cloneDeep(qset);
-
-        const removeSingleTermFromQueryset = (value, key) => {
-            const space = '%20and%20';
-
-            if (Array.isArray(returnedQueryset[key])) {
-                returnedQueryset[key] = returnedQueryset[key].filter(item => item !== value);
-                if (returnedQueryset[key].length < 1) {
-                    delete returnedQueryset[key];
-                }
-            } else if (singleSearchParam && _.get(returnedQueryset, singleSearchParam, []).includes(space)) {
-                const searchParamParts = returnedQueryset[singleSearchParam].split(space);
-                // The value side of each paramPart might have been encoded in
-                // SmartSearch.splitFilterIntoTerms
-                _.each(searchParamParts, (paramPart, paramPartIndex) => {
-                    searchParamParts[paramPartIndex] = decodeURIComponent(paramPart);
-                });
-
-                const paramPartIndex = searchParamParts.indexOf(value);
-
-                if (paramPartIndex !== -1) {
-                    searchParamParts.splice(paramPartIndex, 1);
-                }
-
-                returnedQueryset[singleSearchParam] = searchParamParts.join(space);
-
-            } else {
-                delete returnedQueryset[key];
-            }
-        };
-
-        const termParts = SmartSearchService.splitTermIntoParts(term);
-
-        let removed;
-
-        if (termParts.length === 1) {
-            removed = searchWithoutKey(term, singleSearchParam);
-        } else if (isRelatedField(termParts)) {
-            removed = qs.encodeParam({ term, singleSearchParam, related: true });
-        } else {
-            removed = qs.encodeParam({ term, singleSearchParam });
-        }
-
-        if (!removed) {
-            removed = searchWithoutKey(termParts[termParts.length - 1], singleSearchParam);
-        }
-
-        _.each(removed, removeSingleTermFromQueryset);
-
-        return returnedQueryset;
-    }
-
     // remove tag, merge new queryset, $state.go
     $scope.removeTerm = index => {
         const { singleSearchParam } = $scope;
         const [term] = $scope.searchTags.splice(index, 1);
 
-        const modifiedQueryset = removeTermsFromQueryset(queryset, term, singleSearchParam);
+        const modifiedQueryset = qs.removeTermsFromQueryset(queryset, term, isRelatedField, singleSearchParam);
 
         if (!$scope.querySet) {
             $state.go('.', { [searchKey]: modifiedQueryset })
@@ -365,8 +199,7 @@ function SmartSearchController (
                     // for some reason deleting a tag from a list in a modal does not
                     // remove the param from $stateParams.  Here we'll manually check to make sure
                     // that that happened and remove it if it didn't.
-                    const clearedParams = removeTermsFromQueryset(
+                    const clearedParams = qs.removeTermsFromQueryset($stateParams[searchKey], term, isRelatedField, singleSearchParam);
-                        $stateParams[searchKey], term, singleSearchParam);
                     $stateParams[searchKey] = clearedParams;
                 });
         }
@@ -416,7 +249,6 @@ SmartSearchController.$inject = [
     'GetBasePath',
     'i18n',
     'QuerySet',
-    'SmartSearchService',
 ];
 
 export default SmartSearchController;