/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
'use strict';
angular.module('dgc.search').controller('searchController', ['$scope', '$location', '$http', '$state', '$stateParams', 'lodash', 'searchResource', 'detailsResource', 'notificationService', 'atlasConfig',
function($scope, $location, $http, $state, $stateParams, _, searchResource, detailsResource, notificationService, atlasConfig) {
$scope.results = [];
$scope.resultCount = 0;
$scope.isCollapsed = true;
$scope.currentPage = 1;
$scope.itemsPerPage = 10;
$scope.filteredResults = [];
$scope.resultRows = [];
$scope.resultType = '';
$scope.isObject = angular.isObject;
$scope.isString = angular.isString;
$scope.isArray = angular.isArray;
$scope.isNumber = angular.isNumber;
$scope.searchTypes = atlasConfig.SEARCH_TYPE;
$scope.mapAttr = ['guid', 'typeName', 'owner', 'description', 'createTime', '$traits$', '$id$', 'comment', 'dataType'];
$scope.setPage = function(pageNo) {
$scope.currentPage = pageNo;
};
$scope.initalPage = function() {
var begin = (($scope.currentPage - 1) * $scope.itemsPerPage),
end = begin + $scope.itemsPerPage,
currentRowData = [],
res = [],
count = 0,
firstRowId = 0;
if ($scope.transformedResults && $scope.transformedResults.length > 0) {
currentRowData = $scope.transformedResults.slice(begin, end);
}
var getName = function(gid, obj) {
detailsResource.get({
id: gid
}, function(data) {
if (data.values && data.values.name) {
obj.name = data.values.name;
} else {
obj.name = gid;
}
res.push(obj);
count++;
$scope.filteredResults = res;
if (!$scope.transformedProperties && firstRowId === gid) {
$scope.transformedProperties = $scope.filterProperties();
}
});
};
angular.forEach(currentRowData, function(value) {
var objVal = value;
if (objVal.guid && !objVal.name) {
if (!firstRowId) {
firstRowId = objVal.guid;
}
getName(objVal.guid, objVal);
} else {
res.push(objVal);
count++;
$scope.filteredResults = res;
if (!$scope.transformedProperties) {
$scope.transformedProperties = $scope.filterProperties();
}
}
});
};
$scope.search = function(query, type) {
$scope.results = [];
notificationService.reset();
$scope.limit = 4;
$scope.searchMessage = 'load-gif';
$scope.$parent.query = query;
searchResource.search({
query: query,
searchType: type
}, function searchSuccess(response) {
$scope.resultCount = response.count;
$scope.results = response.results;
$scope.resultRows = ($scope.results && $scope.results.rows) ? $scope.results.rows : $scope.results;
$scope.totalItems = $scope.resultCount;
$scope.transformedResults = {};
$scope.dataTransitioned = false;
$scope.showText = "Text";
if ($scope.searchTypeModel === $scope.searchTypes.dsl.value) {
$scope.showText = "DSL";
}
if ($scope.results) {
if (response.dataType) {
$scope.resultType = response.dataType.typeName;
} else if (response.results.dataType) {
$scope.resultType = response.results.dataType.typeName;
} else if (typeof response.dataType === 'undefined') {
$scope.resultType = "full text";
}
$scope.searchMessage = $scope.resultCount + ' results matching your ' + $scope.showText + ' search query ' + $scope.query + ' were found';
} else {
$scope.searchMessage = '0 results matching your ' + $scope.showText + ' search query ' + $scope.query + ' were found';
}
if (response.dataType && response.dataType.typeName && response.dataType.typeName.toLowerCase().indexOf('table') === -1) {
$scope.dataTransitioned = true;
var attrDef = response.dataType.attributeDefinitions;
if (attrDef.length === 1) {
$scope.searchKey = attrDef[0].name;
} else {
angular.forEach(attrDef, function(value) {
if (value.dataTypeName === '__IdType') {
$scope.searchKey = value.name;
}
});
if ($scope.searchKey === undefined || $scope.searchKey === '') {
$scope.searchKey = '';
}
}
$scope.transformedResults = $scope.filterResults();
} else if (typeof response.dataType === 'undefined') {
$scope.dataTransitioned = true;
$scope.searchKey = '';
$scope.transformedResults = $scope.filterResults();
} else if (response.dataType.typeName && response.dataType.typeName.toLowerCase().indexOf('table') !== -1) {
$scope.searchKey = "Table";
$scope.transformedResults = $scope.resultRows;
} else if (response.results.dataType && response.results.dataType.typeName && response.results.dataType.typeName.toLowerCase().indexOf('table') !== -1) {
$scope.searchKey = "Table";
$scope.transformedResults = $scope.resultRows;
}
$scope.$watch('currentPage + itemsPerPage', function() {
$scope.initalPage();
$scope.pageCount = function() {
return Math.ceil($scope.resultCount / $scope.itemsPerPage);
};
if ($scope.results.length < 1) {
notificationService.error('No Result found', false);
}
});
}, function searchError(err) {
if (($scope.searchTypeModel === $scope.searchTypes.dsl.value) && err.status === 400) {
$scope.searchMessageDsl = true;
$scope.searchMessage = false;
} else {
$scope.searchMessageDsl = false;
$scope.searchMessage = true;
$scope.searchMessage = '0 results matching your ' + $scope.showText + ' search query ' + $scope.query + ' were found';
notificationService.error('Error occurred during executing search query, error status code = ' + err.status + ', status text = ' + err.statusText, false);
}
});
$state.go('search', {
query: query
}, {
location: 'replace'
});
};
$scope.filterResults = function() {
var res = [];
if ($scope.searchKey !== '') {
angular.forEach($scope.resultRows, function(value) {
var objVal = value[$scope.searchKey];
res.push(objVal);
});
} else {
angular.forEach($scope.resultRows, function(value) {
var objVal = {},
curVal = value,
onlyId = false,
traits = false;
if (curVal.name) {
objVal.name = curVal.name;
delete curVal.name;
}
angular.forEach(curVal, function(vl, ky) {
if ($scope.mapAttr.indexOf(ky) !== -1 || ky.indexOf('_col_') !== -1 || ky.toLowerCase().indexOf('name') !== -1) {
if (ky === '$id$') {
objVal.id = curVal[ky].id;
onlyId = true;
traits = true;
} else if (ky.indexOf('$') === -1) {
objVal[ky] = vl;
onlyId = false;
}
}
});
if (traits) {
objVal.$traits$ = curVal.$traits$ || {};
objVal.Tools = objVal.id;
}
if (onlyId) {
objVal.guid = objVal.id;
}
res.push(objVal);
});
}
return res;
};
$scope.filterProperties = function() {
var results = $scope.transformedResults,
pro = [];
if (results && results.length > 0) {
var result = results[0];
if (typeof result === 'object') {
angular.forEach(result, function(value, key) {
if (key.indexOf('$typeName$') === -1) {
pro.push(key);
}
});
} else {
pro.push($scope.searchKey);
}
}
return pro;
};
$scope.doToggle = function($event, el) {
this.isCollapsed = !el;
};
$scope.openAddTagHome = function(traitId) {
$state.go('addTagHome', {
tId: traitId
});
};
$scope.isTag = function(typename) {
if (typename.indexOf("__tempQueryResultStruct") > -1 || $scope.searchKey === '') {
return true;
} else {
return false;
}
};
$scope.getResourceDataHome = function(event, id) {
detailsResource.get({
id: id
}, function(data) {
if ($scope.filteredResults !== null && Object.keys($scope.filteredResults).length > 0) {
angular.forEach($scope.filteredResults, function(obj, trait) {
if ((obj.$id$ && obj.$id$.id.indexOf(id) > -1) || (obj.id && obj.id.indexOf(id) > -1)) {
$scope.filteredResults[trait].$traits$ = data.traits;
}
});
}
});
};
$scope.$on('refreshResourceData', $scope.getResourceDataHome);
$scope.filterSearchResults = function(items) {
var res = {};
var count = 0;
items = _.omit(items, ['name', 'description', 'guid']);
angular.forEach(items, function(value, key) {
if (typeof value !== 'object' && (key.indexOf('$$') < 0)) {
res[key] = value;
count++;
}
});
$scope.keyLength = count;
return res;
};
$scope.searchQuery = $location.search();
$scope.searchTypeModel = $stateParams.searchType;
$scope.query = ($location.search()).query;
$stateParams.searchType = ($location.search()).searchType;
if ($scope.query && $stateParams.searchType) {
$scope.search($scope.query, $stateParams.searchType);
}
$scope.searchTypeChanged = function() {
if ($scope.query) {
$state.go('search', {
query: $scope.query,
searchType: $scope.searchTypeModel
}, {
location: 'replace'
});
}
};
}
]);