private List<Linkable> findPossibleEntryPointsByName(
final SecurityContext securityContext, HttpServletRequest request, final String name)
throws FrameworkException {
List<Linkable> possibleEntryPoints =
(List<Linkable>) request.getAttribute(POSSIBLE_ENTRY_POINTS);
if (CollectionUtils.isNotEmpty(possibleEntryPoints)) {
return possibleEntryPoints;
}
if (name.length() > 0) {
logger.log(Level.FINE, "Requested name: {0}", name);
final Query query = StructrApp.getInstance(securityContext).nodeQuery();
query.and(AbstractNode.name, name);
query.and().orType(Page.class).orTypes(File.class);
// Searching for pages needs super user context anyway
Result results = query.getResult();
logger.log(Level.FINE, "{0} results", results.size());
request.setAttribute(POSSIBLE_ENTRY_POINTS, results.getResults());
return (List<Linkable>) results.getResults();
}
return Collections.EMPTY_LIST;
}
/**
* Find the page with the lowest position value which is visible in the current securit context
*
* @param securityContext
* @return
* @throws FrameworkException
*/
private Page findIndexPage(final SecurityContext securityContext) throws FrameworkException {
Result<Page> results =
StructrApp.getInstance(securityContext)
.nodeQuery(Page.class)
.sort(Page.position)
.order(false)
.getResult();
Collections.sort(
results.getResults(),
new GraphObjectComparator(Page.position, GraphObjectComparator.ASCENDING));
// Find first visible page
Page page = null;
if (!results.isEmpty()) {
int i = 0;
while (page == null || (i < results.size() && !securityContext.isVisible(page))) {
page = results.get(i++);
}
}
return page;
}
@Override
public Result doGet(
PropertyKey sortKey, boolean sortDescending, int pageSize, int page, String offsetId)
throws FrameworkException {
if (wrappedResource != null) {
Result result = wrappedResource.doGet(sortKey, sortDescending, pageSize, page, offsetId);
try {
Collections.sort(result.getResults(), new GraphObjectComparator(sortKey, sortOrder));
} catch (Throwable t) {
throw new FrameworkException("base", new InvalidSortKey(sortKey));
}
return result;
}
throw new IllegalPathException();
}
/**
* Find first node whose name matches the last part of the given path
*
* @param securityContext
* @param request
* @param path
* @return
* @throws FrameworkException
*/
private AbstractNode findFirstNodeByName(
final SecurityContext securityContext, HttpServletRequest request, final String path)
throws FrameworkException {
final String name = PathHelper.getName(path);
if (!name.isEmpty()) {
logger.log(Level.FINE, "Requested name: {0}", name);
final Result results =
StructrApp.getInstance(securityContext)
.nodeQuery()
.and(AbstractNode.name, name)
.getResult();
logger.log(Level.FINE, "{0} results", results.size());
request.setAttribute(POSSIBLE_ENTRY_POINTS, results.getResults());
return (results.size() > 0 ? (AbstractNode) results.get(0) : null);
}
return null;
}
@Override
public void execute(Map<String, Object> attributes) throws FrameworkException {
final String entityType = (String) attributes.get("type");
final String relType = (String) attributes.get("relType");
final GraphDatabaseService graphDb = (GraphDatabaseService) arguments.get("graphDb");
final SecurityContext superUserContext = SecurityContext.getSuperUserInstance();
final NodeFactory nodeFactory = new NodeFactory(superUserContext);
final RelationshipFactory relFactory = new RelationshipFactory(superUserContext);
if (entityType != null) {
final Class type = EntityContext.getEntityClassForRawType(entityType);
if (type != null) {
// final Result<AbstractNode> result = Services.command(securityContext,
// SearchNodeCommand.class).execute(true, false, Search.andExactType(type.getSimpleName()));
final Result<AbstractNode> result =
nodeFactory.instantiateAll(GlobalGraphOperations.at(graphDb).getAllNodes());
final List<AbstractNode> nodes = new ArrayList<AbstractNode>();
for (AbstractNode node : result.getResults()) {
if (node.getClass().equals(type)) {
nodes.add(node);
}
}
logger.log(
Level.INFO,
"Start (re-)indexing all nodes of type {0}",
new Object[] {type.getSimpleName()});
long count =
bulkGraphOperation(
securityContext,
nodes,
1000,
"RebuildIndex",
new BulkGraphOperation<AbstractNode>() {
@Override
public void handleGraphObject(
SecurityContext securityContext, AbstractNode node) {
node.updateInIndex();
}
@Override
public void handleThrowable(
SecurityContext securityContext, Throwable t, AbstractNode node) {
logger.log(
Level.WARNING,
"Unable to index node {0}: {1}",
new Object[] {node, t.getMessage()});
}
@Override
public void handleTransactionFailure(
SecurityContext securityContext, Throwable t) {
logger.log(Level.WARNING, "Unable to index node: {0}", t.getMessage());
}
});
logger.log(Level.INFO, "Done with (re-)indexing {0} nodes", count);
return;
}
} else if (relType != null) {
// final Result<AbstractNode> result = Services.command(securityContext,
// SearchNodeCommand.class).execute(true, false, Search.andExactType(type.getSimpleName()));
final List<AbstractRelationship> unfilteredRels =
relFactory.instantiate(GlobalGraphOperations.at(graphDb).getAllRelationships());
final List<AbstractRelationship> rels = new ArrayList<AbstractRelationship>();
for (AbstractRelationship rel : unfilteredRels) {
if (!rel.getType().equals(relType)) {
rels.add(rel);
}
}
logger.log(Level.INFO, "Start setting UUID on all rels of type {0}", new Object[] {relType});
long count =
bulkGraphOperation(
securityContext,
rels,
1000,
"SetRelationshipUuid",
new BulkGraphOperation<AbstractRelationship>() {
@Override
public void handleGraphObject(
SecurityContext securityContext, AbstractRelationship rel) {
rel.updateInIndex();
}
@Override
public void handleThrowable(
SecurityContext securityContext, Throwable t, AbstractRelationship rel) {
logger.log(
Level.WARNING,
"Unable to index relationship {0}: {1}",
new Object[] {rel, t.getMessage()});
}
@Override
public void handleTransactionFailure(SecurityContext securityContext, Throwable t) {
logger.log(Level.WARNING, "Unable to index relationship: {0}", t.getMessage());
}
});
logger.log(Level.INFO, "Done with (re-)indexing {0} relationships", count);
return;
}
logger.log(Level.INFO, "Unable to determine entity type to re-index.");
}
@Override
public List<GraphObject> getData(
final SecurityContext securityContext,
final RenderContext renderContext,
final String restQuery)
throws FrameworkException {
Map<Pattern, Class<? extends Resource>> resourceMap = new LinkedHashMap<>();
ResourceProvider resourceProvider =
renderContext == null ? null : renderContext.getResourceProvider();
if (resourceProvider == null) {
try {
resourceProvider = UiResourceProvider.class.newInstance();
} catch (Throwable t) {
logger.log(Level.SEVERE, "Couldn't establish a resource provider", t);
return Collections.EMPTY_LIST;
}
}
// inject resources
resourceMap.putAll(resourceProvider.getResources());
Value<String> propertyView = new ThreadLocalPropertyView();
propertyView.set(securityContext, PropertyView.Ui);
// initialize variables
// mimic HTTP request
HttpServletRequest request =
new HttpServletRequestWrapper(
renderContext == null ? securityContext.getRequest() : renderContext.getRequest()) {
@Override
public Enumeration<String> getParameterNames() {
return new IteratorEnumeration(getParameterMap().keySet().iterator());
}
@Override
public String getParameter(String key) {
String[] p = getParameterMap().get(key);
return p != null ? p[0] : null;
}
@Override
public Map<String, String[]> getParameterMap() {
String[] parts = StringUtils.split(getQueryString(), "&");
Map<String, String[]> parameterMap = new HashMap();
for (String p : parts) {
String[] kv = StringUtils.split(p, "=");
if (kv.length > 1) {
parameterMap.put(kv[0], new String[] {kv[1]});
}
}
return parameterMap;
}
@Override
public String getQueryString() {
return StringUtils.substringAfter(restQuery, "?");
}
@Override
public String getPathInfo() {
return StringUtils.substringBefore(restQuery, "?");
}
@Override
public StringBuffer getRequestURL() {
return new StringBuffer(restQuery);
}
};
// update request in security context
securityContext.setRequest(request);
// HttpServletResponse response = renderContext.getResponse();
Resource resource =
ResourceHelper.applyViewTransformation(
request,
securityContext,
ResourceHelper.optimizeNestedResourceChain(
ResourceHelper.parsePath(
securityContext, request, resourceMap, propertyView, GraphObject.id),
GraphObject.id),
propertyView);
// TODO: decide if we need to rest the REST request here
// securityContext.checkResourceAccess(request, resource.getResourceSignature(),
// resource.getGrant(request, response), PropertyView.Ui);
// add sorting & paging
String pageSizeParameter = request.getParameter(JsonRestServlet.REQUEST_PARAMETER_PAGE_SIZE);
String pageParameter = request.getParameter(JsonRestServlet.REQUEST_PARAMETER_PAGE_NUMBER);
String offsetId = request.getParameter(JsonRestServlet.REQUEST_PARAMETER_OFFSET_ID);
String sortOrder = request.getParameter(JsonRestServlet.REQUEST_PARAMETER_SORT_ORDER);
String sortKeyName = request.getParameter(JsonRestServlet.REQUEST_PARAMETER_SORT_KEY);
boolean sortDescending = (sortOrder != null && "desc".equals(sortOrder.toLowerCase()));
int pageSize = parseInt(pageSizeParameter, NodeFactory.DEFAULT_PAGE_SIZE);
int page = parseInt(pageParameter, NodeFactory.DEFAULT_PAGE);
PropertyKey sortKey = null;
// set sort key
if (sortKeyName != null) {
Class<? extends GraphObject> type = resource.getEntityClass();
if (type == null) {
// fallback to default implementation
// if no type can be determined
type = AbstractNode.class;
}
sortKey = StructrApp.getConfiguration().getPropertyKeyForDatabaseName(type, sortKeyName);
}
// do action
Result result = resource.doGet(sortKey, sortDescending, pageSize, page, offsetId);
result.setIsCollection(resource.isCollectionResource());
result.setIsPrimitiveArray(resource.isPrimitiveArray());
// Integer rawResultCount = (Integer) Services.getAttribute(NodeFactory.RAW_RESULT_COUNT +
// Thread.currentThread().getId());
PagingHelper.addPagingParameter(result, pageSize, page);
List<GraphObject> res = result.getResults();
if (renderContext != null) {
renderContext.setResult(result);
}
return res != null ? res : Collections.EMPTY_LIST;
}