package org.drools.guvnor.server.builder.pagerow;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import org.drools.guvnor.client.rpc.PackageConfigData;
import org.drools.guvnor.client.rpc.QueryMetadataPageRequest;
import org.drools.guvnor.client.rpc.QueryPageRow;
import org.drools.guvnor.server.CategoryFilter;
import org.drools.guvnor.server.PackageFilter;
import org.drools.guvnor.server.security.RoleTypes;
import org.drools.guvnor.server.util.QueryPageRowCreator;
import org.drools.repository.AssetItem;
import org.drools.repository.CategoryItem;
import org.drools.repository.RepositoryFilter;
public class QueryMetadataPageRowBuilder {
public List<QueryPageRow> createRows(QueryMetadataPageRequest pageRequest,
Iterator<AssetItem> iterator) {
int skipped = 0;
Integer pageSize = pageRequest.getPageSize();
int startRowIndex = pageRequest.getStartRowIndex();
RepositoryFilter packageFilter = new PackageFilter();
RepositoryFilter categoryFilter = new CategoryFilter();
List<QueryPageRow> rowList = new ArrayList<QueryPageRow>();
while ( iterator.hasNext() && (pageSize == null || rowList.size() < pageSize) ) {
AssetItem assetItem = (AssetItem) iterator.next();
// Filter surplus assets
if ( checkPackagePermissionHelper( packageFilter,
assetItem,
RoleTypes.PACKAGE_READONLY ) || checkCategoryPermissionHelper( categoryFilter,
assetItem,
RoleTypes.ANALYST_READ ) ) {
// Cannot use AssetItemIterator.skip() as it skips non-filtered
// assets whereas startRowIndex is the index of the
// first displayed asset (i.e. filtered)
if ( skipped >= startRowIndex ) {
rowList.add( QueryPageRowCreator.makeQueryPageRow( assetItem ) );
}
skipped++;
}
}
return rowList;
}
private boolean checkCategoryPermissionHelper(RepositoryFilter filter,
AssetItem item,
String roleType) {
List<CategoryItem> tempCateList = item.getCategories();
for ( Iterator<CategoryItem> i = tempCateList.iterator(); i.hasNext(); ) {
CategoryItem categoryItem = i.next();
if ( filter.accept( categoryItem.getName(),
roleType ) ) {
return true;
}
}
return false;
}
private boolean checkPackagePermissionHelper(RepositoryFilter filter,
AssetItem item,
String roleType) {
return filter.accept( getConfigDataHelper( item.getPackage().getUUID() ),
roleType );
}
private PackageConfigData getConfigDataHelper(String uuidStr) {
PackageConfigData data = new PackageConfigData();
data.uuid = uuidStr;
return data;
}
}