Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/**
* This file is part of Graylog.
*
* Graylog is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* Graylog is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Graylog. If not, see .
*/
package org.graylog.plugins.views.search.rest;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import org.apache.shiro.authz.annotation.RequiresAuthentication;
import org.graylog.plugins.views.audit.ViewsAuditEventTypes;
import org.graylog.plugins.views.search.views.ViewDTO;
import org.graylog.plugins.views.search.views.ViewService;
import org.graylog.plugins.views.search.views.sharing.IsViewSharedForUser;
import org.graylog.plugins.views.search.views.sharing.ViewSharing;
import org.graylog.plugins.views.search.views.sharing.ViewSharingService;
import org.graylog2.audit.jersey.AuditEvent;
import org.graylog2.database.PaginatedList;
import org.graylog2.plugin.database.ValidationException;
import org.graylog2.plugin.database.users.User;
import org.graylog2.plugin.rest.PluginRestResource;
import org.graylog2.rest.models.PaginatedResponse;
import org.graylog2.search.SearchQuery;
import org.graylog2.search.SearchQueryField;
import org.graylog2.search.SearchQueryParser;
import org.graylog2.shared.rest.resources.RestResource;
import org.graylog2.shared.security.RestPermissions;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.inject.Inject;
import javax.validation.Valid;
import javax.validation.constraints.NotEmpty;
import javax.ws.rs.BadRequestException;
import javax.ws.rs.DELETE;
import javax.ws.rs.DefaultValue;
import javax.ws.rs.GET;
import javax.ws.rs.NotFoundException;
import javax.ws.rs.POST;
import javax.ws.rs.PUT;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.MediaType;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import java.util.Set;
import static com.google.common.base.Strings.isNullOrEmpty;
import static java.util.Locale.ENGLISH;
@Api(value = "Views")
@Path("/views")
@Produces(MediaType.APPLICATION_JSON)
@RequiresAuthentication
public class ViewsResource extends RestResource implements PluginRestResource {
private static final Logger LOG = LoggerFactory.getLogger(ViewsResource.class);
private static final ImmutableMap SEARCH_FIELD_MAPPING = ImmutableMap.builder()
.put("id", SearchQueryField.create(ViewDTO.FIELD_ID))
.put("title", SearchQueryField.create(ViewDTO.FIELD_TITLE))
.put("summary", SearchQueryField.create(ViewDTO.FIELD_DESCRIPTION))
.build();
private final ViewService dbService;
private final SearchQueryParser searchQueryParser;
private final ViewSharingService viewSharingService;
private final IsViewSharedForUser isViewSharedForUser;
@Inject
public ViewsResource(ViewService dbService,
ViewSharingService viewSharingService,
IsViewSharedForUser isViewSharedForUser) {
this.dbService = dbService;
this.viewSharingService = viewSharingService;
this.isViewSharedForUser = isViewSharedForUser;
this.searchQueryParser = new SearchQueryParser(ViewDTO.FIELD_TITLE, SEARCH_FIELD_MAPPING);
}
@GET
@ApiOperation("Get a list of all views")
public PaginatedResponse views(@ApiParam(name = "page") @QueryParam("page") @DefaultValue("1") int page,
@ApiParam(name = "per_page") @QueryParam("per_page") @DefaultValue("50") int perPage,
@ApiParam(name = "sort",
value = "The field to sort the result on",
required = true,
allowableValues = "id,title,created_at") @DefaultValue(ViewDTO.FIELD_TITLE) @QueryParam("sort") String sortField,
@ApiParam(name = "order", value = "The sort direction", allowableValues = "asc, desc") @DefaultValue("asc") @QueryParam("order") String order,
@ApiParam(name = "query") @QueryParam("query") String query) {
if (!ViewDTO.SORT_FIELDS.contains(sortField.toLowerCase(ENGLISH))) {
sortField = ViewDTO.FIELD_TITLE;
}
try {
final SearchQuery searchQuery = searchQueryParser.parse(query);
final PaginatedList result = dbService.searchPaginated(
searchQuery,
view -> {
final Optional viewSharing = viewSharingService.forView(view.id());
return isPermitted(ViewsRestPermissions.VIEW_READ, view.id())
|| (view.type().equals(ViewDTO.Type.DASHBOARD) && isPermitted(RestPermissions.DASHBOARDS_READ, view.id()))
|| viewSharing.map(sharing -> isViewSharedForUser.isAllowedToSee(getCurrentUser(), sharing)).orElse(false);
},
order,
sortField,
page,
perPage);
return PaginatedResponse.create("views", result, query);
} catch (IllegalArgumentException e) {
throw new BadRequestException(e.getMessage(), e);
}
}
@GET
@Path("{id}")
@ApiOperation("Get a single view")
public ViewDTO get(@ApiParam(name="id") @PathParam("id") @NotEmpty String id) {
if ("default".equals(id)) {
// If the user is not permitted to access the default view, return a 404
return dbService.getDefault()
.filter(dto -> isPermitted(ViewsRestPermissions.VIEW_READ, dto.id()))
.orElseThrow(() -> new NotFoundException("Default view doesn't exist"));
}
final Optional viewSharing = viewSharingService.forView(id);
final ViewDTO view = loadView(id);
if (isPermitted(ViewsRestPermissions.VIEW_READ, id)
|| (view.type().equals(ViewDTO.Type.DASHBOARD) && isPermitted(RestPermissions.DASHBOARDS_READ, view.id()))
|| viewSharing.map(sharing -> isViewSharedForUser.isAllowedToSee(getCurrentUser(), sharing)).orElse(false)) {
return view;
}
throw viewNotFoundException(id);
}
@POST
@ApiOperation("Create a new view")
@AuditEvent(type = ViewsAuditEventTypes.VIEW_CREATE)
public ViewDTO create(@ApiParam @Valid ViewDTO dto) throws ValidationException {
final String username = getCurrentUser() == null ? null : getCurrentUser().getName();
final ViewDTO savedDto = dbService.save(dto.toBuilder().owner(username).build());
ensureUserPermissions(savedDto);
return savedDto;
}
@PUT
@Path("{id}")
@ApiOperation("Update view")
@AuditEvent(type = ViewsAuditEventTypes.VIEW_UPDATE)
public ViewDTO update(@ApiParam(name="id") @PathParam("id") @NotEmpty String id,
@ApiParam @Valid ViewDTO dto) {
if (dto.type().equals(ViewDTO.Type.DASHBOARD)) {
checkAnyPermission(new String[]{
ViewsRestPermissions.VIEW_EDIT,
RestPermissions.DASHBOARDS_EDIT
}, id);
} else {
checkPermission(ViewsRestPermissions.VIEW_EDIT, id);
}
loadView(id);
return dbService.update(dto.toBuilder().id(id).build());
}
@PUT
@Path("{id}/default")
@ApiOperation("Configures the view as default view")
@AuditEvent(type = ViewsAuditEventTypes.DEFAULT_VIEW_SET)
public void setDefault(@ApiParam(name="id") @PathParam("id") @NotEmpty String id) {
checkPermission(ViewsRestPermissions.VIEW_READ, id);
checkPermission(ViewsRestPermissions.DEFAULT_VIEW_SET);
dbService.saveDefault(loadView(id));
}
@DELETE
@Path("{id}")
@ApiOperation("Delete view")
@AuditEvent(type = ViewsAuditEventTypes.VIEW_DELETE)
public ViewDTO delete(@ApiParam(name="id") @PathParam("id") @NotEmpty String id) {
checkPermission(ViewsRestPermissions.VIEW_DELETE, id);
final ViewDTO dto = loadView(id);
dbService.delete(id);
removeUserPermissions(dto);
return dto;
}
private ViewDTO loadView(String id) {
try {
return dbService.get(id).orElseThrow(() -> viewNotFoundException(id));
} catch (IllegalArgumentException ignored) {
throw viewNotFoundException(id);
}
}
private NotFoundException viewNotFoundException(String id) {
return new NotFoundException("View " + id + " doesn't exist");
}
private void ensureUserPermissions(ViewDTO dto) throws ValidationException {
final User user = getCurrentUser();
if (user != null && !user.isLocalAdmin()) {
final List permissions = ImmutableList.builder()
.addAll(user.getPermissions())
.addAll(getViewPermissions(dto))
.build();
user.setPermissions(permissions);
userService.save(user);
}
}
// TODO: Should be moved to org.graylog2.users.UserPermissionsCleanupListener once view are merged into the server
private void removeUserPermissions(ViewDTO dto) {
userService.loadAll().forEach(user -> {
final List newPermissions = new ArrayList<>(user.getPermissions());
boolean modifiedPermissions = newPermissions.removeAll(getViewPermissions(dto));
if (modifiedPermissions) {
user.setPermissions(newPermissions);
try {
userService.save(user);
LOG.debug("Successfully updated permissions of user <{}>: {}", user.getName(), newPermissions);
} catch (ValidationException e) {
LOG.warn("Unable to save user <{}> while removing permissions of deleted dashboard: ", user.getName(), e);
}
}
});
}
private Set getViewPermissions(ViewDTO dto) {
if (isNullOrEmpty(dto.id())) {
throw new IllegalArgumentException("ViewDTO needs an ID to create permissions");
}
return ImmutableSet.of(
ViewsRestPermissions.VIEW_READ + ":" + dto.id(),
ViewsRestPermissions.VIEW_EDIT + ":" + dto.id()
);
}
}